jak utworzyć katalog tymczasowy i uzyskać ścieżkę / nazwę pliku w pythonie
jak utworzyć katalog tymczasowy i uzyskać ścieżkę / nazwę pliku w pythonie
Odpowiedzi:
Skorzystaj z mkdtemp()
funkcji z tempfile
modułu:
import tempfile
import shutil
dirpath = tempfile.mkdtemp()
# ... do stuff with dirpath
shutil.rmtree(dirpath)
with tempfile.TemporaryDirectory() as dirpath:
, a katalog tymczasowy zostanie automatycznie wyczyszczony po wyjściu z menedżera kontekstu. docs.python.org/3.4/library/…
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ę”.
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
W Pythonie 3.2 i nowszych istnieje do tego przydatny menedżer kontekstu w stdlib https://docs.python.org/3/library/tempfile.html#tempfile.TemporaryDirectory
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)