如何用一个装饰器替代Python中的多个装饰器？

当你有一堆函数都用相同的装饰器堆叠时，把它们合并成一个统一的装饰器绝对是明智之举——既能减少重复代码，还能让后续维护更轻松。我给你分享几种常用的实现方式：

1. 把多个装饰器打包成一个新装饰器 #

最直接的方式就是定义一个“组合装饰器”，内部把你需要的装饰器按顺序依次应用到函数上。这样每个目标函数只需要加这一个装饰器就够了。

举个例子，假设你原本有两个装饰器decorator1和decorator2：

# 已有的两个装饰器
def decorator1(func):
    def wrapper(*args, **kwargs):
        print("运行装饰器1的逻辑")
        return func(*args, **kwargs)
    return wrapper

def decorator2(func):
    def wrapper(*args, **kwargs):
        print("运行装饰器2的逻辑")
        return func(*args, **kwargs)
    return wrapper

# 组合成一个新的装饰器
def combined_decorator(func):
    # 注意顺序：这里的嵌套顺序和直接写@decorator1 @decorator2是一致的
    return decorator1(decorator2(func))

# 现在只需要给函数加这一个装饰器
@combined_decorator
def func_a():
    print("执行函数A")

@combined_decorator
def func_b():
    print("执行函数B")

小提示：装饰器的顺序很关键！直接堆叠装饰器时是从下往上执行，所以组合时的嵌套顺序要和原来保持一致，不然逻辑执行顺序会变。

2. 批量给多个函数应用装饰器 #

如果你的函数数量特别多，甚至可以写个批量处理的逻辑，不用挨个加装饰器：

# 把需要装饰的函数列出来
target_functions = [func_a, func_b, func_c]

# 批量应用组合后的装饰器
for func in target_functions:
    decorated_func = combined_decorator(func)
    # 更新全局命名空间里的函数指向
    globals()[func.__name__] = decorated_func

3. 给类的所有方法统一应用装饰器 #

如果这些函数是某个类的方法，那可以用类装饰器一次性给所有方法加上装饰器：

def decorate_all_methods(decorator):
    def class_decorator(cls):
        # 遍历类的所有属性，给可调用的方法加装饰器
        for name, attr in vars(cls).items():
            if callable(attr):
                setattr(cls, name, decorator(attr))
        return cls
    return class_decorator

# 给整个类应用装饰器，里面的所有方法都会被combined_decorator装饰
@decorate_all_methods(combined_decorator)
class MyService:
    def process_data(self):
        print("处理数据")
    
    def validate_input(self):
        print("验证输入")

额外补充：带参数的装饰器组合 #

如果你的装饰器本身需要接收参数，也可以写一个带参数的组合装饰器：

def combined_decorator_with_args(decorator1_arg, decorator2_arg):
    def outer(func):
        # 把参数传递给对应的装饰器
        return decorator1(decorator1_arg)(decorator2(decorator2_arg)(func))
    return outer

# 使用时传入参数
@combined_decorator_with_args("日志级别: INFO", "超时时间: 10s")
def func_with_args():
    print("执行带参数装饰的函数")

这样调整后，你的代码会变得更简洁，后续如果要修改装饰器的逻辑（比如增减一个装饰器、调整参数），只需要修改组合装饰器的定义，不用挨个修改每个函数的装饰器，维护效率大大提升！

内容的提问来源于stack exchange，提问作者AXE Labs