120 lines
3.8 KiB
Python
120 lines
3.8 KiB
Python
|
import logging
|
|||
|
|
import os
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
from config.network_config import network_config
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Logger:
|
|||
|
|
"""
|
|||
|
|
简单的日志操作类，支持控制台和文件输出
|
|||
|
|
可以使用现有的logger对象或创建新的实例
|
|||
|
|
"""
|
|||
|
|
_instance = None
|
|||
|
|
|
|||
|
|
def __new__(cls, existing_logger=None, *args, **kwargs):
|
|||
|
|
if existing_logger is not None:
|
|||
|
|
# 如果提供了现有的logger对象，创建新实例并使用该对象
|
|||
|
|
instance = super().__new__(cls)
|
|||
|
|
instance.logger = existing_logger
|
|||
|
|
return instance
|
|||
|
|
|
|||
|
|
# 否则使用单例模式
|
|||
|
|
if cls._instance is None:
|
|||
|
|
cls._instance = super().__new__(cls)
|
|||
|
|
return cls._instance
|
|||
|
|
|
|||
|
|
def __init__(self, existing_logger=None, log_file="logs/app.log"):
|
|||
|
|
# 如果使用现有logger，直接返回
|
|||
|
|
if existing_logger is not None:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 如果已经初始化过，直接返回
|
|||
|
|
if hasattr(self, 'logger'):
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 创建logs目录（如果不存在）
|
|||
|
|
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
|||
|
|
|
|||
|
|
# 初始化logger
|
|||
|
|
self.logger = logging.getLogger('AppLogger')
|
|||
|
|
self.logger.setLevel(logging.DEBUG)
|
|||
|
|
|
|||
|
|
# 清除可能存在的处理器
|
|||
|
|
self.logger.handlers.clear()
|
|||
|
|
|
|||
|
|
# 设置日志格式
|
|||
|
|
formatter = logging.Formatter(
|
|||
|
|
'%(asctime)s [%(levelname)s] %(message)s',
|
|||
|
|
datefmt='%Y-%m-%d %H:%M:%S'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 添加控制台处理器
|
|||
|
|
console_handler = logging.StreamHandler()
|
|||
|
|
console_handler.setFormatter(formatter)
|
|||
|
|
self.logger.addHandler(console_handler)
|
|||
|
|
|
|||
|
|
# 添加文件处理器
|
|||
|
|
file_handler = logging.FileHandler(log_file, encoding='utf-8')
|
|||
|
|
file_handler.setFormatter(formatter)
|
|||
|
|
self.logger.addHandler(file_handler)
|
|||
|
|
|
|||
|
|
def debug(self, message, strategy_id=None):
|
|||
|
|
"""记录调试级别的日志"""
|
|||
|
|
self.logger.debug(message)
|
|||
|
|
try:
|
|||
|
|
network_config.insert_logs(strategy_id, "debug", message)
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def info(self, message, strategy_id=None):
|
|||
|
|
"""记录信息级别的日志"""
|
|||
|
|
self.logger.info(message)
|
|||
|
|
try:
|
|||
|
|
network_config.insert_logs(strategy_id, "info ", message)
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def warning(self, message, strategy_id=None):
|
|||
|
|
"""记录警告级别的日志"""
|
|||
|
|
self.logger.warning(message)
|
|||
|
|
try:
|
|||
|
|
network_config.insert_logs(strategy_id, "warn", message)
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def error(self, message, strategy_id=None):
|
|||
|
|
"""记录错误级别的日志"""
|
|||
|
|
self.logger.error(message)
|
|||
|
|
try:
|
|||
|
|
network_config.insert_logs(strategy_id, "error", message)
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def critical(self, message, strategy_id=None):
|
|||
|
|
"""记录严重错误级别的日志"""
|
|||
|
|
self.logger.critical(message)
|
|||
|
|
try:
|
|||
|
|
network_config.insert_logs(strategy_id, "critical", message)
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
# # 使用示例
|
|||
|
|
# if __name__ == "__main__":
|
|||
|
|
# # 示例1：创建新的logger实例
|
|||
|
|
# logger = Logger()
|
|||
|
|
# logger.info("这是一个新创建的logger实例")
|
|||
|
|
#
|
|||
|
|
# # 示例2：使用现有的logger对象
|
|||
|
|
# existing_logger = logging.getLogger('ExistingLogger')
|
|||
|
|
# existing_logger.setLevel(logging.DEBUG)
|
|||
|
|
# handler = logging.StreamHandler()
|
|||
|
|
# handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
|
|||
|
|
# existing_logger.addHandler(handler)
|
|||
|
|
#
|
|||
|
|
# logger2 = Logger(existing_logger=existing_logger)
|
|||
|
|
# logger2.info("这是使用现有logger对象创建的实例")
|
|||
|
|
#
|
|||
|
|
# # 示例3：测试单例模式（不传入existing_logger时）
|
|||
|
|
# logger3 = Logger()
|
|||
|
|
# logger3.info("这条日志来自第三个logger实例（与第一个是同一个实例）")
|