135 lines
4.3 KiB
Python
135 lines
4.3 KiB
Python
"""样本保险办理 CRUD。"""
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.core.utils import generate_uuid
|
|
from app.models.models import BizSampleInsurance
|
|
from app.services.sample_info_progress import apply_handle_payload, insurance_handled
|
|
|
|
|
|
def _fmt_dt(t: Optional[datetime]) -> str:
|
|
return t.strftime("%Y-%m-%d %H:%M:%S") if t else ""
|
|
|
|
|
|
def get_by_sample_code(
|
|
db: Session, tenant_no: str, sample_code: str
|
|
) -> Optional[BizSampleInsurance]:
|
|
code = (sample_code or "").strip()
|
|
if not code:
|
|
return None
|
|
return (
|
|
db.query(BizSampleInsurance)
|
|
.filter(
|
|
BizSampleInsurance.tenant_no == tenant_no,
|
|
BizSampleInsurance.sample_code == code,
|
|
BizSampleInsurance.del_flag == 0,
|
|
)
|
|
.first()
|
|
)
|
|
|
|
|
|
def get_map_by_sample_codes(
|
|
db: Session, tenant_no: str, codes: List[str]
|
|
) -> Dict[str, BizSampleInsurance]:
|
|
cleaned = list({(c or "").strip() for c in codes if (c or "").strip()})
|
|
if not cleaned:
|
|
return {}
|
|
rows = (
|
|
db.query(BizSampleInsurance)
|
|
.filter(
|
|
BizSampleInsurance.tenant_no == tenant_no,
|
|
BizSampleInsurance.sample_code.in_(cleaned),
|
|
BizSampleInsurance.del_flag == 0,
|
|
)
|
|
.all()
|
|
)
|
|
return {row.sample_code: row for row in rows}
|
|
|
|
|
|
def upsert_handle(
|
|
db: Session,
|
|
*,
|
|
tenant_no: str,
|
|
sample_code: str,
|
|
collection_protocol_id: Optional[str],
|
|
user_id: Optional[str],
|
|
handle_time: Optional[datetime],
|
|
handler_user_id: Optional[str],
|
|
insurance_attachments: str = "",
|
|
handle_remark: str = "",
|
|
policy_send_time: Optional[datetime] = None,
|
|
update_policy_send_time: bool = False,
|
|
) -> BizSampleInsurance:
|
|
code = (sample_code or "").strip()
|
|
if not code:
|
|
raise ValueError("sample_code is required")
|
|
|
|
payload = apply_handle_payload(
|
|
handle_time=handle_time,
|
|
handler_user_id=handler_user_id,
|
|
attachments=insurance_attachments or "",
|
|
remark=handle_remark or "",
|
|
)
|
|
now = datetime.now()
|
|
row = get_by_sample_code(db, tenant_no, code)
|
|
if row is None:
|
|
row = BizSampleInsurance(
|
|
id=generate_uuid(),
|
|
sample_code=code,
|
|
collection_protocol_id=(collection_protocol_id or "").strip() or None,
|
|
tenant_no=tenant_no,
|
|
order_num=0,
|
|
todo_status="0",
|
|
create_by=user_id,
|
|
create_dept_id="",
|
|
create_time=now,
|
|
update_by=user_id,
|
|
update_time=now,
|
|
del_flag=0,
|
|
)
|
|
db.add(row)
|
|
else:
|
|
if collection_protocol_id and not row.collection_protocol_id:
|
|
row.collection_protocol_id = (collection_protocol_id or "").strip() or None
|
|
row.update_by = user_id
|
|
row.update_time = now
|
|
|
|
row.handle_time = payload["handle_time"]
|
|
row.handler_user_id = payload["handler_user_id"]
|
|
row.todo_status = payload["make_or_todo_status"]
|
|
row.handle_date = payload["legacy_date"] or None
|
|
row.insurance_attachments = payload["attachments"] or None
|
|
row.handle_remark = payload["remark"] or None
|
|
if update_policy_send_time:
|
|
row.policy_send_time = policy_send_time
|
|
|
|
db.flush()
|
|
return row
|
|
|
|
|
|
def row_to_dict(row: BizSampleInsurance) -> Dict[str, Any]:
|
|
return {
|
|
"id": row.id,
|
|
"sampleCode": row.sample_code or "",
|
|
"collectionProtocolId": row.collection_protocol_id or "",
|
|
"todoStatus": row.todo_status or "0",
|
|
"handleDate": (row.handle_date or "").strip() if row.handle_date else "",
|
|
"handleTime": _fmt_dt(row.handle_time) if getattr(row, "handle_time", None) else "",
|
|
"handlerUserId": getattr(row, "handler_user_id", None) or "",
|
|
"handleRemark": row.handle_remark or "",
|
|
"insuranceAttachments": row.insurance_attachments or "",
|
|
"policySendTime": _fmt_dt(row.policy_send_time)
|
|
if getattr(row, "policy_send_time", None)
|
|
else "",
|
|
"insuranceHandled": insurance_handled(row),
|
|
"tenantNo": row.tenant_no,
|
|
"orderNum": row.order_num if row.order_num is not None else 0,
|
|
"remark": row.remark or "",
|
|
"createTime": _fmt_dt(row.create_time),
|
|
"updateTime": _fmt_dt(row.update_time),
|
|
}
|