Python PubSub订阅装饰器问题：传递函数参数至包装器遇报错

嘿，我明白你遇到的问题了——我当初第一次用pubsub结合装饰器的时候也踩过类似的坑！你用了functools.wraps以为能让包装器和原函数签名一致，但pubsub的sendMessage还是报错说有未知额外参数，核心原因其实是**functools.wraps只复制了函数的元数据（比如函数名、文档字符串），但没有真正复制原函数的参数签名**，而pubsub内部很可能是用inspect模块来解析订阅函数的参数，以此校验sendMessage传递的参数是否合法。

问题根源分析 #

当你用*args, **kwargs定义wrapper时，pubsub解析到的是一个“接受任意参数”的函数签名，但实际你的原函数可能只接受特定参数。这就导致两种混乱情况：

• 如果你给sendMessage传了原函数不接受的参数，pubsub会认为wrapper能处理，不会拦截，但原函数调用时会报错；
• 更糟的是，如果pubsub的校验逻辑是基于订阅函数的参数来限制sendMessage的参数，它可能误判允许/禁止传递某些参数，就出现了你遇到的“未知额外参数”提示。

解决方案：让包装器签名与原函数完全一致 #

我们需要让包装器的参数签名和原函数完全对齐，让pubsub能正确识别原函数的参数要求。可以通过以下步骤实现：

1. 用inspect.signature获取原函数的参数签名；
2. 给包装器的__signature__属性赋值原函数的签名，让inspect（也就是pubsub）能识别到正确的参数；
3. 在wrapper内部用签名绑定参数，确保传递给原函数的参数是合法的。

完整装饰器代码示例

from pubsub import pub
from functools import wraps
import inspect

def subscribe(topic):
    def decorator(func):
        # 获取原函数的参数签名
        original_sig = inspect.signature(func)
        
        @wraps(func)
        def wrapper(*args, **kwargs):
            # 可选：在这里添加你的装饰器逻辑，比如日志、参数校验等
            print(f"[Decorator] Processing message for topic: {topic}")
            
            # 绑定参数到原函数的签名，确保参数匹配
            bound_args = original_sig.bind(*args, **kwargs)
            bound_args.apply_defaults()
            return func(*bound_args.args, **bound_args.kwargs)
        
        # 关键：将原函数的签名赋值给wrapper，让pubsub识别正确的参数
        wrapper.__signature__ = original_sig
        
        # 订阅包装后的函数到指定主题
        pub.subscribe(wrapper, topic)
        return wrapper
    return decorator

使用示例

# 用装饰器订阅主题
@subscribe("user.register")
def handle_user_register(username, email, age=18):
    print(f"New user registered: {username}, email: {email}, age: {age}")

# 合法发送消息：参数完全匹配原函数
pub.sendMessage("user.register", username="mike", email="mike@example.com")
# 带默认参数的合法调用
pub.sendMessage("user.register", username="anna", email="anna@example.com", age=25)

# 非法调用：传递原函数不接受的参数，pubsub会提示未知额外参数（符合预期）
# pub.sendMessage("user.register", username="bob", email="bob@example.com", extra_param="test")

简化场景：如果装饰器不需要额外逻辑 #

如果你的装饰器只是单纯用来注册订阅，不需要在函数执行前后加逻辑，其实可以直接订阅原函数，完全跳过包装器：

def subscribe(topic):
    def decorator(func):
        pub.subscribe(func, topic)
        return func  # 直接返回原函数，不包装
    return decorator

这种情况下，pubsub直接订阅原函数，参数签名完全正确，自然不会有参数校验的问题。

内容的提问来源于stack exchange，提问作者Michael K.