Como criar um diretório temporário e obter o caminho / nome do arquivo em Python

Respostas:

210

Use a mkdtemp()função do tempfilemódulo:

import tempfile
import shutil

dirpath = tempfile.mkdtemp()
# ... do stuff with dirpath
shutil.rmtree(dirpath)
Philipp
fonte
7
Se você usar isso em um teste, certifique-se de remover (shutil.rmtree) o diretório, pois ele não é excluído automaticamente após o uso. "O usuário do mkdtemp () é responsável por excluir o diretório temporário e seu conteúdo quando terminar com ele." Veja: docs.python.org/2/library/tempfile.html#tempfile.mkdtemp
Niels Bom
97
No python3, você pode fazer with tempfile.TemporaryDirectory() as dirpath:, e o diretório temporário será limpo automaticamente ao sair do gerenciador de contexto. docs.python.org/3.4/library/…
Symmetric
41

No Python 3, o TemporaryDirectory no módulo tempfile pode ser usado.

Isso é direto dos exemplos :

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

Se você deseja manter o diretório um pouco mais, algo como isso pode ser feito (não no exemplo):

import tempfile
import shutil

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

Como o @MatthiasRoelandts apontou, a documentação também diz que "o diretório pode ser explicitamente limpo chamando o cleanup()método".

Nagev
fonte
2
shutil.rmtree (temp_dir.name) não é necessário.
sidcha
37

Para expandir para outra resposta, aqui está um exemplo bastante completo que pode limpar o tmpdir mesmo com exceções:

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
fonte
3

Se eu receber sua pergunta corretamente, você também deseja saber os nomes dos arquivos gerados dentro do diretório temporário? Se sim, tente o seguinte:

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
fonte