python with 语句妙用
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python with 语句妙用相关的知识,希望对你有一定的参考价值。
class aa(): def bb(self): print("hhhh") return "hello world" def __enter__(self): # 必须有__enter__ print("enter") return self def cc(self): print("www") def __exit__(self, exc_type, exc_val, exc_tb): # 必须有结束__exit__ print("exit") def ll(): with aa() as a: return a.bb() res = ll() print(res)
结果:
enter
hhhh
exit
hello world
原理:自定义上下文管理器
开发人员可以自定义支持上下文管理协议的类。自定义的上下文管理器要实现上下文管理协议所需要的 __enter__() 和 __exit__() 两个方法: context_manager.__enter__() :进入上下文管理器的运行时上下文,在语句体执行前调用。with 语句将该方法的返回值赋值给 as 子句中的 target,如果指定了 as 子句的话 context_manager.__exit__(exc_type, exc_value, exc_traceback) :退出与上下文管理器相关的运行时上下文,返回一个布尔值表示是否对发生的异常进行处理。
参数表示引起退出操作的异常,如果退出时没有发生异常,则3个参数都为None。如果发生异常,返回 True 表示不处理异常,否则会在退出该方法后重新抛出异常以由 with 语句之外的代码逻辑进行处理。如果该方法内部产生异常,则会取代由 statement-body 中语句产生的异常。
要处理异常时,不要显示重新抛出异常,即不能重新抛出通过参数传递进来的异常,只需要将返回值设置为 False 就可以了。之后,上下文管理代码会检测是否 __exit__() 失败来处理异常
以上是关于python with 语句妙用的主要内容,如果未能解决你的问题,请参考以下文章