基础知识回顾——上下文管理器

上下文管理机制

很多情况，当我们使用完一个资源后，我们需要手动的关闭掉它，比如操作文件，建立数据库连接等。但是，在使用资源的过程中，如果遇到异常，很可能错误被直接抛出，导致来不及关闭资源。所以在大部分时候，我们使用”try-finally”语句来确保资源会关闭。

try:
    f = open('test.txt', 'w+')
    print(f.closed)        #判断文件是否关闭
    f.write('Hello World!')
finally:
    f.close()
    print(f.closed)        #判断文件是否关闭

运行结果：

False
True    

Python语言里提供的with语句功能几乎同”try-finally”一样，代码更简洁。

with open("test.txt", "w") as f:
    print(f.closed)
    f.write("Hello World!")
print(f.closed)

运行结果：

False
True 

以上即是上下文管理器的用法，Python要求在with语句进入时自动调用__enter__()方法，其返回值会赋给as关键字后的变量；在with语句块退出后自动调用__exit__()方法。对于文件对象f来说，它定义了__enter__()和__exit__()方法(可以通过dir(f)看到)。在f的__exit__()方法中，有self.close()语句，这样就不用明文写f.close()了。

 

自定义上下文管理器

要自定义类似with语句的类，其内部必须提供两个内置方法__enter__以及__exit__。前者在主体代码执行前执行，后则在主体代码执行后执行。__enter__()返回的一对象作为as所指的变量。

class ContextManagerDemo(object):
    def __init__(self,text):
        self.text = text

    def __enter__(self):
        self.text = "I say: " + self.text
        #print self     <__main__.ContextManagerDemo object at 0x023557D0>
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):  
        self.text = self.text + "!"
#参数exc_type, exc_val, exc_tb分别代表异常类型，异常值，和异常的Traceback。当处理完异常后，可以让”__exit__()”方法返回True，此时该异常就不会再被抛出。正常程序中三个参数都是None

with ContextManagerDemo('hello') as f:      #上下文管理器会使用__enter__()返回的这一对象作为as所指的变量f，如果没有返回对象，text属性则无法调用
    print f.text

print f.text

运行结果：

I say: hello
I say: hello!

 

上下文管理器应用

1.管理多个文件

with open('data.txt') as source, open('target.txt', 'w') as target:
            target.write(source.read())

2.管理数据库游标

import pymysql 

def get_conn(**kwargs): 
　　return pymysql.connect(host=kwargs.get('host', 'localhost'),
　　　　　　　　　　　　　　　　 port=kwargs.get('port', 8080), 
　　　　　　　　　　　　　　　　 user=kwargs.get('user'), 
　　　　　　　　　　　　　　　　 passwd=kwargs.get('passwd')) 

def main(): 
　　conn = get_conn(user='Admin', passwd='123456') 
　　with conn as cur: 
　　　　 cur.execute('show databases')
 　　　　print cur.fetchall()

 if __name__ == '__main__':
 　　main()