python中类装饰器的实现示例

目录

• 1. 类装饰器的基本语法
• 2. 类装饰器的工作原理
• 3. 类装饰器的示例
• 3.1 添加类属性
• 3.2 添加实例方法
• 3.3 修改类的行为
• 4. 类装饰器与函数装饰器的区别
• 5. 类装饰器的高级用法
• 5.1 使用类装饰器实现单例模式
• 5.2 使用类装饰器实现属性验证
• 6. 总结

在 Python 中，类装饰器是一种强大的工具，可以用来增强或修改类的行为。类装饰器本质上是一个函数，它接受一个类作为参数，并返回一个新的类或对原类进行修改。通过使用类装饰器，你可以实现诸如日志记录、权限检查、属性验证等功能，而无需修改类的内部实现。

类装饰器的基本语法如下：

@decorator
class MyClass:
    pass

这等价于：

class MyClass:
    pass

MyClass = decorator(MyClass)

类装饰器是一个函数，它接受一个类作为参数，并返回一个新的类或对原类进行修改。以下是类装饰器的基本实现：

def decorator(cls):
    # 修改类或返回一个新的类
    return cls

假设我们想为一个类添加一个类属性，可以使用类装饰器来实现：

def add_class_attribute(attr_name, attr_value):
    def decorator(cls):
        setattr(cls, attr_name, attr_value)
        return cls
    return decorator

@add_class_attribute('new_attribute', 42)
class MyClass:
    pass

print(MyClass.new_attribute)  # 输出 42

假设我们想为一个类添加一个实例方法，可以使用类装饰器来实现：

def add_instance_method(method_name, method):
    def decorator(cls):
        setattr(cls, method_name, method)
        return cls
    return decorator

def greet(self):
    return f"Hello, my name is {self.name}"

@add_instance_method('greet', greet)
class Person:
    def __init__(self, name):
        self.name = name

person = Person("Alice")
print(person.greet())  # 输出 "Hello, my name is Alice"

假设我们想在类的初始化时记录日志，可以使用类装饰器来实现：

import logging

def log_initialization(cls):
    original_init = cls.__init__

    def new_init(self, *args, **kwargs):
        logging.info(f"Initializing {cls.__name__} with args={args} and kwargs={kwargs}")
        original_init(self, *args, **kwargs)

    cls.__init__ = new_init
    return cls

@log_initialization
class MyClass:
    def __init__(self, name):
        self.name = name

obj = MyClass("Test")
# 日志输出: INFO:root:Initializing MyClass with args=('Test',) and kwargs={}

• 函数装饰器：用于装饰函数，通常用于修改函数的行为。函数装饰器接受一个函数作为参数，并返回一个新的函数。
• 类装饰器：用于装饰类，通常用于修改类的行为。类装饰器接受一个类作为参数，并返回一个新的类或对原类进行修改。

单例模式是一种常见的设计模式，确保一个类只有一个实例。可以使用类装饰器来实现单例模式：

def singleton(cls):
    instance = None

    def wrapper(*args, **kwargs):
        nonlocal instance
        if instance is None:
            instance = cls(*args, **kwargs)
        return instance

    return wrapper

@singleton
class MyClass:
    def __init__(self, name):
        self.name = name

obj1 = MyClass("Alice")
obj2 = MyClass("Bob")

print(obj1 is obj2)  # 输出 True
print(obj1.name)  # 输出 Alice
print(obj2.name)  # 输出 Alice

假设我们想为类的属性添加验证逻辑，可以使用类装饰器来实现：

def validate_attributes(**validations):
    def decorator(cls):
        original_init = cls.__init__

        def new_init(self, *args, **kwargs):
            original_init(self, *args, **kwargs)
            for attr_name, validation_func in validations.items():
                value = getattr(self, attr_name)
                if not validation_func(value):
                    raise ValueError(f"Invalid value for {attr_name}: {value}")

        cls.__init__ = new_init
        return cls
    return decorator

def is_positive(value):
    return value > 0

@validate_attributes(age=is_positive)
class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age

person = Person("Alice", 30)
# person = Person("Alice", -1)  # 抛出 ValueError: Invalid value for age: -1

类装饰器是 Python 中一个非常强大的工具，可以用来增强或修改类的行为。通过使用类装饰器，你可以实现诸如添加类属性、添加实例方法、修改类的行为、实现单例模式、属性验证等功能，而无需修改类的内部实现。类装饰器不仅提高了代码的复用性和可维护性，还使得代码更加简洁和易读。

到此这篇关于python中类装饰器的实现示例的文章就介绍到这了,更多相关python 类装饰器内容请搜索风君子博客以前的文章或继续浏览下面的相关文章希望大家以后多多支持风君子博客！

您可能感兴趣的文章:

• Python装饰器之类装饰器详解
• Python类装饰器实现方法详解
• python类装饰器用法实例