温馨提示×

在Python中,如何自定义log函数

小樊
85
2024-08-18 20:42:33
栏目: 编程语言

要自定义log函数,可以使用Python中的logging模块来实现。以下是一个示例代码:

import logging

def custom_log(message, level="info"):
    logger = logging.getLogger(__name__)  # 创建logger对象
    formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')  # 设置日志格式

    # 创建文件处理器
    file_handler = logging.FileHandler('custom.log')
    file_handler.setFormatter(formatter)
    logger.addHandler(file_handler)

    if level == "info":
        logger.info(message)
    elif level == "warning":
        logger.warning(message)
    elif level == "error":
        logger.error(message)
    else:
        logger.debug(message)

# 使用自定义log函数
custom_log("This is an info message", "info")
custom_log("This is a warning message", "warning")
custom_log("This is an error message", "error")

在上面的示例中,我们定义了一个custom_log函数,该函数接受两个参数:message和level。根据level的不同,将日志记录到不同的级别中。通过调用custom_log函数,并传入不同的message和level参数,可以实现自定义的log功能。

0