要自定义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功能。