python中類(lèi)裝飾器的實(shí)現(xiàn)示例

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

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

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

1. 類(lèi)裝飾器的基本語(yǔ)法

類(lèi)裝飾器的基本語(yǔ)法如下：

@decorator
class MyClass:
    pass

這等價(jià)于：

class MyClass:
    pass

MyClass = decorator(MyClass)

2. 類(lèi)裝飾器的工作原理

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

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

3. 類(lèi)裝飾器的示例

3.1 添加類(lèi)屬性

假設(shè)我們想為一個(gè)類(lèi)添加一個(gè)類(lèi)屬性，可以使用類(lèi)裝飾器來(lái)實(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í)例方法

假設(shè)我們想為一個(gè)類(lèi)添加一個(gè)實(shí)例方法，可以使用類(lèi)裝飾器來(lái)實(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 修改類(lèi)的行為

假設(shè)我們想在類(lèi)的初始化時(shí)記錄日志，可以使用類(lèi)裝飾器來(lái)實(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. 類(lèi)裝飾器與函數(shù)裝飾器的區(qū)別

函數(shù)裝飾器：用于裝飾函數(shù)，通常用于修改函數(shù)的行為。函數(shù)裝飾器接受一個(gè)函數(shù)作為參數(shù)，并返回一個(gè)新的函數(shù)。
類(lèi)裝飾器：用于裝飾類(lèi)，通常用于修改類(lèi)的行為。類(lèi)裝飾器接受一個(gè)類(lèi)作為參數(shù)，并返回一個(gè)新的類(lèi)或?qū)υ?lèi)進(jìn)行修改。

5. 類(lèi)裝飾器的高級(jí)用法

5.1 使用類(lèi)裝飾器實(shí)現(xiàn)單例模式

單例模式是一種常見(jiàn)的設(shè)計(jì)模式，確保一個(gè)類(lèi)只有一個(gè)實(shí)例?？梢允褂妙?lèi)裝飾器來(lái)實(shí)現(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 使用類(lèi)裝飾器實(shí)現(xiàn)屬性驗(yàn)證

假設(shè)我們想為類(lèi)的屬性添加驗(yàn)證邏輯，可以使用類(lèi)裝飾器來(lái)實(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é)

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

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

您可能感興趣的文章: