Comment supprimer un répertoire créé avec tempfile.mkdtemp?

De gestion des ressources (comme des fichiers) en Python, la meilleure pratique consiste à utiliser la avec mot-clé, qui libère automatiquement les ressources (c'est à dire, nettoie, comme la fermeture des fichiers); il est disponible à partir de Python 2.5.

De Python 3.2, vous pouvez utiliser tempfile.TemporaryDirectory() au lieu de tempfile.mkdtmp() – c'est utilisable dans with et nettoie automatiquement le répertoire:

from tempfile import TemporaryDirectory

with TemporaryDirectory() as temp_dir:
    # ... do something with temp_dir
# automatically cleaned up when context exited

Si vous utilisez une version antérieure de Python (au moins 2,5, ont donc with), vous pouvez utiliser backports.tempfile; voir Nicolas, Évêque de réponse à tempfile.TemporaryDirectory gestionnaire de contexte dans Python 2.7.

C'est facile et instructif de rouler votre propre classe, appelée
gestionnaire de contexte. La valeur de retour de la __enter__() méthode est liée à la cible de la as clause, alors que le __exit__() méthode est appelée lorsque le contexte est sorti – même par exception, et effectue le nettoyage.

import shutil
import tempfile

class TemporaryDirectory(object):
    """Context manager for tempfile.mkdtemp() so it's usable with "with" statement."""
    def __enter__(self):
        self.name = tempfile.mkdtemp()
        return self.name

    def __exit__(self, exc_type, exc_value, traceback):
        shutil.rmtree(self.name)

Vous pouvez simplifier cette avec la @contextlib.contextmanager décorateur, de sorte que vous n'avez pas besoin d'écrire un gestionnaire de contexte manuellement. Le code avant la yield est exécutée lors de la saisie du contexte, de la donné valeur est liée à la cible de l' as, et le code après le yield est exécutée lorsque l'on sort du contexte. C'est fondamentalement un coroutine qui encapsule l'acquisition des ressources et de la libération, avec le yield cession du contrôle de la suite (corps) de la with clause. Notez que vous ne besoin d'avoir un try...finally bloc, comme @contextlib.contextmanager ne pas intercepter des exceptions dans le yield – ce juste les facteurs de la gestion des ressources dans une coroutine.

from contextlib import contextmanager
import tempfile
import shutil

@contextmanager
def TemporaryDirectory():
    name = tempfile.mkdtemp()
    try:
        yield name
    finally:
        shutil.rmtree(name)

Comme simplylizz des notes, si vous n'avez pas l'esprit de l'annuaire déjà été supprimés (dont le code ci-dessus suppose ne pas être le cas), vous pouvez prendre le “No such file or directory” exception comme suit:

import errno
# ...
try:
    shutil.rmtree(self.name)
except OSError as e:
    # Reraise unless ENOENT: No such file or directory
    # (ok if directory has already been deleted)
    if e.errno != errno.ENOENT:
        raise

Vous pouvez comparer avec le niveau de mise en œuvre dans tempfile.py; même ce simple de la classe a eu des bugs et a évolué au fil des ans.

De fond sur with, voir:

• Le Tutoriel Python: Méthodes des Objets de Fichier
• Avec L'Énoncé De Contexte Gestionnaires
• PEP 343 -- La "avec la" Déclaration de

J'ai eu le même problème à l'aide de TemporaryDirectory() qui est essentiellement couvrant les fonctionnalités dont vous avez défini ci-dessus.

Mon problème était dû à l'utilisation du répertoire temporaire. J'ai utilisé pour remplir le contenu par clonage d'un dépôt git, il est arrivé qu'il y avait en lecture seule les fichiers créés au cours du processus et avec une sortie normale de ces en lecture seule les fichiers temporaires étaient à l'origine de la complète répertoire temporaire d'y rester.

N'héritent de la TemporaryDirectory à ma propre classe et de ne remplacer la méthode de la classe _cleanup avec le code suivant.

Le code avant de super() peut être optimisé, mais pour moi, la performance n'était pas le problème.

J'ai fait usage de la force et de lire la source de "tempfile"

import tempfile
import shutil
import stat 

class myTempDir(tempfile.TemporaryDirectory):
    @classmethod
    def _cleanup(self,name, warn_message):
        for root, dirs, files in os.walk(name):
            for fname in files:
                full_path = os.path.join(root, fname)
                os.chmod(full_path ,stat.S_IWRITE)
        super()

Solution a fonctionné pour Windows 10 avec Python 3