如何在 Python 中重写装饰器参数

wufei123 2025-01-26 阅读:42 评论:0

要修改子类中父类方法的装饰器参数，您必须在子类中重写该方法。仅仅在子类中声明同名的类变量并不会影响装饰器参数，除非您显式地重新定义该方法。示例代码将以下代码保存为 test.py 文件： def my_decorator_with_a...

如何在 python 中重写装饰器参数

要修改子类中父类方法的装饰器参数，您必须在子类中重写该方法。仅仅在子类中声明同名的类变量并不会影响装饰器参数，除非您显式地重新定义该方法。

示例代码

将以下代码保存为 test.py 文件：

PHP

def my_decorator_with_args(param1, param2):
    """带参数的装饰器"""

    def actual_decorator(func):
        def wrapper(self, *args, **kwargs):
            print(f"[装饰器] param1={param1}, param2={param2}")
            return func(self, *args, **kwargs)

        return wrapper

    return actual_decorator


class baseclass:
    @my_decorator_with_args(param1="base_param1", param2="base_param2")
    def greet(self):
        print("来自基类的问候！")


class derivedclass(baseclass):
    """
    尝试通过定义类变量来覆盖装饰器参数，
    但由于没有重新定义greet()方法，
    实际使用的是父类的装饰器参数。
    """

    param1 = "derived_param1"
    param2 = "derived_param2"
    # 故意不重新定义 greet() 方法


class derivedclassoverride(baseclass):
    """
    此模式重新定义(覆盖) greet() 方法
    以更改装饰器参数。
    """

    @my_decorator_with_args(param1="override_param1", param2="override_param2")
    def greet(self):
        print("来自派生类的问候！")


if __name__ == "__main__":
    print("=== 基类的 greet 方法 ===")
    b = baseclass()
    b.greet()

    print("
=== 派生类的 greet 方法（未覆盖）===")
    d = derivedclass()
    d.greet()

    print("
=== 派生类的 greet 方法（已覆盖）===")
    d_o = derivedclassoverride()
    d_o.greet()

运行代码：

PHP

python test.py

结果

PHP

=== 基类的 greet 方法 ===
[装饰器] param1=base_param1, param2=base_param2
来自基类的问候！

=== 派生类的 greet 方法（未覆盖）===
[装饰器] param1=base_param1, param2=base_param2
来自基类的问候！

=== 派生类的 greet 方法（已覆盖）===
[装饰器] param1=override_param1, param2=override_param2
来自派生类的问候！