74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
|
import logging
|
|||
|
|
import os
|
|||
|
|
from datetime import datetime
|
|||
|
|
from logging.handlers import RotatingFileHandler
|
|||
|
|
from extensions import redis_client
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
# 创建日志目录
|
|||
|
|
LOG_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'logs')
|
|||
|
|
os.makedirs(LOG_DIR, exist_ok=True)
|
|||
|
|
|
|||
|
|
class SystemLogger:
|
|||
|
|
def __init__(self):
|
|||
|
|
self.logger = logging.getLogger('vision_ai')
|
|||
|
|
self.logger.setLevel(logging.INFO)
|
|||
|
|
|
|||
|
|
# 文件处理器 (自动轮转，最大10MB，保留5个备份)
|
|||
|
|
file_handler = RotatingFileHandler(
|
|||
|
|
os.path.join(LOG_DIR, 'system.log'),
|
|||
|
|
maxBytes=10*1024*1024,
|
|||
|
|
backupCount=5,
|
|||
|
|
encoding='utf-8'
|
|||
|
|
)
|
|||
|
|
file_handler.setLevel(logging.INFO)
|
|||
|
|
|
|||
|
|
# 控制台处理器
|
|||
|
|
console_handler = logging.StreamHandler()
|
|||
|
|
console_handler.setLevel(logging.INFO)
|
|||
|
|
|
|||
|
|
# 格式化
|
|||
|
|
formatter = logging.Formatter(
|
|||
|
|
'[%(asctime)s] %(levelname)s - %(message)s',
|
|||
|
|
datefmt='%Y-%m-%d %H:%M:%S'
|
|||
|
|
)
|
|||
|
|
file_handler.setFormatter(formatter)
|
|||
|
|
console_handler.setFormatter(formatter)
|
|||
|
|
|
|||
|
|
self.logger.addHandler(file_handler)
|
|||
|
|
self.logger.addHandler(console_handler)
|
|||
|
|
|
|||
|
|
def _push_to_redis(self, level, message, extra=None):
|
|||
|
|
"""同时推送到 Redis 用于前端实时查看"""
|
|||
|
|
try:
|
|||
|
|
log_entry = {
|
|||
|
|
"time": datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
|
|||
|
|
"level": level,
|
|||
|
|
"message": message,
|
|||
|
|
"extra": extra or {}
|
|||
|
|
}
|
|||
|
|
# 保留最近 2000 条日志
|
|||
|
|
redis_client.lpush('system_logs', json.dumps(log_entry, ensure_ascii=False))
|
|||
|
|
redis_client.ltrim('system_logs', 0, 1999)
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def info(self, message, **kwargs):
|
|||
|
|
self.logger.info(message)
|
|||
|
|
self._push_to_redis('INFO', message, kwargs)
|
|||
|
|
|
|||
|
|
def warning(self, message, **kwargs):
|
|||
|
|
self.logger.warning(message)
|
|||
|
|
self._push_to_redis('WARNING', message, kwargs)
|
|||
|
|
|
|||
|
|
def error(self, message, **kwargs):
|
|||
|
|
self.logger.error(message)
|
|||
|
|
self._push_to_redis('ERROR', message, kwargs)
|
|||
|
|
|
|||
|
|
def debug(self, message, **kwargs):
|
|||
|
|
self.logger.debug(message)
|
|||
|
|
self._push_to_redis('DEBUG', message, kwargs)
|
|||
|
|
|
|||
|
|
# 全局日志实例
|
|||
|
|
system_logger = SystemLogger()
|