解析 Python 裝飾器高級用法六項
作者：手把手PythonAI編程
本文通過六個高級用法詳細探討了Python裝飾器的強大功能，通過實際的代碼示例和逐步引導，希望讀者能夠深入理解裝飾器的原理和應用。
在Python編程中，裝飾器（Decorators）是一種強大且靈活的工具，它允許你在不修改原有函數(shù)或方法定義的情況下，為其添加新的功能。裝飾器是高級主題，但掌握它后，你將能寫出更加簡潔和可重用的代碼。本文將深入探討Python裝飾器的6項高級用法，通過詳細的例子和逐步的引導，幫助初學者理解和掌握這一強大特性。
1. 帶參數(shù)的裝飾器
裝飾器本身也可以接受參數(shù)，這使得裝飾器更加靈活。我們可以通過定義一個外層的函數(shù)來傳遞參數(shù)給裝飾器。
def my_decorator(arg):
def wrapper(func):
def inner(*args, **kwargs):
print(f"Decorator argument: {arg}")
return func(*args, **kwargs)
return inner
return wrapper
@my_decorator("Hello, World!")
def say_hello():
print("Hello from the function!")
say_hello() 輸出：
Decorator argument: Hello, World!
Hello from the function! 代碼解釋：
- my_decorator 是一個接受參數(shù)的裝飾器工廠。
- wrapper 是實際的裝飾器，它接受一個函數(shù) func 作為參數(shù)。
- inner 是實際調(diào)用原始函數(shù)并添加額外功能的包裝函數(shù)。
2. 類裝飾器
除了函數(shù)裝飾器，Python還支持類裝飾器。類裝飾器允許你使用類來定義裝飾器邏輯。
class MyDecorator:
def __init__(self, func):
self.func = func
def __call__(self, *args, **kwargs):
print("Before the function is called.")
result = self.func(*args, **kwargs)
print("After the function is called.")
return result
@MyDecorator
def greet(name):
print(f"Hello, {name}!")
greet("Alice") 輸出：
Before the function is called.
Hello, Alice!
After the function is called. 代碼解釋：
- MyDecorator 是一個類，它的 __init__ 方法接受一個函數(shù) func。
- __call__ 方法使得類的實例可以像函數(shù)一樣被調(diào)用。
3. 裝飾器鏈
你可以將多個裝飾器應用于同一個函數(shù)，這些裝飾器會按照從外到內(nèi)的順序被調(diào)用。
def decorator_one(func):
def wrapper(*args, **kwargs):
print("Decorator One")
return func(*args, **kwargs)
return wrapper
def decorator_two(func):
def wrapper(*args, **kwargs):
print("Decorator Two")
return func(*args, **kwargs)
return wrapper
@decorator_one
@decorator_two
def my_function():
print("My Function")
my_function() 輸出：
Decorator One
Decorator Two
My Function 代碼解釋：
- decorator_one 和 decorator_two 是兩個裝飾器。
- @decorator_one 和 @decorator_two 按照從外到內(nèi)的順序應用于 my_function。
4. 裝飾器中的裝飾器
裝飾器內(nèi)部也可以嵌套使用其他裝飾器，從而組合多個裝飾器的功能。
def decorator_a(func):
def wrapper(*args, **kwargs):
print("Decorator A")
return func(*args, **kwargs)
return wrapper
def decorator_b(func):
def wrapper(*args, **kwargs):
print("Decorator B")
return func(*args, **kwargs)
return wrapper
def combine_decorators(func):
return decorator_a(decorator_b(func))
@combine_decorators
def my_function():
print("My Function")
my_function() 輸出：
Decorator A
Decorator B
My Function 代碼解釋：
- combine_decorators 是一個裝飾器，它將 decorator_a 和 decorator_b 組合在一起。
5. 裝飾器與異步函數(shù)
Python 3.5 引入了異步編程的支持，裝飾器也可以用于異步函數(shù)。
import asyncio
def async_decorator(func):
async def wrapper(*args, **kwargs):
print("Before the async function is called.")
await func(*args, **kwargs)
print("After the async function is called.")
return wrapper
@async_decorator
async def async_greet(name):
await asyncio.sleep(1)
print(f"Hello, {name}!")
async def main():
await async_greet("Bob")
asyncio.run(main()) 輸出：
Before the async function is called.
Hello, Bob!
After the async function is called. 代碼解釋：
- async_decorator 是一個裝飾器，用于異步函數(shù)。
- async_greet 是一個異步函數(shù)，通過 await 關(guān)鍵字等待異步操作完成。
6. 裝飾器與屬性
在類的方法上應用裝飾器時，可以訪問類的屬性和其他方法。
class MyClass:
def __init__(self):
self.count = 0
def decorator(self, func):
def wrapper(*args, **kwargs):
self.count += 1
print(f"Count before call: {self.count}")
result = func(*args, **kwargs)
print(f"Count after call: {self.count}")
return result
return wrapper
@decorator
def my_method(self):
print("Method is called")
obj = MyClass()
obj.my_method()
obj.my_method() 輸出：
Count before call: 1
Method is called
Count after call: 2
Count before call: 3
Method is called
Count after call: 4 代碼解釋：
- MyClass 類中有一個裝飾器 decorator，它訪問類的 count 屬性。
- @decorator 應用于 my_method，每次調(diào)用 my_method 時都會更新 count 屬性。
實戰(zhàn)案例：日志記錄裝飾器
在實際開發(fā)中，日志記錄是一個常見的需求。我們可以編寫一個裝飾器來自動記錄函數(shù)的調(diào)用情況，包括調(diào)用時間、參數(shù)和返回值。
import time
def log_decorator(func):
def wrapper(*args, **kwargs):
start_time = time.time()
print(f"Calling {func.__name__} with args: {args}, kwargs: {kwargs}")
result = func(*args, **kwargs)
end_time = time.time()
print(f"{func.__name__} returned {result} in {end_time - start_time:.4f} seconds")
return result
return wrapper
@log_decorator
def complex_calculation(a, b):
time.sleep(2) # Simulate a long-running calculation
return a + b
result = complex_calculation(3, 4)
print(f"Final result: {result}") 輸出：
Calling complex_calculation with args: (3, 4), kwargs: {}
complex_calculation returned 7 in 2.0021 seconds
Final result: 7 案例分析：- log_decorator 裝飾器記錄了 complex_calculation 函數(shù)的調(diào)用時間、參數(shù)和返回值。- 這種方式非常適用于調(diào)試和性能分析，而不需要修改原始函數(shù)的代碼。
總結(jié)
本文通過六個高級用法詳細探討了Python裝飾器的強大功能，包括帶參數(shù)的裝飾器、類裝飾器、裝飾器鏈、裝飾器中的裝飾器、裝飾器與異步函數(shù)以及裝飾器與屬性的結(jié)合使用。通過實際的代碼示例和逐步引導，希望讀者能夠深入理解裝飾器的原理和應用。
責任編輯：趙寧寧
來源：
手把手PythonAI編程
