149 lines
4.8 KiB
Python
149 lines
4.8 KiB
Python
# coding=utf-8
|
||
"""合同起草真实入口。Studio 编译不会改这个文件。"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import os
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
|
||
def _repo_root() -> Path:
|
||
for parent in Path(__file__).resolve().parents:
|
||
if (parent / "rpa_yisaiqi" / "__init__.py").exists():
|
||
return parent
|
||
raise RuntimeError("找不到仓库根目录（缺少 rpa_yisaiqi/__init__.py）")
|
||
|
||
|
||
_REPO_ROOT = _repo_root()
|
||
if str(_REPO_ROOT) not in sys.path:
|
||
sys.path.insert(0, str(_REPO_ROOT))
|
||
|
||
from rpa_yisaiqi.apa_browser_adapter import ApaBrowserAdapter # noqa: E402
|
||
from rpa_yisaiqi.callback_client import StateCenterClient # noqa: E402
|
||
from rpa_yisaiqi.contract_create_node import ( # noqa: E402
|
||
ContractCreateError,
|
||
build_spec_from_data,
|
||
load_input_data,
|
||
)
|
||
from rpa_yisaiqi.contract_flow_steps import ( # noqa: E402
|
||
enter_contract_create,
|
||
fill_contract_amount_and_dates,
|
||
fill_contract_basic_info,
|
||
fill_contract_goods_lines,
|
||
fill_contract_payment_info,
|
||
fill_contract_signatory_info,
|
||
login_contract_system,
|
||
open_contract_system,
|
||
read_contract_number,
|
||
select_contract_template,
|
||
submit_contract_for_approval,
|
||
upload_contract_document_if_needed,
|
||
)
|
||
|
||
|
||
def _first(*values) -> str:
|
||
for value in values:
|
||
if value is not None and str(value).strip():
|
||
return str(value).strip()
|
||
return ""
|
||
|
||
|
||
def _parse_input_arg(argv: list[str]) -> str:
|
||
i = 0
|
||
while i < len(argv):
|
||
arg = argv[i]
|
||
if arg in ("-i", "--input", "--input-json") and i + 1 < len(argv):
|
||
return argv[i + 1]
|
||
if arg.startswith("--input=") or arg.startswith("--input-json="):
|
||
return arg.split("=", 1)[1]
|
||
i += 1
|
||
return ""
|
||
|
||
|
||
def main() -> int:
|
||
print("合同起草流程启动", flush=True)
|
||
|
||
raw = _parse_input_arg(sys.argv[1:])
|
||
if not raw:
|
||
raw = os.getenv("CONTRACT_INPUT_JSON", "").strip()
|
||
if not raw:
|
||
raw = str(_REPO_ROOT / "rpa_yisaiqi" / "examples" / "contract_create_input_dry_run.json")
|
||
print("未传入 -i/--input，使用默认联调入参：", raw, flush=True)
|
||
|
||
print("[S01 初始化]", flush=True)
|
||
data = load_input_data(raw)
|
||
state_center_url = _first(
|
||
data.get("stateCenterUrl"),
|
||
data.get("stateCenterURL"),
|
||
os.getenv("STATE_CENTER_URL"),
|
||
"http://192.168.105.69:8000",
|
||
)
|
||
agent_token = _first(data.get("agentToken"), os.getenv("AGENT_TOKEN"), "dev-agent-token")
|
||
spec = build_spec_from_data(data)
|
||
ctx = spec.flow_context
|
||
if ctx is None:
|
||
raise ContractCreateError("初始化失败：flow_context 为空")
|
||
client = StateCenterClient(state_center_url, agent_token)
|
||
browser = ApaBrowserAdapter(url=ctx.contract_url, flow="CONTRACT_CREATE")
|
||
is_dry_run = spec.submit_mode == "dry_run"
|
||
print(" jobId=", spec.job_id, " submitMode=", spec.submit_mode, flush=True)
|
||
print(" contractDocumentPath=", ctx.contract_document_path or ctx.merged_pdf_path, flush=True)
|
||
|
||
if not is_dry_run:
|
||
print("回调状态中心 start", spec.job_id, flush=True)
|
||
client.start_node(spec.job_id, "CONTRACT_CREATE")
|
||
|
||
print("[S02 打开合同系统]", flush=True)
|
||
open_contract_system(browser, ctx)
|
||
|
||
print("[S03 登录合同系统]", flush=True)
|
||
login_contract_system(browser, ctx)
|
||
|
||
print("[S04 进入合同起草]", flush=True)
|
||
enter_contract_create(browser)
|
||
|
||
print("[S05 选择模板]", flush=True)
|
||
select_contract_template(browser, ctx)
|
||
|
||
print("[S06 填写签约方]", flush=True)
|
||
fill_contract_signatory_info(browser, ctx)
|
||
|
||
print("[S07 填写基础信息]", flush=True)
|
||
fill_contract_basic_info(browser, ctx)
|
||
|
||
print("[S08 填写金额日期]", flush=True)
|
||
fill_contract_amount_and_dates(browser, ctx)
|
||
|
||
print("[S09 填写货品明细]", flush=True)
|
||
fill_contract_goods_lines(browser, ctx)
|
||
|
||
print("[S10 填写付款信息]", flush=True)
|
||
fill_contract_payment_info(browser, ctx)
|
||
|
||
print("[S11 上传合同PDF]", flush=True)
|
||
upload_contract_document_if_needed(browser, ctx)
|
||
|
||
print("[S12 提交送审]", flush=True)
|
||
submitted = submit_contract_for_approval(browser, ctx)
|
||
|
||
contract_no = ""
|
||
if submitted:
|
||
print("[S13 读取合同编号]", flush=True)
|
||
contract_no = read_contract_number(browser, ctx)
|
||
|
||
print("[S14 结束]", flush=True)
|
||
if not is_dry_run and contract_no:
|
||
print("回调状态中心 complete", spec.job_id, contract_no, flush=True)
|
||
client.complete_node(
|
||
spec.job_id,
|
||
"CONTRACT_CREATE",
|
||
{"businessRefs": {"contractSystemNo": contract_no}},
|
||
)
|
||
print(json.dumps({"status": "done", "contractSystemNo": contract_no}, ensure_ascii=False), flush=True)
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|