with 上下文管理器了解一下

with 上下文管理器,这个用过很多次,比如文本内容的读写,打开文本写入内容关闭文本,又或者是在执行 mysql 语句的时候要先链接数据库,获取游标,执行 sql 语句,关闭连接。
如果每次执行一条语句都要做这么多操作,就会产生大量重复的代码,这个时候使用上下文管理器即可美观又轻松的解决这个问题。
with 管理上下文的作用是对一些重复的代码简单化,并且能优化 try/except/finally 的写法。
上下文的实现是通过两个魔法函数 enter 和 exit 实现,后来更新再造后,使用 contextlib 提供的 API 可以更加方便的完成。
下面看看如何实现吧。
举个例子:夏目去买猫饼干,每次买的时候都要做这些动作,掏出钱包,花出 x 元,收回钱包。每次都要重复掏钱包收钱包这个动作有些麻烦,并且万一忘了掏钱包就不能付钱,忘了收钱包的话,钱包就掉了。每次掏钱包收钱包都要写代码,有啥简介的方式嘛?

使用 enter 和 exit 实现 (1)

class wallet(object):
def __init__(self,man):
self.man=man

def __enter__(self):
print(self.man + '放心大胆的掏出了钱包')

def __exit__(self, exc_type, exc_val, exc_tb):
print(self.man + '小心翼翼的收起来钱包')

def use_money(man):
return wallet(man)

with use_money('夏目) as a:
print('花了600块钱')

返回结果:

夏目放心大胆的掏出了钱包
花了600块钱
夏目小心翼翼的收起来钱包

使用 enter 和 exit 实现 (2)

当然也可以这么写:

class wallet(object):
def __init__(self,man):
self.man=man

def __enter__(self):
print(self.man + '放心大胆的掏出了钱包')
return self
# return self 这一步非常重要,作用是返回实例

def __exit__(self, exc_type, exc_val, exc_tb):
print(self.man + '小心翼翼的收起来钱包')

def use_money(self,money):
print(self.man + '花了' + money +'元')

with wallet('浪子')as a:
a.use_money('600')

运行结果:

浪子放心大胆的掏出了钱包
浪子花了600元
浪子小心翼翼的收起来钱包

使用 contextlib 实现

import contextlib

@contextlib.contextmanager
def use_money(man):
try:
print(man + '十分放心大胆的掏出钱包')
yield  None
# yield生成器,运行到这里会返回一个值(你随便写一个就行)
finally:
print(man + '万分谨慎的收起了钱包')

with use_money('langzi')as a:
print('花了1块钱')

返回结果:

langzi十分放心大胆的掏出钱包
花了1块钱
langzi万分谨慎的收起了钱包

这只是一个创建上下文管理器的方法,大家记住格式就行.
以上有不足的地方请提出来以便改正,谢谢!