98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
"""任务编排、日志查询模板。
|
||
|
||
通用业务编排层：负责参数校验、鉴权、调用具体业务实现、写入任务日志。
|
||
复制后在 cmd_run 中实现真正的业务逻辑。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import sys
|
||
from typing import Optional
|
||
|
||
from db import task_logs_repository as tlr
|
||
from service.entitlement_service import check_entitlement
|
||
from util.constants import SKILL_SLUG, SKILL_VERSION
|
||
from util.timeutil import unix_to_iso
|
||
|
||
|
||
def cmd_run(target: Optional[str] = None, input_id: Optional[str] = None) -> int:
|
||
"""通用任务执行入口模板。复制后请实现真实业务逻辑。"""
|
||
_ = (target, input_id)
|
||
ok, reason = check_entitlement(SKILL_SLUG)
|
||
if not ok:
|
||
print(f"❌ {reason}")
|
||
return 1
|
||
print("❌ 这是模板仓库，请复制后在 scripts/service/task_service.py 中实现 cmd_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_health() -> int:
|
||
return 0 if sys.version_info >= (3, 10) else 1
|
||
|
||
|
||
def cmd_version() -> int:
|
||
print(json.dumps({"version": SKILL_VERSION, "skill": SKILL_SLUG}, ensure_ascii=False))
|
||
return 0
|