Python装饰器是一种强大的工具,用于在不修改原有函数代码的基础上为其添加额外的功能。以下是一些具体使用装饰器的示例:
示例1:日志记录
装饰器可以用来为函数添加日志记录功能,以便追踪函数的调用情况。
import logging
def log_decorator(func):
def wrapper(*args, **kwargs):
logging.info(f"Calling function {
func.__name__} with args={
args}, kwargs={
kwargs}")
result = func(*args, **kwargs)
logging.info(f"Function {
func.__name__} returned: {
result}")
return result
return wrapper
@log_decorator
def calculate_sum(a, b):
return a + b
calculate_sum(3, 5) # 输出类似:Calling function calculate_sum with args=(3, 5), kwargs={}
# Function calculate_sum returned: 8
在这个例子中,log_decorator 装饰器会在调用 calculate_sum 函数前后分别记录一条日志,显示函数名、传入的参数和返回的结果。
示例2:性能计时
装饰器可以用来测量函数的执行时间,有助于性能分析和优化。
import time
def timer_decorator(func):
def wrapper(*args, **kwargs):
start_time = time.time()
result = func(*args, **kwargs)
end_time = time.time()
execution_time = end_time - start_time
print(f"Function