欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Python with语句 处理上下文 + contextmanager装饰器(1ni)

程序员文章站 2024-02-20 11:42:16
...
# *_*coding:utf-8 *_*

"""
with 用来处理上下文
比如数据库的 连接和关闭

定义的类里面需要定义 __enter__ 和 __exit__方法
"""


class MyResource:
    def __enter__(self):
        print('connect to resource')
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('close resource connection')
        return True

    def query(self):
        print('query data')


with MyResource() as resource:
    resource.query()

1.上面是一个最基础的 with语句

这个 with XXX as 后面是表示的一个对象,也就是我们 __enter__ return 的对象,返回了直接,所以我们在下面的代码里是可以调用我们的 query方法的。

2.__exit__方法的重要性

除了self参数,还需要传入 exc_type,exc_val,exc_tb参数,我们运行上面的代码是空的,原因很简单,因为我们没有wrong 即我们没有抛出异常

下面我们改写一下代码 如下:

# *_*coding:utf-8 *_*

"""
with 用来处理上下文
比如数据库的 连接和关闭

定义的类里面需要定义 __enter__ 和 __exit__方法
"""


class MyResource:
    def __enter__(self):
        print('connect to resource')
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('close resource connection')
        return True

    def query(self):
        print('query data')


with MyResource() as resource:
    print(1/0)
    resource.query()

就是加了 1/0 我们当然知道这是错误,这时候我们 断点调试一下

可以看到如下的信息:

Python with语句 处理上下文 + contextmanager装饰器(1ni)Python with语句 处理上下文 + contextmanager装饰器(1ni)

这两个报错肯定看得懂。

但是假如我们运行第一个代码,也就是没有错误的代码,这些值是空的,因为根本没有错。

还有就是__exit__方法的 return 值了

只能为 True or False

我个人理解是  with语句 是否正确的退出了,简而言之就是 with里面的逻辑是否正确

这就换来了两种思考

1:return True

就是告诉python 我们在__exit__里面已经处理好异常了,外面就不必处理了

2:return False

就是我们并没有处理异常,你在外面处理吧

我们可以用 try 和 catch来测试

# *_*coding:utf-8 *_*

"""
with 用来处理上下文
比如数据库的 连接和关闭

定义的类里面需要定义 __enter__ 和 __exit__方法
"""


class MyResource:
    def __enter__(self):
        print('connect to resource')
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('close resource connection')
        return False

    def query(self):
        print('query data')


try:
    with MyResource() as resource:
        print(1/0)
        resource.query()
except Exception as ex:
    print('来了')
    raise ex

这个return 也可以是空的 ,就是None 其实 None在这里也是 False的意思,就是交给外部处理了

 

总结:

具体流程就是 我们在进入的时候 先调用 __enter__ 方法 ,在退出的时候 调用 __exit__方法,具体异常就看代码逻辑

我们 用 with 处理文件就是这个道理

下面我们用contextmanager来处理我们的上下文(在某种程度上 '简化了代码' )

并且使某些类变成上下文管理器,比如框架提供给我们的类

# *_*coding:utf-8 *_*


class MyResource:


    def query(self):
        print('query data')


"""
装饰器
"""
from contextlib import contextmanager


@contextmanager
def make_myresource():
    print('connect to resource')
    yield MyResource()
    print('close resource connection')


with make_myresource() as r:
    r.query()

 

相关标签: with