python如何创建支持with语句的对象

小编给大家分享一下python如何创建支持with语句的对象，希望大家阅读完这篇文章之后都有所收获，下面让我们一起去探讨吧！

创建支持「with」语句的对象

举例而言，我们都知道如何使用「with」语句打开文件或获取锁，但是我们可以实现自己上下文表达式吗？是的，我们可以使用「enter」和「exit」来实现上下文管理协议:

class Connection:      def __init__(self):          ...      def __enter__(self):          # Initialize connection...      def __exit__(self, type, value, traceback):          # Close connection...  with Connection() as c:      # __enter__() executes      ...      # conn.__exit__() executes

这是在 Python 中最常见的实现上下文管理的方法，但是还有更简单的方法:

from contextlib import contextmanager  @contextmanager  def tag(name):      print(f"<{name}>")      yield      print(f"")  with tag("h2"):      print("This is Title.")

上面这段代码使用 contextmanager 的 manager 装饰器实现了内容管理协议。在进入 with 块时 tag 函数的第一部分（在 yield 之前的部分）就已经执行了，然后 with 块才被执行，最后执行 tag 函数的其余部分。

看完了这篇文章，相信你对"python如何创建支持with语句的对象"有了一定的了解，如果想了解更多相关知识，欢迎关注行业资讯频道，感谢各位的阅读！