web-dev-qa-db-fra.com

Comment créer un répertoire temporaire et obtenir le nom du chemin / fichier dans Python

comment créer un répertoire temporaire et obtenir le chemin/nom du fichier en python

pythontemporary-filestemporary-directory
114
duhhunjonn

Utilisez la fonction mkdtemp() du module tempfile :

import tempfile
import shutil

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

Pour développer une autre réponse, voici un exemple assez complet qui peut nettoyer le répertoire tmpdir, même en cas d’exceptions:

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
35
cdunn2001

Dans python 3.2 et versions ultérieures, il existe un gestionnaire de contexte utile dans stdlib https://docs.python.org/3/library/tempfile.html#tempfile.TemporaryDirectory

12
André Keller

Dans Python 3, TemporaryDirectory dans le module fichier temporaire peut être utilisé.

Ceci vient directement de exemples :

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

Si vous souhaitez conserver le répertoire un peu plus longtemps, vous pouvez faire quelque chose comme ceci (pas à partir de l'exemple):

import tempfile
import shutil

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