74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
# logger_utils.py
|
||
|
||
import queue
|
||
import datetime
|
||
from colorama import Fore, Style, init
|
||
|
||
# 初始化 colorama（仅 Windows 需要）
|
||
init(autoreset=True)
|
||
|
||
# 日志等级颜色映射（控制台 + UI）
|
||
# LOG_COLORS → 控制台日志颜色（命令行）
|
||
# LOG_STYLES → UI 日志样式（图形界面）
|
||
LOG_COLORS = {
|
||
'default': Fore.CYAN,
|
||
'info': Fore.CYAN,
|
||
'warning': Fore.YELLOW,
|
||
'error': Fore.RED,
|
||
'trigger': Fore.GREEN,
|
||
}
|
||
|
||
# 【新增】UI 样式映射
|
||
LOG_STYLES = {
|
||
'default': {'color': 'cyan'},
|
||
'info': {'color': 'cyan'},
|
||
'warning': {'color': 'yellow'},
|
||
'error': {'color': 'red'},
|
||
'trigger': {'color': '#28a745'},
|
||
}
|
||
|
||
class Logger:
|
||
def __init__(self):
|
||
self.log_queue = queue.Queue()
|
||
self.append_log_func = None # 主程序的日志回调函数
|
||
|
||
def set_append_log(self, func):
|
||
"""设置主程序中的 append_log 函数"""
|
||
self.append_log_func = func
|
||
|
||
def log(self, message, level='default'):
|
||
timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||
full_message = f"[{timestamp}] {message}"
|
||
|
||
# 控制台带颜色输出
|
||
color = LOG_COLORS.get(level.lower(), Fore.WHITE)
|
||
print(f"{color}{full_message}{Style.RESET_ALL}")
|
||
|
||
# 触发UI更新（如果已注册）
|
||
if self.append_log_func:
|
||
log_type = level.lower()
|
||
style = LOG_STYLES.get(log_type, LOG_STYLES['default'])
|
||
|
||
# 调用主程序的 append_log 方法，只传递纯文本和日志类型
|
||
self.append_log_func(full_message, log_type)
|
||
|
||
def info(self, message): self.log(message, 'info')
|
||
def warning(self, message): self.log(message, 'warning')
|
||
def error(self, message): self.log(message, 'error')
|
||
def trigger(self, message): self.log(message, 'trigger')
|
||
def debug(self, message): self.log(message, 'debug')
|
||
|
||
|
||
# 全局日志实例
|
||
global_logger = Logger()
|
||
|
||
# 便捷方法，方便其他模块直接使用
|
||
def setup_logger(append_log_func):
|
||
global_logger.set_append_log(append_log_func)
|
||
|
||
def log_info(msg): global_logger.info(msg)
|
||
def log_warning(msg): global_logger.warning(msg)
|
||
def log_error(msg): global_logger.error(msg)
|
||
def log_trigger(msg): global_logger.trigger(msg)
|
||
def log_debug(msg): global_logger.debug(msg)
|