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外的语句不运行方法的效果

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值