diff --git a/app/ai.py b/app/ai.py index 986eac0..e43a8cf 100644 --- a/app/ai.py +++ b/app/ai.py @@ -1082,6 +1082,15 @@ def _emit_generation_error(callback, task, phase, step, exc, error): "error": diagnostics.redact_log_text(error), "exception": exc, } + code = getattr(exc, "code", None) + status = getattr(exc, "status", None) + retryable = getattr(exc, "retryable", None) + if code is not None: + payload["code"] = str(code) + if status is not None: + payload["status"] = status + if retryable is not None: + payload["retryable"] = bool(retryable) try: callback(payload) except Exception: diff --git a/app/gui/tabs/generate.py b/app/gui/tabs/generate.py index e0141c1..9cb84a2 100644 --- a/app/gui/tabs/generate.py +++ b/app/gui/tabs/generate.py @@ -102,6 +102,8 @@ class GenerateTab(QWidget): self.cover_progress_bar.setValue(0) self.failed_progress_label = QLabel("失败 0") self.failed_progress_label.setObjectName("generateFailedProgressLabel") + self.cmhub_balance_label = QLabel("cmhub余额：未获取") + self.cmhub_balance_label.setObjectName("generateCmhubBalanceLabel") left_panel = QWidget() left_layout = QVBoxLayout(left_panel) @@ -196,6 +198,7 @@ class GenerateTab(QWidget): cover_progress_layout.addWidget(self.cover_progress_label) cover_progress_layout.addWidget(self.cover_progress_bar, 1) cover_progress_layout.addWidget(self.failed_progress_label) + cover_progress_layout.addWidget(self.cmhub_balance_label) progress_layout = QVBoxLayout() progress_layout.addLayout(title_progress_layout) progress_layout.addLayout(cover_progress_layout) @@ -239,6 +242,7 @@ class GenerateTab(QWidget): self.refresh_cover_templates() self.refresh_tasks() self._show_current_run_log_empty() + self._update_cmhub_balance_label({}) def _set_status(self, message): if self.status_callback is not None: @@ -278,7 +282,7 @@ class GenerateTab(QWidget): payload = { key: value for key, value in self.config.items() - if key not in {"config_path", "ai_models_path"} + if key not in {"config_path", "ai_models_path", "cmhub_config_path"} } payload["ai"] = ai_settings try: @@ -289,7 +293,7 @@ class GenerateTab(QWidget): internal = { key: value for key, value in self.config.items() - if key in {"config_path", "ai_models_path"} + if key in {"config_path", "ai_models_path", "cmhub_config_path"} } self.config.clear() self.config.update(saved) @@ -489,6 +493,7 @@ class GenerateTab(QWidget): worker.finished.connect(self._on_generate_finished) worker.cancelled.connect(self._on_generate_cancelled) self._reset_generate_run_log(tasks, generate_cover) + self._reset_cmhub_balance_label() thread = run_worker(worker, thread_name="GenerateWorker", start=False) thread.finished.connect(lambda: self._forget_generate_thread(thread)) self.generate_worker = worker @@ -640,6 +645,7 @@ class GenerateTab(QWidget): def _on_generate_progress(self, payload): self._update_generate_progress(payload) + self._update_cmhub_balance_label(payload) self._set_status("生成进度：" + self._generate_progress_text(payload)) def _on_generate_row_updated(self, task_id, fields): @@ -652,6 +658,11 @@ class GenerateTab(QWidget): self._set_generate_running(False) self.refresh_tasks() self._update_generate_progress(payload) + self._update_cmhub_balance_label(payload) + billing_error = payload.get("billing_error") + if billing_error: + self._show_billing_error(billing_error) + return if payload.get("error"): self._set_status(f"AI 生成失败：{payload.get('error')}") return @@ -661,6 +672,11 @@ class GenerateTab(QWidget): self._set_generate_running(False) self.refresh_tasks() self._update_generate_progress(payload) + self._update_cmhub_balance_label(payload) + billing_error = payload.get("billing_error") + if billing_error: + self._show_billing_error(billing_error) + return self._set_status("AI 生成已停止：" + self._generate_progress_text(payload)) def _update_generate_progress(self, payload): @@ -676,6 +692,36 @@ class GenerateTab(QWidget): self._set_progress_bar(self.title_progress_bar, title_done, total) self._set_progress_bar(self.cover_progress_bar, cover_done, cover_total) + def _reset_cmhub_balance_label(self): + if self._is_cmhub_backend(): + self.cmhub_balance_label.setVisible(True) + self.cmhub_balance_label.setText("cmhub余额：生成后刷新") + else: + self.cmhub_balance_label.setVisible(False) + + def _update_cmhub_balance_label(self, payload): + if not self._is_cmhub_backend(): + self.cmhub_balance_label.setVisible(False) + return + self.cmhub_balance_label.setVisible(True) + balance = payload.get("points_balance") if isinstance(payload, dict) else None + if balance is None: + if not self.cmhub_balance_label.text(): + self.cmhub_balance_label.setText("cmhub余额：未获取") + return + self.cmhub_balance_label.setText(f"cmhub余额：{balance}") + + def _is_cmhub_backend(self): + try: + return appconfig.ai_config(self.config).get("backend") == "cmhub" + except Exception: + return False + + def _show_billing_error(self, billing_error): + message = billing_error.get("message") or "点数不足，请先充值。" + QMessageBox.warning(self, "AI生成", message) + self._set_status("AI 生成已中止：" + message) + def _set_progress_bar(self, bar, done, total): maximum = max(1, int(total or 0)) value = min(max(0, int(done or 0)), maximum) diff --git a/app/gui/workers.py b/app/gui/workers.py index c5c6431..96cb2a9 100644 --- a/app/gui/workers.py +++ b/app/gui/workers.py @@ -24,6 +24,10 @@ class GenerateWorker(BaseWorker): self._account_by_alias = {} self._task_positions = {} self._eligible_total = 0 + self._last_progress_payload = {} + self._cmhub_points_balance = None + self._billing_error = None + self._billing_stop_requested = False def execute(self): account_rows = accounts.list_accounts(path=self.db_path, config=self.config) @@ -72,8 +76,8 @@ class GenerateWorker(BaseWorker): "on_error": self._on_generation_error, "generate_cover": generate_cover, }, - on_progress=self.progress.emit, - should_stop=self.should_cancel, + on_progress=self._emit_generate_progress, + should_stop=self._should_stop_generation, ) except Exception as exc: error = diagnostics.redact_log_text(str(exc) or exc.__class__.__name__) @@ -100,30 +104,59 @@ class GenerateWorker(BaseWorker): payload={"error": error}, exc=exc, ) + if self._cmhub_points_balance is not None: + summary["points_balance"] = self._cmhub_points_balance + if self._billing_error is not None: + summary["billing_error"] = dict(self._billing_error) + summary["ok"] = False + summary["cancelled"] = True summary["run_id"] = self._run_id summary["batch_ids"] = batch_ids - status = "cancelled" if summary.get("cancelled") else "done" - level = "warning" if summary.get("cancelled") or summary.get("error") else "info" + status = "failed" if summary.get("billing_error") or summary.get("error") else ("cancelled" if summary.get("cancelled") else "done") + level = "error" if summary.get("billing_error") or summary.get("error") else ("warning" if summary.get("cancelled") else "info") self._log_run_event(self._format_generate_completion(summary), level=level) self._finish_run_log(status, summary) return summary + def _emit_generate_progress(self, payload): + progress = dict(payload or {}) + if self._cmhub_points_balance is not None: + progress["points_balance"] = self._cmhub_points_balance + if self._billing_error is not None: + progress["billing_error"] = dict(self._billing_error) + self._last_progress_payload = dict(progress) + self.progress.emit(progress) + + def _should_stop_generation(self): + return self.should_cancel() or self._billing_stop_requested + def _emit_row_update(self, task_id, fields): self.row_updated.emit(int(task_id), dict(fields or {})) def _on_generation_event(self, payload): task = payload.get("task") + self._remember_cmhub_metadata(payload) message = self._format_generation_event(payload) if not message: return self._log_run_event(message, task=task, level=payload.get("level") or "info") + def _remember_cmhub_metadata(self, payload): + metadata = payload.get("metadata") + if not isinstance(metadata, dict): + return + if metadata.get("points_balance") is not None: + self._cmhub_points_balance = metadata.get("points_balance") + self._emit_generate_progress(self._last_progress_payload) + def _format_generation_event(self, payload): task = payload.get("task") phase = payload.get("phase") or "generate" step = payload.get("step") or "unknown" result = payload.get("result") or "start" detail = self._short_detail(payload.get("detail")) + if isinstance(payload.get("metadata"), dict): + return self._format_cmhub_billing_event(task, phase, payload.get("metadata")) if phase == "title": if result == "start" and step == "title_submit": return f"[标题] {self._task_progress_label(task)} 开始生成" @@ -154,6 +187,22 @@ class GenerateWorker(BaseWorker): return None return None + def _format_cmhub_billing_event(self, task, phase, metadata): + label = "标题" if phase == "title" else ("图片" if phase == "cover" else "AI") + parts = [] + alias = metadata.get("alias") or metadata.get("model_used") + if alias: + parts.append(f"别名 {alias}") + if metadata.get("points_cost") is not None: + parts.append(f"扣点 {metadata.get('points_cost')}") + if metadata.get("points_balance") is not None: + parts.append(f"余额 {metadata.get('points_balance')}") + if metadata.get("call_id"): + parts.append(f"call_id={metadata.get('call_id')}") + if not parts: + return None + return f"[计费] {self._task_plain_label(task)} {label}生成：" + "，".join(str(part) for part in parts) + def _retry_message(self, label, task, payload, detail): attempt = int(payload.get("attempt", 0) or 0) attempts = int(payload.get("attempts", 0) or 0) @@ -192,6 +241,9 @@ class GenerateWorker(BaseWorker): def _format_generate_completion(self, summary): progress = self._summary_text(summary) + billing_error = summary.get("billing_error") or {} + if billing_error: + return f"[失败] AI 生成已中止：{billing_error.get('message') or '点数不足，请先充值'}，{progress}" if summary.get("cancelled"): return f"[停止] AI 生成已停止：{progress}" if summary.get("error"): @@ -212,14 +264,40 @@ class GenerateWorker(BaseWorker): task = payload.get("task") phase = payload.get("phase") or "generate" step = payload.get("step") or "unknown" + exception = payload.get("exception") + code = payload.get("code") or getattr(exception, "code", None) + status = payload.get("status") or getattr(exception, "status", None) error = diagnostics.redact_log_text(payload.get("error") or "未知错误") + if str(code or "") == "insufficient_points": + self._billing_stop_requested = True + message = "点数不足，请先充值。本轮未开始任务将停止。" + self._billing_error = { + "code": "insufficient_points", + "message": message, + "phase": phase, + "task_id": getattr(task, "id", None), + "item_id": getattr(task, "item_id", None), + } + if status is not None: + self._billing_error["status"] = status + self._log_run_event( + f"[计费] {self._task_plain_label(task)} 点数不足，请先充值；本轮未开始任务将停止", + task=task, + level="error", + ) + self._emit_generate_progress(self._last_progress_payload) + diagnostic_payload = {"phase": phase, "error": error} + if code is not None: + diagnostic_payload["code"] = str(code) + if status is not None: + diagnostic_payload["status"] = status self._write_diagnostic_log( "AI生成任务失败", level="ERROR", step=step, task=task, - payload={"phase": phase, "error": error}, - exc=payload.get("exception"), + payload=diagnostic_payload, + exc=exception, ) def _batch_ids(self, tasks): @@ -245,6 +323,7 @@ class GenerateWorker(BaseWorker): "title_concurrency": ai_cfg.get("title_concurrency"), "image_concurrency": ai_cfg.get("image_concurrency"), "generate_cover": ai_cfg.get("generate_cover", False), + "backend": ai_cfg.get("backend", "direct"), }, path=self.db_path, ) diff --git a/docs/06-tasks.md b/docs/06-tasks.md index 6fa758b..57e6c7b 100644 --- a/docs/06-tasks.md +++ b/docs/06-tasks.md @@ -126,7 +126,7 @@ | --- | --- | --- | --- | --- | | T-526 | `app/ai.py` + `appconfig` 接入 cmhub backend | T-301, T-303, T-520 | 依据 `docs/cmhub-integration-design.md` v3.2。`config.json` 的 `ai` 段加 `backend`（`cmhub`/`direct`）+ `cmhub` 子段（`base_url`/`title_alias`/`image_alias`/`connect_timeout` 等）；为保护既有用户，全新配置默认 `backend=direct`、加载既有配置缺 `backend` 时也补 `direct`，`cmhub` 一律由用户在⑤显式 opt-in；`backend=cmhub` 但 `base_url`/Key 缺失时须抛清晰「请去⑤配置 cmhub」错误（`CMHubError`/`AIError`），不崩溃、不静默直连。cmhub API Key 固定存 `config/cmhub.json`（schema `{ "api_key": "..." }`），新增读写/打码 helper 并把该文件加入 `.gitignore`；日志脱敏。`gen_title`/`gen_cover` **返回值不变**，内层按 backend 分流并保留 `direct`；计费元数据不塞进返回值，允许给二者**新增一个可选事件回调参数**（如 `on_meta`/`on_event`）承载，属向后兼容加参，`generate_batch` 显式传回调不受影响。cmhub 分支：生文 `POST /api/v1/generate/title` 体 `{prompt,model:别名,resolution?}`、取 `titles[0]`、空则 `AIError`；生图 `POST /api/v1/generate/image` 体 `{prompt,model:别名,image_base64:<旧封面>,resolution,aspect_ratio:"1:1"}`、拿 `image_url` 后**立即下载**再走 `_save_jpeg`；`resolution` 归一大写 `512/1K/2K/4K`。新增 `CMHubError(AIError)`，带 `code/status/retryable/retry_after`，错误按 `code` 优先分支（`insufficient_points`/`unauthorized`/`account_disabled`/`bad_request`/`model_not_allowed`/`no_pricing_rule`/`content_blocked`/`upstream_error`/`rate_limited`，未知 code 当不可重试）；cmhub HTTP helper 需区分 connect/read timeout（优先用 `requests timeout=(connect, read)`），只对 502/429/连接超时重试，生图读超时绝不自动重发，读超时按分辨率封顶 600s。`points_cost`/`points_balance`/`call_id` 不改返回值，通过 `on_step`/事件回调上报；T-526 只保证 metadata 事件完整传出，T-528 再由 GUI worker 脱敏写 run_logs 和余额展示。`image_url` 下载必须限制 http/https、拒绝内网/回环地址、校验域名解析后的 IP 仍不是内网/回环/本机地址，并设置超时和大小上限。新增 `fetch_cmhub_models(base_url, api_key)` helper 调 `GET /api/v1/models` 返回别名清单（`alias/operation_type/requires_image/pricing_status/prices`）供 T-527 渲染下拉，错误脱敏。不碰 editor/cdp/chrome/accounts/excel/db，也不改 ①③④流程。`tests/test_ai.py` 加 cmhub mock（titles 列表、image_url 下载、安全下载、错误码与重试、读超时不重发、metadata 事件）、`tests/test_appconfig.py` 加 schema 和 key 文件 helper，direct 用例保持绿 | DONE | | T-527 | ⑤设置 cmhub 网关面板 | T-526, T-517 | 依据 `docs/cmhub-integration-design.md` v3.2。⑤ AI 设置按 `backend` 切换：cmhub 模式显示「网关 Base URL + API Key（打码，提示从网页端复制、仅显示一次）+ 生文别名 + 生图别名 + 测试连接/查余额」；**别名从 `GET /api/v1/models`（T-526 的 `fetch_cmhub_models`）动态拉取渲染下拉**，按 `operation_type` 分生文/生图，过滤 `pricing_status="unpriced"` 的别名，可展示单价与 `requires_image` 提示，选中值持久化到 `ai.cmhub.title_alias/image_alias`（网关临时不可达时回退已存值）；不写死别名。direct 模式保留现有 AI 模型 master-detail。保存写 `config.json` 的 `ai` 段与 `config/cmhub.json`；切换 backend 时不删除 legacy `config/ai_models.json`。测试连接/查余额经后台 worker 调 cmhub（复用 `AIModelTestWorker` 思路或新增 worker），错误必须脱敏并给用户可读提示。同步 GUI 设置测试；不改 Shopee/CDP 流程 | DONE | -| T-528 | ② 计费错误提示 + 余额展示 | T-526, T-527, T-303 | 依据 `docs/cmhub-integration-design.md` v3.2。② AI生成页把 cmhub 计费失败态显式化：通过 `CMHubError.code` 识别 `insufficient_points`，弹明确提示「点数不足，请先充值」并引导去网页端充值，本轮未开始任务可提前中止，不靠中文字符串匹配、不淹没在失败计数里；用 T-526 成功响应事件里的 `points_balance` 刷新②页剩余点数显示，`/balance` 仅作手动刷新/可选批量前预检；`points_cost`/`call_id` 记入脱敏 run_logs。只改② UI、`GenerateWorker` 事件处理/文案及 GUI 单测；不改 AI HTTP 协议、DB schema、Excel、Shopee/CDP 流程 | TODO | +| T-528 | ② 计费错误提示 + 余额展示 | T-526, T-527, T-303 | 依据 `docs/cmhub-integration-design.md` v3.2。② AI生成页把 cmhub 计费失败态显式化：通过 `CMHubError.code` 识别 `insufficient_points`，弹明确提示「点数不足，请先充值」并引导去网页端充值，本轮未开始任务可提前中止，不靠中文字符串匹配、不淹没在失败计数里；用 T-526 成功响应事件里的 `points_balance` 刷新②页剩余点数显示，`/balance` 仅作手动刷新/可选批量前预检；`points_cost`/`call_id` 记入脱敏 run_logs。只改② UI、`GenerateWorker` 事件处理/文案及 GUI 单测；不改 AI HTTP 协议、DB schema、Excel、Shopee/CDP 流程 | DONE | ## Phase 8 · 工程基础设施后续（`docs/engineering-review.md`） diff --git a/docs/current-state.md b/docs/current-state.md index 0dbe5a5..0560228 100644 --- a/docs/current-state.md +++ b/docs/current-state.md @@ -6,10 +6,10 @@ ## 当前快照 - 日期：2026-07-04 -- 阶段：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-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 网关面板。 +- 阶段：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-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 ②计费错误提示与余额展示。 - 技术栈：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（direct: `config/ai_models.json` 通用 HTTP；cmhub: `config.json` backend + `config/cmhub.json` Key），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 更新安全与执行模式默认值，`config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接与 OpenAI-compatible base URL 自动补 endpoint，`config/cmhub.json` cmhub Key 读写/打码 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 通过事件回调传出，且保留重试、错误脱敏、图片 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 服务、端口默认分配、启动登录、检测登录、生成快捷方式；`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生成运行日志/重置生成结果与 `GenerateWorker`、③ 更新shopee批次/店铺/商品ID/状态筛选栏/任务列表/开始更新主按钮/重置更新状态右键菜单/更新安全开关拦截与「前往设置」跳转/「检查本轮更新」按钮/开始更新确认弹窗/确认后 `ApplyWorker` 按每批最大更新条数分批执行当前筛选全部可更新记录/账号就绪和端口冲突预检/按账号并行可选/逐条 `set_applied`/运行日志/自动回写结果到 Excel/结束汇总弹窗/手动回写重试按钮、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏、密码明文保存提示、⑤ 设置按 AI backend 切换：direct 模式保留 AI 模型同列上下布局、三列详情/参数表单、API Key 明文保存提示和 `AIModelTestWorker` 后台测试连接，cmhub 模式显示网关 Base URL、API Key、动态生文/生图别名下拉、刷新别名与测试连接/查余额；通用角色/生成参数、路径/端口、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/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；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 更新安全与执行模式默认值，`config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接与 OpenAI-compatible base URL 自动补 endpoint，`config/cmhub.json` cmhub Key 读写/打码 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 通过事件回调传出，且保留重试、错误脱敏、图片 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 服务、端口默认分配、启动登录、检测登录、生成快捷方式；`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/结束汇总弹窗/手动回写重试按钮、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏、密码明文保存提示、⑤ 设置按 AI backend 切换：direct 模式保留 AI 模型同列上下布局、三列详情/参数表单、API Key 明文保存提示和 `AIModelTestWorker` 后台测试连接，cmhub 模式显示网关 Base URL、API Key、动态生文/生图别名下拉、刷新别名与测试连接/查余额；通用角色/生成参数、路径/端口、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/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余额显示/计费日志/点数不足中止提示；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。 ## 既定设计要点（文档已定） @@ -38,19 +38,19 @@ | `requirements-build.txt` | 已有 | T-524 产出：打包环境依赖，先安装运行依赖再安装 PyInstaller 5.13.2；不作为运行时依赖文件 | | `.github/workflows/tests.yml` | 已有 | T-522 产出：push / pull_request 自动在 `windows-latest` + Python 3.11 安装 `requirements.txt`，设置 `QT_QPA_PLATFORM=offscreen`，运行 `python -m compileall app main.py` 与 `python -m unittest discover -s tests` | | `cmshopee.spec` / `scripts/build_exe.ps1` | 已有 | T-524 产出：PyInstaller onedir 构建配置与发布目录校验脚本；`cmshopee.spec` 不声明本地数据 `datas`，脚本检查 `dist\\cmshopee` 不含配置、DB、图片、日志、Chrome 登录态或提示词 | -| `app/gui/` | 已有 | T-523 产出：由旧 `app/gui.py` 拆分的 PySide6 GUI 包，包入口 `__init__.py` 继续兼容 `from app import gui` / `from app.gui import MainWindow`；`main_window.py` 放 MainWindow，`models.py` 放 3 个 TableModel，`widgets.py` 放色板/空状态/批次总览/helper，`workers.py` 放具体 GUI workers，`tabs/` 放 ①~⑤ Tab； T-104/T-105/T-106/T-202/T-202b/T-203/T-204/T-204b/T-205/T-206/T-207/T-302/T-302p/T-303/T-303b/T-401/T-402/T-403/T-404a/T-501/T-501b/T-501c/T-503/T-504/T-506/T-507/T-508/T-509/T-510/T-505/T-511/T-512/T-513/T-514/T-515/T-516/T-517/T-518/T-519/T-520/T-527 产出：PySide6 `QMainWindow` + 五 Tab；顶部 Tab 栏防误点样式；统一语义色板、①②③状态列前景色、③开始更新 warning 样式、①导入校验数字标红、④登录状态点、③Tab warning 小圆点、删除按钮 danger 样式、①②③首次空状态引导卡片和批次阶段进度总览；① 导入采集导入按钮、导入汇总栏、批次/店铺/商品ID/状态筛选、删除批次软删除按钮、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker（旧封面路径按批次/店铺/任务细分）、采集前账号就绪预检与④引导、采集完成自动回写与手动重试、采集运行日志视图；② AI生成左右布局、提示词管理、批次/店铺/商品ID/状态筛选栏、任务列表、新标题列本地编辑、开始生成/停止/标题与图片双进度条、双击新旧封面预览、用户可读自动滚动AI生成运行日志、重置生成结果与 `GenerateWorker`；③ 更新shopee批次/店铺/商品ID/状态筛选栏、任务列表、开始更新主按钮、重置更新状态右键菜单、Shopee 更新安全拦截与「前往设置」跳转、「检查本轮更新」按钮、开始更新确认弹窗、`ApplyWorker` 检查/分批串行/按账号并行、账号与端口预检、运行日志、逐条 `set_applied`、自动回写结果到 Excel、结束汇总弹窗与手动回写重试；④ 账号管理表格、账号弹窗、密码本地明文保存提示、启动登录、检测登录、快捷方式；⑤ 设置 AI backend 选择，direct 模式保留模型下拉、新增/删除、详情编辑、密钥打码与本地明文保存提示、测试连接 worker、默认角色下拉，cmhub 模式提供 Base URL、API Key、生文/生图动态别名、刷新别名、测试连接/查余额和 `config/cmhub.json` 保存；通用生成参数、路径端口配置、三列 Shopee 更新安全与执行模式设置仍保留，高频安全/执行模式分区前置、基础设施分区后置，`test_item_id`/`dry_run` 兼容字段无用户入口且普通更新不再阻断正式更新；②封面模板「另存为/重命名/删除」低频操作已收敛进「模板操作」菜单；②AI生成底部已增加标题/图片双进度条，运行日志已改为用户可读、自动滚动、脱敏的长任务日志，且②本轮「生成封面图片（成本较高）」开关默认关闭并持久化到 `ai.generate_cover`，关闭时只生成标题并进入可更新状态 | +| `app/gui/` | 已有 | T-523 产出：由旧 `app/gui.py` 拆分的 PySide6 GUI 包，包入口 `__init__.py` 继续兼容 `from app import gui` / `from app.gui import MainWindow`；`main_window.py` 放 MainWindow，`models.py` 放 3 个 TableModel，`widgets.py` 放色板/空状态/批次总览/helper，`workers.py` 放具体 GUI workers，`tabs/` 放 ①~⑤ Tab； T-104/T-105/T-106/T-202/T-202b/T-203/T-204/T-204b/T-205/T-206/T-207/T-302/T-302p/T-303/T-303b/T-401/T-402/T-403/T-404a/T-501/T-501b/T-501c/T-503/T-504/T-506/T-507/T-508/T-509/T-510/T-505/T-511/T-512/T-513/T-514/T-515/T-516/T-517/T-518/T-519/T-520/T-527/T-528 产出：PySide6 `QMainWindow` + 五 Tab；顶部 Tab 栏防误点样式；统一语义色板、①②③状态列前景色、③开始更新 warning 样式、①导入校验数字标红、④登录状态点、③Tab warning 小圆点、删除按钮 danger 样式、①②③首次空状态引导卡片和批次阶段进度总览；① 导入采集导入按钮、导入汇总栏、批次/店铺/商品ID/状态筛选、删除批次软删除按钮、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker（旧封面路径按批次/店铺/任务细分）、采集前账号就绪预检与④引导、采集完成自动回写与手动重试、采集运行日志视图；② AI生成左右布局、提示词管理、批次/店铺/商品ID/状态筛选栏、任务列表、新标题列本地编辑、开始生成/停止/标题与图片双进度条、双击新旧封面预览、用户可读自动滚动AI生成运行日志、重置生成结果与 `GenerateWorker`；③ 更新shopee批次/店铺/商品ID/状态筛选栏、任务列表、开始更新主按钮、重置更新状态右键菜单、Shopee 更新安全拦截与「前往设置」跳转、「检查本轮更新」按钮、开始更新确认弹窗、`ApplyWorker` 检查/分批串行/按账号并行、账号与端口预检、运行日志、逐条 `set_applied`、自动回写结果到 Excel、结束汇总弹窗与手动回写重试；④ 账号管理表格、账号弹窗、密码本地明文保存提示、启动登录、检测登录、快捷方式；⑤ 设置 AI backend 选择，direct 模式保留模型下拉、新增/删除、详情编辑、密钥打码与本地明文保存提示、测试连接 worker、默认角色下拉，cmhub 模式提供 Base URL、API Key、生文/生图动态别名、刷新别名、测试连接/查余额和 `config/cmhub.json` 保存；通用生成参数、路径端口配置、三列 Shopee 更新安全与执行模式设置仍保留，高频安全/执行模式分区前置、基础设施分区后置，`test_item_id`/`dry_run` 兼容字段无用户入口且普通更新不再阻断正式更新；②封面模板「另存为/重命名/删除」低频操作已收敛进「模板操作」菜单；②AI生成底部已增加标题/图片双进度条，运行日志已改为用户可读、自动滚动、脱敏的长任务日志，且②本轮「生成封面图片（成本较高）」开关默认关闭并持久化到 `ai.generate_cover`，关闭时只生成标题并进入可更新状态；②cmhub模式会显示生成后的剩余点数，记录 points_cost/call_id 计费日志，并在点数不足时弹提示且中止未开始任务 | | `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/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 产出：`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 产出：`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/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/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` 快捷方式 | -| `tests/` | 已有 | T-006/T-201/T-202/T-202b/T-203/T-204/T-204b/T-205/T-206/T-207/T-301/T-302/T-302p/T-303/T-303b/T-401/T-402/T-403/T-501/T-501b/T-501c/T-502/T-503/T-504/T-506/T-507/T-508/T-509/T-510/T-505/T-511/T-512/T-513/T-514/T-515/T-516/T-517/T-518/T-519/T-520/T-523/T-523a/T-404b/T-524 产出：stdlib unittest 基座；覆盖 appconfig/db/config/accounts/chrome/editor/excel/ai/image_paths/prompts/gui/workers/packaging | +| `tests/` | 已有 | T-006/T-201/T-202/T-202b/T-203/T-204/T-204b/T-205/T-206/T-207/T-301/T-302/T-302p/T-303/T-303b/T-401/T-402/T-403/T-501/T-501b/T-501c/T-502/T-503/T-504/T-506/T-507/T-508/T-509/T-510/T-505/T-511/T-512/T-513/T-514/T-515/T-516/T-517/T-518/T-519/T-520/T-523/T-523a/T-404b/T-524/T-526/T-527/T-528 产出：stdlib unittest 基座；覆盖 appconfig/db/config/accounts/chrome/editor/excel/ai/image_paths/prompts/gui/workers/packaging | | `app/excel.py` | 已有 | T-201/T-204/T-403 产出：多文件 Excel 输入列解析、必需列整文件拒绝、脏行逐行跳过、批次/任务入库、匹配统计；按源文件/工作表/行号回写旧标题与旧封面路径；按源文件/工作表/行号回写新标题、新封面路径、更新状态；支持原文件被占用时另存副本 | | `shopee待处理任务模板.xlsx` | 已有，已提交 | 标准空 Excel 模板；单工作表 `待处理任务`，表头 `账号名 | 别名 | 商品id | 旧标题 | 旧封面图片路径 | 新标题 | 新封面图片路径 | 更新状态`；运营复制后填写，填写副本不提交 | | `config.json` / `config/ai_models.json` / `config/cmhub.json` / `cmshopee.db` / `cmshopee.db-*` / `chrome_user_data_dir/` / `images/` / `logs/` / `prompts/` / `title_prompt.txt` | 本地存在或按需生成，已忽略 | 含配置、密钥、业务、登录态、图片、本地诊断日志和用户提示词，不提交版本库；密码/API Key 保存或变更时提示，展示/日志/导出脱敏 | @@ -68,8 +68,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-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-525 ruff 顺延到 cmhub 对接后。 +- 已完成：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-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-525（引入 ruff lint + format）；Phase 7 cmhub 对接已完成。 ## 当前已知限制 @@ -90,13 +90,14 @@ - T-518 已完成：②AI生成左栏保留模板选择、编辑内容、新建/保存等高频入口，把封面模板「另存为 / 重命名 / 删除」收敛进②本页「模板操作」菜单；动作仍调用原有保存/重命名/删除方法，删除二次确认、错误处理和 prompts 文件结构不变，不涉及 AI/Shopee/CDP。 - T-519 已完成：②AI生成底部增加标题/图片两条独立进度条，现有「AI生成运行日志」改为用户可读、自动滚动、脱敏的长任务日志；日志显示本轮开始、标题/图片开始与成功、商品ID/店铺、调用失败重试、失败、停止请求和完成汇总，不显示 API Key、密码、Cookie、token、完整请求体、base64 图片或超长 prompt，不改变 AI/DB/Excel/Shopee/CDP 核心流程。 - T-520 已完成：②AI生成页增加「生成封面图片（成本较高）」开关，默认关闭以避免无意产生图片模型成本；关闭时标题生成成功即可写库进入 generated，`new_cover_path=NULL`，③只更新标题且不受「允许更新封面」阻断；开启时保持标题后图片两段生成流程。未改 DB schema、AI HTTP 协议、Excel、Shopee/CDP 更新流程。 +- T-528 已完成：②AI生成页在 cmhub backend 下显示生成后的剩余点数；GenerateWorker 读取计费 metadata，把 points_cost、points_balance、call_id 写入脱敏运行日志；遇到 CMHubError.code=insufficient_points 时弹出「点数不足，请先充值」并中止本轮未开始任务，不靠中文错误字符串匹配。未改 AI HTTP 协议、DB schema、Excel、Shopee/CDP 流程。 - T-521 已完成：根目录新增 `requirements.txt` 并锁定当前运行依赖版本；`docs/03-tech-stack.md` 和 `docs/README.md` 已改为 `python -m pip install -r requirements.txt` 安装；未新增运行时依赖、未改业务代码。 - T-522 已完成：新增 `.github/workflows/tests.yml`，在 push / pull_request 上用 Windows + Python 3.11 安装 `requirements.txt`，设置 `QT_QPA_PLATFORM=offscreen`，自动运行语法检查和全量单元/GUI 测试；CI 不连接真实 Shopee 或真实 AI。 - T-523 已完成：旧 `app/gui.py` 已拆为 `app/gui/` 包，公开导入路径保持兼容；本轮为纯结构重构，未改 Shopee/CDP、AI、DB、Excel 行为。 - 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 已接 `app/ai.py`/`appconfig` backend，T-527 已完成⑤设置页 cmhub backend 切换、动态别名下拉、测试连接/查余额 worker 和 mock 单测；下一步 T-528 做②计费错误提示与余额展示；T-525 ruff 因业务优先级顺延。 +- Phase 7 cmhub 网关对接已完成：T-526 已接 `app/ai.py`/`appconfig` backend，T-527 已完成⑤设置页 cmhub backend 切换、动态别名下拉、测试连接/查余额 worker 和 mock 单测，T-528 已完成②cmhub余额展示、计费日志和点数不足中止提示；下一步回到 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，且不要提交该文件。 diff --git a/progress.md b/progress.md index 6e48d5f..386a120 100644 --- a/progress.md +++ b/progress.md @@ -1124,4 +1124,14 @@ - 测试：`tests/test_ai.py` 覆盖 cmhub 余额 helper；`tests/test_gui.py` 覆盖 `CMHubSettingsWorker` 成功/失败脱敏、设置页 cmhub backend 切换、保存 `config.json` + `config/cmhub.json`、别名过滤与已存值保留。 - 文档：`docs/06-tasks.md` 将 T-527 标为 DONE；同步 `docs/current-state.md`，下一步为 T-528。 - 验证：`python -m py_compile tests\test_ai.py tests\test_gui.py` 通过；`python -m unittest discover -s tests -p "test_ai.py"` 通过（17 tests）；`python -m unittest discover -s tests -p "test_gui.py"` 通过（77 tests）；`python -m compileall app main.py` 通过；`python -m unittest discover -s tests` 通过（184 tests）；`git diff --check` 无空白错误（仅 LF/CRLF 提示）。全量测试仍有本机 PySide6 字体目录提示，不影响结果。 -- 下一步：T-528 ②计费错误提示 + 余额展示。 \ No newline at end of file +- 下一步：T-528 ②计费错误提示 + 余额展示。 + +## 【2026-07-04】T-528 完成 · ②计费错误提示 + 余额展示 + +- 状态：DONE +- 代码：②AI生成页新增 cmhub 余额显示，仅在 `backend=cmhub` 时展示；`GenerateWorker` 读取 cmhub metadata，把 `points_cost`、`points_balance`、`call_id` 以脱敏用户日志写入 `run_logs/run_log_events`，并把最新 `points_balance` 回传给页面。 +- 点数不足：`app/ai.py` 的生成错误回调会透传 `CMHubError.code/status/retryable`；`GenerateWorker` 通过 `code=insufficient_points` 识别点数不足，设置本轮停止标记，弹「点数不足，请先充值。本轮未开始任务将停止。」并让未开始任务不再继续，不靠中文错误字符串匹配。 +- 边界：只改②AI生成 UI、worker 事件处理和测试；未改 AI HTTP 协议、DB schema、Excel、Shopee/CDP 流程。 +- 文档：`docs/06-tasks.md` 将 T-528 标为 DONE；同步 `docs/current-state.md`，下一个可领取任务回到 T-525 ruff。 +- 测试：`python -m unittest discover -s tests -p test_gui.py` 通过（80 tests）；`python -m unittest discover -s tests -p test_ai.py` 通过（17 tests）；`python -m compileall app main.py` 通过；`python -m unittest discover -s tests` 通过（187 tests）；`git diff --check` 通过，仅有本机 LF/CRLF 提示。全量测试仍有本机 PySide6 字体目录提示，不影响结果。 +- 下一步：T-525「引入 ruff（lint + format）+ 可选 pre-commit」。 diff --git a/tests/test_gui.py b/tests/test_gui.py index f479e64..1b8bfe1 100644 --- a/tests/test_gui.py +++ b/tests/test_gui.py @@ -12,7 +12,7 @@ sys.path.insert(0, os.path.dirname(__file__)) from _helpers import TempDirMixin from app import gui -from app import accounts, appconfig, db, prompts +from app import accounts, ai, appconfig, db, prompts if gui.QT_IMPORT_ERROR is not None: raise unittest.SkipTest("PySide6 未安装") @@ -962,6 +962,9 @@ class GuiTests(TempDirMixin, unittest.TestCase): self.assertEqual("标题 0/0", tab.title_progress_label.text()) self.assertEqual("图片 0/0", tab.cover_progress_label.text()) self.assertEqual("失败 0", tab.failed_progress_label.text()) + self.assertEqual("generateCmhubBalanceLabel", tab.cmhub_balance_label.objectName()) + self.assertEqual("cmhub余额：未获取", tab.cmhub_balance_label.text()) + self.assertTrue(tab.cmhub_balance_label.isHidden()) self.assertEqual(0, tab.title_progress_bar.value()) self.assertEqual(0, tab.cover_progress_bar.value()) self.assertEqual(1, tab.title_progress_bar.maximum()) @@ -981,6 +984,52 @@ class GuiTests(TempDirMixin, unittest.TestCase): self.assert_removed(temp_dir) + def test_generate_tab_shows_cmhub_balance_and_billing_error(self): + with self.make_temp_dir() as temp_dir: + cfg = self.make_config(temp_dir) + cfg["ai"] = appconfig.ai_config(cfg) + cfg["ai"]["backend"] = "cmhub" + statuses = [] + tab = GenerateTab(config=cfg, status_callback=statuses.append) + self.addCleanup(tab.close) + + self.assertFalse(tab.cmhub_balance_label.isHidden()) + self.assertEqual("cmhub余额：未获取", tab.cmhub_balance_label.text()) + + tab._on_generate_progress( + { + "total": 1, + "title_done": 1, + "cover_done": 0, + "cover_total": 0, + "failed": 0, + "generate_cover": False, + "points_balance": 88, + } + ) + self.assertEqual("cmhub余额：88", tab.cmhub_balance_label.text()) + + with mock.patch("app.gui.QMessageBox.warning") as warning: + tab._on_generate_finished( + { + "total": 1, + "title_done": 0, + "cover_done": 0, + "cover_total": 0, + "failed": 1, + "billing_error": { + "code": "insufficient_points", + "message": "点数不足，请先充值。本轮未开始任务将停止。", + }, + } + ) + + warning.assert_called_once() + self.assertIn("点数不足，请先充值", warning.call_args[0][2]) + self.assertIn("AI 生成已中止", statuses[-1]) + + self.assert_removed(temp_dir) + def test_generate_tab_persists_generate_cover_toggle(self): with self.make_temp_dir() as temp_dir: cfg = self.make_config(temp_dir) @@ -1262,6 +1311,169 @@ class GuiTests(TempDirMixin, unittest.TestCase): self.assert_removed(temp_dir) + def test_generate_worker_records_cmhub_billing_metadata(self): + with self.make_temp_dir() as temp_dir: + cfg = self.make_config(temp_dir) + cfg["ai"] = appconfig.ai_config(cfg) + cfg["ai"]["backend"] = "cmhub" + cfg["ai"]["generate_cover"] = True + 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"]) + tasks = db.list_tasks(batch_id=batch_id, path=cfg["db_path"]) + progress = [] + logs = [] + + def fake_generate_batch(tasks_arg, prompt_values, ai_cfg=None, on_progress=None, should_stop=None): + on_progress({"total": 1, "title_done": 1, "cover_done": 0, "failed": 0}) + ai_cfg["on_event"]( + { + "task": tasks_arg[0], + "phase": "title", + "step": "title_request", + "result": "meta", + "metadata": { + "alias": "title-standard", + "points_cost": 1, + "points_balance": 88, + "call_id": "call-1", + }, + } + ) + return { + "ok": True, + "total": 1, + "title_done": 1, + "cover_done": 1, + "cover_total": 1, + "generated_done": 1, + "failed": 0, + "generate_cover": True, + } + + worker = GenerateWorker( + tasks, + {"title": "标题提示", "cover": "封面提示"}, + db_path=cfg["db_path"], + config=cfg, + ) + worker.progress.connect(progress.append) + worker.log.connect(logs.append) + + with mock.patch("app.gui.ai.generate_batch", side_effect=fake_generate_batch): + summary = worker.execute() + + self.assertEqual(88, summary["points_balance"]) + self.assertEqual(88, progress[-1]["points_balance"]) + joined_logs = "\n".join(logs) + self.assertIn("[计费] 商品 51100639510", joined_logs) + self.assertIn("别名 title-standard", joined_logs) + self.assertIn("扣点 1", joined_logs) + self.assertIn("余额 88", joined_logs) + self.assertIn("call_id=call-1", joined_logs) + run_log = db.list_run_logs(limit=1, run_type="generate", path=cfg["db_path"])[0] + self.assertEqual("cmhub", run_log.options["backend"]) + events = db.list_run_log_events(summary["run_id"], path=cfg["db_path"]) + messages = "\n".join(event.message for event in events) + self.assertIn("[计费] 商品 51100639510", messages) + self.assertIn("余额 88", messages) + + self.assert_removed(temp_dir) + + def test_generate_worker_stops_on_cmhub_insufficient_points(self): + with self.make_temp_dir() as temp_dir: + cfg = self.make_config(temp_dir) + cfg["ai"] = appconfig.ai_config(cfg) + cfg["ai"]["backend"] = "cmhub" + 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"]) + tasks = db.list_tasks(batch_id=batch_id, path=cfg["db_path"]) + progress = [] + logs = [] + + def fake_generate_batch(tasks_arg, prompt_values, ai_cfg=None, on_progress=None, should_stop=None): + exc = ai.CMHubError("insufficient_points", "点数不足，请先充值", status=402) + ai_cfg["on_error"]( + { + "task": tasks_arg[0], + "phase": "title", + "step": "title_request", + "error": str(exc), + "exception": exc, + "code": exc.code, + "status": exc.status, + } + ) + self.assertTrue(should_stop()) + return { + "ok": False, + "total": 1, + "title_done": 0, + "cover_done": 0, + "cover_total": 0, + "generated_done": 0, + "failed": 1, + "cancelled": should_stop(), + "generate_cover": False, + } + + worker = GenerateWorker( + tasks, + {"title": "标题提示", "cover": "封面提示"}, + db_path=cfg["db_path"], + config=cfg, + ) + worker.progress.connect(progress.append) + worker.log.connect(logs.append) + + with mock.patch("app.gui.ai.generate_batch", side_effect=fake_generate_batch): + summary = worker.execute() + + self.assertFalse(summary["ok"]) + self.assertTrue(summary["cancelled"]) + self.assertEqual("insufficient_points", summary["billing_error"]["code"]) + self.assertIn("点数不足，请先充值", summary["billing_error"]["message"]) + self.assertEqual("insufficient_points", progress[-1]["billing_error"]["code"]) + joined_logs = "\n".join(logs) + self.assertIn("[计费] 商品 51100639510", joined_logs) + self.assertIn("点数不足，请先充值", joined_logs) + run_log = db.list_run_logs(limit=1, run_type="generate", path=cfg["db_path"])[0] + self.assertEqual("failed", run_log.status) + self.assertEqual("insufficient_points", run_log.summary["billing_error"]["code"]) + + self.assert_removed(temp_dir) + def test_generate_worker_writes_run_log_and_diagnostic_log_on_image_failure(self): with self.make_temp_dir() as temp_dir: cfg = self.make_config(temp_dir)