blob: ecc21c9397ce530e5c6be2656d295e9c564ae8d2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from __future__ import absolute_import
try:
# Python 3.2
from tempfile import TemporaryDirectory
except ImportError:
import tempfile
import shutil
from contextlib import contextmanager
@contextmanager
def TemporaryDirectory(*args, **kwds):
d = tempfile.mkdtemp(*args, **kwds)
try:
yield d
finally:
shutil.rmtree(d)
|
