使用metaclass元类装饰方法时如何正确处理staticmethod静态方法

原有实现的问题集中在两点：

1. 装饰器显式声明了self作为第一个入参，无法兼容无self参数的静态方法，只能通过try except兜底，会误吞原函数本身抛出的业务异常，稳定性极差
2. 元类的方法识别逻辑仅匹配普通函数类型，staticmethod/classmethod属于Python描述符类型，默认会被过滤；且遍历dir(klass)会拿到父类继承的方法，容易出现重复装饰的问题

我们可以通过参数透传+按方法类型分类处理解决该问题，优化后的完整代码如下：

from functools import wraps
import types

def decorator_function(input_function):
    @wraps(input_function)
    def wrapper(*args, **kwargs):
        if kwargs.get("test_parameter"):
            kwargs["test_parameter"] = 999
        # 直接透传所有参数，不需要区分是否有self
        return input_function(*args, **kwargs)
    return wrapper


class DecoratorMetaClass(type):
    def __new__(meta, name, bases, class_dict):
        # 仅遍历当前类定义的属性，避免处理父类继承的方法
        for key, value in class_dict.items():
            # 跳过系统自带的魔法方法
            if key.startswith('__') and key.endswith('__'):
                continue
            # 处理静态方法
            if isinstance(value, staticmethod):
                original_func = value.__func__
                wrapped_func = decorator_function(original_func)
                class_dict[key] = staticmethod(wrapped_func)
            # 处理类方法（不需要可以删除该分支）
            elif isinstance(value, classmethod):
                original_func = value.__func__
                wrapped_func = decorator_function(original_func)
                class_dict[key] = classmethod(wrapped_func)
            # 处理普通实例方法
            elif isinstance(value, types.FunctionType):
                class_dict[key] = decorator_function(value)
        # 处理完属性后再创建类
        return super().__new__(meta, name, bases, class_dict)


class InterfaceClass(metaclass=DecoratorMetaClass):
    def function(self, test_parameter=1):
        print(f"function - Test Parameter= {test_parameter}")

    @staticmethod
    def static_function(test_parameter=1):
        print(f"static_function - Test Parameter= {test_parameter}")


# 继承InterfaceClass会自动继承元类，不需要重复指定metaclass
class UserClass(InterfaceClass):
    def __init__(self):
        pass
    def function_2(self, test_parameter=1):
        print(f"function_2 - Test Parameter= {test_parameter}")


instance = UserClass()
instance.function(test_parameter=2)
instance.function_2(test_parameter=2)
instance.static_function(test_parameter=2)
print(isinstance(instance, InterfaceClass))

优化点说明 #

• 装饰器去掉了强制的self入参，用*args承接所有位置参数直接透传，完全兼容实例方法、静态方法、类方法的参数结构，不需要异常兜底
• 元类创建类前先修改类属性字典，按方法类型分类处理：静态方法/类方法需要先取出描述符包裹的实际函数，装饰完成后再重新封装为对应类型放回
• 仅处理当前类定义的非魔法方法，不会重复装饰父类方法，也不会影响类的继承关系，isinstance检查完全正常

执行代码后输出如下，符合预期：

function - Test Parameter= 999
function_2 - Test Parameter= 999
static_function - Test Parameter= 999
True

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