collection_protocol.py 51 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374
  1. """细胞存储 — 采集协议单 / 审核 / 财务确认 / 分期预警 API"""
  2. from datetime import datetime, timedelta
  3. from typing import Any, Dict, List, Optional
  4. from fastapi import APIRouter, Depends, HTTPException, Query
  5. from pydantic import BaseModel, ConfigDict, Field
  6. from sqlalchemy.orm import Session
  7. from sqlalchemy import and_ as sa_and, func as sa_func
  8. from app.core.database import get_db
  9. from app.core.data_scope import get_user_dept_id
  10. from app.core.menu_guard import require_admin_or_super_admin
  11. from app.core.response import error_response, success_response
  12. from app.core.utils import parse_datetime
  13. from app.crud import collection_protocol as cp_svc
  14. from app.models.models import (
  15. BizCollectionProtocol,
  16. BizProtocolPartyCustomer,
  17. BizProtocolSample,
  18. BizProtocolStorageCert,
  19. SysUser,
  20. )
  21. from app.services.party_customer_sync import sync_party_customers
  22. router = APIRouter(prefix="/web/admin", tags=["细胞存储"])
  23. _PAYMENT_DETAIL_SENSITIVE_FIELDS = {
  24. "pay_status",
  25. "payStatus",
  26. "finance_confirm_attachments",
  27. "financeConfirmAttachments",
  28. "finance_confirm_attachment_files",
  29. "financeConfirmAttachmentFiles",
  30. "finance_remark",
  31. "financeRemark",
  32. "refund_amount",
  33. "refundAmount",
  34. "refund_remark",
  35. "refundRemark",
  36. "refund_attachments",
  37. "refundAttachments",
  38. "refund_attachment_files",
  39. "refundAttachmentFiles",
  40. "refund_finance_confirm_attachments",
  41. "refundFinanceConfirmAttachments",
  42. "refund_finance_confirm_attachment_files",
  43. "refundFinanceConfirmAttachmentFiles",
  44. "refund_finance_remark",
  45. "refundFinanceRemark",
  46. }
  47. def _get_tenant_no(db: Session, user: dict) -> str:
  48. user_id = user.get("user_id")
  49. if not user_id:
  50. raise PermissionError("用户身份信息缺失,无法确定租户")
  51. sys_user = db.query(SysUser).filter(SysUser.id == user_id).first()
  52. if not sys_user or not sys_user.tenant_no:
  53. raise PermissionError("用户租户信息缺失")
  54. return sys_user.tenant_no
  55. def _get_current_user_id(user: dict) -> str:
  56. user_id = user.get("user_id")
  57. if not user_id:
  58. raise PermissionError("用户身份信息缺失")
  59. return str(user_id)
  60. def _protocol_accessible(
  61. db: Session, protocol_id: str, tenant_no: str, user: dict
  62. ) -> bool:
  63. """与列表接口一致:按租户 + 角色数据权限判断是否可访问该协议单。"""
  64. from app.core.data_scope import apply_data_scope
  65. q = db.query(BizCollectionProtocol).filter(
  66. BizCollectionProtocol.id == protocol_id,
  67. BizCollectionProtocol.del_flag == 0,
  68. BizCollectionProtocol.tenant_no == tenant_no,
  69. )
  70. q = apply_data_scope(q, BizCollectionProtocol, user, db, tenant_no)
  71. return q.first() is not None
  72. _CAMEL_TO_SNAKE_DATETIME = {
  73. "newbornDeliveryTime": "newborn_delivery_time",
  74. "sampleCollectTime": "sample_collect_time",
  75. "maternalBloodCollectTime": "maternal_blood_collect_time",
  76. }
  77. def _parse_datetime_fields(d: dict, camel_keys: list) -> None:
  78. for camel in camel_keys:
  79. val = d.get(camel)
  80. if val is not None and isinstance(val, str):
  81. snake = _CAMEL_TO_SNAKE_DATETIME.get(camel, camel)
  82. parsed = parse_datetime(val)
  83. if parsed:
  84. d.pop(camel, None)
  85. d[snake] = parsed
  86. else:
  87. d.pop(camel, None)
  88. d[snake] = None
  89. # ---------------------------------------------------------------------------
  90. # 采集协议单
  91. # ---------------------------------------------------------------------------
  92. @router.get("/collection-protocols")
  93. async def list_collection_protocols(
  94. page: int = Query(1, ge=1),
  95. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  96. keyword: Optional[str] = None,
  97. protocolStatus: Optional[str] = Query(None, alias="protocolStatus"),
  98. createTimeStart: Optional[str] = Query(None, alias="createTimeStart"),
  99. createTimeEnd: Optional[str] = Query(None, alias="createTimeEnd"),
  100. sampleCollectTimeStart: Optional[str] = Query(None, alias="sampleCollectTimeStart"),
  101. sampleCollectTimeEnd: Optional[str] = Query(None, alias="sampleCollectTimeEnd"),
  102. user: dict = Depends(require_admin_or_super_admin),
  103. db: Session = Depends(get_db),
  104. ):
  105. tenant_no = _get_tenant_no(db, user)
  106. items, total = cp_svc.list_collection_protocols(
  107. db,
  108. page=page,
  109. page_size=pageSize,
  110. keyword=keyword,
  111. protocol_status=protocolStatus,
  112. create_time_start=createTimeStart,
  113. create_time_end=createTimeEnd,
  114. sample_collect_time_start=sampleCollectTimeStart,
  115. sample_collect_time_end=sampleCollectTimeEnd,
  116. tenant_no=tenant_no,
  117. user_info=user,
  118. )
  119. tp = cp_svc.total_pages(total, pageSize)
  120. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  121. @router.get("/collection-protocols/{protocol_id}")
  122. async def get_collection_protocol_detail(
  123. protocol_id: str,
  124. user: dict = Depends(require_admin_or_super_admin),
  125. db: Session = Depends(get_db),
  126. ):
  127. tenant_no = _get_tenant_no(db, user)
  128. if not _protocol_accessible(db, protocol_id, tenant_no, user):
  129. return error_response(404, "协议单不存在或无权查看")
  130. bundle = cp_svc.get_protocol_detail_bundle(db, protocol_id, tenant_no)
  131. if not bundle:
  132. return error_response(404, "协议单不存在或无权查看")
  133. return success_response(bundle)
  134. class ProtocolCreateBody(BaseModel):
  135. model_config = ConfigDict(populate_by_name=True)
  136. protocol_no: Optional[str] = Field(None, alias="protocolNo")
  137. protocol_no_id: Optional[str] = Field(None, alias="protocolNoId")
  138. protocol_status: str = Field("dsx", alias="protocolStatus")
  139. protocol_attachments: Optional[str] = Field(None, alias="protocolAttachments")
  140. consent_form_urls: Optional[str] = Field(None, alias="consentFormUrls")
  141. health_survey_form_urls: Optional[str] = Field(None, alias="healthSurveyFormUrls")
  142. register_date: Optional[str] = Field(None, alias="registerDate")
  143. collector_user_id: Optional[str] = Field(None, alias="collectorUserId")
  144. protocol_total_amount: Optional[float] = Field(None, alias="protocolTotalAmount")
  145. source_presign_order_id: Optional[str] = Field(None, alias="sourcePresignOrderId")
  146. payment_model_version: Optional[int] = Field(2, alias="paymentModelVersion")
  147. remark: Optional[str] = None
  148. party_customer: Optional[dict] = Field(None, alias="partyCustomer")
  149. collection_info: Optional[dict] = Field(None, alias="collectionInfo")
  150. samples: Optional[list] = Field(None, alias="samples")
  151. payment: Optional[dict] = Field(None, alias="payment")
  152. payment_details: Optional[list] = Field(None, alias="paymentDetails")
  153. sample_payments: Optional[list] = Field(None, alias="samplePayments")
  154. attachments: Optional[list] = Field(None, alias="attachments")
  155. consent_attachments: Optional[list] = Field(None, alias="consentAttachments")
  156. health_survey_attachments: Optional[list] = Field(None, alias="healthSurveyAttachments")
  157. storage_cert: Optional[dict] = Field(None, alias="storageCert")
  158. class ProtocolUpdateBody(BaseModel):
  159. model_config = ConfigDict(populate_by_name=True)
  160. protocol_no: Optional[str] = Field(None, alias="protocolNo")
  161. protocol_status: Optional[str] = Field(None, alias="protocolStatus")
  162. protocol_attachments: Optional[str] = Field(None, alias="protocolAttachments")
  163. consent_form_urls: Optional[str] = Field(None, alias="consentFormUrls")
  164. health_survey_form_urls: Optional[str] = Field(None, alias="healthSurveyFormUrls")
  165. register_date: Optional[str] = Field(None, alias="registerDate")
  166. effective_time: Optional[str] = Field(None, alias="effectiveTime")
  167. effective_desc: Optional[str] = Field(None, alias="effectiveDesc")
  168. expire_time: Optional[str] = Field(None, alias="expireTime")
  169. expire_desc: Optional[str] = Field(None, alias="expireDesc")
  170. void_time: Optional[str] = Field(None, alias="voidTime")
  171. void_desc: Optional[str] = Field(None, alias="voidDesc")
  172. void_type: Optional[str] = Field(None, alias="voidType")
  173. to_research_time: Optional[str] = Field(None, alias="toResearchTime")
  174. to_research_desc: Optional[str] = Field(None, alias="toResearchDesc")
  175. collector_user_id: Optional[str] = Field(None, alias="collectorUserId")
  176. protocol_total_amount: Optional[float] = Field(None, alias="protocolTotalAmount")
  177. source_presign_order_id: Optional[str] = Field(None, alias="sourcePresignOrderId")
  178. remark: Optional[str] = None
  179. party_customer: Optional[dict] = Field(None, alias="partyCustomer")
  180. collection_info: Optional[dict] = Field(None, alias="collectionInfo")
  181. samples: Optional[list] = Field(None, alias="samples")
  182. payment: Optional[dict] = Field(None, alias="payment")
  183. payment_details: Optional[list] = Field(None, alias="paymentDetails")
  184. sample_payments: Optional[list] = Field(None, alias="samplePayments")
  185. attachments: Optional[list] = Field(None, alias="attachments")
  186. consent_attachments: Optional[list] = Field(None, alias="consentAttachments")
  187. health_survey_attachments: Optional[list] = Field(None, alias="healthSurveyAttachments")
  188. storage_cert: Optional[dict] = Field(None, alias="storageCert")
  189. def assert_protocol_sensitive_fields(
  190. db: Session,
  191. user: dict,
  192. body: BaseModel,
  193. *,
  194. creating: bool,
  195. ) -> None:
  196. """防止通过普通协议保存接口注入财务确认或退费状态字段(不做菜单权限校验)。"""
  197. for detail in getattr(body, "payment_details", None) or []:
  198. normalized_detail = (
  199. cp_svc._normalize_fields(detail) if isinstance(detail, dict) else {}
  200. )
  201. if _PAYMENT_DETAIL_SENSITIVE_FIELDS.intersection(normalized_detail.keys()):
  202. raise HTTPException(
  203. status_code=400,
  204. detail="普通协议保存不允许提交财务确认或退费状态字段",
  205. )
  206. def _resolve_protocol_attachments(
  207. protocol_attachments: Optional[str],
  208. attachments: Optional[list],
  209. ) -> Optional[str]:
  210. from_urls = cp_svc.attachments_to_urls(attachments)
  211. if from_urls:
  212. return from_urls
  213. return protocol_attachments
  214. def _resolve_consent_form_urls(
  215. consent_form_urls: Optional[str],
  216. consent_attachments: Optional[list],
  217. ) -> Optional[str]:
  218. from_urls = cp_svc.attachments_to_urls(consent_attachments)
  219. if from_urls:
  220. return from_urls
  221. return consent_form_urls
  222. def _resolve_health_survey_form_urls(
  223. health_survey_form_urls: Optional[str],
  224. health_survey_attachments: Optional[list],
  225. ) -> Optional[str]:
  226. from_urls = cp_svc.attachments_to_urls(health_survey_attachments)
  227. if from_urls:
  228. return from_urls
  229. return health_survey_form_urls
  230. def _persist_protocol_sub_resources(
  231. db: Session,
  232. protocol_id: str,
  233. tenant_no: str,
  234. user_id: str,
  235. *,
  236. party_customer: Optional[dict],
  237. collection_info: Optional[dict],
  238. samples: Optional[list],
  239. payment: Optional[dict],
  240. payment_details: Optional[list],
  241. sample_payments: Optional[list] = None,
  242. storage_cert: Optional[dict],
  243. payment_model_version: int = 2,
  244. ) -> None:
  245. """持久化协议子资源。
  246. version=1:缴费(含明细校验)须在样本之前,避免校验失败后已占用样本编码。
  247. version=2:样本先落库占码,再按 sample_code 挂样本缴费方案。
  248. """
  249. if party_customer:
  250. _parse_datetime_fields(party_customer, ["newbornDeliveryTime"])
  251. party_customer = sync_party_customers(
  252. db, party_customer, tenant_no=tenant_no, user_id=user_id
  253. )
  254. cp_svc.upsert_party_customer(db, protocol_id, tenant_no, user_id, **party_customer)
  255. if collection_info:
  256. _parse_datetime_fields(collection_info, ["sampleCollectTime", "maternalBloodCollectTime"])
  257. cp_svc.upsert_collection_info(db, protocol_id, tenant_no, user_id, **collection_info)
  258. version = int(payment_model_version or 2)
  259. if version >= 2:
  260. if samples is not None:
  261. cp_svc.save_samples(db, protocol_id, tenant_no, user_id, samples)
  262. if sample_payments is not None:
  263. from app.crud import protocol_sample_payment as sp_pay_svc
  264. sp_pay_svc.save_sample_payments_for_protocol(
  265. db,
  266. collection_protocol_id=protocol_id,
  267. tenant_no=tenant_no,
  268. user_id=user_id,
  269. sample_payments=sample_payments,
  270. )
  271. else:
  272. # 旧模型:缴费先于样本
  273. if payment or payment_details is not None:
  274. cp_svc.save_payment_with_details(
  275. db,
  276. protocol_id,
  277. tenant_no,
  278. user_id,
  279. payment=payment,
  280. payment_details=payment_details,
  281. )
  282. if samples is not None:
  283. cp_svc.save_samples(db, protocol_id, tenant_no, user_id, samples)
  284. if storage_cert:
  285. cp_svc.upsert_storage_cert(db, protocol_id, tenant_no, user_id, **storage_cert)
  286. @router.post("/collection-protocols")
  287. async def create_collection_protocol(
  288. body: ProtocolCreateBody,
  289. user: dict = Depends(require_admin_or_super_admin),
  290. db: Session = Depends(get_db),
  291. ):
  292. assert_protocol_sensitive_fields(db, user, body, creating=True)
  293. tenant_no = _get_tenant_no(db, user)
  294. user_id = _get_current_user_id(user)
  295. protocol_id: Optional[str] = None
  296. try:
  297. protocol_attachments = _resolve_protocol_attachments(
  298. body.protocol_attachments,
  299. body.attachments,
  300. )
  301. consent_form_urls = _resolve_consent_form_urls(
  302. body.consent_form_urls,
  303. body.consent_attachments,
  304. )
  305. health_survey_form_urls = _resolve_health_survey_form_urls(
  306. body.health_survey_form_urls,
  307. body.health_survey_attachments,
  308. )
  309. collector = (body.collector_user_id or "").strip()
  310. if not collector:
  311. collector = user_id
  312. obj = cp_svc.create_collection_protocol(
  313. db,
  314. create_dept_id=get_user_dept_id(db, user_id) or "",
  315. protocol_no=body.protocol_no or "",
  316. protocol_no_id=body.protocol_no_id,
  317. protocol_status=body.protocol_status,
  318. protocol_attachments=protocol_attachments,
  319. consent_form_urls=consent_form_urls,
  320. health_survey_form_urls=health_survey_form_urls,
  321. register_date=body.register_date,
  322. collector_user_id=collector,
  323. protocol_total_amount=body.protocol_total_amount,
  324. source_presign_order_id=body.source_presign_order_id,
  325. payment_model_version=body.payment_model_version
  326. if body.payment_model_version is not None
  327. else 2,
  328. remark=body.remark,
  329. tenant_no=tenant_no,
  330. user_id=user_id,
  331. )
  332. protocol_id = obj.id
  333. _persist_protocol_sub_resources(
  334. db,
  335. protocol_id,
  336. tenant_no,
  337. user_id,
  338. party_customer=body.party_customer,
  339. collection_info=body.collection_info,
  340. samples=body.samples,
  341. payment=body.payment,
  342. payment_details=body.payment_details,
  343. sample_payments=body.sample_payments,
  344. storage_cert=body.storage_cert,
  345. payment_model_version=int(
  346. getattr(obj, "payment_model_version", None) or 2
  347. ),
  348. )
  349. return success_response(cp_svc._protocol_to_dict(obj), message="创建成功")
  350. except ValueError as e:
  351. # 子表分段 commit 后校验失败时,补偿软删主单及样本,释放样本编码占用
  352. if protocol_id:
  353. try:
  354. cp_svc.soft_delete_collection_protocol(
  355. db,
  356. protocol_id,
  357. tenant_no=tenant_no,
  358. user_id=user_id,
  359. )
  360. except Exception:
  361. pass
  362. return error_response(400, str(e))
  363. @router.put("/collection-protocols/{protocol_id}")
  364. async def update_collection_protocol(
  365. protocol_id: str,
  366. body: ProtocolUpdateBody,
  367. user: dict = Depends(require_admin_or_super_admin),
  368. db: Session = Depends(get_db),
  369. ):
  370. assert_protocol_sensitive_fields(db, user, body, creating=False)
  371. tenant_no = _get_tenant_no(db, user)
  372. user_id = _get_current_user_id(user)
  373. if not _protocol_accessible(db, protocol_id, tenant_no, user):
  374. return error_response(404, "协议单不存在或无权操作")
  375. try:
  376. payload = body.model_dump(exclude_unset=True, by_alias=False)
  377. sub_keys = {
  378. "party_customer", "collection_info", "samples",
  379. "payment", "payment_details", "sample_payments", "attachments",
  380. "consent_attachments", "health_survey_attachments", "storage_cert",
  381. }
  382. main_payload = {k: v for k, v in payload.items() if k not in sub_keys}
  383. if body.attachments is not None:
  384. resolved = _resolve_protocol_attachments(
  385. main_payload.get("protocol_attachments"),
  386. body.attachments,
  387. )
  388. if resolved is not None:
  389. main_payload["protocol_attachments"] = resolved
  390. if body.consent_attachments is not None:
  391. resolved = _resolve_consent_form_urls(
  392. main_payload.get("consent_form_urls"),
  393. body.consent_attachments,
  394. )
  395. if resolved is not None:
  396. main_payload["consent_form_urls"] = resolved
  397. if body.health_survey_attachments is not None:
  398. resolved = _resolve_health_survey_form_urls(
  399. main_payload.get("health_survey_form_urls"),
  400. body.health_survey_attachments,
  401. )
  402. if resolved is not None:
  403. main_payload["health_survey_form_urls"] = resolved
  404. obj = cp_svc.update_collection_protocol(
  405. db,
  406. protocol_id,
  407. tenant_no=tenant_no,
  408. user_id=user_id,
  409. **{
  410. k: v
  411. for k, v in main_payload.items()
  412. if v is not None or k == "protocol_total_amount"
  413. },
  414. )
  415. _persist_protocol_sub_resources(
  416. db,
  417. protocol_id,
  418. tenant_no,
  419. user_id,
  420. party_customer=body.party_customer,
  421. collection_info=body.collection_info,
  422. samples=body.samples,
  423. payment=body.payment,
  424. payment_details=body.payment_details,
  425. sample_payments=body.sample_payments,
  426. storage_cert=body.storage_cert,
  427. payment_model_version=int(
  428. getattr(obj, "payment_model_version", None) or 1
  429. ),
  430. )
  431. return success_response(cp_svc._protocol_to_dict(obj), message="更新成功")
  432. except ValueError as e:
  433. return error_response(400, str(e))
  434. @router.delete("/collection-protocols/{protocol_id}")
  435. async def delete_collection_protocol(
  436. protocol_id: str,
  437. user: dict = Depends(require_admin_or_super_admin),
  438. db: Session = Depends(get_db),
  439. ):
  440. tenant_no = _get_tenant_no(db, user)
  441. user_id = _get_current_user_id(user)
  442. if not _protocol_accessible(db, protocol_id, tenant_no, user):
  443. return error_response(404, "协议单不存在或无权操作")
  444. try:
  445. cp_svc.soft_delete_collection_protocol(
  446. db,
  447. protocol_id,
  448. tenant_no=tenant_no,
  449. user_id=user_id,
  450. )
  451. return success_response(None, message="删除成功")
  452. except ValueError as e:
  453. return error_response(400, str(e))
  454. # ---------------------------------------------------------------------------
  455. # 审核
  456. # ---------------------------------------------------------------------------
  457. @router.get("/collection-protocol-audits")
  458. async def list_collection_protocol_audits(
  459. page: int = Query(1, ge=1),
  460. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  461. keyword: Optional[str] = None,
  462. auditStatus: Optional[str] = Query(None, alias="auditStatus"),
  463. collectionProtocolId: Optional[str] = Query(None, alias="collectionProtocolId"),
  464. user: dict = Depends(require_admin_or_super_admin),
  465. db: Session = Depends(get_db),
  466. ):
  467. tenant_no = _get_tenant_no(db, user)
  468. items, total = cp_svc.list_audits(
  469. db,
  470. page=page,
  471. page_size=pageSize,
  472. keyword=keyword,
  473. audit_status=auditStatus,
  474. collection_protocol_id=collectionProtocolId,
  475. tenant_no=tenant_no,
  476. user_info=user,
  477. )
  478. tp = cp_svc.total_pages(total, pageSize)
  479. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  480. class AuditCreateBody(BaseModel):
  481. model_config = ConfigDict(populate_by_name=True)
  482. audit_status: str = Field(..., alias="auditStatus")
  483. audit_remark: Optional[str] = Field(None, alias="auditRemark")
  484. @router.post("/collection-protocols/{protocol_id}/audit")
  485. async def submit_collection_protocol_audit(
  486. protocol_id: str,
  487. body: AuditCreateBody,
  488. user: dict = Depends(require_admin_or_super_admin),
  489. db: Session = Depends(get_db),
  490. ):
  491. tenant_no = _get_tenant_no(db, user)
  492. if not _protocol_accessible(db, protocol_id, tenant_no, user):
  493. return error_response(403, "无权访问该协议或协议不存在")
  494. try:
  495. cp_svc.create_audit(
  496. db,
  497. collection_protocol_id=protocol_id,
  498. audit_status=body.audit_status,
  499. audit_remark=body.audit_remark,
  500. tenant_no=tenant_no,
  501. user_id=user.get("user_id"),
  502. )
  503. p = db.query(BizCollectionProtocol).filter(BizCollectionProtocol.id == protocol_id).first()
  504. pn = (p.protocol_no if p else "") or ""
  505. return success_response({"protocolNo": pn}, message="审核已提交")
  506. except ValueError as e:
  507. return error_response(400, str(e))
  508. @router.get("/collection-protocols/{protocol_id}/status-changes")
  509. async def list_collection_protocol_status_changes(
  510. protocol_id: str,
  511. user: dict = Depends(require_admin_or_super_admin),
  512. db: Session = Depends(get_db),
  513. ):
  514. tenant_no = _get_tenant_no(db, user)
  515. if not _protocol_accessible(db, protocol_id, tenant_no, user):
  516. return error_response(403, "无权访问该协议或协议不存在")
  517. items = cp_svc.list_protocol_status_changes(
  518. db,
  519. collection_protocol_id=protocol_id,
  520. tenant_no=tenant_no,
  521. )
  522. return success_response({"list": items})
  523. # ---------------------------------------------------------------------------
  524. # 收款确认
  525. # ---------------------------------------------------------------------------
  526. @router.get("/protocol-payment-details")
  527. async def list_payment_details(
  528. page: int = Query(1, ge=1),
  529. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  530. payStatus: str = Query("dfj", alias="payStatus"),
  531. keyword: str = Query("", alias="keyword"),
  532. transferType: str = Query("", alias="transferType"),
  533. installmentDueDateStart: Optional[str] = Query(None, alias="installmentDueDateStart"),
  534. installmentDueDateEnd: Optional[str] = Query(None, alias="installmentDueDateEnd"),
  535. user: dict = Depends(require_admin_or_super_admin),
  536. db: Session = Depends(get_db),
  537. ):
  538. tenant_no = _get_tenant_no(db, user)
  539. items, total = cp_svc.list_payment_details_page(
  540. db,
  541. page=page,
  542. page_size=pageSize,
  543. pay_status=payStatus,
  544. keyword=keyword or None,
  545. transfer_type=transferType or None,
  546. installment_due_date_start=installmentDueDateStart,
  547. installment_due_date_end=installmentDueDateEnd,
  548. tenant_no=tenant_no,
  549. user_info=user,
  550. )
  551. tp = cp_svc.total_pages(total, pageSize)
  552. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  553. class BatchDetailIdsBody(BaseModel):
  554. model_config = ConfigDict(populate_by_name=True)
  555. detail_ids: List[str] = Field(..., alias="detailIds", min_length=1)
  556. class PaymentDetailDraftBody(BaseModel):
  557. model_config = ConfigDict(populate_by_name=True)
  558. transfer_type: Optional[str] = Field(None, alias="transferType")
  559. transfer_amount: Optional[float] = Field(None, alias="transferAmount")
  560. transfer_time: Optional[str] = Field(None, alias="transferTime")
  561. transfer_attachments: Optional[str] = Field(None, alias="transferAttachments")
  562. remark: Optional[str] = None
  563. def _draft_fields_from_body(body: PaymentDetailDraftBody) -> dict:
  564. return body.model_dump(exclude_unset=True)
  565. @router.put("/protocol-payment-details/{detail_id}")
  566. async def put_payment_detail_draft(
  567. detail_id: str,
  568. body: PaymentDetailDraftBody,
  569. user: dict = Depends(require_admin_or_super_admin),
  570. db: Session = Depends(get_db),
  571. ):
  572. tenant_no = _get_tenant_no(db, user)
  573. fields = _draft_fields_from_body(body)
  574. if not fields:
  575. return error_response(400, "请提供需要保存的字段")
  576. try:
  577. cp_svc.update_payment_detail_draft(
  578. db,
  579. detail_id=detail_id,
  580. tenant_no=tenant_no,
  581. user_id=user.get("user_id"),
  582. draft_fields=fields,
  583. )
  584. return success_response(None, message="保存成功")
  585. except ValueError as e:
  586. return error_response(400, str(e))
  587. class RefundFinanceSubmitBody(BaseModel):
  588. model_config = ConfigDict(populate_by_name=True)
  589. refund_amount: float = Field(..., alias="refundAmount", gt=0)
  590. refund_remark: Optional[str] = Field(None, alias="refundRemark")
  591. refund_attachments: Optional[str] = Field(None, alias="refundAttachments")
  592. class BatchRefundFinanceSubmitBody(RefundFinanceSubmitBody):
  593. model_config = ConfigDict(populate_by_name=True)
  594. detail_ids: List[str] = Field(..., alias="detailIds", min_length=1)
  595. @router.post("/protocol-payment-details/{detail_id}/submit-receipt-finance")
  596. async def post_submit_receipt_finance(
  597. detail_id: str,
  598. body: PaymentDetailDraftBody = PaymentDetailDraftBody(),
  599. user: dict = Depends(require_admin_or_super_admin),
  600. db: Session = Depends(get_db),
  601. ):
  602. tenant_no = _get_tenant_no(db, user)
  603. fields = _draft_fields_from_body(body)
  604. try:
  605. cp_svc.submit_receipt_finance(
  606. db,
  607. detail_id=detail_id,
  608. tenant_no=tenant_no,
  609. user_id=user.get("user_id"),
  610. draft_fields=fields or None,
  611. )
  612. return success_response(None, message="提交成功")
  613. except ValueError as e:
  614. return error_response(400, str(e))
  615. @router.post("/protocol-payment-details/submit-receipt-finance-batch")
  616. async def post_submit_receipt_finance_batch(
  617. body: BatchDetailIdsBody,
  618. user: dict = Depends(require_admin_or_super_admin),
  619. db: Session = Depends(get_db),
  620. ):
  621. tenant_no = _get_tenant_no(db, user)
  622. try:
  623. count = cp_svc.batch_submit_receipt_finance(
  624. db,
  625. detail_ids=body.detail_ids,
  626. tenant_no=tenant_no,
  627. user_id=user.get("user_id"),
  628. )
  629. return success_response({"count": count}, message=f"已成功提交 {count} 条")
  630. except ValueError as e:
  631. return error_response(400, str(e))
  632. @router.post("/protocol-payment-details/{detail_id}/submit-refund-finance")
  633. async def post_submit_refund_finance(
  634. detail_id: str,
  635. body: RefundFinanceSubmitBody,
  636. user: dict = Depends(require_admin_or_super_admin),
  637. db: Session = Depends(get_db),
  638. ):
  639. tenant_no = _get_tenant_no(db, user)
  640. try:
  641. cp_svc.submit_refund_finance(
  642. db,
  643. detail_id=detail_id,
  644. tenant_no=tenant_no,
  645. user_id=user.get("user_id"),
  646. refund_amount=body.refund_amount,
  647. refund_remark=body.refund_remark,
  648. refund_attachments=body.refund_attachments,
  649. )
  650. return success_response(None, message="提交成功")
  651. except ValueError as e:
  652. return error_response(400, str(e))
  653. @router.post("/protocol-payment-details/submit-refund-finance-batch")
  654. async def post_submit_refund_finance_batch(
  655. body: BatchRefundFinanceSubmitBody,
  656. user: dict = Depends(require_admin_or_super_admin),
  657. db: Session = Depends(get_db),
  658. ):
  659. tenant_no = _get_tenant_no(db, user)
  660. try:
  661. count = cp_svc.batch_submit_refund_finance(
  662. db,
  663. detail_ids=body.detail_ids,
  664. tenant_no=tenant_no,
  665. user_id=user.get("user_id"),
  666. refund_amount=body.refund_amount,
  667. refund_remark=body.refund_remark,
  668. refund_attachments=body.refund_attachments,
  669. )
  670. return success_response({"count": count}, message=f"已成功提交 {count} 条")
  671. except ValueError as e:
  672. return error_response(400, str(e))
  673. @router.get("/protocol-payment-details/receipt-confirm")
  674. async def list_receipt_confirm(
  675. page: int = Query(1, ge=1),
  676. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  677. keyword: Optional[str] = None,
  678. installmentDueDateStart: Optional[str] = Query(None, alias="installmentDueDateStart"),
  679. installmentDueDateEnd: Optional[str] = Query(None, alias="installmentDueDateEnd"),
  680. user: dict = Depends(require_admin_or_super_admin),
  681. db: Session = Depends(get_db),
  682. ):
  683. tenant_no = _get_tenant_no(db, user)
  684. items, total = cp_svc.list_receipt_confirm_details(
  685. db,
  686. page=page,
  687. page_size=pageSize,
  688. keyword=keyword or None,
  689. installment_due_date_start=installmentDueDateStart,
  690. installment_due_date_end=installmentDueDateEnd,
  691. tenant_no=tenant_no,
  692. user_info=user,
  693. )
  694. tp = cp_svc.total_pages(total, pageSize)
  695. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  696. @router.get("/protocol-payment-details/receipt-confirmed")
  697. async def list_receipt_confirmed(
  698. page: int = Query(1, ge=1),
  699. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  700. keyword: Optional[str] = None,
  701. installmentDueDateStart: Optional[str] = Query(None, alias="installmentDueDateStart"),
  702. installmentDueDateEnd: Optional[str] = Query(None, alias="installmentDueDateEnd"),
  703. user: dict = Depends(require_admin_or_super_admin),
  704. db: Session = Depends(get_db),
  705. ):
  706. tenant_no = _get_tenant_no(db, user)
  707. items, total = cp_svc.list_receipt_confirmed_details(
  708. db,
  709. page=page,
  710. page_size=pageSize,
  711. keyword=keyword or None,
  712. installment_due_date_start=installmentDueDateStart,
  713. installment_due_date_end=installmentDueDateEnd,
  714. tenant_no=tenant_no,
  715. user_info=user,
  716. )
  717. tp = cp_svc.total_pages(total, pageSize)
  718. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  719. class FinanceConfirmBody(BaseModel):
  720. model_config = ConfigDict(populate_by_name=True)
  721. finance_confirm_attachments: Optional[str] = Field(None, alias="financeConfirmAttachments")
  722. finance_remark: Optional[str] = Field(None, alias="financeRemark")
  723. class RefundFinanceConfirmBody(BaseModel):
  724. model_config = ConfigDict(populate_by_name=True)
  725. refund_finance_confirm_attachments: Optional[str] = Field(
  726. None, alias="refundFinanceConfirmAttachments"
  727. )
  728. refund_finance_remark: Optional[str] = Field(None, alias="refundFinanceRemark")
  729. @router.post("/protocol-payment-details/{detail_id}/receipt-confirm")
  730. async def post_receipt_confirm(
  731. detail_id: str,
  732. body: FinanceConfirmBody,
  733. user: dict = Depends(require_admin_or_super_admin),
  734. db: Session = Depends(get_db),
  735. ):
  736. tenant_no = _get_tenant_no(db, user)
  737. try:
  738. cp_svc.confirm_receipt(
  739. db,
  740. detail_id=detail_id,
  741. tenant_no=tenant_no,
  742. user_id=user.get("user_id"),
  743. finance_confirm_attachments=body.finance_confirm_attachments,
  744. finance_remark=body.finance_remark,
  745. )
  746. return success_response(None, message="确认成功")
  747. except ValueError as e:
  748. return error_response(400, str(e))
  749. @router.post("/protocol-payment-details/{detail_id}/receipt-reject")
  750. async def post_receipt_reject(
  751. detail_id: str,
  752. user: dict = Depends(require_admin_or_super_admin),
  753. db: Session = Depends(get_db),
  754. ):
  755. tenant_no = _get_tenant_no(db, user)
  756. try:
  757. cp_svc.reject_receipt(
  758. db,
  759. detail_id=detail_id,
  760. tenant_no=tenant_no,
  761. user_id=user.get("user_id"),
  762. )
  763. return success_response(None, message="退回成功")
  764. except ValueError as e:
  765. return error_response(400, str(e))
  766. # ---------------------------------------------------------------------------
  767. # 退费确认
  768. # ---------------------------------------------------------------------------
  769. @router.get("/protocol-payment-details/refund-confirm")
  770. async def list_refund_confirm(
  771. page: int = Query(1, ge=1),
  772. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  773. keyword: Optional[str] = None,
  774. installmentDueDateStart: Optional[str] = Query(None, alias="installmentDueDateStart"),
  775. installmentDueDateEnd: Optional[str] = Query(None, alias="installmentDueDateEnd"),
  776. user: dict = Depends(require_admin_or_super_admin),
  777. db: Session = Depends(get_db),
  778. ):
  779. tenant_no = _get_tenant_no(db, user)
  780. items, total = cp_svc.list_refund_confirm_details(
  781. db,
  782. page=page,
  783. page_size=pageSize,
  784. keyword=keyword or None,
  785. installment_due_date_start=installmentDueDateStart,
  786. installment_due_date_end=installmentDueDateEnd,
  787. tenant_no=tenant_no,
  788. user_info=user,
  789. )
  790. tp = cp_svc.total_pages(total, pageSize)
  791. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  792. @router.get("/protocol-payment-details/refund-confirmed")
  793. async def list_refund_confirmed(
  794. page: int = Query(1, ge=1),
  795. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  796. keyword: Optional[str] = None,
  797. installmentDueDateStart: Optional[str] = Query(None, alias="installmentDueDateStart"),
  798. installmentDueDateEnd: Optional[str] = Query(None, alias="installmentDueDateEnd"),
  799. user: dict = Depends(require_admin_or_super_admin),
  800. db: Session = Depends(get_db),
  801. ):
  802. tenant_no = _get_tenant_no(db, user)
  803. items, total = cp_svc.list_refund_confirmed_details(
  804. db,
  805. page=page,
  806. page_size=pageSize,
  807. keyword=keyword or None,
  808. installment_due_date_start=installmentDueDateStart,
  809. installment_due_date_end=installmentDueDateEnd,
  810. tenant_no=tenant_no,
  811. user_info=user,
  812. )
  813. tp = cp_svc.total_pages(total, pageSize)
  814. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize, "totalPages": tp})
  815. @router.post("/protocol-payment-details/{detail_id}/refund-confirm")
  816. async def post_refund_confirm(
  817. detail_id: str,
  818. body: RefundFinanceConfirmBody,
  819. user: dict = Depends(require_admin_or_super_admin),
  820. db: Session = Depends(get_db),
  821. ):
  822. tenant_no = _get_tenant_no(db, user)
  823. try:
  824. cp_svc.confirm_refund(
  825. db,
  826. detail_id=detail_id,
  827. tenant_no=tenant_no,
  828. user_id=user.get("user_id"),
  829. refund_finance_confirm_attachments=body.refund_finance_confirm_attachments,
  830. refund_finance_remark=body.refund_finance_remark,
  831. )
  832. return success_response(None, message="确认成功")
  833. except ValueError as e:
  834. return error_response(400, str(e))
  835. @router.post("/protocol-payment-details/{detail_id}/refund-reject")
  836. async def post_refund_reject(
  837. detail_id: str,
  838. user: dict = Depends(require_admin_or_super_admin),
  839. db: Session = Depends(get_db),
  840. ):
  841. tenant_no = _get_tenant_no(db, user)
  842. try:
  843. cp_svc.reject_refund(
  844. db,
  845. detail_id=detail_id,
  846. tenant_no=tenant_no,
  847. user_id=user.get("user_id"),
  848. )
  849. return success_response(None, message="退回成功")
  850. except ValueError as e:
  851. return error_response(400, str(e))
  852. # ---------------------------------------------------------------------------
  853. # 储存证明管理 / 满月待办
  854. # ---------------------------------------------------------------------------
  855. @router.get("/storage-certs")
  856. async def list_storage_certs(
  857. page: int = Query(1, ge=1),
  858. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  859. keyword: Optional[str] = None,
  860. makeStatus: Optional[str] = Query(None, alias="makeStatus"),
  861. storageInDateStart: Optional[str] = Query(None, alias="storageInDateStart"),
  862. storageInDateEnd: Optional[str] = Query(None, alias="storageInDateEnd"),
  863. user: dict = Depends(require_admin_or_super_admin),
  864. db: Session = Depends(get_db),
  865. ):
  866. tenant_no = _get_tenant_no(db, user)
  867. page_items, total = cp_svc.list_storage_certs(
  868. db,
  869. tenant_no=tenant_no,
  870. keyword=keyword,
  871. make_status=makeStatus,
  872. storage_in_date_start=storageInDateStart,
  873. storage_in_date_end=storageInDateEnd,
  874. page=page,
  875. page_size=pageSize,
  876. user_info=user,
  877. )
  878. page_safe = max(1, page)
  879. page_size_safe = min(max(1, pageSize), 100)
  880. tp = cp_svc.total_pages(total, page_size_safe)
  881. return success_response(
  882. {"list": page_items, "total": total, "page": page_safe, "pageSize": page_size_safe, "totalPages": tp}
  883. )
  884. @router.get("/fullmoon-todos")
  885. async def list_fullmoon_todos(
  886. page: int = Query(1, ge=1),
  887. pageSize: int = Query(10, ge=1, le=100, alias="pageSize"),
  888. keyword: Optional[str] = None,
  889. todoStatus: Optional[str] = Query(None, alias="todoStatus"),
  890. fullmoonDateStart: Optional[str] = Query(None, alias="fullmoonDateStart"),
  891. fullmoonDateEnd: Optional[str] = Query(None, alias="fullmoonDateEnd"),
  892. user: dict = Depends(require_admin_or_super_admin),
  893. db: Session = Depends(get_db),
  894. ):
  895. tenant_no = _get_tenant_no(db, user)
  896. party_join = sa_and(
  897. BizProtocolPartyCustomer.collection_protocol_id == BizCollectionProtocol.id,
  898. BizProtocolPartyCustomer.del_flag == 0,
  899. BizProtocolPartyCustomer.tenant_no == tenant_no,
  900. )
  901. q = (
  902. db.query(BizCollectionProtocol, BizProtocolPartyCustomer)
  903. .outerjoin(BizProtocolPartyCustomer, party_join)
  904. .filter(
  905. BizCollectionProtocol.del_flag == 0,
  906. BizCollectionProtocol.tenant_no == tenant_no,
  907. BizCollectionProtocol.protocol_status == "sxz",
  908. )
  909. )
  910. kw = (keyword or "").strip()
  911. if kw:
  912. like = f"%{kw}%"
  913. q = q.filter(BizProtocolPartyCustomer.mother_name.like(like))
  914. from app.core.data_scope import apply_data_scope
  915. q = apply_data_scope(q, BizCollectionProtocol, user, db, tenant_no)
  916. rows = q.order_by(BizCollectionProtocol.create_time.desc()).all()
  917. protocol_ids = [r.id for r, _ in rows]
  918. ft_map: Dict[str, Any] = {}
  919. if protocol_ids:
  920. from app.models.models import BizProtocolFullmoonTodo
  921. ft_rows = (
  922. db.query(BizProtocolFullmoonTodo)
  923. .filter(
  924. BizProtocolFullmoonTodo.collection_protocol_id.in_(protocol_ids),
  925. BizProtocolFullmoonTodo.tenant_no == tenant_no,
  926. BizProtocolFullmoonTodo.del_flag == 0,
  927. )
  928. .all()
  929. )
  930. for ft in ft_rows:
  931. ft_map[ft.collection_protocol_id] = ft
  932. now = datetime.now()
  933. _empty_ft: Dict[str, Any] = {
  934. "todoStatus": "0",
  935. "handleDate": "",
  936. "handleRemark": "",
  937. "insuranceAttachments": "",
  938. "insuranceCompanyId": "",
  939. "insuranceCompanyName": "",
  940. "insuranceType": "",
  941. "insurancePolicyNo": "",
  942. "insuranceEffectiveDate": "",
  943. "insuranceExpiryDate": "",
  944. "insuranceSubscribed": "0",
  945. "insuranceFee": None,
  946. }
  947. items: List[Dict[str, Any]] = []
  948. for proto, pc in rows:
  949. dt = pc.newborn_delivery_time if pc else None
  950. fullmoon_date_str = ""
  951. is_fullmoon_passed = False
  952. if dt:
  953. fullmoon_date = dt + timedelta(days=30)
  954. fullmoon_date_str = fullmoon_date.strftime("%Y-%m-%d")
  955. is_fullmoon_passed = fullmoon_date <= now
  956. ft = ft_map.get(proto.id)
  957. ts = (ft.todo_status or "0") if ft else "0"
  958. if todoStatus and todoStatus.strip() and ts != todoStatus.strip():
  959. continue
  960. fm_start = (fullmoonDateStart or "").strip()[:10]
  961. fm_end = (fullmoonDateEnd or "").strip()[:10]
  962. if fm_start or fm_end:
  963. if not fullmoon_date_str:
  964. continue
  965. if fm_start and fullmoon_date_str < fm_start:
  966. continue
  967. if fm_end and fullmoon_date_str > fm_end:
  968. continue
  969. ft_dict = cp_svc._fullmoon_todo_to_dict(ft) if ft else dict(_empty_ft)
  970. items.append(
  971. {
  972. "protocolId": proto.id,
  973. "id": proto.id,
  974. "protocolNo": proto.protocol_no or "",
  975. "protocolStatus": proto.protocol_status or "",
  976. "motherName": (pc.mother_name if pc else "") or "",
  977. "motherPhone": (pc.mother_phone if pc else "") or "",
  978. "newbornName": (pc.newborn_name if pc else "") or "",
  979. "newbornDeliveryTime": cp_svc._fmt_dt(dt) if dt else "",
  980. "fullmoonDate": fullmoon_date_str,
  981. "isFullmoonPassed": is_fullmoon_passed,
  982. **ft_dict,
  983. }
  984. )
  985. total = len(items)
  986. page_safe = max(1, page)
  987. page_size_safe = min(max(1, pageSize), 100)
  988. start = (page_safe - 1) * page_size_safe
  989. page_items = items[start : start + page_size_safe]
  990. tp = cp_svc.total_pages(total, page_size_safe)
  991. return success_response(
  992. {"list": page_items, "total": total, "page": page_safe, "pageSize": page_size_safe, "totalPages": tp}
  993. )
  994. class FullmoonTodoInsuranceBody(BaseModel):
  995. model_config = ConfigDict(populate_by_name=True)
  996. insurance_company_id: Optional[str] = Field(None, alias="insuranceCompanyId")
  997. insurance_company_name: Optional[str] = Field(None, alias="insuranceCompanyName")
  998. insurance_type: Optional[str] = Field(None, alias="insuranceType")
  999. insurance_policy_no: Optional[str] = Field(None, alias="insurancePolicyNo")
  1000. insurance_effective_date: Optional[str] = Field(None, alias="insuranceEffectiveDate")
  1001. insurance_expiry_date: Optional[str] = Field(None, alias="insuranceExpiryDate")
  1002. insurance_subscribed: Optional[str] = Field(None, alias="insuranceSubscribed")
  1003. insurance_fee: Optional[float] = Field(None, alias="insuranceFee")
  1004. insurance_attachments: Optional[str] = Field(None, alias="insuranceAttachments")
  1005. todo_status: Optional[str] = Field(None, alias="todoStatus")
  1006. handle_date: Optional[str] = Field(None, alias="handleDate")
  1007. handle_remark: Optional[str] = Field(None, alias="handleRemark")
  1008. @router.put("/fullmoon-todos/{protocol_id}/insurance")
  1009. async def update_fullmoon_todo_insurance(
  1010. protocol_id: str,
  1011. body: FullmoonTodoInsuranceBody,
  1012. user: dict = Depends(require_admin_or_super_admin),
  1013. db: Session = Depends(get_db),
  1014. ):
  1015. tenant_no = _get_tenant_no(db, user)
  1016. user_id = _get_current_user_id(user)
  1017. try:
  1018. payload = body.model_dump(exclude_unset=True, by_alias=False)
  1019. ft = cp_svc.upsert_fullmoon_todo(
  1020. db,
  1021. protocol_id,
  1022. tenant_no,
  1023. user_id,
  1024. **payload,
  1025. )
  1026. return success_response(cp_svc._fullmoon_todo_to_dict(ft), message="更新成功")
  1027. except ValueError as e:
  1028. return error_response(400, str(e))
  1029. class StorageCertHandleBody(BaseModel):
  1030. model_config = ConfigDict(populate_by_name=True)
  1031. handle_time: Optional[str] = Field(None, alias="handleTime")
  1032. handler_user_id: Optional[str] = Field(None, alias="handlerUserId")
  1033. cert_image_urls: Optional[str] = Field(None, alias="certImageUrls")
  1034. remark: Optional[str] = None
  1035. class FullmoonHandleBody(BaseModel):
  1036. model_config = ConfigDict(populate_by_name=True)
  1037. handle_time: Optional[str] = Field(None, alias="handleTime")
  1038. handler_user_id: Optional[str] = Field(None, alias="handlerUserId")
  1039. insurance_attachments: Optional[str] = Field(None, alias="insuranceAttachments")
  1040. handle_remark: Optional[str] = Field(None, alias="handleRemark")
  1041. def _attach_handler_name(db: Session, payload: Dict[str, Any]) -> Dict[str, Any]:
  1042. uid = (payload.get("handlerUserId") or "").strip()
  1043. name = ""
  1044. if uid:
  1045. u = db.query(SysUser).filter(SysUser.id == uid, SysUser.del_flag == 0).first()
  1046. name = (u.nick_name or "") if u else ""
  1047. payload["handlerUserName"] = name
  1048. return payload
  1049. @router.put("/storage-certs/{protocol_id}")
  1050. async def update_storage_cert_handle(
  1051. protocol_id: str,
  1052. body: StorageCertHandleBody,
  1053. user: dict = Depends(require_admin_or_super_admin),
  1054. db: Session = Depends(get_db),
  1055. ):
  1056. tenant_no = _get_tenant_no(db, user)
  1057. user_id = _get_current_user_id(user)
  1058. if not _protocol_accessible(db, protocol_id, tenant_no, user):
  1059. return error_response(404, "协议单不存在")
  1060. try:
  1061. ht = parse_datetime(body.handle_time) if body.handle_time else None
  1062. sc = cp_svc.upsert_storage_cert_handle(
  1063. db,
  1064. protocol_id,
  1065. tenant_no,
  1066. user_id,
  1067. handle_time=ht,
  1068. handler_user_id=body.handler_user_id,
  1069. cert_image_urls=body.cert_image_urls,
  1070. remark=body.remark,
  1071. )
  1072. return success_response(
  1073. _attach_handler_name(db, cp_svc._storage_cert_to_dict(sc)), message="更新成功"
  1074. )
  1075. except ValueError as e:
  1076. return error_response(400, str(e))
  1077. @router.put("/fullmoon-todos/{protocol_id}/handle")
  1078. async def update_fullmoon_todo_handle(
  1079. protocol_id: str,
  1080. body: FullmoonHandleBody,
  1081. user: dict = Depends(require_admin_or_super_admin),
  1082. db: Session = Depends(get_db),
  1083. ):
  1084. tenant_no = _get_tenant_no(db, user)
  1085. user_id = _get_current_user_id(user)
  1086. if not _protocol_accessible(db, protocol_id, tenant_no, user):
  1087. return error_response(404, "协议单不存在")
  1088. try:
  1089. ht = parse_datetime(body.handle_time) if body.handle_time else None
  1090. ft = cp_svc.upsert_fullmoon_todo_handle(
  1091. db,
  1092. protocol_id,
  1093. tenant_no,
  1094. user_id,
  1095. handle_time=ht,
  1096. handler_user_id=body.handler_user_id,
  1097. insurance_attachments=body.insurance_attachments,
  1098. handle_remark=body.handle_remark,
  1099. )
  1100. return success_response(
  1101. _attach_handler_name(db, cp_svc._fullmoon_todo_to_dict(ft)), message="更新成功"
  1102. )
  1103. except ValueError as e:
  1104. return error_response(400, str(e))
  1105. class InvoiceRequestCreateBody(BaseModel):
  1106. model_config = ConfigDict(populate_by_name=True)
  1107. payment_detail_ids: List[str] = Field(..., alias="paymentDetailIds", min_length=1)
  1108. invoice_title: Optional[str] = Field(None, alias="invoiceTitle")
  1109. tax_no: Optional[str] = Field(None, alias="taxNo")
  1110. company_address: Optional[str] = Field(None, alias="companyAddress")
  1111. company_phone: Optional[str] = Field(None, alias="companyPhone")
  1112. bank_name: Optional[str] = Field(None, alias="bankName")
  1113. bank_account: Optional[str] = Field(None, alias="bankAccount")
  1114. apply_remark: Optional[str] = Field(None, alias="applyRemark")
  1115. class InvoiceRequestCompleteBody(BaseModel):
  1116. model_config = ConfigDict(populate_by_name=True)
  1117. invoice_attachments: str = Field(..., alias="invoiceAttachments")
  1118. finance_remark: Optional[str] = Field(None, alias="financeRemark")
  1119. @router.get("/collection-protocols/{protocol_id}/invoice-eligible-payments")
  1120. async def get_invoice_eligible_payments(
  1121. protocol_id: str,
  1122. user: dict = Depends(require_admin_or_super_admin),
  1123. db: Session = Depends(get_db),
  1124. ):
  1125. tenant_no = _get_tenant_no(db, user)
  1126. items = cp_svc.list_invoice_eligible_payment_details(
  1127. db, collection_protocol_id=protocol_id, tenant_no=tenant_no
  1128. )
  1129. return success_response({"list": items, "total": len(items)})
  1130. @router.get("/collection-protocols/{protocol_id}/invoice-requests")
  1131. async def get_protocol_invoice_requests(
  1132. protocol_id: str,
  1133. user: dict = Depends(require_admin_or_super_admin),
  1134. db: Session = Depends(get_db),
  1135. ):
  1136. tenant_no = _get_tenant_no(db, user)
  1137. items = cp_svc.list_invoice_requests_by_protocol(
  1138. db, collection_protocol_id=protocol_id, tenant_no=tenant_no
  1139. )
  1140. return success_response({"list": items, "total": len(items)})
  1141. @router.post("/collection-protocols/{protocol_id}/invoice-requests")
  1142. async def post_protocol_invoice_request(
  1143. protocol_id: str,
  1144. body: InvoiceRequestCreateBody,
  1145. user: dict = Depends(require_admin_or_super_admin),
  1146. db: Session = Depends(get_db),
  1147. ):
  1148. tenant_no = _get_tenant_no(db, user)
  1149. try:
  1150. row = cp_svc.create_invoice_request(
  1151. db,
  1152. collection_protocol_id=protocol_id,
  1153. payment_detail_ids=body.payment_detail_ids,
  1154. tenant_no=tenant_no,
  1155. user_info=user,
  1156. invoice_title=body.invoice_title,
  1157. tax_no=body.tax_no,
  1158. company_address=body.company_address,
  1159. company_phone=body.company_phone,
  1160. bank_name=body.bank_name,
  1161. bank_account=body.bank_account,
  1162. apply_remark=body.apply_remark,
  1163. )
  1164. except ValueError as e:
  1165. return error_response(400, str(e))
  1166. return success_response(row, message="提交成功")
  1167. @router.get("/protocol-invoice-requests")
  1168. async def list_protocol_invoice_requests_api(
  1169. status: str = Query(..., description="dkp|ykp"),
  1170. page: int = Query(1, ge=1),
  1171. pageSize: int = Query(10, ge=1, le=100),
  1172. keyword: Optional[str] = Query(None),
  1173. user: dict = Depends(require_admin_or_super_admin),
  1174. db: Session = Depends(get_db),
  1175. ):
  1176. tenant_no = _get_tenant_no(db, user)
  1177. items, total = cp_svc.list_protocol_invoice_requests(
  1178. db,
  1179. status=status,
  1180. page=page,
  1181. page_size=pageSize,
  1182. keyword=keyword,
  1183. tenant_no=tenant_no,
  1184. user_info=user,
  1185. )
  1186. return success_response({"list": items, "total": total, "page": page, "pageSize": pageSize})
  1187. @router.get("/protocol-invoice-requests/pending-count")
  1188. async def protocol_invoice_pending_count(
  1189. user: dict = Depends(require_admin_or_super_admin),
  1190. db: Session = Depends(get_db),
  1191. ):
  1192. tenant_no = _get_tenant_no(db, user)
  1193. count = cp_svc.count_pending_protocol_invoice_requests(
  1194. db, tenant_no=tenant_no, user_info=user
  1195. )
  1196. return success_response({"count": count})
  1197. @router.put("/protocol-invoice-requests/{request_id}/complete")
  1198. async def complete_protocol_invoice_request(
  1199. request_id: str,
  1200. body: InvoiceRequestCompleteBody,
  1201. user: dict = Depends(require_admin_or_super_admin),
  1202. db: Session = Depends(get_db),
  1203. ):
  1204. tenant_no = _get_tenant_no(db, user)
  1205. try:
  1206. row = cp_svc.complete_invoice_request(
  1207. db,
  1208. request_id=request_id,
  1209. tenant_no=tenant_no,
  1210. invoice_attachments=body.invoice_attachments,
  1211. finance_remark=body.finance_remark,
  1212. user_info=user,
  1213. )
  1214. except ValueError as e:
  1215. return error_response(400, str(e))
  1216. return success_response(row, message="已完成开票")