使用装饰器实现自动化日志系统

Bruce_Liuxiaowei2024-05-26 22:29

使用装饰器实现自动化日志系统

使用装饰器实现自动化日志系统的示例代码如下：

python 复制代码

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 函数中。

输出结果将是：

python 复制代码

INFO:root:调用函数：加法
INFO:root:函数执行完成：加法，结果为：5
INFO:root:调用函数：乘法
INFO:root:函数执行完成：乘法，结果为：20