如何理解上下文管理器, 那就是有先 有后, 中间是要处理的东西。 举个例子: 别墅独立小院, 是不是有个大门。每天进大门然后出大门(有先有后),中间呢 是你在这个小院里操作各种内容,比如打水。
上下文管理器指符合上下文协议的一个类,类中实现了__enter__方法 和 exit 方法。 在一段代码执行前, 先执行__enter__中的代码用于预处理,然后执行 with 中间的代码,最后执行完这段代码之后再执行 exit 中的代码进行清理工作。 常见的如: 文件的操作:打开文件,关闭文件 打开数据库,关闭数据库 入栈,出栈 都可以用上下文管理器来完成。 具体可以参考官网里这块的代码实现逻辑来加深理解。
上下文管理器格式:
class A():
def __enter__(self):
print('连接')
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print('断开连接')
if(exc_tb):
print('出现异常')
else:
print('无异常')
return True
__enter__方法实现资源的连接过程 __exit__方式实现了资源的断开过程
with语句基本结构
with A() as b:
-------------------------
with A():
在with语句中的操作代码执行前,先执行__enter__中的代码;操作代码执行完后,再执行__exit__中的代码 enter=>with=>exit
这段代码里需要理解一下as b 在 with A() as b 中,b的值为__enter__方法的返回值,一般情况下会写self即返回上下文管理器对象本身,但是也可以返回其他内容 __exit__方法中的参数除了self外还必须有exc_type,exc_val,exc_tb;当操作代码部分没有出现异常时,三个参数的值都为空,当操作代码部分出现异常时分别传入异常的类型、信息、错误堆栈 __exit__的返回值可以是True或False,当返回True时若操作代码出现异常则在with语句外将不会给出异常(在__exit__中处理异常),当返回False时若操作代码出现异常则在with语句外给出异常(由外部处理异常)
举个数据库的简单例子
class DataBase():
def query(self):
print('查询数据库')
def __enter__(self):
print('连接数据库')
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print('断开数据库连接')
if(exc_tb):
print('出现异常')
else:
print('无异常')
return True
with DataBase() as db:
db.query()
|