143 lines
4.1 KiB
Python
143 lines
4.1 KiB
Python
"""返利班结果：查号 / 送审 → 表 4 更新与插入、表 3 状态。"""
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass, field
|
|
|
|
KIND_OK = "ok"
|
|
KIND_FAIL = "fail"
|
|
KIND_NOT_FOUND = "not_found"
|
|
STAMP_CONTRACT = "合同班"
|
|
STAMP_CLAIMING = "查号中"
|
|
STAMP_DONE = "返利班"
|
|
PROCESS_RESULT_MAX = 1000
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class AgreementOutcome:
|
|
kind: str
|
|
agreement_id: str | None
|
|
agreement_dtl_id: str | None
|
|
reason: str | None
|
|
|
|
|
|
@dataclass
|
|
class RebateGroup:
|
|
contract_no: str
|
|
drafts: list[dict] = field(default_factory=list)
|
|
protocols: list[dict] = field(default_factory=list)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class GroupSave:
|
|
table4_updates: list[dict]
|
|
table4_inserts: list[dict]
|
|
table4_release_fids: list[int]
|
|
table3_updates: list[dict]
|
|
|
|
|
|
def clip(text: str, max_len: int) -> str:
|
|
text = (text or "").strip()
|
|
if len(text) <= max_len:
|
|
return text
|
|
return text[: max_len - 1] + "…"
|
|
|
|
|
|
def agreement_ok(outcome: AgreementOutcome) -> bool:
|
|
return outcome.kind == KIND_OK and bool((outcome.agreement_id or "").strip())
|
|
|
|
|
|
def build_not_found(group: RebateGroup) -> GroupSave:
|
|
return GroupSave(
|
|
table4_updates=[],
|
|
table4_inserts=[],
|
|
table4_release_fids=[int(row["fid"]) for row in group.drafts],
|
|
table3_updates=[
|
|
{"fid": int(row["fid"]), "process_status": 0, "process_result": ""}
|
|
for row in group.protocols
|
|
],
|
|
)
|
|
|
|
|
|
def build_found(group: RebateGroup, outcome: AgreementOutcome) -> GroupSave:
|
|
ok = agreement_ok(outcome)
|
|
number = (group.contract_no or "").strip()
|
|
updates = [
|
|
{
|
|
"fid": int(row["fid"]),
|
|
"contract_status": 2,
|
|
"update_user_name": STAMP_DONE,
|
|
}
|
|
for row in group.drafts
|
|
]
|
|
inserts: list[dict] = []
|
|
seen_protocol: dict[int, dict] = {}
|
|
for row in group.protocols:
|
|
seen_protocol.setdefault(int(row["oms_id"]), row)
|
|
for row in seen_protocol.values():
|
|
inserts.append(_action1_success(row, number))
|
|
if ok:
|
|
seen_oms: dict[int, dict] = {}
|
|
for row in group.drafts + group.protocols:
|
|
seen_oms.setdefault(int(row["oms_id"]), row)
|
|
for row in seen_oms.values():
|
|
inserts.append(_action2(row, number, outcome))
|
|
process_status = 1
|
|
process_result = "协议已录入"
|
|
else:
|
|
process_status = 2
|
|
process_result = clip(outcome.reason or "协议录入失败", PROCESS_RESULT_MAX)
|
|
table3_updates = [
|
|
{
|
|
"fid": int(row["fid"]),
|
|
"process_status": process_status,
|
|
"process_result": process_result,
|
|
}
|
|
for row in group.protocols
|
|
]
|
|
return GroupSave(
|
|
table4_updates=updates,
|
|
table4_inserts=inserts,
|
|
table4_release_fids=[],
|
|
table3_updates=table3_updates,
|
|
)
|
|
|
|
|
|
def _action1_success(row: dict, contract_no: str) -> dict:
|
|
return {
|
|
"oms_id": int(row["oms_id"]),
|
|
"action_code": 1,
|
|
"contract_status": 2,
|
|
"contract_fail_reason": None,
|
|
"contract_no": contract_no,
|
|
"agreement_status": None,
|
|
"agreement_fail_reason": None,
|
|
"agreement_id": None,
|
|
"agreement_dtl_id": None,
|
|
"push_status": 0,
|
|
"org_code": row.get("org_code"),
|
|
"create_user_id": 0,
|
|
"create_user_name": STAMP_DONE,
|
|
"update_user_id": 0,
|
|
"update_user_name": STAMP_DONE,
|
|
}
|
|
|
|
|
|
def _action2(row: dict, contract_no: str, outcome: AgreementOutcome) -> dict:
|
|
return {
|
|
"oms_id": int(row["oms_id"]),
|
|
"action_code": 2,
|
|
"contract_status": None,
|
|
"contract_fail_reason": None,
|
|
"contract_no": contract_no,
|
|
"agreement_status": 1,
|
|
"agreement_fail_reason": None,
|
|
"agreement_id": (outcome.agreement_id or "").strip() or None,
|
|
"agreement_dtl_id": (outcome.agreement_dtl_id or "").strip() or None,
|
|
"push_status": 0,
|
|
"org_code": row.get("org_code"),
|
|
"create_user_id": 0,
|
|
"create_user_name": STAMP_DONE,
|
|
"update_user_id": 0,
|
|
"update_user_name": STAMP_DONE,
|
|
}
|