Jak utworzyć katalog tymczasowy i uzyskać ścieżkę / nazwę pliku w Pythonie

Odpowiedzi:

210

Skorzystaj z mkdtemp()funkcji z tempfilemodułu:

import tempfile
import shutil

dirpath = tempfile.mkdtemp()
# ... do stuff with dirpath
shutil.rmtree(dirpath)
Philipp
źródło
7
Jeśli użyjesz tego w teście, pamiętaj, aby usunąć (shutil.rmtree) katalog, ponieważ nie jest on automatycznie usuwany po użyciu. „Użytkownik mkdtemp () jest odpowiedzialny za usunięcie katalogu tymczasowego i jego zawartości po zakończeniu pracy”. Zobacz: docs.python.org/2/library/tempfile.html#tempfile.mkdtemp
Niels Bom
97
W python3 możesz to zrobić with tempfile.TemporaryDirectory() as dirpath:, a katalog tymczasowy zostanie automatycznie wyczyszczony po wyjściu z menedżera kontekstu. docs.python.org/3.4/library/…
Symmetric
41

W Pythonie 3 można użyć TemporaryDirectory w module tempfile .

Wynika to prosto z przykładów :

import tempfile
with tempfile.TemporaryDirectory() as tmpdirname:
     print('created temporary directory', tmpdirname)
# directory and contents have been removed

Jeśli chciałbyś zachować katalog trochę dłużej, możesz zrobić coś takiego (nie z przykładu):

import tempfile
import shutil

temp_dir = tempfile.TemporaryDirectory()
print(temp_dir.name)
# use temp_dir, and when done:
shutil.rmtree(temp_dir.name)

Jak zauważył @MatthiasRoelandts, dokumentacja mówi również, że „katalog można jawnie wyczyścić, wywołując cleanup()metodę”.

Nagev
źródło
2
shutil.rmtree (temp_dir.name) nie jest konieczne.
sidcha
37

Aby rozwinąć inną odpowiedź, oto dość kompletny przykład, który może wyczyścić tmpdir nawet w wyjątkach:

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
cdunn2001
źródło
3

Jeśli poprawnie odpowiem na Twoje pytanie, chcesz poznać także nazwy plików wygenerowanych w katalogu tymczasowym? Jeśli tak, spróbuj tego:

import os
import tempfile

with tempfile.TemporaryDirectory() as tmp_dir:
    # generate some random files in it
     files_in_dir = os.listdir(tmp_dir)
Shoval Sadde
źródło