python 上下文管理器使用方法小结
上下文管理器最常用的是确保正确关闭文件，
withopen('/path/to/file','r')asf:
f.read()
with语句的基本语法，
withexpression[asvariable]:
with-block
expression是一个上下文管理器，其实现了enter和exit两个函数。当我们调用一个with语句时，依次执行一下步骤，
1.首先生成一个上下文管理器expression，比如open('xx.txt').
2.执行expression.enter()。如果指定了[asvariable]说明符，将enter()的返回值赋给variable.
3.执行with-block语句块.
4.执行expression.exit(),在exit()函数中可以进行资源清理工作.
with语句不仅可以管理文件，还可以管理锁、连接等等，如，
#管理锁 importthreading lock=threading.lock() withlock: #执行一些操作 pass #数据库连接管理 deftest_write(): sql="""#具体的sql语句 """ con=DBConnection() withconascursor: cursor.execute(sql) cursor.execute(sql) cursor.execute(sql)
自定义上下文管理器
上下文管理器就是实现了上下文协议的类,也就是要实现__enter__()和__exit__()两个方法。
__enter__()：主要执行一些环境准备工作，同时返回一资源对象。如果上下文管理器open("test.txt")的enter()函数返回一个文件对象。
__exit__()：完整形式为exit(type,value,traceback),这三个参数和调用sys.exec_info()函数返回值是一样的，分别为异常类型、异常信息和堆栈。如果执行体语句没有引发异常，则这三个参数均被设为None。否则，它们将包含上下文的异常信息。_exit()方法返回True或False,分别指示被引发的异常有没有被处理，如果返回False，引发的异常将会被传递出上下文。如果exit()函数内部引发了异常，则会覆盖掉执行体的中引发的异常。处理异常时，不需要重新抛出异常，只需要返回False，with语句会检测exit()返回False来处理异常。
classtest_query:
def__init__(self):
pass
defquery(self):
print('query')
def__enter__(self):
#如果是数据库连接就可以返回cursor对象
#cursor=self.cursor
#returncursor
print('beginquery,')
returnself#由于这里没有资源对象就返回对象
def__exit__(self,exc_type,exc_value,traceback):
iftracebackisNone:
print('End')
else:
print('Error')
#如果是数据库连接提交操作
#iftracebackisNone:
#self.commit()
#else:
#self.rollback()
if__name__=='__main__':
withtest_query()asq:
q.query()
contextlib
编写enter和exit仍然很繁琐，因此Python的标准库contextlib提供了更简单的写法，
基本语法
@contextmanager defsome_generator(): try: yield finally:
生成器函数some_generator就和我们普通的函数一样，它的原理如下：
some_generator函数在在yield之前的代码等同于上下文管理器中的enter函数.
yield的返回值等同于enter函数的返回值，即如果with语句声明了as
然后执行
例，
自动加括号
importcontextlib
@contextlib.contextmanager
deftag(name):
print('<{}>'.format(name))
yield
print(''.format(name))
if__name__=='__main__':
withtag('h1')ast:
print('hello')
print('context')
管理锁
@contextmanager deflocked(lock): lock.acquire() try: yield finally: lock.release() withlocked(myLock): #代码执行到这里时，myLock已经自动上锁 pass #执行完后会，会自动释放锁
管理文件关闭
@contextmanager
defmyopen(filename,mode="r"):
f=open(filename,mode)
try:
yieldf
finally:
f.close()
withmyopen("test.txt")asf:
forlineinf:
print(line)
管理数据库回滚
@contextmanager deftransaction(db): db.begin() try： yield except: db.rollback() raise else: db.commit() withtransaction(mydb): mydb.cursor.execute(sql) mydb.cursor.execute(sql) mydb.cursor.execute(sql) mydb.cursor.execute(sql)
这就很方便！
如果一个对象没有实现上下文，我们就不能把它用于with语句。这个时候，可以用closing()来把该对象变为上下文对象。它的exit函数仅仅调用传入参数的close函数.
例如，用with语句使用urlopen()：
importcontextlib
fromurllib.requestimporturlopen
if__name__=='__main__':
withcontextlib.closing(urlopen('https://www.python.org'))aspage:
forlineinpage:
print(line)
closing也是一个经过@contextmanager装饰的generator，这个generator编写起来其实非常简单：
@contextmanager
defclosing(thing):
try:
yieldthing
finally:
thing.close()
嵌套使用
importcontextlib
fromurllib.requestimporturlopen
if__name__=='__main__':
withcontextlib.closing(urlopen('https://www.python.org'))aspage,\
contextlib.closing(urlopen('https://www.python.org'))asp:
forlineinpage:
print(line)
print(p)
在2.x中需要使用contextlib.nested()才能使用嵌套，3.x中可以直接使用。
更多函数可以参考官方文档https://docs.python.org/3/library/contextlib.html