上下文管理(with)
with是一种上下文管理协议,目的在于从流程图中把 try,except 和finally 关键字和资源分配释放相关代码统统去掉,简化try….except….finlally的处理流程。with通过enter方法初始化,然后在exit中做善后以及处理异常。所以使用with处理的对象必须有enter()和exit()这两个方法。其中enter()方法在语句体(with语句包裹起来的代码块)执行之前进入运行,exit()方法在语句体执行完毕退出后运行。 with 语句适用于对资源进行访问的场合,确保不管使用过程中是否发生异常都会执行必要的“清理”操作,释放资源,比如文件使用后自动关闭、线程中锁的自动获取和释放等。
with的使用场景 如果某项工作完成后需要有释放资源或者其他清理工作,比如说文件操作时,就可以使用with优雅的处理,不用自己手动关闭文件句柄,而且with还能很好的管理上下文异常。
with open('a.py') as fp: for line in fp: print(line,end=" ")
with工作原理 with后面的语句被求值后,该语句返回的对象的enter()方法被调用,这个方法将返回的值赋给as后面的变量,当with包围的语句块全部执行完毕后,自动调用对象的exit()方法。 with处理异常会更方便,省去try…else…finally复杂的流程,这个用到的就是对象的exit()方法: exit( exc_type, exc_value, exc_trackback) 后面三个参数是固定的,用来接收with执行过程中异常类型,异常名称和异常的详细信息
其他场景应用
import contextlib @contextlib.contextmanager def database(): db = Database() try: if not db.connected: db.connect() yield db except Exception as e: db.close() def handle_query(): with database() as db: print 'handle ---', db.query()