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

# *_*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 我们当然知道这是错误，这时候我们 断点调试一下

可以看到如下的信息：

这两个报错肯定看得懂。

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

还有就是__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()