使用装饰器实现自动化日志系统
使用装饰器实现自动化日志系统的示例代码如下:
import logging
from functools import wraps
def log_function(func):
@wraps(func)
def wrapper(*args, **kwargs):
logging.info(f"调用函数:{func.__name__}")
result = func(*args, **kwargs)
logging.info(f"函数执行完成:{func.__name__},结果为:{result}")
return result
return wrapper
@log_function
def 加法(a, b):
return a + b
@log_function
def 乘法(a, b):
return a * b
logging.basicConfig(level=logging.INFO)
print(加法(2, 3))
print(乘法(4, 5))
在这个示例中,我们定义了一个装饰器 log_function,它记录函数调用前的和后的信息。@wraps(func) 用于保留原始函数的元数据(名称、文档字符串等)。
然后,我们将这个装饰器应用于两个函数:加法 和 乘法。当这些函数被调用时,它们将被包围在 logging 函数中。
输出结果将是:
INFO:root:调用函数:加法
INFO:root:函数执行完成:加法,结果为:5
INFO:root:调用函数:乘法
INFO:root:函数执行完成:乘法,结果为:20