Quelles sont les méthodes __enter__ et __exit__ en Python et comment sont-elles utilisées?

Les méthodes __enter__ et __exit__ sont des méthodes du protocole des gestionnaires de contexte en Python. Elles sont utilisées pour définir un objet pouvant être utilisé dans la construction with pour gérer des ressources (par exemple, des fichiers, des connexions réseau).

__enter__ est appelée lors de l'entrée dans le bloc with. Elle doit retourner la ressource qui sera accessible à l'intérieur du bloc sous le nom de la variable as.

__exit__ est appelée lors de la sortie du bloc with, indépendamment de si une exception s'est produite. Elle prend trois arguments : le type d'exception, la valeur de l'exception et l'objet traceback si une exception a été levée. Si aucune exception n'a eu lieu, tous les arguments seront None. Si __exit__ retourne True, l'exception est supprimée ; sinon, elle est propagée.

Exemple d'utilisation :

# Définition d'une classe avec gestionnaire de contexte
class MyResource:
    def __enter__(self):
        # Actions à l'entrée dans le contexte
        print("Entrée dans le contexte de la ressource.")
        return self  # Retourne l'objet lui-même

    def __exit__(self, exc_type, exc_val, exc_tb):
        # Actions à la sortie du contexte
        print("Sortie du contexte de la ressource.")
        if exc_type is not None:
            print(f"Une exception est survenue : {exc_type.__name__}")
        # Si retourner True, l'exception sera supprimée
        return False # Propagation de l'exception si elle existe

# Utilisation dans une construction with
with MyResource() as resource:
    print("Nous sommes dans le bloc with.")
    # Peut-être ici, traitement avec la ressource 'resource'
    # raise ValueError("Quelque chose a mal tourné!") # Décommenter pour tester l'exception

Exemple courant — travail avec des fichiers :

# Le fichier est automatiquement fermé
with open("my_file.txt", "w") as file:
    file.write("Bonjour, le monde!")

Cela garantit une libération correcte de la ressource, même si une exception se produit à l'intérieur du bloc with.