是否有库函数可将上下文管理器转换为装饰器？

当然有现成的成熟方案啦！Python标准库的contextlib模块里就提供了专门的工具，帮你轻松实现上下文管理器到装饰器的转换，完全不用自己维护代码逻辑。

方法1：使用ContextDecorator类 #

ContextDecorator是contextlib里的一个基类，只要让你的自定义上下文管理器继承它，这个上下文管理器就自动具备了装饰器的功能——既可以用在with语句里，也可以直接用来装饰函数。

举个实际的例子：

from contextlib import ContextDecorator
import time

# 自定义支持装饰器的上下文管理器
class TimingContext(ContextDecorator):
    def __enter__(self):
        self.start = time.time()
        return self
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        print(f"执行耗时: {time.time() - self.start:.2f}秒")
        # 返回False表示不捕获异常，让异常正常抛出
        return False

# 作为上下文管理器使用
with TimingContext():
    time.sleep(1)

# 直接作为装饰器使用
@TimingContext()
def slow_function():
    time.sleep(1)

slow_function()

方法2：用contextmanager装饰器生成可作为装饰器的上下文 #

如果你是用@contextmanager装饰器定义的上下文管理器，它天生就支持作为装饰器使用，不需要额外转换：

from contextlib import contextmanager

@contextmanager
def log_context():
    print("进入上下文：准备执行函数")
    yield
    print("退出上下文：函数执行完毕")

# 直接装饰函数
@log_context()
def my_func():
    print("正在执行函数逻辑")

my_func()

针对已有上下文管理器的转换 #

如果是已经存在的上下文管理器（比如标准库的threading.Lock，或者第三方库的上下文），不想修改它的实现，你写的context_manager_to_decorator函数逻辑是完全正确的！不过也可以借助ContextDecorator来包装一下，让它更贴合标准库的实现风格：

from contextlib import ContextDecorator
from functools import wraps
from typing import ContextManager

def context_manager_to_decorator(context_manager: ContextManager):
    class WrappedContext(ContextDecorator):
        def __enter__(self):
            return context_manager.__enter__()
        
        def __exit__(self, *args, **kwargs):
            return context_manager.__exit__(*args, **kwargs)
    
    def decorator(func):
        if not context_manager:
            return func
        @wraps(func)
        def wrapped(*args, **kwargs):
            with WrappedContext():
                return func(*args, **kwargs)
        return wrapped
    return decorator

# 用法示例：用threading.Lock做演示
import threading
lock = threading.Lock()

@context_manager_to_decorator(lock)
def thread_safe_func():
    print("线程安全的函数执行")

总的来说，优先用标准库提供的工具，既成熟又能避免自己维护代码可能带来的bug~

内容的提问来源于stack exchange，提问作者Nan Hua