¿Cómo creo un directorio temporal en Python?

¿Cómo puedo crear un directorio temporal y obtener su ruta en Python?

En Python 3, TemporaryDirectory desde el tempfile Se puede utilizar el módulo.

Desde el ejemplos:

import tempfile

with tempfile.TemporaryDirectory() as tmpdirname:
     print('created temporary directory', tmpdirname)

# directory and contents have been removed

Para controlar manualmente cuándo se elimina el directorio, no utilice un administrador de contexto, como en el siguiente ejemplo:

import tempfile

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

La documentación también dice:

Al finalizar el contexto o la destrucción del objeto de directorio temporal, el directorio temporal recién creado y todo su contenido se eliminan del sistema de archivos.

Al final del programa, por ejemplo, Python limpiará el directorio si no fue eliminado, por ejemplo, por el administrador de contexto o el cleanup() método. de pitón unittest puede quejarse de ResourceWarning: Implicitly cleaning up <TemporaryDirectory... si confías en esto, sin embargo.

Utilizar el mkdtemp() función de la tempfile módulo:

import tempfile
import shutil

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

Para ampliar otra respuesta, aquí hay un ejemplo bastante completo que puede limpiar el tmpdir incluso en excepciones:

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

En python 3.2 y versiones posteriores, hay un administrador de contexto útil para esto en stdlib https://docs.python.org/3/library/tempfile.html#tempfile.TemporaryDirectory

Los documentos sugieren usar el TemporaryDirectory administrador de contexto, que crea un directorio temporal y lo elimina automáticamente al salir del administrador de contexto:

import tempfile

with tempfile.TemporaryDirectory() as tmpdirname:
    print('created temporary directory', tmpdirname)

# Outside the context manager, directory and contents have been removed.

Usando pathlib para facilitar la manipulación de rutas además de tempfile permite crear nuevos caminos con la / operador de ruta de pathlib:

import tempfile
from pathlib import Path

with tempfile.TemporaryDirectory() as tmpdirname:
    temp_dir = Path(tmpdirname)
    file_name = temp_dir / "test.txt"
    file_name.write_text("bla bla bla")

    print(temp_dir, temp_dir.exists())
    # /tmp/tmp81iox6s2 True

    print(file_name, "contains", file_name.open().read())
    # /tmp/tmp81iox6s2/test.txt contains bla bla bla

Fuera del administrador de contexto, los archivos han sido destruidos

print(temp_dir, temp_dir.exists())
# /tmp/tmp81iox6s2 False

print(file_name, file_name.exists())
# /tmp/tmp81iox6s2/test.txt False

Si respondo correctamente su pregunta, ¿también desea saber los nombres de los archivos generados dentro del directorio temporal? Si es así, prueba esto:

import os
import tempfile

with tempfile.TemporaryDirectory() as tmp_dir:
    # generate some random files in it
     files_in_dir = os.listdir(tmp_dir)