拼多多集成
This commit is contained in:
387
main.py
Normal file
387
main.py
Normal file
@@ -0,0 +1,387 @@
|
||||
import sys
|
||||
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)
|
||||
|
||||
import config
|
||||
from WebSocket.backend_singleton import get_websocket_manager
|
||||
|
||||
|
||||
# 新增: 用户名密码输入对话框类
|
||||
|
||||
class LoginWindow(QMainWindow):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.jd_worker = None
|
||||
self.progress_dialog = None
|
||||
self.douyin_worker = None
|
||||
self.qian_niu_worker = None
|
||||
self.pdd_worker = None
|
||||
|
||||
# 重复执行防护
|
||||
self.platform_combo_connected = False
|
||||
self.last_login_time = 0
|
||||
self.login_cooldown = 2 # 登录冷却时间（秒）
|
||||
|
||||
# 日志管理相关变量已删除
|
||||
|
||||
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) # 设置最小尺寸保证显示完整
|
||||
|
||||
# 创建中央widget
|
||||
central_widget = QWidget()
|
||||
self.setCentralWidget(central_widget)
|
||||
|
||||
# 创建主布局
|
||||
main_layout = QVBoxLayout()
|
||||
main_layout.setSpacing(25) # 增加间距
|
||||
main_layout.setContentsMargins(40, 40, 40, 40) # 增加边距
|
||||
central_widget.setLayout(main_layout)
|
||||
|
||||
# 添加标题
|
||||
title_label = QLabel('AI回复连接入口')
|
||||
title_label.setAlignment(Qt.AlignCenter)
|
||||
title_label.setFont(QFont('Microsoft YaHei', 24, QFont.Bold)) # 使用系统自带字体
|
||||
title_label.setStyleSheet("color: #2c3e50;")
|
||||
main_layout.addWidget(title_label)
|
||||
|
||||
# 添加分隔线
|
||||
line = QFrame()
|
||||
line.setFrameShape(QFrame.HLine)
|
||||
line.setFrameShadow(QFrame.Sunken)
|
||||
line.setStyleSheet("color: #bdc3c7;")
|
||||
main_layout.addWidget(line)
|
||||
|
||||
# 在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)
|
||||
|
||||
# 创建令牌输入区域
|
||||
token_layout = QHBoxLayout()
|
||||
token_layout.setSpacing(15)
|
||||
|
||||
token_label = QLabel('令牌:')
|
||||
token_label.setFont(QFont('Microsoft YaHei', 12))
|
||||
token_label.setFixedWidth(80)
|
||||
|
||||
self.token_input = QLineEdit()
|
||||
self.token_input.setPlaceholderText('请输入您的访问令牌')
|
||||
self.token_input.setEchoMode(QLineEdit.Password)
|
||||
self.token_input.setFont(QFont('Microsoft YaHei', 11))
|
||||
# noinspection PyUnresolvedReferences
|
||||
self.token_input.returnPressed.connect(self.login) # 表示回车提交
|
||||
self.token_input.setMinimumHeight(40) # 增加输入框高度
|
||||
# 预填已保存的令牌（如果存在）
|
||||
try:
|
||||
from config import get_saved_token
|
||||
saved = get_saved_token()
|
||||
if saved:
|
||||
self.token_input.setText(saved)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
token_layout.addWidget(token_label)
|
||||
token_layout.addWidget(self.token_input)
|
||||
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.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.log_display = None
|
||||
|
||||
# 应用现代化样式
|
||||
self.apply_modern_styles()
|
||||
|
||||
# 系统初始化日志输出到终端
|
||||
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;
|
||||
}
|
||||
|
||||
QLabel {
|
||||
color: #34495e;
|
||||
}
|
||||
|
||||
QLineEdit {
|
||||
padding: 10px;
|
||||
border: 2px solid #dfe6e9;
|
||||
border-radius: 6px;
|
||||
font-size: 14px;
|
||||
background-color: white;
|
||||
selection-background-color: #3498db;
|
||||
selection-color: white;
|
||||
}
|
||||
|
||||
QLineEdit:focus {
|
||||
border-color: #3498db;
|
||||
}
|
||||
|
||||
QPushButton {
|
||||
background-color: #3498db;
|
||||
border: none;
|
||||
color: white;
|
||||
padding: 12px 24px;
|
||||
border-radius: 6px;
|
||||
font-size: 14px;
|
||||
min-width: 120px;
|
||||
}
|
||||
|
||||
QPushButton:hover {
|
||||
background-color: #2980b9;
|
||||
}
|
||||
|
||||
QPushButton:pressed {
|
||||
background-color: #1a6a9c;
|
||||
}
|
||||
|
||||
QPushButton:disabled {
|
||||
background-color: #bdc3c7;
|
||||
color: #7f8c8d;
|
||||
}
|
||||
|
||||
QTextEdit {
|
||||
border: 2px solid #dfe6e9;
|
||||
border-radius: 6px;
|
||||
background-color: white;
|
||||
padding: 8px;
|
||||
font-family: 'Consolas', 'Microsoft YaHei', monospace;
|
||||
}
|
||||
|
||||
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;
|
||||
}
|
||||
""")
|
||||
|
||||
# 设置全局字体，确保各Windows版本显示一致
|
||||
font = QFont('Microsoft YaHei', 10) # Windows系统自带字体
|
||||
QApplication.setFont(font)
|
||||
|
||||
# 设置调色板确保颜色一致性
|
||||
palette = QPalette()
|
||||
palette.setColor(QPalette.Window, QColor(249, 249, 249))
|
||||
palette.setColor(QPalette.WindowText, QColor(52, 73, 94))
|
||||
palette.setColor(QPalette.Base, QColor(255, 255, 255))
|
||||
palette.setColor(QPalette.AlternateBase, QColor(233, 235, 239))
|
||||
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.ButtonText, QColor(255, 255, 255))
|
||||
palette.setColor(QPalette.BrightText, QColor(255, 255, 255))
|
||||
palette.setColor(QPalette.Highlight, QColor(52, 152, 219))
|
||||
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("连接中...")
|
||||
|
||||
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")
|
||||
|
||||
except Exception as e:
|
||||
self.add_log(f"连接失败: {e}", "ERROR")
|
||||
finally:
|
||||
self.login_btn.setEnabled(True)
|
||||
self.login_btn.setText("开始连接")
|
||||
|
||||
def verify_token(self, token):
|
||||
"""简化的令牌校验：非空即通过（实际校验由后端承担）"""
|
||||
return bool(token)
|
||||
|
||||
def closeEvent(self, event):
|
||||
"""窗口关闭事件处理"""
|
||||
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:
|
||||
workers.append(self.jd_worker)
|
||||
if hasattr(self, 'douyin_worker') and self.douyin_worker:
|
||||
workers.append(self.douyin_worker)
|
||||
if hasattr(self, 'qian_niu_worker') and self.qian_niu_worker:
|
||||
workers.append(self.qian_niu_worker)
|
||||
if hasattr(self, 'pdd_worker') and self.pdd_worker:
|
||||
workers.append(self.pdd_worker)
|
||||
|
||||
# 停止所有线程
|
||||
for worker in workers:
|
||||
if worker.isRunning():
|
||||
worker.stop()
|
||||
worker.quit()
|
||||
worker.wait(1000)
|
||||
|
||||
# 强制关闭事件循环
|
||||
for worker in workers:
|
||||
if hasattr(worker, 'loop') and worker.loop and not worker.loop.is_closed():
|
||||
worker.loop.close()
|
||||
|
||||
except Exception as e:
|
||||
print(f"关闭窗口时发生错误: {str(e)}")
|
||||
finally:
|
||||
event.accept()
|
||||
|
||||
|
||||
def main():
|
||||
"""主函数，用于测试界面"""
|
||||
app = QApplication(sys.argv)
|
||||
|
||||
# 设置应用程序属性
|
||||
app.setApplicationName(config.WINDOW_TITLE)
|
||||
app.setApplicationVersion(config.VERSION)
|
||||
|
||||
# 创建主窗口
|
||||
window = LoginWindow()
|
||||
window.show() # 程序启动断点
|
||||
|
||||
# 运行应用程序
|
||||
sys.exit(app.exec_())
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main() # sd_aAoHIO9fDRIkePZEhW6oaFgK6IzAPxuB 测试令牌(token)
|
||||
# username = "KLD测试"
|
||||
# password = "kld168168"
|
||||
# taobao nickname = "tb420723827:redboat"
|
||||
|
||||
Reference in New Issue
Block a user