回答:
モジュールのmkdtemp()
関数を使用しtempfile
ます。
import tempfile
import shutil
dirpath = tempfile.mkdtemp()
# ... do stuff with dirpath
shutil.rmtree(dirpath)
with tempfile.TemporaryDirectory() as dirpath:
。一時ディレクトリは、コンテキストマネージャーの終了時に自動的にクリーンアップされます。docs.python.org/3.4/library/...
Python 3では、tempfileモジュールのTemporaryDirectoryを使用できます。
import tempfile
with tempfile.TemporaryDirectory() as tmpdirname:
print('created temporary directory', tmpdirname)
# directory and contents have been removed
ディレクトリをもう少し長くしたい場合は、次のようにすることができます(例ではありません)。
import tempfile
import shutil
temp_dir = tempfile.TemporaryDirectory()
print(temp_dir.name)
# use temp_dir, and when done:
shutil.rmtree(temp_dir.name)
@MatthiasRoelandtsが指摘したように、ドキュメントには「cleanup()
メソッドを呼び出すことによってディレクトリを明示的にクリーンアップできる」とも記載されています。
別の答えに拡張するために、例外があってもtmpdirをクリーンアップできるかなり完全な例を次に示します。
import contextlib
import os
import shutil
import tempfile
@contextlib.contextmanager
def cd(newdir, cleanup=lambda: True):
prevdir = os.getcwd()
os.chdir(os.path.expanduser(newdir))
try:
yield
finally:
os.chdir(prevdir)
cleanup()
@contextlib.contextmanager
def tempdir():
dirpath = tempfile.mkdtemp()
def cleanup():
shutil.rmtree(dirpath)
with cd(dirpath, cleanup):
yield dirpath
def main():
with tempdir() as dirpath:
pass # do something here
Python 3.2以降では、stdlib https://docs.python.org/3/library/tempfile.html#tempfile.TemporaryDirectoryに、このための便利なコンテキストマネージャがあります。