如何用自定义装饰器为已有库装饰器按需添加额外参数？

你的问题出在自定义装饰器的层级逻辑错误：你把check_condition当成了能接收第三方装饰器的高阶工具，但它本质是个普通装饰器，只能直接包装业务函数。当你用@check_condition(existing_decorator('This is argument'))时，existing_decorator返回的装饰器被当成了check_condition的参数f，后续调用时参数传递完全混乱，才会抛出TypeError。

正确实现方式 #

我们需要把check_condition改成高阶装饰器工厂，让它能接收第三方装饰器及其参数，再根据条件动态决定如何应用装饰器：

from functools import wraps

def is_production():
    # 替换为你的环境判断逻辑
    return True

def check_condition(decorator, *dec_args, **dec_kwargs):
    def wrapper_decorator(func):
        if is_production():
            # 生产环境：给装饰器追加additional=False参数
            return decorator(*dec_args, additional=False, **dec_kwargs)(func)
        else:
            # 非生产环境：直接用原参数调用装饰器
            return decorator(*dec_args, **dec_kwargs)(func)
    return wrapper_decorator

# 模拟第三方装饰器
def existing_decorator(arg, additional=True):
    def decorator(func):
        @wraps(func)
        def inner(*args, **kwargs):
            print(f"第三方装饰器参数：arg={arg}, additional={additional}")
            return func(*args, **kwargs)
        return inner
    return decorator

# 使用方式
@check_condition(existing_decorator, 'This is argument')
def my_business_func():
    print("业务函数执行")

my_business_func()

逻辑说明 #

1. check_condition先接收第三方装饰器decorator和它所需的参数dec_args、dec_kwargs
2. 返回的wrapper_decorator会接收真正的业务函数func
3. 根据is_production()的结果，决定是否给第三方装饰器追加additional=False参数，再完成装饰

另一种更简洁的写法（参数固定场景） #

如果existing_decorator的参数不需要动态传递，也可以直接把逻辑写在装饰器里，省去高阶工厂的层级：

def check_condition(func):
    if is_production():
        return existing_decorator('This is argument', additional=False)(func)
    else:
        return existing_decorator('This is argument')(func)

# 使用方式
@check_condition
def my_business_func():
    print("业务函数执行")

这种写法更简单，但灵活性不如第一种，适合装饰器参数固定的场景。

内容的提问来源于stack exchange，提问作者Denis Vitez