matinal：python 上下文管理(with、contextmanager)

matinal

发布于 2023-10-14 14:16:09

3570

简介

python中存在这么一个非常好用的东西，能够帮助我们更好的管理上下文，这种东西就成为上下文管理器，例如我们最常见的with open，在读取文件时可以很方便的进行文件描述符和文件对象的打开和关闭，防止溢出，下面我们就介绍以下上下文管理器的实现的两个方式。

使用内置方法实现

实现__enter__和__exit__方法，注意的时__enter__返回的是self，即返回的对象，__exit__方法无论是否出现问题都会执行，类似与try-except中的finally。

class myContextManager:
    def __enter__(self):
        print("__enter__")
        return self

    # exec_val:异常种类， exec_val:异常值， exec_traceback：异常的堆栈
    def __exit__(self, exec_type, exec_val, exec_traceback):
        print(exec_val)
        print("-------------")
        print(exec_type)
        print("-------------")
        print(exec_traceback)
        print("__exit__")


with myContextManager() as f:
    raise IndexError

使用contextManager装饰器实现

使用contextManager装饰器需要注意，必须要使用yield，yield前类似于__enter__,yield返回的对象就是__enter__返回的对象，yield后便是类似于__exit__的方法

@contextmanager
def some_generator(<arguments>):
    <setup>
    try:
        yield <value>
    finally:
        <cleanup>

对于有些上下文在出现异常后需要主动关闭时，需要主动关闭上下文对象，使用到closing

with closing(<module>.open(<arguments>)) as f:
    <block>

总结

上下文管理器可以帮助我们做好管理，例如打开关闭数据库动作，

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2023-10-11，如有侵权请联系 cloudcommunity@tencent.com 删除

python

matinal：python 上下文管理(with、contextmanager)

matinal：python 上下文管理(with、contextmanager)

简介

使用内置方法实现

使用contextManager装饰器实现

总结

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐