query-payslip-default/examples/real_browser_rpa/scripts/service/task_service.py

"""轻量任务编排示例（无真实数据库依赖）。"""

from __future__ import annotations

import asyncio
import uuid
from typing import Any, Optional

from service.account_client import (
    AccountManagerError,
    pick_web_account,
    release_lease,
    resolve_account_selectors_from_config,
)
from service.task_rpa import (
    ScrapeRunResult,
    format_stop_reason_for_user,
    run_keyword_search_async,
)
from util.constants import (
    DEFAULT_MAX_ITEMS,
    DEFAULT_MAX_SCROLLS,
    SKILL_SLUG,
    TARGET_PLATFORM,
)


def _error_payload(
    code: str,
    message: str,
    *,
    browser_started: Optional[bool] = None,
    stage: Optional[str] = None,
) -> dict[str, Any]:
    payload: dict[str, Any] = {
        "success": False,
        "error": {"code": code, "message": message},
    }
    if browser_started is not None:
        payload["browser_started"] = browser_started
    if stage:
        payload["stage"] = stage
    return payload


def _success_payload(
    *,
    keyword: str,
    collected_count: int,
    items: list[dict[str, Any]],
    stop_reason: Optional[str],
) -> dict[str, Any]:
    return {
        "success": True,
        "skill": SKILL_SLUG,
        "keyword": keyword,
        "collected_count": collected_count,
        "items": items,
        "stop_reason": stop_reason,
        "stop_reason_message": format_stop_reason_for_user(stop_reason, collected_count),
    }


async def run_keyword_search_task(
    keyword: str,
    *,
    max_items: int = DEFAULT_MAX_ITEMS,
    max_scrolls: int = DEFAULT_MAX_SCROLLS,
    account_id: Optional[str] = None,
    platform: str = TARGET_PLATFORM,
) -> dict[str, Any]:
    """
    编排一次关键词搜索采集。

    真实 skill 中还应：
    - 写 SQLite 入库（items 表）
    - 使用 RpaVideoSession 录制步骤留痕
    - 失败时 capture_failure 截图
    """
    batch_id = uuid.uuid4().hex[:12]
    data_dir = ""  # 示例占位；真实 skill 使用 get_skill_data_dir()
    lease_token: Optional[str] = None
    account: Optional[dict[str, Any]] = None

    try:
        cfg_account_id, cfg_login_id = resolve_account_selectors_from_config()
        account = pick_web_account(
            platform,
            account_id=account_id or cfg_account_id,
            login_id=cfg_login_id,
        )
        lease_token = account.get("lease_token")

        scrape_result: ScrapeRunResult = await run_keyword_search_async(
            account,
            keyword,
            max_items=max_items,
            max_scrolls=max_scrolls,
            data_dir=data_dir,
            batch_id=batch_id,
        )

        if not scrape_result.success:
            err = scrape_result.error
            return _error_payload(
                err.code if err else "UNKNOWN_ERROR",
                err.message if err else "未知错误",
                browser_started=err.browser_started if err else True,
                stage=err.stage if err else "rpa",
            )

        collected_count = len(scrape_result.items)
        return _success_payload(
            keyword=keyword,
            collected_count=collected_count,
            items=scrape_result.items[:10],
            stop_reason=scrape_result.stop_reason,
        )
    except AccountManagerError as exc:
        return _error_payload(
            exc.code,
            exc.message,
            browser_started=False,
            stage="account_prepare",
        )
    finally:
        release_lease(lease_token)


def run_keyword_search_task_sync(
    keyword: str,
    *,
    max_items: int = DEFAULT_MAX_ITEMS,
    max_scrolls: int = DEFAULT_MAX_SCROLLS,
    account_id: Optional[str] = None,
    platform: str = TARGET_PLATFORM,
) -> dict[str, Any]:
    return asyncio.run(
        run_keyword_search_task(
            keyword,
            max_items=max_items,
            max_scrolls=max_scrolls,
            account_id=account_id,
            platform=platform,
        )
    )