Как создать временный каталог и получить путь / имя файла в Python

Ответы:

211

Используйте mkdtemp()функцию из tempfileмодуля:

import tempfile
import shutil

dirpath = tempfile.mkdtemp()
# ... do stuff with dirpath
shutil.rmtree(dirpath)
Philipp
источник
7
Если вы используете это в тесте, обязательно удалите (shutil.rmtree) каталог, потому что он не удаляется автоматически после использования. «Пользователь mkdtemp () отвечает за удаление временного каталога и его содержимого, когда с ним покончено». См .: docs.python.org/2/library/tempfile.html#tempfile.mkdtemp
Нильс Бом,
98
В python3 вы можете это сделать with tempfile.TemporaryDirectory() as dirpath:, и временный каталог будет автоматически очищен при выходе из диспетчера контекста. docs.python.org/3.4/library/…
симметричный
41

В Python 3 можно использовать TemporaryDirectory в модуле tempfile .

Это прямо из примеров :

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()метод».

Nagev
источник
2
shutil.rmtree (temp_dir.name) не требуется.
Сидча
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
cdunn2001
источник
3

Если я правильно понял ваш вопрос, вы также хотите знать имена файлов, сгенерированных во временном каталоге? Если так, попробуйте это:

import os
import tempfile

with tempfile.TemporaryDirectory() as tmp_dir:
    # generate some random files in it
     files_in_dir = os.listdir(tmp_dir)
Шовал Садде
источник