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