一時ディレクトリを作成し、Pythonでパス/ファイル名を取得する方法


回答:


210

モジュールのmkdtemp()関数を使用しtempfileます。

import tempfile
import shutil

dirpath = tempfile.mkdtemp()
# ... do stuff with dirpath
shutil.rmtree(dirpath)

7
これをテストで使用する場合、ディレクトリは使用後に自動的に削除されないため、必ず削除(shutil.rmtree)してください。「mkdtemp()のユーザーは、一時ディレクトリとその内容を削除したときにその内容を削除する必要があります。」参照:docs.python.org/2/library/tempfile.html#tempfile.mkdtemp
Niels Bom

97
python3では、を実行できますwith tempfile.TemporaryDirectory() as dirpath:。一時ディレクトリは、コンテキストマネージャーの終了時に自動的にクリーンアップされます。docs.python.org/3.4/library/...
対称

41

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()メソッドを呼び出すことによってディレクトリを明示的にクリーンアップできる」とも記載されています。


2
shutil.rmtree(temp_dir.name)は必要ありません。
sidcha

37

別の答えに拡張するために、例外があっても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


3

質問を正しく受け取ったら、一時ディレクトリ内に生成されたファイルの名前も知りたいですか?もしそうなら、これを試してください:

import os
import tempfile

with tempfile.TemporaryDirectory() as tmp_dir:
    # generate some random files in it
     files_in_dir = os.listdir(tmp_dir)
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.