1.与装饰器的区别
上下文管理器是装饰器的近亲,装饰器用于包装函数,上下文管理器用于包装任意代码块.
上下文管理器最常用的场合--作为确保资源被正确清理的一种方式.
2.上下文管理器举例
打开文件-打开文件必须确保其能关闭,这就构成了一种上下文的关系
3.with语句
我们知道with语句打开文件会自动帮我们关闭文件,这个with语句就起到了上下文管理器的作用.下面会介绍原理.
4.__enter__和__exit__方法
with语句会调用对象的__enter__和__exit__方法,在上面的例子中open就是一个对象(Python中一切皆对象).__enter__方法的返回值会赋值给后面的变量.__enter__就是上文管理__exit__就是下文管理.
写一个自己的上下文管理器.
后面会解释__exit__方法中的后三个参数
5.with调用__enter__和__exit__的过程
使用with语句会先调用with后面类的__enter__方法,with语句块结束后会调用__exit__方法
6.处理异常
__exit__方法可以捕获包装代码块中的异常.
对__exit__函数中的参数解释一下:
exc_type:异常类型
exc_val:异常实例
exc_tb:回溯
如果没有异常则以上三个参数均为None
__exit__对异常的处理:
返回True终止异常
返回False传播异常
什么都不return:
output:
return True:
output:
return False:
output:
7.使用场景
关闭资源
比如连接数据库时必须要关闭数据库,像这种打开后必须关闭的操作可以用上下文管理器的操作.
Python连接MySQL数据库的上下文管理器例子:
output:
处理异常
见上条内容
8.contextlib.contextmanager装饰器
可以利用这个装饰器将一个函数变成上下文管理器
output:
divisionby zero
注意:被装饰的函数需要返回单个值(yield)
网友评论