从根本上说,它是一个用自定义逻辑来划分代码块的对象,自定义逻辑在入口和出口调用,并且可以在它的构造中接受参数。可以使用类定义自定义上下文管理器:class ContextManager(object):
def __init__(self, args):
pass
def __enter__(self):
# Entrance logic here, called before entry of with block
pass
def __exit__(self, exception_type, exception_val, trace):
# Exit logic here, called at exit of with block
return True
然后,入口被传递一个contextmanager类的实例,并且可以引用在__init__方法中创建的任何内容(文件、套接字等)。exit方法还接收内部块和堆栈跟踪对象中引发的任何异常,如果逻辑在没有引发的情况下完成,则返回None。
我们可以这样使用它:with ContextManager(myarg):
# ... code here ...
这对于管理资源生命周期、释放文件描述符、管理异常以及构建嵌入式dsl等更复杂的用途非常有用。
另一种(但相当于)构造方法是contextlibdecorator,它使用一个生成器来分离入口和出口逻辑。from contextlib import contextmanager
@contextmanager
def ContextManager(args):
# Entrance logic here
yield
# Exit logic here