如何自动记录Python类中含foo参数方法的传入值？

无需手动修改每个方法，也不用正则分析代码文本，Python内置的inspect模块可以精准获取方法的参数签名，实现自动记录foo参数值的需求，且能严格区分无foo参数和**foo传入None**的情况。

核心思路 #

1. 用inspect.signature获取每个方法的参数签名，判断是否包含foo参数；
2. 给符合条件的方法添加包装逻辑，在方法调用时绑定传入的参数，提取foo的值并记录；
3. 通过类装饰器批量处理整个类的方法，避免重复代码。

完整代码示例 #

import inspect

def track_foo_params(cls):
    # 遍历类中定义的所有方法
    for attr_name, attr_value in cls.__dict__.items():
        if inspect.isfunction(attr_value):
            method_sig = inspect.signature(attr_value)
            # 检查方法签名中是否存在foo参数
            if 'foo' in method_sig.parameters:
                # 定义包装函数，替换原方法
                def wrapped_method(self, *args, **kwargs):
                    # 绑定传入的参数到方法签名，兼容位置参数和关键字参数
                    bound_args = method_sig.bind(self, *args, **kwargs)
                    bound_args.apply_defaults()
                    # 提取foo参数的值
                    foo_val = bound_args.arguments['foo']
                    # 这里可替换为日志写入、变量存储等自定义逻辑
                    print(f"[{attr_name}] foo参数值: {foo_val}")
                    # 执行原方法
                    return attr_value(self, *args, **kwargs)
                # 替换类中的原方法
                setattr(cls, attr_name, wrapped_method)
    return cls

# 给目标类应用装饰器
@track_foo_params
class MyClass:
    def fun_one(self, foo: int):
        pass

    def fun_two(self, foo: int, flu: int):
        pass

    def fun_three(self, flu: str, foo: int):
        pass

    def fun_four(self):
        pass

# 测试验证
if __name__ == "__main__":
    obj = MyClass()
    obj.fun_one(123)                  # 输出: [fun_one] foo参数值: 123
    obj.fun_two(456, 789)             # 输出: [fun_two] foo参数值: 456
    obj.fun_three("test_str", None)   # 输出: [fun_three] foo参数值: None
    obj.fun_four()                    # 无输出（无foo参数）

方案优势对比正则 #

• 准确性更高：inspect直接获取运行时的方法参数签名，不会误判注释里的foo、重名局部变量等情况；
• 兼容性更好：自动处理位置参数、关键字参数、默认值等各种传参方式，正则分析代码文本很难覆盖所有场景；
• 可扩展性强：后续可以轻松修改记录逻辑（比如写入日志文件、统计参数分布），无需调整正则表达式。

注意事项 #

• 上述代码默认只处理当前类自身定义的方法，如果需要追踪父类继承来的方法，可以将遍历对象改为dir(cls)，并通过getattr(cls, attr_name)获取方法；
• 如果类中包含静态方法/类方法，需要调整包装函数的参数（比如类方法的第一个参数是cls而非self），可以通过inspect判断方法类型后做对应处理。

内容的提问来源于stack exchange，提问作者2br-2b