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 我们当然知道这是错误,这时候我们 断点调试一下
可以看到如下的信息:
这两个报错肯定看得懂。
但是假如我们运行第一个代码,也就是没有错误的代码,这些值是空的,因为根本没有错。
还有就是__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()