在Python中,裝飾器是一種特殊類型的函數,它可以用來修改其他函數的行為。裝飾器可以接受參數,這使得它們更加靈活和可重用。要處理裝飾器的參數,你需要使用functools.wraps
來保留原函數的元數據,并使用一個外部函數來接收和處理這些參數。
下面是一個簡單的例子,展示了如何創建一個帶參數的裝飾器:
import functools
def decorator_with_args(arg1, arg2):
def decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
print(f"裝飾器參數:{arg1}, {arg2}")
result = func(*args, **kwargs)
return result
return wrapper
return decorator
@decorator_with_args("Hello", "World")
def my_function(a, b):
return a + b
result = my_function(1, 2)
print(f"函數結果:{result}")
在這個例子中,我們創建了一個名為decorator_with_args
的外部函數,它接受兩個參數arg1
和arg2
。這個函數返回一個裝飾器,該裝飾器接受一個函數func
作為參數。在裝飾器內部,我們定義了一個名為wrapper
的嵌套函數,它接受任意數量的位置參數和關鍵字參數。在wrapper
函數內部,我們可以訪問外部函數的參數arg1
和arg2
,并在調用原始函數之前或之后執行任何自定義邏輯。
當我們使用@decorator_with_args("Hello", "World")
裝飾my_function
時,裝飾器會打印出傳遞給它的參數,然后調用原始函數并返回其結果。