使用类装饰器扩展已定义库类时触发错误，寻求解决方案

错误原因 #

你用@decorator修饰已定义好的类是语法错误，@装饰器语法只能放在类/函数的定义语句上方，用来在定义时直接修改这个类/函数，不能对已经存在的类对象使用这个语法。

正确的两种实现方式 #

方式1：定义类时直接应用装饰器（适合自定义类场景） #

如果是你自己定义的类，可以直接在类定义前加@decorator，这样定义出来的类就是被装饰后的子类：

bar = "some value"  # 先定义bar，避免NameError

def decorator(general_library_class):
    class MyClass(general_library_class):
        def __init__(self, *args, **kwargs):
            super().__init__(*args, **kwargs)  # 传递参数给父类，适配更通用的场景
            # 这里添加你的自定义功能，比如新增属性
            self.new_attr = "custom attribute"
    
    return MyClass

@decorator
class library_class():
    def __init__(self):
        self.foo = bar

# 测试使用
obj = library_class()
print(obj.foo)          # 输出"some value"
print(obj.new_attr)     # 输出"custom attribute"

方式2：直接调用装饰器处理第三方类（推荐用于第三方库场景） #

对于第三方库已有的类，你没法修改它的定义代码，直接调用装饰器函数把第三方类传入，得到新的子类即可：

# 假设这是第三方库的类
class library_class():
    def __init__(self):
        self.foo = "bar"

def decorator(general_library_class):
    class MyClass(general_library_class):
        def __init__(self, *args, **kwargs):
            super().__init__(*args, **kwargs)
            # 添加自定义功能，比如新增方法
            def custom_func(self):
                print("custom method triggered")
            self.custom_func = custom_func.__get__(self)
    
    return MyClass

# 调用装饰器生成扩展后的子类
ExtendedLibraryClass = decorator(library_class)

# 测试使用扩展类
obj = ExtendedLibraryClass()
print(obj.foo)          # 输出"bar"
obj.custom_func()       # 输出"custom method triggered"

额外优化建议 #

• 子类的__init__最好加上*args, **kwargs，这样能兼容父类所有的初始化参数，不用因为父类参数变化而修改代码。
• 可以给装饰器生成的子类设置和父类一致的名称，比如MyClass.__name__ = general_library_class.__name__，这样调试时更友好。

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