基本概念
·上下文管理协议(Context Management Protocol)
包含方法__enter__ 和 __exit__,支持该协议的对象要实现这两个方法。
·上下文管理器( Content Manager )
支持上下文管理协议的对象,实现了__enter__ / __exit__方法。上下文管理器定义执行with语句时要建立的运行时上下文,负责执行with语句块上下文的进入和退出操作。通常使用with语句调用上下文管理器。
·运行时上下文( runtime context )
由上下文管理器创建,通过上下文管理器的__enter和__exit__方法实现,__enter方法在语句体执行之前进入运行时上下文,__exit__在语句题执行完成后从上下文运行时退出。
·上下文表达式(Context Expression)
with 语句中跟在关键字with之后的表达式,该表达式返回一个上下文管理器对象。
·语句体(with-body)
with语句包裹起来的代码块,在执行语句体之前调用上下文管理器的__enter__方法,执行完语句体之后会执行__exit__方法。
with不是上下文管理器。with只是一个语句。实现了__enter__和__exit__这两个上下文管理协议方法的对象,都被称作上下文管理器。 上下文管理器被with调用时,会创建运行时上下文,__enter__在语句体执行之前进入运行时上下文,__exit__负责在语句执行完成 之后退出。
with
with语句用上下文管理器定义的方法包裹一段代码的执行,等价于try...except..finally。with的主要作用是相当于finally。
一、多个上下文管理器
with A() as a,B() as b: suite
二、上下文管理器类型
context manager 是Python中with语句用来定义运行时上下文的对象,上下文管理器控制着进/出运行时上下文的功能,上下文管理器通常由with语句出发,也可以直接通过他们的方法来调用它们。上下文管理器通常用于保存和恢复各式各样的全局状态、加解锁资源和关闭文件的文件等。
自定义上下文管理器
class TraceBlock: def message(self, arg): print('running ' + arg) def __enter__(self): print('starting with block') return self def __exit__(self, exc_type, exc_val, exc_tb): if exc_type is None: print('exited normally') else: print('raise an exception' + str(exc_type)) return False if __name__ == '__main__': with TraceBlock() as action: action.message('test1') raise Exception print('reached')
上下文装饰器
因为自定义装饰器比较麻烦,用contextlib.contextmanager将生成器转化为上下文管理器,不必创建一个类或单独指定__enter__和__exit__方法。
@contextmanager def message(arg): print('running' + arg) yield print('hello') if __name__ == '__main__': with message('hell'): print('ok')
装饰器,用来装饰一个生成器函数,使这个生成器函数称为一个上下文管理器。
包含yield的函数才是生成器函数,即contextmanager装饰的函数必须含有yield!
@contextmanager def make_context(): print('enter') try: yield {} except Exception as err: print('error', err) finally: print('exit') if __name__ == '__main__': with make_context() as value: raise Exception print(value)
生成器函数中yield之前的语句在__enter__方法中执行;yield之后的语句在__exit__中执行;yield产生的值赋值给as子句中的variable变量。如果有异常产生,在上面的例子中,能够被except捕获。
python学习网,免费的在线学习python平台,欢迎关注!