Explication détaillée du mot-clé "with" et gestionnaire de contexte en python

Cet article présente principalement les informations pertinentes sur le mot-clé « with » et le gestionnaire de contexte en Python. L'introduction dans l'article est très détaillée. Je pense qu'elle a une certaine valeur de référence pour que tous les amis qui en ont besoin apprennent ou l'utilisent. on peut lire ci-dessous. Jetons un coup d'oeil.

Avant-propos

Si vous avez l'habitude de lire le code source, vous constaterez peut-être que certains excellents codes apparaissent souvent avec le mot clé "with" déclaration, dans quels scénarios est-il habituellement utilisé ? Aujourd’hui, parlons avec et des gestionnaires de contexte.

Pour les ressources système telles que les fichiers, les connexions à la base de données et les sockets, une fois que l'application a ouvert ces ressources et exécuté la logique métier, une chose qu'elle doit faire est de fermer (déconnecter) la ressource.

Par exemple, un programme Python ouvre un fichier et écrit le contenu dans le fichier. Après l'écriture, le fichier doit être fermé. Sinon, que se passera-t-il ? Dans des cas extrêmes, des erreurs « Trop de fichiers ouverts » peuvent survenir car le nombre maximum de fichiers que le système vous permet d'ouvrir est limité.

De même, pour la base de données, s'il y a trop de connexions et qu'elles ne sont pas fermées à temps, "Impossible de se connecter au serveur MySQL Trop de connexions" peut apparaître, car les connexions à la base de données sont une ressource très coûteuse et ne peuvent pas être fermé dans le temps. Peut être créé sans limite.

Voyons comment fermer correctement un fichier.

Version normale :

def m1():
 f = open("output.txt", "w")
 f.write("python之禅")
 f.close()

Il y a un problème potentiel lors de l'écriture comme celui-ci. Si une exception se produit lors de l'appel à l'écriture et que le code suivant ne peut pas continuer à s'exécuter, le méthode close Elle ne peut pas être appelée normalement, donc les ressources seront toujours libérées par l'occupant du programme. Alors, comment pouvons-nous améliorer le code ?

Version avancée :

def m2():
 f = open("output.txt", "w")
 try:
 f.write("python之禅")
 except IOError:
 print("oops error")
 finally:
 f.close()

La version améliorée du programme consiste à essayer de capturer le code là où des exceptions peuvent survenir, en utilisant l'instruction try/finally, ce qui signifie si dans le code try block Si une exception se produit dans le programme, le code suivant ne sera plus exécuté et passera directement au bloc de code except. Quoi qu’il en soit, le code du bloc final finira par être exécuté. Par conséquent, tant que close est placé dans le code final, le fichier sera définitivement fermé.

Version avancée :

def m3():
 with open("output.txt", "w") as f:
 f.write("Python之禅")

Une manière plus concise et élégante consiste à utiliser le mot-clé with. La valeur de retour de la méthode open est attribuée à la variable f. Lorsque vous quittez le bloc de code with, le système appelle automatiquement la méthode f.close() . La fonction de with est la même que celle de l'instruction try/finally. Alors quel est son principe de mise en œuvre ?

Avant de parler du principe du with, il convient de mentionner un autre concept, qui est le gestionnaire de contexte.

Gestionnaire de contexte

Tout objet qui implémente les méthodes __enter__() et __exit__() peut être appelé un gestionnaire de contexte. utilisez le mot-clé with. Évidemment, les objets fichier implémentent également des gestionnaires de contexte.

Alors, comment l'objet fichier implémente-t-il ces deux méthodes ? Nous pouvons simuler l'implémentation de notre propre classe de fichiers et laisser la classe implémenter les méthodes __enter__() et __exit__() . La méthode

class File():

 def __init__(self, filename, mode):
 self.filename = filename
 self.mode = mode

 def __enter__(self):
 print("entering")
 self.f = open(self.filename, self.mode)
 return self.f

 def __exit__(self, *args):
 print("will exit")
 self.f.close()

__enter__() renvoie l'objet ressource, qui est l'objet fichier que vous êtes sur le point d'ouvrir. La méthode __exit__() gère un travail de nettoyage.

Étant donné que la classe File implémente un gestionnaire de contexte, vous pouvez désormais utiliser l'instruction with.

with File('out.txt', 'w') as f:
 print("writing")
 f.write('hello, python')

De cette façon, vous n'avez pas besoin d'appeler explicitement la méthode close. Le système l'appellera automatiquement, même si une exception est rencontrée au milieu, la méthode close sera appelée.

contextlib

Python fournit également un décorateur contextmanager, qui simplifie encore la mise en œuvre du gestionnaire de contexte. La fonction est divisée en deux parties par rendement. L'instruction avant rendement est exécutée dans la méthode __enter__ et l'instruction après rendement est exécutée dans la méthode __exit__ . La valeur immédiatement après rendement est la valeur de retour de la fonction.

from contextlib import contextmanager

@contextmanager
def my_open(path, mode):
 f = open(path, mode)
 yield f
 f.close()

Appel

with my_open('out.txt', 'w') as f:
 f.write("hello , the simplest context manager")

Résumé

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!