fix: 采集后关闭自动新建商品页
- 新增 close_tab 和 create_tab_info，区分断开 CDP 连接与关闭浏览器 target - open_product 标记商品页是否本轮自动新建，collect 结束后只关闭自动新建 tab - 保留用户原本打开的商品 tab，③ 更新流程暂不自动关闭页面 - 补充 editor tab 生命周期测试并同步架构、API、任务和进度文档
This commit is contained in:
chengma
+22
-3
@@ -30,6 +30,19 @@ def http_get(path, host=None):
|
||||
return s.get(f"{_base(host)}{path}", timeout=10).json()
|
||||
|
||||
|
||||
def close_tab(target_id, host=None):
|
||||
"""Close a Chrome target/page by target id."""
|
||||
|
||||
if not target_id:
|
||||
return False
|
||||
import requests
|
||||
|
||||
s = requests.Session()
|
||||
s.trust_env = False
|
||||
response = s.get(f"{_base(host)}/json/close/{target_id}", timeout=10)
|
||||
return response.ok
|
||||
|
||||
|
||||
class CDP:
|
||||
"""单个 target 的 CDP 客户端：命令同步、事件回调异步。"""
|
||||
|
||||
@@ -133,8 +146,8 @@ def find_product_tab(item_id, host=None):
|
||||
return None
|
||||
|
||||
|
||||
def create_tab(url, host=None):
|
||||
"""用 browser 级 Target.createTarget 新建 tab，返回其 page websocket。"""
|
||||
def create_tab_info(url, host=None):
|
||||
"""用 browser 级 Target.createTarget 新建 tab，返回其 page target 信息。"""
|
||||
ver = http_get("/json/version", host=host)
|
||||
b = CDP(ver["webSocketDebuggerUrl"])
|
||||
try:
|
||||
@@ -145,6 +158,12 @@ def create_tab(url, host=None):
|
||||
while time.time() < end:
|
||||
for t in http_get("/json", host=host):
|
||||
if t.get("id") == tid and t.get("webSocketDebuggerUrl"):
|
||||
return t["webSocketDebuggerUrl"]
|
||||
return t
|
||||
time.sleep(0.5)
|
||||
raise TimeoutError("等待新 tab websocket 超时")
|
||||
|
||||
|
||||
def create_tab(url, host=None):
|
||||
"""用 browser 级 Target.createTarget 新建 tab，返回其 page websocket。"""
|
||||
|
||||
return create_tab_info(url, host=host)["webSocketDebuggerUrl"]
|
||||
|
||||
+22
-2
@@ -5,7 +5,7 @@ import os
|
||||
import time
|
||||
from urllib.parse import urlparse
|
||||
|
||||
from .cdp import CDP, create_tab, find_product_tab, http_get
|
||||
from .cdp import CDP, close_tab, create_tab, create_tab_info, find_product_tab, http_get
|
||||
|
||||
|
||||
DEFAULT_REGION_HOST = "seller.shopee.tw"
|
||||
@@ -272,8 +272,14 @@ def open_product(account, item_id) -> CDP:
|
||||
item_id = str(item_id)
|
||||
url = _product_url(account, item_id)
|
||||
tab = find_product_tab(item_id, host=host)
|
||||
ws = tab["webSocketDebuggerUrl"] if tab else create_tab(url, host=host)
|
||||
created_by_app = tab is None
|
||||
if tab is None:
|
||||
tab = create_tab_info(url, host=host)
|
||||
ws = tab["webSocketDebuggerUrl"]
|
||||
cdp = CDP(ws)
|
||||
cdp.target_id = tab.get("id")
|
||||
cdp.created_by_app = created_by_app
|
||||
cdp.cdp_host = host
|
||||
_ensure_page_domains(cdp)
|
||||
try:
|
||||
cdp.send("Page.bringToFront")
|
||||
@@ -338,7 +344,21 @@ def collect(account, task) -> dict:
|
||||
"old_cover_path": old_cover_path,
|
||||
}
|
||||
finally:
|
||||
_close_collected_product(cdp)
|
||||
|
||||
|
||||
def _close_collected_product(cdp):
|
||||
target_id = getattr(cdp, "target_id", None)
|
||||
created_by_app = bool(getattr(cdp, "created_by_app", False))
|
||||
host = getattr(cdp, "cdp_host", None)
|
||||
try:
|
||||
cdp.close()
|
||||
finally:
|
||||
if created_by_app and target_id:
|
||||
try:
|
||||
close_tab(target_id, host=host)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
|
||||
def change_title(cdp, new_title) -> dict:
|
||||
|
||||
@@ -286,6 +286,7 @@ images/<slug>/<item_id>_new.<ext> # AI 生成的新封面
|
||||
### 6.1 采集（① Tab，只读）
|
||||
|
||||
- 用账号 Chrome 打开商品页，等就绪，读旧标题（标题输入框 value）。
|
||||
- `open_product` 先复用已打开的同商品 tab；没有才新建商品编辑页 tab。采集完成后只关闭本次程序自动新建的商品 tab，不关闭用户原本已经打开的 tab。`CDP.close()` 只断开 WebSocket 控制连接，不等于关闭浏览器 tab。
|
||||
- 旧封面：取第一张 itembox 的 `img.src`（CDN 链接），下载到 `images/<slug>/<item_id>_old`。
|
||||
- 写 `old_title/old_cover_path`、stage=collected；批量回写 Excel 旧字段。
|
||||
- 采集任务结束且本轮有成功采集行时，自动触发当前批次旧字段回写；原 Excel 被锁时不影响 SQLite 结果，提示关闭后重试，并保留手动「回写旧数据到 Excel」入口。
|
||||
@@ -329,6 +330,7 @@ images/<slug>/<item_id>_new.<ext> # AI 生成的新封面
|
||||
| Chrome 启动参数 | 全关后带 `--remote-debugging-port=<port> --remote-allow-origins=* --user-data-dir=<dir>`；缺 allow-origins 则 WebSocket 403 |
|
||||
| 代理干扰 | 清除 `*_proxy`（requests `trust_env=False`），否则连本地 CDP 超时 |
|
||||
| WebSocket Origin | `websocket-client` `suppress_origin=True` |
|
||||
| 关闭连接 vs 关闭 tab | `CDP.close()` 只关闭 WebSocket；需要关闭浏览器页面时必须调用浏览器 target 关闭接口。采集只关闭本轮自动新建的商品页，复用的用户已有 tab 不关闭 |
|
||||
| SPA 就绪 | 不用 load 事件；轮询“标题输入框 + 图片 itembox + 上传输入框”三者都在 |
|
||||
| 标题输入框 | XPath `//input[@class='eds-input__input' and string-length(@modelvalue)>24]` |
|
||||
| 写标题 | 原生 setter + 派发 `input`/`change`；`value`==`modelvalue`==新值 |
|
||||
|
||||
@@ -54,6 +54,7 @@
|
||||
| T-204 | 回写旧字段到原 Excel（含文件锁处理） | T-203, T-201 | `excel.write_back()` 按源文件/工作表/行号回写旧标题、旧封面；`export_copy()` 另存副本；Tab① 用 `WriteBackWorker` 后台回写，文件被占用时提示关闭后重试 | DONE |
|
||||
| T-204b | 采集完成后自动回写旧字段到 Excel | T-204 | `CollectWorker` 完成后自动触发 `excel.write_back()` 回写当前批次旧字段；成功时状态栏/日志提示“已回写”；原文件被锁时不影响 SQLite，提示关闭后点「回写旧数据到 Excel」手动重试或另存副本 | DONE |
|
||||
| T-205 | 首次未配账号 / Chrome 未启动 / 未登录的引导保护 | T-105, T-203 | 无账号、匹配账号未启动 CDP 端口或未登录时，① 执行按钮禁用或采集前汇总提示，并引导去④；可提供“打开账号管理/启动登录”入口，但不无提示批量启动所有账号 Chrome | DONE |
|
||||
| T-205b | 采集后关闭程序自动新建的商品编辑页 tab | T-203 | `open_product` 区分复用旧 tab 与本次新建 tab；`CDP.close()` 仍只断开连接；采集完成后只关闭本次自动新建的商品页，不关闭用户原本打开的 tab；失败/异常也执行清理；③ 更新流程暂不自动关页，失败时保留现场便于排查 | DONE |
|
||||
|
||||
## Phase 3 · AI 生成（②）
|
||||
|
||||
|
||||
+12
-3
@@ -191,8 +191,9 @@ is_running(port, host="127.0.0.1") -> bool
|
||||
|
||||
```python
|
||||
CDP_HOST: str
|
||||
http_get(path, host=None); find_product_tab(item_id, host=None); create_tab(url, host=None)
|
||||
class CDP: send/ev/val/object_id/drag/close # suppress_origin、trust_env=False
|
||||
http_get(path, host=None); find_product_tab(item_id, host=None); create_tab_info(url, host=None); create_tab(url, host=None)
|
||||
close_tab(target_id, host=None) -> bool # 关闭浏览器 target/page；区别于 CDP.close()
|
||||
class CDP: send/ev/val/object_id/drag/close # close 只断开 WebSocket；suppress_origin、trust_env=False
|
||||
```
|
||||
|
||||
## editor 模块（`app/editor.py`，已建，重构自现有脚本）
|
||||
@@ -200,7 +201,7 @@ class CDP: send/ev/val/object_id/drag/close # suppress_origin、trust_env=Fals
|
||||
```python
|
||||
login_status(account, timeout=8) -> dict # {logged_in, reason, url, host, cookie_names}
|
||||
is_logged_in(account) -> bool # login_status(...).logged_in；重定向登录页或缺 SPC_ST/SPC_U → False
|
||||
open_product(account, item_id) -> CDP # 连端口、导航商品页、等就绪
|
||||
open_product(account, item_id) -> CDP # 连端口、导航商品页、等就绪；标记该 tab 是否本轮自动新建
|
||||
|
||||
# 采集（只读）
|
||||
read_title(cdp) -> str
|
||||
@@ -218,6 +219,13 @@ apply_task(account, task) -> dict # 对已生成任务：换标题+
|
||||
|
||||
`login_status()` 不自动登录；无 Shopee tab 时打开卖家中心根地址 `https://<region_host>/`（默认 `https://seller.shopee.tw/`）用于检测/人工登录。判断规则：最终 URL 是登录页 → `LOGIN_PAGE`；缺少 `SPC_ST`/`SPC_U` → `NO_SESSION_COOKIE`；有会话 Cookie → 已登录。
|
||||
|
||||
采集 tab 生命周期：
|
||||
|
||||
- `CDP.close()` 只断开当前 websocket 控制连接，不关闭 Chrome 页面。
|
||||
- `open_product()` 若复用已存在商品 tab，则标记为用户已有页面；若调用 `create_tab()` 新建，则记录 target id。
|
||||
- `collect()` 结束时只关闭本轮自动新建的商品编辑页 tab；用户原本打开的商品 tab 不关闭。
|
||||
- ③ 更新流程暂不自动关闭商品页：成功后是否关闭留待后续体验任务，失败时保留现场便于排查。
|
||||
|
||||
## ai 模块（`app/ai.py`，已建，外部 AI，通用 HTTP）
|
||||
|
||||
```python
|
||||
@@ -314,6 +322,7 @@ TAB_STYLE: str # 顶层 Tab 栏防误点样式：
|
||||
- 账号列优先显示匹配到的 `accounts.account_name`；未匹配账号时保留 Excel 输入账号名。
|
||||
- 别名未匹配 `accounts.alias` 时列表阶段列显示“略过”；点击「采集旧标题/旧封面」后由 `CollectWorker` 逐条写库为 `skipped`，原因 `别名未匹配账号`。
|
||||
- 「采集旧标题/旧封面」通过 `CollectWorker` 后台执行，只处理 `stage=imported` 的任务；采集前先做账号就绪预检。无账号、当前批次匹配账号未启动 CDP 端口或未登录时，返回 `blocked=True`，GUI 弹窗汇总并跳转/引导去④账号管理，不进入逐条采集、不写 skipped/failed。预检通过后，已匹配任务调用 `editor.collect()` 下载旧封面到 `image_dir/<slug>/<item_id>_old.jpg` 并 `db.set_collected()`；别名未匹配任务仍逐条 `mark_skipped`；单条失败 `mark_failed(..., "collect", error)` 后继续。采集完成且本轮成功采集数量大于 0 时，自动触发当前批次旧字段回写；锁文件失败时只提示，不回滚 SQLite。
|
||||
- 采集打开商品页时，若本轮自动新建 tab，采集完成/失败后会关闭该 tab；若复用用户已打开的商品页，只断开 CDP 连接不关闭页面。
|
||||
- 「停止」调用 worker 的协作式 `cancel()`，已开始的单条跑到安全边界后结束。
|
||||
- 「回写旧数据到 Excel」通过 `WriteBackWorker` 后台调用 `excel.write_back()`，把已采集旧标题/旧封面路径按原 Excel 行定位写回；该按钮主要作为自动回写失败后的手动重试入口。原文件被占用时弹窗提示关闭后重试，SQLite 采集结果不回滚。
|
||||
- 采集前由 `CollectWorker` 做账号就绪预检：无账号、当前批次匹配账号未启动 CDP 端口或未登录时，返回 `blocked=True`，GUI 弹窗汇总并跳转/引导去④账号管理；不无提示批量启动所有账号 Chrome。匹配账号未登录属于预检阻断，不是逐条 skipped。
|
||||
|
||||
@@ -6,10 +6,10 @@
|
||||
## 当前快照
|
||||
|
||||
- 日期：2026-06-27
|
||||
- 阶段：V0 单账号 CDP 流程已验证；V1 已完成 T-000 正式代码包结构、T-001 `app/editor.py` 模块化、T-002 `app/appconfig.py` 应用配置、T-003 SQLite 持久化地基、T-004 本地数据忽略规则、T-005 AI 模型清单后端、T-006 单元测试基座、T-101 账号 user-data-dir 工具、T-102 Chrome 启动器、T-103 登录保活与检测、T-104 PySide6 主窗口骨架、T-104b PySide6 worker 基座、T-105 Tab④ 账号管理、T-106 账号快捷方式、T-201 Excel 导入入库、T-202 Tab① 任务列表与导入按钮、T-202b Tab① 导入汇总栏、T-203 采集旧标题旧封面、T-204 回写旧字段到原 Excel、T-204b 采集完成自动回写、T-205 首次未配账号/Chrome 未启动/未登录引导保护、T-301 AI 生成接口、T-302 Tab② AI 生成布局与任务列表、T-302p 提示词管理、T-303 Tab② 开始生成/停止/进度。
|
||||
- 阶段：V0 单账号 CDP 流程已验证；V1 已完成 T-000 正式代码包结构、T-001 `app/editor.py` 模块化、T-002 `app/appconfig.py` 应用配置、T-003 SQLite 持久化地基、T-004 本地数据忽略规则、T-005 AI 模型清单后端、T-006 单元测试基座、T-101 账号 user-data-dir 工具、T-102 Chrome 启动器、T-103 登录保活与检测、T-104 PySide6 主窗口骨架、T-104b PySide6 worker 基座、T-105 Tab④ 账号管理、T-106 账号快捷方式、T-201 Excel 导入入库、T-202 Tab① 任务列表与导入按钮、T-202b Tab① 导入汇总栏、T-203 采集旧标题旧封面、T-204 回写旧字段到原 Excel、T-204b 采集完成自动回写、T-205 首次未配账号/Chrome 未启动/未登录引导保护、T-205b 采集后关闭自动新建商品页 tab、T-301 AI 生成接口、T-302 Tab② AI 生成布局与任务列表、T-302p 提示词管理、T-303 Tab② 开始生成/停止/进度。
|
||||
- 技术栈：Python 3.10+，自研 CDP（websocket-client + requests），SQLite（sqlite3）+ `config.json` + openpyxl + AI（`config/ai_models.json` 通用 HTTP，chat JSON / images_edits），GUI PySide6 5 Tab（已定）。
|
||||
- 生产代码：已建立 `app/` 包 + 根入口 `main.py`；`app/cdp.py` 为已验证 CDP 底座；`app/editor.py` 已封装登录状态检测、标题/封面/采集/更新按钮能力；`app/appconfig.py` 已实现 `config.json` 默认值/读写/更新、AI 参数与端口读取，以及 `config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接；`app/ai.py` 已实现 `gen_title()`/`gen_cover()`/`generate_batch()`，按默认文本/图片模型发起通用 HTTP 调用，支持重试、错误脱敏、图片 URL/base64 解析、resolution resize、jpg_quality 保存、先并发标题再并发封面、逐条 `set_generated`、失败 `mark_failed` 与停止取消未开始项；`app/prompts.py` 已实现标题提示词读写、封面模板 CRUD 与变量替换；`app/db.py` 已实现 SQLite schema、连接 PRAGMA、批次/账号/任务与阶段写库函数；`app/excel.py` 已实现多 Excel 输入列解析、整文件列校验、脏行统计跳过、导入批次与任务入库、别名匹配统计、旧标题/旧封面路径回写原 Excel 与另存副本；`app/config.py` 已实现账号 slug 与 user-data-dir 创建；`app/accounts.py` 已实现账号 CRUD 服务、端口默认分配、启动登录、检测登录、生成快捷方式；`app/chrome.py` 已实现 Chrome 参数拼装、启动、CDP 端口探测、PowerShell `.lnk` 快捷方式生成；`app/gui.py` 已实现 PySide6 `MainWindow`、五 Tab、顶部 Tab 栏防误点样式、① 导入采集的 Excel 导入按钮/导入汇总栏/QTableView 任务列表/未匹配筛选与略过标记/采集旧标题旧封面 worker/采集前账号就绪预检与④引导/采集完成自动回写/旧数据回写重试按钮与 worker、② AI生成左右布局/标题与封面提示词管理/批次店铺状态筛选/任务列表/变量预览/开始生成/停止/进度/双击新旧封面预览与 `GenerateWorker`、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏；`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。
|
||||
- 测试：`tests/` 已建立；T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`，当前覆盖 appconfig/db/config/accounts/chrome 启动与快捷方式/editor 登录检测/excel 导入/ai 标题与封面 HTTP 解析/`generate_batch` 正常、失败与停止/prompts 读写与渲染/gui ① 导入采集/gui ② AI生成布局筛选、提示词管理、生成 worker 与双击预览/gui ④ 账号管理/worker signal 与线程包装；CDP/Shopee 改动仍需测试商品手动验证。
|
||||
- 生产代码：已建立 `app/` 包 + 根入口 `main.py`；`app/cdp.py` 为已验证 CDP 底座，已区分 `CDP.close()` 断开 WebSocket 与 `close_tab()` 关闭浏览器 target；`app/editor.py` 已封装登录状态检测、标题/封面/采集/更新按钮能力，并在采集结束后只关闭本轮自动新建的商品编辑页 tab、保留用户已有 tab；`app/appconfig.py` 已实现 `config.json` 默认值/读写/更新、AI 参数与端口读取，以及 `config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接；`app/ai.py` 已实现 `gen_title()`/`gen_cover()`/`generate_batch()`，按默认文本/图片模型发起通用 HTTP 调用，支持重试、错误脱敏、图片 URL/base64 解析、resolution resize、jpg_quality 保存、先并发标题再并发封面、逐条 `set_generated`、失败 `mark_failed` 与停止取消未开始项；`app/prompts.py` 已实现标题提示词读写、封面模板 CRUD 与变量替换；`app/db.py` 已实现 SQLite schema、连接 PRAGMA、批次/账号/任务与阶段写库函数；`app/excel.py` 已实现多 Excel 输入列解析、整文件列校验、脏行统计跳过、导入批次与任务入库、别名匹配统计、旧标题/旧封面路径回写原 Excel 与另存副本；`app/config.py` 已实现账号 slug 与 user-data-dir 创建；`app/accounts.py` 已实现账号 CRUD 服务、端口默认分配、启动登录、检测登录、生成快捷方式；`app/chrome.py` 已实现 Chrome 参数拼装、启动、CDP 端口探测、PowerShell `.lnk` 快捷方式生成；`app/gui.py` 已实现 PySide6 `MainWindow`、五 Tab、顶部 Tab 栏防误点样式、① 导入采集的 Excel 导入按钮/导入汇总栏/QTableView 任务列表/未匹配筛选与略过标记/采集旧标题旧封面 worker/采集前账号就绪预检与④引导/采集完成自动回写/旧数据回写重试按钮与 worker、② AI生成左右布局/标题与封面提示词管理/批次店铺状态筛选/任务列表/变量预览/开始生成/停止/进度/双击新旧封面预览与 `GenerateWorker`、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏；`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。
|
||||
- 测试：`tests/` 已建立；T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`，当前覆盖 appconfig/db/config/accounts/chrome 启动与快捷方式/editor 登录检测与商品 tab 生命周期/excel 导入/ai 标题与封面 HTTP 解析/`generate_batch` 正常、失败与停止/prompts 读写与渲染/gui ① 导入采集/gui ② AI生成布局筛选、提示词管理、生成 worker 与双击预览/gui ④ 账号管理/worker signal 与线程包装；CDP/Shopee 改动仍需测试商品手动验证。
|
||||
- 数据：`config.json`、`config/ai_models.json`、`cmshopee.db`、`chrome_user_data_dir/`、`images/` 已由 `.gitignore` 排除；运营填写后的 Excel 业务文件默认忽略，标准空模板 `shopee待处理任务模板.xlsx` 可提交；`app/appconfig.py` 首次读取缺失的 `config.json` 时会在本地写默认配置，`app/db.py` 调用 `init_db()` 时会在本地创建 SQLite DB。
|
||||
|
||||
## 既定设计要点（文档已定）
|
||||
@@ -28,14 +28,14 @@
|
||||
| 路径 | 状态 | 说明 |
|
||||
| --- | --- | --- |
|
||||
| `docs/` | 已有 | 本 harness coding 文档集合 |
|
||||
| `app/cdp.py` | 已有 | CDP 底座：连接/找 tab/开 tab/执行 JS/拖拽 |
|
||||
| `app/cdp.py` | 已有 | CDP 底座：连接/找 tab/开 tab/关闭 target/执行 JS/拖拽；`CDP.close()` 只断开 WebSocket，`close_tab()` 才关闭浏览器页面 |
|
||||
| `prototypes/` | 已有 | 已验证原型/探查脚本（demo/set_title/set_cover/get_title/cookies/inspect_images/grab/1.py），保留作人工回归与探查参考；见 `prototypes/README.md` |
|
||||
| `chrome-remote-debug-lan.md` | 已有 | WSL→Windows CDP 转发排查记录 |
|
||||
| `app/__init__.py` / `app/__main__.py` / `main.py` | 已有 | 正式包与启动入口；`python main.py` / `python -m app` 可运行占位入口 |
|
||||
| `app/gui.py` | 已有 | T-104/T-105/T-106/T-202/T-202b/T-203/T-204/T-204b/T-205/T-302/T-302p/T-303 产出：PySide6 `QMainWindow` + 五 Tab；顶部 Tab 栏防误点样式；① 导入采集导入按钮、导入汇总栏、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker、采集前账号就绪预检与④引导、采集完成自动回写与手动重试；② AI生成左右布局、提示词管理、筛选栏、任务列表、开始生成/停止/进度、双击新旧封面预览与 `GenerateWorker`；④ 账号管理表格、账号弹窗、启动登录、检测登录、快捷方式 |
|
||||
| `app/workers.py` | 已有 | T-104b 产出：`BaseWorker` + 通用 signals + 取消标记 + `run_worker()` QThread 包装 |
|
||||
| `app/accounts.py` | 已有 | T-105/T-106 产出：账号 CRUD 服务、目录创建、端口分配、启动登录、检测登录、快捷方式 |
|
||||
| `app/editor.py` | 已有 | T-001/T-103 产出：登录状态检测、打开商品页、读/写标题、读/下载封面、上传拖封面、更新按钮、apply_task |
|
||||
| `app/editor.py` | 已有 | T-001/T-103/T-205b 产出：登录状态检测、打开商品页、读/写标题、读/下载封面、采集后关闭自动新建商品页 tab、上传拖封面、更新按钮、apply_task |
|
||||
| `app/appconfig.py` | 已有 | T-002 产出：`config.json` 默认值、读写、更新、路径/端口/AI 参数读取；拒绝敏感字段写入 |
|
||||
| `app/ai.py` | 已有 | T-301/T-303 产出：`gen_title()`/`gen_cover()`/`generate_batch()`；读取默认模型；通用 HTTP 调用；失败重试；错误脱敏；封面按 resolution/jpg_quality 保存；批量生成先标题后封面、进度回调、逐条落库、失败标记、停止取消未开始项 |
|
||||
| `app/prompts.py` | 已有 | T-302p 产出：标题提示词读写、封面模板列表/读取/保存/重命名/删除、变量替换 |
|
||||
@@ -60,7 +60,7 @@
|
||||
|
||||
任务状态以 [`06-tasks.md`](06-tasks.md) 为准，历史记录见 [`../progress.md`](../progress.md)。
|
||||
|
||||
- 已完成：T-000（正式代码包结构）、T-001（`app/editor.py` 模块化）、T-002（`app/appconfig.py` + `config.json`）、T-003（SQLite 建表）、T-004（本地数据 gitignore）、T-005（AI 模型清单后端）、T-006（单元测试基座）、T-101（账号 slug/user-data-dir）、T-102（Chrome 启动器）、T-103（登录保活与检测）、T-104（PySide6 五 Tab 主窗口骨架）、T-104b（PySide6 worker 基座）、T-105（Tab④ 账号管理）、T-106（账号快捷方式）、T-201（Excel 导入：解析多文件输入列入库）、T-202（Tab① 任务列表 + 导入按钮 + 别名匹配标记）、T-202b（Tab① 导入汇总栏）、T-203（采集旧标题+旧封面）、T-204（回写旧字段到原 Excel）、T-204b（采集完成自动回写旧字段）、T-205（首次未配账号 / Chrome 未启动 / 未登录引导保护）、T-301（AI 生成接口）、T-302（Tab② 左右布局与任务列表）、T-302p（提示词管理）、T-303（Tab② 开始生成 + 停止 + 进度）。
|
||||
- 已完成：T-000（正式代码包结构）、T-001（`app/editor.py` 模块化）、T-002（`app/appconfig.py` + `config.json`）、T-003（SQLite 建表）、T-004（本地数据 gitignore）、T-005（AI 模型清单后端）、T-006（单元测试基座）、T-101（账号 slug/user-data-dir）、T-102（Chrome 启动器）、T-103（登录保活与检测）、T-104（PySide6 五 Tab 主窗口骨架）、T-104b（PySide6 worker 基座）、T-105（Tab④ 账号管理）、T-106（账号快捷方式）、T-201（Excel 导入：解析多文件输入列入库）、T-202（Tab① 任务列表 + 导入按钮 + 别名匹配标记）、T-202b（Tab① 导入汇总栏）、T-203（采集旧标题+旧封面）、T-204（回写旧字段到原 Excel）、T-204b（采集完成自动回写旧字段）、T-205（首次未配账号 / Chrome 未启动 / 未登录引导保护）、T-205b（采集后关闭自动新建商品页 tab）、T-301（AI 生成接口）、T-302（Tab② 左右布局与任务列表）、T-302p（提示词管理）、T-303（Tab② 开始生成 + 停止 + 进度）。
|
||||
- 正在进行：无。
|
||||
- 下一个可领取任务：**T-401（Tab③ 列出已生成任务 + 按批次/店铺/状态筛选 + 开始更新确认弹窗）**。
|
||||
|
||||
|
||||
+1
-1
@@ -52,7 +52,7 @@
|
||||
|
||||
- 导入：openpyxl 解析**输入列**（账号名/别名/商品id）入 SQLite。
|
||||
- **导入汇总栏**（导入后即时刷新，跑采集前的校验关口）：显示 文件数、解析行数（原始数据量）、有效/无效行、匹配账号行数（按账号细分）、未匹配行数。未匹配/无效数字标红可点，点击在列表筛出便于定位纠错。
|
||||
- 采集：用该账号已启动并登录的 Chrome 只读打开商品页，读旧标题、下载旧封面到本地图片目录，写 `old_title/old_cover_path`，stage=collected。
|
||||
- 采集：用该账号已启动并登录的 Chrome 只读打开商品页，读旧标题、下载旧封面到本地图片目录，写 `old_title/old_cover_path`，stage=collected。若程序为采集自动新建商品页 tab，采集结束后自动关闭；若复用用户原本打开的 tab，则不关闭。
|
||||
- 回写：采集完成后自动把旧标题/旧封面路径批量回写原 Excel；保留「回写旧数据到 Excel」作为手动重试入口（原文件被锁→提示关闭后重试/另存）。
|
||||
- 别名未匹配账号 / 账号未登录 → 该行 skipped 并记原因。
|
||||
|
||||
|
||||
@@ -509,3 +509,12 @@
|
||||
- 文档：`docs/06-tasks.md` 将 T-303 标为 DONE；同步 `docs/api.md`、`docs/routes.md`、`docs/current-state.md`，下一个可领取任务更新为 T-401。
|
||||
- 验证：`python -m unittest discover -s tests -p "test_ai.py"` 通过（6 tests）；`python -m unittest discover -s tests -p "test_gui.py"` 通过（22 tests）；`python -m compileall app main.py tests` 通过；`python -m unittest discover -s tests` 通过（73 tests）。
|
||||
- 注意：本轮仍为 mock AI 单测，未调用真实 AI 服务；真实生成需在 `config/ai_models.json` 配好可用 text/image 模型后做小样本实测。
|
||||
|
||||
## 【2026-06-27】T-205b 采集后关闭自动新建商品页 tab
|
||||
|
||||
- 状态：DONE
|
||||
- 背景：真机采集时发现程序为不同商品新建编辑页 tab，采集标题和封面后只断开 CDP 连接，不会关闭浏览器页面，批量采集后 Chrome 会堆积多个商品页。
|
||||
- 方案：区分“复用用户已有 tab”和“本轮程序自动新建 tab”。`CDP.close()` 仍只断开 WebSocket；新增 `close_tab(target_id)` 调 Chrome `/json/close/<targetId>` 关闭浏览器 target；`open_product()` 给 CDP 对象标记 `target_id/created_by_app/cdp_host`；`collect()` finally 中只关闭 `created_by_app=True` 的商品页，用户原本打开的 tab 不关闭。③ 更新流程暂不自动关页，失败时保留现场便于排查。
|
||||
- 变更：更新 `app/cdp.py`、`app/editor.py`；同步 `docs/06-tasks.md`、`docs/04-architecture.md`、`docs/api.md`、`docs/routes.md`、`docs/current-state.md`。
|
||||
- 测试：`tests/test_editor_login.py` 增加 `open_product()` 新/旧 tab 标记、`collect()` 关闭自动新建 tab、不关闭复用 tab 的覆盖。
|
||||
- 验证：`python -m unittest discover -s tests -p "test_editor_login.py"` 通过（9 tests）；`python -m compileall app main.py tests` 通过；`python -m unittest discover -s tests` 通过（77 tests）。
|
||||
|
||||
@@ -37,6 +37,20 @@ class FakeCDP:
|
||||
self.closed = True
|
||||
|
||||
|
||||
class FakeProductCDP:
|
||||
def __init__(self, ws):
|
||||
self.ws = ws
|
||||
self.closed = False
|
||||
self.sent = []
|
||||
|
||||
def send(self, method, params=None):
|
||||
self.sent.append((method, params or {}))
|
||||
return {}
|
||||
|
||||
def close(self):
|
||||
self.closed = True
|
||||
|
||||
|
||||
def cookie(name, domain=".shopee.tw"):
|
||||
return {"name": name, "domain": domain}
|
||||
|
||||
@@ -133,6 +147,103 @@ class EditorLoginTests(unittest.TestCase):
|
||||
):
|
||||
self.assertTrue(editor.is_logged_in({"debug_port": 9222}))
|
||||
|
||||
def test_open_product_marks_reused_existing_tab(self):
|
||||
fake = FakeProductCDP("ws-existing")
|
||||
with mock.patch(
|
||||
"app.editor.find_product_tab",
|
||||
return_value={"id": "target-existing", "webSocketDebuggerUrl": "ws-existing"},
|
||||
) as find_product_tab, mock.patch(
|
||||
"app.editor.create_tab_info",
|
||||
) as create_tab_info, mock.patch(
|
||||
"app.editor.CDP",
|
||||
return_value=fake,
|
||||
), mock.patch("app.editor._ensure_page_domains"), mock.patch("app.editor._wait_ready"):
|
||||
cdp = editor.open_product({"debug_port": 9222}, "51100639510")
|
||||
|
||||
self.assertIs(fake, cdp)
|
||||
self.assertEqual("target-existing", cdp.target_id)
|
||||
self.assertFalse(cdp.created_by_app)
|
||||
self.assertEqual("127.0.0.1:9222", cdp.cdp_host)
|
||||
find_product_tab.assert_called_once_with("51100639510", host="127.0.0.1:9222")
|
||||
create_tab_info.assert_not_called()
|
||||
self.assertIn(
|
||||
(
|
||||
"Page.navigate",
|
||||
{
|
||||
"url": (
|
||||
"https://seller.shopee.tw/portal/product/51100639510"
|
||||
"?pageEntry=product_list&ignore-html-cache=1"
|
||||
)
|
||||
},
|
||||
),
|
||||
fake.sent,
|
||||
)
|
||||
|
||||
def test_open_product_marks_auto_created_tab(self):
|
||||
fake = FakeProductCDP("ws-new")
|
||||
with mock.patch("app.editor.find_product_tab", return_value=None), mock.patch(
|
||||
"app.editor.create_tab_info",
|
||||
return_value={"id": "target-new", "webSocketDebuggerUrl": "ws-new"},
|
||||
) as create_tab_info, mock.patch(
|
||||
"app.editor.CDP",
|
||||
return_value=fake,
|
||||
), mock.patch("app.editor._ensure_page_domains"), mock.patch("app.editor._wait_ready"):
|
||||
cdp = editor.open_product({"debug_port": 9223}, "51100639511")
|
||||
|
||||
self.assertEqual("target-new", cdp.target_id)
|
||||
self.assertTrue(cdp.created_by_app)
|
||||
self.assertEqual("127.0.0.1:9223", cdp.cdp_host)
|
||||
create_tab_info.assert_called_once()
|
||||
|
||||
def test_collect_closes_only_auto_created_product_tab(self):
|
||||
cdp = FakeProductCDP("ws-new")
|
||||
cdp.target_id = "target-new"
|
||||
cdp.created_by_app = True
|
||||
cdp.cdp_host = "127.0.0.1:9222"
|
||||
|
||||
with mock.patch("app.editor.open_product", return_value=cdp), mock.patch(
|
||||
"app.editor.read_title",
|
||||
return_value="旧标题",
|
||||
), mock.patch(
|
||||
"app.editor.read_cover_src",
|
||||
return_value="https://down-ws-sg.vod.susercontent.com/cover.jpg",
|
||||
), mock.patch(
|
||||
"app.editor.download_cover",
|
||||
return_value="images/main/51100639510_old.jpg",
|
||||
), mock.patch("app.editor.close_tab", return_value=True) as close_tab:
|
||||
result = editor.collect(
|
||||
{"debug_port": 9222},
|
||||
{"item_id": "51100639510", "old_cover_path": "old.jpg"},
|
||||
)
|
||||
|
||||
self.assertEqual("旧标题", result["old_title"])
|
||||
self.assertTrue(cdp.closed)
|
||||
close_tab.assert_called_once_with("target-new", host="127.0.0.1:9222")
|
||||
|
||||
def test_collect_keeps_reused_product_tab_open(self):
|
||||
cdp = FakeProductCDP("ws-existing")
|
||||
cdp.target_id = "target-existing"
|
||||
cdp.created_by_app = False
|
||||
cdp.cdp_host = "127.0.0.1:9222"
|
||||
|
||||
with mock.patch("app.editor.open_product", return_value=cdp), mock.patch(
|
||||
"app.editor.read_title",
|
||||
return_value="旧标题",
|
||||
), mock.patch(
|
||||
"app.editor.read_cover_src",
|
||||
return_value="https://down-ws-sg.vod.susercontent.com/cover.jpg",
|
||||
), mock.patch(
|
||||
"app.editor.download_cover",
|
||||
return_value="images/main/51100639510_old.jpg",
|
||||
), mock.patch("app.editor.close_tab") as close_tab:
|
||||
editor.collect(
|
||||
{"debug_port": 9222},
|
||||
{"item_id": "51100639510", "old_cover_path": "old.jpg"},
|
||||
)
|
||||
|
||||
self.assertTrue(cdp.closed)
|
||||
close_tab.assert_not_called()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user