Python中能否在闭包内用with语句延长上下文管理器生命周期？

你希望上下文管理器的生命周期和闭包（嵌套函数）保持一致，即闭包创建时进入上下文，逻辑执行完成或闭包销毁时退出上下文。但原代码中with块会在closure()返回嵌套函数后立即结束，导致__exit__提前触发，无法达到预期效果。

原示例代码：

class CtxMgr:
    def __enter__(self):
        print('enter')
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('exit')
        return False

def closure():
    with CtxMgr() as mgr:
        def enclosed():
            print('called')
        return enclosed


cl = closure()
print('generated')
cl()

当前输出：

enter
exit
generated
called

期望输出：

enter
generated
called
exit

下面提供两种可行的实现方案：

方法一：手动管控上下文（适合闭包仅调用一次的场景） #

直接在closure()中手动调用上下文管理器的__enter__，并在嵌套函数的finally块中触发__exit__，确保逻辑执行完毕后再退出上下文：

class CtxMgr:
    def __enter__(self):
        print('enter')
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('exit')
        return False

def closure():
    mgr = CtxMgr()
    mgr.__enter__()  # 手动进入上下文
    def enclosed():
        try:
            print('called')
        finally:
            # 执行完业务逻辑后，手动退出上下文
            mgr.__exit__(None, None, None)
    return enclosed

cl = closure()
print('generated')
cl()

执行输出完全匹配预期：

enter
generated
called
exit

注意：这种方式下如果闭包被多次调用，每次都会触发__exit__，不符合上下文管理器“一次进入、一次退出”的常规语义，因此更适合闭包仅调用一次的场景。

方法二：绑定垃圾回收时机（适合闭包多次调用的场景） #

如果需要上下文在闭包的整个生命周期（从创建到被垃圾回收）保持活跃，可以将上下文管理器绑定到闭包上，并利用__del__方法在闭包被销毁时触发__exit__：

class CtxMgr:
    def __enter__(self):
        print('enter')
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('exit')
        return False

def closure():
    mgr = CtxMgr()
    mgr.__enter__()
    def enclosed():
        print('called')
    
    # 将上下文管理器绑定到闭包对象，避免被提前回收
    enclosed._ctx_mgr = mgr
    
    # 定义清理函数，在闭包被销毁时执行上下文退出逻辑
    def cleanup():
        mgr.__exit__(None, None, None)
    enclosed.__del__ = cleanup
    
    return enclosed

cl = closure()
print('generated')
cl()
# 手动删除闭包引用并触发垃圾回收（实际场景中Python会自动处理，此处为了立即看到exit输出）
del cl
import gc
gc.collect()

执行输出：

enter
generated
called
exit

这种方式下，上下文会在闭包创建时进入，直到闭包被垃圾回收时才退出，无论闭包被调用多少次，上下文只会退出一次。

内容的提问来源于stack exchange，提问作者Andreas Franek