怎么用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上下文管理器”的学习就结束了，希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习，快去试试吧！若想继续学习更多相关知识，请继续关注亿速云网站，小编会继续努力为大家带来更多实用的文章！