Coverage for airflow/utils/file.py : 50%

# -*- coding: utf-8 -*- 

# 

# Licensed under the Apache License, Version 2.0 (the "License"); 

# you may not use this file except in compliance with the License. 

# You may obtain a copy of the License at 

# 

# http://www.apache.org/licenses/LICENSE-2.0 

# 

# Unless required by applicable law or agreed to in writing, software 

# distributed under the License is distributed on an "AS IS" BASIS, 

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

# See the License for the specific language governing permissions and 

# limitations under the License. 

# 

from __future__ import absolute_import 

from __future__ import unicode_literals 

import errno 

import shutil 

from tempfile import mkdtemp 

from contextlib import contextmanager 

@contextmanager 

def TemporaryDirectory(suffix='', prefix=None, dir=None): 

name = mkdtemp(suffix=suffix, prefix=prefix, dir=dir) 

try: 

yield name 

finally: 

try: 

shutil.rmtree(name) 

except OSError as e: 

# ENOENT - no such file or directory 

if e.errno != errno.ENOENT: 

raise e