445 lines
16 KiB
Python
445 lines
16 KiB
Python
"""PDD 任务页的事件绑定。
|
||
|
||
把 `pdd_ui.py` 里控件的信号，接到查询、任务引擎等实际动作上。
|
||
|
||
改动本文件前必读 `client/AGENTS.md`。以下几条最容易踩：
|
||
|
||
- **禁止在 Qt 主线程做任何阻塞的事**：Admin 请求、uiautomator2/ADB 调用、
|
||
`time.sleep()`、轮询、大 XML 解析、批量写文件。做了界面就会卡死转圈。
|
||
- 长任务统一用 `QObject` + `moveToThread` 的 Worker 写法，
|
||
模板照抄 `docs/client/02-architecture.md` §5.1。
|
||
**不要**用 `QThread` 子类、`QRunnable` 或 Python 的 `threading`。
|
||
- 后台结果只能通过**信号**回到主线程，Worker 里一行界面代码都不许有。
|
||
- 窗口关闭时要断开信号并置标志位，否则迟到的后台结果会访问
|
||
已经销毁的控件、直接崩溃。做法见同文档 §5.2。
|
||
- 数据库读写走 Repository，**不要在这里拼业务 SQL**。
|
||
- “获取任务”会真的去操作手机、可能下单；“搜索”只读本地数据库。
|
||
两者必须分开，不得共用入口。
|
||
- 普通成功不弹窗，更新界面即可；可恢复错误用 `InfoBar`
|
||
（模板见 `docs/client/05-ui-specification.md` §9.1）；
|
||
只有必须让用户当场做决定时才用模态对话框。
|
||
"""
|
||
|
||
from typing import Dict, Optional
|
||
|
||
from PyQt5.QtCore import QCoreApplication, QObject, QThread, pyqtSignal, pyqtSlot
|
||
from qfluentwidgets import InfoBar, InfoBarPosition
|
||
|
||
from .admin_gateway import (
|
||
AdminGatewayError,
|
||
AdminTask,
|
||
AndroidDeviceInfo,
|
||
ClaimCapabilities,
|
||
ClientInfo,
|
||
TaskClaimGateway,
|
||
)
|
||
from .current_client_service import CurrentClientService
|
||
from .http_admin_gateway import DEFAULT_ADMIN_BASE_URL, HttpAdminGateway
|
||
from .pdd_ui import PDDTaskPage, TaskRow
|
||
from .selected_android_device_service import SelectedAndroidDeviceService
|
||
from .settings_repository import SettingsRepository
|
||
from .task_models import (
|
||
NewClaimedTask,
|
||
TaskFilters,
|
||
TaskStatus,
|
||
TaskSummary,
|
||
TaskType,
|
||
)
|
||
from .task_repository import DuplicateTaskError, TaskRepository
|
||
|
||
|
||
TASK_TYPE_BY_TEXT = {
|
||
"采集": TaskType.COLLECT,
|
||
"采购": TaskType.PURCHASE,
|
||
}
|
||
|
||
TASK_STATUS_BY_TEXT = {
|
||
"待执行": TaskStatus.CLAIMED,
|
||
"执行中": TaskStatus.RUNNING,
|
||
"结果待提交": TaskStatus.RESULT_PENDING,
|
||
"等待重试": TaskStatus.RETRY_WAIT,
|
||
"需要人工处理": TaskStatus.MANUAL_REVIEW,
|
||
"已完成": TaskStatus.SUCCEEDED,
|
||
"失败": TaskStatus.FAILED,
|
||
"已取消": TaskStatus.CANCELLED,
|
||
}
|
||
|
||
TASK_TYPE_TEXT = {
|
||
TaskType.COLLECT: "采集",
|
||
TaskType.PURCHASE: "采购",
|
||
}
|
||
|
||
TASK_STATUS_TEXT = {
|
||
TaskStatus.CLAIMED: "待执行",
|
||
TaskStatus.RUNNING: "执行中",
|
||
TaskStatus.RESULT_PENDING: "结果待提交",
|
||
TaskStatus.RETRY_WAIT: "等待重试",
|
||
TaskStatus.MANUAL_REVIEW: "需要人工处理",
|
||
TaskStatus.SUCCEEDED: "已完成",
|
||
TaskStatus.FAILED: "失败",
|
||
TaskStatus.CANCELLED: "已取消",
|
||
}
|
||
|
||
|
||
def admin_task_to_new_claimed_task(task: AdminTask) -> NewClaimedTask:
|
||
"""把 Admin 任务显式映射为本地任务，避免字段名自动展开出错。"""
|
||
|
||
if task.task_type is not TaskType.COLLECT:
|
||
raise ValueError(f"任务 {task.task_id} 不是采集任务")
|
||
|
||
payload = dict(task.payload)
|
||
goods_url = payload.get("goods_url")
|
||
goods_id = payload.get("goods_id")
|
||
if not isinstance(goods_url, str) or not goods_url.strip():
|
||
raise ValueError("payload.goods_url 不能为空")
|
||
if goods_id is not None and not isinstance(goods_id, str):
|
||
raise ValueError("payload.goods_id 必须是文本")
|
||
|
||
original_task = {
|
||
"id": task.task_id,
|
||
"type": task.task_type.value,
|
||
"version": task.version,
|
||
"priority": task.priority,
|
||
"payload": payload,
|
||
"created_at": task.created_at,
|
||
"updated_at": task.updated_at,
|
||
}
|
||
return NewClaimedTask(
|
||
remote_task_id=task.task_id,
|
||
task_type=task.task_type,
|
||
goods_url=goods_url,
|
||
goods_id=goods_id.strip() if isinstance(goods_id, str) else None,
|
||
priority=task.priority,
|
||
version=task.version,
|
||
admin_payload=original_task,
|
||
)
|
||
|
||
|
||
class ClaimTaskWorker(QObject):
|
||
"""在后台领取至多一个采集任务，并先写入本地数据库。"""
|
||
|
||
noTask = pyqtSignal()
|
||
taskSaved = pyqtSignal(str)
|
||
duplicateTask = pyqtSignal(str)
|
||
localSaveFailed = pyqtSignal(str, str)
|
||
failed = pyqtSignal(str)
|
||
completed = pyqtSignal()
|
||
|
||
def __init__(
|
||
self,
|
||
gateway: TaskClaimGateway,
|
||
task_repository: TaskRepository,
|
||
client_service: CurrentClientService,
|
||
android_device_service: SelectedAndroidDeviceService,
|
||
) -> None:
|
||
super().__init__()
|
||
self._gateway = gateway
|
||
self._task_repository = task_repository
|
||
self._client_service = client_service
|
||
self._android_device_service = android_device_service
|
||
self._cancelled = False
|
||
|
||
def cancel(self) -> None:
|
||
"""阻止尚未开始的领取；已领取的任务仍必须保存到本地。"""
|
||
|
||
self._cancelled = True
|
||
|
||
@pyqtSlot()
|
||
def run(self) -> None:
|
||
claimed_task: Optional[AdminTask] = None
|
||
try:
|
||
if self._cancelled:
|
||
return
|
||
client_settings = self._client_service.load()
|
||
if not client_settings.client_id:
|
||
self.failed.emit("请先在设置页保存当前设备号和设备名")
|
||
return
|
||
android_serial = self._android_device_service.load()
|
||
if not android_serial:
|
||
self.failed.emit("请先在设置页选择并保存 Android 设备")
|
||
return
|
||
|
||
capabilities = ClaimCapabilities(
|
||
device=AndroidDeviceInfo(android_serial),
|
||
supported_types=(TaskType.COLLECT,),
|
||
purchase_mode="dry_run",
|
||
schema_versions=(1,),
|
||
)
|
||
claimed_task = self._gateway.claim_next(
|
||
ClientInfo(
|
||
client_settings.client_id,
|
||
client_settings.client_name,
|
||
),
|
||
capabilities,
|
||
)
|
||
if claimed_task is None:
|
||
if not self._cancelled:
|
||
self.noTask.emit()
|
||
return
|
||
|
||
local_task = admin_task_to_new_claimed_task(claimed_task)
|
||
try:
|
||
self._task_repository.add_claimed_task(local_task)
|
||
except DuplicateTaskError:
|
||
if not self._cancelled:
|
||
self.duplicateTask.emit(claimed_task.task_id)
|
||
return
|
||
except Exception as exc:
|
||
if not self._cancelled:
|
||
self.localSaveFailed.emit(claimed_task.task_id, str(exc))
|
||
return
|
||
|
||
if not self._cancelled:
|
||
self.taskSaved.emit(claimed_task.task_id)
|
||
except AdminGatewayError as exc:
|
||
if not self._cancelled:
|
||
request_hint = (
|
||
f"，请求编号：{exc.request_id}" if exc.request_id else ""
|
||
)
|
||
self.failed.emit(f"{exc}{request_hint}")
|
||
except Exception as exc:
|
||
if not self._cancelled:
|
||
if claimed_task is not None:
|
||
self.localSaveFailed.emit(claimed_task.task_id, str(exc))
|
||
else:
|
||
self.failed.emit(f"领取任务失败：{exc}")
|
||
finally:
|
||
self.completed.emit()
|
||
|
||
|
||
class PDDTaskPageEvent(QObject):
|
||
"""把 PDD 页面只读操作连接到本地任务 Repository。"""
|
||
|
||
def __init__(
|
||
self,
|
||
page: PDDTaskPage,
|
||
repository: Optional[TaskRepository] = None,
|
||
parent=None,
|
||
claim_gateway: Optional[TaskClaimGateway] = None,
|
||
settings_repository: Optional[SettingsRepository] = None,
|
||
):
|
||
super().__init__(parent or page)
|
||
self._page = page
|
||
self._repository = repository or TaskRepository()
|
||
self._filters = TaskFilters()
|
||
self._closing = False
|
||
self._claim_busy = False
|
||
self._claim_thread: Optional[QThread] = None
|
||
self._claim_worker: Optional[ClaimTaskWorker] = None
|
||
|
||
settings = settings_repository or SettingsRepository()
|
||
self._client_service = CurrentClientService(settings)
|
||
self._selected_android_device_service = SelectedAndroidDeviceService(
|
||
settings
|
||
)
|
||
self._claim_gateway = claim_gateway
|
||
self._claim_gateway_error = ""
|
||
if self._claim_gateway is None:
|
||
base_url = settings.get("admin.base_url", DEFAULT_ADMIN_BASE_URL)
|
||
timeout_value = settings.get("admin.request_timeout_seconds", 3.0)
|
||
try:
|
||
timeout_seconds = float(timeout_value)
|
||
self._claim_gateway = HttpAdminGateway(
|
||
base_url if isinstance(base_url, str) else "",
|
||
timeout_seconds=timeout_seconds,
|
||
)
|
||
except (TypeError, ValueError) as exc:
|
||
self._claim_gateway_error = str(exc)
|
||
|
||
page.searchRequested.connect(self.search_tasks)
|
||
page.refreshRequested.connect(self.refresh_tasks)
|
||
page.autoFetchRequested.connect(self._request_claim_task)
|
||
page.taskModel.loadMoreRequested.connect(self._load_page)
|
||
page.autoFetchButton.setText("获取任务")
|
||
page.destroyed.connect(self.shutdown)
|
||
application = QCoreApplication.instance()
|
||
if application is not None:
|
||
application.aboutToQuit.connect(self.shutdown)
|
||
|
||
def load_initial_tasks(self) -> None:
|
||
"""应用启动后读取第一页本地任务。"""
|
||
|
||
self._reload()
|
||
|
||
def search_tasks(self, values: Dict[str, str]) -> None:
|
||
"""把界面中文筛选值转换成领域筛选，并重新查询。"""
|
||
|
||
self._filters = TaskFilters(
|
||
task_type=TASK_TYPE_BY_TEXT.get(values.get("task_type", "全部")),
|
||
status=TASK_STATUS_BY_TEXT.get(values.get("status", "全部")),
|
||
keyword=values.get("keyword", "").strip(),
|
||
)
|
||
self._reload()
|
||
|
||
def refresh_tasks(self) -> None:
|
||
"""使用当前筛选条件刷新列表。"""
|
||
|
||
self._reload()
|
||
|
||
@pyqtSlot()
|
||
def _request_claim_task(self) -> None:
|
||
"""启动一次后台领取；重复点击和关闭期间直接忽略。"""
|
||
|
||
if self._closing or self._claim_busy:
|
||
return
|
||
if self._claim_gateway_error or self._claim_gateway is None:
|
||
message = self._claim_gateway_error or "Admin 领取服务未初始化"
|
||
self._page.set_engine_status(f"领取失败：{message}")
|
||
self._show_claim_error("领取任务失败", message)
|
||
return
|
||
|
||
self._claim_busy = True
|
||
self._page.autoFetchButton.setEnabled(False)
|
||
self._page.set_engine_status("正在领取一个采集任务，请稍候…")
|
||
|
||
thread = QThread(self)
|
||
worker = ClaimTaskWorker(
|
||
self._claim_gateway,
|
||
self._repository,
|
||
self._client_service,
|
||
self._selected_android_device_service,
|
||
)
|
||
worker.moveToThread(thread)
|
||
thread.started.connect(worker.run)
|
||
worker.noTask.connect(self._on_no_claimed_task)
|
||
worker.taskSaved.connect(self._on_claimed_task_saved)
|
||
worker.duplicateTask.connect(self._on_duplicate_claimed_task)
|
||
worker.localSaveFailed.connect(self._on_claimed_task_save_failed)
|
||
worker.failed.connect(self._on_claim_failed)
|
||
worker.completed.connect(thread.quit)
|
||
worker.completed.connect(worker.deleteLater)
|
||
thread.finished.connect(thread.deleteLater)
|
||
thread.finished.connect(self._on_claim_thread_finished)
|
||
self._claim_thread = thread
|
||
self._claim_worker = worker
|
||
thread.start()
|
||
|
||
@pyqtSlot()
|
||
def _on_no_claimed_task(self) -> None:
|
||
if not self._closing:
|
||
self._page.set_engine_status("暂无可领取的采集任务")
|
||
|
||
@pyqtSlot(str)
|
||
def _on_claimed_task_saved(self, task_id: str) -> None:
|
||
if self._closing:
|
||
return
|
||
self._page.set_engine_status(
|
||
f"已领取任务 {task_id}，已保存到本地任务列表"
|
||
)
|
||
self._reload()
|
||
|
||
@pyqtSlot(str)
|
||
def _on_duplicate_claimed_task(self, task_id: str) -> None:
|
||
if not self._closing:
|
||
self._page.set_engine_status(
|
||
f"任务 {task_id} 本地已有，未重复保存"
|
||
)
|
||
|
||
@pyqtSlot(str, str)
|
||
def _on_claimed_task_save_failed(self, task_id: str, message: str) -> None:
|
||
if not self._closing:
|
||
content = (
|
||
f"任务 {task_id} 已在服务端领取，但本地保存失败：{message}。"
|
||
"请记下这个任务号联系维护者。"
|
||
)
|
||
self._page.set_engine_status(content)
|
||
self._show_claim_error("本地任务保存失败", content)
|
||
|
||
@pyqtSlot(str)
|
||
def _on_claim_failed(self, message: str) -> None:
|
||
if not self._closing:
|
||
content = message or "领取任务失败，请稍后重试"
|
||
self._page.set_engine_status(content)
|
||
self._show_claim_error("领取任务失败", content)
|
||
|
||
def _show_claim_error(self, title: str, content: str) -> None:
|
||
"""显示不会自动消失的可恢复错误，同时保留底部状态文字。"""
|
||
|
||
InfoBar.error(
|
||
title=title,
|
||
content=content,
|
||
isClosable=True,
|
||
duration=-1,
|
||
position=InfoBarPosition.TOP_RIGHT,
|
||
parent=self._page,
|
||
)
|
||
|
||
@pyqtSlot()
|
||
def _on_claim_thread_finished(self) -> None:
|
||
self._claim_worker = None
|
||
self._claim_thread = None
|
||
self._claim_busy = False
|
||
if not self._closing:
|
||
self._page.autoFetchButton.setText("获取任务")
|
||
self._page.autoFetchButton.setEnabled(True)
|
||
|
||
def _reload(self) -> None:
|
||
self._page.begin_task_reload()
|
||
self._page.taskModel.fetchMore()
|
||
|
||
def _load_page(self, offset: int, limit: int) -> None:
|
||
try:
|
||
summaries = self._repository.list_tasks(
|
||
filters=self._filters,
|
||
limit=limit,
|
||
offset=offset,
|
||
)
|
||
except Exception:
|
||
self._page.set_load_error("无法读取本地任务，请检查数据库后重试。")
|
||
return
|
||
|
||
rows = [summary_to_row(summary) for summary in summaries]
|
||
self._page.append_task_page(rows, has_more=len(rows) == limit)
|
||
|
||
@pyqtSlot()
|
||
def shutdown(self) -> None:
|
||
"""窗口关闭时停止新的领取，并等待已领取任务完成本地保存。"""
|
||
|
||
if self._closing:
|
||
return
|
||
self._closing = True
|
||
|
||
worker = self._claim_worker
|
||
thread = self._claim_thread
|
||
if worker is not None:
|
||
try:
|
||
worker.cancel()
|
||
signal_slots = (
|
||
(worker.noTask, self._on_no_claimed_task),
|
||
(worker.taskSaved, self._on_claimed_task_saved),
|
||
(worker.duplicateTask, self._on_duplicate_claimed_task),
|
||
(
|
||
worker.localSaveFailed,
|
||
self._on_claimed_task_save_failed,
|
||
),
|
||
(worker.failed, self._on_claim_failed),
|
||
)
|
||
except RuntimeError:
|
||
signal_slots = ()
|
||
for signal, slot in signal_slots:
|
||
try:
|
||
signal.disconnect(slot)
|
||
except (TypeError, RuntimeError):
|
||
pass
|
||
|
||
if thread is not None and thread.isRunning():
|
||
thread.quit()
|
||
thread.wait(10_000)
|
||
|
||
|
||
def summary_to_row(summary: TaskSummary) -> TaskRow:
|
||
"""把领域摘要转换成只供表格显示的轻量行。"""
|
||
|
||
return TaskRow(
|
||
remote_task_id=summary.remote_task_id,
|
||
task_type=TASK_TYPE_TEXT[summary.task_type],
|
||
goods_id=summary.goods_id or "",
|
||
title=summary.title or "",
|
||
color=summary.target_color or "",
|
||
size=summary.target_size or "",
|
||
price_cents=summary.price_cent,
|
||
quantity=summary.quantity,
|
||
status=TASK_STATUS_TEXT[summary.status],
|
||
updated_at=summary.updated_at,
|
||
)
|