怎么用Python实现with上下文管理器

怎么用Python实现with上下文管理器

这篇文章主要介绍“怎么用Python实现with上下文管理器”，在日常操作中，相信很多人在怎么用Python实现with上下文管理器问题上存在疑惑，小编查阅了各式资料，整理出简单好用的操作方法，希望对大家解答”怎么用Python实现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('langzi') as a:
print('花了600块钱')

返回结果：

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

使用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万分谨慎的收起了钱包

到此，关于“怎么用Python实现with上下文管理器”的学习就结束了，希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习，快去试试吧！若想继续学习更多相关知识，请继续关注恰卡编程网网站，小编会继续努力为大家带来更多实用的文章！