pl_freight_settlement_service.py 50 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503
  1. """运费结算业务服务"""
  2. from __future__ import annotations
  3. import json
  4. from datetime import date, datetime
  5. from decimal import Decimal
  6. from typing import Any, Dict, List, Optional, Tuple
  7. from fastapi import HTTPException
  8. from sqlalchemy import false, func, or_
  9. from sqlalchemy.orm import Session
  10. from app.core.utils import generate_uuid
  11. from app.core.delete_guard import assert_row_deletable_by_creator, user_info_from_context
  12. from app.models.models import SysUser
  13. from app.services.pl_data_scope import apply_pl_scope, assert_pl_row_in_scope
  14. from app.models.pl_models import (
  15. PlFreightSettlement,
  16. PlFreightSettlementApproval,
  17. PlFreightSettlementTransport,
  18. PlPlan,
  19. PlRemittanceVoucher,
  20. PlTransport,
  21. )
  22. from app.schemas.pl.freight import (
  23. FreightSettlementDisburseBody,
  24. FreightSettlementSubmitApprovalBody,
  25. FreightSettlementUpdate,
  26. )
  27. _FMT = "%Y-%m-%d %H:%M:%S"
  28. _DRAFT_SETTLEMENT_STATUSES = ("pending", "rejected")
  29. TAB_STATUS_MAP = {
  30. "bills": ("pending", "rejected", "approving", "approved", "confirmed", "cancelled"),
  31. "approval_pending": ("approving",),
  32. "approval_done": ("approved", "confirmed"),
  33. "disburse": ("approved",),
  34. "settled": ("confirmed",),
  35. }
  36. def _resolve_user_display_name(db: Session, user_id: str) -> str:
  37. uid = (user_id or "").strip()
  38. if not uid:
  39. return ""
  40. u = db.query(SysUser).filter(SysUser.id == uid, SysUser.del_flag == 0).first()
  41. if not u:
  42. return ""
  43. return (u.nick_name or u.user_no or "").strip()
  44. def _occupied_transport_ids_subquery(db: Session, tenant_no: str):
  45. return (
  46. db.query(PlFreightSettlementTransport.transport_id)
  47. .join(
  48. PlFreightSettlement,
  49. PlFreightSettlement.id == PlFreightSettlementTransport.settlement_id,
  50. )
  51. .filter(
  52. PlFreightSettlementTransport.tenant_no == tenant_no,
  53. PlFreightSettlementTransport.del_flag == 0,
  54. PlFreightSettlement.tenant_no == tenant_no,
  55. PlFreightSettlement.del_flag == 0,
  56. PlFreightSettlement.settlement_status != "cancelled",
  57. )
  58. )
  59. def get_settlement_row(
  60. db: Session,
  61. tenant_no: str,
  62. settlement_id: str,
  63. *,
  64. user=None,
  65. skip_scope: bool = False,
  66. ) -> PlFreightSettlement:
  67. row = (
  68. db.query(PlFreightSettlement)
  69. .filter(
  70. PlFreightSettlement.id == settlement_id,
  71. PlFreightSettlement.tenant_no == tenant_no,
  72. PlFreightSettlement.del_flag == 0,
  73. )
  74. .first()
  75. )
  76. if not row:
  77. raise HTTPException(status_code=404, detail="结算单不存在")
  78. if user and not skip_scope:
  79. assert_pl_row_in_scope(db, row, PlFreightSettlement, user, tenant_no)
  80. return row
  81. def get_settlement_by_transport_id(
  82. db: Session,
  83. *,
  84. tenant_no: str,
  85. transport_id: str,
  86. user=None,
  87. ) -> Optional[Dict[str, Any]]:
  88. if not transport_id:
  89. return None
  90. row = (
  91. db.query(PlFreightSettlement)
  92. .join(
  93. PlFreightSettlementTransport,
  94. PlFreightSettlementTransport.settlement_id == PlFreightSettlement.id,
  95. )
  96. .filter(
  97. PlFreightSettlementTransport.transport_id == transport_id,
  98. PlFreightSettlementTransport.tenant_no == tenant_no,
  99. PlFreightSettlementTransport.del_flag == 0,
  100. PlFreightSettlement.tenant_no == tenant_no,
  101. PlFreightSettlement.del_flag == 0,
  102. )
  103. .order_by(PlFreightSettlement.create_time.desc())
  104. .first()
  105. )
  106. if not row:
  107. return None
  108. if user:
  109. assert_pl_row_in_scope(db, row, PlFreightSettlement, user, tenant_no)
  110. return {
  111. "id": row.id,
  112. "settlementNo": row.settlement_no,
  113. "invoiceAmount": row.invoice_amount,
  114. "invoiceNo": row.invoice_no,
  115. "invoiceDate": row.invoice_date,
  116. "invoiceAttachmentUrls": row.invoice_attachment_urls,
  117. "settlementStatus": row.settlement_status,
  118. }
  119. def unsettled_plans_q_filter(q: str):
  120. """简易搜索:计划编号 / 客户(供应商) / 车牌 / 承运商名称。"""
  121. return or_(
  122. PlPlan.plan_no.contains(q),
  123. PlPlan.customer_name.contains(q),
  124. PlPlan.supplier_name.contains(q),
  125. PlPlan.license_plate.contains(q),
  126. PlTransport.license_plate.contains(q),
  127. PlTransport.carrier_name.contains(q),
  128. )
  129. def list_unsettled_plans(
  130. db: Session,
  131. *,
  132. tenant_no: str,
  133. user=None,
  134. q: Optional[str] = None,
  135. carrier_id: Optional[str] = None,
  136. plan_no: Optional[str] = None,
  137. plan_type: Optional[str] = None,
  138. plan_type_in: Optional[str] = None,
  139. license_plate: Optional[str] = None,
  140. transport_company: Optional[str] = None,
  141. freight_amount_op: Optional[str] = None,
  142. freight_amount: Optional[str] = None,
  143. freight_amount_min: Optional[str] = None,
  144. freight_amount_max: Optional[str] = None,
  145. freight_type: Optional[str] = None,
  146. freight_type_in: Optional[str] = None,
  147. transport_qty_op: Optional[str] = None,
  148. transport_qty: Optional[str] = None,
  149. transport_qty_min: Optional[str] = None,
  150. transport_qty_max: Optional[str] = None,
  151. has_invoice: Optional[str] = None,
  152. has_invoice_in: Optional[str] = None,
  153. freight_remark: Optional[str] = None,
  154. ids: Optional[List[str]] = None,
  155. page: int = 1,
  156. page_size: int = 20,
  157. max_page_size: Optional[int] = None,
  158. ) -> Tuple[int, List[Dict[str, Any]]]:
  159. occupied = _occupied_transport_ids_subquery(db, tenant_no).subquery()
  160. query = (
  161. db.query(PlPlan, PlTransport)
  162. .join(PlTransport, PlTransport.id == PlPlan.transport_id)
  163. .filter(
  164. PlPlan.tenant_no == tenant_no,
  165. PlPlan.del_flag == 0,
  166. PlPlan.archive_status == "archived",
  167. PlPlan.has_freight == 1,
  168. PlPlan.follow_status != "cancelled",
  169. PlPlan.transport_id.isnot(None),
  170. PlTransport.del_flag == 0,
  171. PlTransport.freight_amount > 0,
  172. ~PlPlan.transport_id.in_(occupied),
  173. )
  174. )
  175. if q:
  176. query = query.filter(unsettled_plans_q_filter(q))
  177. if carrier_id:
  178. query = query.filter(PlTransport.carrier_id == carrier_id)
  179. from app.services.pl.freight_list_filters import build_unsettled_plan_column_filters
  180. column_filters = build_unsettled_plan_column_filters(
  181. plan_no=plan_no,
  182. plan_type=plan_type,
  183. plan_type_in=plan_type_in,
  184. license_plate=license_plate,
  185. transport_company=transport_company,
  186. freight_amount_op=freight_amount_op,
  187. freight_amount=freight_amount,
  188. freight_amount_min=freight_amount_min,
  189. freight_amount_max=freight_amount_max,
  190. freight_type=freight_type,
  191. freight_type_in=freight_type_in,
  192. transport_qty_op=transport_qty_op,
  193. transport_qty=transport_qty,
  194. transport_qty_min=transport_qty_min,
  195. transport_qty_max=transport_qty_max,
  196. has_invoice=has_invoice,
  197. has_invoice_in=has_invoice_in,
  198. freight_remark=freight_remark,
  199. )
  200. if column_filters:
  201. query = query.filter(*column_filters)
  202. if ids:
  203. query = query.filter(PlPlan.id.in_(ids))
  204. if user:
  205. query = apply_pl_scope(query, PlPlan, user, db, tenant_no)
  206. total = query.count()
  207. page = max(1, page)
  208. cap = int(max_page_size) if max_page_size else 500
  209. page_size = min(max(1, page_size), cap)
  210. rows = (
  211. query.order_by(PlPlan.update_time.desc())
  212. .offset((page - 1) * page_size)
  213. .limit(page_size)
  214. .all()
  215. )
  216. items: List[Dict[str, Any]] = []
  217. for plan, transport in rows:
  218. items.append(
  219. {
  220. "id": plan.id,
  221. "planNo": plan.plan_no,
  222. "planDate": plan.plan_date,
  223. "planType": plan.plan_type,
  224. "customerName": plan.customer_name,
  225. "supplierName": plan.supplier_name,
  226. "licensePlate": plan.license_plate or transport.license_plate,
  227. "transportCompany": transport.carrier_name,
  228. "freightAmount": plan.freight_amount or transport.freight_amount,
  229. "freightType": transport.freight_type,
  230. "transportQty": transport.transport_qty,
  231. "freightPrice": transport.freight_price,
  232. "hasInvoice": int(transport.has_invoice or 0),
  233. "freightRemark": plan.freight_remark,
  234. "archiveTime": plan.update_time,
  235. "transportId": transport.id,
  236. "transportNo": transport.transport_no,
  237. "driverName": transport.driver_name,
  238. "transportMedium": transport.transport_medium,
  239. "startPoint": transport.start_point,
  240. "endPoint": transport.end_point,
  241. }
  242. )
  243. return total, items
  244. def _resolve_transports_for_create(
  245. db: Session, tenant_no: str, *, plan_ids: Optional[List[str]], transport_ids: Optional[List[str]]
  246. ) -> List[PlTransport]:
  247. ids: List[str] = []
  248. if plan_ids:
  249. plans = (
  250. db.query(PlPlan)
  251. .filter(
  252. PlPlan.id.in_(plan_ids),
  253. PlPlan.tenant_no == tenant_no,
  254. PlPlan.del_flag == 0,
  255. )
  256. .all()
  257. )
  258. if len(plans) != len(set(plan_ids)):
  259. raise HTTPException(status_code=400, detail="存在无效的计划")
  260. for p in plans:
  261. if p.archive_status != "archived":
  262. raise HTTPException(status_code=400, detail=f"计划 {p.plan_no} 未归档,不能结算")
  263. if int(p.has_freight or 0) != 1:
  264. raise HTTPException(status_code=400, detail=f"计划 {p.plan_no} 无运输费用")
  265. if not p.transport_id:
  266. raise HTTPException(status_code=400, detail=f"计划 {p.plan_no} 未安排运输")
  267. ids.append(p.transport_id)
  268. if transport_ids:
  269. ids.extend(transport_ids)
  270. ids = list(dict.fromkeys(ids))
  271. if not ids:
  272. raise HTTPException(status_code=400, detail="请选择有效的运输单")
  273. occupied = {r[0] for r in _occupied_transport_ids_subquery(db, tenant_no).all()}
  274. conflict = [tid for tid in ids if tid in occupied]
  275. if conflict:
  276. raise HTTPException(status_code=400, detail="所选运输单已纳入其他结算单")
  277. transports = (
  278. db.query(PlTransport)
  279. .filter(
  280. PlTransport.id.in_(ids),
  281. PlTransport.tenant_no == tenant_no,
  282. PlTransport.del_flag == 0,
  283. )
  284. .all()
  285. )
  286. if len(transports) != len(ids):
  287. raise HTTPException(status_code=400, detail="请选择有效的运输单")
  288. return transports
  289. def create_settlement(
  290. db: Session,
  291. *,
  292. tenant_no: str,
  293. plan_ids: Optional[List[str]],
  294. transport_ids: Optional[List[str]],
  295. user_id: str,
  296. dept_id: str,
  297. settlement_no: str,
  298. ) -> PlFreightSettlement:
  299. transports = _resolve_transports_for_create(
  300. db, tenant_no, plan_ids=plan_ids, transport_ids=transport_ids
  301. )
  302. total_freight = sum(Decimal(str(t.freight_amount or 0)) for t in transports)
  303. settlement = PlFreightSettlement(
  304. id=generate_uuid(),
  305. tenant_no=tenant_no,
  306. settlement_no=settlement_no,
  307. total_freight=total_freight,
  308. settlement_status="pending",
  309. create_by=user_id,
  310. create_dept_id=dept_id or "",
  311. )
  312. db.add(settlement)
  313. for t in transports:
  314. db.add(
  315. PlFreightSettlementTransport(
  316. id=generate_uuid(),
  317. tenant_no=tenant_no,
  318. settlement_id=settlement.id,
  319. transport_id=t.id,
  320. freight_amount=t.freight_amount,
  321. create_by=user_id,
  322. create_dept_id=dept_id or "",
  323. )
  324. )
  325. db.commit()
  326. db.refresh(settlement)
  327. return settlement
  328. def _recalc_settlement_total_freight_from_links(
  329. db: Session,
  330. settlement_id: str,
  331. *,
  332. user_id: str,
  333. now: Optional[datetime] = None,
  334. ) -> None:
  335. # SessionLocal 关闭了 autoflush,须先 flush 再聚合查询,否则 SUM 仍读到旧明细金额
  336. db.flush()
  337. total = (
  338. db.query(func.coalesce(func.sum(PlFreightSettlementTransport.freight_amount), 0))
  339. .filter(
  340. PlFreightSettlementTransport.settlement_id == settlement_id,
  341. PlFreightSettlementTransport.del_flag == 0,
  342. )
  343. .scalar()
  344. )
  345. settlement = (
  346. db.query(PlFreightSettlement)
  347. .filter(
  348. PlFreightSettlement.id == settlement_id,
  349. PlFreightSettlement.del_flag == 0,
  350. )
  351. .first()
  352. )
  353. if not settlement:
  354. return
  355. ts = now or datetime.now()
  356. settlement.total_freight = Decimal(str(total or 0))
  357. settlement.update_by = user_id
  358. settlement.update_time = ts
  359. def sync_draft_settlement_freight_for_transport(
  360. db: Session,
  361. *,
  362. tenant_no: str,
  363. transport_id: str,
  364. freight_amount: Decimal,
  365. user_id: str,
  366. ) -> None:
  367. """运输单运费变更后,同步关联草稿结算单的明细快照与合计。"""
  368. tid = (transport_id or "").strip()
  369. if not tid:
  370. return
  371. links = (
  372. db.query(PlFreightSettlementTransport)
  373. .join(
  374. PlFreightSettlement,
  375. PlFreightSettlement.id == PlFreightSettlementTransport.settlement_id,
  376. )
  377. .filter(
  378. PlFreightSettlementTransport.transport_id == tid,
  379. PlFreightSettlementTransport.tenant_no == tenant_no,
  380. PlFreightSettlementTransport.del_flag == 0,
  381. PlFreightSettlement.tenant_no == tenant_no,
  382. PlFreightSettlement.del_flag == 0,
  383. PlFreightSettlement.settlement_status.in_(_DRAFT_SETTLEMENT_STATUSES),
  384. )
  385. .all()
  386. )
  387. if not links:
  388. return
  389. now = datetime.now()
  390. amount = Decimal(str(freight_amount or 0))
  391. settlement_ids = set()
  392. for link in links:
  393. link.freight_amount = amount
  394. link.update_by = user_id
  395. link.update_time = now
  396. settlement_ids.add(link.settlement_id)
  397. for settlement_id in settlement_ids:
  398. _recalc_settlement_total_freight_from_links(
  399. db, settlement_id, user_id=user_id, now=now
  400. )
  401. def refresh_draft_settlement_freight(
  402. db: Session,
  403. settlement: PlFreightSettlement,
  404. *,
  405. user_id: str = "",
  406. ) -> bool:
  407. """草稿/已驳回结算单按当前运输单运费刷新明细快照与合计。"""
  408. if settlement.settlement_status not in _DRAFT_SETTLEMENT_STATUSES:
  409. return False
  410. links = (
  411. db.query(PlFreightSettlementTransport)
  412. .filter(
  413. PlFreightSettlementTransport.settlement_id == settlement.id,
  414. PlFreightSettlementTransport.del_flag == 0,
  415. )
  416. .all()
  417. )
  418. if not links:
  419. return False
  420. transport_ids = [link.transport_id for link in links]
  421. transports = {
  422. t.id: t
  423. for t in db.query(PlTransport)
  424. .filter(PlTransport.id.in_(transport_ids), PlTransport.del_flag == 0)
  425. .all()
  426. }
  427. old_total = Decimal(str(settlement.total_freight or 0))
  428. now = datetime.now()
  429. changed = False
  430. for link in links:
  431. transport = transports.get(link.transport_id)
  432. if not transport:
  433. continue
  434. new_amount = Decimal(str(transport.freight_amount or 0))
  435. if Decimal(str(link.freight_amount or 0)) != new_amount:
  436. link.freight_amount = new_amount
  437. link.update_time = now
  438. if user_id:
  439. link.update_by = user_id
  440. changed = True
  441. _recalc_settlement_total_freight_from_links(
  442. db, settlement.id, user_id=user_id, now=now
  443. )
  444. new_total = Decimal(str(settlement.total_freight or 0))
  445. return changed or new_total != old_total
  446. def delete_draft_settlement(
  447. db: Session, *, tenant_no: str, settlement_id: str, user_id: str, user=None
  448. ) -> None:
  449. settlement = get_settlement_row(db, tenant_no, settlement_id, user=user)
  450. if settlement.settlement_status not in _DRAFT_SETTLEMENT_STATUSES:
  451. raise HTTPException(status_code=400, detail="仅草稿或已驳回结算单可删除")
  452. if user:
  453. assert_row_deletable_by_creator(settlement, user_info_from_context(user))
  454. elif user_id:
  455. assert_row_deletable_by_creator(settlement, {"user_id": user_id})
  456. now = datetime.now()
  457. settlement.del_flag = 1
  458. settlement.update_by = user_id
  459. settlement.update_time = now
  460. links = (
  461. db.query(PlFreightSettlementTransport)
  462. .filter(
  463. PlFreightSettlementTransport.settlement_id == settlement_id,
  464. PlFreightSettlementTransport.del_flag == 0,
  465. )
  466. .all()
  467. )
  468. for link in links:
  469. link.del_flag = 1
  470. link.update_time = now
  471. db.commit()
  472. def cancel_draft_settlement(
  473. db: Session, *, tenant_no: str, settlement_id: str, user_id: str, user=None
  474. ) -> PlFreightSettlement:
  475. settlement = get_settlement_row(db, tenant_no, settlement_id, user=user)
  476. if settlement.settlement_status not in _DRAFT_SETTLEMENT_STATUSES:
  477. raise HTTPException(status_code=400, detail="仅草稿或已驳回结算单可作废")
  478. if user:
  479. assert_row_deletable_by_creator(settlement, user_info_from_context(user))
  480. elif user_id:
  481. assert_row_deletable_by_creator(settlement, {"user_id": user_id})
  482. now = datetime.now()
  483. settlement.settlement_status = "cancelled"
  484. settlement.update_by = user_id
  485. settlement.update_time = now
  486. links = (
  487. db.query(PlFreightSettlementTransport)
  488. .filter(
  489. PlFreightSettlementTransport.settlement_id == settlement_id,
  490. PlFreightSettlementTransport.del_flag == 0,
  491. )
  492. .all()
  493. )
  494. for link in links:
  495. link.del_flag = 1
  496. link.update_time = now
  497. db.commit()
  498. db.refresh(settlement)
  499. return settlement
  500. def _get_transport_ids(db: Session, settlement_id: str) -> List[str]:
  501. return [
  502. r.transport_id
  503. for r in db.query(PlFreightSettlementTransport)
  504. .filter(
  505. PlFreightSettlementTransport.settlement_id == settlement_id,
  506. PlFreightSettlementTransport.del_flag == 0,
  507. )
  508. .all()
  509. ]
  510. def _get_plans_for_transports(db: Session, tenant_no: str, transport_ids: List[str]) -> List[PlPlan]:
  511. if not transport_ids:
  512. return []
  513. return (
  514. db.query(PlPlan)
  515. .filter(
  516. PlPlan.tenant_no == tenant_no,
  517. PlPlan.del_flag == 0,
  518. PlPlan.transport_id.in_(transport_ids),
  519. )
  520. .all()
  521. )
  522. def _approval_to_dict(row: PlFreightSettlementApproval) -> Dict[str, Any]:
  523. return {
  524. "id": row.id,
  525. "step": row.step,
  526. "stepName": row.step_name,
  527. "approverId": row.approver_id,
  528. "approverName": row.approver_name,
  529. "approvalType": row.approval_type,
  530. "approvalResult": row.approval_result,
  531. "approvalOpinion": row.approval_opinion,
  532. "approvalTime": row.approval_time,
  533. "isCurrent": row.is_current,
  534. }
  535. def create_settlement_approval_records(
  536. db: Session,
  537. *,
  538. tenant_no: str,
  539. settlement_id: str,
  540. approval_type: str,
  541. approver_ids: List[str],
  542. user_id: str,
  543. dept_id: str,
  544. ) -> None:
  545. approval_type = (approval_type or "sequential").strip() or "sequential"
  546. cleaned = [aid.strip() for aid in approver_ids if (aid or "").strip()]
  547. if not cleaned:
  548. raise HTTPException(status_code=400, detail="请选择审批人")
  549. if approval_type in ("parallel", "or"):
  550. step = 1
  551. for idx, aid in enumerate(cleaned):
  552. nick = _resolve_user_display_name(db, aid) or f"用户{aid}"
  553. db.add(
  554. PlFreightSettlementApproval(
  555. id=generate_uuid(),
  556. settlement_id=settlement_id,
  557. step=step,
  558. step_name="会签审批" if approval_type == "parallel" else "或签审批",
  559. approver_id=aid,
  560. approver_name=nick,
  561. approval_type=approval_type,
  562. approval_result="pending",
  563. is_current=1,
  564. tenant_no=tenant_no,
  565. create_by=user_id,
  566. create_dept_id=dept_id or "",
  567. )
  568. )
  569. else:
  570. step = 1
  571. for idx, aid in enumerate(cleaned):
  572. nick = _resolve_user_display_name(db, aid) or f"用户{aid}"
  573. db.add(
  574. PlFreightSettlementApproval(
  575. id=generate_uuid(),
  576. settlement_id=settlement_id,
  577. step=step,
  578. step_name=f"第{step}步审批",
  579. approver_id=aid,
  580. approver_name=nick,
  581. approval_type="sequential",
  582. approval_result="pending",
  583. is_current=1 if idx == 0 else 0,
  584. tenant_no=tenant_no,
  585. create_by=user_id,
  586. create_dept_id=dept_id or "",
  587. )
  588. )
  589. step += 1
  590. db.flush()
  591. def _clear_settlement_approval_records(db: Session, settlement_id: str) -> None:
  592. """软删除结算单历史审批记录(重新提交审批前清理)"""
  593. now = datetime.now()
  594. rows = (
  595. db.query(PlFreightSettlementApproval)
  596. .filter(
  597. PlFreightSettlementApproval.settlement_id == settlement_id,
  598. PlFreightSettlementApproval.del_flag == 0,
  599. )
  600. .all()
  601. )
  602. for row in rows:
  603. row.del_flag = 1
  604. row.update_time = now
  605. if rows:
  606. db.flush()
  607. def _sync_settlement_status_if_approval_complete(
  608. db: Session, settlement: PlFreightSettlement
  609. ) -> bool:
  610. """审批记录已全部完成但结算单仍停留在 approving 时自动纠正(兼容历史脏数据)"""
  611. if settlement.settlement_status != "approving":
  612. return False
  613. if not _settlement_all_approved(db, settlement.id):
  614. return False
  615. settlement.settlement_status = "approved"
  616. settlement.update_time = datetime.now()
  617. return True
  618. def _sync_settlement_status_if_disbursed(
  619. db: Session, settlement: PlFreightSettlement
  620. ) -> bool:
  621. """已有汇款凭证但结算单仍停留在 approved 时自动纠正(兼容历史脏数据)"""
  622. if settlement.settlement_status != "approved":
  623. return False
  624. voucher = (
  625. db.query(PlRemittanceVoucher)
  626. .filter(
  627. PlRemittanceVoucher.settlement_id == settlement.id,
  628. PlRemittanceVoucher.del_flag == 0,
  629. )
  630. .first()
  631. )
  632. if not voucher:
  633. return False
  634. settlement.settlement_status = "confirmed"
  635. settlement.update_time = datetime.now()
  636. return True
  637. def _sync_disbursed_settlements(db: Session, tenant_no: str) -> None:
  638. """批量纠正已有汇款凭证但仍为 approved 的结算单"""
  639. rows = (
  640. db.query(PlFreightSettlement)
  641. .filter(
  642. PlFreightSettlement.tenant_no == tenant_no,
  643. PlFreightSettlement.del_flag == 0,
  644. PlFreightSettlement.settlement_status == "approved",
  645. db.query(PlRemittanceVoucher.id)
  646. .filter(
  647. PlRemittanceVoucher.settlement_id == PlFreightSettlement.id,
  648. PlRemittanceVoucher.del_flag == 0,
  649. )
  650. .exists(),
  651. )
  652. .all()
  653. )
  654. if not rows:
  655. return
  656. now = datetime.now()
  657. for row in rows:
  658. row.settlement_status = "confirmed"
  659. row.update_time = now
  660. db.commit()
  661. def can_user_approve_settlement(db: Session, settlement_id: str, user_id: str) -> bool:
  662. if not settlement_id or not user_id:
  663. return False
  664. approvals = (
  665. db.query(PlFreightSettlementApproval)
  666. .filter(
  667. PlFreightSettlementApproval.settlement_id == settlement_id,
  668. PlFreightSettlementApproval.del_flag == 0,
  669. )
  670. .order_by(PlFreightSettlementApproval.step.asc())
  671. .all()
  672. )
  673. if not approvals:
  674. return False
  675. user_approval = next((a for a in approvals if a.approver_id == user_id), None)
  676. if not user_approval or user_approval.approval_result != "pending" or user_approval.is_current != 1:
  677. return False
  678. approval_type = user_approval.approval_type or "sequential"
  679. if approval_type == "sequential":
  680. for approval in approvals:
  681. if approval.step == user_approval.step:
  682. break
  683. if approval.approval_result != "approved":
  684. return False
  685. return True
  686. def _settlement_all_approved(db: Session, settlement_id: str) -> bool:
  687. pending = (
  688. db.query(PlFreightSettlementApproval)
  689. .filter(
  690. PlFreightSettlementApproval.settlement_id == settlement_id,
  691. PlFreightSettlementApproval.approval_result == "pending",
  692. PlFreightSettlementApproval.del_flag == 0,
  693. )
  694. .count()
  695. )
  696. return pending == 0
  697. def _advance_after_approval(
  698. db: Session, settlement: PlFreightSettlement, current: PlFreightSettlementApproval, result: str
  699. ) -> None:
  700. now = datetime.now()
  701. if result == "rejected":
  702. settlement.settlement_status = "rejected"
  703. settlement.update_time = now
  704. others = (
  705. db.query(PlFreightSettlementApproval)
  706. .filter(
  707. PlFreightSettlementApproval.settlement_id == settlement.id,
  708. PlFreightSettlementApproval.id != current.id,
  709. PlFreightSettlementApproval.approval_result == "pending",
  710. PlFreightSettlementApproval.del_flag == 0,
  711. )
  712. .all()
  713. )
  714. for row in others:
  715. row.approval_result = "skipped"
  716. row.is_current = 0
  717. row.update_time = now
  718. return
  719. approval_type = current.approval_type or "sequential"
  720. step = current.step
  721. if approval_type == "or":
  722. others = (
  723. db.query(PlFreightSettlementApproval)
  724. .filter(
  725. PlFreightSettlementApproval.settlement_id == settlement.id,
  726. PlFreightSettlementApproval.step == step,
  727. PlFreightSettlementApproval.id != current.id,
  728. PlFreightSettlementApproval.del_flag == 0,
  729. )
  730. .all()
  731. )
  732. for row in others:
  733. if row.approval_result == "pending":
  734. row.approval_result = "skipped"
  735. row.is_current = 0
  736. row.update_time = now
  737. elif approval_type == "parallel":
  738. step_pending = (
  739. db.query(PlFreightSettlementApproval)
  740. .filter(
  741. PlFreightSettlementApproval.settlement_id == settlement.id,
  742. PlFreightSettlementApproval.step == step,
  743. PlFreightSettlementApproval.approval_result == "pending",
  744. PlFreightSettlementApproval.del_flag == 0,
  745. )
  746. .count()
  747. )
  748. if step_pending > 0:
  749. return
  750. else:
  751. nxt = (
  752. db.query(PlFreightSettlementApproval)
  753. .filter(
  754. PlFreightSettlementApproval.settlement_id == settlement.id,
  755. PlFreightSettlementApproval.approval_result == "pending",
  756. PlFreightSettlementApproval.step > step,
  757. PlFreightSettlementApproval.del_flag == 0,
  758. )
  759. .order_by(PlFreightSettlementApproval.step.asc())
  760. .first()
  761. )
  762. if nxt:
  763. nxt.is_current = 1
  764. nxt.update_time = now
  765. return
  766. if _settlement_all_approved(db, settlement.id):
  767. settlement.settlement_status = "approved"
  768. settlement.update_time = now
  769. def submit_settlement_approval(
  770. db: Session,
  771. *,
  772. tenant_no: str,
  773. settlement_id: str,
  774. body: FreightSettlementSubmitApprovalBody,
  775. user_id: str,
  776. dept_id: str,
  777. user=None,
  778. ) -> PlFreightSettlement:
  779. settlement = get_settlement_row(db, tenant_no, settlement_id, user=user)
  780. if settlement.settlement_status not in _DRAFT_SETTLEMENT_STATUSES:
  781. raise HTTPException(status_code=400, detail="仅草稿或已驳回状态可提交审批")
  782. settlement.settlement_status = "approving"
  783. if body.remark:
  784. settlement.remark = body.remark
  785. settlement.update_by = user_id
  786. db.flush()
  787. _clear_settlement_approval_records(db, settlement_id)
  788. create_settlement_approval_records(
  789. db,
  790. tenant_no=tenant_no,
  791. settlement_id=settlement_id,
  792. approval_type=body.approval_type,
  793. approver_ids=body.approver_ids,
  794. user_id=user_id,
  795. dept_id=dept_id,
  796. )
  797. db.commit()
  798. db.refresh(settlement)
  799. return settlement
  800. def process_settlement_approval(
  801. db: Session,
  802. *,
  803. tenant_no: str,
  804. approval_id: str,
  805. action: str,
  806. comment: Optional[str],
  807. user_id: str,
  808. ) -> PlFreightSettlement:
  809. approval = (
  810. db.query(PlFreightSettlementApproval)
  811. .filter(
  812. PlFreightSettlementApproval.id == approval_id,
  813. PlFreightSettlementApproval.tenant_no == tenant_no,
  814. PlFreightSettlementApproval.del_flag == 0,
  815. )
  816. .first()
  817. )
  818. if not approval:
  819. raise HTTPException(status_code=404, detail="审批记录不存在")
  820. settlement = get_settlement_row(db, tenant_no, approval.settlement_id, skip_scope=True)
  821. if settlement.settlement_status != "approving":
  822. raise HTTPException(status_code=400, detail="结算单不在审批中")
  823. if approval.approver_id != user_id:
  824. raise HTTPException(status_code=403, detail="只有审批人才能审批")
  825. if not can_user_approve_settlement(db, settlement.id, user_id):
  826. raise HTTPException(status_code=400, detail="当前不可审批")
  827. result = "approved" if action == "approve" else "rejected"
  828. now = datetime.now()
  829. approval.approval_result = result
  830. approval.approval_opinion = comment or ""
  831. approval.approval_time = now
  832. approval.is_current = 0
  833. approval.update_by = user_id
  834. approval.update_time = now
  835. # SessionLocal 关闭了 autoflush,须先 flush 再查询 pending 计数,否则当前审批仍会被算作 pending
  836. db.flush()
  837. _advance_after_approval(db, settlement, approval, result)
  838. settlement.update_by = user_id
  839. db.commit()
  840. db.refresh(settlement)
  841. return settlement
  842. def withdraw_settlement_approval(
  843. db: Session, *, tenant_no: str, settlement_id: str, user_id: str, user=None
  844. ) -> PlFreightSettlement:
  845. settlement = get_settlement_row(db, tenant_no, settlement_id, user=user)
  846. if settlement.settlement_status != "approving":
  847. raise HTTPException(status_code=400, detail="仅审批中可撤回")
  848. if settlement.create_by != user_id:
  849. raise HTTPException(status_code=403, detail="仅提交人可撤回")
  850. now = datetime.now()
  851. rows = (
  852. db.query(PlFreightSettlementApproval)
  853. .filter(
  854. PlFreightSettlementApproval.settlement_id == settlement_id,
  855. PlFreightSettlementApproval.del_flag == 0,
  856. )
  857. .all()
  858. )
  859. for row in rows:
  860. row.del_flag = 1
  861. row.update_time = now
  862. settlement.settlement_status = "pending"
  863. settlement.update_by = user_id
  864. settlement.update_time = now
  865. db.commit()
  866. db.refresh(settlement)
  867. return settlement
  868. def disburse_settlement(
  869. db: Session,
  870. *,
  871. tenant_no: str,
  872. settlement_id: str,
  873. body: FreightSettlementDisburseBody,
  874. user_id: str,
  875. dept_id: str,
  876. user=None,
  877. ) -> PlFreightSettlement:
  878. settlement = get_settlement_row(db, tenant_no, settlement_id, user=user)
  879. if settlement.settlement_status != "approved":
  880. raise HTTPException(status_code=400, detail="仅待发放状态可登记")
  881. total = Decimal(str(settlement.total_freight or 0))
  882. amount = Decimal(str(body.remittance_amount))
  883. if amount != total and not (body.remark or "").strip():
  884. raise HTTPException(status_code=400, detail="汇款金额与运费合计不一致时备注必填")
  885. db.add(
  886. PlRemittanceVoucher(
  887. id=generate_uuid(),
  888. tenant_no=tenant_no,
  889. settlement_id=settlement.id,
  890. remittance_date=body.remittance_date,
  891. remittance_amount=body.remittance_amount,
  892. voucher_urls=json.dumps(body.voucher_urls),
  893. remark=body.remark,
  894. create_by=user_id,
  895. create_dept_id=dept_id or "",
  896. )
  897. )
  898. settlement.settlement_status = "confirmed"
  899. settlement.update_by = user_id
  900. db.commit()
  901. db.refresh(settlement)
  902. return settlement
  903. def get_approval_flow(db: Session, settlement_id: str) -> List[Dict[str, Any]]:
  904. rows = (
  905. db.query(PlFreightSettlementApproval)
  906. .filter(
  907. PlFreightSettlementApproval.settlement_id == settlement_id,
  908. PlFreightSettlementApproval.del_flag == 0,
  909. )
  910. .order_by(PlFreightSettlementApproval.step.asc(), PlFreightSettlementApproval.create_time.asc())
  911. .all()
  912. )
  913. return [_approval_to_dict(r) for r in rows]
  914. def get_approval_history(db: Session, settlement_id: str) -> List[Dict[str, Any]]:
  915. """返回全部已完成的审批动作(含撤回/重提前的历史轮次),按时间正序。"""
  916. rows = (
  917. db.query(PlFreightSettlementApproval)
  918. .filter(
  919. PlFreightSettlementApproval.settlement_id == settlement_id,
  920. PlFreightSettlementApproval.approval_result.in_(["approved", "rejected"]),
  921. )
  922. .order_by(
  923. PlFreightSettlementApproval.approval_time.asc(),
  924. PlFreightSettlementApproval.create_time.asc(),
  925. )
  926. .all()
  927. )
  928. return [_approval_to_dict(r) for r in rows]
  929. def _current_approver_name(db: Session, settlement_id: str) -> Optional[str]:
  930. row = (
  931. db.query(PlFreightSettlementApproval)
  932. .filter(
  933. PlFreightSettlementApproval.settlement_id == settlement_id,
  934. PlFreightSettlementApproval.approval_result == "pending",
  935. PlFreightSettlementApproval.is_current == 1,
  936. PlFreightSettlementApproval.del_flag == 0,
  937. )
  938. .order_by(PlFreightSettlementApproval.step.asc())
  939. .first()
  940. )
  941. return row.approver_name if row else None
  942. def _current_approval_id(db: Session, settlement_id: str, user_id: str) -> Optional[str]:
  943. if not can_user_approve_settlement(db, settlement_id, user_id):
  944. return None
  945. row = (
  946. db.query(PlFreightSettlementApproval)
  947. .filter(
  948. PlFreightSettlementApproval.settlement_id == settlement_id,
  949. PlFreightSettlementApproval.approver_id == user_id,
  950. PlFreightSettlementApproval.approval_result == "pending",
  951. PlFreightSettlementApproval.is_current == 1,
  952. PlFreightSettlementApproval.del_flag == 0,
  953. )
  954. .first()
  955. )
  956. return row.id if row else None
  957. def _last_approval_time(db: Session, settlement_id: str) -> Optional[datetime]:
  958. row = (
  959. db.query(PlFreightSettlementApproval)
  960. .filter(
  961. PlFreightSettlementApproval.settlement_id == settlement_id,
  962. PlFreightSettlementApproval.approval_result == "approved",
  963. PlFreightSettlementApproval.del_flag == 0,
  964. )
  965. .order_by(PlFreightSettlementApproval.approval_time.desc())
  966. .first()
  967. )
  968. return row.approval_time if row else None
  969. def enrich_settlement_list_item(
  970. db: Session, settlement: PlFreightSettlement, user_id: str
  971. ) -> Dict[str, Any]:
  972. transport_ids = _get_transport_ids(db, settlement.id)
  973. plans = _get_plans_for_transports(db, settlement.tenant_no, transport_ids)
  974. plates = [p.license_plate for p in plans if p.license_plate]
  975. if not plates:
  976. transports = (
  977. db.query(PlTransport)
  978. .filter(PlTransport.id.in_(transport_ids), PlTransport.del_flag == 0)
  979. .all()
  980. if transport_ids
  981. else []
  982. )
  983. plates = [t.license_plate for t in transports if t.license_plate]
  984. voucher = (
  985. db.query(PlRemittanceVoucher)
  986. .filter(
  987. PlRemittanceVoucher.settlement_id == settlement.id,
  988. PlRemittanceVoucher.del_flag == 0,
  989. )
  990. .order_by(PlRemittanceVoucher.create_time.desc())
  991. .first()
  992. )
  993. submitter = _resolve_user_display_name(db, settlement.create_by or "")
  994. item = {
  995. "id": settlement.id,
  996. "settlementNo": settlement.settlement_no,
  997. "totalFreight": settlement.total_freight,
  998. "invoiceAmount": settlement.invoice_amount,
  999. "invoiceNo": settlement.invoice_no,
  1000. "invoiceDate": settlement.invoice_date,
  1001. "invoiceAttachmentUrls": settlement.invoice_attachment_urls,
  1002. "settlementStatus": settlement.settlement_status,
  1003. "remark": settlement.remark,
  1004. "tenantNo": settlement.tenant_no,
  1005. "orderNum": settlement.order_num,
  1006. "createBy": settlement.create_by,
  1007. "createDeptId": settlement.create_dept_id,
  1008. "createTime": settlement.create_time,
  1009. "updateBy": settlement.update_by,
  1010. "updateTime": settlement.update_time,
  1011. "delFlag": settlement.del_flag,
  1012. "planCount": len(plans),
  1013. "licensePlateSummary": "、".join(plates[:3]) + ("…" if len(plates) > 3 else ""),
  1014. "currentApproverName": _current_approver_name(db, settlement.id),
  1015. "canApprove": can_user_approve_settlement(db, settlement.id, user_id),
  1016. "canWithdraw": settlement.settlement_status == "approving" and settlement.create_by == user_id,
  1017. "currentApprovalId": _current_approval_id(db, settlement.id, user_id),
  1018. "approvalFinishedTime": _last_approval_time(db, settlement.id),
  1019. "disburseTime": voucher.create_time if voucher else None,
  1020. "disburseAmount": voucher.remittance_amount if voucher else None,
  1021. "submitterName": submitter or None,
  1022. }
  1023. return item
  1024. def _split_status_csv(raw: Optional[str]) -> List[str]:
  1025. if not raw or not str(raw).strip():
  1026. return []
  1027. return [s.strip() for s in str(raw).split(",") if s.strip()]
  1028. def list_settlements_enriched(
  1029. db: Session,
  1030. *,
  1031. tenant_no: str,
  1032. user_id: str,
  1033. user=None,
  1034. q: Optional[str] = None,
  1035. carrier_id: Optional[str] = None,
  1036. settlement_no: Optional[str] = None,
  1037. settlement_status: Optional[str] = None,
  1038. settlement_status_in: Optional[str] = None,
  1039. tab: Optional[str] = None,
  1040. ids: Optional[List[str]] = None,
  1041. page: int = 1,
  1042. page_size: int = 20,
  1043. max_page_size: Optional[int] = None,
  1044. ) -> Tuple[int, List[Dict[str, Any]]]:
  1045. _sync_disbursed_settlements(db, tenant_no)
  1046. query = db.query(PlFreightSettlement).filter(
  1047. PlFreightSettlement.tenant_no == tenant_no,
  1048. PlFreightSettlement.del_flag == 0,
  1049. )
  1050. selected_statuses = _split_status_csv(settlement_status_in)
  1051. if not selected_statuses and settlement_status and str(settlement_status).strip():
  1052. selected_statuses = [str(settlement_status).strip()]
  1053. if tab and tab in TAB_STATUS_MAP:
  1054. tab_statuses = list(TAB_STATUS_MAP[tab])
  1055. if selected_statuses:
  1056. intersected = [s for s in selected_statuses if s in tab_statuses]
  1057. if not intersected:
  1058. query = query.filter(false())
  1059. else:
  1060. query = query.filter(PlFreightSettlement.settlement_status.in_(intersected))
  1061. else:
  1062. query = query.filter(PlFreightSettlement.settlement_status.in_(tab_statuses))
  1063. elif selected_statuses:
  1064. query = query.filter(PlFreightSettlement.settlement_status.in_(selected_statuses))
  1065. from app.services.pl.freight_list_filters import build_settlement_column_filters
  1066. column_filters = build_settlement_column_filters(settlement_no=settlement_no)
  1067. if column_filters:
  1068. query = query.filter(*column_filters)
  1069. if q:
  1070. query = query.filter(PlFreightSettlement.settlement_no.contains(q))
  1071. if carrier_id:
  1072. carrier_settlement_ids = (
  1073. db.query(PlFreightSettlementTransport.settlement_id)
  1074. .join(PlTransport, PlTransport.id == PlFreightSettlementTransport.transport_id)
  1075. .filter(
  1076. PlFreightSettlementTransport.tenant_no == tenant_no,
  1077. PlFreightSettlementTransport.del_flag == 0,
  1078. PlTransport.del_flag == 0,
  1079. PlTransport.carrier_id == carrier_id,
  1080. )
  1081. .distinct()
  1082. )
  1083. query = query.filter(PlFreightSettlement.id.in_(carrier_settlement_ids))
  1084. if ids:
  1085. query = query.filter(PlFreightSettlement.id.in_(ids))
  1086. if user and tab != "approval_pending":
  1087. query = apply_pl_scope(query, PlFreightSettlement, user, db, tenant_no)
  1088. total = query.count()
  1089. page = max(1, page)
  1090. cap = int(max_page_size) if max_page_size else 500
  1091. page_size = min(max(1, page_size), cap)
  1092. rows = (
  1093. query.order_by(PlFreightSettlement.create_time.desc())
  1094. .offset((page - 1) * page_size)
  1095. .limit(page_size)
  1096. .all()
  1097. )
  1098. status_fixed = False
  1099. freight_fixed = False
  1100. for row in rows:
  1101. if _sync_settlement_status_if_approval_complete(db, row):
  1102. status_fixed = True
  1103. if _sync_settlement_status_if_disbursed(db, row):
  1104. status_fixed = True
  1105. if refresh_draft_settlement_freight(db, row, user_id=user_id):
  1106. freight_fixed = True
  1107. if status_fixed or freight_fixed:
  1108. db.commit()
  1109. for row in rows:
  1110. db.refresh(row)
  1111. if tab == "bills":
  1112. bills_statuses = set(TAB_STATUS_MAP["bills"])
  1113. rows = [r for r in rows if r.settlement_status in bills_statuses]
  1114. return total, [enrich_settlement_list_item(db, r, user_id) for r in rows]
  1115. def get_settlement_detail(
  1116. db: Session, *, tenant_no: str, settlement_id: str, user_id: str, user=None
  1117. ) -> Dict[str, Any]:
  1118. settlement = get_settlement_row(db, tenant_no, settlement_id, user=user)
  1119. if _sync_settlement_status_if_approval_complete(db, settlement):
  1120. db.commit()
  1121. db.refresh(settlement)
  1122. if _sync_settlement_status_if_disbursed(db, settlement):
  1123. db.commit()
  1124. db.refresh(settlement)
  1125. if refresh_draft_settlement_freight(db, settlement, user_id=user_id):
  1126. db.commit()
  1127. db.refresh(settlement)
  1128. base = enrich_settlement_list_item(db, settlement, user_id)
  1129. transport_ids = _get_transport_ids(db, settlement_id)
  1130. transports = (
  1131. db.query(PlTransport)
  1132. .filter(PlTransport.id.in_(transport_ids), PlTransport.del_flag == 0)
  1133. .all()
  1134. if transport_ids
  1135. else []
  1136. )
  1137. plans = _get_plans_for_transports(db, tenant_no, transport_ids)
  1138. plan_summaries = [
  1139. {
  1140. "id": p.id,
  1141. "planNo": p.plan_no,
  1142. "planDate": p.plan_date,
  1143. "licensePlate": p.license_plate,
  1144. "freightAmount": p.freight_amount,
  1145. "freightRemark": p.freight_remark,
  1146. }
  1147. for p in plans
  1148. ]
  1149. vouchers = (
  1150. db.query(PlRemittanceVoucher)
  1151. .filter(
  1152. PlRemittanceVoucher.settlement_id == settlement_id,
  1153. PlRemittanceVoucher.del_flag == 0,
  1154. )
  1155. .order_by(PlRemittanceVoucher.create_time.desc())
  1156. .all()
  1157. )
  1158. transport_payload = []
  1159. for t in transports:
  1160. transport_payload.append(
  1161. {
  1162. "id": t.id,
  1163. "transportNo": t.transport_no,
  1164. "licensePlate": t.license_plate,
  1165. "driverName": t.driver_name,
  1166. "carrierName": t.carrier_name,
  1167. "transportMedium": t.transport_medium,
  1168. "startPoint": t.start_point,
  1169. "endPoint": t.end_point,
  1170. "freightAmount": t.freight_amount,
  1171. "transportStatus": t.transport_status,
  1172. }
  1173. )
  1174. voucher_payload = []
  1175. for v in vouchers:
  1176. voucher_payload.append(
  1177. {
  1178. "id": v.id,
  1179. "settlementId": v.settlement_id,
  1180. "remittanceDate": v.remittance_date,
  1181. "remittanceAmount": v.remittance_amount,
  1182. "voucherUrls": v.voucher_urls,
  1183. "remark": v.remark,
  1184. "createBy": v.create_by,
  1185. "createTime": v.create_time,
  1186. }
  1187. )
  1188. base["transports"] = transport_payload
  1189. base["plans"] = plan_summaries
  1190. base["approvalFlow"] = get_approval_flow(db, settlement_id)
  1191. base["approvalHistory"] = get_approval_history(db, settlement_id)
  1192. base["vouchers"] = voucher_payload
  1193. return base
  1194. def _validate_transports_for_settlement_update(
  1195. db: Session,
  1196. *,
  1197. tenant_no: str,
  1198. settlement_id: str,
  1199. transport_ids: List[str],
  1200. ) -> List[PlTransport]:
  1201. cleaned = list(dict.fromkeys([tid.strip() for tid in transport_ids if (tid or "").strip()]))
  1202. if not cleaned:
  1203. raise HTTPException(status_code=400, detail="至少保留一条关联运输")
  1204. transports = (
  1205. db.query(PlTransport)
  1206. .filter(
  1207. PlTransport.id.in_(cleaned),
  1208. PlTransport.tenant_no == tenant_no,
  1209. PlTransport.del_flag == 0,
  1210. )
  1211. .all()
  1212. )
  1213. if len(transports) != len(cleaned):
  1214. raise HTTPException(status_code=400, detail="存在无效的运输单")
  1215. transport_map = {t.id: t for t in transports}
  1216. current_ids = set(_get_transport_ids(db, settlement_id))
  1217. added_ids = [tid for tid in cleaned if tid not in current_ids]
  1218. if added_ids:
  1219. occupied = {r[0] for r in _occupied_transport_ids_subquery(db, tenant_no).all()}
  1220. conflict = [tid for tid in added_ids if tid in occupied]
  1221. if conflict:
  1222. raise HTTPException(status_code=400, detail="所选运输单已纳入其他结算单")
  1223. plans = (
  1224. db.query(PlPlan)
  1225. .filter(
  1226. PlPlan.tenant_no == tenant_no,
  1227. PlPlan.del_flag == 0,
  1228. PlPlan.transport_id.in_(added_ids),
  1229. )
  1230. .all()
  1231. )
  1232. plan_by_transport = {p.transport_id: p for p in plans if p.transport_id}
  1233. for tid in added_ids:
  1234. plan = plan_by_transport.get(tid)
  1235. if not plan:
  1236. raise HTTPException(status_code=400, detail="存在未关联计划的运输单")
  1237. if plan.archive_status != "archived":
  1238. raise HTTPException(status_code=400, detail=f"计划 {plan.plan_no} 未归档,不能结算")
  1239. if int(plan.has_freight or 0) != 1:
  1240. raise HTTPException(status_code=400, detail=f"计划 {plan.plan_no} 无运输费用")
  1241. if plan.follow_status == "cancelled":
  1242. raise HTTPException(status_code=400, detail=f"计划 {plan.plan_no} 已作废")
  1243. amount = Decimal(str(transport_map[tid].freight_amount or 0))
  1244. if amount <= 0:
  1245. raise HTTPException(status_code=400, detail=f"运输单 {transport_map[tid].transport_no} 运费无效")
  1246. return [transport_map[tid] for tid in cleaned]
  1247. def _sync_settlement_transport_links(
  1248. db: Session,
  1249. *,
  1250. tenant_no: str,
  1251. settlement_id: str,
  1252. transport_ids: List[str],
  1253. transports: List[PlTransport],
  1254. user_id: str,
  1255. dept_id: str,
  1256. ) -> None:
  1257. cleaned = list(dict.fromkeys([tid.strip() for tid in transport_ids if (tid or "").strip()]))
  1258. transport_map = {t.id: t for t in transports}
  1259. now = datetime.now()
  1260. current_links = (
  1261. db.query(PlFreightSettlementTransport)
  1262. .filter(
  1263. PlFreightSettlementTransport.settlement_id == settlement_id,
  1264. PlFreightSettlementTransport.del_flag == 0,
  1265. )
  1266. .all()
  1267. )
  1268. current_map = {link.transport_id: link for link in current_links}
  1269. new_set = set(cleaned)
  1270. for tid, link in current_map.items():
  1271. if tid not in new_set:
  1272. link.del_flag = 1
  1273. link.update_by = user_id
  1274. link.update_time = now
  1275. for tid in cleaned:
  1276. transport = transport_map[tid]
  1277. amount = transport.freight_amount
  1278. if tid in current_map:
  1279. link = current_map[tid]
  1280. if Decimal(str(link.freight_amount or 0)) != Decimal(str(amount or 0)):
  1281. link.freight_amount = amount
  1282. link.update_by = user_id
  1283. link.update_time = now
  1284. continue
  1285. db.add(
  1286. PlFreightSettlementTransport(
  1287. id=generate_uuid(),
  1288. tenant_no=tenant_no,
  1289. settlement_id=settlement_id,
  1290. transport_id=tid,
  1291. freight_amount=amount,
  1292. create_by=user_id,
  1293. create_dept_id=dept_id or "",
  1294. )
  1295. )
  1296. _recalc_settlement_total_freight_from_links(db, settlement_id, user_id=user_id, now=now)
  1297. def update_settlement_draft(
  1298. db: Session,
  1299. *,
  1300. tenant_no: str,
  1301. settlement_id: str,
  1302. payload: FreightSettlementUpdate,
  1303. user_id: str,
  1304. user=None,
  1305. ) -> PlFreightSettlement:
  1306. settlement = get_settlement_row(db, tenant_no, settlement_id, user=user)
  1307. status = settlement.settlement_status
  1308. if status not in (*_DRAFT_SETTLEMENT_STATUSES, "approved"):
  1309. raise HTTPException(status_code=400, detail="当前状态不可编辑")
  1310. update_data = payload.model_dump(exclude_unset=True, by_alias=False)
  1311. transport_ids = update_data.pop("transport_ids", None)
  1312. if status == "approved":
  1313. if transport_ids is not None:
  1314. raise HTTPException(status_code=400, detail="当前状态不可修改关联运输")
  1315. allowed = {"invoice_amount", "invoice_no", "invoice_date", "invoice_attachment_urls", "remark"}
  1316. update_data = {k: v for k, v in update_data.items() if k in allowed}
  1317. if not update_data:
  1318. raise HTTPException(status_code=400, detail="无有效更新字段")
  1319. elif status in _DRAFT_SETTLEMENT_STATUSES and transport_ids is not None:
  1320. dept_id = getattr(user, "dept_id", "") if user else ""
  1321. transports = _validate_transports_for_settlement_update(
  1322. db,
  1323. tenant_no=tenant_no,
  1324. settlement_id=settlement_id,
  1325. transport_ids=transport_ids,
  1326. )
  1327. _sync_settlement_transport_links(
  1328. db,
  1329. tenant_no=tenant_no,
  1330. settlement_id=settlement_id,
  1331. transport_ids=transport_ids,
  1332. transports=transports,
  1333. user_id=user_id,
  1334. dept_id=dept_id,
  1335. )
  1336. for key, value in update_data.items():
  1337. setattr(settlement, key, value)
  1338. settlement.update_by = user_id
  1339. db.commit()
  1340. db.refresh(settlement)
  1341. return settlement