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

这只是一个创建上下文管理器的方法,大家记住格式就行。

猜你喜欢

转载自blog.csdn.net/weixin_44786530/article/details/89737838