""" 日志模块 Python: ========================= from umi_log import logger logger.debug("调试信息") logger.info("普通信息") logger.warning("警告信息") logger.error("错误信息")) logger.critical("严重错误信息") # exc_info 只能在 except 块中开启 logger.error("错误信息", exc_info=True, stack_info=True) # 覆盖 LogRecord 的属性 logger.debug("信息", extra={"cover": {"filename": "test.txt", "lineno": 999}} Qml: ========================= console.log("调试信息") console.info("普通信息") console.warn("警告信息") console.error("错误信息") console.trace() // 堆栈信息，级别debug，含函数名、文件名、行号 """ import os import sys import json import logging from datetime import datetime from logging.handlers import RotatingFileHandler from logging import LogRecord # 保存的日志级别，可在UI修改 Save_Log_Level: int = logging.ERROR # 日志保存目录 Logs_Dir = "./logs" Logs_Dir = os.path.abspath(Logs_Dir) # 日志级别，对应的int值由小到大 _Log_Levels = { "DEBUG": logging.DEBUG, "INFO": logging.INFO, "WARNING": logging.WARNING, "ERROR": logging.ERROR, "CRITICAL": logging.CRITICAL, "NONE": logging.CRITICAL + 10, # 最大，表示不记录日志 } # 忽略日志列表，屏蔽QT框架自发产生的一些日志。（目前仅对 get_qt_message_handler 生效） Log_Ignore_List = [ "Retrying to obtain clipboard.", # https://bugreports.qt.io/browse/QTBUG-97930 "Unable to obtain clipboard.", ] # 覆盖过滤器 class _CoverFilter(logging.Filter): def filter(self, record: LogRecord): try: # 提取自定义信息，覆盖给 record cover = record.__dict__.get("cover", {}) for k, v in cover.items(): if hasattr(record, k): setattr(record, k, v) return True except Exception: logger.error("日志过滤错误", exc_info=True, stack_info=True) return False # 简化日志级别符号的格式化器 class _LevelFormatter(logging.Formatter): # 定义日志级别和对应符号的映射 LEVEL_SYMBOLS = { "DEBUG": " ", "INFO": "√", "WARNING": "?", "ERROR": "×", "CRITICAL": "×××", } def format(self, record): # 获取符号，如果没有定义则使用默认级别名称 levelname = record.levelname record.levelsymbol = self.LEVEL_SYMBOLS.get(levelname, levelname) return super().format(record) # json 日志文件处理器 class _JsonRotatingFileHandler(RotatingFileHandler): # 日志信息转字典 def _record_to_dict(self, record: LogRecord): # 时间戳格式化 dt_object = datetime.fromtimestamp(record.created) formatted_time = dt_object.strftime("%Y-%m-%d %H:%M:%S.%f") # 构造消息字典 log_dict = { # 时间 "time": formatted_time, # 日志级别 ( DEBUG, INFO, WARNING, ERROR, CRITICAL ) "level": record.levelname, # 日志消息 "message": record.getMessage(), # ===== # 代码所在文件 "filename": record.filename, # 代码行号 "lineno": record.lineno, # 模块名 "module": record.module, # 函数名 "funcName": record.funcName, # 异常信息，需在 except 块中开启 exc_info=True "exc_text": record.exc_text, # 堆栈信息，需开启 stack_info=True "stack_info": record.stack_info, # ===== # 线程标识符 "thread": record.thread, # 线程名称 "threadName": record.threadName, # 进程标识符 "process": record.process, # 进程名称 "processName": record.processName, # 日志记录器的名称 "name": record.name, } return log_dict # 发送日志 def emit(self, record: LogRecord): # 跳过忽略等级 if record.levelno < Save_Log_Level: return # 检查文件大小并进行轮转 if self.shouldRollover(record): self.doRollover() # 日志信息转字典 try: log_dict = self._record_to_dict(record) except Exception: self.handleError(record) # 输出到日志文件 try: with open(self.baseFilename, "a", encoding=self.encoding) as f: json.dump(log_dict, f, ensure_ascii=False) f.write("\n") except Exception: self.handleError(record) # TODO: 输出到UI界面 # 日志记录器 管理类 class _LogManager: @staticmethod # 控制台处理器 def _get_console_handler(): # 显式规定输出到 stderr ，避免干涉命令行使用 console_handler = logging.StreamHandler(sys.stderr) console_handler.setLevel(logging.DEBUG) fmt = "%(asctime)s %(levelsymbol)s %(funcName)s | %(message)s" formatter = _LevelFormatter(fmt, datefmt="%H:%M:%S") # 使用自定义格式化器 console_handler.setFormatter(formatter) return console_handler @staticmethod # json处理器，输出到本地文件及UI def _get_json_handler(): # 确保日志目录存在 if not os.path.exists(Logs_Dir): os.makedirs(Logs_Dir) # 获取当前日期 current_date = datetime.now().strftime("%Y-%m-%d") # 构造日志文件路径 log_file = os.path.join(Logs_Dir, f"log_{current_date}.jsonl.txt") # 创建json处理器 json_handler = _JsonRotatingFileHandler( log_file, mode="a", # 追加写入 maxBytes=10485760, # 单个文件最大：10MB backupCount=3, # 文件备份数量 encoding="utf-8", # 文件编码 delay=True, # 延迟创建文件 ) json_handler.setLevel(logging.DEBUG) return json_handler @staticmethod def create_logger(name): """创建并返回一个新的日志记录器。""" logger = logging.getLogger(name) logger.addFilter(_CoverFilter()) # 添加覆盖过滤器 logger.setLevel(logging.DEBUG) logger.addHandler(_LogManager._get_console_handler()) logger.addHandler(_LogManager._get_json_handler()) return logger # 全局单例日志记录器 logger = _LogManager.create_logger("Umi-OCR") # 获取 QT 日志重定向器 def get_qt_message_handler(): # 确保在初次调用时才导入QT模块 from PySide2.QtCore import QtMsgType, QMessageLogContext # 处理 QT (QML) 抛出的日志 def qt_message_handler(mode: QtMsgType, context: QMessageLogContext, msg: str): try: if msg in Log_Ignore_List: return # 提取信息 filepath = getattr(context, "file", "?") if filepath: filename = os.path.basename(filepath) else: filepath = filename = "?" funcName = getattr(context, "function", "?") if not funcName: # 匿名函数 funcName = r"()=>{}" # 覆盖字典 extra = { "cover": { "category": getattr(context, "category", "?"), "filename": filename, "funcName": funcName, "lineno": getattr(context, "line", "?"), "version": getattr(context, "version", "?"), "module": "qml", } } if mode == QtMsgType.QtDebugMsg: logger.debug(msg, extra=extra) elif mode == QtMsgType.QtInfoMsg: logger.info(msg, extra=extra) elif mode == QtMsgType.QtWarningMsg: logger.warning(msg, extra=extra) elif mode == QtMsgType.QtCriticalMsg: logger.error(msg, extra=extra) elif mode == QtMsgType.QtFatalMsg: logger.critical(msg, extra=extra) except Exception: logger.warning( "qt_message_handler error", exc_info=True, stack_info=True, ) return qt_message_handler # 更改保存的日志级别，成功返回T def change_save_log_level(levelname): global Save_Log_Level if levelname in _Log_Levels.keys(): Save_Log_Level = _Log_Levels[levelname] logger.info(f"设置保存日志级别： {levelname}") return True logger.error(f"设置保存日志级别 {levelname} 失败。") return False # 打开日志保存目录 def open_logs_dir(): os.startfile(Logs_Dir)