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