pl_freight_settlement_service.py 47 KB

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