wuyang6
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
from product_data_portable import (
|
|
DEFAULT_REQUEST_ID,
|
|
DEFAULT_TIMESTAMP_STEP_MS,
|
|
emit_error,
|
|
emit_success,
|
|
load_json_payload,
|
|
normalize_dataset_draft,
|
|
resolve_portable_path,
|
|
)
|
|
|
|
|
|
def main() -> int:
|
|
try:
|
|
payload = load_json_payload()
|
|
draft_text = str(payload.get("draft_text") or "")
|
|
if not draft_text and payload.get("draft_path"):
|
|
draft_text = resolve_portable_path(str(payload["draft_path"])).read_text(encoding="utf-8")
|
|
source_type = str(payload.get("source_type") or "generated")
|
|
confirmed_plan_id = str(payload.get("confirmed_plan_id") or "").strip()
|
|
if source_type == "generated" and not confirmed_plan_id:
|
|
raise ValueError("confirmed_plan_id is required for generated data")
|
|
result = normalize_dataset_draft(
|
|
draft_text,
|
|
batch_id=str(payload.get("batch_id") or DEFAULT_REQUEST_ID),
|
|
source_type=source_type,
|
|
base_timestamp=payload.get("base_timestamp"),
|
|
timestamp_step_ms=int(payload.get("timestamp_step_ms") or DEFAULT_TIMESTAMP_STEP_MS),
|
|
default_request_id=str(payload.get("default_request_id") or DEFAULT_REQUEST_ID),
|
|
)
|
|
if confirmed_plan_id:
|
|
result["confirmed_plan_id"] = confirmed_plan_id
|
|
emit_success(result)
|
|
return 0
|
|
except Exception as exc: # noqa: BLE001 - CLI 需要把错误稳定转成 JSON
|
|
emit_error(exc)
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|