diff --git a/app/ai.py b/app/ai.py index d078b93..e8440cc 100644 --- a/app/ai.py +++ b/app/ai.py @@ -247,7 +247,11 @@ def generation_needs(task, generate_cover=False): return {"title": False, "cover": False} if stage not in {"collected", "generated"}: return {"title": False, "cover": False} - if stage == "generated" and _task_attempt_count(task, "apply_attempts") > 0: + if ( + stage == "generated" + and status == "failed" + and _task_attempt_count(task, "apply_attempts") > 0 + ): return {"title": False, "cover": False} needs_title = not _has_generated_title(task) needs_cover = bool(generate_cover) and not _has_generated_cover(task) diff --git a/app/db.py b/app/db.py index 7c7b6ca..7c10d50 100644 --- a/app/db.py +++ b/app/db.py @@ -743,32 +743,45 @@ def set_applied(task_id, committed, error=None, path=None, conn=None) -> None: ) -def reset_generated(task_id, delete_file=False, path=None, conn=None) -> dict: - """Clear local AI result and move the task back to collected state.""" +def reset_generated( + task_id, + reset_title=True, + reset_cover=True, + delete_file=False, + path=None, + conn=None, +) -> dict: + """Clear selected local AI result components and keep the task generatable.""" + if not reset_title and not reset_cover: + raise DbError("至少需要选择重置标题或重置封面") + if delete_file and not reset_cover: + raise DbError("只有重置封面时才允许删除本地新封面文件") with _connection(conn, path) as database: before = get_task(task_id, conn=database) if before is None: raise DbError(f"任务不存在: {task_id}") new_cover_path = before.new_cover_path deleted_file = None - if delete_file: - raise DbError("重置生成结果默认不删除本地新封面文件，请手动确认后清理") + if delete_file and reset_cover and new_cover_path and os.path.isfile(str(new_cover_path)): + os.remove(str(new_cover_path)) + deleted_file = new_cover_path now = _now() + new_title = None if reset_title else before.new_title + new_cover = None if reset_cover else before.new_cover_path with database: database.execute( """ UPDATE tasks - SET new_title = NULL, - new_cover_path = NULL, - stage = 'collected', + SET new_title = ?, + new_cover_path = ?, + stage = 'generated', status = 'success', last_error = NULL, - generated_at = NULL, updated_at = ? WHERE id = ? """, - (now, int(task_id)), + (new_title, new_cover, now, int(task_id)), ) after = get_task(task_id, conn=database) return { @@ -777,6 +790,8 @@ def reset_generated(task_id, delete_file=False, path=None, conn=None) -> dict: "after": after, "new_cover_path": new_cover_path, "deleted_file": deleted_file, + "reset_title": bool(reset_title), + "reset_cover": bool(reset_cover), } diff --git a/app/gui/tabs/generate.py b/app/gui/tabs/generate.py index 7e042e0..16b6ce3 100644 --- a/app/gui/tabs/generate.py +++ b/app/gui/tabs/generate.py @@ -21,6 +21,11 @@ class GenerateTab(QWidget): ("略过", "skipped"), ("已更新", "applied"), ] + RESET_MODE_LABELS = { + "title": "标题", + "cover": "封面", + "all": "标题和封面", + } def __init__( self, @@ -162,7 +167,7 @@ class GenerateTab(QWidget): self.model = GenerateTaskTableModel(self.task_table, db_path=self.db_path, status_callback=self._set_status) self.task_table.setModel(self.model) self.task_table.setSelectionBehavior(QAbstractItemView.SelectRows) - self.task_table.setSelectionMode(QAbstractItemView.SingleSelection) + self.task_table.setSelectionMode(QAbstractItemView.ExtendedSelection) self.task_table.setEditTriggers(QAbstractItemView.DoubleClicked | QAbstractItemView.EditKeyPressed) self.task_table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch) self.task_table.verticalHeader().setVisible(False) @@ -531,57 +536,191 @@ class GenerateTab(QWidget): if self.generate_thread is not None: self._set_status("AI 生成正在进行，不能重置") return - task = self._selected_task() - if task is None: - self._set_status("请选择要重置生成结果的任务") + scope, tasks = self._reset_scope_tasks() + if not tasks: + self._set_status("当前没有可重置生成结果的任务") return - has_ai_result = bool(getattr(task, "new_title", None) or getattr(task, "new_cover_path", None)) - if not has_ai_result and getattr(task, "stage", None) not in {"generated", "applied"}: - self._set_status("选中任务没有可重置的生成结果") + if not self._resettable_tasks(tasks, reset_title=True, reset_cover=True): + self._set_status("当前范围没有可重置的生成结果") return - lines = [ - "确定重置当前选中任务的本地生成结果吗？", - "", - f"商品ID：{task.item_id}", - f"店铺：{self.model.account_name_for(task)}", - "", - "将清空新标题、新封面路径和错误信息，并退回到已采集状态。", - "默认不删除本地新封面文件，不触碰 Shopee，也不会自动回写 Excel。", - ] - if getattr(task, "new_cover_path", None): - lines.append(f"本地新封面文件保留：{task.new_cover_path}") - if getattr(task, "committed", 0): - lines.extend([ - "", - "注意：该记录曾经提交过线上。本地重置不会回滚 Shopee，后续重新生成/更新可能再次提交线上。", - ]) - answer = QMessageBox.question( - self, - "重置生成结果", - "\n".join(lines), - QMessageBox.Yes | QMessageBox.No, - QMessageBox.No, - ) - if answer != QMessageBox.Yes: + mode = self._ask_reset_generated_mode(tasks, scope) + if mode is None: self._set_status("已取消重置生成结果") return + reset_title, reset_cover = self._reset_mode_flags(mode) + target_tasks = self._resettable_tasks( + tasks, + reset_title=reset_title, + reset_cover=reset_cover, + ) + if not target_tasks: + self._set_status(f"当前范围没有可重置的{self.RESET_MODE_LABELS[mode]}") + return + label = self.RESET_MODE_LABELS[mode] try: - db.reset_generated(task.id, path=self.db_path) - message = ( - "action=reset_generated step=db_write result=success " - f"detail=清空AI生成结果 task_id={task.id}" + for task in target_tasks: + db.reset_generated( + task.id, + reset_title=reset_title, + reset_cover=reset_cover, + path=self.db_path, + ) + run_id = self._write_generated_reset_run_log( + target_tasks, + scope, + mode, + reset_title=reset_title, + reset_cover=reset_cover, ) - run_id = _write_reset_run_log(self.db_path, task, "reset_generated", message) except Exception as exc: QMessageBox.warning(self, "重置生成结果", str(exc)) self._set_status(f"重置生成结果失败：{exc}") return + message = ( + "action=reset_generated step=db_write result=success " + f"detail=重置{label} scope={scope} total={len(target_tasks)}" + ) self.refresh_tasks() self._append_generate_log(message) self._set_status( - f"已重置生成结果：商品 {task.item_id}，run_id={run_id}" + f"已重置生成结果：{len(target_tasks)} 条，内容：{label}，run_id={run_id}" ) + def _reset_scope_tasks(self): + selection = self.task_table.selectionModel() + selected_rows = [] + if selection is not None: + selected_rows = sorted({index.row() for index in selection.selectedRows()}) + if selected_rows: + tasks = [self.model.task_at(row) for row in selected_rows] + return "selected", [task for task in tasks if task is not None] + return "filtered", list(self.model.tasks) + + def _ask_reset_generated_mode(self, tasks, scope): + scope_text = "选中任务" if scope == "selected" else "当前筛选结果" + title_count = sum(1 for task in tasks if getattr(task, "new_title", None)) + cover_count = sum(1 for task in tasks if getattr(task, "new_cover_path", None)) + committed_count = sum(1 for task in tasks if int(getattr(task, "committed", 0) or 0)) + running_count = sum(1 for task in tasks if getattr(task, "status", None) == "running") + sample_ids = [str(getattr(task, "item_id", "") or "") for task in tasks[:5]] + sample_ids = [item_id for item_id in sample_ids if item_id] + sample_text = "、".join(sample_ids) + if len(tasks) > 5: + sample_text += f" 等 {len(tasks)} 条" + lines = [ + "请选择要重置的本地生成内容。", + "", + f"作用范围：{scope_text} {len(tasks)} 条", + f"包含新标题：{title_count} 条", + f"包含新封面：{cover_count} 条", + ] + if sample_text: + lines.append(f"商品ID：{sample_text}") + if running_count: + lines.append(f"运行中任务会跳过：{running_count} 条") + lines.extend([ + "", + "重置标题：只清空新标题，保留新封面。", + "重置封面：只清空新封面，保留新标题和手动微调标题。", + "重置全部：清空新标题和新封面。", + "", + "不会触碰 Shopee，也不会自动回写 Excel；后续重新生成会消耗 cmhub 点数。", + "默认不删除本地新封面文件。", + ]) + if committed_count: + lines.extend([ + "", + f"注意：其中 {committed_count} 条已经提交过线上。本地重置不会回滚 Shopee，重生成后再更新会再次提交线上。", + ]) + box = QMessageBox(self) + box.setWindowTitle("重置生成结果") + box.setText("\n".join(lines)) + title_button = box.addButton("重置标题", QMessageBox.AcceptRole) + cover_button = box.addButton("重置封面", QMessageBox.AcceptRole) + all_button = box.addButton("重置全部", QMessageBox.DestructiveRole) + cancel_button = box.addButton("取消", QMessageBox.RejectRole) + box.setDefaultButton(cancel_button) + box.exec() + clicked = box.clickedButton() + if clicked is title_button: + return "title" + if clicked is cover_button: + return "cover" + if clicked is all_button: + return "all" + return None + + def _reset_mode_flags(self, mode): + if mode == "title": + return True, False + if mode == "cover": + return False, True + return True, True + + def _resettable_tasks(self, tasks, reset_title=True, reset_cover=True): + resettable = [] + for task in tasks: + if getattr(task, "status", None) == "running": + continue + has_title = bool(getattr(task, "new_title", None)) + has_cover = bool(getattr(task, "new_cover_path", None)) + if reset_title and has_title: + resettable.append(task) + continue + if reset_cover and has_cover: + resettable.append(task) + return resettable + + def _write_generated_reset_run_log( + self, + tasks, + scope, + mode, + reset_title=True, + reset_cover=True, + ): + task_ids = [getattr(task, "id", None) for task in tasks] + committed_count = sum(1 for task in tasks if int(getattr(task, "committed", 0) or 0)) + options = { + "action": "reset_generated", + "scope": scope, + "mode": mode, + "reset_title": bool(reset_title), + "reset_cover": bool(reset_cover), + "task_ids": task_ids, + "committed_count": committed_count, + } + run_id = db.create_run_log( + "reset", + total=len(tasks), + options=options, + path=self.db_path, + ) + label = self.RESET_MODE_LABELS[mode] + for task in tasks: + message = ( + "action=reset_generated step=db_write result=success " + f"detail=重置{label} task_id={getattr(task, 'id', None)}" + ) + db.add_run_log_event( + run_id, + message, + task_id=getattr(task, "id", None), + alias=getattr(task, "alias", None), + item_id=getattr(task, "item_id", None), + path=self.db_path, + ) + db.finish_run_log( + run_id, + status="done", + done=len(tasks), + success_count=len(tasks), + failed_count=0, + summary_json=options, + path=self.db_path, + ) + return run_id + def show_task_images(self, index): if index.isValid() and index.column() == 3: return diff --git a/docs/06-tasks.md b/docs/06-tasks.md index d6e761b..f152fdb 100644 --- a/docs/06-tasks.md +++ b/docs/06-tasks.md @@ -77,7 +77,7 @@ | T-401 | Tab③ 列出已生成任务 + 按批次/店铺/商品ID/状态筛选 + 开始更新确认弹窗 | T-303 | 顶部批次/店铺/商品ID/状态筛选；「开始更新」仅作用于当前筛选结果；弹窗显示筛选条件/任务数/线上提交风险；取消不执行；状态=失败可重试；无常驻提交开关 | DONE | | T-402 | 串行执行 apply：批量确认后换标题+换封面+点更新提交，单条失败继续，立即写库 | T-401, T-001, T-104b | 确认后 worker 串行执行并逐条 set_applied；失败继续；未确认时不调用 apply；真实更新前检查本轮账号 Chrome/CDP/登录态，未启动或未登录时整体阻断并提示去④，不自动打开 Chrome | DONE | | T-403 | 回写结果到原 Excel（新标题/新封面/更新状态）+ 结束弹窗汇总 | T-402, T-204 | 回写原文件（锁处理）；弹窗 成功/失败/略过 | DONE | -| T-404a | ②/③ 选中记录重置（联调重试辅助） | T-303, T-403, T-504 | 现状：② AI生成和③ 更新shopee列表没有重置选中记录的功能，生成/上传/提交成功后难以用同一商品重复测试。方案：② 增加「重置生成结果」按钮，仅作用当前选中单条，确认后清空本地 `new_title/new_cover_path`、清空 `last_error`，把任务退回 `stage=collected/status=success`，默认不删除本地新封面文件，可在确认框提示路径；③ 增加「重置更新状态」按钮，仅作用当前选中单条，保留 `new_title/new_cover_path`，确认后把本地状态退回可更新（`stage=generated/status=pending`、清空 `last_error`），用于重复测试上传/提交。若记录 `committed=1`，确认框必须醒目提示“线上已提交过，本地重置不会回滚 Shopee，重复更新会再次提交线上”，并保留 committed 历史事实/运行日志；运行中禁用；两类重置都写 `run_type=reset` 运行日志，不触碰 Shopee、不回写 Excel，后续由手动回写同步本地状态 | DONE | +| T-404a | ②/③ 选中记录重置（联调重试辅助） | T-303, T-403, T-504 | 初版解决 ② AI生成和③ 更新shopee列表无法重置选中记录的问题：③「重置更新状态」保留 `new_title/new_cover_path`，把本地状态退回可更新（`stage=generated/status=pending`、清空 `last_error`），用于重复测试上传/提交。②「重置生成结果」后续已由 T-534 升级，不再按初版退回 `collected` 或仅作用单选；当前以 T-534 为准：支持多选/当前筛选结果，确认框可选「重置标题 / 重置封面 / 重置全部」，只清对应 `new_title` 或 `new_cover_path` 与 `last_error`，保持 `stage=generated/status=success`，由 T-533 后续只补缺失组件；只重置封面时必须保留 `new_title` 和 T-509 手动标题。若记录 `committed=1`，确认框必须醒目提示“线上已提交过，本地重置不会回滚 Shopee，重复更新会再次提交线上”，并保留 committed 历史事实/运行日志；运行中禁用；两类重置都写 `run_type=reset` 运行日志，不触碰 Shopee、不回写 Excel，后续由手动回写同步本地状态 | DONE | | T-404 | 真实 Shopee 单条更新冒烟验收 | T-403, T-501c | 已完成真实验收：2026-07-01 用户手动导入包含 5 个商品 ID 的 Excel，完成 AI 标题/图片生成，并通过 Tab③「开始更新」流程更新到 Shopee；覆盖真实编辑页、标题/封面更新、Shopee 站点侧确认框、提交后结果判断与本地状态链路。后续若 Shopee 页面结构变化，仍需按 CDP 规则补回归验证 | DONE | | T-404b | 商品详情页加载失败 toast 自动捕获 | T-404, T-505 | 问题：③ 更新或①采集打开商品详情页时，如果商品 ID 已失效、无权限或店铺不匹配，Shopee 可能只弹出很快消失的错误 toast，例如 `please input correct product id`；用户来不及复制 HTML，GUI 也可能只报“商品详情页加载失败/等待超时”，无法判断真实原因。方案：在 `open_product()`/等待商品详情页就绪前安装或启用 toast 捕获，监听并兜底读取 `.eds-toasts .eds-toast__content`，保存最近 toast 的文本、`outerHTML`、当前 URL、时间和可见状态到页面内缓存；当标题输入框/图片管理器/上传入口等关键元素超时或页面不是商品编辑页时，优先把最近错误 toast 提升为失败原因，写入 `tasks.last_error`、`run_logs/run_log_events` 和本地 `logs/` 诊断日志，并可保存失败现场 HTML/toast JSON 片段用于排查。若 toast 明确属于商品失效/商品不存在/无权限类错误，则底层仍保持 `stage=imported/status=failed/last_error=商品失效：<原始toast>`，①导入采集列表“阶段”列显示“商品失效”；其他打开失败仍显示“失败”。如果失败发生在 `open_product()` 内部、尚未把 `cdp` 返回给上层，仍必须由 `open_product()` 自己清理：本轮自动新建的商品 tab 要断开 CDP 后关闭浏览器 target；复用用户原本打开的 tab 只断开 CDP，不关闭页面。不得记录 Cookie、密码、token；不绕过 Shopee 校验，不改变上传、删图、提交逻辑。同步 editor/CDP mock 和 GUI model 测试覆盖短暂 toast、隐藏 toast、商品 ID 失效错误文案上浮、自动新开 tab 在商品失效时关闭、复用 tab 不关闭、仅明确失效 toast 才显示“商品失效” | DONE | @@ -133,7 +133,7 @@ | T-531 | ⑤ 设置未保存状态追踪 + 离开确认 | T-530, T-529, T-527 | 现象：在⑤填写 cmhub Base URL/API Key（或改任何设置）后，没有任何"未保存"提示；切换到其它 Tab、关闭窗口都不拦截，输入静默丢失。且「刷新别名/测试连接」用输入框实时文本、不落盘，成功后易误以为已保存。采用**方式 B（显式保存 + 脏状态追踪 + 离开确认）**，不改自动保存（API Key/半份 cmhub 配置不宜自动写盘）。方案：① 脏追踪——⑤ 所有可编辑控件（cmhub Base URL/API Key/别名下拉、模型详情、生成参数、路径端口、Shopee 更新安全等）接 `textChanged`/`currentIndexChanged`/`toggled`/`valueChanged` 置 `self._dirty=True`；「保存设置」成功后清 `False`；`save_app_settings()` 或等价保存入口必须返回成功/失败，失败时保留 dirty 并阻止离开。② 程序化回填保护——加载配置、保存后重载、刷新别名写回下拉、`_populate_app_settings()` / `_populate_cmhub_alias_combos()` 等必须用 `_suspend_dirty` 或 `QSignalBlocker` 避免误置脏；用户手动输入后由刷新/测试触发的 Base URL 规整仍应保持 dirty，提醒其保存。③ 离开拦截——`main_window` 切 Tab 与 `closeEvent` 若发现⑤ `_dirty` 为真，弹三选一「保存 / 放弃 / 取消」：保存则调保存入口且成功后再切/关；放弃必须重新从 `config.json` + `config/cmhub.json` 载入并回填控件，再清 dirty，不能只清标志；取消则留在⑤不切不关。`QTabWidget.currentChanged` 是切换后信号，实现时需维护上一 Tab index，并用 `_reverting_tab_change`/类似 guard 防止 `setCurrentIndex()` 递归触发。④ 视觉提示——「保存设置」旁显示「● 未保存更改」，保存或放弃回填后消失。⑤ 「刷新别名/测试连接」成功后提示语补「别名已拉取，记得点『保存设置』持久化」，**不**顺带偷偷保存 url/key（保持显式保存一致性）。只改 ⑤ 设置页、`main_window` Tab 切换/关闭守卫和 GUI 单测（脏标志置位/清除、切 Tab 保存/放弃/取消/保存失败分支、程序化回填不置脏、放弃会还原输入、currentChanged 回退不递归）；不改配置 schema、AI HTTP 协议、DB schema、Excel、Shopee/CDP 流程 | DONE | | T-532 | ⑤ cmhub 连接成功提示显示账号名 | T-531, T-527, T-528 | 问题：⑤「测试连接/查余额」或刷新别名成功后，界面只提示 `cmhub 连接成功：...`，用户无法确认当前 API Key 实际属于哪个 cmhub 账号，尤其多账号或换 Key 时容易误判。方案：成功 payload 中若 `/api/v1/balance` 返回可识别账号信息，则在结果 label 和状态栏优先显示 `cmhub 账号「<账号名>」连接成功：生文别名 X 个，生图别名 X 个；余额 Y；别名已拉取，记得点『保存设置』持久化`；当前接口结构按 `{ "user": "cmhub_user", "points_balance": 88, "account": { "username": "cmhub_user", "display_name": "主账号" } }` 兼容，显示名优先取 `account.display_name/name/account_name/username/email/id`，再兜底 `user` 字符串或 `user.name/account_name/username/email/id`、顶层 `display_name/name/username/email/id`；邮箱需要脱敏展示；没有账号字段、仅刷新别名不查余额或接口未返回用户信息时，保留现有 `cmhub 连接成功：...` 文案。不得把 API Key、token、完整邮箱或原始敏感响应写入 GUI、run log、诊断日志；不改变 cmhub HTTP 协议、配置 schema、保存逻辑、AI 生成流程或 Shopee/CDP 流程。同步 GUI 单测覆盖有账号名、当前接口结构、只有邮箱、无账号名四种成功提示 | DONE | | T-533 | ② 增量生成：按缺失组件补生成（先标题后封面分离） | T-303, T-520, T-509 | 现象：先关闭「生成封面图片」只生成了标题（`stage=generated/status=success`、`new_cover_path` 为空），之后勾选封面再点「开始生成」，提示「当前筛选结果没有待生成任务」，无法补生成封面；若改用「重置生成结果」会连标题一起清掉、重生成时白扣一次标题点并覆盖 T-509 手动改的标题。根因：`ai.is_generatable_task` 把 `generated+success` 一律判为不可生成，且 `generate_batch` 对每条 eligible 任务都先重生成标题再生成封面。方案：把「开始生成」改为**按缺失组件增量补齐**——① 判定放宽：`generated+success` 且（封面开关 ON 且 `new_cover_path` 为空）判为可生成（只补封面）；② `generate_batch` 组件级跳过：`new_title` 已存在则**跳过标题生成**（复用现有标题，不再调 cmhub 生文、不扣标题点、**不覆盖用户手动标题**），封面开关 ON 且 `new_cover_path` 为空才生成封面；③ 进度计数分组件——`title_total`=缺标题任务数、`cover_total`=（封面开关 ON 时）缺封面任务数，不再用 `len(eligible)` 一刀切；④「没有待生成任务」提示只在**按当前设置真的无缺口**（标题全有 且（封面关 或 封面全有））时出现；⑤ 幂等/点数：全组件齐备的任务不纳入、不重复调用扣点。保持 `set_generated` 写库、run_logs、诊断日志、停止取消未开始项语义不变；不改 cmhub HTTP 协议、DB schema、Excel、Shopee/CDP 流程。单测覆盖：只补封面（跳过标题、标题点不增、手动标题不被覆盖）、混合批（部分缺标题+部分缺封面）、无缺口提示无任务、封面关时标题态任务不纳入 | DONE | -| T-534 | ② 重置增强：多选/筛选范围 + 按组件（标题/封面/全部）重置 | T-533, T-404a, T-509 | 现状：②「重置生成结果」只能对单条、且整条全清（`db.reset_generated` 同时清 `new_title` 和 `new_cover_path` 退回 collected）。真实诉求是精准重置——一批标题不满意但封面 OK（只重标题、保封面、不重扣封面点）、或封面差但标题 OK（只重封面、**保留 `new_title` 保护 T-509 手动标题**、不重扣标题点）；且需要作用于多条。方案：与 T-533 的组件模型咬合——① 后端 `db.reset_generated` 加组件开关 `reset_title`/`reset_cover`（`delete_file` 仅在重置封面时可选），只清对应字段与 `last_error`，stage 保持 `generated`，由 T-533 泛化后的 `is_generatable_task`（缺任一必需组件即可生成）识别缺口后由「开始生成」只补被清组件；② UI 重置入口给三选项「重置标题 / 重置封面 / 重置全部」；③ 作用域从单条扩到**多选（表格改 `ExtendedSelection`）或当前筛选结果**，与③「作用于当前筛选结果」心智一致；④ 二次确认框显示将重置的标题/封面条数、其中 `committed=1` 条数，并提示「线上已提交、本地重置不回滚 Shopee、重生成后再更新会再次提交线上」+ 重生成会消耗 cmhub 点数；⑤ 运行中禁用；批量写 `run_type=reset` 运行日志，不触碰 Shopee、不回写 Excel。**只重置封面时绝不改 `new_title`**。不改 cmhub HTTP 协议、Excel、Shopee/CDP 流程。单测覆盖：只重标题保留封面、只重封面保留标题（含手动标题不被清）、重置全部、多选/筛选批量、committed 提示、运行中禁用 | TODO | +| T-534 | ② 重置增强：多选/筛选范围 + 按组件（标题/封面/全部）重置 | T-533, T-404a, T-509 | 现状：②「重置生成结果」只能对单条、且整条全清（`db.reset_generated` 同时清 `new_title` 和 `new_cover_path` 退回 collected）。真实诉求是精准重置——一批标题不满意但封面 OK（只重标题、保封面、不重扣封面点）、或封面差但标题 OK（只重封面、**保留 `new_title` 保护 T-509 手动标题**、不重扣标题点）；且需要作用于多条。方案：与 T-533 的组件模型咬合——① 后端 `db.reset_generated` 加组件开关 `reset_title`/`reset_cover`（`delete_file` 仅在重置封面时可选），只清对应字段与 `last_error`，stage 保持 `generated`，由 T-533 泛化后的 `is_generatable_task`（缺任一必需组件即可生成）识别缺口后由「开始生成」只补被清组件；② UI 重置入口给三选项「重置标题 / 重置封面 / 重置全部」；③ 作用域从单条扩到**多选（表格改 `ExtendedSelection`）或当前筛选结果**，与③「作用于当前筛选结果」心智一致；④ 二次确认框显示将重置的标题/封面条数、其中 `committed=1` 条数，并提示「线上已提交、本地重置不回滚 Shopee、重生成后再更新会再次提交线上」+ 重生成会消耗 cmhub 点数；⑤ 运行中禁用；批量写 `run_type=reset` 运行日志，不触碰 Shopee、不回写 Excel。**只重置封面时绝不改 `new_title`**。不改 cmhub HTTP 协议、Excel、Shopee/CDP 流程。单测覆盖：只重标题保留封面、只重封面保留标题（含手动标题不被清）、重置全部、多选/筛选批量、committed 提示、运行中禁用 | DONE | ## Phase 8 · 工程基础设施后续（`docs/engineering-review.md`） diff --git a/docs/current-state.md b/docs/current-state.md index 6973cea..0a25f74 100644 --- a/docs/current-state.md +++ b/docs/current-state.md @@ -6,10 +6,10 @@ ## 当前快照 - 日期：2026-07-06 -- 阶段：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-103b 登录检测补充识别 Shopee accounts 登录页、T-104 PySide6 主窗口骨架、T-104b PySide6 worker 基座、T-105 Tab④ 账号管理、T-105b ④启动登录复用已打开 Chrome、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-206 Tab① 删除指定批次软删除、T-301 AI 生成接口、T-302 Tab② AI 生成布局与任务列表、T-302p 提示词管理、T-303 Tab② 开始生成/停止/进度、T-303b ②/③ 商品ID筛选、T-401 Tab③ 更新列表筛选与开始更新确认、T-402 Tab③ 确认后串行更新、T-403 Tab③ 结果回写与结束汇总、T-501 Tab⑤ AI 模型管理 UI、T-501b Tab⑤ 角色与生成参数、T-501c Tab⑤ Shopee 更新安全开关、T-502 换封面删第一张再上传、T-503 敏感信息本地明文保存提示与日志脱敏、T-504 多账号并行/dry-run/运行日志、T-207 ①采集诊断日志、② AI生成图片失败诊断日志补丁、T-404a ②/③ 选中记录重置、T-404 真实 Shopee 单条更新冒烟验收、T-506 正式使用批量更新体验、T-507 正式批量更新移除普通流程测试商品 ID 限制、T-508 ③ 更新shopee生产化操作区、T-509 ② 新标题人工微调、T-510 ③ 检查本轮更新文案统一、T-505 全流程诊断日志扩展、T-511 语义色板与任务状态列上色、T-512 ③高风险按钮上色与①导入校验数字标红、T-513 登录点/③Tab危险标识/破坏性按钮上色、T-514 ①②③ 首次空状态引导卡片、T-515 批次阶段进度总览、T-516 ①筛选对齐②③、T-517 ⑤设置分区与兼容字段清理、T-518 ②左栏提示词区组件密度优化、T-519 ②AI生成长任务进度条与用户可读滚动日志、T-520 ②AI生成封面可选生成开关、T-521 依赖清单（锁版本 requirements）、T-522 CI 自动跑语法 + 单元 / GUI 测试、T-523 拆分 `app/gui.py` 为 `app/gui/` 包、T-523a ②/③ 新一轮运行前清空界面日志显示、T-404b 商品详情页加载失败 toast 自动捕获、T-524 PyInstaller 打包为免安装 exe、T-526 `app/ai.py` + `appconfig` 接入 cmhub backend、T-527 ⑤设置 cmhub 网关面板、T-528 ②计费错误提示与余额展示、T-529 默认 cmhub 网关并隐藏 AI 后端选择、T-530 cmhub Base URL 规整与 404 明确提示、T-531 ⑤设置未保存状态追踪与离开确认、T-532 ⑤cmhub连接成功提示显示账号名、T-533 ②增量生成：按缺失组件补生成。 +- 阶段：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-103b 登录检测补充识别 Shopee accounts 登录页、T-104 PySide6 主窗口骨架、T-104b PySide6 worker 基座、T-105 Tab④ 账号管理、T-105b ④启动登录复用已打开 Chrome、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-206 Tab① 删除指定批次软删除、T-301 AI 生成接口、T-302 Tab② AI 生成布局与任务列表、T-302p 提示词管理、T-303 Tab② 开始生成/停止/进度、T-303b ②/③ 商品ID筛选、T-401 Tab③ 更新列表筛选与开始更新确认、T-402 Tab③ 确认后串行更新、T-403 Tab③ 结果回写与结束汇总、T-501 Tab⑤ AI 模型管理 UI、T-501b Tab⑤ 角色与生成参数、T-501c Tab⑤ Shopee 更新安全开关、T-502 换封面删第一张再上传、T-503 敏感信息本地明文保存提示与日志脱敏、T-504 多账号并行/dry-run/运行日志、T-207 ①采集诊断日志、② AI生成图片失败诊断日志补丁、T-404a ②/③ 选中记录重置、T-404 真实 Shopee 单条更新冒烟验收、T-506 正式使用批量更新体验、T-507 正式批量更新移除普通流程测试商品 ID 限制、T-508 ③ 更新shopee生产化操作区、T-509 ② 新标题人工微调、T-510 ③ 检查本轮更新文案统一、T-505 全流程诊断日志扩展、T-511 语义色板与任务状态列上色、T-512 ③高风险按钮上色与①导入校验数字标红、T-513 登录点/③Tab危险标识/破坏性按钮上色、T-514 ①②③ 首次空状态引导卡片、T-515 批次阶段进度总览、T-516 ①筛选对齐②③、T-517 ⑤设置分区与兼容字段清理、T-518 ②左栏提示词区组件密度优化、T-519 ②AI生成长任务进度条与用户可读滚动日志、T-520 ②AI生成封面可选生成开关、T-521 依赖清单（锁版本 requirements）、T-522 CI 自动跑语法 + 单元 / GUI 测试、T-523 拆分 `app/gui.py` 为 `app/gui/` 包、T-523a ②/③ 新一轮运行前清空界面日志显示、T-404b 商品详情页加载失败 toast 自动捕获、T-524 PyInstaller 打包为免安装 exe、T-526 `app/ai.py` + `appconfig` 接入 cmhub backend、T-527 ⑤设置 cmhub 网关面板、T-528 ②计费错误提示与余额展示、T-529 默认 cmhub 网关并隐藏 AI 后端选择、T-530 cmhub Base URL 规整与 404 明确提示、T-531 ⑤设置未保存状态追踪与离开确认、T-532 ⑤cmhub连接成功提示显示账号名、T-533 ②增量生成：按缺失组件补生成、T-534 ②重置增强：多选/筛选范围 + 按组件重置。 - 技术栈：Python 3.10+，根目录 `requirements.txt` 锁定运行依赖，GitHub Actions 使用 Windows + Python 3.11 自动跑语法和单元/GUI 测试，PyInstaller onedir 打包为 Windows 免安装 exe；自研 CDP（websocket-client + requests），SQLite（sqlite3）+ `config.json` + openpyxl + AI（默认 cmhub: `config.json` 网关配置 + `config/cmhub.json` Key；direct: `config/ai_models.json` 通用 HTTP 内部兼容），GUI PySide6 5 Tab（已定）。 -- 生产代码：已建立 `app/` 包 + 根入口 `main.py`；`app/cdp.py` 为已验证 CDP 底座，已区分 `CDP.close()` 断开 WebSocket 与 `close_tab()` 关闭浏览器 target；`app/editor.py` 已封装登录状态检测、标题/封面/采集/更新按钮能力，打开商品详情页时会安装 toast 捕获并在详情页加载失败时上浮商品失效/无权限等错误原因，`open_product()` 内部失败会关闭本轮自动新建的失败 tab，复用用户已有 tab 不关闭，并在采集结束后只关闭本轮自动新建的商品编辑页 tab、保留用户已有 tab，更新提交成功且设置开启时可关闭本轮自动新建商品页，确认后跳回商品列表页时关闭前等待 2 秒；`click_update()` 已补 Shopee 站点侧确认框处理，页面主「更新」后若出现 `確定您要更新商品嗎?` modal（`.eds-modal__content` / `.eds-modal__box`），只点击弹窗主按钮「更新」并避开「立即優化」，且记录确认后是否跳回商品列表页；`replace_cover()` 已实现更新封面统一先校验本地旧封面备份，再点第一张删除、可见确认框、等待图片管理器稳定和上传 input 恢复，随后先点击上传块模拟人工入口、短暂等待、重新获取 input、注入文件上传新图并等待 Shopee CDN 后拖到第一位的代码路径；已修正 2026-06-30 稳定等待回归：删除第一张前不要求上传 input 可用，只等图片列表稳定；删除后再等上传入口恢复；`重複/重复/duplicate` 上传 toast 会立即判为新封面重复错误；`app/image_paths.py` 已统一新采集/新生成图片路径为 `images///__old/new.jpg`，历史 DB 路径继续按原路径读取；`app/appconfig.py` 已实现 `config.json` 默认值/读写/更新、AI 参数、端口读取、Shopee 更新安全与执行模式默认值，`ai.backend` 默认 `cmhub`，`config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接与 OpenAI-compatible base URL 自动补 endpoint，`config/cmhub.json` cmhub Key 读写/打码 helper、cmhub Base URL 规整与请求 URL 拼接 helper，以及 `mask_secret()`、`sanitize_for_log()`、`redact_secrets()` 敏感信息脱敏工具；`app/ai.py` 已实现 `gen_title()`/`gen_cover()`/`generate_batch()`，支持 direct 通用 HTTP 和 cmhub 网关 backend，direct 按默认文本/图片模型调用，cmhub 按 Base URL + 生文/生图别名调用 title/image/models/balance 接口，生图 `image_url` 安全下载后转本地 JPEG，计费 metadata 通过事件回调传出，HTTP 404 映射为 `not_found` 并给出中文排障提示，且保留重试、错误脱敏、图片 URL/base64 解析、resolution resize、jpg_quality 保存、按 `ai.generate_cover` 选择只生成标题或先并发标题再并发封面、逐条 `set_generated`、失败 `mark_failed`、步骤级事件/错误回调与停止取消未开始项；`app/prompts.py` 已实现标题提示词读写、封面模板 CRUD 与变量替换；`app/db.py` 已实现 SQLite schema、连接 PRAGMA、批次/账号/任务与阶段写库函数、T-206 批次软删除标记与默认业务查询过滤、T-404a 本地生成结果/更新状态重置函数、T-509 `update_generated_title()` 本地新标题微调函数，以及 `run_logs/run_log_events` 运行日志函数；`app/diagnostics.py` 已实现 gitignore 本地诊断日志、滚动写入、结构化 payload 和自由文本脱敏；T-505 已把 Excel 导入/回写、③更新shopee、④Chrome 启动/登录检测、⑤AI模型测试连接接入 `run_logs/run_log_events` 与本地脱敏诊断日志；`app/excel.py` 已实现多 Excel 输入列解析、整文件列校验、脏行统计跳过、导入批次与任务入库、别名匹配统计、旧标题/旧封面路径回写原 Excel、更新结果回写原 Excel 与另存副本；`app/config.py` 已实现账号 slug 与 user-data-dir 创建；`app/accounts.py` 已实现账号 CRUD 服务、端口默认分配、启动登录幂等复用已打开 Chrome、检测登录、生成快捷方式；`app/chrome.py` 已实现 Chrome 参数拼装、启动、CDP 端口探测、PowerShell `.lnk` 快捷方式生成；`app/gui/` 已由 T-523 拆分为 PySide6 GUI 包，包入口 `__init__.py` 兼容旧导入，`main_window.py` 放 `MainWindow`，`models.py` 放 3 个 TableModel，`widgets.py` 放色板/空状态/批次总览/helper，`workers.py` 放具体 GUI worker，`tabs/` 放 ①~⑤ Tab；整体仍实现 PySide6 `MainWindow`、五 Tab、顶部 Tab 栏防误点样式、统一语义色板、①②③任务状态列前景色、③「开始更新」warning 描边/文字色和①导入校验数字标红、④登录状态点上色、③更新shopee Tab warning 小圆点和删除类按钮 danger 样式、①②③首次空状态引导卡片、①②③批次阶段进度总览、① 导入采集的 Excel 导入按钮/导入汇总栏/批次筛选与删除批次软删除入口/QTableView 任务列表/未匹配筛选与略过标记/采集旧标题旧封面 worker/采集前账号就绪预检与④引导/采集完成自动回写/旧数据回写重试按钮与 worker/采集运行日志视图、② AI生成左右布局/标题与封面提示词管理/批次/店铺/商品ID/状态筛选/任务列表/新标题列本地微调/变量预览/生成封面图片成本开关/开始生成/停止/标题与图片双进度条/双击新旧封面预览/用户可读自动滚动AI生成运行日志/cmhub余额显示/点数不足弹窗/计费日志/重置生成结果与 `GenerateWorker`、③ 更新shopee批次/店铺/商品ID/状态筛选栏/任务列表/开始更新主按钮/重置更新状态右键菜单/更新安全开关拦截与「前往设置」跳转/「检查本轮更新」按钮/开始更新确认弹窗/确认后 `ApplyWorker` 按每批最大更新条数分批执行当前筛选全部可更新记录/账号就绪和端口冲突预检/按账号并行可选/逐条 `set_applied`/运行日志/自动回写结果到 Excel/结束汇总弹窗/手动回写重试按钮、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏、密码明文保存提示、⑤ 设置页 T-529 后默认展示 cmhub 网关 Base URL、API Key、动态生文/生图别名下拉、Base URL 网关根提示、刷新别名与测试连接/查余额，不再显示「AI 后端」label/dropdown、direct 模型选择、模型详情或标题/图片模型角色下拉；保存设置固定写 `ai.backend=cmhub`，保存/刷新前会把 Base URL 规整为网关根，允许先保存不完整 cmhub 配置，生成时再提示补齐；direct 模型配置、`AIModelTestWorker` 和 `config/ai_models.json` 仅作内部兼容/手工回滚；通用生成参数、路径/端口、Shopee 更新安全与多账号并行设置继续持久化 `config.json`，cmhub Key 单独写入 `config/cmhub.json`，保存成功后弹轻量提示框；⑤ 设置页已将「Shopee 更新安全 / 执行模式」前置、将「基础设施（路径与端口）」后置，`test_item_id` 与 `dry_run` 不再有用户可操作控件，保存时保留 `test_item_id` 兼容值并固定 `dry_run=false`；③ 普通正式更新不再用 `test_item_id` 阻断非测试商品，确认弹窗不再显示测试商品 ID；`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。 -- 测试：`tests/` 已建立；T-522 已新增 GitHub Actions 在 push / pull_request 自动运行语法检查与全量 unittest；T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`，当前覆盖 appconfig/db/config/accounts（含 T-105b 启动登录复用已运行 Chrome）/chrome 启动与快捷方式/editor 登录检测（含 Shopee accounts 登录页）与商品 tab 生命周期、商品详情页失效/隐藏 toast 捕获、更新成功后关闭本轮新开 tab、更新封面备份缺失阻断/8张与9张先删第一张/删除确认/上传入口初始不可用/重复图片 toast/物流错误不阻断上传/删图后稳定等待/上传状态诊断/拖首位 mock 路径/Shopee 更新确认框、成功跳回商品列表和残留错误 toast 不覆盖成功 mock 路径/excel 导入/旧字段与更新结果回写/ai 标题与封面 HTTP 解析/`generate_batch` 正常、失败与停止/prompts 读写与渲染/gui ① 导入采集、删除批次软删除与采集诊断日志/gui ② AI生成布局与批次/店铺/商品ID/状态筛选、提示词管理、生成 worker、图片失败诊断日志、双击预览、重置生成结果与新标题本地编辑/gui ③ 更新shopee批次/店铺/商品ID/状态筛选列表、确认弹窗、开始更新主按钮、重置更新状态右键菜单、Shopee 更新安全拦截与前往设置、`ApplyWorker` 串行/检查/分批/按账号并行/端口冲突预检、运行日志、结果回写与汇总/gui ④ 账号管理、密码打码与明文保存提示/gui ⑤ AI 模型管理、API Key 打码与明文保存提示、角色/生成参数设置、Shopee 更新安全设置、检查按钮/每批最大更新条数/多账号并行设置、测试商品 ID 限制移除/worker signal 与线程包装、T-505 全流程诊断日志（import/write_back/apply/chrome_launch/login_check/ai_model_test）与本地日志脱敏、gui T-511 语义色板和任务状态列前景色、T-512 高风险按钮与导入校验数字样式、T-513 登录状态/Tab 标识/删除按钮样式、T-514 首次空状态引导卡片、T-515 批次阶段进度总览、T-516 ①筛选对齐②③、T-517 ⑤设置分区与兼容字段清理、T-518 ②左栏提示词区组件密度优化、T-519 ②AI生成长任务进度条与用户可读滚动日志、T-520 ②AI生成封面可选生成开关、T-524 PyInstaller 打包入口/构建脚本/发布目录校验、T-526 cmhub backend mock、T-527 设置页 cmhub backend 切换/别名下拉/worker mock、T-528 ②cmhub余额显示/计费日志/点数不足中止提示、T-529 默认 cmhub/隐藏 AI 后端选择/保存固定 cmhub/direct 兼容、T-530 cmhub Base URL 规整/404 not_found 明确提示、T-531 设置页未保存状态/离开确认/放弃还原、T-532 cmhub连接成功账号名提示；2026-07-01 已完成 5 个真实商品的 T-404 更新验收，后续 CDP/Shopee 改动仍需测试商品手动验证。 +- 生产代码：已建立 `app/` 包 + 根入口 `main.py`；`app/cdp.py` 为已验证 CDP 底座，已区分 `CDP.close()` 断开 WebSocket 与 `close_tab()` 关闭浏览器 target；`app/editor.py` 已封装登录状态检测、标题/封面/采集/更新按钮能力，打开商品详情页时会安装 toast 捕获并在详情页加载失败时上浮商品失效/无权限等错误原因，`open_product()` 内部失败会关闭本轮自动新建的失败 tab，复用用户已有 tab 不关闭，并在采集结束后只关闭本轮自动新建的商品编辑页 tab、保留用户已有 tab，更新提交成功且设置开启时可关闭本轮自动新建商品页，确认后跳回商品列表页时关闭前等待 2 秒；`click_update()` 已补 Shopee 站点侧确认框处理，页面主「更新」后若出现 `確定您要更新商品嗎?` modal（`.eds-modal__content` / `.eds-modal__box`），只点击弹窗主按钮「更新」并避开「立即優化」，且记录确认后是否跳回商品列表页；`replace_cover()` 已实现更新封面统一先校验本地旧封面备份，再点第一张删除、可见确认框、等待图片管理器稳定和上传 input 恢复，随后先点击上传块模拟人工入口、短暂等待、重新获取 input、注入文件上传新图并等待 Shopee CDN 后拖到第一位的代码路径；已修正 2026-06-30 稳定等待回归：删除第一张前不要求上传 input 可用，只等图片列表稳定；删除后再等上传入口恢复；`重複/重复/duplicate` 上传 toast 会立即判为新封面重复错误；`app/image_paths.py` 已统一新采集/新生成图片路径为 `images///__old/new.jpg`，历史 DB 路径继续按原路径读取；`app/appconfig.py` 已实现 `config.json` 默认值/读写/更新、AI 参数、端口读取、Shopee 更新安全与执行模式默认值，`ai.backend` 默认 `cmhub`，`config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接与 OpenAI-compatible base URL 自动补 endpoint，`config/cmhub.json` cmhub Key 读写/打码 helper、cmhub Base URL 规整与请求 URL 拼接 helper，以及 `mask_secret()`、`sanitize_for_log()`、`redact_secrets()` 敏感信息脱敏工具；`app/ai.py` 已实现 `gen_title()`/`gen_cover()`/`generate_batch()`，支持 direct 通用 HTTP 和 cmhub 网关 backend，direct 按默认文本/图片模型调用，cmhub 按 Base URL + 生文/生图别名调用 title/image/models/balance 接口，生图 `image_url` 安全下载后转本地 JPEG，计费 metadata 通过事件回调传出，HTTP 404 映射为 `not_found` 并给出中文排障提示，且保留重试、错误脱敏、图片 URL/base64 解析、resolution resize、jpg_quality 保存、按 `ai.generate_cover` 选择只生成标题或先并发标题再并发封面、逐条 `set_generated`、失败 `mark_failed`、步骤级事件/错误回调与停止取消未开始项；`app/prompts.py` 已实现标题提示词读写、封面模板 CRUD 与变量替换；`app/db.py` 已实现 SQLite schema、连接 PRAGMA、批次/账号/任务与阶段写库函数、T-206 批次软删除标记与默认业务查询过滤、T-404a/T-534 本地生成结果/更新状态重置函数（②生成结果可按标题/封面组件重置并保持 generated）、T-509 `update_generated_title()` 本地新标题微调函数，以及 `run_logs/run_log_events` 运行日志函数；`app/diagnostics.py` 已实现 gitignore 本地诊断日志、滚动写入、结构化 payload 和自由文本脱敏；T-505 已把 Excel 导入/回写、③更新shopee、④Chrome 启动/登录检测、⑤AI模型测试连接接入 `run_logs/run_log_events` 与本地脱敏诊断日志；`app/excel.py` 已实现多 Excel 输入列解析、整文件列校验、脏行统计跳过、导入批次与任务入库、别名匹配统计、旧标题/旧封面路径回写原 Excel、更新结果回写原 Excel 与另存副本；`app/config.py` 已实现账号 slug 与 user-data-dir 创建；`app/accounts.py` 已实现账号 CRUD 服务、端口默认分配、启动登录幂等复用已打开 Chrome、检测登录、生成快捷方式；`app/chrome.py` 已实现 Chrome 参数拼装、启动、CDP 端口探测、PowerShell `.lnk` 快捷方式生成；`app/gui/` 已由 T-523 拆分为 PySide6 GUI 包，包入口 `__init__.py` 兼容旧导入，`main_window.py` 放 `MainWindow`，`models.py` 放 3 个 TableModel，`widgets.py` 放色板/空状态/批次总览/helper，`workers.py` 放具体 GUI worker，`tabs/` 放 ①~⑤ Tab；整体仍实现 PySide6 `MainWindow`、五 Tab、顶部 Tab 栏防误点样式、统一语义色板、①②③任务状态列前景色、③「开始更新」warning 描边/文字色和①导入校验数字标红、④登录状态点上色、③更新shopee Tab warning 小圆点和删除类按钮 danger 样式、①②③首次空状态引导卡片、①②③批次阶段进度总览、① 导入采集的 Excel 导入按钮/导入汇总栏/批次筛选与删除批次软删除入口/QTableView 任务列表/未匹配筛选与略过标记/采集旧标题旧封面 worker/采集前账号就绪预检与④引导/采集完成自动回写/旧数据回写重试按钮与 worker/采集运行日志视图、② AI生成左右布局/标题与封面提示词管理/批次/店铺/商品ID/状态筛选/任务列表/新标题列本地微调/变量预览/生成封面图片成本开关/开始生成/停止/标题与图片双进度条/双击新旧封面预览/用户可读自动滚动AI生成运行日志/cmhub余额显示/点数不足弹窗/计费日志/重置生成结果与 `GenerateWorker`，且②重置支持多选/当前筛选结果并可只重置标题、只重置封面或重置全部；③ 更新shopee批次/店铺/商品ID/状态筛选栏/任务列表/开始更新主按钮/重置更新状态右键菜单/更新安全开关拦截与「前往设置」跳转/「检查本轮更新」按钮/开始更新确认弹窗/确认后 `ApplyWorker` 按每批最大更新条数分批执行当前筛选全部可更新记录/账号就绪和端口冲突预检/按账号并行可选/逐条 `set_applied`/运行日志/自动回写结果到 Excel/结束汇总弹窗/手动回写重试按钮、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏、密码明文保存提示、⑤ 设置页 T-529 后默认展示 cmhub 网关 Base URL、API Key、动态生文/生图别名下拉、Base URL 网关根提示、刷新别名与测试连接/查余额，不再显示「AI 后端」label/dropdown、direct 模型选择、模型详情或标题/图片模型角色下拉；保存设置固定写 `ai.backend=cmhub`，保存/刷新前会把 Base URL 规整为网关根，允许先保存不完整 cmhub 配置，生成时再提示补齐；direct 模型配置、`AIModelTestWorker` 和 `config/ai_models.json` 仅作内部兼容/手工回滚；通用生成参数、路径/端口、Shopee 更新安全与多账号并行设置继续持久化 `config.json`，cmhub Key 单独写入 `config/cmhub.json`，保存成功后弹轻量提示框；⑤ 设置页已将「Shopee 更新安全 / 执行模式」前置、将「基础设施（路径与端口）」后置，`test_item_id` 与 `dry_run` 不再有用户可操作控件，保存时保留 `test_item_id` 兼容值并固定 `dry_run=false`；③ 普通正式更新不再用 `test_item_id` 阻断非测试商品，确认弹窗不再显示测试商品 ID；`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。 +- 测试：`tests/` 已建立；T-522 已新增 GitHub Actions 在 push / pull_request 自动运行语法检查与全量 unittest；T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`，当前覆盖 appconfig/db/config/accounts（含 T-105b 启动登录复用已运行 Chrome）/chrome 启动与快捷方式/editor 登录检测（含 Shopee accounts 登录页）与商品 tab 生命周期、商品详情页失效/隐藏 toast 捕获、更新成功后关闭本轮新开 tab、更新封面备份缺失阻断/8张与9张先删第一张/删除确认/上传入口初始不可用/重复图片 toast/物流错误不阻断上传/删图后稳定等待/上传状态诊断/拖首位 mock 路径/Shopee 更新确认框、成功跳回商品列表和残留错误 toast 不覆盖成功 mock 路径/excel 导入/旧字段与更新结果回写/ai 标题与封面 HTTP 解析/`generate_batch` 正常、失败与停止/prompts 读写与渲染/gui ① 导入采集、删除批次软删除与采集诊断日志/gui ② AI生成布局与批次/店铺/商品ID/状态筛选、提示词管理、生成 worker、图片失败诊断日志、双击预览、重置生成结果与新标题本地编辑/gui ③ 更新shopee批次/店铺/商品ID/状态筛选列表、确认弹窗、开始更新主按钮、重置更新状态右键菜单、Shopee 更新安全拦截与前往设置、`ApplyWorker` 串行/检查/分批/按账号并行/端口冲突预检、运行日志、结果回写与汇总/gui ④ 账号管理、密码打码与明文保存提示/gui ⑤ AI 模型管理、API Key 打码与明文保存提示、角色/生成参数设置、Shopee 更新安全设置、检查按钮/每批最大更新条数/多账号并行设置、测试商品 ID 限制移除/worker signal 与线程包装、T-505 全流程诊断日志（import/write_back/apply/chrome_launch/login_check/ai_model_test）与本地日志脱敏、gui T-511 语义色板和任务状态列前景色、T-512 高风险按钮与导入校验数字样式、T-513 登录状态/Tab 标识/删除按钮样式、T-514 首次空状态引导卡片、T-515 批次阶段进度总览、T-516 ①筛选对齐②③、T-517 ⑤设置分区与兼容字段清理、T-518 ②左栏提示词区组件密度优化、T-519 ②AI生成长任务进度条与用户可读滚动日志、T-520 ②AI生成封面可选生成开关、T-524 PyInstaller 打包入口/构建脚本/发布目录校验、T-526 cmhub backend mock、T-527 设置页 cmhub backend 切换/别名下拉/worker mock、T-528 ②cmhub余额显示/计费日志/点数不足中止提示、T-529 默认 cmhub/隐藏 AI 后端选择/保存固定 cmhub/direct 兼容、T-530 cmhub Base URL 规整/404 not_found 明确提示、T-531 设置页未保存状态/离开确认/放弃还原、T-532 cmhub连接成功账号名提示、T-533 增量生成组件口径、T-534 ②按标题/封面组件重置与多选/筛选范围重置；2026-07-01 已完成 5 个真实商品的 T-404 更新验收，后续 CDP/Shopee 改动仍需测试商品手动验证。 - 数据：`config.json`、`config/ai_models.json`、`config/cmhub.json`、`cmshopee.db`、`cmshopee.db-*`、`chrome_user_data_dir/`、`images/`、`logs/`、`prompts/`、`title_prompt.txt`、`build/`、`dist/` 已由 `.gitignore` 排除；密码与 API Key 本地明文保存但保存/变更时提示，UI 打码，日志/导出必须脱敏；运营填写后的 Excel 业务文件默认忽略，标准空模板 `shopee待处理任务模板.xlsx` 可提交；`app/appconfig.py` 首次读取缺失的 `config.json` 时会在本地写默认配置，`app/db.py` 调用 `init_db()` 时会在本地创建 SQLite DB。 ## 既定设计要点（文档已定） @@ -45,9 +45,9 @@ | `app/editor.py` | 已有 | T-001/T-103/T-205b/T-207/T-501c/T-502/T-404b + T-404 补丁产出：登录状态检测、打开商品页、详情页加载失败 toast 捕获与商品失效归因、读/写标题、读/下载封面、采集步骤回调、更新步骤回调、采集后关闭自动新建商品页 tab、上传前等待图片管理器稳定、点击上传块后注入文件并检测 CDN 后拖封面、更新封面统一先校验旧封面备份再删线上第一张、重复图片 toast 明确失败、页面主更新按钮、Shopee 站点侧确认框主按钮、apply_task；更新成功后可按设置关闭本轮自动新建商品页 | | `app/appconfig.py` | 已有 | T-002/T-501c/T-503/T-504/T-520/T-526/T-527 产出：`config.json` 默认值、读写、更新、路径/端口/AI 参数读取、AI `backend=direct/cmhub` 和 `ai.cmhub` 默认值、`config/cmhub.json` Key helper、Shopee 更新安全与 dry-run/多账号并行默认值，②生成封面默认关闭的 `ai.generate_cover`；拒绝敏感字段写入；提供敏感值打码、结构化日志脱敏与自由文本替换工具 | | `app/diagnostics.py` | 已有 | T-207 + AI生成诊断补丁 + T-505 产出：本地 `logs/cmshopee.log` 诊断日志、大小滚动、异常类型/traceback/step/耗时记录，结构化 payload 和自由文本脱敏后写入 | -| `app/ai.py` | 已有 | T-301/T-303 + AI生成诊断补丁 + T-519/T-520/T-526/T-528 产出：`gen_title()`/`gen_cover()`/`generate_batch()`；支持 direct 默认模型直连和 cmhub 网关 backend；cmhub 使用 `CMHubError` 结构化错误、tuple timeout、非幂等生图读超时不重发、`image_url` 安全下载、`fetch_cmhub_models()` 别名发现、`fetch_cmhub_balance()` 余额查询和 metadata 事件；direct 保留通用 HTTP 调用、失败重试、错误脱敏；封面按 resolution/jpg_quality 保存并按批次/店铺/任务路径落盘；批量生成可按 `ai.generate_cover` 跳过封面阶段，标题成功即写库进入 generated；开启封面时先标题后封面、进度回调、逐条落库、失败标记、事件/错误回调、停止取消未开始项 | +| `app/ai.py` | 已有 | T-301/T-303 + AI生成诊断补丁 + T-519/T-520/T-526/T-528/T-533 产出：`gen_title()`/`gen_cover()`/`generate_batch()`；支持 direct 默认模型直连和 cmhub 网关 backend；cmhub 使用 `CMHubError` 结构化错误、tuple timeout、非幂等生图读超时不重发、`image_url` 安全下载、`fetch_cmhub_models()` 别名发现、`fetch_cmhub_balance()` 余额查询和 metadata 事件；direct 保留通用 HTTP 调用、失败重试、错误脱敏；封面按 resolution/jpg_quality 保存并按批次/店铺/任务路径落盘；批量生成可按 `ai.generate_cover` 跳过封面阶段，开启封面时按缺失组件增量补齐，已有标题不重生、不覆盖手动标题，已有封面不重生；进度回调按标题/封面组件统计，逐条落库、失败标记、事件/错误回调、停止取消未开始项 | | `app/prompts.py` | 已有 | T-302p 产出：标题提示词读写、封面模板列表/读取/保存/重命名/删除、变量替换 | -| `app/db.py` | 已有 | T-003/T-206/T-404a/T-504/T-509 产出：batches/accounts/tasks schema；batches 软删除字段与默认业务查询过滤；run_logs/run_log_events；WAL/busy_timeout/foreign_keys；账号/批次/任务与 set_* 阶段写库；生成结果/更新状态本地重置；`update_generated_title()` 本地新标题微调；运行日志写入与查询 | +| `app/db.py` | 已有 | T-003/T-206/T-404a/T-504/T-509/T-534 产出：batches/accounts/tasks schema；batches 软删除字段与默认业务查询过滤；run_logs/run_log_events；WAL/busy_timeout/foreign_keys；账号/批次/任务与 set_* 阶段写库；生成结果/更新状态本地重置，②生成结果支持按标题/封面组件清空并保持 generated；`update_generated_title()` 本地新标题微调；运行日志写入与查询 | | `app/config.py` | 已有 | T-101 产出：别名→稳定 slug；创建并返回绝对 user-data-dir | | `app/image_paths.py` | 已有 | 本地图片路径 helper：新采集旧封面和新生成封面统一写入 `images///__old/new.jpg`；历史 DB 已存路径继续按原路径读取 | | `app/chrome.py` | 已有 | T-102/T-106 产出：Chrome 启动参数、`subprocess.Popen` 启动、`/json/version` 端口探测、PowerShell `.lnk` 快捷方式 | @@ -70,8 +70,8 @@ 任务状态以 [`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-103b（登录检测补充识别 Shopee accounts 登录页）、T-104（PySide6 五 Tab 主窗口骨架）、T-104b（PySide6 worker 基座）、T-105（Tab④ 账号管理）、T-105b（④启动登录复用已打开 Chrome）、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-206（Tab① 删除指定批次软删除）、T-301（AI 生成接口）、T-302（Tab② 左右布局与任务列表）、T-302p（提示词管理）、T-303（Tab② 开始生成 + 停止 + 进度）、T-303b（②/③ 商品ID筛选）、T-401（Tab③ 更新列表筛选 + 开始更新确认弹窗）、T-402（Tab③ 确认后串行更新）、T-403（Tab③ 结果回写与结束汇总）、T-501（Tab⑤ AI 模型管理 UI）、T-501b（Tab⑤ 角色与生成参数）、T-501c（Tab⑤ Shopee 更新安全开关）、T-502（换封面删第一张再上传）、T-503（敏感信息本地明文保存提示与日志脱敏）、T-504（多账号并行 / dry-run / 运行日志）、T-207（①采集诊断日志）、② AI生成图片失败诊断日志补丁、T-404a（②/③ 选中记录重置）、T-404（真实 Shopee 单条更新冒烟验收）、T-506（正式使用批量更新体验）、T-507（正式批量更新：移除普通流程测试商品 ID 限制）、T-508（③ 更新shopee生产化操作区）、T-509（② 新标题人工微调）、T-510（③ 检查本轮更新文案统一）、T-505（全流程诊断日志扩展）、T-511（语义色板 + ①②③任务状态列上色）、T-512（③高风险按钮上色 + ①导入校验数字标红）、T-513（登录点 / ③Tab危险标识 / 破坏性按钮上色）、T-514（①②③ 首次空状态引导卡片）、T-515（批次阶段进度总览）、T-516（①筛选对齐②③）、T-517（⑤设置分区 + 清理兼容字段）、T-518（②左栏提示词区组件密度优化）、T-519（②AI生成长任务进度条 + 用户可读滚动日志）、T-520（②AI生成封面可选生成开关）、T-521（依赖清单：锁版本 requirements）、T-522（CI：自动跑语法 + 单元 / GUI 测试）、T-523（拆分 `app/gui.py` 为 `app/gui/` 包）、T-523a（②/③ 新一轮运行前清空界面日志显示）、T-404b（商品详情页加载失败 toast 自动捕获）、T-524（PyInstaller 打包为免安装 exe）、T-526（`app/ai.py` + `appconfig` 接入 cmhub backend）、T-527（⑤设置 cmhub 网关面板）、T-528（②计费错误提示 + 余额展示）、T-529（默认 cmhub 网关并隐藏 AI 后端选择）、T-530（cmhub Base URL 规整 + 404 明确提示）、T-531（⑤设置未保存状态追踪 + 离开确认）、T-532（⑤cmhub连接成功提示显示账号名）。 -- 最近完成补充：T-533（②增量生成：按缺失组件补生成）。 -- 下一个可领取任务：T-534（② 重置增强：多选/筛选范围 + 按组件重置）；T-525 ruff 继续顺延。 +- 最近完成补充：T-533（②增量生成：按缺失组件补生成）、T-534（②重置增强：多选/筛选范围 + 按组件重置）。 +- 下一个可领取任务：T-525（引入 ruff lint/format，先以安全规则为主）。 ## 当前已知限制 @@ -106,12 +106,12 @@ - T-523a 已完成：②点击「开始生成」、③点击「检查本轮更新」或「开始更新」时，先清空对应界面的旧日志显示并写入本轮开始摘要；运行中只追加本轮日志。历史 `run_logs/run_log_events` 和本地 `logs/` 不删除、不自动混入当前运行界面。 - T-404b 已完成：`open_product()` 在导航商品详情页前后安装 toast 捕获，等待详情页关键元素超时时会读取最近 `.eds-toasts`/toast/message 节点及页面缓存，把明确商品失效/商品不存在/无权限类 toast 上浮为 `商品失效：<原始toast>`；如果失败发生在 `open_product()` 内部，本轮自动新建的失败 tab 会关闭，复用用户已有 tab 不关闭；①列表“阶段”列只在这类明确失效错误时显示“商品失效”，底层仍保持 `stage=imported/status=failed`，其他打开失败仍显示“失败”。 - T-524 已完成：新增 PyInstaller onedir 打包配置、构建脚本和发布目录校验；打包版启动时工作目录切到 `cmshopee.exe` 所在目录，首次运行在 exe 同级生成/使用本地配置、DB、图片、日志、登录态和提示词；发布包不内置这些本地数据，用户后续更新采用关闭程序后覆盖程序文件、保留本地数据的方式。 -- Phase 7 cmhub 网关对接已完成 T-526~T-533；②生成阶段已支持按标题/封面缺失组件增量生成，避免补封面时重跑标题。下一步按看板进入 T-534；T-525 ruff 继续顺延。 +- Phase 7 cmhub 网关对接已完成 T-526~T-534；②生成阶段已支持按标题/封面缺失组件增量生成，②重置生成结果已支持多选/筛选范围和按标题/封面组件重置。下一步按看板回到 T-525 ruff。 - T-301/T-303 已完成通用 HTTP AI 接口、批量生成编排和 GUI 接入 mock 单测；真实 AI 生成还需要在 `config/ai_models.json` 填入可用 url/model/api_key 后做一次成本可控的小样本实测；URL 可填完整 endpoint 或 OpenAI-compatible base URL。 - 本地 `config/ai_models.json` 若由旧版本或手工维护，可能缺少 `category`；启动报 “AI 模型 category 必须是 text 或 image” 时，按 [`troubleshooting.md`](troubleshooting.md) 只补 `category` / `enabled` 等非密钥字段，保留 API Key，且不要提交该文件。 - T-403/T-501c 已完成③更新结果回写、结束汇总与真实更新安全开关；2026-06-29 已用测试商品跑到真实编辑页并完成标题/封面替换、页面主「更新」点击，但被 Shopee 站点侧确认框拦住。当前代码已补确认框处理、确认后列表页跳转观测、上传状态诊断、删除后稳定等待和 mock 单测；2026-06-30 用户复跑发现稳定等待顺序回归：满 9 张时删除前要求上传 input 可用，导致不删除第一张图；当前已修正为删除前只等图片列表稳定、删除后再等上传入口恢复，并补 mock 回归测试；2026-06-30 又发现物流/备货页面级校验 toast 会被误判成封面上传失败，当前已修复为上传阶段只识别图片/文件/上传相关错误，物流错误留到提交阶段处理。2026-06-30 用户复跑确认浏览器已成功更新标题/图片并跳转商品列表，但 GUI 仍报 `POST_UPDATE_ERROR`；已修正提交后观测优先级：列表页跳转优先判成功，残留 error toast 不再覆盖成功跳转，错误 toast 仅在未跳转且持续存在时判失败。当前又发现商品 26887160467 手动上传 220KB 新图成功，但代码直接注入文件后上传组件长期转圈；T-404 当前补丁已让 `replace_cover()` 上传前先点击上传块模拟人工入口、短暂等待、重新获取 input 后再 `DOM.setFileInputFiles`，并补 mock 回归测试。25120403046 只有 8 张商品图时未删除第一张导致 Shopee 提示 `有1張重複的圖片`；当前已落地为更新封面统一先删当前第一张再上传，并把 `重複/重复/duplicate` toast 识别为封面上传失败。新采集/新生成图片路径已改为按批次细分，避免同一账号多批次图片混放；历史 DB 已存路径继续可用。确认后若返回我的商品列表页且设置了成功后关闭本次新开编辑页，关闭前等待 2 秒。2026-07-01 用户已用包含 5 个商品 ID 的 Excel 完成导入、AI 生成标题/图片、Tab③ 更新到 Shopee 的真实链路验收，T-404 判定完成。 -- T-404a 已完成：②「重置生成结果」清空本地 AI 结果并退回 collected，默认不删除本地新封面文件；③「重置更新状态」保留新标题/新封面并退回 generated/pending，当前入口已按 T-508 移到任务表右键菜单；两者只作用当前选中单条、运行中禁用、写 `run_type=reset` 运行日志。③ 对 `committed=1` 会提示线上已提交过、本地重置不回滚 Shopee、重复更新会再次提交。 +- T-404a/T-534 已完成：②「重置生成结果」已从单条整清升级为多选/当前筛选结果批量重置，确认框提供「重置标题 / 重置封面 / 重置全部」三种组件操作；只重置封面时保留 `new_title` 和 T-509 手动标题，只重置标题时保留 `new_cover_path`，重置后任务保持 `stage=generated/status=success` 供 T-533 按缺口补生成。默认不删除本地新封面文件，不触碰 Shopee、不自动回写 Excel；已提交线上记录会提示本地重置不回滚 Shopee、重生成后再更新会再次提交线上；运行中禁用并写 `run_type=reset` 运行日志。③「重置更新状态」仍保留新标题/新封面并退回 generated/pending，入口按 T-508 保持在任务表右键菜单。 - T-502 换封面删除流程已完成代码路径、mock 单测和真实 9 图商品不提交流程实测；更新封面统一先删当前第一张，删除前必须已有本地旧封面备份（`old_cover_path` 存在且文件存在），缺失备份时拒绝删除线上图片；mock 已覆盖 8 张图也先删第一张、重复图片 toast 立即失败。本轮实测只操作编辑页并关闭测试 tab，未点击「更新」保存线上。`images/` 新路径按 `batch_id/slug/task_id_item_id` 细分，旧 DB 路径兼容不迁移。 ## 当前可运行内容 diff --git a/progress.md b/progress.md index d9523a2..4c805f9 100644 --- a/progress.md +++ b/progress.md @@ -1235,3 +1235,13 @@ - 边界：未改 cmhub HTTP 协议、配置 schema、DB schema、Excel、Shopee/CDP 流程；全组件齐备任务不纳入本轮生成。 - 测试：新增/调整 AI 与 GUI 单测覆盖只补封面、混合标题/封面缺口、封面关闭时标题态任务不纳入、全组件齐备不重复生成、②入口可启动补封面。 - 验证：`python -m py_compile app\ai.py app\gui\workers.py app\gui\tabs\generate.py tests\test_ai.py tests\test_gui.py` 通过；`python -m unittest discover -s tests -p test_ai.py` 通过（23 tests）；相关 GUI 用例通过；`python -m compileall app main.py` 通过；`python -m unittest discover -s tests` 通过（207 tests）；`git diff --check` 通过，仅有本机 LF/CRLF 提示。 + +## 【2026-07-06】T-534 完成 · ②按组件批量重置生成结果 + +- 状态：DONE +- 代码：`db.reset_generated()` 增加 `reset_title` / `reset_cover` 组件开关和仅重置封面时可选的 `delete_file`，重置后保持 `stage=generated/status=success`，让 T-533 能按缺口只补被清掉的标题或封面；已提交线上记录保留 `committed/apply_attempts` 历史。 +- AI：`generation_needs()` 只阻断 `status=failed` 且有 `apply_attempts` 的更新失败记录；用户主动重置后的成功态记录即使保留提交历史，也允许按缺口补生成。 +- GUI：②任务表改为多选；「重置生成结果」确认框提供「重置标题 / 重置封面 / 重置全部 / 取消」，有选中行时作用于选中任务，没有选中行时作用于当前筛选结果；确认框显示标题/封面数量、运行中跳过数、已提交线上数量和重复提交风险；批量写 `run_type=reset` 日志。 +- 边界：不改 cmhub HTTP 协议、配置 schema、Excel、Shopee/CDP 流程；默认不删除本地新封面文件，不触碰 Shopee，不自动回写 Excel。 +- 测试：新增/调整 DB、AI、GUI 单测覆盖只重标题保留封面、只重封面保留手动标题、重置全部、多选/筛选范围、committed 警告、运行中禁用、提交历史下补封面。 +- 验证：`python -m py_compile app\db.py app\ai.py app\gui\tabs\generate.py tests\test_db.py tests\test_ai.py tests\test_gui.py` 通过；`python -m unittest discover -s tests -p test_db.py` 通过（7 tests）；`python -m unittest discover -s tests -p test_ai.py` 通过（24 tests）；相关 GUI 用例通过；`python -m compileall app main.py` 通过；`python -m unittest discover -s tests` 通过（212 tests）。 diff --git a/tests/test_ai.py b/tests/test_ai.py index ba1a218..e6e8310 100644 --- a/tests/test_ai.py +++ b/tests/test_ai.py @@ -802,6 +802,51 @@ class AITests(TempDirMixin, unittest.TestCase): self.assert_removed(temp_dir) + def test_generate_batch_fills_reset_cover_after_committed_history(self): + with self.make_temp_dir() as temp_dir: + cfg = self._config() + cfg["db_path"] = os.path.join(temp_dir, "cmshopee.db") + cfg["image_dir"] = os.path.join(temp_dir, "images") + cfg["ai"]["generate_cover"] = True + batch_id, tasks = self._collected_tasks(temp_dir, cfg, ["旧标题"]) + db.set_generated(tasks[0].id, "手动标题", "old-new.jpg", path=cfg["db_path"]) + db.set_applied(tasks[0].id, True, path=cfg["db_path"]) + db.reset_generated( + tasks[0].id, + reset_title=False, + reset_cover=True, + path=cfg["db_path"], + ) + cover_only_task = db.list_tasks(batch_id=batch_id, path=cfg["db_path"])[0] + + def fake_cover(cover_prompt, old_cover_path, out_path, **kwargs): + os.makedirs(os.path.dirname(out_path), exist_ok=True) + with open(out_path, "wb") as fh: + fh.write(b"jpeg") + return out_path + + with mock.patch("app.ai.gen_title") as gen_title, \ + mock.patch("app.ai.gen_cover", side_effect=fake_cover) as gen_cover: + summary = ai.generate_batch( + [cover_only_task], + {"title": "标题提示", "cover": "封面 {新标题}"}, + ai_cfg={"config": cfg, "db_path": cfg["db_path"]}, + ) + + self.assertTrue(summary["ok"]) + self.assertEqual(1, summary["total"]) + self.assertEqual(0, summary["title_total"]) + self.assertEqual(1, summary["cover_total"]) + gen_title.assert_not_called() + gen_cover.assert_called_once() + updated = db.list_tasks(batch_id=batch_id, path=cfg["db_path"])[0] + self.assertEqual("手动标题", updated.new_title) + self.assertTrue(os.path.exists(updated.new_cover_path)) + self.assertEqual(1, updated.committed) + self.assertEqual(1, updated.apply_attempts) + + self.assert_removed(temp_dir) + def test_generate_batch_handles_mixed_title_and_cover_gaps(self): with self.make_temp_dir() as temp_dir: cfg = self._config() diff --git a/tests/test_db.py b/tests/test_db.py index 54b8493..04d258d 100644 --- a/tests/test_db.py +++ b/tests/test_db.py @@ -168,7 +168,7 @@ class DbTests(TempDirMixin, unittest.TestCase): self.assertEqual(new_cover, reset_generated["new_cover_path"]) self.assertIsNone(reset_generated["deleted_file"]) self.assertTrue(os.path.exists(new_cover)) - self.assertEqual("collected", after_generated.stage) + self.assertEqual("generated", after_generated.stage) self.assertEqual("success", after_generated.status) self.assertIsNone(after_generated.new_title) self.assertIsNone(after_generated.new_cover_path) @@ -177,6 +177,81 @@ class DbTests(TempDirMixin, unittest.TestCase): self.assert_removed(temp_dir) + def test_reset_generated_can_reset_title_or_cover_components(self): + with self.make_temp_dir() as temp_dir: + db_path = os.path.join(temp_dir, "cmshopee.db") + db.init_db(db_path) + batch_id = db.create_batch(["input.xlsx"], path=db_path) + db.insert_tasks( + batch_id, + [ + { + "source_file_abs": os.path.join(temp_dir, "input.xlsx"), + "source_sheet": "Sheet1", + "source_row": 2, + "account_name": "shop", + "alias": "alias", + "item_id": "51100639510", + }, + { + "source_file_abs": os.path.join(temp_dir, "input.xlsx"), + "source_sheet": "Sheet1", + "source_row": 3, + "account_name": "shop", + "alias": "alias", + "item_id": "51100639511", + }, + ], + path=db_path, + ) + first, second = db.list_tasks(batch_id=batch_id, path=db_path) + first_cover = os.path.join(temp_dir, "first_new.jpg") + second_cover = os.path.join(temp_dir, "second_new.jpg") + for cover in [first_cover, second_cover]: + with open(cover, "wb") as fh: + fh.write(b"jpeg") + db.set_collected(first.id, "旧标题A", "old-a.jpg", path=db_path) + db.set_generated(first.id, "手动标题A", first_cover, path=db_path) + db.mark_failed(first.id, "generate", "图片不满意", path=db_path) + db.set_collected(second.id, "旧标题B", "old-b.jpg", path=db_path) + db.set_generated(second.id, "手动标题B", second_cover, path=db_path) + + title_reset = db.reset_generated( + first.id, + reset_title=True, + reset_cover=False, + path=db_path, + ) + after_title = title_reset["after"] + self.assertEqual("generated", after_title.stage) + self.assertEqual("success", after_title.status) + self.assertIsNone(after_title.new_title) + self.assertEqual(first_cover, after_title.new_cover_path) + self.assertIsNone(after_title.last_error) + self.assertTrue(os.path.exists(first_cover)) + + cover_reset = db.reset_generated( + second.id, + reset_title=False, + reset_cover=True, + delete_file=True, + path=db_path, + ) + after_cover = cover_reset["after"] + self.assertEqual("generated", after_cover.stage) + self.assertEqual("success", after_cover.status) + self.assertEqual("手动标题B", after_cover.new_title) + self.assertIsNone(after_cover.new_cover_path) + self.assertEqual(second_cover, cover_reset["deleted_file"]) + self.assertFalse(os.path.exists(second_cover)) + + with self.assertRaises(db.DbError): + db.reset_generated(first.id, reset_title=False, reset_cover=False, path=db_path) + with self.assertRaises(db.DbError): + db.reset_generated(first.id, reset_title=True, reset_cover=False, delete_file=True, path=db_path) + + self.assert_removed(temp_dir) + def test_delete_batch_soft_hides_batch_and_tasks(self): with self.make_temp_dir() as temp_dir: diff --git a/tests/test_gui.py b/tests/test_gui.py index b1491a7..04ae1f6 100644 --- a/tests/test_gui.py +++ b/tests/test_gui.py @@ -17,6 +17,7 @@ from app import accounts, ai, appconfig, db, prompts if gui.QT_IMPORT_ERROR is not None: raise unittest.SkipTest("PySide6 未安装") +from PySide6.QtCore import QItemSelectionModel from PySide6.QtGui import QTextCursor from PySide6.QtWidgets import QApplication, QCheckBox, QLineEdit, QPlainTextEdit, QProgressBar, QTableView @@ -1540,14 +1541,15 @@ class GuiTests(TempDirMixin, unittest.TestCase): self.addCleanup(tab.close) tab.task_table.selectRow(0) - with mock.patch("app.gui.QMessageBox.question", return_value=gui.QMessageBox.Yes) as question: + message_box, boxes = self.make_fake_message_box("重置全部") + with mock.patch("app.gui.tabs.generate.QMessageBox", message_box): tab.reset_generated_result() - message = question.call_args[0][2] + message = boxes[0].text self.assertIn("默认不删除本地新封面文件", message) - self.assertIn(new_cover, message) + self.assertIn("重置全部", message) updated = db.get_task(task.id, path=cfg["db_path"]) - self.assertEqual("collected", updated.stage) + self.assertEqual("generated", updated.stage) self.assertEqual("success", updated.status) self.assertIsNone(updated.new_title) self.assertIsNone(updated.new_cover_path) @@ -1555,12 +1557,147 @@ class GuiTests(TempDirMixin, unittest.TestCase): run_log = db.list_run_logs(limit=1, run_type="reset", path=cfg["db_path"])[0] self.assertEqual("done", run_log.status) self.assertEqual("reset_generated", run_log.options["action"]) + self.assertEqual("all", run_log.options["mode"]) events = db.list_run_log_events(run_log.id, path=cfg["db_path"]) self.assertTrue(any("action=reset_generated" in event.message for event in events)) self.assertIn("已重置生成结果", statuses[-1]) self.assert_removed(temp_dir) + def test_generate_tab_resets_titles_for_current_filtered_tasks(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", + }, + { + "source_file_abs": os.path.join(temp_dir, "input.xlsx"), + "source_sheet": "商品", + "source_row": 3, + "account_name": "Excel主店", + "alias": "alias-a", + "item_id": "51100639511", + }, + ], + path=cfg["db_path"], + ) + tasks = db.list_tasks(batch_id=batch_id, path=cfg["db_path"]) + for index, task in enumerate(tasks): + db.set_collected(task.id, f"旧标题{index}", f"old-{index}.jpg", path=cfg["db_path"]) + db.set_generated(task.id, f"新标题{index}", f"new-{index}.jpg", path=cfg["db_path"]) + statuses = [] + tab = GenerateTab(config=cfg, status_callback=statuses.append) + self.addCleanup(tab.close) + tab.task_table.clearSelection() + + message_box, boxes = self.make_fake_message_box("重置标题") + with mock.patch("app.gui.tabs.generate.QMessageBox", message_box): + tab.reset_generated_result() + + self.assertIn("当前筛选结果 2 条", boxes[0].text) + updated = db.list_tasks(batch_id=batch_id, path=cfg["db_path"]) + self.assertEqual([None, None], [task.new_title for task in updated]) + self.assertEqual(["new-0.jpg", "new-1.jpg"], [task.new_cover_path for task in updated]) + self.assertTrue(all(task.stage == "generated" for task in updated)) + run_log = db.list_run_logs(limit=1, run_type="reset", path=cfg["db_path"])[0] + self.assertEqual("filtered", run_log.options["scope"]) + self.assertEqual("title", run_log.options["mode"]) + self.assertTrue(run_log.options["reset_title"]) + self.assertFalse(run_log.options["reset_cover"]) + self.assertEqual(2, run_log.total) + self.assertIn("已重置生成结果：2 条，内容：标题", statuses[-1]) + + self.assert_removed(temp_dir) + + def test_generate_tab_resets_selected_covers_and_keeps_manual_titles(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", + }, + { + "source_file_abs": os.path.join(temp_dir, "input.xlsx"), + "source_sheet": "商品", + "source_row": 3, + "account_name": "Excel主店", + "alias": "alias-a", + "item_id": "51100639511", + }, + ], + path=cfg["db_path"], + ) + tasks = db.list_tasks(batch_id=batch_id, path=cfg["db_path"]) + for index, task in enumerate(tasks): + db.set_collected(task.id, f"旧标题{index}", f"old-{index}.jpg", path=cfg["db_path"]) + db.set_generated(task.id, f"手动标题{index}", f"new-{index}.jpg", path=cfg["db_path"]) + db.set_applied(tasks[0].id, True, path=cfg["db_path"]) + tab = GenerateTab(config=cfg) + self.addCleanup(tab.close) + selection = tab.task_table.selectionModel() + for row in [0, 1]: + selection.select( + tab.model.index(row, 0), + QItemSelectionModel.Select | QItemSelectionModel.Rows, + ) + + message_box, boxes = self.make_fake_message_box("重置封面") + with mock.patch("app.gui.tabs.generate.QMessageBox", message_box): + tab.reset_generated_result() + + message = boxes[0].text + self.assertIn("选中任务 2 条", message) + self.assertIn("已经提交过线上", message) + self.assertIn("重生成后再更新会再次提交线上", message) + updated = db.list_tasks(batch_id=batch_id, path=cfg["db_path"]) + self.assertEqual(["手动标题0", "手动标题1"], [task.new_title for task in updated]) + self.assertEqual([None, None], [task.new_cover_path for task in updated]) + self.assertEqual("generated", updated[0].stage) + self.assertEqual(1, updated[0].committed) + run_log = db.list_run_logs(limit=1, run_type="reset", path=cfg["db_path"])[0] + self.assertEqual("selected", run_log.options["scope"]) + self.assertEqual("cover", run_log.options["mode"]) + self.assertFalse(run_log.options["reset_title"]) + self.assertTrue(run_log.options["reset_cover"]) + self.assertEqual(1, run_log.options["committed_count"]) + + self.assert_removed(temp_dir) + + def test_generate_tab_disables_reset_while_generation_running(self): + with self.make_temp_dir() as temp_dir: + cfg = self.make_config(temp_dir) + statuses = [] + tab = GenerateTab(config=cfg, status_callback=statuses.append) + self.addCleanup(tab.close) + + tab._set_generate_running(True) + self.assertFalse(tab.reset_generate_button.isEnabled()) + tab.generate_thread = object() + tab.reset_generated_result() + + self.assertIn("AI 生成正在进行，不能重置", statuses[-1]) + + self.assert_removed(temp_dir) + def test_generate_worker_calls_generate_batch_and_emits_signals(self): with self.make_temp_dir() as temp_dir: cfg = self.make_config(temp_dir)