快捷導(dǎo)航

python中類裝飾器的實現(xiàn)示例

更新時間：2026年02月04日 09:51:48 作者：哈里謝頓

在 Python 中,類裝飾器是一種強大的工具,可以用來增強或修改類的行為,類裝飾器本質(zhì)上是一個函數(shù),它接受一個類作為參數(shù),并返回一個新的類或?qū)υ愡M行修改,下面就來詳細的介紹一下如何實現(xiàn)

在 Python 中，類裝飾器是一種強大的工具，可以用來增強或修改類的行為。類裝飾器本質(zhì)上是一個函數(shù)，它接受一個類作為參數(shù)，并返回一個新的類或?qū)υ愡M行修改。通過使用類裝飾器，你可以實現(xiàn)諸如日志記錄、權(quán)限檢查、屬性驗證等功能，而無需修改類的內(nèi)部實現(xiàn)。

1. 類裝飾器的基本語法

類裝飾器的基本語法如下：

@decorator
class MyClass:
    pass

這等價于：

class MyClass:
    pass

MyClass = decorator(MyClass)

2. 類裝飾器的工作原理

類裝飾器是一個函數(shù)，它接受一個類作為參數(shù)，并返回一個新的類或?qū)υ愡M行修改。以下是類裝飾器的基本實現(xiàn)：

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

3. 類裝飾器的示例

3.1 添加類屬性

假設(shè)我們想為一個類添加一個類屬性，可以使用類裝飾器來實現(xiàn)：

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

3.2 添加實例方法

假設(shè)我們想為一個類添加一個實例方法，可以使用類裝飾器來實現(xiàn)：

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"

3.3 修改類的行為

假設(shè)我們想在類的初始化時記錄日志，可以使用類裝飾器來實現(xiàn)：

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={}

4. 類裝飾器與函數(shù)裝飾器的區(qū)別

函數(shù)裝飾器：用于裝飾函數(shù)，通常用于修改函數(shù)的行為。函數(shù)裝飾器接受一個函數(shù)作為參數(shù)，并返回一個新的函數(shù)。
類裝飾器：用于裝飾類，通常用于修改類的行為。類裝飾器接受一個類作為參數(shù)，并返回一個新的類或?qū)υ愡M行修改。

5. 類裝飾器的高級用法

5.1 使用類裝飾器實現(xiàn)單例模式

單例模式是一種常見的設(shè)計模式，確保一個類只有一個實例?？梢允褂妙愌b飾器來實現(xiàn)單例模式：

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

5.2 使用類裝飾器實現(xiàn)屬性驗證

假設(shè)我們想為類的屬性添加驗證邏輯，可以使用類裝飾器來實現(xiàn)：

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

6. 總結(jié)

類裝飾器是 Python 中一個非常強大的工具，可以用來增強或修改類的行為。通過使用類裝飾器，你可以實現(xiàn)諸如添加類屬性、添加實例方法、修改類的行為、實現(xiàn)單例模式、屬性驗證等功能，而無需修改類的內(nèi)部實現(xiàn)。類裝飾器不僅提高了代碼的復(fù)用性和可維護性，還使得代碼更加簡潔和易讀。

到此這篇關(guān)于python中類裝飾器的實現(xiàn)示例的文章就介紹到這了,更多相關(guān)python 類裝飾器內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: