python 实现with语句

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 语句 一样。

上下文管理器协议包含 enterexit 两个方法。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外的语句不运行方法的效果

你可能感兴趣的:(Python,python)