with 语句的目的是简化 try/finally 模式。这种模式用于保证一段代码运行完毕后执行某 项操作,即便那段代码由于异常、return 语句或 sys.exit() 调用而中止,也会执行指定的 操作。finally 子句中的代码通常用于释放重要的资源,或者还原临时变更的状态。
最常见的例子是确保关闭文件对象:
with open('mirror.py') as fp:
src = fp.read(60)
print(len(src)) # 60
print(fp) #<_io.TextIOWrapper name='mirror.py' mode='r' encoding='UTF-8'>
print(fp.closed, fp.encoding) #(True, utf-8)
但并不是所有的函数都能使用with语句的,对于一些我们自己实现的类方法,也想通过with语句实现该怎么做呢?这里就涉及到了上下文管理器。
上下文管理器对象存在的目的是管理 with 语句,就像迭代器的存在是为了管理 for 语句 一样。
上下文管理器协议包含 enter 和 exit 两个方法。with 语句开始运行时,会在上下 文管理器对象上调用 enter 方法。with 语句运行结束后,会在上下文管理器对象上调 用 exit 方法,以此扮演 finally 子句的角色。
因此,只要在我们的类中自定义实现__enter__方法和__exit__两个方法,就能实现with语句
class MyMethod(object):
def __init__(self, name, age):
self._name = name
self._age = age
def __enter__(self):
self._name.start_something()
def __exit__(self):
self._age.end_something()
#-----------------------
with MyMethod(name, age):
print("text1") # 录屏
print("text2") # 不录屏
MyMethod类实现了with语句,达到在with中的语句运行某些方法,在with外的语句不运行方法的效果