"""任务编排、日志查询模板。 编排层:job_context / emit / checkpoint / finish、鉴权、任务日志与错误转换。 领域业务请放在 ``service/_service.py``,由本模块调用;多入口(CLI/Action/cron)复用同一内核。 复制后在 cmd_run(或 cmd_sync_*)中接线到真实 domain service,勿按 toolbar/cron/agent 分叉业务。 """ from __future__ import annotations import asyncio import json import os import uuid from typing import Any, Dict, Optional from jiangchang_skill_core import collect_runtime_diagnostics, config, format_runtime_health_lines from jiangchang_skill_core.activity import JobStopped, emit, finish, job_context from db import task_logs_repository as tlr from jiangchang_skill_core.rpa.video_session import RpaVideoSession from service.entitlement_service import check_entitlement from service.task_run_support import ( _print_video_summary, build_video_info, merge_video_into_result_summary, ) from util.constants import LOG_LOGGER_NAME, PLATFORM_KIT_MIN_VERSION, SKILL_SLUG, SKILL_VERSION from util.logging_config import get_skill_logger, setup_skill_logging from util.runtime_paths import get_skill_data_dir, get_skill_root from util.timeutil import unix_to_iso def _get_task_logger(): try: return get_skill_logger() except RuntimeError: setup_skill_logging(SKILL_SLUG, LOG_LOGGER_NAME) return get_skill_logger() async def _run_template_demo(target: Optional[str], input_id: Optional[str]) -> tuple[int, dict[str, Any]]: """最小 RpaVideoSession 示范:不启动浏览器、不执行业务,仅演示录屏包裹范式。""" batch_id = uuid.uuid4().hex[:12] data_dir = get_skill_data_dir() async with RpaVideoSession( skill_slug=SKILL_SLUG, skill_data_dir=data_dir, batch_id=batch_id, title="开始执行示例任务", closing_title="示例任务执行完成", ) as video: video.add_step("准备执行示例任务") if target: video.add_step(f"接收目标参数:{target}") if input_id: video.add_step(f"接收输入标识:{input_id}") video.add_step("示例任务执行完成") video_summary = video.summary() video_info = build_video_info(video_summary, {}, {}) return 1, video_info def cmd_run(target: Optional[str] = None, input_id: Optional[str] = None) -> int: """通用任务执行入口模板。复制后请实现真实业务逻辑。""" log = _get_task_logger() task_type = "demo" with job_context(skill=SKILL_SLUG): log.info( "task_start task_type=%s target_id=%s input_id=%s", task_type, target, input_id, ) emit("开始处理任务", skill=SKILL_SLUG, stage="run") try: ok, reason = check_entitlement(SKILL_SLUG) if not ok: log.warning("entitlement_failed task_type=%s reason=%s", task_type, reason) emit("鉴权失败,已记录诊断信息", type="warn", skill=SKILL_SLUG, stage="auth") tlr.save_task_log( task_type=task_type, target_id=target, input_id=input_id, input_title="模板示例任务", status="failed", error_msg=reason, result_summary=json.dumps({"stage": "auth", "error": reason}, ensure_ascii=False), ) log.info( "task_log_saved task_type=%s target_id=%s input_id=%s stage=auth status=failed", task_type, target, input_id, ) finish( status="failed", message=reason, skill=SKILL_SLUG, error_code="ENTITLEMENT_DENIED", stage="auth", ) print(f"❌ {reason}") return 1 log.info("task_run_demo_start target_id=%s input_id=%s", target, input_id) rc, video_info = asyncio.run(_run_template_demo(target, input_id)) log.info("task_run_demo_done target_id=%s input_id=%s status=failed", target, input_id) _print_video_summary(video_info) summary_payload = merge_video_into_result_summary( { "template_demo": True, "target": target, "input_id": input_id, }, video_info, ) tlr.save_task_log( task_type=task_type, target_id=target, input_id=input_id, input_title="模板示例任务", status="failed", error_msg="模板仓库未实现真实业务", result_summary=json.dumps(summary_payload, ensure_ascii=False), ) log.info( "task_log_saved task_type=%s target_id=%s input_id=%s status=failed", task_type, target, input_id, ) finish( status="failed", message="模板仓库未实现真实业务", skill=SKILL_SLUG, template_demo=True, target=target, input_id=input_id, ) print("❌ 这是模板仓库,请复制后在 scripts/service/task_service.py 中实现 cmd_run 的真实业务逻辑。") return rc except JobStopped: raise except Exception: log.exception( "task_failed task_type=%s target_id=%s input_id=%s", task_type, target, input_id, ) emit("任务失败,已记录诊断信息", type="warn", skill=SKILL_SLUG, stage="run") try: tlr.save_task_log( task_type=task_type, target_id=target, input_id=input_id, input_title="模板示例任务", status="failed", error_msg="任务执行异常,详见统一日志", result_summary=json.dumps( {"stage": "run", "error": "unexpected_exception"}, ensure_ascii=False, ), ) log.info( "task_log_saved task_type=%s target_id=%s input_id=%s stage=run status=failed", task_type, target, input_id, ) except Exception: log.exception( "task_log_save_failed task_type=%s target_id=%s input_id=%s", task_type, target, input_id, ) finish( status="failed", message="任务执行异常,详见统一日志", skill=SKILL_SLUG, stage="run", ) return 1 def cmd_logs( limit: int = 10, status: Optional[str] = None, task_type: Optional[str] = None, target_id: Optional[str] = None, ) -> int: if limit <= 0: limit = 10 rows = tlr.list_task_logs(limit, status, task_type, target_id) if not rows: print("暂无任务日志") return 0 sep_line = "_" * 39 for idx, r in enumerate(rows): rid, ttype, tid, iid, ititle, st, err, rsum, cat, uat = r print(f"id:{rid}") print(f"task_type:{ttype or ''}") print(f"target_id:{tid or ''}") print(f"input_id:{iid or ''}") print(f"input_title:{ititle or ''}") print(f"status:{st or ''}") print(f"error_msg:{err or ''}") print(f"result_summary:{rsum or ''}") print(f"created_at:{unix_to_iso(cat) or str(cat or '')}") print(f"updated_at:{unix_to_iso(uat) or str(uat or '')}") if idx != len(rows) - 1: print(sep_line) print() return 0 def cmd_log_get(log_id: str) -> int: if not str(log_id).isdigit(): print("❌ log_id 必须是数字") return 1 row = tlr.get_task_log_by_id(int(log_id)) if not row: print("❌ 没有这条任务日志") return 1 rid, ttype, tid, iid, ititle, st, err, rsum, cat, uat = row print( json.dumps( { "id": int(rid), "task_type": ttype, "target_id": tid, "input_id": iid, "input_title": ititle, "status": st, "error_msg": err, "result_summary": rsum, "created_at": unix_to_iso(cat), "updated_at": unix_to_iso(uat), }, ensure_ascii=False, ) ) return 0 def cmd_config_path() -> int: example_path = os.path.join(get_skill_root(), ".env.example") env_path = config.get_env_file_path() or "" print( json.dumps( { "skill": SKILL_SLUG, "env_path": env_path, "example_path": os.path.abspath(example_path), }, ensure_ascii=False, ) ) return 0 def cmd_health() -> int: runtime = collect_runtime_diagnostics( skill_slug=SKILL_SLUG, platform_kit_min_version=PLATFORM_KIT_MIN_VERSION, skill_root=get_skill_root(), ) example_path = os.path.join(get_skill_root(), ".env.example") env_path = config.get_env_file_path() or "" env_exists = bool(env_path and os.path.isfile(env_path)) health_status = "failed" if runtime.has_fatal_issues else "ok" lines = [ f"{SKILL_SLUG} health: {health_status}", *format_runtime_health_lines(runtime), f"env_path: {env_path}", f"env_exists: {env_exists}", f"example_path: {os.path.abspath(example_path)}", ] for line in lines: print(line) return 1 if runtime.has_fatal_issues else 0 def cmd_version() -> int: print(json.dumps({"version": SKILL_VERSION, "skill": SKILL_SLUG}, ensure_ascii=False)) return 0