haosicheng/shuidrop_gui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Todo: 集成多平台 解决因SaiNiu线程抢占资源问题 本地提交测试环境打包 和 正式打包脚本与正式环境打包bat

Browse Source
This commit is contained in:
haosicheng
2025-09-17 17:48:35 +08:00
parent 256e6c21a5
commit 8b9fc925fa
11 changed files with 2020 additions and 610 deletions
Download Patch File Download Diff File Expand all files Collapse all files

745
main.py
View File

@@ -3,10 +3,20 @@ from PyQt5.QtCore import Qt
from PyQt5.QtGui import QFont, QPalette, QColor
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
QHBoxLayout, QLabel, QPushButton, QLineEdit,
QTextEdit, QFrame, QDialog, QDialogButtonBox, QComboBox)
QTextEdit, QFrame, QDialog, QDialogButtonBox, QComboBox,
QSystemTrayIcon, QMenu, QAction, QMessageBox, QGraphicsDropShadowEffect)
from PyQt5.QtCore import QPropertyAnimation, QEasingCurve, QRect, QTimer, pyqtProperty
from PyQt5.QtGui import QColor
import config
from WebSocket.backend_singleton import get_websocket_manager
from windows_taskbar_fix import setup_windows_taskbar_icon
import os
# ===================== 文件日志系统 - 同时支持开发和exe环境 =====================
# 重定向所有输出到文件，确保有日志记录
from exe_file_logger import setup_file_logging, log_to_file
setup_file_logging()
print("文件日志系统已在main.py中初始化")
# 新增: 用户名密码输入对话框类
@@ -21,22 +31,27 @@ class LoginWindow(QMainWindow):
self.pdd_worker = None
# 重复执行防护
self.platform_combo_connected = False
self.last_login_time = 0
self.login_cooldown = 2 # 登录冷却时间（秒）
# 平台连接状态管理
self.connected_platforms = []
self.status_timer = None
self.last_platform_connect_time = 0 # 记录最后一次平台连接时间
# 日志管理相关变量已删除
self.initUI()
# 设置当前平台为ComboBox的当前值
self.current_platform = self.platform_combo.currentText()
def initUI(self):
# 设置窗口基本属性
self.setWindowTitle('AI回复连接入口-V1.0')
self.setGeometry(300, 300, 800, 600) # 增大窗口尺寸
self.setMinimumSize(700, 500) # 设置最小尺寸保证显示完整
self.setWindowTitle('AI客服智能助手')
self.setGeometry(300, 300, 450, 300) # 进一步减小窗口尺寸
self.setMinimumSize(420, 280) # 设置更小的最小尺寸
self.setMaximumSize(500, 350) # 设置最大尺寸，保持紧凑
# 窗口图标将由统一的任务栏修复模块设置，这里只做备用检查
print("[INFO] 窗口图标将由统一模块设置")
# 创建中央widget
central_widget = QWidget()
@@ -44,60 +59,44 @@ class LoginWindow(QMainWindow):
# 创建主布局
main_layout = QVBoxLayout()
main_layout.setSpacing(25) # 增加间距
main_layout.setContentsMargins(40, 40, 40, 40) # 增加边距
main_layout.setSpacing(15) # 减小间距
main_layout.setContentsMargins(30, 20, 30, 20) # 减小边距
central_widget.setLayout(main_layout)
# 添加标题
title_label = QLabel('AI回复连接入口')
# 添加标题和副标题
title_label = QLabel('AI客服智能助手')
title_label.setObjectName("title")
title_label.setAlignment(Qt.AlignCenter)
title_label.setFont(QFont('Microsoft YaHei', 24, QFont.Bold)) # 使用系统自带字体
title_label.setStyleSheet("color: #2c3e50;")
title_label.setFont(QFont('Microsoft YaHei', 18, QFont.Bold)) # 减小字体
title_label.setStyleSheet("color: #2c3e50; margin-bottom: 3px;")
main_layout.addWidget(title_label)
# 添加分隔线
line = QFrame()
line.setFrameShape(QFrame.HLine)
line.setFrameShadow(QFrame.Sunken)
line.setStyleSheet("color: #bdc3c7;")
main_layout.addWidget(line)
# 添加副标题说明
subtitle_label = QLabel('智能连接多平台客服，提供AI自动回复服务')
subtitle_label.setObjectName("subtitle")
subtitle_label.setAlignment(Qt.AlignCenter)
subtitle_label.setFont(QFont('Microsoft YaHei', 9)) # 减小字体
subtitle_label.setStyleSheet("color: #7f8c8d; margin-bottom: 10px;")
main_layout.addWidget(subtitle_label)
# 在token_input之前添加平台选择区域
platform_layout = QHBoxLayout()
platform_label = QLabel("选择平台:")
self.platform_combo = QComboBox()
self.platform_combo.addItems(["JD", "抖音", "千牛(淘宝)", "拼多多"])
# 防止重复连接信号 - 更强的保护
if not hasattr(self, 'platform_combo_connected') or not self.platform_combo_connected:
try:
self.platform_combo.currentIndexChanged.disconnect()
except TypeError:
pass # 如果没有连接则忽略
self.platform_combo.currentIndexChanged.connect(self.on_platform_changed)
self.platform_combo_connected = True
platform_layout.addWidget(platform_label)
platform_layout.addWidget(self.platform_combo)
# 将platform_layout添加到主布局中，在token_layout之前
main_layout.addLayout(platform_layout)
# 添加少量垂直空间
main_layout.addStretch(1)
# 创建令牌输入区域
token_layout = QHBoxLayout()
token_layout.setSpacing(15)
token_layout = QVBoxLayout()
token_layout.setSpacing(8) # 减小间距
token_label = QLabel('令牌:')
token_label.setFont(QFont('Microsoft YaHei', 12))
token_label.setFixedWidth(80)
token_label = QLabel('访问令牌')
token_label.setFont(QFont('Microsoft YaHei', 11, QFont.Bold)) # 减小字体
token_label.setStyleSheet("color: #34495e;")
self.token_input = QLineEdit()
self.token_input.setPlaceholderText('请输入您的访问令牌')
self.token_input.setPlaceholderText('请输入您的访问令牌以连接服务')
self.token_input.setEchoMode(QLineEdit.Password)
self.token_input.setFont(QFont('Microsoft YaHei', 11))
self.token_input.setFont(QFont('Microsoft YaHei', 10)) # 减小字体
# noinspection PyUnresolvedReferences
self.token_input.returnPressed.connect(self.login) # 表示回车提交
self.token_input.setMinimumHeight(40) # 增加输入框高度
self.token_input.setMinimumHeight(38) # 减小输入框高度
# 预填已保存的令牌（如果存在）
try:
from config import get_saved_token
@@ -112,18 +111,22 @@ class LoginWindow(QMainWindow):
main_layout.addLayout(token_layout)
# 创建连接按钮
self.login_btn = QPushButton('是否连接')
self.login_btn.setFont(QFont('Microsoft YaHei', 12, QFont.Bold))
self.login_btn.setMinimumHeight(45) # 增大按钮高度
self.login_btn = QPushButton('连接服务')
self.login_btn.setFont(QFont('Microsoft YaHei', 11, QFont.Bold)) # 减小字体
self.login_btn.setMinimumHeight(40) # 减小按钮高度
self.login_btn.clicked.connect(self.login) # 表示点击提交
main_layout.addWidget(self.login_btn)
# 添加分隔线
line = QFrame()
line.setFrameShape(QFrame.HLine)
line.setFrameShadow(QFrame.Sunken)
line.setStyleSheet("color: #bdc3c7;")
main_layout.addWidget(line)
# 添加连接状态提示
self.status_label = QLabel('等待连接...')
self.status_label.setObjectName("status")
self.status_label.setAlignment(Qt.AlignCenter)
self.status_label.setFont(QFont('Microsoft YaHei', 9)) # 减小字体
self.status_label.setStyleSheet("color: #95a5a6; margin-top: 5px;")
main_layout.addWidget(self.status_label)
# 添加少量底部空间
main_layout.addStretch(1)
# 日志框已永久删除，只使用终端输出
self.log_display = None
@@ -131,102 +134,142 @@ class LoginWindow(QMainWindow):
# 应用现代化样式
self.apply_modern_styles()
# 添加视觉效果
self.add_visual_effects()
# 初始化系统托盘
self.init_system_tray()
# 系统初始化日志输出到终端
print("[INFO] 系统初始化完成")
# # 在平台选择改变时添加调试日志
# self.platform_combo.currentIndexChanged.connect(self.on_platform_changed)
# 设置默认平台
self.platform_combo.setCurrentText("JD")
print(f"🔥 设置默认平台为: {self.platform_combo.currentText()}")
def apply_modern_styles(self):
"""应用现代化样式，兼容各Windows版本"""
"""应用现代化简约样式 - 精致美化版"""
self.setStyleSheet("""
QMainWindow {
background-color: #f9f9f9;
background: qlineargradient(
x1:0, y1:0, x2:0, y2:1,
stop:0 #f8fafb,
stop:0.5 #f1f4f6,
stop:1 #e8ecef
);
border: 1px solid rgba(255, 255, 255, 0.3);
}
QLabel {
color: #34495e;
color: #2c3e50;
background: transparent;
}
QLabel[objectName="title"] {
color: qlineargradient(
x1:0, y1:0, x2:1, y2:0,
stop:0 #2c3e50,
stop:1 #34495e
);
font-weight: bold;
}
QLabel[objectName="subtitle"] {
color: #7f8c8d;
background: transparent;
}
QLabel[objectName="status"] {
padding: 5px 10px;
border-radius: 12px;
background: rgba(149, 165, 166, 0.1);
font-size: 9px;
}
QLineEdit {
padding: 10px;
border: 2px solid #dfe6e9;
border-radius: 6px;
font-size: 14px;
background-color: white;
selection-background-color: #3498db;
padding: 12px 16px;
border: 2px solid transparent;
border-radius: 12px;
font-size: 13px;
background: qlineargradient(
x1:0, y1:0, x2:0, y2:1,
stop:0 #ffffff,
stop:1 #f8fafb
);
selection-background-color: #007bff;
selection-color: white;
color: #2c3e50;
}
QLineEdit:focus {
border-color: #3498db;
border: 2px solid #007bff;
background: white;
outline: none;
}
QLineEdit:hover {
border: 2px solid #6c757d;
background: white;
}
QPushButton {
background-color: #3498db;
background: qlineargradient(
x1:0, y1:0, x2:0, y2:1,
stop:0 #4285f4,
stop:0.5 #1976d2,
stop:1 #1565c0
);
border: none;
color: white;
padding: 12px 24px;
border-radius: 6px;
font-size: 14px;
min-width: 120px;
border-radius: 12px;
font-size: 13px;
font-weight: 600;
min-width: 180px;
letter-spacing: 0.5px;
}
QPushButton:hover {
background-color: #2980b9;
background: qlineargradient(
x1:0, y1:0, x2:0, y2:1,
stop:0 #5294f5,
stop:0.5 #2986d3,
stop:1 #1e74c1
);
}
QPushButton:pressed {
background-color: #1a6a9c;
background: qlineargradient(
x1:0, y1:0, x2:0, y2:1,
stop:0 #3274d4,
stop:0.5 #1666c2,
stop:1 #1455b0
);
}
QPushButton:disabled {
background-color: #bdc3c7;
color: #7f8c8d;
background: qlineargradient(
x1:0, y1:0, x2:0, y2:1,
stop:0 #f8f9fa,
stop:1 #e9ecef
);
color: #6c757d;
border: 1px solid #dee2e6;
}
QTextEdit {
border: 2px solid #dfe6e9;
border-radius: 6px;
background-color: white;
padding: 8px;
font-family: 'Consolas', 'Microsoft YaHei', monospace;
/* 连接成功状态的按钮 */
QPushButton[objectName="connected"] {
background: qlineargradient(
x1:0, y1:0, x2:0, y2:1,
stop:0 #28a745,
stop:0.5 #20963b,
stop:1 #1e7e34
);
}
QFrame[frameShape="4"] {
color: #dfe6e9;
}
QGroupBox {
font-weight: bold;
border: 2px solid #dfe6e9;
border-radius: 6px;
margin-top: 10px;
padding-top: 10px;
}
QComboBox {
padding: 8px;
border: 2px solid #dfe6e9;
border-radius: 6px;
font-size: 14px;
background-color: white;
min-width: 100px;
}
QComboBox::drop-down {
border: none;
width: 20px;
}
QComboBox QAbstractItemView {
border: 2px solid #dfe6e9;
selection-background-color: #3498db;
selection-color: white;
QPushButton[objectName="connected"]:hover {
background: qlineargradient(
x1:0, y1:0, x2:0, y2:1,
stop:0 #34ce57,
stop:0.5 #28a745,
stop:1 #20963b
);
}
""")
@@ -236,104 +279,256 @@ class LoginWindow(QMainWindow):
# 设置调色板确保颜色一致性
palette = QPalette()
palette.setColor(QPalette.Window, QColor(249, 249, 249))
palette.setColor(QPalette.WindowText, QColor(52, 73, 94))
palette.setColor(QPalette.Window, QColor(248, 250, 251))
palette.setColor(QPalette.WindowText, QColor(44, 62, 80))
palette.setColor(QPalette.Base, QColor(255, 255, 255))
palette.setColor(QPalette.AlternateBase, QColor(233, 235, 239))
palette.setColor(QPalette.AlternateBase, QColor(241, 244, 246))
palette.setColor(QPalette.ToolTipBase, QColor(255, 255, 255))
palette.setColor(QPalette.ToolTipText, QColor(52, 73, 94))
palette.setColor(QPalette.Text, QColor(52, 73, 94))
palette.setColor(QPalette.Button, QColor(52, 152, 219))
palette.setColor(QPalette.ToolTipText, QColor(44, 62, 80))
palette.setColor(QPalette.Text, QColor(44, 62, 80))
palette.setColor(QPalette.Button, QColor(66, 133, 244))
palette.setColor(QPalette.ButtonText, QColor(255, 255, 255))
palette.setColor(QPalette.BrightText, QColor(255, 255, 255))
palette.setColor(QPalette.Highlight, QColor(52, 152, 219))
palette.setColor(QPalette.Highlight, QColor(66, 133, 244))
palette.setColor(QPalette.HighlightedText, QColor(255, 255, 255))
QApplication.setPalette(palette)
def on_platform_changed(self, index):
"""平台选择改变时的处理 - 新增方法"""
platform = self.platform_combo.currentText()
self.current_platform = platform
self.add_log(f"已选择平台: {platform}", "INFO")
print(f"🔥 平台已更改为: {platform}")
def add_log(self, message, log_type="INFO"):
"""添加日志信息 - 只输出到终端"""
from datetime import datetime
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
# 根据日志类型设置颜色（ANSI颜色码）
colors = {
"ERROR": "\033[91m", # 红色
"SUCCESS": "\033[92m", # 绿色
"WARNING": "\033[93m", # 黄色
"INFO": "\033[94m", # 蓝色
"DEBUG": "\033[95m" # 紫色
}
reset = "\033[0m" # 重置颜色
color = colors.get(log_type, colors["INFO"])
log_entry = f"{color}[{timestamp}] [{log_type}] {message}{reset}"
print(log_entry)
def login(self):
"""处理连接逻辑"""
# 防止重复快速点击
import time
current_time = time.time()
if current_time - self.last_login_time < self.login_cooldown:
self.add_log(f"请等待 {self.login_cooldown} 秒后再试", "WARNING")
return
self.last_login_time = current_time
token = self.token_input.text().strip()
if not token:
self.add_log("请输入有效的连接令牌", "ERROR")
return
# 禁用连接按钮，防止重复点击
self.login_btn.setEnabled(False)
self.login_btn.setText("连接中...")
def add_visual_effects(self):
"""添加视觉效果"""
try:
# 使用 WebSocket 管理器处理连接
ws_manager = get_websocket_manager()
# 设置回调函数
ws_manager.set_callbacks(
log=self.add_log,
success=lambda: self.add_log("WebSocket连接管理器连接成功", "SUCCESS"),
error=lambda error: self.add_log(f"WebSocket连接管理器错误: {error}", "ERROR")
)
# 连接后端
success = ws_manager.connect_backend(token)
if success:
self.add_log("已启动WebSocket连接管理器", "SUCCESS")
else:
self.add_log("WebSocket连接管理器启动失败", "ERROR")
# 为主窗口添加阴影效果
shadow = QGraphicsDropShadowEffect()
shadow.setBlurRadius(20)
shadow.setXOffset(0)
shadow.setYOffset(5)
shadow.setColor(QColor(0, 0, 0, 40))
# 为输入框添加阴影
input_shadow = QGraphicsDropShadowEffect()
input_shadow.setBlurRadius(8)
input_shadow.setXOffset(0)
input_shadow.setYOffset(2)
input_shadow.setColor(QColor(0, 0, 0, 20))
self.token_input.setGraphicsEffect(input_shadow)
# 为按钮添加阴影
button_shadow = QGraphicsDropShadowEffect()
button_shadow.setBlurRadius(12)
button_shadow.setXOffset(0)
button_shadow.setYOffset(4)
button_shadow.setColor(QColor(66, 133, 244, 80))
self.login_btn.setGraphicsEffect(button_shadow)
# 设置对象名称以应用特定样式
self.token_input.setObjectName("tokenInput")
self.login_btn.setObjectName("loginButton")
print("[INFO] 视觉效果已添加")
except Exception as e:
self.add_log(f"连接失败: {e}", "ERROR")
finally:
self.login_btn.setEnabled(True)
self.login_btn.setText("开始连接")
print(f"[WARNING] 添加视觉效果失败: {e}")
def verify_token(self, token):
"""简化的令牌校验：非空即通过（实际校验由后端承担）"""
return bool(token)
def animate_success(self):
"""连接成功的动画效果"""
try:
# 按钮轻微放大动画
self.button_animation = QPropertyAnimation(self.login_btn, b"geometry")
original_rect = self.login_btn.geometry()
def closeEvent(self, event):
"""窗口关闭事件处理"""
# 计算放大后的位置（居中放大）
expanded_rect = QRect(
original_rect.x() - 5,
original_rect.y() - 2,
original_rect.width() + 10,
original_rect.height() + 4
)
self.button_animation.setDuration(200)
self.button_animation.setStartValue(original_rect)
self.button_animation.setEndValue(expanded_rect)
self.button_animation.setEasingCurve(QEasingCurve.OutBack)
# 动画完成后恢复原始大小
def restore_size():
restore_animation = QPropertyAnimation(self.login_btn, b"geometry")
restore_animation.setDuration(150)
restore_animation.setStartValue(expanded_rect)
restore_animation.setEndValue(original_rect)
restore_animation.setEasingCurve(QEasingCurve.InBack)
restore_animation.start()
self.button_animation.finished.connect(restore_size)
self.button_animation.start()
print("[INFO] 成功动画已启动")
except Exception as e:
print(f"[WARNING] 动画效果失败: {e}")
def on_platform_connected(self, platform_name: str, all_platforms: list):
"""处理平台连接成功事件"""
try:
import time
self.connected_platforms = all_platforms.copy()
self.last_platform_connect_time = time.time()
# 短暂显示平台连接成功提示
self.status_label.setText(f"🟢 {platform_name}平台连接成功！")
self.status_label.setStyleSheet(
"color: #28a745; background: rgba(40, 167, 69, 0.15); border-radius: 12px; padding: 5px 10px; font-weight: bold;")
# 停止之前的定时器
if self.status_timer:
self.status_timer.stop()
self.status_timer = None
self.add_log(f"GUI收到平台连接通知: {platform_name}, 当前已连接: {', '.join(all_platforms)}", "INFO")
# 使用线程定时器，更可靠（资源消耗极小）
import threading
def delayed_update():
import time
time.sleep(3) # 等待3秒
# 使用QTimer.singleShot确保在主线程中执行GUI更新
QTimer.singleShot(0, self.delayed_platform_summary)
# 启动轻量级守护线程（3秒后自动销毁）
timer_thread = threading.Thread(target=delayed_update, daemon=True)
timer_thread.start()
self.add_log("3秒线程定时器已启动，将自动切换到汇总显示", "INFO")
except Exception as e:
self.add_log(f"处理平台连接事件失败: {e}", "ERROR")
def delayed_platform_summary(self):
"""定时器触发的汇总显示更新"""
try:
self.add_log("🎯 定时器触发！开始执行汇总显示更新", "INFO")
self.update_platform_summary()
self.add_log("🎯 定时器执行完成", "INFO")
except Exception as e:
self.add_log(f"定时器执行失败: {e}", "ERROR")
def check_and_update_summary(self):
"""检查是否应该更新汇总显示"""
try:
import time
elapsed = time.time() - self.last_platform_connect_time
self.add_log(f"检查汇总更新: 距离最后连接 {elapsed:.1f} 秒", "INFO")
# 检查是否在最近1秒内还有新的平台连接
if elapsed >= 2.8: # 距离最后连接超过2.8秒
self.add_log("满足条件，开始更新汇总显示", "INFO")
self.update_platform_summary()
else:
# 如果还有新连接，再延迟一会儿
self.add_log(f"还需等待 {2.8 - elapsed:.1f} 秒，延迟更新", "INFO")
QTimer.singleShot(1000, self.check_and_update_summary)
except Exception as e:
self.add_log(f"检查汇总更新失败: {e}", "ERROR")
def update_platform_summary(self):
"""更新平台连接汇总显示"""
try:
self.add_log(f"开始更新汇总显示，当前连接平台: {self.connected_platforms}", "INFO")
if self.connected_platforms:
platforms_text = "、".join(self.connected_platforms)
summary_text = f"🟢 已连接: {platforms_text}"
self.status_label.setText(summary_text)
self.status_label.setStyleSheet(
"color: #28a745; background: rgba(40, 167, 69, 0.1); border-radius: 12px; padding: 5px 10px;")
self.add_log(f"汇总显示已更新: {summary_text}", "INFO")
else:
self.status_label.setText("🟢 连接成功！等待平台指令...")
self.status_label.setStyleSheet(
"color: #28a745; background: rgba(40, 167, 69, 0.1); border-radius: 12px; padding: 5px 10px;")
self.add_log("已更新为等待平台指令状态", "INFO")
except Exception as e:
self.add_log(f"更新平台汇总显示失败: {e}", "ERROR")
def init_system_tray(self):
"""初始化系统托盘"""
# 检查系统是否支持托盘
if not QSystemTrayIcon.isSystemTrayAvailable():
QMessageBox.critical(None, "系统托盘",
"系统托盘不可用")
return
# 创建托盘图标
self.tray_icon = QSystemTrayIcon(self)
# 使用自定义的美观图标
try:
tray_icon_path = os.path.join(os.path.dirname(__file__), "static", "ai_assistant_icon_16.png")
if os.path.exists(tray_icon_path):
from PyQt5.QtGui import QIcon
self.tray_icon.setIcon(QIcon(tray_icon_path))
print(f"[INFO] 已加载托盘图标: {tray_icon_path}")
else:
print(f"[WARNING] 托盘图标文件不存在: {tray_icon_path}")
# 备用方案：使用系统图标
self.tray_icon.setIcon(self.style().standardIcon(self.style().SP_ComputerIcon))
except Exception as e:
print(f"[WARNING] 加载托盘图标失败: {e}, 使用默认图标")
# 备用方案：使用系统图标
self.tray_icon.setIcon(self.style().standardIcon(self.style().SP_ComputerIcon))
# 创建托盘菜单
tray_menu = QMenu()
# 显示窗口动作
show_action = QAction("显示主窗口", self)
show_action.triggered.connect(self.show_window)
tray_menu.addAction(show_action)
# 分隔线
tray_menu.addSeparator()
# 退出动作
quit_action = QAction("退出程序", self)
quit_action.triggered.connect(self.quit_application)
tray_menu.addAction(quit_action)
# 设置托盘菜单
self.tray_icon.setContextMenu(tray_menu)
# 设置托盘提示
self.tray_icon.setToolTip("AI客服智能助手")
# 双击托盘图标显示窗口
self.tray_icon.activated.connect(self.tray_icon_activated)
# 显示托盘图标
self.tray_icon.show()
print("[INFO] 系统托盘已初始化")
def tray_icon_activated(self, reason):
"""托盘图标被激活时的处理"""
if reason == QSystemTrayIcon.DoubleClick:
self.show_window()
def show_window(self):
"""显示主窗口"""
self.show()
self.raise_()
self.activateWindow()
print("[INFO] 主窗口已显示")
def quit_application(self):
"""真正退出应用程序"""
print("[INFO] 正在退出应用程序...")
# 执行原来的关闭逻辑
try:
# 使用 WebSocket 管理器断开所有连接
from WebSocket.backend_singleton import get_websocket_manager
ws_manager = get_websocket_manager()
ws_manager.disconnect_all()
# 停止所有工作线程（向后兼容）
workers = []
if hasattr(self, 'jd_worker') and self.jd_worker:
@@ -358,8 +553,124 @@ class LoginWindow(QMainWindow):
worker.loop.close()
except Exception as e:
print(f"关闭窗口时发生错误: {str(e)}")
print(f"关闭时发生错误: {str(e)}")
finally:
# 隐藏托盘图标
if hasattr(self, 'tray_icon'):
self.tray_icon.hide()
# 退出应用程序
QApplication.quit()
def add_log(self, message, log_type="INFO"):
"""添加日志信息 - 只输出到终端"""
from datetime import datetime
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
# 根据日志类型设置颜色（ANSI颜色码）
colors = {
"ERROR": "\033[91m", # 红色
"SUCCESS": "\033[92m", # 绿色
"WARNING": "\033[93m", # 黄色
"INFO": "\033[94m", # 蓝色
"DEBUG": "\033[95m" # 紫色
}
reset = "\033[0m" # 重置颜色
color = colors.get(log_type, colors["INFO"])
log_entry = f"{color}[{timestamp}] [{log_type}] {message}{reset}"
print(log_entry)
def login(self):
"""处理连接逻辑"""
# 防止重复快速点击
import time
current_time = time.time()
if current_time - self.last_login_time < self.login_cooldown:
self.add_log(f"请等待 {self.login_cooldown} 秒后再试", "WARNING")
return
self.last_login_time = current_time
token = self.token_input.text().strip()
if not token:
self.add_log("请输入有效的连接令牌", "ERROR")
return
# 禁用连接按钮，防止重复点击
self.login_btn.setEnabled(False)
self.login_btn.setText("连接中...")
self.status_label.setText("正在连接服务...")
self.status_label.setStyleSheet("color: #ffc107;")
success = False
try:
# 使用 WebSocket 管理器处理连接
ws_manager = get_websocket_manager()
# 设置回调函数
ws_manager.set_callbacks(
log=self.add_log,
success=lambda: self.add_log("WebSocket连接管理器连接成功", "SUCCESS"),
error=lambda error: self.add_log(f"WebSocket连接管理器错误: {error}", "ERROR"),
platform_connected=self.on_platform_connected # 新增：平台连接回调
)
# 连接后端
success = ws_manager.connect_backend(token)
if success:
self.add_log("已启动WebSocket连接管理器", "SUCCESS")
self.status_label.setText("🟢 连接成功！等待平台指令...")
self.status_label.setStyleSheet(
"color: #28a745; background: rgba(40, 167, 69, 0.1); border-radius: 12px; padding: 5px 10px;")
self.login_btn.setText("✓ 已连接")
self.login_btn.setObjectName("connected")
self.login_btn.setStyleSheet(self.login_btn.styleSheet()) # 刷新样式
# 添加成功动画效果
self.animate_success()
else:
self.add_log("WebSocket连接管理器启动失败", "ERROR")
self.status_label.setText("🔴 连接失败，请检查令牌")
self.status_label.setStyleSheet(
"color: #dc3545; background: rgba(220, 53, 69, 0.1); border-radius: 12px; padding: 5px 10px;")
except Exception as e:
self.add_log(f"连接失败: {e}", "ERROR")
self.status_label.setText(f"🔴 连接失败: {str(e)}")
self.status_label.setStyleSheet(
"color: #dc3545; background: rgba(220, 53, 69, 0.1); border-radius: 12px; padding: 5px 10px;")
finally:
self.login_btn.setEnabled(True)
if not success:
self.login_btn.setText("重新连接")
self.login_btn.setObjectName("loginButton") # 恢复原始样式
def verify_token(self, token):
"""简化的令牌校验：非空即通过（实际校验由后端承担）"""
return bool(token)
def closeEvent(self, event):
"""窗口关闭事件处理 - 最小化到托盘"""
if hasattr(self, 'tray_icon') and self.tray_icon.isVisible():
# 首次最小化时显示提示消息
if not hasattr(self, '_tray_message_shown'):
self.tray_icon.showMessage(
"AI客服智能助手",
"程序已最小化到系统托盘。双击托盘图标可重新显示窗口。",
QSystemTrayIcon.Information,
3000
)
self._tray_message_shown = True
# 隐藏窗口而不是关闭
self.hide()
event.ignore()
print("[INFO] 窗口已最小化到系统托盘")
else:
# 如果托盘不可用，则正常退出
self.quit_application()
event.accept()
@@ -371,8 +682,36 @@ def main():
app.setApplicationName(config.WINDOW_TITLE)
app.setApplicationVersion(config.VERSION)
# 设置应用程序唯一ID（重要：避免和Python默认图标混淆）
try:
import ctypes
# 设置应用程序用户模型ID，让Windows识别为独立应用
ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID("ShuidropAI.CustomerService.1.0")
print("[INFO] 已设置应用程序唯一ID")
except Exception as e:
print(f"[WARNING] 设置应用程序ID失败: {e}")
# 创建主窗口
window = LoginWindow()
# 统一设置所有图标（使用任务栏修复模块）
try:
app_icon = setup_windows_taskbar_icon(app, window)
print("[INFO] 所有图标设置完成")
except Exception as e:
print(f"[WARNING] 图标设置失败: {e}")
# 备用方案：使用简单的图标设置
try:
icon_path = os.path.join(os.path.dirname(__file__), "static", "ai_assistant_icon_32.png")
if os.path.exists(icon_path):
from PyQt5.QtGui import QIcon
backup_icon = QIcon(icon_path)
app.setWindowIcon(backup_icon)
window.setWindowIcon(backup_icon)
print(f"[INFO] 已使用备用图标: {icon_path}")
except Exception as e2:
print(f"[ERROR] 备用图标也设置失败: {e2}")
window.show() # 程序启动断点
# 运行应用程序
@@ -380,7 +719,7 @@ def main():
if __name__ == '__main__':
main() # sd_aAoHIO9fDRIkePZEhW6oaFgK6IzAPxuB 测试令牌(token)
main() # sd_acF0TisgfFOtsBm4ytqb17MQbcxuX9Vp 测试令牌(token)
# username = "KLD测试"
# password = "kld168168"
# taobao nickname = "tb420723827:redboat"