From 5a63af46206affc8d68eceeea1bc202d90911e99 Mon Sep 17 00:00:00 2001 From: chengma Date: Sat, 27 Jun 2026 16:59:46 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AE=8C=E6=88=90T-401=E6=9B=B4?= =?UTF-8?q?=E6=96=B0=E7=A1=AE=E8=AE=A4=E9=A1=B5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增 ApplyTab 和 ApplyTaskTableModel，③ Tab 展示已生成更新候选任务 - 支持按批次、店铺、状态筛选，默认列出待更新任务并展示失败/已更新状态 - 开始更新弹窗确认当前筛选条件、任务数和线上提交风险，取消不执行不改库 - 补充 GUI 测试覆盖筛选、确认弹窗和不调用 apply_task 的 T-401 边界 - 同步任务看板、API、路由、当前状态与 progress 文档 --- app/gui.py | 311 ++++++++++++++++++++++++++++++++++++++++++ docs/06-tasks.md | 2 +- docs/api.md | 14 +- docs/current-state.md | 15 +- docs/routes.md | 2 +- progress.md | 10 ++ tests/test_gui.py | 166 ++++++++++++++++++++++ 7 files changed, 510 insertions(+), 10 deletions(-) diff --git a/app/gui.py b/app/gui.py index fadc497..563c71f 100644 --- a/app/gui.py +++ b/app/gui.py @@ -289,6 +289,98 @@ if QT_IMPORT_ERROR is None: return values[column] if 0 <= column < len(values) else None + class ApplyTaskTableModel(QAbstractTableModel): + """Table model for Tab 3 update candidates.""" + + HEADERS = ["店铺", "商品ID", "新标题", "新封面", "阶段", "结果"] + + STATUS_TEXT = { + "running": "处理中", + "failed": "失败", + "skipped": "略过", + "cancelled": "已取消", + "pending": "待更新", + "success": "成功", + } + + def __init__(self, parent=None): + super().__init__(parent) + self.tasks = [] + self.account_by_alias = {} + + def set_tasks(self, tasks, accounts): + self.beginResetModel() + self.tasks = list(tasks) + self.account_by_alias = { + str(account.alias).strip(): account + for account in accounts + if str(account.alias).strip() + } + self.endResetModel() + + def rowCount(self, parent=QModelIndex()): + return 0 if parent.isValid() else len(self.tasks) + + def columnCount(self, parent=QModelIndex()): + return 0 if parent.isValid() else len(self.HEADERS) + + def headerData(self, section, orientation, role=Qt.DisplayRole): + if role != Qt.DisplayRole: + return None + if orientation == Qt.Horizontal and 0 <= section < len(self.HEADERS): + return self.HEADERS[section] + return section + 1 if orientation == Qt.Vertical else None + + def data(self, index, role=Qt.DisplayRole): + if not index.isValid(): + return None + task = self.tasks[index.row()] + if role == Qt.DisplayRole: + return self._display_value(task, index.column()) + if role == Qt.ToolTipRole and task.last_error: + return task.last_error + return None + + def flags(self, index): + if not index.isValid(): + return Qt.NoItemFlags + return Qt.ItemIsEnabled | Qt.ItemIsSelectable + + def task_at(self, row): + if row < 0 or row >= len(self.tasks): + return None + return self.tasks[row] + + def account_name_for(self, task): + account = self.account_by_alias.get(str(task.alias).strip()) + if account is not None: + return account.account_name + return task.account_name or task.alias or "" + + def _display_value(self, task, column): + values = [ + self.account_name_for(task), + task.item_id, + task.new_title or "", + os.path.basename(task.new_cover_path or ""), + self._stage_text(task), + self._result_text(task), + ] + return values[column] if 0 <= column < len(values) else None + + def _stage_text(self, task): + if task.stage == "generated": + return "待更新" + if task.stage == "applied": + return "已更新" + return task.stage + + def _result_text(self, task): + if task.status == "success" and task.stage == "generated": + return "待更新" + return self.STATUS_TEXT.get(task.status, task.status) + + class GenerateTab(QWidget): """Tab 2: prompt area plus generation task filters/list.""" @@ -848,6 +940,219 @@ if QT_IMPORT_ERROR is None: return True + class ApplyTab(QWidget): + """Tab 3: list generated tasks and confirm the update scope.""" + + STATUS_FILTERS = [ + ("已生成", "generated"), + ("失败", "failed"), + ("已更新", "applied"), + ("略过", "skipped"), + ("全部状态", "all"), + ] + + def __init__(self, parent=None, db_path=None, config=None, status_callback=None): + super().__init__(parent) + self.config = appconfig.load_config() if config is None else config + self.db_path = _database_path(db_path, self.config) + self.status_callback = status_callback + + self.batch_filter = QComboBox() + self.batch_filter.setObjectName("applyBatchFilter") + self.shop_filter = QComboBox() + self.shop_filter.setObjectName("applyShopFilter") + self.status_filter = QComboBox() + self.status_filter.setObjectName("applyStatusFilter") + for label, value in self.STATUS_FILTERS: + self.status_filter.addItem(label, value) + self.refresh_button = QPushButton("刷新") + + filter_layout = QHBoxLayout() + filter_layout.addWidget(QLabel("批次")) + filter_layout.addWidget(self.batch_filter, 2) + filter_layout.addWidget(QLabel("店铺")) + filter_layout.addWidget(self.shop_filter, 1) + filter_layout.addWidget(QLabel("状态")) + filter_layout.addWidget(self.status_filter, 1) + filter_layout.addWidget(self.refresh_button) + + self.summary_label = QLabel("任务 0 条") + self.risk_label = QLabel("点击「开始更新」后会先确认当前筛选范围；确认后才允许后续任务提交线上。") + self.task_table = QTableView() + self.model = ApplyTaskTableModel(self.task_table) + self.task_table.setModel(self.model) + self.task_table.setSelectionBehavior(QAbstractItemView.SelectRows) + self.task_table.setSelectionMode(QAbstractItemView.SingleSelection) + self.task_table.setEditTriggers(QAbstractItemView.NoEditTriggers) + self.task_table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch) + self.task_table.verticalHeader().setVisible(False) + + self.start_update_button = QPushButton("开始更新") + self.stop_update_button = QPushButton("停止") + self.write_back_button = QPushButton("回写结果到 Excel") + self.stop_update_button.setEnabled(False) + self.write_back_button.setEnabled(False) + + action_layout = QHBoxLayout() + action_layout.addWidget(self.start_update_button) + action_layout.addWidget(self.stop_update_button) + action_layout.addStretch(1) + action_layout.addWidget(self.write_back_button) + + layout = QVBoxLayout(self) + layout.setContentsMargins(18, 18, 18, 18) + layout.addLayout(filter_layout) + layout.addWidget(self.risk_label) + layout.addWidget(self.summary_label) + layout.addWidget(self.task_table, 1) + layout.addLayout(action_layout) + + self.batch_filter.currentIndexChanged.connect(self.refresh_tasks) + self.shop_filter.currentIndexChanged.connect(self.refresh_tasks) + self.status_filter.currentIndexChanged.connect(self.refresh_tasks) + self.refresh_button.clicked.connect(self.refresh_tasks) + self.start_update_button.clicked.connect(self.start_update) + + self.refresh_tasks() + + def _set_status(self, message): + if self.status_callback is not None: + self.status_callback(message) + + def refresh_tasks(self, checked=False): + try: + db.init_db(self.db_path) + batches = db.list_batches(path=self.db_path) + account_rows = accounts.list_accounts(path=self.db_path, config=self.config) + selected_batch = self.batch_filter.currentData() + selected_shop = self.shop_filter.currentData() + selected_status = self.status_filter.currentData() or "generated" + self._populate_batch_filter(batches, selected_batch) + selected_batch = self.batch_filter.currentData() + batch_tasks = [ + task for task in db.list_tasks(batch_id=selected_batch, path=self.db_path) + if self._is_update_task(task) + ] + self._populate_shop_filter(batch_tasks, account_rows, selected_shop) + selected_shop = self.shop_filter.currentData() + filtered_tasks = [ + task for task in batch_tasks + if self._matches_shop(task, selected_shop) + and self._matches_status(task, selected_status) + ] + except Exception as exc: + self.model.set_tasks([], []) + self.summary_label.setText("任务读取失败") + self._set_status(f"更新任务读取失败：{exc}") + return + self.model.set_tasks(filtered_tasks, account_rows) + self.summary_label.setText( + f"任务 {len(filtered_tasks)}/{len(batch_tasks)} 条" + ) + + def start_update(self, checked=False): + tasks = list(self.model.tasks) + if not tasks: + self._set_status("当前筛选结果没有可更新任务") + return + answer = QMessageBox.question( + self, + "确认开始更新", + self._confirmation_message(tasks), + QMessageBox.Yes | QMessageBox.No, + QMessageBox.No, + ) + if answer != QMessageBox.Yes: + self._set_status("已取消开始更新") + return + self._set_status( + f"已确认更新范围：{len(tasks)} 条；实际更新执行将在 T-402 接入" + ) + + def _populate_batch_filter(self, batches, selected_batch): + previous = selected_batch if selected_batch in {batch.id for batch in batches} else None + self.batch_filter.blockSignals(True) + self.batch_filter.clear() + self.batch_filter.addItem("全部批次", None) + for batch in batches: + self.batch_filter.addItem(self._batch_label(batch), batch.id) + index = self.batch_filter.findData(previous) + self.batch_filter.setCurrentIndex(index if index >= 0 else 0) + self.batch_filter.blockSignals(False) + + def _populate_shop_filter(self, tasks, account_rows, selected_shop): + account_by_alias = { + str(account.alias).strip(): account + for account in account_rows + if str(account.alias).strip() + } + aliases = [] + for task in tasks: + alias = str(task.alias).strip() + if alias and alias not in aliases: + aliases.append(alias) + previous = selected_shop if selected_shop in aliases else None + self.shop_filter.blockSignals(True) + self.shop_filter.clear() + self.shop_filter.addItem("全部店铺", None) + for alias in sorted(aliases, key=lambda value: self._shop_label(value, account_by_alias)): + self.shop_filter.addItem(self._shop_label(alias, account_by_alias), alias) + index = self.shop_filter.findData(previous) + self.shop_filter.setCurrentIndex(index if index >= 0 else 0) + self.shop_filter.blockSignals(False) + + def _batch_label(self, batch): + source_files = batch.source_files + first_file = os.path.basename(source_files[0]) if source_files else batch.id + return f"{batch.created_at} · {first_file}" + + def _shop_label(self, alias, account_by_alias): + account = account_by_alias.get(alias) + if account is not None: + return f"{account.account_name} ({alias})" + return alias + + def _status_label(self): + return self.status_filter.currentText() or "已生成" + + def _batch_filter_label(self): + return self.batch_filter.currentText() or "全部批次" + + def _shop_filter_label(self): + return self.shop_filter.currentText() or "全部店铺" + + def _is_update_task(self, task): + if task.stage in {"generated", "applied"}: + return True + return bool((task.new_title or task.new_cover_path) and task.status in {"failed", "skipped"}) + + def _matches_shop(self, task, selected_shop): + return selected_shop is None or str(task.alias).strip() == selected_shop + + def _matches_status(self, task, selected_status): + if selected_status in (None, "all"): + return True + if selected_status == "generated": + return task.stage == "generated" and task.status in {"success", "pending"} + if selected_status == "failed": + return task.status == "failed" + if selected_status == "applied": + return task.stage == "applied" + if selected_status == "skipped": + return task.status == "skipped" + return True + + def _confirmation_message(self, tasks): + return ( + "即将按当前筛选结果开始更新 Shopee 线上商品。\n\n" + f"批次：{self._batch_filter_label()}\n" + f"店铺：{self._shop_filter_label()}\n" + f"状态：{self._status_label()}\n" + f"任务数：{len(tasks)}\n\n" + "确认后后续执行会打开商品编辑页、替换标题和封面，并点击「更新」提交线上。" + ) + + class CollectTab(QWidget): """Tab 1: import Excel files and list imported tasks.""" @@ -1897,6 +2202,12 @@ if QT_IMPORT_ERROR is None: config=self.config, status_callback=self.statusBar().showMessage, ) + if title == "③ 更新shopee": + return ApplyTab( + db_path=self.db_path, + config=self.config, + status_callback=self.statusBar().showMessage, + ) if title == "④ 账号管理": return AccountsTab( db_path=self.db_path, diff --git a/docs/06-tasks.md b/docs/06-tasks.md index a3d1fc7..898420a 100644 --- a/docs/06-tasks.md +++ b/docs/06-tasks.md @@ -69,7 +69,7 @@ | ID | 任务 | 依赖 | 验收要点 | 状态 | | --- | --- | --- | --- | --- | -| T-401 | Tab③ 列出已生成任务 + 按批次/店铺/状态筛选 + 开始更新确认弹窗 | T-303 | 顶部批次/店铺/状态筛选；「开始更新」仅作用于当前筛选结果；弹窗显示筛选条件/任务数/线上提交风险；取消不执行；状态=失败可重试；无常驻提交开关 | TODO | +| T-401 | Tab③ 列出已生成任务 + 按批次/店铺/状态筛选 + 开始更新确认弹窗 | T-303 | 顶部批次/店铺/状态筛选；「开始更新」仅作用于当前筛选结果；弹窗显示筛选条件/任务数/线上提交风险；取消不执行；状态=失败可重试；无常驻提交开关 | DONE | | T-402 | 串行执行 apply：批量确认后换标题+换封面+点更新提交，单条失败继续，立即写库 | T-401, T-001, T-104b | 确认后 worker 串行执行并逐条 set_applied；失败继续；未登录引导保护；未确认时不调用 apply | TODO | | T-403 | 回写结果到原 Excel（新标题/新封面/更新状态）+ 结束弹窗汇总 | T-402, T-204 | 回写原文件（锁处理）；弹窗 成功/失败/略过 | TODO | diff --git a/docs/api.md b/docs/api.md index 1137d0f..733b977 100644 --- a/docs/api.md +++ b/docs/api.md @@ -292,18 +292,20 @@ main() -> int # 创建 QApplication + MainWindow class MainWindow(QMainWindow) # QTabWidget: ①②③④⑤；支持注入 db_path/config 便于测试 class CollectTab(QWidget) # ① 导入采集：导入 Excel + 汇总栏 + QTableView 任务列表 + 未匹配略过标记 class GenerateTab(QWidget) # ② AI生成：提示词管理 + 筛选任务 + 开始/停止生成 + 新旧封面预览 +class ApplyTab(QWidget) # ③ 更新shopee：筛选已生成任务 + 开始更新确认弹窗（T-401） class CollectWorker(BaseWorker) # ① 后台采集：账号就绪预检 -> editor.collect -> db.set_collected/mark_skipped/mark_failed class GenerateWorker(BaseWorker) # ② 后台生成：ai.generate_batch -> db.set_generated/mark_failed + 进度 class WriteBackWorker(BaseWorker) # ① 后台回写：excel.write_back(batch_id) 写旧标题/旧封面到原 Excel class TaskTableModel(QAbstractTableModel) # 任务表格模型：账号/别名/商品ID/阶段；未匹配别名显示“略过” class GenerateTaskTableModel(QAbstractTableModel) # ② 任务表格模型：店铺/商品ID/旧标题/新标题/状态 +class ApplyTaskTableModel(QAbstractTableModel) # ③ 任务表格模型：店铺/商品ID/新标题/新封面/阶段/结果 class AccountsTab(QWidget) # ④ 账号管理：表格 + 增删改 + 启动登录 + 检测登录 + 快捷方式 class AccountDialog(QDialog) # 账号编辑弹窗；密码 QLineEdit.Password TAB_TITLES: list[str] # 固定 Tab 顺序 TAB_STYLE: str # 顶层 Tab 栏防误点样式：最小宽度/padding/间距/当前态 ``` -`MainWindow` 已实现五 Tab、① 导入采集任务列表、② AI生成布局/提示词/开始生成/停止/封面对照预览、④ 账号管理；③⑤ 仍为空壳，业务控件按后续 Tab 任务逐步填充。缺 PySide6 时 `main()` 返回 1 并输出明确提示。 +`MainWindow` 已实现五 Tab、① 导入采集任务列表、② AI生成布局/提示词/开始生成/停止/封面对照预览、③ 更新shopee筛选列表与开始更新确认、④ 账号管理；⑤ 仍为空壳，业务控件按后续 Tab 任务逐步填充。缺 PySide6 时 `main()` 返回 1 并输出明确提示。 主 Tab 栏必须在 `MainWindow` 初始化时应用 `TAB_STYLE`：5 个 Tab 不使用 Qt 默认紧凑宽度，需保证点击区域稳定、间距清晰、当前 Tab 高亮明显。该样式属于全局导航基础，不归后续业务 Tab 任务重复实现。 @@ -340,6 +342,16 @@ TAB_STYLE: str # 顶层 Tab 栏防误点样式： - 「停止」调用 worker 的协作式 `cancel()`；未开始的 Future 取消，不记失败；已完成的 `db.set_generated()` 结果保留，可再次开始生成剩余任务。 - 进度标签显示：`标题x/n · 封面x/n · 失败z`；每条封面生成成功后立即 `db.set_generated()`，失败 `db.mark_failed(..., "generate", error)`，GUI 刷新任务表。 +③ 更新shopee当前要点（T-401）： + +- `ApplyTab` 顶部筛选栏包含：批次、店铺、状态、刷新。批次来自 `db.list_batches()`；店铺来自当前更新候选任务别名并优先显示匹配账号名。 +- ③ 只列出已具备新标题/新封面、可进入更新阶段的任务：`stage=generated/applied`，或已有新字段且 `status=failed/skipped` 的任务。 +- 状态筛选支持：已生成（默认，`stage=generated` 且 `status=success/pending`）、失败、已更新、略过、全部状态。 +- 任务列表使用 `QTableView + ApplyTaskTableModel`，列为：店铺、商品ID、新标题、新封面、阶段、结果。 +- 「开始更新」只读取当前筛选结果；无任务时只提示，不弹确认、不改库。 +- 点击「开始更新」会弹窗展示批次/店铺/状态/任务数和提交线上风险；用户点否/取消时不执行、不改库；用户点是时当前版本只记录“已确认范围”，实际串行 `apply_task()` 执行留给 T-402。 +- ③ 没有常驻提交开关；确认弹窗是提交线上前的边界。 + ## workers 模块（`app/workers.py`，已建，PySide6） ```python diff --git a/docs/current-state.md b/docs/current-state.md index 4db2b06..0957f0a 100644 --- a/docs/current-state.md +++ b/docs/current-state.md @@ -6,10 +6,10 @@ ## 当前快照 - 日期：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② 开始生成/停止/进度。 +- 阶段：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③ 更新列表筛选与开始更新确认。 - 技术栈：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 与另存副本；`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`、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏；`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 ④ 账号管理/worker signal 与线程包装；CDP/Shopee 改动仍需测试商品手动验证。 +- 生产代码：已建立 `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 与另存副本；`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筛选栏/任务列表/开始更新确认弹窗、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏；`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筛选列表与确认弹窗/gui ④ 账号管理/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。 ## 既定设计要点（文档已定） @@ -32,7 +32,7 @@ | `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 转发排查记录 | | `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 产出：PySide6 `QMainWindow` + 五 Tab；顶部 Tab 栏防误点样式；① 导入采集导入按钮、导入汇总栏、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker、采集前账号就绪预检与④引导、采集完成自动回写与手动重试；② AI生成左右布局、提示词管理、筛选栏、任务列表、开始生成/停止/进度、双击新旧封面预览与 `GenerateWorker`；④ 账号管理表格、账号弹窗、启动登录、检测登录、快捷方式 | +| `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 产出：PySide6 `QMainWindow` + 五 Tab；顶部 Tab 栏防误点样式；① 导入采集导入按钮、导入汇总栏、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker、采集前账号就绪预检与④引导、采集完成自动回写与手动重试；② AI生成左右布局、提示词管理、筛选栏、任务列表、开始生成/停止/进度、双击新旧封面预览与 `GenerateWorker`；③ 更新shopee筛选栏、任务列表与开始更新确认弹窗；④ 账号管理表格、账号弹窗、启动登录、检测登录、快捷方式 | | `app/workers.py` | 已有 | T-104b 产出：`BaseWorker` + 通用 signals + 取消标记 + `run_worker()` QThread 包装 | | `app/accounts.py` | 已有 | T-105/T-106 产出：账号 CRUD 服务、目录创建、端口分配、启动登录、检测登录、快捷方式 | | `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/config.py` | 已有 | T-101 产出：别名→稳定 slug；创建并返回绝对 user-data-dir | | `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 产出：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 产出：stdlib unittest 基座；覆盖 appconfig/db/config/accounts/chrome/editor/excel/ai/prompts/gui/workers | | `app/excel.py` | 已有 | T-201/T-204 产出：多文件 Excel 输入列解析、必需列整文件拒绝、脏行逐行跳过、批次/任务入库、匹配统计；按源文件/工作表/行号回写旧标题与旧封面路径；支持原文件被占用时另存副本 | | `shopee待处理任务模板.xlsx` | 已有，已提交 | 标准空 Excel 模板；单工作表 `待处理任务`，表头 `账号名 | 别名 | 商品id | 旧标题 | 旧封面图片路径 | 新标题 | 新封面图片路径 | 更新状态`；运营复制后填写，填写副本不提交 | | `config.json` / `config/ai_models.json` / `cmshopee.db` / `chrome_user_data_dir/` / `images/` | 本地存在或按需生成，已忽略 | 含配置、密钥、业务、登录态、图片，不提交版本库 | @@ -60,14 +60,15 @@ 任务状态以 [`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-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-401（Tab③ 列出已生成任务 + 按批次/店铺/状态筛选 + 开始更新确认弹窗）**。 +- 下一个可领取任务：**T-402（串行执行 apply：批量确认后换标题+换封面+点更新提交，单条失败继续，立即写库）**。 ## 当前已知限制 - ① 采集依赖对应账号 Chrome 已用专属 user-data-dir 和 CDP 端口启动并登录；T-205 已在采集前拦截未配置账号、Chrome 未启动、未登录，并引导去④账号管理，但不会无提示批量启动所有账号 Chrome。 - T-301/T-303 已完成通用 HTTP AI 接口、批量生成编排和 GUI 接入 mock 单测；真实 AI 生成还需要在 `config/ai_models.json` 填入可用 url/model/api_key 后做一次成本可控的小样本实测。 +- T-401 已完成③的筛选列表和开始更新确认弹窗；真正打开 Shopee 商品页、换标题/封面并点击「更新」提交线上留给 T-402。 ## 当前可运行内容 diff --git a/docs/routes.md b/docs/routes.md index 6197454..0d3ec98 100644 --- a/docs/routes.md +++ b/docs/routes.md @@ -149,7 +149,7 @@ | `MainWindow(QMainWindow)` | 根窗口 | 持有 `QTabWidget`、状态栏、全局消息 | | `CollectTab(QWidget)` | ① | 导入、任务表、采集、回写 | | `GenerateTab(QWidget)` | ② | 左提示词管理 + 右筛选/任务列表；双击看新旧封面；开始生成/停止/进度已接入 `GenerateWorker` | -| `ApplyTab(QWidget)` | ③ | 已生成任务、开始更新确认、换标题+封面+提交、回写 | +| `ApplyTab(QWidget)` | ③ | 已生成任务筛选 + 开始更新确认已接入；换标题+封面+提交由 T-402 接入，回写由 T-403 接入 | | `AccountsTab(QWidget)` | ④ | 账号增删改、启动登录、检测登录、生成快捷方式 | | `SettingsTab(QWidget)` | ⑤ | AI/目录/Chrome 配置 | | `TaskTableModel(QAbstractTableModel)` | ①②③ | 任务表格数据模型，供 `QTableView` 使用 | diff --git a/progress.md b/progress.md index a759291..90e02b1 100644 --- a/progress.md +++ b/progress.md @@ -518,3 +518,13 @@ - 变更：更新 `app/cdp.py`、`app/editor.py`；同步 `docs/06-tasks.md`、`docs/04-architecture.md`、`docs/api.md`、`docs/routes.md`、`docs/current-state.md`。 - 测试：`tests/test_editor_login.py` 增加 `open_product()` 新/旧 tab 标记、`collect()` 关闭自动新建 tab、不关闭复用 tab 的覆盖。 - 验证：`python -m unittest discover -s tests -p "test_editor_login.py"` 通过（9 tests）；`python -m compileall app main.py tests` 通过；`python -m unittest discover -s tests` 通过（77 tests）。 + +## 【2026-06-27】T-401 Tab③ 更新列表 + 开始更新确认 + +- 状态：DONE +- 变更：`app/gui.py` 新增 `ApplyTaskTableModel` 与 `ApplyTab`，MainWindow 的③从空壳切换为更新页。③ 顶部提供批次、店铺、状态筛选，默认显示 `stage=generated/status=success|pending` 的已生成待更新任务；状态筛选支持失败、已更新、略过和全部。 +- 细节：③任务表列为店铺、商品ID、新标题、新封面、阶段、结果；店铺优先显示匹配账号名。候选范围只包含已具备新标题/新封面、可进入更新阶段的任务；导入/采集/生成失败但没有新字段的任务不会进入③。 +- 确认边界：点击「开始更新」只作用于当前筛选结果；弹窗展示批次、店铺、状态、任务数和提交线上风险。用户取消时不执行、不改库；用户确认后当前 T-401 只记录“已确认范围”，实际 `apply_task()` 串行执行留给 T-402。无常驻提交开关。 +- 测试：`tests/test_gui.py` 覆盖 MainWindow 挂载③、③筛选列表、失败/已更新状态展示、开始更新确认弹窗，以及确认/取消均不调用 `editor.apply_task()`、不改 DB。 +- 文档：`docs/06-tasks.md` 将 T-401 标为 DONE；同步 `docs/api.md`、`docs/routes.md`、`docs/current-state.md`，下一个可领取任务更新为 T-402。 +- 验证：`python -m unittest discover -s tests -p "test_gui.py"` 通过（24 tests）；`python -m compileall app main.py tests` 通过；`python -m unittest discover -s tests` 通过（79 tests）。 diff --git a/tests/test_gui.py b/tests/test_gui.py index a1c0359..d14b0a0 100644 --- a/tests/test_gui.py +++ b/tests/test_gui.py @@ -20,6 +20,7 @@ from PySide6.QtWidgets import QApplication, QLineEdit, QPlainTextEdit, QTableVie from app.gui import ( AccountDialog, AccountsTab, + ApplyTab, CollectWorker, CollectTab, GenerateWorker, @@ -68,6 +69,7 @@ class GuiTests(TempDirMixin, unittest.TestCase): self.assertIn("margin-right: 8px", window.tabs.styleSheet()) self.assertIsInstance(window.tabs.widget(0), CollectTab) self.assertIsInstance(window.tabs.widget(1), GenerateTab) + self.assertIsInstance(window.tabs.widget(2), ApplyTab) self.assertEqual( "回写旧数据到 Excel", window.tabs.widget(0).write_back_button.text(), @@ -233,6 +235,170 @@ class GuiTests(TempDirMixin, unittest.TestCase): self.assert_removed(temp_dir) + def test_apply_tab_lists_generated_tasks_and_filters_by_batch_shop_status(self): + with self.make_temp_dir() as temp_dir: + cfg = self.make_config(temp_dir) + accounts.create_account("主店", "alias-a", debug_port=9222, config=cfg) + accounts.create_account("副店", "alias-b", debug_port=9223, config=cfg) + batch_a = db.create_batch(["input-a.xlsx"], path=cfg["db_path"]) + batch_b = db.create_batch(["input-b.xlsx"], path=cfg["db_path"]) + db.insert_tasks( + batch_a, + [ + { + "source_file_abs": os.path.join(temp_dir, "input-a.xlsx"), + "source_sheet": "商品", + "source_row": 2, + "account_name": "Excel主店", + "alias": "alias-a", + "item_id": "51100639510", + }, + { + "source_file_abs": os.path.join(temp_dir, "input-a.xlsx"), + "source_sheet": "商品", + "source_row": 3, + "account_name": "Excel副店", + "alias": "alias-b", + "item_id": "51100639511", + }, + { + "source_file_abs": os.path.join(temp_dir, "input-a.xlsx"), + "source_sheet": "商品", + "source_row": 4, + "account_name": "Excel主店", + "alias": "alias-a", + "item_id": "51100639512", + }, + { + "source_file_abs": os.path.join(temp_dir, "input-a.xlsx"), + "source_sheet": "商品", + "source_row": 5, + "account_name": "Excel主店", + "alias": "alias-a", + "item_id": "51100639513", + }, + ], + path=cfg["db_path"], + ) + db.insert_tasks( + batch_b, + [ + { + "source_file_abs": os.path.join(temp_dir, "input-b.xlsx"), + "source_sheet": "商品", + "source_row": 2, + "account_name": "Excel主店", + "alias": "alias-a", + "item_id": "51100639514", + } + ], + path=cfg["db_path"], + ) + tasks_a = db.list_tasks(batch_id=batch_a, path=cfg["db_path"]) + tasks_b = db.list_tasks(batch_id=batch_b, path=cfg["db_path"]) + for task in tasks_a[:3] + tasks_b: + db.set_collected(task.id, "旧标题" + task.item_id[-2:], "old.jpg", path=cfg["db_path"]) + db.set_generated( + task.id, + "新标题" + task.item_id[-2:], + os.path.join(temp_dir, task.item_id + "_new.jpg"), + path=cfg["db_path"], + ) + db.mark_failed(tasks_a[2].id, "apply", "更新失败", path=cfg["db_path"]) + db.set_applied(tasks_b[0].id, True, path=cfg["db_path"]) + + tab = ApplyTab(config=cfg) + self.addCleanup(tab.close) + + self.assertIsInstance(tab.task_table, QTableView) + self.assertEqual(["店铺", "商品ID", "新标题", "新封面", "阶段", "结果"], tab.model.HEADERS) + self.assertEqual("开始更新", tab.start_update_button.text()) + self.assertEqual("停止", tab.stop_update_button.text()) + self.assertFalse(tab.stop_update_button.isEnabled()) + self.assertEqual(2, tab.model.rowCount()) + self.assertEqual("主店", tab.model.index(0, 0).data()) + self.assertEqual("51100639510", tab.model.index(0, 1).data()) + self.assertEqual("新标题10", tab.model.index(0, 2).data()) + self.assertEqual("51100639510_new.jpg", tab.model.index(0, 3).data()) + self.assertEqual("待更新", tab.model.index(0, 4).data()) + self.assertEqual("待更新", tab.model.index(0, 5).data()) + self.assertEqual("任务 2/4 条", tab.summary_label.text()) + + tab.status_filter.setCurrentIndex(tab.status_filter.findData("failed")) + self.assertEqual(1, tab.model.rowCount()) + self.assertEqual("失败", tab.model.index(0, 5).data()) + self.assertEqual( + "更新失败", + tab.model.data(tab.model.index(0, 0), gui.Qt.ToolTipRole), + ) + + tab.status_filter.setCurrentIndex(tab.status_filter.findData("generated")) + tab.shop_filter.setCurrentIndex(tab.shop_filter.findData("alias-b")) + self.assertEqual(1, tab.model.rowCount()) + self.assertEqual("副店", tab.model.index(0, 0).data()) + + tab.shop_filter.setCurrentIndex(tab.shop_filter.findData(None)) + tab.batch_filter.setCurrentIndex(tab.batch_filter.findData(batch_b)) + tab.status_filter.setCurrentIndex(tab.status_filter.findData("applied")) + self.assertEqual(1, tab.model.rowCount()) + self.assertEqual("已更新", tab.model.index(0, 4).data()) + self.assertEqual("成功", tab.model.index(0, 5).data()) + + self.assert_removed(temp_dir) + + def test_apply_tab_start_update_requires_confirmation_without_running_apply(self): + with self.make_temp_dir() as temp_dir: + cfg = self.make_config(temp_dir) + accounts.create_account("主店", "alias-a", debug_port=9222, config=cfg) + batch_id = db.create_batch(["input.xlsx"], path=cfg["db_path"]) + db.insert_tasks( + batch_id, + [ + { + "source_file_abs": os.path.join(temp_dir, "input.xlsx"), + "source_sheet": "商品", + "source_row": 2, + "account_name": "Excel主店", + "alias": "alias-a", + "item_id": "51100639510", + } + ], + path=cfg["db_path"], + ) + task = db.list_tasks(batch_id=batch_id, path=cfg["db_path"])[0] + db.set_collected(task.id, "旧标题", "old.jpg", path=cfg["db_path"]) + db.set_generated(task.id, "新标题", "new.jpg", path=cfg["db_path"]) + statuses = [] + tab = ApplyTab(config=cfg, status_callback=statuses.append) + self.addCleanup(tab.close) + + with mock.patch( + "app.gui.QMessageBox.question", + return_value=gui.QMessageBox.No, + ) as question, mock.patch("app.gui.editor.apply_task") as apply_task: + tab.start_update() + + message = question.call_args[0][2] + self.assertIn("任务数：1", message) + self.assertIn("提交线上", message) + self.assertIn("状态：已生成", message) + self.assertEqual("已取消开始更新", statuses[-1]) + apply_task.assert_not_called() + + with mock.patch( + "app.gui.QMessageBox.question", + return_value=gui.QMessageBox.Yes, + ), mock.patch("app.gui.editor.apply_task") as apply_task: + tab.start_update() + + self.assertIn("已确认更新范围：1 条", statuses[-1]) + apply_task.assert_not_called() + unchanged = db.list_tasks(batch_id=batch_id, path=cfg["db_path"])[0] + self.assertEqual("generated", unchanged.stage) + self.assertEqual("success", unchanged.status) + + self.assert_removed(temp_dir) + def test_generate_tab_lists_tasks_and_filters_by_shop_status_and_batch(self): with self.make_temp_dir() as temp_dir: cfg = self.make_config(temp_dir)