fix：matplotlib炸飞主程序
This commit is contained in:
SengokuCola
@@ -1,347 +1,378 @@
|
||||
import customtkinter as ctk
|
||||
import subprocess
|
||||
import threading
|
||||
import queue
|
||||
import re
|
||||
import os
|
||||
import signal
|
||||
from collections import deque
|
||||
# import customtkinter as ctk
|
||||
# import subprocess
|
||||
# import threading
|
||||
# import queue
|
||||
# import re
|
||||
# import os
|
||||
# import signal
|
||||
# from collections import deque
|
||||
# import sys
|
||||
|
||||
# 设置应用的外观模式和默认颜色主题
|
||||
ctk.set_appearance_mode("dark")
|
||||
ctk.set_default_color_theme("blue")
|
||||
# # 设置应用的外观模式和默认颜色主题
|
||||
# ctk.set_appearance_mode("dark")
|
||||
# ctk.set_default_color_theme("blue")
|
||||
|
||||
|
||||
class LogViewerApp(ctk.CTk):
|
||||
"""日志查看器应用的主类，继承自customtkinter的CTk类"""
|
||||
# class LogViewerApp(ctk.CTk):
|
||||
# """日志查看器应用的主类，继承自customtkinter的CTk类"""
|
||||
|
||||
def __init__(self):
|
||||
"""初始化日志查看器应用的界面和状态"""
|
||||
super().__init__()
|
||||
self.title("日志查看器")
|
||||
self.geometry("1200x800")
|
||||
# def __init__(self):
|
||||
# """初始化日志查看器应用的界面和状态"""
|
||||
# super().__init__()
|
||||
# self.title("日志查看器")
|
||||
# self.geometry("1200x800")
|
||||
|
||||
# 初始化进程、日志队列、日志数据等变量
|
||||
self.process = None
|
||||
self.log_queue = queue.Queue()
|
||||
self.log_data = deque(maxlen=10000) # 使用固定长度队列
|
||||
self.available_levels = set()
|
||||
self.available_modules = set()
|
||||
self.sorted_modules = []
|
||||
self.module_checkboxes = {} # 存储模块复选框的字典
|
||||
# # 标记GUI是否运行中
|
||||
# self.is_running = True
|
||||
|
||||
# 日志颜色配置
|
||||
self.color_config = {
|
||||
"time": "#888888",
|
||||
"DEBUG": "#2196F3",
|
||||
"INFO": "#4CAF50",
|
||||
"WARNING": "#FF9800",
|
||||
"ERROR": "#F44336",
|
||||
"module": "#D4D0AB",
|
||||
"default": "#FFFFFF",
|
||||
}
|
||||
# # 程序关闭时的清理操作
|
||||
# self.protocol("WM_DELETE_WINDOW", self._on_closing)
|
||||
|
||||
# 列可见性配置
|
||||
self.column_visibility = {"show_time": True, "show_level": True, "show_module": True}
|
||||
# # 初始化进程、日志队列、日志数据等变量
|
||||
# self.process = None
|
||||
# self.log_queue = queue.Queue()
|
||||
# self.log_data = deque(maxlen=10000) # 使用固定长度队列
|
||||
# self.available_levels = set()
|
||||
# self.available_modules = set()
|
||||
# self.sorted_modules = []
|
||||
# self.module_checkboxes = {} # 存储模块复选框的字典
|
||||
|
||||
# 选中的日志等级和模块
|
||||
self.selected_levels = set()
|
||||
self.selected_modules = set()
|
||||
# # 日志颜色配置
|
||||
# self.color_config = {
|
||||
# "time": "#888888",
|
||||
# "DEBUG": "#2196F3",
|
||||
# "INFO": "#4CAF50",
|
||||
# "WARNING": "#FF9800",
|
||||
# "ERROR": "#F44336",
|
||||
# "module": "#D4D0AB",
|
||||
# "default": "#FFFFFF",
|
||||
# }
|
||||
|
||||
# 创建界面组件并启动日志队列处理
|
||||
self.create_widgets()
|
||||
self.after(100, self.process_log_queue)
|
||||
# # 列可见性配置
|
||||
# self.column_visibility = {"show_time": True, "show_level": True, "show_module": True}
|
||||
|
||||
def create_widgets(self):
|
||||
"""创建应用界面的各个组件"""
|
||||
self.grid_columnconfigure(0, weight=1)
|
||||
self.grid_rowconfigure(1, weight=1)
|
||||
# # 选中的日志等级和模块
|
||||
# self.selected_levels = set()
|
||||
# self.selected_modules = set()
|
||||
|
||||
# 控制面板
|
||||
control_frame = ctk.CTkFrame(self)
|
||||
control_frame.grid(row=0, column=0, sticky="ew", padx=10, pady=5)
|
||||
# # 创建界面组件并启动日志队列处理
|
||||
# self.create_widgets()
|
||||
# self.after(100, self.process_log_queue)
|
||||
|
||||
self.start_btn = ctk.CTkButton(control_frame, text="启动", command=self.start_process)
|
||||
self.start_btn.pack(side="left", padx=5)
|
||||
# def create_widgets(self):
|
||||
# """创建应用界面的各个组件"""
|
||||
# self.grid_columnconfigure(0, weight=1)
|
||||
# self.grid_rowconfigure(1, weight=1)
|
||||
|
||||
self.stop_btn = ctk.CTkButton(control_frame, text="停止", command=self.stop_process, state="disabled")
|
||||
self.stop_btn.pack(side="left", padx=5)
|
||||
# # 控制面板
|
||||
# control_frame = ctk.CTkFrame(self)
|
||||
# control_frame.grid(row=0, column=0, sticky="ew", padx=10, pady=5)
|
||||
|
||||
self.clear_btn = ctk.CTkButton(control_frame, text="清屏", command=self.clear_logs)
|
||||
self.clear_btn.pack(side="left", padx=5)
|
||||
# self.start_btn = ctk.CTkButton(control_frame, text="启动", command=self.start_process)
|
||||
# self.start_btn.pack(side="left", padx=5)
|
||||
|
||||
column_filter_frame = ctk.CTkFrame(control_frame)
|
||||
column_filter_frame.pack(side="left", padx=20)
|
||||
# self.stop_btn = ctk.CTkButton(control_frame, text="停止", command=self.stop_process, state="disabled")
|
||||
# self.stop_btn.pack(side="left", padx=5)
|
||||
|
||||
self.time_check = ctk.CTkCheckBox(column_filter_frame, text="显示时间", command=self.refresh_logs)
|
||||
self.time_check.pack(side="left", padx=5)
|
||||
self.time_check.select()
|
||||
# self.clear_btn = ctk.CTkButton(control_frame, text="清屏", command=self.clear_logs)
|
||||
# self.clear_btn.pack(side="left", padx=5)
|
||||
|
||||
self.level_check = ctk.CTkCheckBox(column_filter_frame, text="显示等级", command=self.refresh_logs)
|
||||
self.level_check.pack(side="left", padx=5)
|
||||
self.level_check.select()
|
||||
# column_filter_frame = ctk.CTkFrame(control_frame)
|
||||
# column_filter_frame.pack(side="left", padx=20)
|
||||
|
||||
self.module_check = ctk.CTkCheckBox(column_filter_frame, text="显示模块", command=self.refresh_logs)
|
||||
self.module_check.pack(side="left", padx=5)
|
||||
self.module_check.select()
|
||||
# self.time_check = ctk.CTkCheckBox(column_filter_frame, text="显示时间", command=self.refresh_logs)
|
||||
# self.time_check.pack(side="left", padx=5)
|
||||
# self.time_check.select()
|
||||
|
||||
# 筛选面板
|
||||
filter_frame = ctk.CTkFrame(self)
|
||||
filter_frame.grid(row=0, column=1, rowspan=2, sticky="ns", padx=5)
|
||||
# self.level_check = ctk.CTkCheckBox(column_filter_frame, text="显示等级", command=self.refresh_logs)
|
||||
# self.level_check.pack(side="left", padx=5)
|
||||
# self.level_check.select()
|
||||
|
||||
ctk.CTkLabel(filter_frame, text="日志等级筛选").pack(pady=5)
|
||||
self.level_scroll = ctk.CTkScrollableFrame(filter_frame, width=150, height=200)
|
||||
self.level_scroll.pack(fill="both", expand=True, padx=5)
|
||||
# self.module_check = ctk.CTkCheckBox(column_filter_frame, text="显示模块", command=self.refresh_logs)
|
||||
# self.module_check.pack(side="left", padx=5)
|
||||
# self.module_check.select()
|
||||
|
||||
ctk.CTkLabel(filter_frame, text="模块筛选").pack(pady=5)
|
||||
self.module_filter_entry = ctk.CTkEntry(filter_frame, placeholder_text="输入模块过滤词")
|
||||
self.module_filter_entry.pack(pady=5)
|
||||
self.module_filter_entry.bind("<KeyRelease>", self.update_module_filter)
|
||||
# # 筛选面板
|
||||
# filter_frame = ctk.CTkFrame(self)
|
||||
# filter_frame.grid(row=0, column=1, rowspan=2, sticky="ns", padx=5)
|
||||
|
||||
self.module_scroll = ctk.CTkScrollableFrame(filter_frame, width=300, height=200)
|
||||
self.module_scroll.pack(fill="both", expand=True, padx=5)
|
||||
# ctk.CTkLabel(filter_frame, text="日志等级筛选").pack(pady=5)
|
||||
# self.level_scroll = ctk.CTkScrollableFrame(filter_frame, width=150, height=200)
|
||||
# self.level_scroll.pack(fill="both", expand=True, padx=5)
|
||||
|
||||
self.log_text = ctk.CTkTextbox(self, wrap="word")
|
||||
self.log_text.grid(row=1, column=0, sticky="nsew", padx=10, pady=5)
|
||||
# ctk.CTkLabel(filter_frame, text="模块筛选").pack(pady=5)
|
||||
# self.module_filter_entry = ctk.CTkEntry(filter_frame, placeholder_text="输入模块过滤词")
|
||||
# self.module_filter_entry.pack(pady=5)
|
||||
# self.module_filter_entry.bind("<KeyRelease>", self.update_module_filter)
|
||||
|
||||
self.init_text_tags()
|
||||
# self.module_scroll = ctk.CTkScrollableFrame(filter_frame, width=300, height=200)
|
||||
# self.module_scroll.pack(fill="both", expand=True, padx=5)
|
||||
|
||||
def update_module_filter(self, event):
|
||||
"""根据模块过滤词更新模块复选框的显示"""
|
||||
filter_text = self.module_filter_entry.get().strip().lower()
|
||||
for module, checkbox in self.module_checkboxes.items():
|
||||
if filter_text in module.lower():
|
||||
checkbox.pack(anchor="w", padx=5, pady=2)
|
||||
else:
|
||||
checkbox.pack_forget()
|
||||
# self.log_text = ctk.CTkTextbox(self, wrap="word")
|
||||
# self.log_text.grid(row=1, column=0, sticky="nsew", padx=10, pady=5)
|
||||
|
||||
def update_filters(self, level, module):
|
||||
"""更新日志等级和模块的筛选器"""
|
||||
if level not in self.available_levels:
|
||||
self.available_levels.add(level)
|
||||
self.add_checkbox(self.level_scroll, level, "level")
|
||||
# self.init_text_tags()
|
||||
|
||||
module_key = self.get_module_key(module)
|
||||
if module_key not in self.available_modules:
|
||||
self.available_modules.add(module_key)
|
||||
self.sorted_modules = sorted(self.available_modules, key=lambda x: x.lower())
|
||||
self.rebuild_module_checkboxes()
|
||||
# def update_module_filter(self, event):
|
||||
# """根据模块过滤词更新模块复选框的显示"""
|
||||
# filter_text = self.module_filter_entry.get().strip().lower()
|
||||
# for module, checkbox in self.module_checkboxes.items():
|
||||
# if filter_text in module.lower():
|
||||
# checkbox.pack(anchor="w", padx=5, pady=2)
|
||||
# else:
|
||||
# checkbox.pack_forget()
|
||||
|
||||
def rebuild_module_checkboxes(self):
|
||||
"""重新构建模块复选框"""
|
||||
# 清空现有复选框
|
||||
for widget in self.module_scroll.winfo_children():
|
||||
widget.destroy()
|
||||
self.module_checkboxes.clear()
|
||||
# def update_filters(self, level, module):
|
||||
# """更新日志等级和模块的筛选器"""
|
||||
# if level not in self.available_levels:
|
||||
# self.available_levels.add(level)
|
||||
# self.add_checkbox(self.level_scroll, level, "level")
|
||||
|
||||
# 重建排序后的复选框
|
||||
for module in self.sorted_modules:
|
||||
self.add_checkbox(self.module_scroll, module, "module")
|
||||
# module_key = self.get_module_key(module)
|
||||
# if module_key not in self.available_modules:
|
||||
# self.available_modules.add(module_key)
|
||||
# self.sorted_modules = sorted(self.available_modules, key=lambda x: x.lower())
|
||||
# self.rebuild_module_checkboxes()
|
||||
|
||||
def add_checkbox(self, parent, text, type_):
|
||||
"""在指定父组件中添加复选框"""
|
||||
# def rebuild_module_checkboxes(self):
|
||||
# """重新构建模块复选框"""
|
||||
# # 清空现有复选框
|
||||
# for widget in self.module_scroll.winfo_children():
|
||||
# widget.destroy()
|
||||
# self.module_checkboxes.clear()
|
||||
|
||||
def update_filter():
|
||||
current = cb.get()
|
||||
if type_ == "level":
|
||||
(self.selected_levels.add if current else self.selected_levels.discard)(text)
|
||||
else:
|
||||
(self.selected_modules.add if current else self.selected_modules.discard)(text)
|
||||
self.refresh_logs()
|
||||
# # 重建排序后的复选框
|
||||
# for module in self.sorted_modules:
|
||||
# self.add_checkbox(self.module_scroll, module, "module")
|
||||
|
||||
cb = ctk.CTkCheckBox(parent, text=text, command=update_filter)
|
||||
cb.select() # 初始选中
|
||||
# def add_checkbox(self, parent, text, type_):
|
||||
# """在指定父组件中添加复选框"""
|
||||
|
||||
# 手动同步初始状态到集合（关键修复）
|
||||
if type_ == "level":
|
||||
self.selected_levels.add(text)
|
||||
else:
|
||||
self.selected_modules.add(text)
|
||||
# def update_filter():
|
||||
# current = cb.get()
|
||||
# if type_ == "level":
|
||||
# (self.selected_levels.add if current else self.selected_levels.discard)(text)
|
||||
# else:
|
||||
# (self.selected_modules.add if current else self.selected_modules.discard)(text)
|
||||
# self.refresh_logs()
|
||||
|
||||
if type_ == "module":
|
||||
self.module_checkboxes[text] = cb
|
||||
cb.pack(anchor="w", padx=5, pady=2)
|
||||
return cb
|
||||
# cb = ctk.CTkCheckBox(parent, text=text, command=update_filter)
|
||||
# cb.select() # 初始选中
|
||||
|
||||
def check_filter(self, entry):
|
||||
"""检查日志条目是否符合当前筛选条件"""
|
||||
level_ok = not self.selected_levels or entry["level"] in self.selected_levels
|
||||
module_key = self.get_module_key(entry["module"])
|
||||
module_ok = not self.selected_modules or module_key in self.selected_modules
|
||||
return level_ok and module_ok
|
||||
# # 手动同步初始状态到集合（关键修复）
|
||||
# if type_ == "level":
|
||||
# self.selected_levels.add(text)
|
||||
# else:
|
||||
# self.selected_modules.add(text)
|
||||
|
||||
def init_text_tags(self):
|
||||
"""初始化日志文本的颜色标签"""
|
||||
for tag, color in self.color_config.items():
|
||||
self.log_text.tag_config(tag, foreground=color)
|
||||
self.log_text.tag_config("default", foreground=self.color_config["default"])
|
||||
# if type_ == "module":
|
||||
# self.module_checkboxes[text] = cb
|
||||
# cb.pack(anchor="w", padx=5, pady=2)
|
||||
# return cb
|
||||
|
||||
def start_process(self):
|
||||
"""启动日志进程并开始读取输出"""
|
||||
self.process = subprocess.Popen(
|
||||
["nb", "run"],
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.STDOUT,
|
||||
text=True,
|
||||
bufsize=1,
|
||||
encoding="utf-8",
|
||||
errors="ignore",
|
||||
)
|
||||
self.start_btn.configure(state="disabled")
|
||||
self.stop_btn.configure(state="normal")
|
||||
threading.Thread(target=self.read_output, daemon=True).start()
|
||||
# def check_filter(self, entry):
|
||||
# """检查日志条目是否符合当前筛选条件"""
|
||||
# level_ok = not self.selected_levels or entry["level"] in self.selected_levels
|
||||
# module_key = self.get_module_key(entry["module"])
|
||||
# module_ok = not self.selected_modules or module_key in self.selected_modules
|
||||
# return level_ok and module_ok
|
||||
|
||||
def stop_process(self):
|
||||
"""停止日志进程并清理相关资源"""
|
||||
if self.process:
|
||||
try:
|
||||
if hasattr(self.process, "pid"):
|
||||
if os.name == "nt":
|
||||
subprocess.run(
|
||||
["taskkill", "/F", "/T", "/PID", str(self.process.pid)], check=True, capture_output=True
|
||||
)
|
||||
else:
|
||||
os.killpg(os.getpgid(self.process.pid), signal.SIGTERM)
|
||||
except (subprocess.CalledProcessError, ProcessLookupError, OSError) as e:
|
||||
print(f"终止进程失败: {e}")
|
||||
finally:
|
||||
self.process = None
|
||||
self.log_queue.queue.clear()
|
||||
self.start_btn.configure(state="normal")
|
||||
self.stop_btn.configure(state="disabled")
|
||||
self.refresh_logs()
|
||||
# def init_text_tags(self):
|
||||
# """初始化日志文本的颜色标签"""
|
||||
# for tag, color in self.color_config.items():
|
||||
# self.log_text.tag_config(tag, foreground=color)
|
||||
# self.log_text.tag_config("default", foreground=self.color_config["default"])
|
||||
|
||||
def read_output(self):
|
||||
"""读取日志进程的输出并放入队列"""
|
||||
try:
|
||||
while self.process and self.process.poll() is None:
|
||||
line = self.process.stdout.readline()
|
||||
if line:
|
||||
self.log_queue.put(line)
|
||||
else:
|
||||
break # 避免空循环
|
||||
self.process.stdout.close() # 确保关闭文件描述符
|
||||
except ValueError: # 处理可能的I/O操作异常
|
||||
pass
|
||||
# def start_process(self):
|
||||
# """启动日志进程并开始读取输出"""
|
||||
# self.process = subprocess.Popen(
|
||||
# ["nb", "run"],
|
||||
# stdout=subprocess.PIPE,
|
||||
# stderr=subprocess.STDOUT,
|
||||
# text=True,
|
||||
# bufsize=1,
|
||||
# encoding="utf-8",
|
||||
# errors="ignore",
|
||||
# )
|
||||
# self.start_btn.configure(state="disabled")
|
||||
# self.stop_btn.configure(state="normal")
|
||||
# threading.Thread(target=self.read_output, daemon=True).start()
|
||||
|
||||
def process_log_queue(self):
|
||||
"""处理日志队列中的日志条目"""
|
||||
while not self.log_queue.empty():
|
||||
line = self.log_queue.get()
|
||||
self.process_log_line(line)
|
||||
self.after(100, self.process_log_queue)
|
||||
# def stop_process(self):
|
||||
# """停止日志进程并清理相关资源"""
|
||||
# if self.process:
|
||||
# try:
|
||||
# if hasattr(self.process, "pid"):
|
||||
# if os.name == "nt":
|
||||
# subprocess.run(
|
||||
# ["taskkill", "/F", "/T", "/PID", str(self.process.pid)], check=True, capture_output=True
|
||||
# )
|
||||
# else:
|
||||
# os.killpg(os.getpgid(self.process.pid), signal.SIGTERM)
|
||||
# except (subprocess.CalledProcessError, ProcessLookupError, OSError) as e:
|
||||
# print(f"终止进程失败: {e}")
|
||||
# finally:
|
||||
# self.process = None
|
||||
# self.log_queue.queue.clear()
|
||||
# self.start_btn.configure(state="normal")
|
||||
# self.stop_btn.configure(state="disabled")
|
||||
# self.refresh_logs()
|
||||
|
||||
def process_log_line(self, line):
|
||||
"""解析单行日志并更新日志数据和筛选器"""
|
||||
match = re.match(
|
||||
r"""^
|
||||
(?:(?P<time>\d{2}:\d{2}(?::\d{2})?)\s*\|\s*)?
|
||||
(?P<level>\w+)\s*\|\s*
|
||||
(?P<module>.*?)
|
||||
\s*[-|]\s*
|
||||
(?P<message>.*)
|
||||
$""",
|
||||
line.strip(),
|
||||
re.VERBOSE,
|
||||
)
|
||||
# def read_output(self):
|
||||
# """读取日志进程的输出并放入队列"""
|
||||
# try:
|
||||
# while self.process and self.process.poll() is None and self.is_running:
|
||||
# line = self.process.stdout.readline()
|
||||
# if line:
|
||||
# self.log_queue.put(line)
|
||||
# else:
|
||||
# break # 避免空循环
|
||||
# self.process.stdout.close() # 确保关闭文件描述符
|
||||
# except ValueError: # 处理可能的I/O操作异常
|
||||
# pass
|
||||
|
||||
if match:
|
||||
groups = match.groupdict()
|
||||
time = groups.get("time", "")
|
||||
level = groups.get("level", "OTHER")
|
||||
module = groups.get("module", "UNKNOWN").strip()
|
||||
message = groups.get("message", "").strip()
|
||||
raw_line = line
|
||||
else:
|
||||
time, level, module, message = "", "OTHER", "UNKNOWN", line
|
||||
raw_line = line
|
||||
# def process_log_queue(self):
|
||||
# """处理日志队列中的日志条目"""
|
||||
# while not self.log_queue.empty():
|
||||
# line = self.log_queue.get()
|
||||
# self.process_log_line(line)
|
||||
|
||||
self.update_filters(level, module)
|
||||
log_entry = {"raw": raw_line, "time": time, "level": level, "module": module, "message": message}
|
||||
self.log_data.append(log_entry)
|
||||
# # 仅在GUI仍在运行时继续处理队列
|
||||
# if self.is_running:
|
||||
# self.after(100, self.process_log_queue)
|
||||
|
||||
if self.check_filter(log_entry):
|
||||
self.display_log(log_entry)
|
||||
# def process_log_line(self, line):
|
||||
# """解析单行日志并更新日志数据和筛选器"""
|
||||
# match = re.match(
|
||||
# r"""^
|
||||
# (?:(?P<time>\d{2}:\d{2}(?::\d{2})?)\s*\|\s*)?
|
||||
# (?P<level>\w+)\s*\|\s*
|
||||
# (?P<module>.*?)
|
||||
# \s*[-|]\s*
|
||||
# (?P<message>.*)
|
||||
# $""",
|
||||
# line.strip(),
|
||||
# re.VERBOSE,
|
||||
# )
|
||||
|
||||
def get_module_key(self, module_name):
|
||||
"""获取模块名称的标准化键"""
|
||||
cleaned = module_name.strip()
|
||||
return re.sub(r":\d+$", "", cleaned)
|
||||
# if match:
|
||||
# groups = match.groupdict()
|
||||
# time = groups.get("time", "")
|
||||
# level = groups.get("level", "OTHER")
|
||||
# module = groups.get("module", "UNKNOWN").strip()
|
||||
# message = groups.get("message", "").strip()
|
||||
# raw_line = line
|
||||
# else:
|
||||
# time, level, module, message = "", "OTHER", "UNKNOWN", line
|
||||
# raw_line = line
|
||||
|
||||
def display_log(self, entry):
|
||||
"""在日志文本框中显示日志条目"""
|
||||
parts = []
|
||||
tags = []
|
||||
# self.update_filters(level, module)
|
||||
# log_entry = {"raw": raw_line, "time": time, "level": level, "module": module, "message": message}
|
||||
# self.log_data.append(log_entry)
|
||||
|
||||
if self.column_visibility["show_time"] and entry["time"]:
|
||||
parts.append(f"{entry['time']} ")
|
||||
tags.append("time")
|
||||
# if self.check_filter(log_entry):
|
||||
# self.display_log(log_entry)
|
||||
|
||||
if self.column_visibility["show_level"]:
|
||||
level_tag = entry["level"] if entry["level"] in self.color_config else "default"
|
||||
parts.append(f"{entry['level']:<8} ")
|
||||
tags.append(level_tag)
|
||||
# def get_module_key(self, module_name):
|
||||
# """获取模块名称的标准化键"""
|
||||
# cleaned = module_name.strip()
|
||||
# return re.sub(r":\d+$", "", cleaned)
|
||||
|
||||
if self.column_visibility["show_module"]:
|
||||
parts.append(f"{entry['module']} ")
|
||||
tags.append("module")
|
||||
# def display_log(self, entry):
|
||||
# """在日志文本框中显示日志条目"""
|
||||
# parts = []
|
||||
# tags = []
|
||||
|
||||
parts.append(f"- {entry['message']}\n")
|
||||
tags.append("default")
|
||||
# if self.column_visibility["show_time"] and entry["time"]:
|
||||
# parts.append(f"{entry['time']} ")
|
||||
# tags.append("time")
|
||||
|
||||
self.log_text.configure(state="normal")
|
||||
for part, tag in zip(parts, tags):
|
||||
self.log_text.insert("end", part, tag)
|
||||
self.log_text.see("end")
|
||||
self.log_text.configure(state="disabled")
|
||||
# if self.column_visibility["show_level"]:
|
||||
# level_tag = entry["level"] if entry["level"] in self.color_config else "default"
|
||||
# parts.append(f"{entry['level']:<8} ")
|
||||
# tags.append(level_tag)
|
||||
|
||||
def refresh_logs(self):
|
||||
"""刷新日志显示，根据筛选条件重新显示日志"""
|
||||
self.column_visibility = {
|
||||
"show_time": self.time_check.get(),
|
||||
"show_level": self.level_check.get(),
|
||||
"show_module": self.module_check.get(),
|
||||
}
|
||||
# if self.column_visibility["show_module"]:
|
||||
# parts.append(f"{entry['module']} ")
|
||||
# tags.append("module")
|
||||
|
||||
self.log_text.configure(state="normal")
|
||||
self.log_text.delete("1.0", "end")
|
||||
# parts.append(f"- {entry['message']}\n")
|
||||
# tags.append("default")
|
||||
|
||||
filtered_logs = [entry for entry in self.log_data if self.check_filter(entry)]
|
||||
# self.log_text.configure(state="normal")
|
||||
# for part, tag in zip(parts, tags):
|
||||
# self.log_text.insert("end", part, tag)
|
||||
# self.log_text.see("end")
|
||||
# self.log_text.configure(state="disabled")
|
||||
|
||||
for entry in filtered_logs:
|
||||
parts = []
|
||||
tags = []
|
||||
# def refresh_logs(self):
|
||||
# """刷新日志显示，根据筛选条件重新显示日志"""
|
||||
# self.column_visibility = {
|
||||
# "show_time": self.time_check.get(),
|
||||
# "show_level": self.level_check.get(),
|
||||
# "show_module": self.module_check.get(),
|
||||
# }
|
||||
|
||||
if self.column_visibility["show_time"] and entry["time"]:
|
||||
parts.append(f"{entry['time']} ")
|
||||
tags.append("time")
|
||||
# self.log_text.configure(state="normal")
|
||||
# self.log_text.delete("1.0", "end")
|
||||
|
||||
if self.column_visibility["show_level"]:
|
||||
level_tag = entry["level"] if entry["level"] in self.color_config else "default"
|
||||
parts.append(f"{entry['level']:<8} ")
|
||||
tags.append(level_tag)
|
||||
# filtered_logs = [entry for entry in self.log_data if self.check_filter(entry)]
|
||||
|
||||
if self.column_visibility["show_module"]:
|
||||
parts.append(f"{entry['module']} ")
|
||||
tags.append("module")
|
||||
# for entry in filtered_logs:
|
||||
# parts = []
|
||||
# tags = []
|
||||
|
||||
parts.append(f"- {entry['message']}\n")
|
||||
tags.append("default")
|
||||
# if self.column_visibility["show_time"] and entry["time"]:
|
||||
# parts.append(f"{entry['time']} ")
|
||||
# tags.append("time")
|
||||
|
||||
for part, tag in zip(parts, tags):
|
||||
self.log_text.insert("end", part, tag)
|
||||
# if self.column_visibility["show_level"]:
|
||||
# level_tag = entry["level"] if entry["level"] in self.color_config else "default"
|
||||
# parts.append(f"{entry['level']:<8} ")
|
||||
# tags.append(level_tag)
|
||||
|
||||
self.log_text.see("end")
|
||||
self.log_text.configure(state="disabled")
|
||||
# if self.column_visibility["show_module"]:
|
||||
# parts.append(f"{entry['module']} ")
|
||||
# tags.append("module")
|
||||
|
||||
def clear_logs(self):
|
||||
"""清空日志文本框中的内容"""
|
||||
self.log_text.configure(state="normal")
|
||||
self.log_text.delete("1.0", "end")
|
||||
self.log_text.configure(state="disabled")
|
||||
# parts.append(f"- {entry['message']}\n")
|
||||
# tags.append("default")
|
||||
|
||||
# for part, tag in zip(parts, tags):
|
||||
# self.log_text.insert("end", part, tag)
|
||||
|
||||
# self.log_text.see("end")
|
||||
# self.log_text.configure(state="disabled")
|
||||
|
||||
# def clear_logs(self):
|
||||
# """清空日志文本框中的内容"""
|
||||
# self.log_text.configure(state="normal")
|
||||
# self.log_text.delete("1.0", "end")
|
||||
# self.log_text.configure(state="disabled")
|
||||
|
||||
# def _on_closing(self):
|
||||
# """处理窗口关闭事件，安全清理资源"""
|
||||
# # 标记GUI已关闭
|
||||
# self.is_running = False
|
||||
|
||||
# # 停止日志进程
|
||||
# self.stop_process()
|
||||
|
||||
# # 安全清理tkinter变量
|
||||
# for attr_name in list(self.__dict__.keys()):
|
||||
# if isinstance(getattr(self, attr_name), (ctk.Variable, ctk.StringVar, ctk.IntVar, ctk.DoubleVar, ctk.BooleanVar)):
|
||||
# try:
|
||||
# var = getattr(self, attr_name)
|
||||
# var.set(None)
|
||||
# except Exception:
|
||||
# pass
|
||||
# setattr(self, attr_name, None)
|
||||
|
||||
# self.quit()
|
||||
# sys.exit(0)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# 启动日志查看器应用
|
||||
app = LogViewerApp()
|
||||
app.mainloop()
|
||||
# if __name__ == "__main__":
|
||||
# # 启动日志查看器应用
|
||||
# app = LogViewerApp()
|
||||
# app.mainloop()
|
||||
|
||||
@@ -1,320 +1,342 @@
|
||||
import os
|
||||
import queue
|
||||
import sys
|
||||
import threading
|
||||
import time
|
||||
from datetime import datetime
|
||||
from typing import Dict, List
|
||||
from typing import Optional
|
||||
# import os
|
||||
# import queue
|
||||
# import sys
|
||||
# import threading
|
||||
# import time
|
||||
# from datetime import datetime
|
||||
# from typing import Dict, List
|
||||
# from typing import Optional
|
||||
|
||||
sys.path.insert(0, sys.path[0] + "/../")
|
||||
sys.path.insert(0, sys.path[0] + "/../")
|
||||
from src.common.logger import get_module_logger
|
||||
# sys.path.insert(0, sys.path[0] + "/../")
|
||||
# sys.path.insert(0, sys.path[0] + "/../")
|
||||
# from src.common.logger import get_module_logger
|
||||
|
||||
import customtkinter as ctk
|
||||
from dotenv import load_dotenv
|
||||
# import customtkinter as ctk
|
||||
# from dotenv import load_dotenv
|
||||
|
||||
logger = get_module_logger("gui")
|
||||
# logger = get_module_logger("gui")
|
||||
|
||||
# 获取当前文件的目录
|
||||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
# 获取项目根目录
|
||||
root_dir = os.path.abspath(os.path.join(current_dir, "..", ".."))
|
||||
sys.path.insert(0, root_dir)
|
||||
from src.common.database import db # noqa: E402
|
||||
# # 获取当前文件的目录
|
||||
# current_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
# # 获取项目根目录
|
||||
# root_dir = os.path.abspath(os.path.join(current_dir, "..", ".."))
|
||||
# sys.path.insert(0, root_dir)
|
||||
# from src.common.database import db # noqa: E402
|
||||
|
||||
# 加载环境变量
|
||||
if os.path.exists(os.path.join(root_dir, ".env.dev")):
|
||||
load_dotenv(os.path.join(root_dir, ".env.dev"))
|
||||
logger.info("成功加载开发环境配置")
|
||||
elif os.path.exists(os.path.join(root_dir, ".env")):
|
||||
load_dotenv(os.path.join(root_dir, ".env"))
|
||||
logger.info("成功加载生产环境配置")
|
||||
else:
|
||||
logger.error("未找到环境配置文件")
|
||||
sys.exit(1)
|
||||
# # 加载环境变量
|
||||
# if os.path.exists(os.path.join(root_dir, ".env.dev")):
|
||||
# load_dotenv(os.path.join(root_dir, ".env.dev"))
|
||||
# logger.info("成功加载开发环境配置")
|
||||
# elif os.path.exists(os.path.join(root_dir, ".env")):
|
||||
# load_dotenv(os.path.join(root_dir, ".env"))
|
||||
# logger.info("成功加载生产环境配置")
|
||||
# else:
|
||||
# logger.error("未找到环境配置文件")
|
||||
# sys.exit(1)
|
||||
|
||||
|
||||
class ReasoningGUI:
|
||||
def __init__(self):
|
||||
# 记录启动时间戳，转换为Unix时间戳
|
||||
self.start_timestamp = datetime.now().timestamp()
|
||||
logger.info(f"程序启动时间戳: {self.start_timestamp}")
|
||||
# class ReasoningGUI:
|
||||
# def __init__(self):
|
||||
# # 记录启动时间戳，转换为Unix时间戳
|
||||
# self.start_timestamp = datetime.now().timestamp()
|
||||
# logger.info(f"程序启动时间戳: {self.start_timestamp}")
|
||||
|
||||
# 设置主题
|
||||
ctk.set_appearance_mode("dark")
|
||||
ctk.set_default_color_theme("blue")
|
||||
# # 设置主题
|
||||
# ctk.set_appearance_mode("dark")
|
||||
# ctk.set_default_color_theme("blue")
|
||||
|
||||
# 创建主窗口
|
||||
self.root = ctk.CTk()
|
||||
self.root.title("麦麦推理")
|
||||
self.root.geometry("800x600")
|
||||
self.root.protocol("WM_DELETE_WINDOW", self._on_closing)
|
||||
# # 创建主窗口
|
||||
# self.root = ctk.CTk()
|
||||
# self.root.title("麦麦推理")
|
||||
# self.root.geometry("800x600")
|
||||
# self.root.protocol("WM_DELETE_WINDOW", self._on_closing)
|
||||
|
||||
# 存储群组数据
|
||||
self.group_data: Dict[str, List[dict]] = {}
|
||||
# # 存储群组数据
|
||||
# self.group_data: Dict[str, List[dict]] = {}
|
||||
|
||||
# 创建更新队列
|
||||
self.update_queue = queue.Queue()
|
||||
# # 创建更新队列
|
||||
# self.update_queue = queue.Queue()
|
||||
|
||||
# 创建主框架
|
||||
self.frame = ctk.CTkFrame(self.root)
|
||||
self.frame.pack(pady=20, padx=20, fill="both", expand=True)
|
||||
# # 创建主框架
|
||||
# self.frame = ctk.CTkFrame(self.root)
|
||||
# self.frame.pack(pady=20, padx=20, fill="both", expand=True)
|
||||
|
||||
# 添加标题
|
||||
self.title = ctk.CTkLabel(self.frame, text="麦麦的脑内所想", font=("Arial", 24))
|
||||
self.title.pack(pady=10, padx=10)
|
||||
# # 添加标题
|
||||
# self.title = ctk.CTkLabel(self.frame, text="麦麦的脑内所想", font=("Arial", 24))
|
||||
# self.title.pack(pady=10, padx=10)
|
||||
|
||||
# 创建左右分栏
|
||||
self.paned = ctk.CTkFrame(self.frame)
|
||||
self.paned.pack(fill="both", expand=True, padx=10, pady=10)
|
||||
# # 创建左右分栏
|
||||
# self.paned = ctk.CTkFrame(self.frame)
|
||||
# self.paned.pack(fill="both", expand=True, padx=10, pady=10)
|
||||
|
||||
# 左侧群组列表
|
||||
self.left_frame = ctk.CTkFrame(self.paned, width=200)
|
||||
self.left_frame.pack(side="left", fill="y", padx=5, pady=5)
|
||||
# # 左侧群组列表
|
||||
# self.left_frame = ctk.CTkFrame(self.paned, width=200)
|
||||
# self.left_frame.pack(side="left", fill="y", padx=5, pady=5)
|
||||
|
||||
self.group_label = ctk.CTkLabel(self.left_frame, text="群组列表", font=("Arial", 16))
|
||||
self.group_label.pack(pady=5)
|
||||
# self.group_label = ctk.CTkLabel(self.left_frame, text="群组列表", font=("Arial", 16))
|
||||
# self.group_label.pack(pady=5)
|
||||
|
||||
# 创建可滚动框架来容纳群组按钮
|
||||
self.group_scroll_frame = ctk.CTkScrollableFrame(self.left_frame, width=180, height=400)
|
||||
self.group_scroll_frame.pack(pady=5, padx=5, fill="both", expand=True)
|
||||
# # 创建可滚动框架来容纳群组按钮
|
||||
# self.group_scroll_frame = ctk.CTkScrollableFrame(self.left_frame, width=180, height=400)
|
||||
# self.group_scroll_frame.pack(pady=5, padx=5, fill="both", expand=True)
|
||||
|
||||
# 存储群组按钮的字典
|
||||
self.group_buttons: Dict[str, ctk.CTkButton] = {}
|
||||
# 当前选中的群组ID
|
||||
self.selected_group_id: Optional[str] = None
|
||||
# # 存储群组按钮的字典
|
||||
# self.group_buttons: Dict[str, ctk.CTkButton] = {}
|
||||
# # 当前选中的群组ID
|
||||
# self.selected_group_id: Optional[str] = None
|
||||
|
||||
# 右侧内容显示
|
||||
self.right_frame = ctk.CTkFrame(self.paned)
|
||||
self.right_frame.pack(side="right", fill="both", expand=True, padx=5, pady=5)
|
||||
# # 右侧内容显示
|
||||
# self.right_frame = ctk.CTkFrame(self.paned)
|
||||
# self.right_frame.pack(side="right", fill="both", expand=True, padx=5, pady=5)
|
||||
|
||||
self.content_label = ctk.CTkLabel(self.right_frame, text="推理内容", font=("Arial", 16))
|
||||
self.content_label.pack(pady=5)
|
||||
# self.content_label = ctk.CTkLabel(self.right_frame, text="推理内容", font=("Arial", 16))
|
||||
# self.content_label.pack(pady=5)
|
||||
|
||||
# 创建富文本显示框
|
||||
self.content_text = ctk.CTkTextbox(self.right_frame, width=500, height=400)
|
||||
self.content_text.pack(pady=5, padx=5, fill="both", expand=True)
|
||||
# # 创建富文本显示框
|
||||
# self.content_text = ctk.CTkTextbox(self.right_frame, width=500, height=400)
|
||||
# self.content_text.pack(pady=5, padx=5, fill="both", expand=True)
|
||||
|
||||
# 配置文本标签 - 只使用颜色
|
||||
self.content_text.tag_config("timestamp", foreground="#888888") # 时间戳使用灰色
|
||||
self.content_text.tag_config("user", foreground="#4CAF50") # 用户名使用绿色
|
||||
self.content_text.tag_config("message", foreground="#2196F3") # 消息使用蓝色
|
||||
self.content_text.tag_config("model", foreground="#9C27B0") # 模型名称使用紫色
|
||||
self.content_text.tag_config("prompt", foreground="#FF9800") # prompt内容使用橙色
|
||||
self.content_text.tag_config("reasoning", foreground="#FF9800") # 推理过程使用橙色
|
||||
self.content_text.tag_config("response", foreground="#E91E63") # 回复使用粉色
|
||||
self.content_text.tag_config("separator", foreground="#666666") # 分隔符使用深灰色
|
||||
# # 配置文本标签 - 只使用颜色
|
||||
# self.content_text.tag_config("timestamp", foreground="#888888") # 时间戳使用灰色
|
||||
# self.content_text.tag_config("user", foreground="#4CAF50") # 用户名使用绿色
|
||||
# self.content_text.tag_config("message", foreground="#2196F3") # 消息使用蓝色
|
||||
# self.content_text.tag_config("model", foreground="#9C27B0") # 模型名称使用紫色
|
||||
# self.content_text.tag_config("prompt", foreground="#FF9800") # prompt内容使用橙色
|
||||
# self.content_text.tag_config("reasoning", foreground="#FF9800") # 推理过程使用橙色
|
||||
# self.content_text.tag_config("response", foreground="#E91E63") # 回复使用粉色
|
||||
# self.content_text.tag_config("separator", foreground="#666666") # 分隔符使用深灰色
|
||||
|
||||
# 底部控制栏
|
||||
self.control_frame = ctk.CTkFrame(self.frame)
|
||||
self.control_frame.pack(fill="x", padx=10, pady=5)
|
||||
# # 底部控制栏
|
||||
# self.control_frame = ctk.CTkFrame(self.frame)
|
||||
# self.control_frame.pack(fill="x", padx=10, pady=5)
|
||||
|
||||
self.clear_button = ctk.CTkButton(self.control_frame, text="清除显示", command=self.clear_display, width=120)
|
||||
self.clear_button.pack(side="left", padx=5)
|
||||
# self.clear_button = ctk.CTkButton(self.control_frame, text="清除显示", command=self.clear_display, width=120)
|
||||
# self.clear_button.pack(side="left", padx=5)
|
||||
|
||||
# 启动自动更新线程
|
||||
self.update_thread = threading.Thread(target=self._auto_update, daemon=True)
|
||||
self.update_thread.start()
|
||||
# # 添加标志，标记GUI是否已关闭
|
||||
# self.is_running = True
|
||||
|
||||
# 启动GUI更新检查
|
||||
self.root.after(100, self._process_queue)
|
||||
# # 启动自动更新线程
|
||||
# self.update_thread = threading.Thread(target=self._auto_update, daemon=True)
|
||||
# self.update_thread.start()
|
||||
|
||||
def _on_closing(self):
|
||||
"""处理窗口关闭事件"""
|
||||
self.root.quit()
|
||||
sys.exit(0)
|
||||
# # 启动GUI更新检查
|
||||
# self.root.after(100, self._process_queue)
|
||||
|
||||
def _process_queue(self):
|
||||
"""处理更新队列中的任务"""
|
||||
try:
|
||||
while True:
|
||||
task = self.update_queue.get_nowait()
|
||||
if task["type"] == "update_group_list":
|
||||
self._update_group_list_gui()
|
||||
elif task["type"] == "update_display":
|
||||
self._update_display_gui(task["group_id"])
|
||||
except queue.Empty:
|
||||
pass
|
||||
finally:
|
||||
# 继续检查队列
|
||||
self.root.after(100, self._process_queue)
|
||||
# def _on_closing(self):
|
||||
# """处理窗口关闭事件"""
|
||||
# # 标记GUI已关闭，防止后台线程继续访问tkinter对象
|
||||
# self.is_running = False
|
||||
|
||||
def _update_group_list_gui(self):
|
||||
"""在主线程中更新群组列表"""
|
||||
# 清除现有按钮
|
||||
for button in self.group_buttons.values():
|
||||
button.destroy()
|
||||
self.group_buttons.clear()
|
||||
# # 安全清理所有可能的tkinter变量
|
||||
# for attr_name in list(self.__dict__.keys()):
|
||||
# if isinstance(getattr(self, attr_name), (ctk.Variable, ctk.StringVar, ctk.IntVar, ctk.DoubleVar, ctk.BooleanVar)):
|
||||
# # 删除变量前安全地将其设置为None
|
||||
# try:
|
||||
# var = getattr(self, attr_name)
|
||||
# var.set(None)
|
||||
# except Exception:
|
||||
# pass
|
||||
# setattr(self, attr_name, None)
|
||||
|
||||
# 创建新的群组按钮
|
||||
for group_id in self.group_data.keys():
|
||||
button = ctk.CTkButton(
|
||||
self.group_scroll_frame,
|
||||
text=f"群号: {group_id}",
|
||||
width=160,
|
||||
height=30,
|
||||
corner_radius=8,
|
||||
command=lambda gid=group_id: self._on_group_select(gid),
|
||||
)
|
||||
button.pack(pady=2, padx=5)
|
||||
self.group_buttons[group_id] = button
|
||||
# # 退出
|
||||
# self.root.quit()
|
||||
# sys.exit(0)
|
||||
|
||||
# 如果有选中的群组，保持其高亮状态
|
||||
if self.selected_group_id and self.selected_group_id in self.group_buttons:
|
||||
self._highlight_selected_group(self.selected_group_id)
|
||||
# def _process_queue(self):
|
||||
# """处理更新队列中的任务"""
|
||||
# try:
|
||||
# while True:
|
||||
# task = self.update_queue.get_nowait()
|
||||
# if task["type"] == "update_group_list":
|
||||
# self._update_group_list_gui()
|
||||
# elif task["type"] == "update_display":
|
||||
# self._update_display_gui(task["group_id"])
|
||||
# except queue.Empty:
|
||||
# pass
|
||||
# finally:
|
||||
# # 继续检查队列，但仅在GUI仍在运行时
|
||||
# if self.is_running:
|
||||
# self.root.after(100, self._process_queue)
|
||||
|
||||
def _on_group_select(self, group_id: str):
|
||||
"""处理群组选择事件"""
|
||||
self._highlight_selected_group(group_id)
|
||||
self._update_display_gui(group_id)
|
||||
# def _update_group_list_gui(self):
|
||||
# """在主线程中更新群组列表"""
|
||||
# # 清除现有按钮
|
||||
# for button in self.group_buttons.values():
|
||||
# button.destroy()
|
||||
# self.group_buttons.clear()
|
||||
|
||||
def _highlight_selected_group(self, group_id: str):
|
||||
"""高亮显示选中的群组按钮"""
|
||||
# 重置所有按钮的颜色
|
||||
for gid, button in self.group_buttons.items():
|
||||
if gid == group_id:
|
||||
# 设置选中按钮的颜色
|
||||
button.configure(fg_color="#1E88E5", hover_color="#1976D2")
|
||||
else:
|
||||
# 恢复其他按钮的默认颜色
|
||||
button.configure(fg_color="#2B2B2B", hover_color="#404040")
|
||||
# # 创建新的群组按钮
|
||||
# for group_id in self.group_data.keys():
|
||||
# button = ctk.CTkButton(
|
||||
# self.group_scroll_frame,
|
||||
# text=f"群号: {group_id}",
|
||||
# width=160,
|
||||
# height=30,
|
||||
# corner_radius=8,
|
||||
# command=lambda gid=group_id: self._on_group_select(gid),
|
||||
# )
|
||||
# button.pack(pady=2, padx=5)
|
||||
# self.group_buttons[group_id] = button
|
||||
|
||||
self.selected_group_id = group_id
|
||||
# # 如果有选中的群组，保持其高亮状态
|
||||
# if self.selected_group_id and self.selected_group_id in self.group_buttons:
|
||||
# self._highlight_selected_group(self.selected_group_id)
|
||||
|
||||
def _update_display_gui(self, group_id: str):
|
||||
"""在主线程中更新显示内容"""
|
||||
if group_id in self.group_data:
|
||||
self.content_text.delete("1.0", "end")
|
||||
for item in self.group_data[group_id]:
|
||||
# 时间戳
|
||||
time_str = item["time"].strftime("%Y-%m-%d %H:%M:%S")
|
||||
self.content_text.insert("end", f"[{time_str}]\n", "timestamp")
|
||||
# def _on_group_select(self, group_id: str):
|
||||
# """处理群组选择事件"""
|
||||
# self._highlight_selected_group(group_id)
|
||||
# self._update_display_gui(group_id)
|
||||
|
||||
# 用户信息
|
||||
self.content_text.insert("end", "用户: ", "timestamp")
|
||||
self.content_text.insert("end", f"{item.get('user', '未知')}\n", "user")
|
||||
# def _highlight_selected_group(self, group_id: str):
|
||||
# """高亮显示选中的群组按钮"""
|
||||
# # 重置所有按钮的颜色
|
||||
# for gid, button in self.group_buttons.items():
|
||||
# if gid == group_id:
|
||||
# # 设置选中按钮的颜色
|
||||
# button.configure(fg_color="#1E88E5", hover_color="#1976D2")
|
||||
# else:
|
||||
# # 恢复其他按钮的默认颜色
|
||||
# button.configure(fg_color="#2B2B2B", hover_color="#404040")
|
||||
|
||||
# 消息内容
|
||||
self.content_text.insert("end", "消息: ", "timestamp")
|
||||
self.content_text.insert("end", f"{item.get('message', '')}\n", "message")
|
||||
# self.selected_group_id = group_id
|
||||
|
||||
# 模型信息
|
||||
self.content_text.insert("end", "模型: ", "timestamp")
|
||||
self.content_text.insert("end", f"{item.get('model', '')}\n", "model")
|
||||
# def _update_display_gui(self, group_id: str):
|
||||
# """在主线程中更新显示内容"""
|
||||
# if group_id in self.group_data:
|
||||
# self.content_text.delete("1.0", "end")
|
||||
# for item in self.group_data[group_id]:
|
||||
# # 时间戳
|
||||
# time_str = item["time"].strftime("%Y-%m-%d %H:%M:%S")
|
||||
# self.content_text.insert("end", f"[{time_str}]\n", "timestamp")
|
||||
|
||||
# Prompt内容
|
||||
self.content_text.insert("end", "Prompt内容:\n", "timestamp")
|
||||
prompt_text = item.get("prompt", "")
|
||||
if prompt_text and prompt_text.lower() != "none":
|
||||
lines = prompt_text.split("\n")
|
||||
for line in lines:
|
||||
if line.strip():
|
||||
self.content_text.insert("end", " " + line + "\n", "prompt")
|
||||
else:
|
||||
self.content_text.insert("end", " 无Prompt内容\n", "prompt")
|
||||
# # 用户信息
|
||||
# self.content_text.insert("end", "用户: ", "timestamp")
|
||||
# self.content_text.insert("end", f"{item.get('user', '未知')}\n", "user")
|
||||
|
||||
# 推理过程
|
||||
self.content_text.insert("end", "推理过程:\n", "timestamp")
|
||||
reasoning_text = item.get("reasoning", "")
|
||||
if reasoning_text and reasoning_text.lower() != "none":
|
||||
lines = reasoning_text.split("\n")
|
||||
for line in lines:
|
||||
if line.strip():
|
||||
self.content_text.insert("end", " " + line + "\n", "reasoning")
|
||||
else:
|
||||
self.content_text.insert("end", " 无推理过程\n", "reasoning")
|
||||
# # 消息内容
|
||||
# self.content_text.insert("end", "消息: ", "timestamp")
|
||||
# self.content_text.insert("end", f"{item.get('message', '')}\n", "message")
|
||||
|
||||
# 回复内容
|
||||
self.content_text.insert("end", "回复: ", "timestamp")
|
||||
self.content_text.insert("end", f"{item.get('response', '')}\n", "response")
|
||||
# # 模型信息
|
||||
# self.content_text.insert("end", "模型: ", "timestamp")
|
||||
# self.content_text.insert("end", f"{item.get('model', '')}\n", "model")
|
||||
|
||||
# 分隔符
|
||||
self.content_text.insert("end", f"\n{'=' * 50}\n\n", "separator")
|
||||
# # Prompt内容
|
||||
# self.content_text.insert("end", "Prompt内容:\n", "timestamp")
|
||||
# prompt_text = item.get("prompt", "")
|
||||
# if prompt_text and prompt_text.lower() != "none":
|
||||
# lines = prompt_text.split("\n")
|
||||
# for line in lines:
|
||||
# if line.strip():
|
||||
# self.content_text.insert("end", " " + line + "\n", "prompt")
|
||||
# else:
|
||||
# self.content_text.insert("end", " 无Prompt内容\n", "prompt")
|
||||
|
||||
# 滚动到顶部
|
||||
self.content_text.see("1.0")
|
||||
# # 推理过程
|
||||
# self.content_text.insert("end", "推理过程:\n", "timestamp")
|
||||
# reasoning_text = item.get("reasoning", "")
|
||||
# if reasoning_text and reasoning_text.lower() != "none":
|
||||
# lines = reasoning_text.split("\n")
|
||||
# for line in lines:
|
||||
# if line.strip():
|
||||
# self.content_text.insert("end", " " + line + "\n", "reasoning")
|
||||
# else:
|
||||
# self.content_text.insert("end", " 无推理过程\n", "reasoning")
|
||||
|
||||
def _auto_update(self):
|
||||
"""自动更新函数"""
|
||||
while True:
|
||||
try:
|
||||
# 从数据库获取最新数据，只获取启动时间之后的记录
|
||||
query = {"time": {"$gt": self.start_timestamp}}
|
||||
logger.debug(f"查询条件: {query}")
|
||||
# # 回复内容
|
||||
# self.content_text.insert("end", "回复: ", "timestamp")
|
||||
# self.content_text.insert("end", f"{item.get('response', '')}\n", "response")
|
||||
|
||||
# 先获取一条记录检查时间格式
|
||||
sample = db.reasoning_logs.find_one()
|
||||
if sample:
|
||||
logger.debug(f"样本记录时间格式: {type(sample['time'])} 值: {sample['time']}")
|
||||
# # 分隔符
|
||||
# self.content_text.insert("end", f"\n{'=' * 50}\n\n", "separator")
|
||||
|
||||
cursor = db.reasoning_logs.find(query).sort("time", -1)
|
||||
new_data = {}
|
||||
total_count = 0
|
||||
# # 滚动到顶部
|
||||
# self.content_text.see("1.0")
|
||||
|
||||
for item in cursor:
|
||||
# 调试输出
|
||||
if total_count == 0:
|
||||
logger.debug(f"记录时间: {item['time']}, 类型: {type(item['time'])}")
|
||||
# def _auto_update(self):
|
||||
# """自动更新函数"""
|
||||
# while True:
|
||||
# if not self.is_running:
|
||||
# break # 如果GUI已关闭，停止线程
|
||||
|
||||
total_count += 1
|
||||
group_id = str(item.get("group_id", "unknown"))
|
||||
if group_id not in new_data:
|
||||
new_data[group_id] = []
|
||||
# try:
|
||||
# # 从数据库获取最新数据，只获取启动时间之后的记录
|
||||
# query = {"time": {"$gt": self.start_timestamp}}
|
||||
# logger.debug(f"查询条件: {query}")
|
||||
|
||||
# 转换时间戳为datetime对象
|
||||
if isinstance(item["time"], (int, float)):
|
||||
time_obj = datetime.fromtimestamp(item["time"])
|
||||
elif isinstance(item["time"], datetime):
|
||||
time_obj = item["time"]
|
||||
else:
|
||||
logger.warning(f"未知的时间格式: {type(item['time'])}")
|
||||
time_obj = datetime.now() # 使用当前时间作为后备
|
||||
# # 先获取一条记录检查时间格式
|
||||
# sample = db.reasoning_logs.find_one()
|
||||
# if sample:
|
||||
# logger.debug(f"样本记录时间格式: {type(sample['time'])} 值: {sample['time']}")
|
||||
|
||||
new_data[group_id].append(
|
||||
{
|
||||
"time": time_obj,
|
||||
"user": item.get("user", "未知"),
|
||||
"message": item.get("message", ""),
|
||||
"model": item.get("model", "未知"),
|
||||
"reasoning": item.get("reasoning", ""),
|
||||
"response": item.get("response", ""),
|
||||
"prompt": item.get("prompt", ""), # 添加prompt字段
|
||||
}
|
||||
)
|
||||
# cursor = db.reasoning_logs.find(query).sort("time", -1)
|
||||
# new_data = {}
|
||||
# total_count = 0
|
||||
|
||||
logger.info(f"从数据库加载了 {total_count} 条记录，分布在 {len(new_data)} 个群组中")
|
||||
# for item in cursor:
|
||||
# # 调试输出
|
||||
# if total_count == 0:
|
||||
# logger.debug(f"记录时间: {item['time']}, 类型: {type(item['time'])}")
|
||||
|
||||
# 更新数据
|
||||
if new_data != self.group_data:
|
||||
self.group_data = new_data
|
||||
logger.info("数据已更新，正在刷新显示...")
|
||||
# 将更新任务添加到队列
|
||||
self.update_queue.put({"type": "update_group_list"})
|
||||
if self.group_data:
|
||||
# 如果没有选中的群组，选择最新的群组
|
||||
if not self.selected_group_id or self.selected_group_id not in self.group_data:
|
||||
self.selected_group_id = next(iter(self.group_data))
|
||||
self.update_queue.put({"type": "update_display", "group_id": self.selected_group_id})
|
||||
except Exception:
|
||||
logger.exception("自动更新出错")
|
||||
# total_count += 1
|
||||
# group_id = str(item.get("group_id", "unknown"))
|
||||
# if group_id not in new_data:
|
||||
# new_data[group_id] = []
|
||||
|
||||
# 每5秒更新一次
|
||||
time.sleep(5)
|
||||
# # 转换时间戳为datetime对象
|
||||
# if isinstance(item["time"], (int, float)):
|
||||
# time_obj = datetime.fromtimestamp(item["time"])
|
||||
# elif isinstance(item["time"], datetime):
|
||||
# time_obj = item["time"]
|
||||
# else:
|
||||
# logger.warning(f"未知的时间格式: {type(item['time'])}")
|
||||
# time_obj = datetime.now() # 使用当前时间作为后备
|
||||
|
||||
def clear_display(self):
|
||||
"""清除显示内容"""
|
||||
self.content_text.delete("1.0", "end")
|
||||
# new_data[group_id].append(
|
||||
# {
|
||||
# "time": time_obj,
|
||||
# "user": item.get("user", "未知"),
|
||||
# "message": item.get("message", ""),
|
||||
# "model": item.get("model", "未知"),
|
||||
# "reasoning": item.get("reasoning", ""),
|
||||
# "response": item.get("response", ""),
|
||||
# "prompt": item.get("prompt", ""), # 添加prompt字段
|
||||
# }
|
||||
# )
|
||||
|
||||
def run(self):
|
||||
"""运行GUI"""
|
||||
self.root.mainloop()
|
||||
# logger.info(f"从数据库加载了 {total_count} 条记录，分布在 {len(new_data)} 个群组中")
|
||||
|
||||
# # 更新数据
|
||||
# if new_data != self.group_data:
|
||||
# self.group_data = new_data
|
||||
# logger.info("数据已更新，正在刷新显示...")
|
||||
# # 将更新任务添加到队列
|
||||
# self.update_queue.put({"type": "update_group_list"})
|
||||
# if self.group_data:
|
||||
# # 如果没有选中的群组，选择最新的群组
|
||||
# if not self.selected_group_id or self.selected_group_id not in self.group_data:
|
||||
# self.selected_group_id = next(iter(self.group_data))
|
||||
# self.update_queue.put({"type": "update_display", "group_id": self.selected_group_id})
|
||||
# except Exception:
|
||||
# logger.exception("自动更新出错")
|
||||
|
||||
# # 每5秒更新一次
|
||||
# time.sleep(5)
|
||||
|
||||
# def clear_display(self):
|
||||
# """清除显示内容"""
|
||||
# self.content_text.delete("1.0", "end")
|
||||
|
||||
# def run(self):
|
||||
# """运行GUI"""
|
||||
# self.root.mainloop()
|
||||
|
||||
|
||||
def main():
|
||||
app = ReasoningGUI()
|
||||
app.run()
|
||||
# def main():
|
||||
# app = ReasoningGUI()
|
||||
# app.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
# if __name__ == "__main__":
|
||||
# main()
|
||||
|
||||
@@ -6,9 +6,9 @@ from typing import Any, Callable, Dict
|
||||
import datetime
|
||||
import asyncio
|
||||
import numpy
|
||||
import matplotlib.pyplot as plt
|
||||
from pathlib import Path
|
||||
import pandas as pd
|
||||
# import matplotlib.pyplot as plt
|
||||
# from pathlib import Path
|
||||
# import pandas as pd
|
||||
|
||||
|
||||
"""
|
||||
@@ -232,7 +232,6 @@ class PersonInfoManager:
|
||||
logger.info(f"个人信息推断启动: {current_time.strftime('%Y-%m-%d %H:%M:%S')}")
|
||||
|
||||
# "msg_interval"推断
|
||||
msg_interval_map = False
|
||||
msg_interval_lists = await self.get_specific_value_list(
|
||||
"msg_interval_list",
|
||||
lambda x: isinstance(x, list) and len(x) >= 100
|
||||
@@ -246,33 +245,7 @@ class PersonInfoManager:
|
||||
time_interval.append(delta)
|
||||
|
||||
if len(time_interval) > 30:
|
||||
time_interval.sort()
|
||||
|
||||
# 画图(log)
|
||||
msg_interval_map = True
|
||||
log_dir = Path("logs/person_info")
|
||||
log_dir.mkdir(parents=True, exist_ok=True)
|
||||
plt.figure(figsize=(10, 6))
|
||||
|
||||
time_series = pd.Series(time_interval)
|
||||
|
||||
# 绘制直方图
|
||||
plt.hist(time_series, bins=50, density=True, alpha=0.4, color='pink', label='Histogram')
|
||||
|
||||
# 绘制KDE曲线（使用相同的实际数据）
|
||||
time_series.plot(kind='kde', color='mediumpurple', linewidth=1, label='Density')
|
||||
|
||||
plt.grid(True, alpha=0.2)
|
||||
plt.xlim(0, 8000)
|
||||
plt.title(f"Message Interval Distribution (User: {person_id[:8]}...)")
|
||||
plt.xlabel("Interval (ms)")
|
||||
plt.ylabel("Density")
|
||||
plt.legend(framealpha=0.9, facecolor='white')
|
||||
|
||||
img_path = log_dir / f"interval_distribution_{person_id[:8]}.png"
|
||||
plt.savefig(img_path)
|
||||
plt.close()
|
||||
# 画图
|
||||
# 移除matplotlib相关的绘图功能
|
||||
|
||||
filtered_intervals = [t for t in time_interval if t >= 500]
|
||||
if len(filtered_intervals) > 25:
|
||||
@@ -285,8 +258,6 @@ class PersonInfoManager:
|
||||
|
||||
# 其他...
|
||||
|
||||
if msg_interval_map:
|
||||
logger.info("已保存分布图到: logs/person_info")
|
||||
logger.info(f"个人信息推断结束: {current_time.strftime('%Y-%m-%d %H:%M:%S')}")
|
||||
await asyncio.sleep(86400)
|
||||
|
||||
@@ -294,4 +265,4 @@ class PersonInfoManager:
|
||||
logger.error(f"个人信息推断运行时出错: {str(e)}")
|
||||
logger.exception("详细错误信息：")
|
||||
|
||||
person_info_manager = PersonInfoManager()
|
||||
person_info_manager = PersonInfoManager()
|
||||
|
||||
Reference in New Issue
Block a user