Python标准模块--ContextManager上下文管理器的具体用法

写代码时，我们希望把一些操作放到一个代码块中，这样在代码块中执行时就可以保持在某种运行状态，而当离开该代码块时就执行另一个操作，结束当前状态；所以，简单来说，上下文管理器的目的就是规定对象的使用范围，如果超出范围就采取“处理”。

这一功能是在Python2.5之后引进的，它的优势在于可以使得你的代码更具可读性，且不容易出错。

1 模块简介

在数年前，Python 2.5 加入了一个非常特殊的关键字，就是with。with语句允许开发者创建上下文管理器。什么是上下文管理器？上下文管理器就是允许你可以自动地开始和结束一些事情。例如，你可能想要打开一个文件，然后写入一些内容，最后再关闭文件。这或许就是上下文管理器中一个最经典的示例。事实上，当你利用with语句打开一个文件时，Python替你自动创建了一个上下文管理器。

with open("test/test.txt","w") as f_obj:  f_obj.write("hello")

如果你使用的是Python 2.4，你不得不以一种老的方式来完成这个任务。

f_obj = open("test/test.txt","w")f_obj.write("hello")f_obj.close()

上下文管理器背后工作的机制是使用Python的方法：__enter__和__exit__。让我们尝试着去创建我们的上下文管理器，以此来了解上下文管理器是如何工作的。

2 模块使用

2.1 创建一个上下文管理器类

与其继续使用Python打开文件这个例子，不如我们创建一个上下文管理器，这个上下文管理器将会创建一个SQLite数据库连接，当任务处理完毕，将会将其关闭。下面就是一个简单的示例。

import sqlite3class DataConn:  def __init__(self,db_name):    self.db_name = db_name  def __enter__(self):    self.conn = sqlite3.connect(self.db_name)    return self.conn  def __exit__(self,exc_type,exc_val,exc_tb):    self.conn.close()    if exc_val:      raiseif __name__ == "__main__":  db = "test/test.db"  with DataConn(db) as conn:    cursor = conn.cursor()

在上述代码中，我们创建了一个类，获取到SQLite数据库文件的路径。__enter__方法将会自动执行，并返回数据库连接对象。现在我们已经获取到数据库连接对象，然后我们创建光标，向数据库写入数据或者对数据库进行查询。当我们退出with语句的时候，它将会调用__exit__方法用于执行和关闭这个连接。

让我们使用其它的方法来创建上下文管理器。

2.2 利用contextlib创建一个上下文管理器

Python 2.5 不仅仅添加了with语句，它也添加了contextlib模块。这就允许我们使用contextlib的contextmanager函数作为装饰器，来创建一个上下文管理器。让我们尝试着用它来创建一个上下文管理器，用于打开和关闭文件。