如何在Python中自动包装含特定参数的类方法？

现有Stuff类，部分方法包含argument_x参数，示例代码如下：

class Stuff:
    def method_1(self, argument_x, **other_args):
        pass
    def method_2(self, argument_x, **other_args):
        pass
    def method_3(self, I_dont_have_argument_x):
        pass

希望创建该类的子类，对所有包含argument_x的方法进行统一包装，手动实现示例如下：

class StuffWithConstantX(Stuff):
    def __init__(self, argument_x_value):
        super().__init__()
        self._argument_x_value = argument_x_value
    def method_1(self, **other_args):
        super().method_1(argument_x=self._argument_x_value, **other_args)
    def method_2(self, **other_args):
        super().method_2(argument_x=self._argument_x_value, **other_args)

不含argument_x的method_3保持不变，请问能否实现自动化？具体怎么做？

可以实现自动化，核心是利用Python的反射机制解析方法参数，动态生成包装后的方法，以下是两种实用方案：

方案一：类装饰器实现 #

通过装饰器在类定义完成后，批量修改符合条件的方法：

import inspect

def wrap_constant_x_methods(cls):
    # 遍历类的所有方法（包括继承自父类的）
    for method_name, method in inspect.getmembers(cls, predicate=inspect.isfunction):
        # 跳过特殊方法（如__init__）
        if method_name.startswith('__'):
            continue
        # 解析方法参数签名
        sig = inspect.signature(method)
        if 'argument_x' in sig.parameters:
            # 定义包装函数，自动注入固定的argument_x值
            def wrapped_method(self, **other_args):
                return super(cls, self).__getattribute__(method_name)(
                    argument_x=self._argument_x_value, **other_args
                )
            # 替换原方法
            setattr(cls, method_name, wrapped_method)
    return cls

# 应用装饰器创建子类
@wrap_constant_x_methods
class StuffWithConstantX(Stuff):
    def __init__(self, argument_x_value):
        super().__init__()
        self._argument_x_value = argument_x_value

方案二：元类实现 #

元类可以在类创建阶段就完成方法包装，适合需要自动处理父类新增方法的场景：

import inspect

class ConstantXMeta(type):
    def __new__(meta, name, bases, attrs):
        # 创建子类
        subclass = super().__new__(meta, name, bases, attrs)
        # 遍历所有父类的方法
        for base in bases:
            for method_name, method in inspect.getmembers(base, predicate=inspect.isfunction):
                if method_name.startswith('__'):
                    continue
                sig = inspect.signature(method)
                if 'argument_x' in sig.parameters:
                    # 生成包装方法
                    def wrapped(self, **other_args):
                        return super(subclass, self).__getattribute__(method_name)(
                            argument_x=self._argument_x_value, **other_args
                        )
                    setattr(subclass, method_name, wrapped)
        return subclass

# 使用元类定义子类
class StuffWithConstantX(Stuff, metaclass=ConstantXMeta):
    def __init__(self, argument_x_value):
        super().__init__()
        self._argument_x_value = argument_x_value

关键说明 #

• 两种方案都用inspect.signature解析方法参数，确保精准识别包含argument_x的方法。
• 包装函数中使用super().__getattribute__是为了避免闭包陷阱，确保调用的是父类对应方法。
• 元类方案会自动处理父类后续新增的含argument_x的方法，装饰器方案则需要重新应用装饰器才能生效。

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