feat: 完成T-302 AI生成页布局
新增GenerateTab和GenerateTaskTableModel,将② AI生成从占位页替换为左右布局:左侧标题/封面提示词多行输入,右侧批次、店铺、状态筛选和任务列表。 任务列表展示店铺、商品ID、旧标题、新标题、状态;筛选只读SQLite,不调用AI、不写库,生成执行留给T-303,提示词管理留给T-302p。 补充GUI测试覆盖Tab②挂载、提示词输入区、任务列表展示与批次/店铺/状态筛选;同步任务看板、API、routes、current-state和progress。
This commit is contained in:
+273
@@ -10,6 +10,7 @@ try:
|
|||||||
from PySide6.QtWidgets import (
|
from PySide6.QtWidgets import (
|
||||||
QAbstractItemView,
|
QAbstractItemView,
|
||||||
QApplication,
|
QApplication,
|
||||||
|
QComboBox,
|
||||||
QDialog,
|
QDialog,
|
||||||
QDialogButtonBox,
|
QDialogButtonBox,
|
||||||
QFileDialog,
|
QFileDialog,
|
||||||
@@ -22,6 +23,7 @@ try:
|
|||||||
QMessageBox,
|
QMessageBox,
|
||||||
QPlainTextEdit,
|
QPlainTextEdit,
|
||||||
QPushButton,
|
QPushButton,
|
||||||
|
QSplitter,
|
||||||
QTableView,
|
QTableView,
|
||||||
QSpinBox,
|
QSpinBox,
|
||||||
QTableWidget,
|
QTableWidget,
|
||||||
@@ -193,6 +195,271 @@ if QT_IMPORT_ERROR is None:
|
|||||||
return values[column] if 0 <= column < len(values) else None
|
return values[column] if 0 <= column < len(values) else None
|
||||||
|
|
||||||
|
|
||||||
|
class GenerateTaskTableModel(QAbstractTableModel):
|
||||||
|
"""Table model for Tab 2 generation candidates."""
|
||||||
|
|
||||||
|
HEADERS = ["店铺", "商品ID", "旧标题", "新标题", "状态"]
|
||||||
|
|
||||||
|
STATUS_TEXT = {
|
||||||
|
"running": "处理中",
|
||||||
|
"failed": "失败",
|
||||||
|
"skipped": "略过",
|
||||||
|
"cancelled": "已取消",
|
||||||
|
}
|
||||||
|
|
||||||
|
STAGE_TEXT = {
|
||||||
|
"imported": "未采集",
|
||||||
|
"collected": "待生成",
|
||||||
|
"generated": "已生成",
|
||||||
|
"applied": "已更新",
|
||||||
|
}
|
||||||
|
|
||||||
|
def __init__(self, parent=None):
|
||||||
|
super().__init__(parent)
|
||||||
|
self.tasks = []
|
||||||
|
self.account_by_alias = {}
|
||||||
|
|
||||||
|
def set_tasks(self, tasks, accounts):
|
||||||
|
self.beginResetModel()
|
||||||
|
self.tasks = list(tasks)
|
||||||
|
self.account_by_alias = {
|
||||||
|
str(account.alias).strip(): account
|
||||||
|
for account in accounts
|
||||||
|
if str(account.alias).strip()
|
||||||
|
}
|
||||||
|
self.endResetModel()
|
||||||
|
|
||||||
|
def rowCount(self, parent=QModelIndex()):
|
||||||
|
return 0 if parent.isValid() else len(self.tasks)
|
||||||
|
|
||||||
|
def columnCount(self, parent=QModelIndex()):
|
||||||
|
return 0 if parent.isValid() else len(self.HEADERS)
|
||||||
|
|
||||||
|
def headerData(self, section, orientation, role=Qt.DisplayRole):
|
||||||
|
if role != Qt.DisplayRole:
|
||||||
|
return None
|
||||||
|
if orientation == Qt.Horizontal and 0 <= section < len(self.HEADERS):
|
||||||
|
return self.HEADERS[section]
|
||||||
|
return section + 1 if orientation == Qt.Vertical else None
|
||||||
|
|
||||||
|
def data(self, index, role=Qt.DisplayRole):
|
||||||
|
if not index.isValid():
|
||||||
|
return None
|
||||||
|
task = self.tasks[index.row()]
|
||||||
|
if role == Qt.DisplayRole:
|
||||||
|
return self._display_value(task, index.column())
|
||||||
|
if role == Qt.ToolTipRole and task.last_error:
|
||||||
|
return task.last_error
|
||||||
|
return None
|
||||||
|
|
||||||
|
def flags(self, index):
|
||||||
|
if not index.isValid():
|
||||||
|
return Qt.NoItemFlags
|
||||||
|
return Qt.ItemIsEnabled | Qt.ItemIsSelectable
|
||||||
|
|
||||||
|
def _account_name(self, task):
|
||||||
|
account = self.account_by_alias.get(str(task.alias).strip())
|
||||||
|
if account is not None:
|
||||||
|
return account.account_name
|
||||||
|
return task.account_name or task.alias or ""
|
||||||
|
|
||||||
|
def _status_text(self, task):
|
||||||
|
if task.status in self.STATUS_TEXT and task.status != "pending":
|
||||||
|
return self.STATUS_TEXT[task.status]
|
||||||
|
return self.STAGE_TEXT.get(task.stage, task.stage)
|
||||||
|
|
||||||
|
def _display_value(self, task, column):
|
||||||
|
values = [
|
||||||
|
self._account_name(task),
|
||||||
|
task.item_id,
|
||||||
|
task.old_title or "",
|
||||||
|
task.new_title or "",
|
||||||
|
self._status_text(task),
|
||||||
|
]
|
||||||
|
return values[column] if 0 <= column < len(values) else None
|
||||||
|
|
||||||
|
|
||||||
|
class GenerateTab(QWidget):
|
||||||
|
"""Tab 2: prompt area plus generation task filters/list."""
|
||||||
|
|
||||||
|
STATUS_FILTERS = [
|
||||||
|
("全部状态", "all"),
|
||||||
|
("待生成", "to_generate"),
|
||||||
|
("已生成", "generated"),
|
||||||
|
("失败", "failed"),
|
||||||
|
("略过", "skipped"),
|
||||||
|
("已更新", "applied"),
|
||||||
|
]
|
||||||
|
|
||||||
|
def __init__(self, parent=None, db_path=None, config=None, status_callback=None):
|
||||||
|
super().__init__(parent)
|
||||||
|
self.config = appconfig.load_config() if config is None else config
|
||||||
|
self.db_path = _database_path(db_path, self.config)
|
||||||
|
self.status_callback = status_callback
|
||||||
|
|
||||||
|
self.title_prompt_edit = QPlainTextEdit()
|
||||||
|
self.title_prompt_edit.setObjectName("titlePromptEdit")
|
||||||
|
self.title_prompt_edit.setPlaceholderText("标题提示词")
|
||||||
|
self.cover_prompt_edit = QPlainTextEdit()
|
||||||
|
self.cover_prompt_edit.setObjectName("coverPromptEdit")
|
||||||
|
self.cover_prompt_edit.setPlaceholderText("封面提示词")
|
||||||
|
|
||||||
|
left_panel = QWidget()
|
||||||
|
left_layout = QVBoxLayout(left_panel)
|
||||||
|
left_layout.setContentsMargins(0, 0, 12, 0)
|
||||||
|
left_layout.addWidget(QLabel("标题提示词"))
|
||||||
|
left_layout.addWidget(self.title_prompt_edit, 1)
|
||||||
|
left_layout.addWidget(QLabel("封面提示词"))
|
||||||
|
left_layout.addWidget(self.cover_prompt_edit, 2)
|
||||||
|
|
||||||
|
self.batch_filter = QComboBox()
|
||||||
|
self.batch_filter.setObjectName("batchFilter")
|
||||||
|
self.shop_filter = QComboBox()
|
||||||
|
self.shop_filter.setObjectName("shopFilter")
|
||||||
|
self.status_filter = QComboBox()
|
||||||
|
self.status_filter.setObjectName("statusFilter")
|
||||||
|
for label, value in self.STATUS_FILTERS:
|
||||||
|
self.status_filter.addItem(label, value)
|
||||||
|
self.refresh_button = QPushButton("刷新")
|
||||||
|
|
||||||
|
filter_layout = QHBoxLayout()
|
||||||
|
filter_layout.addWidget(QLabel("批次"))
|
||||||
|
filter_layout.addWidget(self.batch_filter, 2)
|
||||||
|
filter_layout.addWidget(QLabel("店铺"))
|
||||||
|
filter_layout.addWidget(self.shop_filter, 1)
|
||||||
|
filter_layout.addWidget(QLabel("状态"))
|
||||||
|
filter_layout.addWidget(self.status_filter, 1)
|
||||||
|
filter_layout.addWidget(self.refresh_button)
|
||||||
|
|
||||||
|
self.summary_label = QLabel("任务 0 条")
|
||||||
|
self.task_table = QTableView()
|
||||||
|
self.model = GenerateTaskTableModel(self.task_table)
|
||||||
|
self.task_table.setModel(self.model)
|
||||||
|
self.task_table.setSelectionBehavior(QAbstractItemView.SelectRows)
|
||||||
|
self.task_table.setSelectionMode(QAbstractItemView.SingleSelection)
|
||||||
|
self.task_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
|
||||||
|
self.task_table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
|
||||||
|
self.task_table.verticalHeader().setVisible(False)
|
||||||
|
|
||||||
|
right_panel = QWidget()
|
||||||
|
right_layout = QVBoxLayout(right_panel)
|
||||||
|
right_layout.setContentsMargins(12, 0, 0, 0)
|
||||||
|
right_layout.addLayout(filter_layout)
|
||||||
|
right_layout.addWidget(self.summary_label)
|
||||||
|
right_layout.addWidget(self.task_table, 1)
|
||||||
|
|
||||||
|
self.splitter = QSplitter(Qt.Horizontal)
|
||||||
|
self.splitter.addWidget(left_panel)
|
||||||
|
self.splitter.addWidget(right_panel)
|
||||||
|
self.splitter.setStretchFactor(0, 1)
|
||||||
|
self.splitter.setStretchFactor(1, 3)
|
||||||
|
self.splitter.setSizes([280, 860])
|
||||||
|
|
||||||
|
layout = QVBoxLayout(self)
|
||||||
|
layout.setContentsMargins(18, 18, 18, 18)
|
||||||
|
layout.addWidget(self.splitter, 1)
|
||||||
|
|
||||||
|
self.batch_filter.currentIndexChanged.connect(self.refresh_tasks)
|
||||||
|
self.shop_filter.currentIndexChanged.connect(self.refresh_tasks)
|
||||||
|
self.status_filter.currentIndexChanged.connect(self.refresh_tasks)
|
||||||
|
self.refresh_button.clicked.connect(self.refresh_tasks)
|
||||||
|
|
||||||
|
self.refresh_tasks()
|
||||||
|
|
||||||
|
def _set_status(self, message):
|
||||||
|
if self.status_callback is not None:
|
||||||
|
self.status_callback(message)
|
||||||
|
|
||||||
|
def refresh_tasks(self, checked=False):
|
||||||
|
try:
|
||||||
|
db.init_db(self.db_path)
|
||||||
|
batches = db.list_batches(path=self.db_path)
|
||||||
|
accounts_rows = accounts.list_accounts(path=self.db_path, config=self.config)
|
||||||
|
selected_batch = self.batch_filter.currentData()
|
||||||
|
selected_shop = self.shop_filter.currentData()
|
||||||
|
selected_status = self.status_filter.currentData() or "all"
|
||||||
|
self._populate_batch_filter(batches, selected_batch)
|
||||||
|
selected_batch = self.batch_filter.currentData()
|
||||||
|
batch_tasks = db.list_tasks(batch_id=selected_batch, path=self.db_path)
|
||||||
|
self._populate_shop_filter(batch_tasks, accounts_rows, selected_shop)
|
||||||
|
selected_shop = self.shop_filter.currentData()
|
||||||
|
filtered_tasks = [
|
||||||
|
task for task in batch_tasks
|
||||||
|
if self._matches_shop(task, selected_shop)
|
||||||
|
and self._matches_status(task, selected_status)
|
||||||
|
]
|
||||||
|
except Exception as exc:
|
||||||
|
self.model.set_tasks([], [])
|
||||||
|
self.summary_label.setText("任务读取失败")
|
||||||
|
self._set_status(f"AI 生成任务读取失败:{exc}")
|
||||||
|
return
|
||||||
|
self.model.set_tasks(filtered_tasks, accounts_rows)
|
||||||
|
self.summary_label.setText(
|
||||||
|
f"任务 {len(filtered_tasks)}/{len(batch_tasks)} 条"
|
||||||
|
)
|
||||||
|
|
||||||
|
def _populate_batch_filter(self, batches, selected_batch):
|
||||||
|
previous = selected_batch if selected_batch in {batch.id for batch in batches} else None
|
||||||
|
self.batch_filter.blockSignals(True)
|
||||||
|
self.batch_filter.clear()
|
||||||
|
self.batch_filter.addItem("全部批次", None)
|
||||||
|
for batch in batches:
|
||||||
|
self.batch_filter.addItem(self._batch_label(batch), batch.id)
|
||||||
|
index = self.batch_filter.findData(previous)
|
||||||
|
self.batch_filter.setCurrentIndex(index if index >= 0 else 0)
|
||||||
|
self.batch_filter.blockSignals(False)
|
||||||
|
|
||||||
|
def _populate_shop_filter(self, tasks, account_rows, selected_shop):
|
||||||
|
account_by_alias = {
|
||||||
|
str(account.alias).strip(): account
|
||||||
|
for account in account_rows
|
||||||
|
if str(account.alias).strip()
|
||||||
|
}
|
||||||
|
aliases = []
|
||||||
|
for task in tasks:
|
||||||
|
alias = str(task.alias).strip()
|
||||||
|
if alias and alias not in aliases:
|
||||||
|
aliases.append(alias)
|
||||||
|
previous = selected_shop if selected_shop in aliases else None
|
||||||
|
self.shop_filter.blockSignals(True)
|
||||||
|
self.shop_filter.clear()
|
||||||
|
self.shop_filter.addItem("全部店铺", None)
|
||||||
|
for alias in sorted(aliases, key=lambda value: self._shop_label(value, account_by_alias)):
|
||||||
|
self.shop_filter.addItem(self._shop_label(alias, account_by_alias), alias)
|
||||||
|
index = self.shop_filter.findData(previous)
|
||||||
|
self.shop_filter.setCurrentIndex(index if index >= 0 else 0)
|
||||||
|
self.shop_filter.blockSignals(False)
|
||||||
|
|
||||||
|
def _batch_label(self, batch):
|
||||||
|
source_files = batch.source_files
|
||||||
|
first_file = os.path.basename(source_files[0]) if source_files else batch.id
|
||||||
|
return f"{batch.created_at} · {first_file}"
|
||||||
|
|
||||||
|
def _shop_label(self, alias, account_by_alias):
|
||||||
|
account = account_by_alias.get(alias)
|
||||||
|
if account is not None:
|
||||||
|
return f"{account.account_name} ({alias})"
|
||||||
|
return alias
|
||||||
|
|
||||||
|
def _matches_shop(self, task, selected_shop):
|
||||||
|
return selected_shop is None or str(task.alias).strip() == selected_shop
|
||||||
|
|
||||||
|
def _matches_status(self, task, selected_status):
|
||||||
|
if selected_status in (None, "all"):
|
||||||
|
return True
|
||||||
|
if selected_status == "to_generate":
|
||||||
|
return task.stage == "collected" and task.status in {"success", "pending"}
|
||||||
|
if selected_status == "generated":
|
||||||
|
return task.stage == "generated"
|
||||||
|
if selected_status == "applied":
|
||||||
|
return task.stage == "applied"
|
||||||
|
if selected_status == "failed":
|
||||||
|
return task.status == "failed"
|
||||||
|
if selected_status == "skipped":
|
||||||
|
return task.status == "skipped"
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
class CollectTab(QWidget):
|
class CollectTab(QWidget):
|
||||||
"""Tab 1: import Excel files and list imported tasks."""
|
"""Tab 1: import Excel files and list imported tasks."""
|
||||||
|
|
||||||
@@ -1201,6 +1468,12 @@ if QT_IMPORT_ERROR is None:
|
|||||||
status_callback=self.statusBar().showMessage,
|
status_callback=self.statusBar().showMessage,
|
||||||
open_accounts_callback=lambda: self.open_accounts_tab(),
|
open_accounts_callback=lambda: self.open_accounts_tab(),
|
||||||
)
|
)
|
||||||
|
if title == "② AI生成":
|
||||||
|
return GenerateTab(
|
||||||
|
db_path=self.db_path,
|
||||||
|
config=self.config,
|
||||||
|
status_callback=self.statusBar().showMessage,
|
||||||
|
)
|
||||||
if title == "④ 账号管理":
|
if title == "④ 账号管理":
|
||||||
return AccountsTab(
|
return AccountsTab(
|
||||||
db_path=self.db_path,
|
db_path=self.db_path,
|
||||||
|
|||||||
+1
-1
@@ -60,7 +60,7 @@
|
|||||||
| ID | 任务 | 依赖 | 验收要点 | 状态 |
|
| ID | 任务 | 依赖 | 验收要点 | 状态 |
|
||||||
| --- | --- | --- | --- | --- |
|
| --- | --- | --- | --- | --- |
|
||||||
| T-301 | 确定 AI 服务商/模型并接入 `app/ai.py`(`gen_title`/`gen_cover`,带重试/分辨率/jpg质量) | T-005 | 从 `config/ai_models.json` 读取模型与本地明文 Key;`gen_cover` 支持 resolution+jpg_quality;失败按 retry 重试;错误明确;日志脱敏 | DONE |
|
| T-301 | 确定 AI 服务商/模型并接入 `app/ai.py`(`gen_title`/`gen_cover`,带重试/分辨率/jpg质量) | T-005 | 从 `config/ai_models.json` 读取模型与本地明文 Key;`gen_cover` 支持 resolution+jpg_quality;失败按 retry 重试;错误明确;日志脱敏 | DONE |
|
||||||
| T-302 | Tab② 左右布局:左提示词(标题/封面),右按批次/店铺/状态筛选 + 任务列表 | T-301, T-203 | 左 ~1/4 提示词多行;右筛选+列表(店铺/商品id/旧标题/新标题/状态) | TODO |
|
| T-302 | Tab② 左右布局:左提示词(标题/封面),右按批次/店铺/状态筛选 + 任务列表 | T-301, T-203 | 左 ~1/4 提示词多行;右筛选+列表(店铺/商品id/旧标题/新标题/状态) | DONE |
|
||||||
| T-302p | `app/prompts.py` + Tab② 提示词管理 | T-302 | 标题保存/启动回显 title_prompt.txt;封面多模板(下拉+新建/保存/另存为/重命名/删除,存 prompts/cover/);插入 `{新标题}`;预览变量替换;render_prompt 接入生成 | TODO |
|
| T-302p | `app/prompts.py` + Tab② 提示词管理 | T-302 | 标题保存/启动回显 title_prompt.txt;封面多模板(下拉+新建/保存/另存为/重命名/删除,存 prompts/cover/);插入 `{新标题}`;预览变量替换;render_prompt 接入生成 | TODO |
|
||||||
| T-303 | Tab② 开始生成(单按钮)+ 停止 + 进度:**先并发标题再并发图片** | T-302, T-104b | `generate_batch` 先 title_concurrency 并发标题、再 image_concurrency 并发图片;worker/signal 回传进度;每条 set_generated 立即写库;停止可取消未开始项;进度 标题/封面/失败 计数;双击弹窗看新旧封面 | TODO |
|
| T-303 | Tab② 开始生成(单按钮)+ 停止 + 进度:**先并发标题再并发图片** | T-302, T-104b | `generate_batch` 先 title_concurrency 并发标题、再 image_concurrency 并发图片;worker/signal 回传进度;每条 set_generated 立即写库;停止可取消未开始项;进度 标题/封面/失败 计数;双击弹窗看新旧封面 | TODO |
|
||||||
|
|
||||||
|
|||||||
+10
-1
@@ -279,16 +279,18 @@ render_prompt(template_text, task) -> str
|
|||||||
main() -> int # 创建 QApplication + MainWindow
|
main() -> int # 创建 QApplication + MainWindow
|
||||||
class MainWindow(QMainWindow) # QTabWidget: ①②③④⑤;支持注入 db_path/config 便于测试
|
class MainWindow(QMainWindow) # QTabWidget: ①②③④⑤;支持注入 db_path/config 便于测试
|
||||||
class CollectTab(QWidget) # ① 导入采集:导入 Excel + 汇总栏 + QTableView 任务列表 + 未匹配略过标记
|
class CollectTab(QWidget) # ① 导入采集:导入 Excel + 汇总栏 + QTableView 任务列表 + 未匹配略过标记
|
||||||
|
class GenerateTab(QWidget) # ② AI生成:左标题/封面提示词,右批次/店铺/状态筛选 + 任务列表
|
||||||
class CollectWorker(BaseWorker) # ① 后台采集:账号就绪预检 -> editor.collect -> db.set_collected/mark_skipped/mark_failed
|
class CollectWorker(BaseWorker) # ① 后台采集:账号就绪预检 -> editor.collect -> db.set_collected/mark_skipped/mark_failed
|
||||||
class WriteBackWorker(BaseWorker) # ① 后台回写:excel.write_back(batch_id) 写旧标题/旧封面到原 Excel
|
class WriteBackWorker(BaseWorker) # ① 后台回写:excel.write_back(batch_id) 写旧标题/旧封面到原 Excel
|
||||||
class TaskTableModel(QAbstractTableModel) # 任务表格模型:账号/别名/商品ID/阶段;未匹配别名显示“略过”
|
class TaskTableModel(QAbstractTableModel) # 任务表格模型:账号/别名/商品ID/阶段;未匹配别名显示“略过”
|
||||||
|
class GenerateTaskTableModel(QAbstractTableModel) # ② 任务表格模型:店铺/商品ID/旧标题/新标题/状态
|
||||||
class AccountsTab(QWidget) # ④ 账号管理:表格 + 增删改 + 启动登录 + 检测登录 + 快捷方式
|
class AccountsTab(QWidget) # ④ 账号管理:表格 + 增删改 + 启动登录 + 检测登录 + 快捷方式
|
||||||
class AccountDialog(QDialog) # 账号编辑弹窗;密码 QLineEdit.Password
|
class AccountDialog(QDialog) # 账号编辑弹窗;密码 QLineEdit.Password
|
||||||
TAB_TITLES: list[str] # 固定 Tab 顺序
|
TAB_TITLES: list[str] # 固定 Tab 顺序
|
||||||
TAB_STYLE: str # 顶层 Tab 栏防误点样式:最小宽度/padding/间距/当前态
|
TAB_STYLE: str # 顶层 Tab 栏防误点样式:最小宽度/padding/间距/当前态
|
||||||
```
|
```
|
||||||
|
|
||||||
`MainWindow` 已实现五 Tab、① 导入采集任务列表与 ④ 账号管理;②③⑤ 仍为空壳,业务控件按后续 Tab 任务逐步填充。缺 PySide6 时 `main()` 返回 1 并输出明确提示。
|
`MainWindow` 已实现五 Tab、① 导入采集任务列表、② AI生成布局与任务列表、④ 账号管理;③⑤ 仍为空壳,业务控件按后续 Tab 任务逐步填充。缺 PySide6 时 `main()` 返回 1 并输出明确提示。
|
||||||
|
|
||||||
主 Tab 栏必须在 `MainWindow` 初始化时应用 `TAB_STYLE`:5 个 Tab 不使用 Qt 默认紧凑宽度,需保证点击区域稳定、间距清晰、当前 Tab 高亮明显。该样式属于全局导航基础,不归后续业务 Tab 任务重复实现。
|
主 Tab 栏必须在 `MainWindow` 初始化时应用 `TAB_STYLE`:5 个 Tab 不使用 Qt 默认紧凑宽度,需保证点击区域稳定、间距清晰、当前 Tab 高亮明显。该样式属于全局导航基础,不归后续业务 Tab 任务重复实现。
|
||||||
|
|
||||||
@@ -311,6 +313,13 @@ TAB_STYLE: str # 顶层 Tab 栏防误点样式:
|
|||||||
- 「回写旧数据到 Excel」通过 `WriteBackWorker` 后台调用 `excel.write_back()`,把已采集旧标题/旧封面路径按原 Excel 行定位写回;该按钮主要作为自动回写失败后的手动重试入口。原文件被占用时弹窗提示关闭后重试,SQLite 采集结果不回滚。
|
- 「回写旧数据到 Excel」通过 `WriteBackWorker` 后台调用 `excel.write_back()`,把已采集旧标题/旧封面路径按原 Excel 行定位写回;该按钮主要作为自动回写失败后的手动重试入口。原文件被占用时弹窗提示关闭后重试,SQLite 采集结果不回滚。
|
||||||
- 采集前由 `CollectWorker` 做账号就绪预检:无账号、当前批次匹配账号未启动 CDP 端口或未登录时,返回 `blocked=True`,GUI 弹窗汇总并跳转/引导去④账号管理;不无提示批量启动所有账号 Chrome。匹配账号未登录属于预检阻断,不是逐条 skipped。
|
- 采集前由 `CollectWorker` 做账号就绪预检:无账号、当前批次匹配账号未启动 CDP 端口或未登录时,返回 `blocked=True`,GUI 弹窗汇总并跳转/引导去④账号管理;不无提示批量启动所有账号 Chrome。匹配账号未登录属于预检阻断,不是逐条 skipped。
|
||||||
|
|
||||||
|
② AI生成当前要点(T-302):
|
||||||
|
|
||||||
|
- 左右 `QSplitter`:左侧约 1/4 为标题提示词、封面提示词两个多行输入;右侧为筛选栏 + 任务列表。
|
||||||
|
- 筛选栏包含:批次、店铺、状态、刷新。批次来自 `db.list_batches()`;店铺来自当前任务别名并优先显示匹配账号名;状态支持全部/待生成/已生成/失败/略过/已更新。
|
||||||
|
- 任务列表使用 `QTableView + GenerateTaskTableModel`,列为:店铺、商品ID、旧标题、新标题、状态。`stage=collected` 显示“待生成”,`stage=generated` 显示“已生成”,`status=failed/skipped/running` 优先显示对应状态。
|
||||||
|
- T-302 只实现布局与只读列表筛选,不调用 `app.ai`、不写 SQLite;提示词保存/模板管理留给 T-302p,开始生成/停止/进度与 `set_generated()` 留给 T-303。
|
||||||
|
|
||||||
## workers 模块(`app/workers.py`,已建,PySide6)
|
## workers 模块(`app/workers.py`,已建,PySide6)
|
||||||
|
|
||||||
```python
|
```python
|
||||||
|
|||||||
@@ -6,10 +6,10 @@
|
|||||||
## 当前快照
|
## 当前快照
|
||||||
|
|
||||||
- 日期:2026-06-27
|
- 日期:2026-06-27
|
||||||
- 阶段:V0 单账号 CDP 流程已验证;V1 已完成 T-000 正式代码包结构、T-001 `app/editor.py` 模块化、T-002 `app/appconfig.py` 应用配置、T-003 SQLite 持久化地基、T-004 本地数据忽略规则、T-005 AI 模型清单后端、T-006 单元测试基座、T-101 账号 user-data-dir 工具、T-102 Chrome 启动器、T-103 登录保活与检测、T-104 PySide6 主窗口骨架、T-104b PySide6 worker 基座、T-105 Tab④ 账号管理、T-106 账号快捷方式、T-201 Excel 导入入库、T-202 Tab① 任务列表与导入按钮、T-202b Tab① 导入汇总栏、T-203 采集旧标题旧封面、T-204 回写旧字段到原 Excel、T-204b 采集完成自动回写、T-205 首次未配账号/Chrome 未启动/未登录引导保护、T-301 AI 生成接口。
|
- 阶段: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 生成布局与任务列表。
|
||||||
- 技术栈:Python 3.10+,自研 CDP(websocket-client + requests),SQLite(sqlite3)+ `config.json` + openpyxl + AI(`config/ai_models.json` 通用 HTTP,chat JSON / images_edits),GUI PySide6 5 Tab(已定)。
|
- 技术栈:Python 3.10+,自研 CDP(websocket-client + requests),SQLite(sqlite3)+ `config.json` + openpyxl + AI(`config/ai_models.json` 通用 HTTP,chat JSON / images_edits),GUI PySide6 5 Tab(已定)。
|
||||||
- 生产代码:已建立 `app/` 包 + 根入口 `main.py`;`app/cdp.py` 为已验证 CDP 底座;`app/editor.py` 已封装登录状态检测、标题/封面/采集/更新按钮能力;`app/appconfig.py` 已实现 `config.json` 默认值/读写/更新、AI 参数与端口读取,以及 `config/ai_models.json` 模型清单 CRUD/过滤/打码/测试连接;`app/ai.py` 已实现 `gen_title()`/`gen_cover()`,按默认文本/图片模型发起通用 HTTP 调用,支持重试、错误脱敏、图片 URL/base64 解析、resolution resize 与 jpg_quality 保存;`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、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏;`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。
|
- 生产代码:已建立 `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()`,按默认文本/图片模型发起通用 HTTP 调用,支持重试、错误脱敏、图片 URL/base64 解析、resolution resize 与 jpg_quality 保存;`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生成左右布局/标题与封面提示词多行输入/批次店铺状态筛选/任务列表、④ 账号管理表格/弹窗/按钮/快捷方式与状态栏;`app/workers.py` 已实现 `BaseWorker`、通用 signals、取消标记和 `QThread` 启动包装。
|
||||||
- 测试:`tests/` 已建立;T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`,当前覆盖 appconfig/db/config/accounts/chrome 启动与快捷方式/editor 登录检测/excel 导入/ai 标题与封面 HTTP 解析/gui ① 导入采集/gui ④ 账号管理/worker signal 与线程包装,并对尚未实现的 app.prompts 做契约占位 skip;CDP/Shopee 改动仍需测试商品手动验证。
|
- 测试:`tests/` 已建立;T-006 后纯逻辑改动必须运行 `python -m unittest discover -s tests`,当前覆盖 appconfig/db/config/accounts/chrome 启动与快捷方式/editor 登录检测/excel 导入/ai 标题与封面 HTTP 解析/gui ① 导入采集/gui ② AI生成布局与筛选/gui ④ 账号管理/worker signal 与线程包装,并对尚未实现的 app.prompts 做契约占位 skip;CDP/Shopee 改动仍需测试商品手动验证。
|
||||||
- 数据:`config.json`、`config/ai_models.json`、`cmshopee.db`、`chrome_user_data_dir/`、`images/` 已由 `.gitignore` 排除;运营填写后的 Excel 业务文件默认忽略,标准空模板 `shopee待处理任务模板.xlsx` 可提交;`app/appconfig.py` 首次读取缺失的 `config.json` 时会在本地写默认配置,`app/db.py` 调用 `init_db()` 时会在本地创建 SQLite DB。
|
- 数据:`config.json`、`config/ai_models.json`、`cmshopee.db`、`chrome_user_data_dir/`、`images/` 已由 `.gitignore` 排除;运营填写后的 Excel 业务文件默认忽略,标准空模板 `shopee待处理任务模板.xlsx` 可提交;`app/appconfig.py` 首次读取缺失的 `config.json` 时会在本地写默认配置,`app/db.py` 调用 `init_db()` 时会在本地创建 SQLite DB。
|
||||||
|
|
||||||
## 既定设计要点(文档已定)
|
## 既定设计要点(文档已定)
|
||||||
@@ -32,7 +32,7 @@
|
|||||||
| `prototypes/` | 已有 | 已验证原型/探查脚本(demo/set_title/set_cover/get_title/cookies/inspect_images/grab/1.py),保留作人工回归与探查参考;见 `prototypes/README.md` |
|
| `prototypes/` | 已有 | 已验证原型/探查脚本(demo/set_title/set_cover/get_title/cookies/inspect_images/grab/1.py),保留作人工回归与探查参考;见 `prototypes/README.md` |
|
||||||
| `chrome-remote-debug-lan.md` | 已有 | WSL→Windows CDP 转发排查记录 |
|
| `chrome-remote-debug-lan.md` | 已有 | WSL→Windows CDP 转发排查记录 |
|
||||||
| `app/__init__.py` / `app/__main__.py` / `main.py` | 已有 | 正式包与启动入口;`python main.py` / `python -m app` 可运行占位入口 |
|
| `app/__init__.py` / `app/__main__.py` / `main.py` | 已有 | 正式包与启动入口;`python main.py` / `python -m app` 可运行占位入口 |
|
||||||
| `app/gui.py` | 已有 | T-104/T-105/T-106/T-202/T-202b/T-203/T-204/T-204b/T-205 产出:PySide6 `QMainWindow` + 五 Tab;顶部 Tab 栏防误点样式;① 导入采集导入按钮、导入汇总栏、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker、采集前账号就绪预检与④引导、采集完成自动回写与手动重试;④ 账号管理表格、账号弹窗、启动登录、检测登录、快捷方式 |
|
| `app/gui.py` | 已有 | T-104/T-105/T-106/T-202/T-202b/T-203/T-204/T-204b/T-205/T-302 产出:PySide6 `QMainWindow` + 五 Tab;顶部 Tab 栏防误点样式;① 导入采集导入按钮、导入汇总栏、`QTableView` 任务列表、未匹配筛选与略过标记、采集旧标题旧封面 worker、采集前账号就绪预检与④引导、采集完成自动回写与手动重试;② AI生成左右布局、提示词多行输入、筛选栏和任务列表;④ 账号管理表格、账号弹窗、启动登录、检测登录、快捷方式 |
|
||||||
| `app/workers.py` | 已有 | T-104b 产出:`BaseWorker` + 通用 signals + 取消标记 + `run_worker()` QThread 包装 |
|
| `app/workers.py` | 已有 | T-104b 产出:`BaseWorker` + 通用 signals + 取消标记 + `run_worker()` QThread 包装 |
|
||||||
| `app/accounts.py` | 已有 | T-105/T-106 产出:账号 CRUD 服务、目录创建、端口分配、启动登录、检测登录、快捷方式 |
|
| `app/accounts.py` | 已有 | T-105/T-106 产出:账号 CRUD 服务、目录创建、端口分配、启动登录、检测登录、快捷方式 |
|
||||||
| `app/editor.py` | 已有 | T-001/T-103 产出:登录状态检测、打开商品页、读/写标题、读/下载封面、上传拖封面、更新按钮、apply_task |
|
| `app/editor.py` | 已有 | T-001/T-103 产出:登录状态检测、打开商品页、读/写标题、读/下载封面、上传拖封面、更新按钮、apply_task |
|
||||||
@@ -41,7 +41,7 @@
|
|||||||
| `app/db.py` | 已有 | T-003 产出:batches/accounts/tasks schema;WAL/busy_timeout/foreign_keys;账号/批次/任务与 set_* 阶段写库 |
|
| `app/db.py` | 已有 | T-003 产出:batches/accounts/tasks schema;WAL/busy_timeout/foreign_keys;账号/批次/任务与 set_* 阶段写库 |
|
||||||
| `app/config.py` | 已有 | T-101 产出:别名→稳定 slug;创建并返回绝对 user-data-dir |
|
| `app/config.py` | 已有 | T-101 产出:别名→稳定 slug;创建并返回绝对 user-data-dir |
|
||||||
| `app/chrome.py` | 已有 | T-102/T-106 产出:Chrome 启动参数、`subprocess.Popen` 启动、`/json/version` 端口探测、PowerShell `.lnk` 快捷方式 |
|
| `app/chrome.py` | 已有 | T-102/T-106 产出:Chrome 启动参数、`subprocess.Popen` 启动、`/json/version` 端口探测、PowerShell `.lnk` 快捷方式 |
|
||||||
| `tests/` | 已有 | T-006/T-201/T-202/T-202b/T-203/T-204/T-204b/T-205/T-301 产出:stdlib unittest 基座;覆盖 appconfig/db/config/accounts/chrome/editor/excel/ai/gui/workers;prompts 模块契约占位测试 |
|
| `tests/` | 已有 | T-006/T-201/T-202/T-202b/T-203/T-204/T-204b/T-205/T-301/T-302 产出:stdlib unittest 基座;覆盖 appconfig/db/config/accounts/chrome/editor/excel/ai/gui/workers;prompts 模块契约占位测试 |
|
||||||
| `app/excel.py` | 已有 | T-201/T-204 产出:多文件 Excel 输入列解析、必需列整文件拒绝、脏行逐行跳过、批次/任务入库、匹配统计;按源文件/工作表/行号回写旧标题与旧封面路径;支持原文件被占用时另存副本 |
|
| `app/excel.py` | 已有 | T-201/T-204 产出:多文件 Excel 输入列解析、必需列整文件拒绝、脏行逐行跳过、批次/任务入库、匹配统计;按源文件/工作表/行号回写旧标题与旧封面路径;支持原文件被占用时另存副本 |
|
||||||
| `shopee待处理任务模板.xlsx` | 已有,已提交 | 标准空 Excel 模板;单工作表 `待处理任务`,表头 `账号名 | 别名 | 商品id | 旧标题 | 旧封面图片路径 | 新标题 | 新封面图片路径 | 更新状态`;运营复制后填写,填写副本不提交 |
|
| `shopee待处理任务模板.xlsx` | 已有,已提交 | 标准空 Excel 模板;单工作表 `待处理任务`,表头 `账号名 | 别名 | 商品id | 旧标题 | 旧封面图片路径 | 新标题 | 新封面图片路径 | 更新状态`;运营复制后填写,填写副本不提交 |
|
||||||
| `config.json` / `config/ai_models.json` / `cmshopee.db` / `chrome_user_data_dir/` / `images/` | 本地存在或按需生成,已忽略 | 含配置、密钥、业务、登录态、图片,不提交版本库 |
|
| `config.json` / `config/ai_models.json` / `cmshopee.db` / `chrome_user_data_dir/` / `images/` | 本地存在或按需生成,已忽略 | 含配置、密钥、业务、登录态、图片,不提交版本库 |
|
||||||
@@ -59,9 +59,9 @@
|
|||||||
|
|
||||||
任务状态以 [`06-tasks.md`](06-tasks.md) 为准,历史记录见 [`../progress.md`](../progress.md)。
|
任务状态以 [`06-tasks.md`](06-tasks.md) 为准,历史记录见 [`../progress.md`](../progress.md)。
|
||||||
|
|
||||||
- 已完成:T-000(正式代码包结构)、T-001(`app/editor.py` 模块化)、T-002(`app/appconfig.py` + `config.json`)、T-003(SQLite 建表)、T-004(本地数据 gitignore)、T-005(AI 模型清单后端)、T-006(单元测试基座)、T-101(账号 slug/user-data-dir)、T-102(Chrome 启动器)、T-103(登录保活与检测)、T-104(PySide6 五 Tab 主窗口骨架)、T-104b(PySide6 worker 基座)、T-105(Tab④ 账号管理)、T-106(账号快捷方式)、T-201(Excel 导入:解析多文件输入列入库)、T-202(Tab① 任务列表 + 导入按钮 + 别名匹配标记)、T-202b(Tab① 导入汇总栏)、T-203(采集旧标题+旧封面)、T-204(回写旧字段到原 Excel)、T-204b(采集完成自动回写旧字段)、T-205(首次未配账号 / Chrome 未启动 / 未登录引导保护)、T-301(AI 生成接口)。
|
- 已完成: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-302(Tab② 左右布局:左提示词,右筛选 + 任务列表)**。
|
- 下一个可领取任务:**T-302p(`app/prompts.py` + Tab② 提示词管理)**。
|
||||||
|
|
||||||
## 当前已知限制
|
## 当前已知限制
|
||||||
|
|
||||||
|
|||||||
+1
-1
@@ -148,7 +148,7 @@
|
|||||||
| --- | --- | --- |
|
| --- | --- | --- |
|
||||||
| `MainWindow(QMainWindow)` | 根窗口 | 持有 `QTabWidget`、状态栏、全局消息 |
|
| `MainWindow(QMainWindow)` | 根窗口 | 持有 `QTabWidget`、状态栏、全局消息 |
|
||||||
| `CollectTab(QWidget)` | ① | 导入、任务表、采集、回写 |
|
| `CollectTab(QWidget)` | ① | 导入、任务表、采集、回写 |
|
||||||
| `GenerateTab(QWidget)` | ② | 左提示词 + 右筛选/任务列表、双击看新旧封面、开始生成 |
|
| `GenerateTab(QWidget)` | ② | 左提示词 + 右筛选/任务列表;双击看图与开始生成由后续 T-303 接入 |
|
||||||
| `ApplyTab(QWidget)` | ③ | 已生成任务、开始更新确认、换标题+封面+提交、回写 |
|
| `ApplyTab(QWidget)` | ③ | 已生成任务、开始更新确认、换标题+封面+提交、回写 |
|
||||||
| `AccountsTab(QWidget)` | ④ | 账号增删改、启动登录、检测登录、生成快捷方式 |
|
| `AccountsTab(QWidget)` | ④ | 账号增删改、启动登录、检测登录、生成快捷方式 |
|
||||||
| `SettingsTab(QWidget)` | ⑤ | AI/目录/Chrome 配置 |
|
| `SettingsTab(QWidget)` | ⑤ | AI/目录/Chrome 配置 |
|
||||||
|
|||||||
@@ -481,3 +481,12 @@
|
|||||||
- 测试:新增 `tests/test_ai.py`,覆盖文本生成重试与 extra_body、缺字段错误不泄露 Key、图片生成保存 JPEG。
|
- 测试:新增 `tests/test_ai.py`,覆盖文本生成重试与 extra_body、缺字段错误不泄露 Key、图片生成保存 JPEG。
|
||||||
- 验证:`python -m unittest discover -s tests -p "test_ai.py"` 通过(3 tests);`python -m compileall app main.py tests` 通过;`python -m unittest discover -s tests` 通过(63 tests,skipped=1);`py -3 -m compileall app main.py tests` 通过;`py -3 -m unittest discover -s tests -p "test_ai.py"` 通过(3 tests,skipped=1,py -3 环境缺 Pillow);`py -3 -m unittest discover -s tests` 通过(37 tests,skipped=5,py -3 环境缺 openpyxl/PySide6/Pillow,相关测试按设计跳过)。
|
- 验证:`python -m unittest discover -s tests -p "test_ai.py"` 通过(3 tests);`python -m compileall app main.py tests` 通过;`python -m unittest discover -s tests` 通过(63 tests,skipped=1);`py -3 -m compileall app main.py tests` 通过;`py -3 -m unittest discover -s tests -p "test_ai.py"` 通过(3 tests,skipped=1,py -3 环境缺 Pillow);`py -3 -m unittest discover -s tests` 通过(37 tests,skipped=5,py -3 环境缺 openpyxl/PySide6/Pillow,相关测试按设计跳过)。
|
||||||
- 注意:T-301 只做通用 AI 接口与 mock 单测;真实 AI 调用还需要在 `config/ai_models.json` 填入可用 url/model/api_key 后做一次小样本实测。生成批次编排和 GUI 接入留给 T-303/T-302。
|
- 注意:T-301 只做通用 AI 接口与 mock 单测;真实 AI 调用还需要在 `config/ai_models.json` 填入可用 url/model/api_key 后做一次小样本实测。生成批次编排和 GUI 接入留给 T-303/T-302。
|
||||||
|
|
||||||
|
## 【2026-06-27】T-302 Tab② AI生成布局与任务列表
|
||||||
|
|
||||||
|
- 状态:DONE
|
||||||
|
- 变更:`app/gui.py` 新增 `GenerateTab` 与 `GenerateTaskTableModel`;MainWindow 的②从占位页切换为实际 AI 生成页。左侧使用 `QSplitter` 约 1/4 宽度放标题提示词、封面提示词两个多行输入;右侧提供批次、店铺、状态筛选和任务列表。
|
||||||
|
- 细节:任务列表列为店铺、商品ID、旧标题、新标题、状态;店铺优先使用匹配账号名;状态将 `stage=collected` 显示为“待生成”、`stage=generated` 显示为“已生成”,并优先显示 failed/skipped/running。筛选只读 SQLite,不调用 AI、不写库。
|
||||||
|
- 测试:`tests/test_gui.py` 增加 MainWindow 挂载②、提示词多行输入、任务列表展示,以及按批次/店铺/状态筛选的覆盖。
|
||||||
|
- 文档:`docs/06-tasks.md` 将 T-302 标为 DONE;同步 `docs/api.md`、`docs/current-state.md`,下一个可领取任务更新为 T-302p。
|
||||||
|
- 验证:`python -m unittest discover -s tests -p "test_gui.py"` 通过(20 tests);`python -m compileall app main.py tests` 通过;`python -m unittest discover -s tests` 通过(65 tests,skipped=1);`py -3 -m compileall app main.py tests` 通过;`py -3 -m unittest discover -s tests` 通过(37 tests,skipped=5,py -3 环境缺 openpyxl/PySide6/Pillow,相关测试按设计跳过)。
|
||||||
|
|||||||
+108
-1
@@ -14,13 +14,14 @@ from app import accounts, db
|
|||||||
if gui.QT_IMPORT_ERROR is not None:
|
if gui.QT_IMPORT_ERROR is not None:
|
||||||
raise unittest.SkipTest("PySide6 未安装")
|
raise unittest.SkipTest("PySide6 未安装")
|
||||||
|
|
||||||
from PySide6.QtWidgets import QApplication, QLineEdit, QTableView
|
from PySide6.QtWidgets import QApplication, QLineEdit, QPlainTextEdit, QTableView
|
||||||
|
|
||||||
from app.gui import (
|
from app.gui import (
|
||||||
AccountDialog,
|
AccountDialog,
|
||||||
AccountsTab,
|
AccountsTab,
|
||||||
CollectWorker,
|
CollectWorker,
|
||||||
CollectTab,
|
CollectTab,
|
||||||
|
GenerateTab,
|
||||||
MainWindow,
|
MainWindow,
|
||||||
TAB_STYLE,
|
TAB_STYLE,
|
||||||
TAB_TITLES,
|
TAB_TITLES,
|
||||||
@@ -64,6 +65,7 @@ class GuiTests(TempDirMixin, unittest.TestCase):
|
|||||||
self.assertIn("padding: 8px 18px", window.tabs.styleSheet())
|
self.assertIn("padding: 8px 18px", window.tabs.styleSheet())
|
||||||
self.assertIn("margin-right: 8px", window.tabs.styleSheet())
|
self.assertIn("margin-right: 8px", window.tabs.styleSheet())
|
||||||
self.assertIsInstance(window.tabs.widget(0), CollectTab)
|
self.assertIsInstance(window.tabs.widget(0), CollectTab)
|
||||||
|
self.assertIsInstance(window.tabs.widget(1), GenerateTab)
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
"回写旧数据到 Excel",
|
"回写旧数据到 Excel",
|
||||||
window.tabs.widget(0).write_back_button.text(),
|
window.tabs.widget(0).write_back_button.text(),
|
||||||
@@ -71,6 +73,111 @@ class GuiTests(TempDirMixin, unittest.TestCase):
|
|||||||
|
|
||||||
self.assert_removed(temp_dir)
|
self.assert_removed(temp_dir)
|
||||||
|
|
||||||
|
def test_generate_tab_has_prompt_editors_and_task_table(self):
|
||||||
|
with self.make_temp_dir() as temp_dir:
|
||||||
|
tab = GenerateTab(config=self.make_config(temp_dir))
|
||||||
|
self.addCleanup(tab.close)
|
||||||
|
|
||||||
|
self.assertIsInstance(tab.title_prompt_edit, QPlainTextEdit)
|
||||||
|
self.assertIsInstance(tab.cover_prompt_edit, QPlainTextEdit)
|
||||||
|
self.assertIsInstance(tab.task_table, QTableView)
|
||||||
|
self.assertEqual("标题提示词", tab.title_prompt_edit.placeholderText())
|
||||||
|
self.assertEqual("封面提示词", tab.cover_prompt_edit.placeholderText())
|
||||||
|
self.assertEqual(["店铺", "商品ID", "旧标题", "新标题", "状态"], tab.model.HEADERS)
|
||||||
|
self.assertEqual("任务 0/0 条", tab.summary_label.text())
|
||||||
|
|
||||||
|
self.assert_removed(temp_dir)
|
||||||
|
|
||||||
|
def test_generate_tab_lists_tasks_and_filters_by_shop_status_and_batch(self):
|
||||||
|
with self.make_temp_dir() as temp_dir:
|
||||||
|
cfg = self.make_config(temp_dir)
|
||||||
|
accounts.create_account("主店", "alias-a", debug_port=9222, config=cfg)
|
||||||
|
accounts.create_account("副店", "alias-b", debug_port=9223, config=cfg)
|
||||||
|
batch_a = db.create_batch(["input-a.xlsx"], path=cfg["db_path"])
|
||||||
|
batch_b = db.create_batch(["input-b.xlsx"], path=cfg["db_path"])
|
||||||
|
db.insert_tasks(
|
||||||
|
batch_a,
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"source_file_abs": os.path.join(temp_dir, "input-a.xlsx"),
|
||||||
|
"source_sheet": "商品",
|
||||||
|
"source_row": 2,
|
||||||
|
"account_name": "Excel主店",
|
||||||
|
"alias": "alias-a",
|
||||||
|
"item_id": "51100639510",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"source_file_abs": os.path.join(temp_dir, "input-a.xlsx"),
|
||||||
|
"source_sheet": "商品",
|
||||||
|
"source_row": 3,
|
||||||
|
"account_name": "Excel副店",
|
||||||
|
"alias": "alias-b",
|
||||||
|
"item_id": "51100639511",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"source_file_abs": os.path.join(temp_dir, "input-a.xlsx"),
|
||||||
|
"source_sheet": "商品",
|
||||||
|
"source_row": 4,
|
||||||
|
"account_name": "Excel主店",
|
||||||
|
"alias": "alias-a",
|
||||||
|
"item_id": "51100639512",
|
||||||
|
},
|
||||||
|
],
|
||||||
|
path=cfg["db_path"],
|
||||||
|
)
|
||||||
|
db.insert_tasks(
|
||||||
|
batch_b,
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"source_file_abs": os.path.join(temp_dir, "input-b.xlsx"),
|
||||||
|
"source_sheet": "商品",
|
||||||
|
"source_row": 2,
|
||||||
|
"account_name": "Excel主店",
|
||||||
|
"alias": "alias-a",
|
||||||
|
"item_id": "51100639513",
|
||||||
|
}
|
||||||
|
],
|
||||||
|
path=cfg["db_path"],
|
||||||
|
)
|
||||||
|
tasks_a = db.list_tasks(batch_id=batch_a, path=cfg["db_path"])
|
||||||
|
tasks_b = db.list_tasks(batch_id=batch_b, path=cfg["db_path"])
|
||||||
|
db.set_collected(tasks_a[0].id, "旧标题A", "old-a.jpg", path=cfg["db_path"])
|
||||||
|
db.set_collected(tasks_a[1].id, "旧标题B", "old-b.jpg", path=cfg["db_path"])
|
||||||
|
db.set_collected(tasks_a[2].id, "旧标题C", "old-c.jpg", path=cfg["db_path"])
|
||||||
|
db.set_generated(tasks_a[2].id, "新标题C", "new-c.jpg", path=cfg["db_path"])
|
||||||
|
db.mark_failed(tasks_b[0].id, "generate", "生成失败", path=cfg["db_path"])
|
||||||
|
|
||||||
|
tab = GenerateTab(config=cfg)
|
||||||
|
self.addCleanup(tab.close)
|
||||||
|
|
||||||
|
self.assertEqual(4, tab.model.rowCount())
|
||||||
|
self.assertEqual("主店", tab.model.index(0, 0).data())
|
||||||
|
self.assertEqual("51100639510", tab.model.index(0, 1).data())
|
||||||
|
self.assertEqual("旧标题A", tab.model.index(0, 2).data())
|
||||||
|
self.assertEqual("", tab.model.index(0, 3).data())
|
||||||
|
self.assertEqual("待生成", tab.model.index(0, 4).data())
|
||||||
|
self.assertEqual("已生成", tab.model.index(2, 4).data())
|
||||||
|
|
||||||
|
tab.status_filter.setCurrentIndex(tab.status_filter.findData("to_generate"))
|
||||||
|
self.assertEqual(2, tab.model.rowCount())
|
||||||
|
self.assertTrue(all(tab.model.index(row, 4).data() == "待生成" for row in range(2)))
|
||||||
|
self.assertEqual("任务 2/4 条", tab.summary_label.text())
|
||||||
|
|
||||||
|
tab.shop_filter.setCurrentIndex(tab.shop_filter.findData("alias-b"))
|
||||||
|
self.assertEqual(1, tab.model.rowCount())
|
||||||
|
self.assertEqual("副店", tab.model.index(0, 0).data())
|
||||||
|
|
||||||
|
tab.status_filter.setCurrentIndex(tab.status_filter.findData("all"))
|
||||||
|
tab.batch_filter.setCurrentIndex(tab.batch_filter.findData(batch_b))
|
||||||
|
self.assertEqual(1, tab.model.rowCount())
|
||||||
|
self.assertEqual("失败", tab.model.index(0, 4).data())
|
||||||
|
self.assertEqual(
|
||||||
|
"生成失败",
|
||||||
|
tab.model.data(tab.model.index(0, 0), gui.Qt.ToolTipRole),
|
||||||
|
)
|
||||||
|
|
||||||
|
self.assert_removed(temp_dir)
|
||||||
|
|
||||||
def test_tab_switch_updates_status_bar(self):
|
def test_tab_switch_updates_status_bar(self):
|
||||||
with self.make_temp_dir() as temp_dir:
|
with self.make_temp_dir() as temp_dir:
|
||||||
window = MainWindow(config=self.make_config(temp_dir))
|
window = MainWindow(config=self.make_config(temp_dir))
|
||||||
|
|||||||
Reference in New Issue
Block a user