上下文管理(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()