Python裝飾器有哪些最佳實(shí)踐

小樊
81
2024-11-02 05:21:54

Python裝飾器是一種強(qiáng)大的工具，它允許你在不修改函數(shù)代碼的情況下，增加函數(shù)的功能。以下是一些關(guān)于Python裝飾器的最佳實(shí)踐：

裝飾器的基本用法

裝飾器是一個(gè)函數(shù)，它接受一個(gè)函數(shù)作為參數(shù)，并返回一個(gè)新的函數(shù)。裝飾器的基本用法包括日志記錄、性能測(cè)試、權(quán)限驗(yàn)證等。

def log_decorator(func):
    def wrapper(*args, **kwargs):
        print(f"Calling function {func.__name__}")
        result = func(*args, **kwargs)
        print(f"Function {func.__name__} finished")
        return result
    return wrapper

@log_decorator
def add(a, b):
    return a + b

add(3, 4)

使用functools.wraps保留原函數(shù)的信息

當(dāng)使用裝飾器時(shí)，原函數(shù)的元信息（如函數(shù)名、文檔字符串等）可能會(huì)丟失。使用functools.wraps可以保留這些信息。

from functools import wraps

def my_decorator(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        return func(*args, **kwargs)
    return wrapper

@my_decorator
def greet(name):
    print(f"Hello, {name}!")

greet("World")

帶參數(shù)的裝飾器

帶參數(shù)的裝飾器允許你根據(jù)傳入的參數(shù)動(dòng)態(tài)修改裝飾器的行為。

def repeat_decorator(times):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for _ in range(times):
                result = func(*args, **kwargs)
            return result
        return wrapper
    return decorator

@repeat_decorator(3)
def say_hello():
    print("Hello!")

say_hello()

類裝飾器

類裝飾器是裝飾類的特殊裝飾器，它通過(guò)實(shí)現(xiàn)__call__方法來(lái)允許類的實(shí)例像函數(shù)一樣被調(diào)用。

class Cache:
    def __init__(self, func):
        self.func = func
        self.cache = {}

    def __call__(self, *args):
        if args in self.cache:
            print("從緩存中獲取結(jié)果")
            return self.cache[args]
        result = self.func(*args)
        self.cache[args] = result
        return result

@Cache()
def fibonacci(n):
    if n < 2:
        return n
    return fibonacci(n - 1) + fibonacci(n - 2)

print(fibonacci(50))

裝飾器的執(zhí)行順序

裝飾器的執(zhí)行順序是從下往上的，即最內(nèi)層的裝飾器先執(zhí)行，然后是外層的裝飾器。

注意事項(xiàng)

  • 裝飾器應(yīng)盡量簡(jiǎn)單，避免過(guò)度裝飾。
  • 裝飾器可能會(huì)影響代碼的可讀性，因此應(yīng)謹(jǐn)慎使用。

通過(guò)遵循這些最佳實(shí)踐，你可以更有效地使用Python裝飾器來(lái)增強(qiáng)你的代碼功能，同時(shí)保持代碼的清晰和可維護(hù)性。

0 踩