feat: 完成T-501设置页AI模型管理
- 新增SettingsTab,接入AI模型下拉、新增、删除、详情编辑和保存 - 新增AIModelTestWorker,后台调用appconfig.test_ai_model测试连接 - 密钥输入使用密码框打码,删除保护保持text/image模型类别约束 - 补充GUI测试覆盖模型加载、保存删除和测试连接worker - 同步任务看板、current-state、routes、api和progress
This commit is contained in:
+408
-7
@@ -11,6 +11,7 @@ try:
|
|||||||
from PySide6.QtWidgets import (
|
from PySide6.QtWidgets import (
|
||||||
QAbstractItemView,
|
QAbstractItemView,
|
||||||
QApplication,
|
QApplication,
|
||||||
|
QCheckBox,
|
||||||
QComboBox,
|
QComboBox,
|
||||||
QDialog,
|
QDialog,
|
||||||
QDialogButtonBox,
|
QDialogButtonBox,
|
||||||
@@ -2416,6 +2417,402 @@ if QT_IMPORT_ERROR is None:
|
|||||||
return {"alias": self.account.alias, "status": status}
|
return {"alias": self.account.alias, "status": status}
|
||||||
|
|
||||||
|
|
||||||
|
class AIModelTestWorker(BaseWorker):
|
||||||
|
"""Test one AI model connection without blocking the GUI thread."""
|
||||||
|
|
||||||
|
def __init__(self, model_name, ai_models_path=None):
|
||||||
|
super().__init__()
|
||||||
|
self.model_name = model_name
|
||||||
|
self.ai_models_path = ai_models_path or appconfig.AI_MODELS_PATH
|
||||||
|
|
||||||
|
def execute(self):
|
||||||
|
result = appconfig.test_ai_model(self.model_name, path=self.ai_models_path)
|
||||||
|
payload = dict(result or {})
|
||||||
|
payload["name"] = self.model_name
|
||||||
|
return payload
|
||||||
|
|
||||||
|
|
||||||
|
class SettingsTab(QWidget):
|
||||||
|
"""Tab 5: AI model definitions stored in config/ai_models.json."""
|
||||||
|
|
||||||
|
CATEGORY_ITEMS = [("文本", "text"), ("图像", "image")]
|
||||||
|
API_TYPE_ITEMS = [("chat", "chat"), ("images_edits", "images_edits"), ("auto", "auto")]
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
parent=None,
|
||||||
|
config=None,
|
||||||
|
ai_models_path=None,
|
||||||
|
status_callback=None,
|
||||||
|
):
|
||||||
|
super().__init__(parent)
|
||||||
|
self.config = appconfig.load_config() if config is None else config
|
||||||
|
self.ai_models_path = (
|
||||||
|
ai_models_path
|
||||||
|
or self.config.get("ai_models_path")
|
||||||
|
or appconfig.AI_MODELS_PATH
|
||||||
|
)
|
||||||
|
self.status_callback = status_callback
|
||||||
|
self.models = []
|
||||||
|
self.current_model_name = None
|
||||||
|
self.test_worker = None
|
||||||
|
self.test_thread = None
|
||||||
|
|
||||||
|
self.model_combo = QComboBox()
|
||||||
|
self.model_combo.setObjectName("aiModelCombo")
|
||||||
|
self.add_model_button = QPushButton("新增")
|
||||||
|
self.delete_model_button = QPushButton("删除")
|
||||||
|
|
||||||
|
left_panel = QWidget()
|
||||||
|
left_layout = QVBoxLayout(left_panel)
|
||||||
|
left_layout.setContentsMargins(0, 0, 12, 0)
|
||||||
|
left_layout.addWidget(QLabel("AI 模型"))
|
||||||
|
left_layout.addWidget(self.model_combo)
|
||||||
|
left_toolbar = QHBoxLayout()
|
||||||
|
left_toolbar.addWidget(self.add_model_button)
|
||||||
|
left_toolbar.addWidget(self.delete_model_button)
|
||||||
|
left_toolbar.addStretch(1)
|
||||||
|
left_layout.addLayout(left_toolbar)
|
||||||
|
left_layout.addStretch(1)
|
||||||
|
|
||||||
|
self.enabled_checkbox = QCheckBox("启用")
|
||||||
|
self.name_edit = QLineEdit()
|
||||||
|
self.name_edit.setObjectName("modelNameEdit")
|
||||||
|
self.category_combo = QComboBox()
|
||||||
|
self.category_combo.setObjectName("modelCategoryCombo")
|
||||||
|
for label, value in self.CATEGORY_ITEMS:
|
||||||
|
self.category_combo.addItem(label, value)
|
||||||
|
self.api_type_combo = QComboBox()
|
||||||
|
self.api_type_combo.setObjectName("modelApiTypeCombo")
|
||||||
|
for label, value in self.API_TYPE_ITEMS:
|
||||||
|
self.api_type_combo.addItem(label, value)
|
||||||
|
self.model_id_edit = QLineEdit()
|
||||||
|
self.model_id_edit.setObjectName("modelIdEdit")
|
||||||
|
self.url_edit = QLineEdit()
|
||||||
|
self.url_edit.setObjectName("modelUrlEdit")
|
||||||
|
self.api_key_edit = QLineEdit()
|
||||||
|
self.api_key_edit.setObjectName("modelApiKeyEdit")
|
||||||
|
self.api_key_edit.setEchoMode(QLineEdit.Password)
|
||||||
|
self.connect_timeout_spin = QSpinBox()
|
||||||
|
self.connect_timeout_spin.setObjectName("connectTimeoutSpin")
|
||||||
|
self.connect_timeout_spin.setRange(1, 3600)
|
||||||
|
self.connect_timeout_spin.setValue(30)
|
||||||
|
self.save_model_button = QPushButton("保存")
|
||||||
|
self.test_connection_button = QPushButton("测试连接")
|
||||||
|
self.test_result_label = QLabel("")
|
||||||
|
self.test_result_label.setWordWrap(True)
|
||||||
|
|
||||||
|
form = QFormLayout()
|
||||||
|
form.addRow("", self.enabled_checkbox)
|
||||||
|
form.addRow("服务商名", self.name_edit)
|
||||||
|
form.addRow("类别", self.category_combo)
|
||||||
|
form.addRow("api_type", self.api_type_combo)
|
||||||
|
form.addRow("模型ID", self.model_id_edit)
|
||||||
|
form.addRow("网址", self.url_edit)
|
||||||
|
form.addRow("密钥", self.api_key_edit)
|
||||||
|
form.addRow("连接超时(秒)", self.connect_timeout_spin)
|
||||||
|
|
||||||
|
action_layout = QHBoxLayout()
|
||||||
|
action_layout.addWidget(self.save_model_button)
|
||||||
|
action_layout.addWidget(self.test_connection_button)
|
||||||
|
action_layout.addStretch(1)
|
||||||
|
|
||||||
|
right_panel = QWidget()
|
||||||
|
right_layout = QVBoxLayout(right_panel)
|
||||||
|
right_layout.setContentsMargins(12, 0, 0, 0)
|
||||||
|
right_layout.addWidget(QLabel("模型详情"))
|
||||||
|
right_layout.addLayout(form)
|
||||||
|
right_layout.addLayout(action_layout)
|
||||||
|
right_layout.addWidget(self.test_result_label)
|
||||||
|
right_layout.addStretch(1)
|
||||||
|
|
||||||
|
self.splitter = QSplitter(Qt.Horizontal)
|
||||||
|
self.splitter.addWidget(left_panel)
|
||||||
|
self.splitter.addWidget(right_panel)
|
||||||
|
self.splitter.setStretchFactor(0, 1)
|
||||||
|
self.splitter.setStretchFactor(1, 3)
|
||||||
|
self.splitter.setSizes([280, 860])
|
||||||
|
|
||||||
|
layout = QVBoxLayout(self)
|
||||||
|
layout.setContentsMargins(18, 18, 18, 18)
|
||||||
|
layout.addWidget(self.splitter, 1)
|
||||||
|
|
||||||
|
self.model_combo.currentIndexChanged.connect(self.load_selected_model)
|
||||||
|
self.add_model_button.clicked.connect(self.add_model)
|
||||||
|
self.delete_model_button.clicked.connect(self.delete_model)
|
||||||
|
self.save_model_button.clicked.connect(self.save_model)
|
||||||
|
self.test_connection_button.clicked.connect(self.test_connection)
|
||||||
|
|
||||||
|
self.refresh_models()
|
||||||
|
|
||||||
|
def _set_status(self, message):
|
||||||
|
if self.status_callback is not None:
|
||||||
|
self.status_callback(message)
|
||||||
|
|
||||||
|
def refresh_models(self, selected=None):
|
||||||
|
try:
|
||||||
|
self.models = appconfig.list_ai_models(
|
||||||
|
path=self.ai_models_path,
|
||||||
|
reveal_api_key=True,
|
||||||
|
)
|
||||||
|
except Exception as exc:
|
||||||
|
self.models = []
|
||||||
|
self.current_model_name = None
|
||||||
|
self._show_error(exc)
|
||||||
|
|
||||||
|
current = selected or self.current_model_name
|
||||||
|
self.model_combo.blockSignals(True)
|
||||||
|
self.model_combo.clear()
|
||||||
|
for model in self.models:
|
||||||
|
label = f"{model['name']} · {self._category_label(model['category'])}"
|
||||||
|
if not model.get("enabled", True):
|
||||||
|
label += " · 已停用"
|
||||||
|
self.model_combo.addItem(label, model["name"])
|
||||||
|
index = self.model_combo.findData(current)
|
||||||
|
self.model_combo.setCurrentIndex(index if index >= 0 else (0 if self.models else -1))
|
||||||
|
self.model_combo.blockSignals(False)
|
||||||
|
self.load_selected_model()
|
||||||
|
|
||||||
|
def load_selected_model(self, index=None):
|
||||||
|
name = self.model_combo.currentData()
|
||||||
|
model = self._model_by_name(name)
|
||||||
|
self.current_model_name = model["name"] if model else None
|
||||||
|
self._populate_form(model)
|
||||||
|
self._update_button_state()
|
||||||
|
|
||||||
|
def add_model(self, checked=False):
|
||||||
|
name = self._unique_model_name("新文本模型")
|
||||||
|
model = {
|
||||||
|
"name": name,
|
||||||
|
"category": "text",
|
||||||
|
"enabled": True,
|
||||||
|
"url": "",
|
||||||
|
"model": "",
|
||||||
|
"api_key": "",
|
||||||
|
"api_type": "chat",
|
||||||
|
"connect_timeout_seconds": 30,
|
||||||
|
"timeout_seconds": 0,
|
||||||
|
"extra_body": {},
|
||||||
|
}
|
||||||
|
try:
|
||||||
|
appconfig.add_ai_model(model, path=self.ai_models_path)
|
||||||
|
except Exception as exc:
|
||||||
|
self._show_error(exc)
|
||||||
|
return
|
||||||
|
self.refresh_models(selected=name)
|
||||||
|
self._set_status(f"AI 模型已新增:{name}")
|
||||||
|
|
||||||
|
def save_model(self, checked=False):
|
||||||
|
model = self._form_values()
|
||||||
|
if model is None:
|
||||||
|
return
|
||||||
|
try:
|
||||||
|
if self.current_model_name is None:
|
||||||
|
appconfig.add_ai_model(model, path=self.ai_models_path)
|
||||||
|
else:
|
||||||
|
appconfig.update_ai_model(
|
||||||
|
self.current_model_name,
|
||||||
|
path=self.ai_models_path,
|
||||||
|
**model,
|
||||||
|
)
|
||||||
|
except Exception as exc:
|
||||||
|
self._show_error(exc)
|
||||||
|
return
|
||||||
|
self.refresh_models(selected=model["name"])
|
||||||
|
self._set_status(f"AI 模型已保存:{model['name']}")
|
||||||
|
|
||||||
|
def delete_model(self, checked=False):
|
||||||
|
model = self._current_model()
|
||||||
|
if model is None:
|
||||||
|
return
|
||||||
|
if not self._can_delete_model(model):
|
||||||
|
self._set_status("每个类别至少保留一个模型,当前模型不能删除")
|
||||||
|
return
|
||||||
|
answer = QMessageBox.question(
|
||||||
|
self,
|
||||||
|
"删除 AI 模型",
|
||||||
|
f"确认删除模型「{model['name']}」?",
|
||||||
|
QMessageBox.Yes | QMessageBox.No,
|
||||||
|
QMessageBox.No,
|
||||||
|
)
|
||||||
|
if answer != QMessageBox.Yes:
|
||||||
|
return
|
||||||
|
try:
|
||||||
|
appconfig.delete_ai_model(model["name"], path=self.ai_models_path)
|
||||||
|
except Exception as exc:
|
||||||
|
self._show_error(exc)
|
||||||
|
return
|
||||||
|
self.refresh_models()
|
||||||
|
self._set_status(f"AI 模型已删除:{model['name']}")
|
||||||
|
|
||||||
|
def test_connection(self, checked=False):
|
||||||
|
if self.test_thread is not None:
|
||||||
|
self._set_status("模型连接测试正在进行...")
|
||||||
|
return
|
||||||
|
model = self._current_model()
|
||||||
|
if model is None:
|
||||||
|
return
|
||||||
|
if self.name_edit.text().strip() != model["name"]:
|
||||||
|
self._set_status("请先保存模型名称变更后再测试连接")
|
||||||
|
return
|
||||||
|
worker = AIModelTestWorker(model["name"], ai_models_path=self.ai_models_path)
|
||||||
|
worker.finished.connect(self._on_test_finished)
|
||||||
|
worker.failed.connect(self._on_test_failed)
|
||||||
|
thread = run_worker(worker, thread_name="AIModelTestWorker", start=False)
|
||||||
|
thread.finished.connect(lambda: self._forget_test_thread(thread))
|
||||||
|
self.test_worker = worker
|
||||||
|
self.test_thread = thread
|
||||||
|
self._set_test_running(True)
|
||||||
|
self.test_result_label.setText("正在测试连接...")
|
||||||
|
self._set_status(f"正在测试 AI 模型连接:{model['name']}")
|
||||||
|
thread.start()
|
||||||
|
|
||||||
|
def _form_values(self):
|
||||||
|
current = self._current_model() or {}
|
||||||
|
name = self.name_edit.text().strip()
|
||||||
|
if not name:
|
||||||
|
self._show_error("AI 模型服务商名不能为空")
|
||||||
|
return None
|
||||||
|
extra_body = current.get("extra_body", {})
|
||||||
|
if not isinstance(extra_body, dict):
|
||||||
|
extra_body = {}
|
||||||
|
return {
|
||||||
|
"name": name,
|
||||||
|
"category": self.category_combo.currentData() or "text",
|
||||||
|
"enabled": self.enabled_checkbox.isChecked(),
|
||||||
|
"url": self.url_edit.text().strip(),
|
||||||
|
"model": self.model_id_edit.text().strip(),
|
||||||
|
"api_key": self.api_key_edit.text(),
|
||||||
|
"api_type": self.api_type_combo.currentData() or "auto",
|
||||||
|
"connect_timeout_seconds": self.connect_timeout_spin.value(),
|
||||||
|
"timeout_seconds": int(current.get("timeout_seconds", 0) or 0),
|
||||||
|
"extra_body": dict(extra_body),
|
||||||
|
}
|
||||||
|
|
||||||
|
def _populate_form(self, model):
|
||||||
|
widgets = [
|
||||||
|
self.enabled_checkbox,
|
||||||
|
self.name_edit,
|
||||||
|
self.category_combo,
|
||||||
|
self.api_type_combo,
|
||||||
|
self.model_id_edit,
|
||||||
|
self.url_edit,
|
||||||
|
self.api_key_edit,
|
||||||
|
self.connect_timeout_spin,
|
||||||
|
]
|
||||||
|
for widget in widgets:
|
||||||
|
widget.blockSignals(True)
|
||||||
|
if model is None:
|
||||||
|
self.enabled_checkbox.setChecked(False)
|
||||||
|
self.name_edit.clear()
|
||||||
|
self.category_combo.setCurrentIndex(0)
|
||||||
|
self.api_type_combo.setCurrentIndex(0)
|
||||||
|
self.model_id_edit.clear()
|
||||||
|
self.url_edit.clear()
|
||||||
|
self.api_key_edit.clear()
|
||||||
|
self.connect_timeout_spin.setValue(30)
|
||||||
|
else:
|
||||||
|
self.enabled_checkbox.setChecked(bool(model.get("enabled", True)))
|
||||||
|
self.name_edit.setText(model.get("name", ""))
|
||||||
|
self._set_combo_by_data(self.category_combo, model.get("category", "text"))
|
||||||
|
self._set_combo_by_data(self.api_type_combo, model.get("api_type", "auto"))
|
||||||
|
self.model_id_edit.setText(model.get("model", ""))
|
||||||
|
self.url_edit.setText(model.get("url", ""))
|
||||||
|
self.api_key_edit.setText(model.get("api_key", ""))
|
||||||
|
self.connect_timeout_spin.setValue(
|
||||||
|
int(model.get("connect_timeout_seconds", 30) or 30)
|
||||||
|
)
|
||||||
|
for widget in widgets:
|
||||||
|
widget.blockSignals(False)
|
||||||
|
|
||||||
|
def _set_combo_by_data(self, combo, value):
|
||||||
|
index = combo.findData(value)
|
||||||
|
combo.setCurrentIndex(index if index >= 0 else 0)
|
||||||
|
|
||||||
|
def _update_button_state(self):
|
||||||
|
has_model = self._current_model() is not None
|
||||||
|
testing = self.test_thread is not None
|
||||||
|
for widget in (
|
||||||
|
self.enabled_checkbox,
|
||||||
|
self.name_edit,
|
||||||
|
self.category_combo,
|
||||||
|
self.api_type_combo,
|
||||||
|
self.model_id_edit,
|
||||||
|
self.url_edit,
|
||||||
|
self.api_key_edit,
|
||||||
|
self.connect_timeout_spin,
|
||||||
|
self.save_model_button,
|
||||||
|
):
|
||||||
|
widget.setEnabled(has_model and not testing)
|
||||||
|
self.add_model_button.setEnabled(not testing)
|
||||||
|
self.delete_model_button.setEnabled(
|
||||||
|
has_model and not testing and self._can_delete_model(self._current_model())
|
||||||
|
)
|
||||||
|
self.test_connection_button.setEnabled(has_model and not testing)
|
||||||
|
|
||||||
|
def _set_test_running(self, running):
|
||||||
|
self._update_button_state()
|
||||||
|
self.test_connection_button.setEnabled(
|
||||||
|
not running and self._current_model() is not None
|
||||||
|
)
|
||||||
|
|
||||||
|
def _forget_test_thread(self, thread):
|
||||||
|
if self.test_thread is thread:
|
||||||
|
self.test_thread = None
|
||||||
|
self.test_worker = None
|
||||||
|
self._set_test_running(False)
|
||||||
|
|
||||||
|
def _on_test_finished(self, payload):
|
||||||
|
if payload.get("ok"):
|
||||||
|
status = payload.get("status")
|
||||||
|
suffix = f"(HTTP {status})" if status else ""
|
||||||
|
message = f"测试连接成功:{payload.get('name')}{suffix}"
|
||||||
|
else:
|
||||||
|
error = payload.get("error") or "连接失败"
|
||||||
|
status = payload.get("status")
|
||||||
|
status_text = f"HTTP {status}," if status else ""
|
||||||
|
message = f"测试连接失败:{status_text}{error}"
|
||||||
|
self.test_result_label.setText(message)
|
||||||
|
self._set_status(message)
|
||||||
|
|
||||||
|
def _on_test_failed(self, _task_id, error):
|
||||||
|
message = f"测试连接失败:{error}"
|
||||||
|
self.test_result_label.setText(message)
|
||||||
|
self._set_status(message)
|
||||||
|
|
||||||
|
def _show_error(self, error):
|
||||||
|
message = str(error)
|
||||||
|
QMessageBox.warning(self, "设置", message)
|
||||||
|
self._set_status(message)
|
||||||
|
|
||||||
|
def _current_model(self):
|
||||||
|
return self._model_by_name(self.current_model_name)
|
||||||
|
|
||||||
|
def _model_by_name(self, name):
|
||||||
|
for model in self.models:
|
||||||
|
if model.get("name") == name:
|
||||||
|
return model
|
||||||
|
return None
|
||||||
|
|
||||||
|
def _unique_model_name(self, base):
|
||||||
|
names = {model.get("name") for model in self.models}
|
||||||
|
if base not in names:
|
||||||
|
return base
|
||||||
|
counter = 2
|
||||||
|
while f"{base} {counter}" in names:
|
||||||
|
counter += 1
|
||||||
|
return f"{base} {counter}"
|
||||||
|
|
||||||
|
def _can_delete_model(self, model):
|
||||||
|
if model is None:
|
||||||
|
return False
|
||||||
|
category = model.get("category")
|
||||||
|
return sum(1 for item in self.models if item.get("category") == category) > 1
|
||||||
|
|
||||||
|
def _category_label(self, category):
|
||||||
|
return {"text": "文本", "image": "图像"}.get(category, category)
|
||||||
|
|
||||||
|
|
||||||
class AccountsTab(QWidget):
|
class AccountsTab(QWidget):
|
||||||
COLUMNS = ["账号名", "别名", "地区", "端口", "登录状态", "备注"]
|
COLUMNS = ["账号名", "别名", "地区", "端口", "登录状态", "备注"]
|
||||||
|
|
||||||
@@ -2672,10 +3069,15 @@ if QT_IMPORT_ERROR is None:
|
|||||||
class MainWindow(QMainWindow):
|
class MainWindow(QMainWindow):
|
||||||
"""Main application window with the fixed five-tab workflow."""
|
"""Main application window with the fixed five-tab workflow."""
|
||||||
|
|
||||||
def __init__(self, db_path=None, config=None):
|
def __init__(self, db_path=None, config=None, ai_models_path=None):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
self.config = appconfig.load_config() if config is None else config
|
self.config = appconfig.load_config() if config is None else config
|
||||||
self.db_path = _database_path(db_path, self.config)
|
self.db_path = _database_path(db_path, self.config)
|
||||||
|
self.ai_models_path = (
|
||||||
|
ai_models_path
|
||||||
|
or self.config.get("ai_models_path")
|
||||||
|
or appconfig.AI_MODELS_PATH
|
||||||
|
)
|
||||||
self.setWindowTitle("cmshopee")
|
self.setWindowTitle("cmshopee")
|
||||||
self.resize(1180, 760)
|
self.resize(1180, 760)
|
||||||
self.tabs = QTabWidget()
|
self.tabs = QTabWidget()
|
||||||
@@ -2714,12 +3116,11 @@ if QT_IMPORT_ERROR is None:
|
|||||||
config=self.config,
|
config=self.config,
|
||||||
status_callback=self.statusBar().showMessage,
|
status_callback=self.statusBar().showMessage,
|
||||||
)
|
)
|
||||||
widget = QWidget()
|
return SettingsTab(
|
||||||
widget.setObjectName(title)
|
config=self.config,
|
||||||
layout = QVBoxLayout(widget)
|
ai_models_path=self.ai_models_path,
|
||||||
layout.setContentsMargins(18, 18, 18, 18)
|
status_callback=self.statusBar().showMessage,
|
||||||
layout.addStretch(1)
|
)
|
||||||
return widget
|
|
||||||
|
|
||||||
def _on_tab_changed(self, index):
|
def _on_tab_changed(self, index):
|
||||||
self.statusBar().showMessage(f"当前:{self.tabs.tabText(index)}")
|
self.statusBar().showMessage(f"当前:{self.tabs.tabText(index)}")
|
||||||
|
|||||||
+1
-1
@@ -78,7 +78,7 @@
|
|||||||
|
|
||||||
| ID | 任务 | 依赖 | 验收要点 | 状态 |
|
| ID | 任务 | 依赖 | 验收要点 | 状态 |
|
||||||
| --- | --- | --- | --- | --- |
|
| --- | --- | --- | --- | --- |
|
||||||
| T-501 | Tab⑤ 设置 · AI 模型管理 UI(master-detail):模型下拉 + 新增/删除 + 详情(服务商名/类别/api_type/模型ID/网址/密钥打码/连接超时) + 测试连接 | T-005, T-104 | 复用 T-005 后端;name 唯一;类别 text/image;至少各一个;密钥本地明文保存、UI 打码;测试连接可用 | TODO |
|
| T-501 | Tab⑤ 设置 · AI 模型管理 UI(master-detail):模型下拉 + 新增/删除 + 详情(服务商名/类别/api_type/模型ID/网址/密钥打码/连接超时) + 测试连接 | T-005, T-104 | 复用 T-005 后端;name 唯一;类别 text/image;至少各一个;密钥本地明文保存、UI 打码;测试连接可用 | DONE |
|
||||||
| T-501b | Tab⑤ 设置 · 角色与生成参数:标题大模型/图片大模型下拉、分辨率(512/1k/2k/4k)、返回超时随分辨率、并发/重试/jpg质量、路径/端口 | T-501 | 角色下拉按类别过滤;分辨率选后返回超时按 resolution_timeouts 自动;持久化 config.json | TODO |
|
| T-501b | Tab⑤ 设置 · 角色与生成参数:标题大模型/图片大模型下拉、分辨率(512/1k/2k/4k)、返回超时随分辨率、并发/重试/jpg质量、路径/端口 | T-501 | 角色下拉按类别过滤;分辨率选后返回超时按 resolution_timeouts 自动;持久化 config.json | TODO |
|
||||||
| T-501c | Tab⑤ 设置 · Shopee 更新安全开关 | T-501b | 配置测试商品 ID、是否允许真实提交、是否允许更新封面、单次最大更新条数、成功后是否自动关闭本次新开编辑 tab;默认关闭真实提交和封面更新;Tab③ 执行前读取并拦截不符合安全配置的更新 | TODO |
|
| T-501c | Tab⑤ 设置 · Shopee 更新安全开关 | T-501b | 配置测试商品 ID、是否允许真实提交、是否允许更新封面、单次最大更新条数、成功后是否自动关闭本次新开编辑 tab;默认关闭真实提交和封面更新;Tab③ 执行前读取并拦截不符合安全配置的更新 | TODO |
|
||||||
| T-502 | 满 9 张封面:删第一张再上传 | T-001 | 实测删除确认框并处理;满 9 张流程正确 | TODO |
|
| T-502 | 满 9 张封面:删第一张再上传 | T-001 | 实测删除确认框并处理;满 9 张流程正确 | TODO |
|
||||||
|
|||||||
+13
-2
@@ -293,14 +293,16 @@ render_prompt(template_text, task) -> str
|
|||||||
```python
|
```python
|
||||||
# GUI 入口
|
# GUI 入口
|
||||||
main() -> int # 创建 QApplication + MainWindow
|
main() -> int # 创建 QApplication + MainWindow
|
||||||
class MainWindow(QMainWindow) # QTabWidget: ①②③④⑤;支持注入 db_path/config 便于测试
|
class MainWindow(QMainWindow) # QTabWidget: ①②③④⑤;支持注入 db_path/config/ai_models_path 便于测试
|
||||||
class CollectTab(QWidget) # ① 导入采集:导入 Excel + 汇总栏 + QTableView 任务列表 + 未匹配略过标记
|
class CollectTab(QWidget) # ① 导入采集:导入 Excel + 汇总栏 + QTableView 任务列表 + 未匹配略过标记
|
||||||
class GenerateTab(QWidget) # ② AI生成:提示词管理 + 筛选任务 + 开始/停止生成 + 新旧封面预览
|
class GenerateTab(QWidget) # ② AI生成:提示词管理 + 筛选任务 + 开始/停止生成 + 新旧封面预览
|
||||||
class ApplyTab(QWidget) # ③ 更新shopee:筛选已生成任务 + 确认后串行执行更新
|
class ApplyTab(QWidget) # ③ 更新shopee:筛选已生成任务 + 确认后串行执行更新
|
||||||
|
class SettingsTab(QWidget) # ⑤ 设置:AI 模型 master-detail 管理
|
||||||
class CollectWorker(BaseWorker) # ① 后台采集:账号就绪预检 -> editor.collect -> db.set_collected/mark_skipped/mark_failed
|
class CollectWorker(BaseWorker) # ① 后台采集:账号就绪预检 -> editor.collect -> db.set_collected/mark_skipped/mark_failed
|
||||||
class GenerateWorker(BaseWorker) # ② 后台生成:ai.generate_batch -> db.set_generated/mark_failed + 进度
|
class GenerateWorker(BaseWorker) # ② 后台生成:ai.generate_batch -> db.set_generated/mark_failed + 进度
|
||||||
class ApplyWorker(BaseWorker) # ③ 后台更新:账号就绪预检 -> editor.apply_task -> db.set_applied/mark_skipped
|
class ApplyWorker(BaseWorker) # ③ 后台更新:账号就绪预检 -> editor.apply_task -> db.set_applied/mark_skipped
|
||||||
class WriteBackWorker(BaseWorker) # ①/③ 后台回写:旧字段或更新结果写回原 Excel
|
class WriteBackWorker(BaseWorker) # ①/③ 后台回写:旧字段或更新结果写回原 Excel
|
||||||
|
class AIModelTestWorker(BaseWorker) # ⑤ 后台测试 AI 模型连接:appconfig.test_ai_model
|
||||||
class TaskTableModel(QAbstractTableModel) # 任务表格模型:账号/别名/商品ID/阶段;未匹配别名显示“略过”
|
class TaskTableModel(QAbstractTableModel) # 任务表格模型:账号/别名/商品ID/阶段;未匹配别名显示“略过”
|
||||||
class GenerateTaskTableModel(QAbstractTableModel) # ② 任务表格模型:店铺/商品ID/旧标题/新标题/状态
|
class GenerateTaskTableModel(QAbstractTableModel) # ② 任务表格模型:店铺/商品ID/旧标题/新标题/状态
|
||||||
class ApplyTaskTableModel(QAbstractTableModel) # ③ 任务表格模型:店铺/商品ID/新标题/新封面/阶段/结果
|
class ApplyTaskTableModel(QAbstractTableModel) # ③ 任务表格模型:店铺/商品ID/新标题/新封面/阶段/结果
|
||||||
@@ -310,7 +312,7 @@ TAB_TITLES: list[str] # 固定 Tab 顺序
|
|||||||
TAB_STYLE: str # 顶层 Tab 栏防误点样式:最小宽度/padding/间距/当前态
|
TAB_STYLE: str # 顶层 Tab 栏防误点样式:最小宽度/padding/间距/当前态
|
||||||
```
|
```
|
||||||
|
|
||||||
`MainWindow` 已实现五 Tab、① 导入采集任务列表、② AI生成布局/提示词/开始生成/停止/封面对照预览、③ 更新shopee筛选列表与确认后串行更新、④ 账号管理;⑤ 仍为空壳,业务控件按后续 Tab 任务逐步填充。缺 PySide6 时 `main()` 返回 1 并输出明确提示。
|
`MainWindow` 已实现五 Tab、① 导入采集任务列表、② AI生成布局/提示词/开始生成/停止/封面对照预览、③ 更新shopee筛选列表与确认后串行更新、④ 账号管理、⑤ AI 模型管理。缺 PySide6 时 `main()` 返回 1 并输出明确提示。
|
||||||
|
|
||||||
主 Tab 栏必须在 `MainWindow` 初始化时应用 `TAB_STYLE`:5 个 Tab 不使用 Qt 默认紧凑宽度,需保证点击区域稳定、间距清晰、当前 Tab 高亮明显。该样式属于全局导航基础,不归后续业务 Tab 任务重复实现。
|
主 Tab 栏必须在 `MainWindow` 初始化时应用 `TAB_STYLE`:5 个 Tab 不使用 Qt 默认紧凑宽度,需保证点击区域稳定、间距清晰、当前 Tab 高亮明显。该样式属于全局导航基础,不归后续业务 Tab 任务重复实现。
|
||||||
|
|
||||||
@@ -320,6 +322,15 @@ TAB_STYLE: str # 顶层 Tab 栏防误点样式:
|
|||||||
- 弹窗字段:账号名、别名、地区、调试端口、密码、备注、slug、数据目录;密码框使用打码显示。
|
- 弹窗字段:账号名、别名、地区、调试端口、密码、备注、slug、数据目录;密码框使用打码显示。
|
||||||
- 「启动登录」只启动 Chrome,人工登录;「检测登录」用 worker 跑 `accounts.detect_login()` 并刷新状态列;「快捷方式」调用 `accounts.create_shortcut()` 生成默认桌面 `.lnk`。
|
- 「启动登录」只启动 Chrome,人工登录;「检测登录」用 worker 跑 `accounts.detect_login()` 并刷新状态列;「快捷方式」调用 `accounts.create_shortcut()` 生成默认桌面 `.lnk`。
|
||||||
|
|
||||||
|
⑤ 设置当前要点(T-501):
|
||||||
|
|
||||||
|
- `SettingsTab` 左侧为 AI 模型下拉与新增/删除按钮,右侧为模型详情表单。
|
||||||
|
- 详情字段:启用、服务商名、类别(text/image)、api_type、模型 ID、网址、密钥、连接超时。
|
||||||
|
- 模型数据读写复用 `appconfig.list_ai_models(..., reveal_api_key=True)`、`add_ai_model()`、`update_ai_model()`、`delete_ai_model()`;保存时保留现有 `extra_body` 与 `timeout_seconds`,不接管 T-501b 的生成参数。
|
||||||
|
- 密钥字段使用 `QLineEdit.Password` 打码显示;明文只写入已 gitignore 的 `config/ai_models.json`,不得进入日志/导出。
|
||||||
|
- 删除按钮在当前类别只剩 1 个模型时禁用;后端仍以“至少启用一个 text 和 image 模型”为硬约束。
|
||||||
|
- 「测试连接」创建 `AIModelTestWorker` 后台调用 `appconfig.test_ai_model()`,GUI 主线程不直接发网络请求。
|
||||||
|
|
||||||
① 导入采集当前要点(T-202/T-202b):
|
① 导入采集当前要点(T-202/T-202b):
|
||||||
|
|
||||||
- 「导入 Excel...」按钮调用 `excel.import_tasks(file_paths, path=db_path)`,导入成功后刷新当前批次任务。
|
- 「导入 Excel...」按钮调用 `excel.import_tasks(file_paths, path=db_path)`,导入成功后刷新当前批次任务。
|
||||||
|
|||||||
@@ -6,10 +6,10 @@
|
|||||||
## 当前快照
|
## 当前快照
|
||||||
|
|
||||||
- 日期:2026-06-27
|
- 日期:2026-06-27
|
||||||
- 阶段:V0 单账号 CDP 流程已验证;V1 已完成 T-000 正式代码包结构、T-001 `app/editor.py` 模块化、T-002 `app/appconfig.py` 应用配置、T-003 SQLite 持久化地基、T-004 本地数据忽略规则、T-005 AI 模型清单后端、T-006 单元测试基座、T-101 账号 user-data-dir 工具、T-102 Chrome 启动器、T-103 登录保活与检测、T-104 PySide6 主窗口骨架、T-104b PySide6 worker 基座、T-105 Tab④ 账号管理、T-106 账号快捷方式、T-201 Excel 导入入库、T-202 Tab① 任务列表与导入按钮、T-202b Tab① 导入汇总栏、T-203 采集旧标题旧封面、T-204 回写旧字段到原 Excel、T-204b 采集完成自动回写、T-205 首次未配账号/Chrome 未启动/未登录引导保护、T-205b 采集后关闭自动新建商品页 tab、T-301 AI 生成接口、T-302 Tab② AI 生成布局与任务列表、T-302p 提示词管理、T-303 Tab② 开始生成/停止/进度、T-401 Tab③ 更新列表筛选与开始更新确认、T-402 Tab③ 确认后串行更新、T-403 Tab③ 结果回写与结束汇总。
|
- 阶段:V0 单账号 CDP 流程已验证;V1 已完成 T-000 正式代码包结构、T-001 `app/editor.py` 模块化、T-002 `app/appconfig.py` 应用配置、T-003 SQLite 持久化地基、T-004 本地数据忽略规则、T-005 AI 模型清单后端、T-006 单元测试基座、T-101 账号 user-data-dir 工具、T-102 Chrome 启动器、T-103 登录保活与检测、T-104 PySide6 主窗口骨架、T-104b PySide6 worker 基座、T-105 Tab④ 账号管理、T-106 账号快捷方式、T-201 Excel 导入入库、T-202 Tab① 任务列表与导入按钮、T-202b Tab① 导入汇总栏、T-203 采集旧标题旧封面、T-204 回写旧字段到原 Excel、T-204b 采集完成自动回写、T-205 首次未配账号/Chrome 未启动/未登录引导保护、T-205b 采集后关闭自动新建商品页 tab、T-301 AI 生成接口、T-302 Tab② AI 生成布局与任务列表、T-302p 提示词管理、T-303 Tab② 开始生成/停止/进度、T-401 Tab③ 更新列表筛选与开始更新确认、T-402 Tab③ 确认后串行更新、T-403 Tab③ 结果回写与结束汇总、T-501 Tab⑤ AI 模型管理 UI。
|
||||||
- 技术栈:Python 3.10+,自研 CDP(websocket-client + requests),SQLite(sqlite3)+ `config.json` + openpyxl + AI(`config/ai_models.json` 通用 HTTP,chat JSON / images_edits),GUI PySide6 5 Tab(已定)。
|
- 技术栈:Python 3.10+,自研 CDP(websocket-client + requests),SQLite(sqlite3)+ `config.json` + openpyxl + AI(`config/ai_models.json` 通用 HTTP,chat JSON / images_edits),GUI PySide6 5 Tab(已定)。
|
||||||
- 生产代码:已建立 `app/` 包 + 根入口 `main.py`;`app/cdp.py` 为已验证 CDP 底座,已区分 `CDP.close()` 断开 WebSocket 与 `close_tab()` 关闭浏览器 target;`app/editor.py` 已封装登录状态检测、标题/封面/采集/更新按钮能力,并在采集结束后只关闭本轮自动新建的商品编辑页 tab、保留用户已有 tab;`app/appconfig.py` 已实现 `config.json` 默认值/读写/更新、AI 参数与端口读取,以及 `config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接;`app/ai.py` 已实现 `gen_title()`/`gen_cover()`/`generate_batch()`,按默认文本/图片模型发起通用 HTTP 调用,支持重试、错误脱敏、图片 URL/base64 解析、resolution resize、jpg_quality 保存、先并发标题再并发封面、逐条 `set_generated`、失败 `mark_failed` 与停止取消未开始项;`app/prompts.py` 已实现标题提示词读写、封面模板 CRUD 与变量替换;`app/db.py` 已实现 SQLite schema、连接 PRAGMA、批次/账号/任务与阶段写库函数;`app/excel.py` 已实现多 Excel 输入列解析、整文件列校验、脏行统计跳过、导入批次与任务入库、别名匹配统计、旧标题/旧封面路径回写原 Excel、更新结果回写原 Excel 与另存副本;`app/config.py` 已实现账号 slug 与 user-data-dir 创建;`app/accounts.py` 已实现账号 CRUD 服务、端口默认分配、启动登录、检测登录、生成快捷方式;`app/chrome.py` 已实现 Chrome 参数拼装、启动、CDP 端口探测、PowerShell `.lnk` 快捷方式生成;`app/gui.py` 已实现 PySide6 `MainWindow`、五 Tab、顶部 Tab 栏防误点样式、① 导入采集的 Excel 导入按钮/导入汇总栏/QTableView 任务列表/未匹配筛选与略过标记/采集旧标题旧封面 worker/采集前账号就绪预检与④引导/采集完成自动回写/旧数据回写重试按钮与 worker、② AI生成左右布局/标题与封面提示词管理/批次店铺状态筛选/任务列表/变量预览/开始生成/停止/进度/双击新旧封面预览与 `GenerateWorker`、③ 更新shopee筛选栏/任务列表/开始更新确认弹窗/确认后 `ApplyWorker` 串行更新/账号就绪预检与④引导/逐条 `set_applied`/自动回写结果到 Excel/结束汇总弹窗/手动回写重试按钮、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏;`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。
|
- 生产代码:已建立 `app/` 包 + 根入口 `main.py`;`app/cdp.py` 为已验证 CDP 底座,已区分 `CDP.close()` 断开 WebSocket 与 `close_tab()` 关闭浏览器 target;`app/editor.py` 已封装登录状态检测、标题/封面/采集/更新按钮能力,并在采集结束后只关闭本轮自动新建的商品编辑页 tab、保留用户已有 tab;`app/appconfig.py` 已实现 `config.json` 默认值/读写/更新、AI 参数与端口读取,以及 `config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接;`app/ai.py` 已实现 `gen_title()`/`gen_cover()`/`generate_batch()`,按默认文本/图片模型发起通用 HTTP 调用,支持重试、错误脱敏、图片 URL/base64 解析、resolution resize、jpg_quality 保存、先并发标题再并发封面、逐条 `set_generated`、失败 `mark_failed` 与停止取消未开始项;`app/prompts.py` 已实现标题提示词读写、封面模板 CRUD 与变量替换;`app/db.py` 已实现 SQLite schema、连接 PRAGMA、批次/账号/任务与阶段写库函数;`app/excel.py` 已实现多 Excel 输入列解析、整文件列校验、脏行统计跳过、导入批次与任务入库、别名匹配统计、旧标题/旧封面路径回写原 Excel、更新结果回写原 Excel 与另存副本;`app/config.py` 已实现账号 slug 与 user-data-dir 创建;`app/accounts.py` 已实现账号 CRUD 服务、端口默认分配、启动登录、检测登录、生成快捷方式;`app/chrome.py` 已实现 Chrome 参数拼装、启动、CDP 端口探测、PowerShell `.lnk` 快捷方式生成;`app/gui.py` 已实现 PySide6 `MainWindow`、五 Tab、顶部 Tab 栏防误点样式、① 导入采集的 Excel 导入按钮/导入汇总栏/QTableView 任务列表/未匹配筛选与略过标记/采集旧标题旧封面 worker/采集前账号就绪预检与④引导/采集完成自动回写/旧数据回写重试按钮与 worker、② AI生成左右布局/标题与封面提示词管理/批次店铺状态筛选/任务列表/变量预览/开始生成/停止/进度/双击新旧封面预览与 `GenerateWorker`、③ 更新shopee筛选栏/任务列表/开始更新确认弹窗/确认后 `ApplyWorker` 串行更新/账号就绪预检与④引导/逐条 `set_applied`/自动回写结果到 Excel/结束汇总弹窗/手动回写重试按钮、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏、⑤ 设置 AI 模型 master-detail 管理与 `AIModelTestWorker` 后台测试连接;`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。
|
||||||
- 测试:`tests/` 已建立;T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`,当前覆盖 appconfig/db/config/accounts/chrome 启动与快捷方式/editor 登录检测与商品 tab 生命周期/excel 导入/旧字段与更新结果回写/ai 标题与封面 HTTP 解析/`generate_batch` 正常、失败与停止/prompts 读写与渲染/gui ① 导入采集/gui ② AI生成布局筛选、提示词管理、生成 worker 与双击预览/gui ③ 更新shopee筛选列表、确认弹窗、`ApplyWorker` 串行更新与账号预检、结果回写与汇总/gui ④ 账号管理/worker signal 与线程包装;CDP/Shopee 改动仍需测试商品手动验证。
|
- 测试:`tests/` 已建立;T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`,当前覆盖 appconfig/db/config/accounts/chrome 启动与快捷方式/editor 登录检测与商品 tab 生命周期/excel 导入/旧字段与更新结果回写/ai 标题与封面 HTTP 解析/`generate_batch` 正常、失败与停止/prompts 读写与渲染/gui ① 导入采集/gui ② AI生成布局筛选、提示词管理、生成 worker 与双击预览/gui ③ 更新shopee筛选列表、确认弹窗、`ApplyWorker` 串行更新与账号预检、结果回写与汇总/gui ④ 账号管理/gui ⑤ AI 模型管理/worker signal 与线程包装;CDP/Shopee 改动仍需测试商品手动验证。
|
||||||
- 数据:`config.json`、`config/ai_models.json`、`cmshopee.db`、`chrome_user_data_dir/`、`images/` 已由 `.gitignore` 排除;运营填写后的 Excel 业务文件默认忽略,标准空模板 `shopee待处理任务模板.xlsx` 可提交;`app/appconfig.py` 首次读取缺失的 `config.json` 时会在本地写默认配置,`app/db.py` 调用 `init_db()` 时会在本地创建 SQLite DB。
|
- 数据:`config.json`、`config/ai_models.json`、`cmshopee.db`、`chrome_user_data_dir/`、`images/` 已由 `.gitignore` 排除;运营填写后的 Excel 业务文件默认忽略,标准空模板 `shopee待处理任务模板.xlsx` 可提交;`app/appconfig.py` 首次读取缺失的 `config.json` 时会在本地写默认配置,`app/db.py` 调用 `init_db()` 时会在本地创建 SQLite DB。
|
||||||
|
|
||||||
## 既定设计要点(文档已定)
|
## 既定设计要点(文档已定)
|
||||||
@@ -32,7 +32,7 @@
|
|||||||
| `prototypes/` | 已有 | 已验证原型/探查脚本(demo/set_title/set_cover/get_title/cookies/inspect_images/grab/1.py),保留作人工回归与探查参考;见 `prototypes/README.md` |
|
| `prototypes/` | 已有 | 已验证原型/探查脚本(demo/set_title/set_cover/get_title/cookies/inspect_images/grab/1.py),保留作人工回归与探查参考;见 `prototypes/README.md` |
|
||||||
| `chrome-remote-debug-lan.md` | 已有 | WSL→Windows CDP 转发排查记录 |
|
| `chrome-remote-debug-lan.md` | 已有 | WSL→Windows CDP 转发排查记录 |
|
||||||
| `app/__init__.py` / `app/__main__.py` / `main.py` | 已有 | 正式包与启动入口;`python main.py` / `python -m app` 可运行占位入口 |
|
| `app/__init__.py` / `app/__main__.py` / `main.py` | 已有 | 正式包与启动入口;`python main.py` / `python -m app` 可运行占位入口 |
|
||||||
| `app/gui.py` | 已有 | T-104/T-105/T-106/T-202/T-202b/T-203/T-204/T-204b/T-205/T-302/T-302p/T-303/T-401/T-402/T-403 产出:PySide6 `QMainWindow` + 五 Tab;顶部 Tab 栏防误点样式;① 导入采集导入按钮、导入汇总栏、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker、采集前账号就绪预检与④引导、采集完成自动回写与手动重试;② AI生成左右布局、提示词管理、筛选栏、任务列表、开始生成/停止/进度、双击新旧封面预览与 `GenerateWorker`;③ 更新shopee筛选栏、任务列表、开始更新确认弹窗、确认后 `ApplyWorker` 串行更新、账号预检与④引导、逐条 `set_applied`、自动回写结果到 Excel、结束汇总弹窗与手动回写重试;④ 账号管理表格、账号弹窗、启动登录、检测登录、快捷方式 |
|
| `app/gui.py` | 已有 | T-104/T-105/T-106/T-202/T-202b/T-203/T-204/T-204b/T-205/T-302/T-302p/T-303/T-401/T-402/T-403/T-501 产出:PySide6 `QMainWindow` + 五 Tab;顶部 Tab 栏防误点样式;① 导入采集导入按钮、导入汇总栏、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker、采集前账号就绪预检与④引导、采集完成自动回写与手动重试;② AI生成左右布局、提示词管理、筛选栏、任务列表、开始生成/停止/进度、双击新旧封面预览与 `GenerateWorker`;③ 更新shopee筛选栏、任务列表、开始更新确认弹窗、确认后 `ApplyWorker` 串行更新、账号预检与④引导、逐条 `set_applied`、自动回写结果到 Excel、结束汇总弹窗与手动回写重试;④ 账号管理表格、账号弹窗、启动登录、检测登录、快捷方式;⑤ 设置 AI 模型下拉、新增/删除、详情编辑、密钥打码、测试连接 worker |
|
||||||
| `app/workers.py` | 已有 | T-104b 产出:`BaseWorker` + 通用 signals + 取消标记 + `run_worker()` QThread 包装 |
|
| `app/workers.py` | 已有 | T-104b 产出:`BaseWorker` + 通用 signals + 取消标记 + `run_worker()` QThread 包装 |
|
||||||
| `app/accounts.py` | 已有 | T-105/T-106 产出:账号 CRUD 服务、目录创建、端口分配、启动登录、检测登录、快捷方式 |
|
| `app/accounts.py` | 已有 | T-105/T-106 产出:账号 CRUD 服务、目录创建、端口分配、启动登录、检测登录、快捷方式 |
|
||||||
| `app/editor.py` | 已有 | T-001/T-103/T-205b 产出:登录状态检测、打开商品页、读/写标题、读/下载封面、采集后关闭自动新建商品页 tab、上传拖封面、更新按钮、apply_task |
|
| `app/editor.py` | 已有 | T-001/T-103/T-205b 产出:登录状态检测、打开商品页、读/写标题、读/下载封面、采集后关闭自动新建商品页 tab、上传拖封面、更新按钮、apply_task |
|
||||||
@@ -42,7 +42,7 @@
|
|||||||
| `app/db.py` | 已有 | T-003 产出:batches/accounts/tasks schema;WAL/busy_timeout/foreign_keys;账号/批次/任务与 set_* 阶段写库 |
|
| `app/db.py` | 已有 | T-003 产出:batches/accounts/tasks schema;WAL/busy_timeout/foreign_keys;账号/批次/任务与 set_* 阶段写库 |
|
||||||
| `app/config.py` | 已有 | T-101 产出:别名→稳定 slug;创建并返回绝对 user-data-dir |
|
| `app/config.py` | 已有 | T-101 产出:别名→稳定 slug;创建并返回绝对 user-data-dir |
|
||||||
| `app/chrome.py` | 已有 | T-102/T-106 产出:Chrome 启动参数、`subprocess.Popen` 启动、`/json/version` 端口探测、PowerShell `.lnk` 快捷方式 |
|
| `app/chrome.py` | 已有 | T-102/T-106 产出:Chrome 启动参数、`subprocess.Popen` 启动、`/json/version` 端口探测、PowerShell `.lnk` 快捷方式 |
|
||||||
| `tests/` | 已有 | T-006/T-201/T-202/T-202b/T-203/T-204/T-204b/T-205/T-301/T-302/T-302p/T-303/T-401/T-402/T-403 产出:stdlib unittest 基座;覆盖 appconfig/db/config/accounts/chrome/editor/excel/ai/prompts/gui/workers |
|
| `tests/` | 已有 | T-006/T-201/T-202/T-202b/T-203/T-204/T-204b/T-205/T-301/T-302/T-302p/T-303/T-401/T-402/T-403/T-501 产出:stdlib unittest 基座;覆盖 appconfig/db/config/accounts/chrome/editor/excel/ai/prompts/gui/workers |
|
||||||
| `app/excel.py` | 已有 | T-201/T-204/T-403 产出:多文件 Excel 输入列解析、必需列整文件拒绝、脏行逐行跳过、批次/任务入库、匹配统计;按源文件/工作表/行号回写旧标题与旧封面路径;按源文件/工作表/行号回写新标题、新封面路径、更新状态;支持原文件被占用时另存副本 |
|
| `app/excel.py` | 已有 | T-201/T-204/T-403 产出:多文件 Excel 输入列解析、必需列整文件拒绝、脏行逐行跳过、批次/任务入库、匹配统计;按源文件/工作表/行号回写旧标题与旧封面路径;按源文件/工作表/行号回写新标题、新封面路径、更新状态;支持原文件被占用时另存副本 |
|
||||||
| `shopee待处理任务模板.xlsx` | 已有,已提交 | 标准空 Excel 模板;单工作表 `待处理任务`,表头 `账号名 | 别名 | 商品id | 旧标题 | 旧封面图片路径 | 新标题 | 新封面图片路径 | 更新状态`;运营复制后填写,填写副本不提交 |
|
| `shopee待处理任务模板.xlsx` | 已有,已提交 | 标准空 Excel 模板;单工作表 `待处理任务`,表头 `账号名 | 别名 | 商品id | 旧标题 | 旧封面图片路径 | 新标题 | 新封面图片路径 | 更新状态`;运营复制后填写,填写副本不提交 |
|
||||||
| `config.json` / `config/ai_models.json` / `cmshopee.db` / `chrome_user_data_dir/` / `images/` | 本地存在或按需生成,已忽略 | 含配置、密钥、业务、登录态、图片,不提交版本库 |
|
| `config.json` / `config/ai_models.json` / `cmshopee.db` / `chrome_user_data_dir/` / `images/` | 本地存在或按需生成,已忽略 | 含配置、密钥、业务、登录态、图片,不提交版本库 |
|
||||||
@@ -60,13 +60,14 @@
|
|||||||
|
|
||||||
任务状态以 [`06-tasks.md`](06-tasks.md) 为准,历史记录见 [`../progress.md`](../progress.md)。
|
任务状态以 [`06-tasks.md`](06-tasks.md) 为准,历史记录见 [`../progress.md`](../progress.md)。
|
||||||
|
|
||||||
- 已完成:T-000(正式代码包结构)、T-001(`app/editor.py` 模块化)、T-002(`app/appconfig.py` + `config.json`)、T-003(SQLite 建表)、T-004(本地数据 gitignore)、T-005(AI 模型清单后端)、T-006(单元测试基座)、T-101(账号 slug/user-data-dir)、T-102(Chrome 启动器)、T-103(登录保活与检测)、T-104(PySide6 五 Tab 主窗口骨架)、T-104b(PySide6 worker 基座)、T-105(Tab④ 账号管理)、T-106(账号快捷方式)、T-201(Excel 导入:解析多文件输入列入库)、T-202(Tab① 任务列表 + 导入按钮 + 别名匹配标记)、T-202b(Tab① 导入汇总栏)、T-203(采集旧标题+旧封面)、T-204(回写旧字段到原 Excel)、T-204b(采集完成自动回写旧字段)、T-205(首次未配账号 / Chrome 未启动 / 未登录引导保护)、T-205b(采集后关闭自动新建商品页 tab)、T-301(AI 生成接口)、T-302(Tab② 左右布局与任务列表)、T-302p(提示词管理)、T-303(Tab② 开始生成 + 停止 + 进度)、T-401(Tab③ 更新列表筛选 + 开始更新确认弹窗)、T-402(Tab③ 确认后串行更新)、T-403(Tab③ 结果回写与结束汇总)。
|
- 已完成:T-000(正式代码包结构)、T-001(`app/editor.py` 模块化)、T-002(`app/appconfig.py` + `config.json`)、T-003(SQLite 建表)、T-004(本地数据 gitignore)、T-005(AI 模型清单后端)、T-006(单元测试基座)、T-101(账号 slug/user-data-dir)、T-102(Chrome 启动器)、T-103(登录保活与检测)、T-104(PySide6 五 Tab 主窗口骨架)、T-104b(PySide6 worker 基座)、T-105(Tab④ 账号管理)、T-106(账号快捷方式)、T-201(Excel 导入:解析多文件输入列入库)、T-202(Tab① 任务列表 + 导入按钮 + 别名匹配标记)、T-202b(Tab① 导入汇总栏)、T-203(采集旧标题+旧封面)、T-204(回写旧字段到原 Excel)、T-204b(采集完成自动回写旧字段)、T-205(首次未配账号 / Chrome 未启动 / 未登录引导保护)、T-205b(采集后关闭自动新建商品页 tab)、T-301(AI 生成接口)、T-302(Tab② 左右布局与任务列表)、T-302p(提示词管理)、T-303(Tab② 开始生成 + 停止 + 进度)、T-401(Tab③ 更新列表筛选 + 开始更新确认弹窗)、T-402(Tab③ 确认后串行更新)、T-403(Tab③ 结果回写与结束汇总)、T-501(Tab⑤ AI 模型管理 UI)。
|
||||||
- 正在进行:无。
|
- 正在进行:无。
|
||||||
- 下一个可领取任务:**T-501(Tab⑤ 设置 · AI 模型管理 UI)**。T-404 真实 Shopee 单条更新冒烟已登记,但依赖 T-501c 设置安全开关完成后再执行。
|
- 下一个可领取任务:**T-501b(Tab⑤ 设置 · 角色与生成参数)**。T-404 真实 Shopee 单条更新冒烟已登记,但依赖 T-501c 设置安全开关完成后再执行。
|
||||||
|
|
||||||
## 当前已知限制
|
## 当前已知限制
|
||||||
|
|
||||||
- ① 采集依赖对应账号 Chrome 已用专属 user-data-dir 和 CDP 端口启动并登录;T-205 已在采集前拦截未配置账号、Chrome 未启动、未登录,并引导去④账号管理,但不会无提示批量启动所有账号 Chrome。
|
- ① 采集依赖对应账号 Chrome 已用专属 user-data-dir 和 CDP 端口启动并登录;T-205 已在采集前拦截未配置账号、Chrome 未启动、未登录,并引导去④账号管理,但不会无提示批量启动所有账号 Chrome。
|
||||||
|
- T-501 已完成 `config/ai_models.json` 模型清单 UI;标题/图片默认模型角色、并发、重试、分辨率、jpg 质量、路径/端口等仍留给 T-501b。
|
||||||
- T-301/T-303 已完成通用 HTTP AI 接口、批量生成编排和 GUI 接入 mock 单测;真实 AI 生成还需要在 `config/ai_models.json` 填入可用 url/model/api_key 后做一次成本可控的小样本实测。
|
- T-301/T-303 已完成通用 HTTP AI 接口、批量生成编排和 GUI 接入 mock 单测;真实 AI 生成还需要在 `config/ai_models.json` 填入可用 url/model/api_key 后做一次成本可控的小样本实测。
|
||||||
- T-403 已完成③更新结果回写与结束汇总;真实 Shopee 更新冒烟暂不执行,延后到 T-501c 设置安全开关完成后做 T-404 单条验收。T-404 默认先只测标题更新,封面更新作为可选子项。
|
- T-403 已完成③更新结果回写与结束汇总;真实 Shopee 更新冒烟暂不执行,延后到 T-501c 设置安全开关完成后做 T-404 单条验收。T-404 默认先只测标题更新,封面更新作为可选子项。
|
||||||
|
|
||||||
|
|||||||
+9
-5
@@ -123,10 +123,13 @@
|
|||||||
|
|
||||||
## ⑤ 设置
|
## ⑤ 设置
|
||||||
|
|
||||||
- AI:服务商 / 文本模型 / 图像模型 / API Key(本地明文存,UI 打码,不入日志)。
|
- AI 模型管理(T-501 已接入):左侧模型下拉 + 新增/删除,右侧详情编辑。
|
||||||
- AI 生成参数:**标题并发数、图片并发数、失败重试次数、分辨率、jpg 质量**。
|
- 字段:启用、服务商名、类别(文本/图像)、api_type、模型ID、网址、密钥、连接超时。
|
||||||
- 本地图片目录(旧封面下载、新封面生成的存放根目录)。
|
- 密钥存在 `config/ai_models.json`,本地明文保存;UI 使用密码框打码显示,不进入日志。
|
||||||
- Chrome 路径、默认调试端口 / 端口范围、超时、DB 路径。
|
- 保存复用 `appconfig` 模型清单后端,保持 name 唯一、至少 text/image 各一个模型。
|
||||||
|
- 「测试连接」通过后台 worker 调 `appconfig.test_ai_model()`,不阻塞主线程。
|
||||||
|
- AI 生成参数(T-501b):标题/图片模型角色下拉、标题并发数、图片并发数、失败重试次数、分辨率、jpg 质量。
|
||||||
|
- 路径与端口(T-501b):本地图片目录、Chrome 路径、默认调试端口 / 端口范围、超时、DB 路径。
|
||||||
|
|
||||||
## 流程导航
|
## 流程导航
|
||||||
|
|
||||||
@@ -153,10 +156,11 @@
|
|||||||
| `GenerateTab(QWidget)` | ② | 左提示词管理 + 右筛选/任务列表;双击看新旧封面;开始生成/停止/进度已接入 `GenerateWorker` |
|
| `GenerateTab(QWidget)` | ② | 左提示词管理 + 右筛选/任务列表;双击看新旧封面;开始生成/停止/进度已接入 `GenerateWorker` |
|
||||||
| `ApplyTab(QWidget)` | ③ | 已生成任务筛选 + 开始更新确认 + 确认后串行更新 + 结果回写与结束汇总 |
|
| `ApplyTab(QWidget)` | ③ | 已生成任务筛选 + 开始更新确认 + 确认后串行更新 + 结果回写与结束汇总 |
|
||||||
| `AccountsTab(QWidget)` | ④ | 账号增删改、启动登录、检测登录、生成快捷方式 |
|
| `AccountsTab(QWidget)` | ④ | 账号增删改、启动登录、检测登录、生成快捷方式 |
|
||||||
| `SettingsTab(QWidget)` | ⑤ | AI/目录/Chrome 配置 |
|
| `SettingsTab(QWidget)` | ⑤ | AI 模型 master-detail 管理;后续接入生成参数、目录、Chrome 配置 |
|
||||||
| `TaskTableModel(QAbstractTableModel)` | ①②③ | 任务表格数据模型,供 `QTableView` 使用 |
|
| `TaskTableModel(QAbstractTableModel)` | ①②③ | 任务表格数据模型,供 `QTableView` 使用 |
|
||||||
| `BaseWorker(QObject)` | 后台 | 定义 `progress/log/row_updated/failed/finished/cancelled` signals |
|
| `BaseWorker(QObject)` | 后台 | 定义 `progress/log/row_updated/failed/finished/cancelled` signals |
|
||||||
| `ApplyWorker(BaseWorker)` | ③ | 账号就绪预检、串行调用 `editor.apply_task()`、逐条 `set_applied()`,失败继续 |
|
| `ApplyWorker(BaseWorker)` | ③ | 账号就绪预检、串行调用 `editor.apply_task()`、逐条 `set_applied()`,失败继续 |
|
||||||
|
| `AIModelTestWorker(BaseWorker)` | ⑤ | 后台调用 `appconfig.test_ai_model()` 测试模型连接 |
|
||||||
| `WriteBackWorker(BaseWorker)` | ①③ | ①回写旧字段;③回写新标题/新封面/更新状态 |
|
| `WriteBackWorker(BaseWorker)` | ①③ | ①回写旧字段;③回写新标题/新封面/更新状态 |
|
||||||
|
|
||||||
> 采集、生成、更新都是耗时操作,使用 `QObject` worker + `QThread`。Worker 不直接操作 QWidget,只通过 signal 通知主线程刷新 UI。
|
> 采集、生成、更新都是耗时操作,使用 `QObject` worker + `QThread`。Worker 不直接操作 QWidget,只通过 signal 通知主线程刷新 UI。
|
||||||
|
|||||||
+10
@@ -575,3 +575,13 @@
|
|||||||
- 细化:取消语义明确为协作式取消,只阻止后续未开始任务,不中断当前正在执行的商品;Tab① 旧标题/旧封面采集已有真机冒烟,未真机闭环的是真实 AI 生成与 Tab③ 在线提交更新。
|
- 细化:取消语义明确为协作式取消,只阻止后续未开始任务,不中断当前正在执行的商品;Tab① 旧标题/旧封面采集已有真机冒烟,未真机闭环的是真实 AI 生成与 Tab③ 在线提交更新。
|
||||||
- 变更:更新 `docs/06-tasks.md`、`docs/current-state.md`、`docs/reviews/2026-06-27-T401-T403-acceptance.md`。
|
- 变更:更新 `docs/06-tasks.md`、`docs/current-state.md`、`docs/reviews/2026-06-27-T401-T403-acceptance.md`。
|
||||||
- 验证:仅文档更新,未运行测试。
|
- 验证:仅文档更新,未运行测试。
|
||||||
|
|
||||||
|
## 【2026-06-27】T-501 Tab⑤ AI 模型管理 UI
|
||||||
|
|
||||||
|
- 状态:DONE
|
||||||
|
- 变更:`app/gui.py` 新增 `SettingsTab`,将⑤设置从空壳替换为 AI 模型 master-detail 管理界面。左侧为模型下拉、新增、删除;右侧为启用、服务商名、类别、api_type、模型ID、网址、密钥、连接超时、保存、测试连接。
|
||||||
|
- 安全:密钥字段使用 `QLineEdit.Password` 打码;模型清单仍写入 gitignore 的 `config/ai_models.json`;删除按钮在当前类别只剩一个模型时禁用,后端继续保证至少有 text/image 各一个启用模型。
|
||||||
|
- Worker:新增 `AIModelTestWorker`,通过后台线程调用 `appconfig.test_ai_model()`,避免在 GUI 主线程发网络请求。
|
||||||
|
- 测试:`tests/test_gui.py` 覆盖 MainWindow 挂载⑤、模型加载与密钥打码、新增/保存/删除、测试连接 worker、`AIModelTestWorker.execute()`。
|
||||||
|
- 文档:`docs/06-tasks.md` 将 T-501 标为 DONE;同步 `docs/routes.md`、`docs/api.md`、`docs/current-state.md`,下一个可领取任务更新为 T-501b。
|
||||||
|
- 验证:`python -m compileall app main.py tests` 通过;`python -m unittest discover -s tests -p test_gui.py` 通过(36 tests);`python -m unittest discover -s tests` 通过(92 tests)。
|
||||||
|
|||||||
+154
-1
@@ -9,7 +9,7 @@ sys.path.insert(0, os.path.dirname(__file__))
|
|||||||
from _helpers import TempDirMixin
|
from _helpers import TempDirMixin
|
||||||
|
|
||||||
from app import gui
|
from app import gui
|
||||||
from app import accounts, db, prompts
|
from app import accounts, appconfig, db, prompts
|
||||||
|
|
||||||
if gui.QT_IMPORT_ERROR is not None:
|
if gui.QT_IMPORT_ERROR is not None:
|
||||||
raise unittest.SkipTest("PySide6 未安装")
|
raise unittest.SkipTest("PySide6 未安装")
|
||||||
@@ -20,6 +20,7 @@ from PySide6.QtWidgets import QApplication, QLineEdit, QPlainTextEdit, QTableVie
|
|||||||
from app.gui import (
|
from app.gui import (
|
||||||
AccountDialog,
|
AccountDialog,
|
||||||
AccountsTab,
|
AccountsTab,
|
||||||
|
AIModelTestWorker,
|
||||||
ApplyTab,
|
ApplyTab,
|
||||||
ApplyWorker,
|
ApplyWorker,
|
||||||
CollectWorker,
|
CollectWorker,
|
||||||
@@ -27,6 +28,7 @@ from app.gui import (
|
|||||||
GenerateWorker,
|
GenerateWorker,
|
||||||
GenerateTab,
|
GenerateTab,
|
||||||
MainWindow,
|
MainWindow,
|
||||||
|
SettingsTab,
|
||||||
TAB_STYLE,
|
TAB_STYLE,
|
||||||
TAB_TITLES,
|
TAB_TITLES,
|
||||||
WriteBackWorker,
|
WriteBackWorker,
|
||||||
@@ -51,6 +53,7 @@ class GuiTests(TempDirMixin, unittest.TestCase):
|
|||||||
"image_dir": os.path.join(temp_dir, "images"),
|
"image_dir": os.path.join(temp_dir, "images"),
|
||||||
"db_path": os.path.join(temp_dir, "cmshopee.db"),
|
"db_path": os.path.join(temp_dir, "cmshopee.db"),
|
||||||
"debug_port_range": [9222, 9260],
|
"debug_port_range": [9222, 9260],
|
||||||
|
"ai_models_path": os.path.join(temp_dir, "ai_models.json"),
|
||||||
}
|
}
|
||||||
|
|
||||||
def test_main_window_has_five_tabs_in_workflow_order(self):
|
def test_main_window_has_five_tabs_in_workflow_order(self):
|
||||||
@@ -71,6 +74,7 @@ class GuiTests(TempDirMixin, unittest.TestCase):
|
|||||||
self.assertIsInstance(window.tabs.widget(0), CollectTab)
|
self.assertIsInstance(window.tabs.widget(0), CollectTab)
|
||||||
self.assertIsInstance(window.tabs.widget(1), GenerateTab)
|
self.assertIsInstance(window.tabs.widget(1), GenerateTab)
|
||||||
self.assertIsInstance(window.tabs.widget(2), ApplyTab)
|
self.assertIsInstance(window.tabs.widget(2), ApplyTab)
|
||||||
|
self.assertIsInstance(window.tabs.widget(4), SettingsTab)
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
"回写旧数据到 Excel",
|
"回写旧数据到 Excel",
|
||||||
window.tabs.widget(0).write_back_button.text(),
|
window.tabs.widget(0).write_back_button.text(),
|
||||||
@@ -78,6 +82,155 @@ class GuiTests(TempDirMixin, unittest.TestCase):
|
|||||||
|
|
||||||
self.assert_removed(temp_dir)
|
self.assert_removed(temp_dir)
|
||||||
|
|
||||||
|
def test_settings_tab_loads_ai_models_and_masks_key_field(self):
|
||||||
|
with self.make_temp_dir() as temp_dir:
|
||||||
|
cfg = self.make_config(temp_dir)
|
||||||
|
models_path = cfg["ai_models_path"]
|
||||||
|
appconfig.save_ai_models_config(
|
||||||
|
{
|
||||||
|
"models": [
|
||||||
|
{
|
||||||
|
"name": "Text A",
|
||||||
|
"category": "text",
|
||||||
|
"enabled": True,
|
||||||
|
"url": "https://example.invalid/text",
|
||||||
|
"model": "text-model",
|
||||||
|
"api_key": "sk-text-secret",
|
||||||
|
"api_type": "chat",
|
||||||
|
"connect_timeout_seconds": 11,
|
||||||
|
"timeout_seconds": 0,
|
||||||
|
"extra_body": {"temperature": 0},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Image A",
|
||||||
|
"category": "image",
|
||||||
|
"enabled": True,
|
||||||
|
"url": "https://example.invalid/image",
|
||||||
|
"model": "image-model",
|
||||||
|
"api_key": "sk-image-secret",
|
||||||
|
"api_type": "auto",
|
||||||
|
"connect_timeout_seconds": 22,
|
||||||
|
"timeout_seconds": 0,
|
||||||
|
"extra_body": {},
|
||||||
|
},
|
||||||
|
]
|
||||||
|
},
|
||||||
|
path=models_path,
|
||||||
|
)
|
||||||
|
|
||||||
|
tab = SettingsTab(config=cfg, ai_models_path=models_path)
|
||||||
|
self.addCleanup(tab.close)
|
||||||
|
|
||||||
|
self.assertEqual(2, tab.model_combo.count())
|
||||||
|
self.assertEqual("Text A", tab.name_edit.text())
|
||||||
|
self.assertEqual("text", tab.category_combo.currentData())
|
||||||
|
self.assertEqual("chat", tab.api_type_combo.currentData())
|
||||||
|
self.assertEqual("text-model", tab.model_id_edit.text())
|
||||||
|
self.assertEqual("https://example.invalid/text", tab.url_edit.text())
|
||||||
|
self.assertEqual("sk-text-secret", tab.api_key_edit.text())
|
||||||
|
self.assertEqual(QLineEdit.Password, tab.api_key_edit.echoMode())
|
||||||
|
self.assertEqual(11, tab.connect_timeout_spin.value())
|
||||||
|
self.assertFalse(tab.delete_model_button.isEnabled())
|
||||||
|
|
||||||
|
self.assert_removed(temp_dir)
|
||||||
|
|
||||||
|
def test_settings_tab_adds_saves_and_deletes_model(self):
|
||||||
|
with self.make_temp_dir() as temp_dir:
|
||||||
|
cfg = self.make_config(temp_dir)
|
||||||
|
models_path = cfg["ai_models_path"]
|
||||||
|
statuses = []
|
||||||
|
tab = SettingsTab(
|
||||||
|
config=cfg,
|
||||||
|
ai_models_path=models_path,
|
||||||
|
status_callback=statuses.append,
|
||||||
|
)
|
||||||
|
self.addCleanup(tab.close)
|
||||||
|
|
||||||
|
tab.add_model()
|
||||||
|
self.assertEqual("新文本模型", tab.current_model_name)
|
||||||
|
self.assertEqual(3, tab.model_combo.count())
|
||||||
|
|
||||||
|
tab.name_edit.setText("Text Custom")
|
||||||
|
tab.url_edit.setText("https://example.invalid/v1/chat/completions")
|
||||||
|
tab.model_id_edit.setText("demo-text")
|
||||||
|
tab.api_key_edit.setText("sk-custom-secret")
|
||||||
|
tab.api_type_combo.setCurrentIndex(tab.api_type_combo.findData("chat"))
|
||||||
|
tab.connect_timeout_spin.setValue(12)
|
||||||
|
tab.save_model()
|
||||||
|
|
||||||
|
saved = appconfig.get_model("Text Custom", path=models_path)
|
||||||
|
self.assertEqual("text", saved["category"])
|
||||||
|
self.assertEqual("demo-text", saved["model"])
|
||||||
|
self.assertEqual("sk-custom-secret", saved["api_key"])
|
||||||
|
self.assertEqual(12, saved["connect_timeout_seconds"])
|
||||||
|
self.assertIn("AI 模型已保存:Text Custom", statuses[-1])
|
||||||
|
self.assertTrue(tab.delete_model_button.isEnabled())
|
||||||
|
|
||||||
|
with mock.patch("app.gui.QMessageBox.question", return_value=gui.QMessageBox.Yes):
|
||||||
|
tab.delete_model()
|
||||||
|
|
||||||
|
names = [model["name"] for model in appconfig.list_ai_models(path=models_path)]
|
||||||
|
self.assertNotIn("Text Custom", names)
|
||||||
|
self.assertIn("AI 模型已删除:Text Custom", statuses[-1])
|
||||||
|
|
||||||
|
self.assert_removed(temp_dir)
|
||||||
|
|
||||||
|
def test_settings_tab_starts_connection_test_worker(self):
|
||||||
|
with self.make_temp_dir() as temp_dir:
|
||||||
|
cfg = self.make_config(temp_dir)
|
||||||
|
models_path = cfg["ai_models_path"]
|
||||||
|
statuses = []
|
||||||
|
tab = SettingsTab(
|
||||||
|
config=cfg,
|
||||||
|
ai_models_path=models_path,
|
||||||
|
status_callback=statuses.append,
|
||||||
|
)
|
||||||
|
self.addCleanup(tab.close)
|
||||||
|
|
||||||
|
class FakeSignal:
|
||||||
|
def __init__(self):
|
||||||
|
self.callbacks = []
|
||||||
|
|
||||||
|
def connect(self, callback):
|
||||||
|
self.callbacks.append(callback)
|
||||||
|
|
||||||
|
class FakeThread:
|
||||||
|
def __init__(self):
|
||||||
|
self.finished = FakeSignal()
|
||||||
|
self.started = False
|
||||||
|
|
||||||
|
def start(self):
|
||||||
|
self.started = True
|
||||||
|
|
||||||
|
fake_thread = FakeThread()
|
||||||
|
with mock.patch("app.gui.run_worker", return_value=fake_thread) as run_worker:
|
||||||
|
tab.test_connection()
|
||||||
|
|
||||||
|
run_worker.assert_called_once()
|
||||||
|
self.assertIsInstance(tab.test_worker, AIModelTestWorker)
|
||||||
|
self.assertIs(tab.test_thread, fake_thread)
|
||||||
|
self.assertTrue(fake_thread.started)
|
||||||
|
self.assertFalse(tab.test_connection_button.isEnabled())
|
||||||
|
self.assertIn("正在测试 AI 模型连接", statuses[-1])
|
||||||
|
|
||||||
|
self.assert_removed(temp_dir)
|
||||||
|
|
||||||
|
def test_ai_model_test_worker_calls_appconfig(self):
|
||||||
|
with self.make_temp_dir() as temp_dir:
|
||||||
|
models_path = os.path.join(temp_dir, "ai_models.json")
|
||||||
|
worker = AIModelTestWorker("Text A", ai_models_path=models_path)
|
||||||
|
|
||||||
|
with mock.patch(
|
||||||
|
"app.gui.appconfig.test_ai_model",
|
||||||
|
return_value={"ok": True, "status": 200},
|
||||||
|
) as test_ai_model:
|
||||||
|
result = worker.execute()
|
||||||
|
|
||||||
|
test_ai_model.assert_called_once_with("Text A", path=models_path)
|
||||||
|
self.assertEqual({"ok": True, "status": 200, "name": "Text A"}, result)
|
||||||
|
|
||||||
|
self.assert_removed(temp_dir)
|
||||||
|
|
||||||
def test_generate_tab_has_prompt_editors_and_task_table(self):
|
def test_generate_tab_has_prompt_editors_and_task_table(self):
|
||||||
with self.make_temp_dir() as temp_dir:
|
with self.make_temp_dir() as temp_dir:
|
||||||
title_prompt_path = os.path.join(temp_dir, "title_prompt.txt")
|
title_prompt_path = os.path.join(temp_dir, "title_prompt.txt")
|
||||||
|
|||||||
Reference in New Issue
Block a user