pl_plan_service.py 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627
  1. """运输计划业务逻辑"""
  2. import json
  3. from datetime import date, datetime
  4. from decimal import Decimal
  5. from typing import Optional, Union
  6. from fastapi import HTTPException
  7. from sqlalchemy import and_, exists, func, or_, select
  8. from sqlalchemy.orm import Session
  9. from app.core.utils import generate_uuid
  10. from app.models.crm_models import CrmProduct
  11. from app.models.pl_models import (
  12. PlCarrier,
  13. PlFreightSettlement,
  14. PlFreightSettlementTransport,
  15. PlPersonnel,
  16. PlPlan,
  17. PlPlanChange,
  18. PlPlanContract,
  19. PlPlanProduct,
  20. PlPlanTransport,
  21. PlTransport,
  22. PlTransportSiteLink,
  23. PlVehicle,
  24. PlVehiclePersonnel,
  25. )
  26. from app.schemas.pl.plan import (
  27. PlanArchiveFreightBlock,
  28. PlanArchiveProductItem,
  29. PlanContractLink,
  30. PlanCreateFull,
  31. PlanDriverFeedbackUpdateBody,
  32. PlanProductCreate,
  33. PlanSiteFeedbackUpdateBody,
  34. PlanTransportCreateBlock,
  35. )
  36. from app.services.pl_internal_customer import is_internal_customer_name
  37. from app.services.pl_brand_archive import persist_plan_product_brand_qtys, validate_brand_qty_rows
  38. from app.services.pl_feedback_brand_qty import persist_transport_product_feedback
  39. from app.services.pl_freight_settlement_service import sync_draft_settlement_freight_for_transport
  40. from app.services.pl_transport_resource_snapshot_service import (
  41. apply_resource_snapshot_json,
  42. build_transport_resource_snapshot,
  43. clear_resource_snapshot,
  44. )
  45. def determine_plan_type(supplier_name: Optional[str], destination_name: Optional[str]) -> str:
  46. if destination_name and ("安庆山峡" in destination_name or "安徽云栋" in destination_name):
  47. return "inbound"
  48. if supplier_name and ("安庆山峡" in supplier_name or "安徽云栋" in supplier_name):
  49. return "outbound"
  50. return "direct"
  51. def is_ton_product_unit(unit: Optional[str]) -> bool:
  52. code = (unit or "").strip()
  53. if not code:
  54. return True
  55. return code in ("dun", "吨")
  56. def _product_quantity(prod: PlanProductCreate) -> Decimal:
  57. if prod.quantity is not None:
  58. return Decimal(str(prod.quantity))
  59. if prod.inbound_estimate_qty is not None:
  60. return Decimal(str(prod.inbound_estimate_qty))
  61. if prod.outbound_estimate_qty is not None:
  62. return Decimal(str(prod.outbound_estimate_qty))
  63. return Decimal("0")
  64. def _product_create_actual_qty(prod: PlanProductCreate) -> Decimal:
  65. if prod.actual_qty is not None:
  66. return Decimal(str(prod.actual_qty))
  67. return Decimal("0")
  68. def _plan_product_actual_qty(prod: PlPlanProduct) -> Decimal:
  69. if prod.actual_qty is not None:
  70. return Decimal(str(prod.actual_qty))
  71. return Decimal("0")
  72. def sum_product_actual_quantities(products: list[PlanProductCreate]) -> Decimal:
  73. return sum(_product_create_actual_qty(p) for p in products)
  74. def sum_plan_product_actual_quantities(products: list[PlPlanProduct]) -> Decimal:
  75. return sum(_plan_product_actual_qty(p) for p in products)
  76. def _estimate_qty_fields(plan_type: str, qty: Optional[Decimal]) -> dict:
  77. if qty is None:
  78. return {}
  79. q = float(qty)
  80. if plan_type == "inbound":
  81. return {"inbound_estimate_qty": q, "outbound_estimate_qty": q}
  82. if plan_type == "outbound":
  83. return {"outbound_estimate_qty": q, "inbound_estimate_qty": q}
  84. return {"inbound_estimate_qty": q, "outbound_estimate_qty": q}
  85. def build_plan_product(
  86. *,
  87. tenant_no: str,
  88. plan_id: str,
  89. plan_type: str,
  90. prod: PlanProductCreate,
  91. order_num: int,
  92. user_id: str,
  93. dept_id: str,
  94. ) -> PlPlanProduct:
  95. qty = _product_quantity(prod)
  96. estimates = _estimate_qty_fields(plan_type, qty if qty else None)
  97. tax_price = prod.tax_price
  98. return PlPlanProduct(
  99. id=generate_uuid(),
  100. tenant_no=tenant_no,
  101. plan_id=plan_id,
  102. product_id=prod.product_id,
  103. product_no=prod.product_no,
  104. product_name=prod.product_name,
  105. brand_mode=prod.brand_mode,
  106. brand_id=prod.brand_id,
  107. brand_name=prod.brand_name,
  108. model=prod.model,
  109. unit=prod.unit or "吨",
  110. category_name=prod.category_name,
  111. package_spec=prod.package_spec,
  112. delivery_method=prod.delivery_method,
  113. quantity=qty if qty else None,
  114. actual_qty=Decimal(str(prod.actual_qty)) if prod.actual_qty is not None else None,
  115. tax_price=tax_price,
  116. tax_amount=prod.tax_amount,
  117. tax_subtotal=prod.tax_subtotal,
  118. is_explosive=prod.is_explosive,
  119. is_toxic=prod.is_toxic,
  120. purchase_cert_url=prod.purchase_cert_url,
  121. transport_cert_url=prod.transport_cert_url,
  122. legal_use_cert_url=prod.legal_use_cert_url,
  123. inbound_price=tax_price,
  124. outbound_price=tax_price,
  125. inbound_amount=prod.tax_subtotal,
  126. outbound_amount=prod.tax_subtotal,
  127. remark=prod.remark,
  128. order_num=order_num,
  129. create_by=user_id,
  130. create_dept_id=dept_id,
  131. **estimates,
  132. )
  133. def _apply_vehicle_snapshot_to_plan_transport(
  134. db: Session,
  135. *,
  136. tenant_no: str,
  137. plan: PlPlan,
  138. transport: Optional[PlTransport],
  139. vehicle: PlVehicle,
  140. driver_id: Optional[str],
  141. ) -> None:
  142. snapshot = build_transport_resource_snapshot(
  143. db,
  144. tenant_no,
  145. vehicle.id,
  146. selected_driver_id=driver_id,
  147. )
  148. apply_resource_snapshot_json(plan=plan, transport=transport, snapshot=snapshot)
  149. def _resolve_driver(
  150. db: Session,
  151. tenant_no: str,
  152. vehicle_id: str,
  153. driver_id: Optional[str],
  154. ) -> tuple[Optional[str], Optional[str], Optional[str]]:
  155. """解析司机:显式选择优先,否则取车辆关联司机;均无则允许为空。"""
  156. if driver_id:
  157. personnel = (
  158. db.query(PlPersonnel)
  159. .filter(
  160. PlPersonnel.id == driver_id,
  161. PlPersonnel.tenant_no == tenant_no,
  162. PlPersonnel.del_flag == 0,
  163. )
  164. .first()
  165. )
  166. if personnel:
  167. return personnel.id, personnel.personnel_name, personnel.phone_no
  168. default_link = (
  169. db.query(PlVehiclePersonnel)
  170. .filter(
  171. PlVehiclePersonnel.vehicle_id == vehicle_id,
  172. PlVehiclePersonnel.tenant_no == tenant_no,
  173. PlVehiclePersonnel.del_flag == 0,
  174. PlVehiclePersonnel.role == "driver",
  175. )
  176. .order_by(PlVehiclePersonnel.is_default.desc(), PlVehiclePersonnel.order_num.asc())
  177. .first()
  178. )
  179. if default_link:
  180. personnel = (
  181. db.query(PlPersonnel)
  182. .filter(
  183. PlPersonnel.id == default_link.personnel_id,
  184. PlPersonnel.del_flag == 0,
  185. )
  186. .first()
  187. )
  188. if personnel:
  189. return personnel.id, personnel.personnel_name, personnel.phone_no
  190. return None, None, None
  191. def _calc_freight_amount(
  192. freight_type: str,
  193. products: list[PlPlanProduct],
  194. transport_qty: Optional[Decimal],
  195. freight_price: Optional[Decimal],
  196. freight_amount: Optional[Decimal],
  197. ) -> Decimal:
  198. if freight_type == "by_trip":
  199. return Decimal(str(freight_amount or 0))
  200. qty = transport_qty
  201. if qty is None:
  202. qty = sum_plan_product_actual_quantities(products)
  203. if freight_price is not None:
  204. computed = qty * Decimal(str(freight_price))
  205. if freight_amount is not None:
  206. return Decimal(str(freight_amount))
  207. return computed
  208. return Decimal(str(freight_amount or 0))
  209. def fetch_hazard_flags_map(
  210. db: Session,
  211. tenant_no: str,
  212. product_ids: Optional[list[str]] = None,
  213. ) -> dict[str, tuple[int, int]]:
  214. """批量从 CRM 产品主数据读取易制毒/易制爆标识。"""
  215. from app.models.crm_models import CrmProduct
  216. ids = [str(pid) for pid in (product_ids or []) if pid]
  217. if not ids:
  218. return {}
  219. rows = (
  220. db.query(CrmProduct.id, CrmProduct.is_toxic, CrmProduct.is_explosive)
  221. .filter(
  222. CrmProduct.id.in_(ids),
  223. CrmProduct.tenant_no == tenant_no,
  224. CrmProduct.del_flag == 0,
  225. )
  226. .all()
  227. )
  228. return {
  229. str(row.id): (int(row.is_toxic or 0), int(row.is_explosive or 0))
  230. for row in rows
  231. }
  232. def resolve_hazard_flags_from_crm(
  233. db: Session,
  234. tenant_no: str,
  235. product_id: Optional[str],
  236. *,
  237. hazard_map: Optional[dict[str, tuple[int, int]]] = None,
  238. ) -> tuple[int, int]:
  239. """从 CRM 产品主数据读取易制毒/易制爆标识(不读计划产品快照)。"""
  240. if not product_id:
  241. return 0, 0
  242. if hazard_map is not None:
  243. return hazard_map.get(str(product_id), (0, 0))
  244. from app.models.crm_models import CrmProduct
  245. row = (
  246. db.query(CrmProduct)
  247. .filter(
  248. CrmProduct.id == product_id,
  249. CrmProduct.tenant_no == tenant_no,
  250. CrmProduct.del_flag == 0,
  251. )
  252. .first()
  253. )
  254. if not row:
  255. return 0, 0
  256. return int(row.is_toxic or 0), int(row.is_explosive or 0)
  257. def plan_has_hazard_products(
  258. db: Session,
  259. tenant_no: str,
  260. products: list[PlPlanProduct],
  261. ) -> bool:
  262. return bool(collect_hazard_product_ids(db, tenant_no, products))
  263. def _cert_url_blank(column):
  264. """证书 URL 为空(NULL / 空串 / 纯空白)。"""
  265. return func.nullif(func.trim(func.coalesce(column, "")), "").is_(None)
  266. def plan_has_hazard_product_exists(tenant_no: str):
  267. """列表筛选:计划下存在 CRM 标识为易制毒/易制爆的产品。"""
  268. return exists(
  269. select(PlPlanProduct.id)
  270. .select_from(PlPlanProduct)
  271. .join(
  272. CrmProduct,
  273. and_(
  274. CrmProduct.id == PlPlanProduct.product_id,
  275. CrmProduct.tenant_no == tenant_no,
  276. CrmProduct.del_flag == 0,
  277. ),
  278. )
  279. .where(
  280. PlPlanProduct.plan_id == PlPlan.id,
  281. PlPlanProduct.tenant_no == tenant_no,
  282. PlPlanProduct.del_flag == 0,
  283. or_(CrmProduct.is_toxic == 1, CrmProduct.is_explosive == 1),
  284. )
  285. )
  286. def plan_has_incomplete_hazard_cert_exists(tenant_no: str):
  287. """列表筛选:存在危化产品且至少一张必需证书未上传。"""
  288. toxic_incomplete = and_(
  289. CrmProduct.is_toxic == 1,
  290. or_(
  291. _cert_url_blank(PlPlanProduct.purchase_cert_url),
  292. _cert_url_blank(PlPlanProduct.transport_cert_url),
  293. ),
  294. )
  295. explosive_incomplete = and_(
  296. CrmProduct.is_explosive == 1,
  297. _cert_url_blank(PlPlanProduct.legal_use_cert_url),
  298. )
  299. return exists(
  300. select(PlPlanProduct.id)
  301. .select_from(PlPlanProduct)
  302. .join(
  303. CrmProduct,
  304. and_(
  305. CrmProduct.id == PlPlanProduct.product_id,
  306. CrmProduct.tenant_no == tenant_no,
  307. CrmProduct.del_flag == 0,
  308. ),
  309. )
  310. .where(
  311. PlPlanProduct.plan_id == PlPlan.id,
  312. PlPlanProduct.tenant_no == tenant_no,
  313. PlPlanProduct.del_flag == 0,
  314. or_(toxic_incomplete, explosive_incomplete),
  315. )
  316. )
  317. def collect_hazard_product_ids(
  318. db: Session,
  319. tenant_no: str,
  320. products: list[Union[PlPlanProduct, PlanProductCreate]],
  321. ) -> frozenset[str]:
  322. product_ids = [str(p.product_id) for p in products if getattr(p, "product_id", None)]
  323. if not product_ids:
  324. return frozenset()
  325. hazard_map = fetch_hazard_flags_map(db, tenant_no, product_ids)
  326. hazard_ids: set[str] = set()
  327. for prod in products:
  328. product_id = getattr(prod, "product_id", None)
  329. if not product_id:
  330. continue
  331. is_toxic, is_explosive = resolve_hazard_flags_from_crm(
  332. db,
  333. tenant_no,
  334. product_id,
  335. hazard_map=hazard_map,
  336. )
  337. if is_toxic or is_explosive:
  338. hazard_ids.add(str(product_id))
  339. return frozenset(hazard_ids)
  340. def clear_hazard_cert_ready(plan: PlPlan) -> None:
  341. plan.hazard_cert_ready = 0
  342. plan.hazard_cert_ready_by = None
  343. plan.hazard_cert_ready_time = None
  344. def _required_hazard_cert_fields(is_toxic: int, is_explosive: int) -> list[str]:
  345. fields: list[str] = []
  346. if is_toxic:
  347. fields.extend(["purchase_cert_url", "transport_cert_url"])
  348. if is_explosive:
  349. fields.append("legal_use_cert_url")
  350. return fields
  351. def _is_hazard_product_cert_complete(
  352. product: PlPlanProduct,
  353. *,
  354. is_toxic: int,
  355. is_explosive: int,
  356. ) -> bool:
  357. for field in _required_hazard_cert_fields(is_toxic, is_explosive):
  358. if not str(getattr(product, field, "") or "").strip():
  359. return False
  360. return True
  361. def all_hazard_products_cert_complete(
  362. db: Session,
  363. tenant_no: str,
  364. products: list[PlPlanProduct],
  365. ) -> bool:
  366. """计划内所有危化品产品的所需证书槽位均已上传。"""
  367. if not products:
  368. return True
  369. hazard_map = fetch_hazard_flags_map(
  370. db,
  371. tenant_no,
  372. [product.product_id for product in products if product.product_id],
  373. )
  374. has_hazard = False
  375. for product in products:
  376. is_toxic, is_explosive = resolve_hazard_flags_from_crm(
  377. db,
  378. tenant_no,
  379. product.product_id,
  380. hazard_map=hazard_map,
  381. )
  382. if not (is_toxic or is_explosive):
  383. continue
  384. has_hazard = True
  385. if not _is_hazard_product_cert_complete(
  386. product,
  387. is_toxic=is_toxic,
  388. is_explosive=is_explosive,
  389. ):
  390. return False
  391. return has_hazard
  392. def maybe_auto_set_hazard_cert_ready(
  393. db: Session,
  394. *,
  395. tenant_no: str,
  396. plan: PlPlan,
  397. products: list[PlPlanProduct],
  398. user_id: str,
  399. ) -> None:
  400. """全部危化品证书已上传时,自动标记证书齐备(与人工确认等效)。"""
  401. if int(plan.hazard_cert_ready or 0):
  402. return
  403. if not all_hazard_products_cert_complete(db, tenant_no, products):
  404. return
  405. plan.hazard_cert_ready = 1
  406. plan.hazard_cert_ready_by = user_id
  407. plan.hazard_cert_ready_time = datetime.now()
  408. def validate_hazard_cert_ready_for_transport(
  409. db: Session,
  410. tenant_no: str,
  411. plan: PlPlan,
  412. products: list[PlPlanProduct],
  413. ) -> None:
  414. if not plan_has_hazard_products(db, tenant_no, products):
  415. return
  416. if int(plan.hazard_cert_ready or 0):
  417. return
  418. if all_hazard_products_cert_complete(db, tenant_no, products):
  419. return
  420. raise HTTPException(
  421. status_code=400,
  422. detail="该计划危化品证书尚未确认齐备,请至危化品证上传页操作",
  423. )
  424. def set_hazard_cert_ready(
  425. db: Session,
  426. *,
  427. tenant_no: str,
  428. plan_id: str,
  429. ready: bool,
  430. user_id: str,
  431. ) -> PlPlan:
  432. plan = (
  433. db.query(PlPlan)
  434. .filter(
  435. PlPlan.id == plan_id,
  436. PlPlan.tenant_no == tenant_no,
  437. PlPlan.del_flag == 0,
  438. )
  439. .first()
  440. )
  441. if not plan:
  442. raise HTTPException(status_code=404, detail="计划不存在")
  443. if plan.archive_status == "archived":
  444. raise HTTPException(status_code=400, detail="已归档的计划不能修改证书齐备状态")
  445. if plan.follow_status == "cancelled":
  446. raise HTTPException(status_code=400, detail="已作废的计划不能修改证书齐备状态")
  447. products = (
  448. db.query(PlPlanProduct)
  449. .filter(
  450. PlPlanProduct.plan_id == plan_id,
  451. PlPlanProduct.tenant_no == tenant_no,
  452. PlPlanProduct.del_flag == 0,
  453. )
  454. .all()
  455. )
  456. has_hazard = plan_has_hazard_products(db, tenant_no, products)
  457. if ready:
  458. if not has_hazard:
  459. raise HTTPException(status_code=400, detail="该计划不含易制毒/易制爆产品,无需确认证书齐备")
  460. plan.hazard_cert_ready = 1
  461. plan.hazard_cert_ready_by = user_id
  462. plan.hazard_cert_ready_time = datetime.now()
  463. else:
  464. clear_hazard_cert_ready(plan)
  465. plan.update_by = user_id
  466. db.commit()
  467. db.refresh(plan)
  468. return plan
  469. def update_plan_product_certs(
  470. db: Session,
  471. *,
  472. tenant_no: str,
  473. plan_id: str,
  474. items: list,
  475. user_id: str,
  476. ) -> list[PlPlanProduct]:
  477. plan = (
  478. db.query(PlPlan)
  479. .filter(
  480. PlPlan.id == plan_id,
  481. PlPlan.tenant_no == tenant_no,
  482. PlPlan.del_flag == 0,
  483. )
  484. .first()
  485. )
  486. if not plan:
  487. raise HTTPException(status_code=404, detail="计划不存在")
  488. if plan.archive_status == "archived":
  489. raise HTTPException(status_code=400, detail="已归档的计划不能修改")
  490. existing_rows = (
  491. db.query(PlPlanProduct)
  492. .filter(
  493. PlPlanProduct.plan_id == plan_id,
  494. PlPlanProduct.tenant_no == tenant_no,
  495. PlPlanProduct.del_flag == 0,
  496. )
  497. .all()
  498. )
  499. hazard_map = fetch_hazard_flags_map(
  500. db,
  501. tenant_no,
  502. [row.product_id for row in existing_rows if row.product_id],
  503. )
  504. updated: list[PlPlanProduct] = []
  505. for item in items:
  506. row = (
  507. db.query(PlPlanProduct)
  508. .filter(
  509. PlPlanProduct.id == item.id,
  510. PlPlanProduct.plan_id == plan_id,
  511. PlPlanProduct.tenant_no == tenant_no,
  512. PlPlanProduct.del_flag == 0,
  513. )
  514. .first()
  515. )
  516. if not row:
  517. raise HTTPException(status_code=404, detail=f"计划产品不存在: {item.id}")
  518. is_toxic, is_explosive = resolve_hazard_flags_from_crm(
  519. db, tenant_no, row.product_id, hazard_map=hazard_map
  520. )
  521. if not (is_toxic or is_explosive):
  522. raise HTTPException(status_code=400, detail=f"产品【{row.product_name}】无需上传危化品证书")
  523. row.purchase_cert_url = (item.purchase_cert_url or "").strip() or None
  524. row.transport_cert_url = (item.transport_cert_url or "").strip() or None
  525. row.legal_use_cert_url = (item.legal_use_cert_url or "").strip() or None
  526. row.update_by = user_id
  527. updated.append(row)
  528. maybe_auto_set_hazard_cert_ready(
  529. db,
  530. tenant_no=tenant_no,
  531. plan=plan,
  532. products=existing_rows,
  533. user_id=user_id,
  534. )
  535. plan.update_by = user_id
  536. db.commit()
  537. for row in updated:
  538. db.refresh(row)
  539. return updated
  540. def _generate_transport_no(db: Session) -> str:
  541. now = date.today()
  542. prefix = f"SXYS{now.strftime('%Y%m')}"
  543. last = (
  544. db.query(PlTransport)
  545. .filter(PlTransport.transport_no.like(f"{prefix}%"))
  546. .order_by(PlTransport.transport_no.desc())
  547. .first()
  548. )
  549. seq = 1
  550. if last and last.transport_no:
  551. try:
  552. seq = int(last.transport_no[-6:]) + 1
  553. except ValueError:
  554. seq = 1
  555. return f"{prefix}{seq:06d}"
  556. def _resolve_transport_context(
  557. db: Session,
  558. *,
  559. tenant_no: str,
  560. plan: PlPlan,
  561. transport_payload: PlanTransportCreateBlock,
  562. products: list[PlPlanProduct],
  563. ) -> tuple[
  564. PlVehicle,
  565. Optional[str],
  566. Optional[str],
  567. Optional[str],
  568. Optional[str],
  569. Optional[str],
  570. Decimal,
  571. Optional[Decimal],
  572. ]:
  573. vehicle = (
  574. db.query(PlVehicle)
  575. .filter(
  576. PlVehicle.id == transport_payload.vehicle_id,
  577. PlVehicle.tenant_no == tenant_no,
  578. PlVehicle.del_flag == 0,
  579. )
  580. .first()
  581. )
  582. if not vehicle:
  583. raise HTTPException(status_code=400, detail="车辆不存在")
  584. driver_id, driver_name, driver_phone = _resolve_driver(
  585. db, tenant_no, transport_payload.vehicle_id, transport_payload.driver_id
  586. )
  587. carrier_id = transport_payload.carrier_id or vehicle.carrier_id
  588. carrier_name = vehicle.carrier_name
  589. if carrier_id:
  590. carrier = (
  591. db.query(PlCarrier)
  592. .filter(PlCarrier.id == carrier_id, PlCarrier.del_flag == 0)
  593. .first()
  594. )
  595. if carrier:
  596. carrier_name = carrier.carrier_name
  597. freight_amount = _calc_freight_amount(
  598. transport_payload.freight_type,
  599. products,
  600. transport_payload.transport_qty,
  601. transport_payload.freight_price,
  602. transport_payload.freight_amount,
  603. )
  604. transport_qty = transport_payload.transport_qty
  605. if transport_qty is None and transport_payload.freight_type == "by_weight":
  606. transport_qty = sum_plan_product_actual_quantities(products)
  607. return (
  608. vehicle,
  609. driver_id,
  610. driver_name,
  611. driver_phone,
  612. carrier_id,
  613. carrier_name,
  614. freight_amount,
  615. transport_qty,
  616. )
  617. def _apply_payload_to_transport_row(
  618. transport: PlTransport,
  619. *,
  620. vehicle: PlVehicle,
  621. driver_id: Optional[str],
  622. driver_name: Optional[str],
  623. driver_phone: Optional[str],
  624. carrier_id: Optional[str],
  625. carrier_name: Optional[str],
  626. plan: PlPlan,
  627. transport_payload: PlanTransportCreateBlock,
  628. freight_amount: Decimal,
  629. transport_qty: Optional[Decimal],
  630. user_id: str,
  631. ) -> None:
  632. transport.vehicle_id = vehicle.id
  633. transport.license_plate = vehicle.license_plate
  634. transport.driver_id = driver_id
  635. transport.driver_name = driver_name
  636. transport.driver_phone = driver_phone
  637. transport.carrier_id = carrier_id
  638. transport.carrier_name = carrier_name
  639. transport.start_point = plan.pickup_address
  640. transport.end_point = plan.destination_address or plan.destination_name
  641. transport.distance_km = transport_payload.distance_km
  642. transport.transport_qty = transport_qty
  643. transport.freight_price = transport_payload.freight_price
  644. transport.freight_amount = freight_amount or 0
  645. transport.freight_type = transport_payload.freight_type
  646. if transport_payload.has_invoice is not None:
  647. transport.has_invoice = transport_payload.has_invoice
  648. transport.unit = "吨" if transport_payload.freight_type == "by_weight" else "趟"
  649. if transport_payload.remark is not None:
  650. transport.remark = transport_payload.remark
  651. transport.update_by = user_id
  652. def upsert_draft_transport_for_plan(
  653. db: Session,
  654. *,
  655. tenant_no: str,
  656. plan: PlPlan,
  657. transport_payload: PlanTransportCreateBlock,
  658. products: list[PlPlanProduct],
  659. user_id: str,
  660. dept_id: str,
  661. ) -> None:
  662. """写入/更新草稿运输单(pl_transport + pl_plan_transport),计划保持待安排。"""
  663. (
  664. vehicle,
  665. driver_id,
  666. driver_name,
  667. driver_phone,
  668. carrier_id,
  669. carrier_name,
  670. freight_amount,
  671. transport_qty,
  672. ) = _resolve_transport_context(
  673. db,
  674. tenant_no=tenant_no,
  675. plan=plan,
  676. transport_payload=transport_payload,
  677. products=products,
  678. )
  679. transport: Optional[PlTransport] = None
  680. if plan.transport_id:
  681. transport = (
  682. db.query(PlTransport)
  683. .filter(
  684. PlTransport.id == plan.transport_id,
  685. PlTransport.tenant_no == tenant_no,
  686. PlTransport.del_flag == 0,
  687. )
  688. .first()
  689. )
  690. if transport:
  691. _apply_payload_to_transport_row(
  692. transport,
  693. vehicle=vehicle,
  694. driver_id=driver_id,
  695. driver_name=driver_name,
  696. driver_phone=driver_phone,
  697. carrier_id=carrier_id,
  698. carrier_name=carrier_name,
  699. plan=plan,
  700. transport_payload=transport_payload,
  701. freight_amount=freight_amount,
  702. transport_qty=transport_qty,
  703. user_id=user_id,
  704. )
  705. if transport.transport_status == "draft":
  706. pass
  707. else:
  708. transport_no = _generate_transport_no(db)
  709. transport = PlTransport(
  710. id=generate_uuid(),
  711. tenant_no=tenant_no,
  712. transport_no=transport_no,
  713. transport_status="draft",
  714. approval_status="none",
  715. remark=transport_payload.remark,
  716. create_by=user_id,
  717. create_dept_id=dept_id,
  718. )
  719. db.add(transport)
  720. db.flush()
  721. db.add(
  722. PlPlanTransport(
  723. id=generate_uuid(),
  724. tenant_no=tenant_no,
  725. plan_id=plan.id,
  726. transport_id=transport.id,
  727. create_by=user_id,
  728. create_dept_id=dept_id,
  729. )
  730. )
  731. plan.transport_id = transport.id
  732. plan.transport_no = transport_no
  733. _apply_payload_to_transport_row(
  734. transport,
  735. vehicle=vehicle,
  736. driver_id=driver_id,
  737. driver_name=driver_name,
  738. driver_phone=driver_phone,
  739. carrier_id=carrier_id,
  740. carrier_name=carrier_name,
  741. plan=plan,
  742. transport_payload=transport_payload,
  743. freight_amount=freight_amount,
  744. transport_qty=transport_qty,
  745. user_id=user_id,
  746. )
  747. plan.vehicle_id = vehicle.id
  748. plan.license_plate = vehicle.license_plate
  749. plan.driver_id = driver_id
  750. plan.driver_name = driver_name
  751. _apply_vehicle_snapshot_to_plan_transport(
  752. db,
  753. tenant_no=tenant_no,
  754. plan=plan,
  755. transport=transport,
  756. vehicle=vehicle,
  757. driver_id=driver_id,
  758. )
  759. plan.freight_amount = freight_amount or 0
  760. if plan.transport_id:
  761. sync_draft_settlement_freight_for_transport(
  762. db,
  763. tenant_no=tenant_no,
  764. transport_id=plan.transport_id,
  765. freight_amount=freight_amount or Decimal("0"),
  766. user_id=user_id,
  767. )
  768. if plan.transport_status != "arranged":
  769. plan.transport_status = "pending"
  770. def clear_draft_transport_for_plan(
  771. db: Session,
  772. *,
  773. tenant_no: str,
  774. plan: PlPlan,
  775. user_id: str,
  776. ) -> None:
  777. """清除计划上的草稿运输单及车辆预选。"""
  778. transport: Optional[PlTransport] = None
  779. if plan.transport_id:
  780. transport = (
  781. db.query(PlTransport)
  782. .filter(
  783. PlTransport.id == plan.transport_id,
  784. PlTransport.tenant_no == tenant_no,
  785. PlTransport.del_flag == 0,
  786. )
  787. .first()
  788. )
  789. if transport and transport.transport_status == "draft":
  790. transport.del_flag = 1
  791. transport.update_by = user_id
  792. db.query(PlPlanTransport).filter(
  793. PlPlanTransport.plan_id == plan.id,
  794. PlPlanTransport.transport_id == transport.id,
  795. PlPlanTransport.tenant_no == tenant_no,
  796. PlPlanTransport.del_flag == 0,
  797. ).update({"del_flag": 1, "update_by": user_id})
  798. plan.transport_id = None
  799. plan.transport_no = None
  800. plan.vehicle_id = None
  801. plan.license_plate = None
  802. plan.driver_id = None
  803. plan.driver_name = None
  804. clear_resource_snapshot(plan=plan, transport=transport)
  805. plan.freight_amount = 0
  806. if plan.transport_status != "arranged":
  807. plan.transport_status = "pending"
  808. def sync_transport_prefill_for_plan(
  809. db: Session,
  810. *,
  811. tenant_no: str,
  812. plan: PlPlan,
  813. transport_payload: PlanTransportCreateBlock,
  814. products: list[PlPlanProduct],
  815. user_id: str,
  816. dept_id: str = "",
  817. ) -> None:
  818. """同步运输预选到草稿运输子表(pl_transport)。"""
  819. upsert_draft_transport_for_plan(
  820. db,
  821. tenant_no=tenant_no,
  822. plan=plan,
  823. transport_payload=transport_payload,
  824. products=products,
  825. user_id=user_id,
  826. dept_id=dept_id or user_id,
  827. )
  828. def apply_vehicle_prefill_to_plan(
  829. db: Session,
  830. *,
  831. tenant_no: str,
  832. plan: PlPlan,
  833. transport_payload: PlanTransportCreateBlock,
  834. products: list[PlPlanProduct],
  835. user_id: str = "",
  836. dept_id: str = "",
  837. ) -> None:
  838. """兼容旧调用:写入草稿运输子表。"""
  839. upsert_draft_transport_for_plan(
  840. db,
  841. tenant_no=tenant_no,
  842. plan=plan,
  843. transport_payload=transport_payload,
  844. products=products,
  845. user_id=user_id,
  846. dept_id=dept_id or user_id,
  847. )
  848. def create_transport_for_plan(
  849. db: Session,
  850. *,
  851. tenant_no: str,
  852. plan: PlPlan,
  853. products: list[PlPlanProduct],
  854. transport_payload: PlanTransportCreateBlock,
  855. user_id: str,
  856. dept_id: str,
  857. generate_transport_no,
  858. ) -> PlTransport:
  859. vehicle = (
  860. db.query(PlVehicle)
  861. .filter(
  862. PlVehicle.id == transport_payload.vehicle_id,
  863. PlVehicle.tenant_no == tenant_no,
  864. PlVehicle.del_flag == 0,
  865. )
  866. .first()
  867. )
  868. if not vehicle:
  869. raise HTTPException(status_code=400, detail="车辆不存在")
  870. validate_hazard_cert_ready_for_transport(db, tenant_no, plan, products)
  871. driver_id, driver_name, driver_phone = _resolve_driver(
  872. db, tenant_no, transport_payload.vehicle_id, transport_payload.driver_id
  873. )
  874. carrier_name = vehicle.carrier_name
  875. carrier_id = transport_payload.carrier_id or vehicle.carrier_id
  876. if carrier_id:
  877. carrier = (
  878. db.query(PlCarrier)
  879. .filter(PlCarrier.id == carrier_id, PlCarrier.del_flag == 0)
  880. .first()
  881. )
  882. if carrier:
  883. carrier_name = carrier.carrier_name
  884. freight_amount = _calc_freight_amount(
  885. transport_payload.freight_type,
  886. products,
  887. transport_payload.transport_qty,
  888. transport_payload.freight_price,
  889. transport_payload.freight_amount,
  890. )
  891. transport_qty = transport_payload.transport_qty
  892. if transport_qty is None and transport_payload.freight_type == "by_weight":
  893. transport_qty = sum_plan_product_actual_quantities(products)
  894. transport_no = generate_transport_no(db)
  895. transport = PlTransport(
  896. id=generate_uuid(),
  897. tenant_no=tenant_no,
  898. transport_no=transport_no,
  899. vehicle_id=vehicle.id,
  900. license_plate=vehicle.license_plate,
  901. driver_id=driver_id,
  902. driver_name=driver_name,
  903. driver_phone=driver_phone,
  904. carrier_id=carrier_id,
  905. carrier_name=carrier_name,
  906. start_point=plan.pickup_address,
  907. end_point=plan.destination_address or plan.destination_name,
  908. distance_km=transport_payload.distance_km,
  909. unit="吨" if transport_payload.freight_type == "by_weight" else "趟",
  910. transport_qty=transport_qty,
  911. freight_price=transport_payload.freight_price,
  912. freight_amount=freight_amount or 0,
  913. freight_type=transport_payload.freight_type,
  914. has_invoice=transport_payload.has_invoice if transport_payload.has_invoice is not None else 0,
  915. transport_status="draft",
  916. approval_status="none",
  917. remark=transport_payload.remark,
  918. create_by=user_id,
  919. create_dept_id=dept_id,
  920. )
  921. db.add(transport)
  922. db.add(
  923. PlPlanTransport(
  924. id=generate_uuid(),
  925. tenant_no=tenant_no,
  926. plan_id=plan.id,
  927. transport_id=transport.id,
  928. create_by=user_id,
  929. create_dept_id=dept_id,
  930. )
  931. )
  932. plan.transport_id = transport.id
  933. plan.transport_no = transport_no
  934. plan.vehicle_id = vehicle.id
  935. plan.license_plate = vehicle.license_plate
  936. plan.driver_id = driver_id
  937. plan.driver_name = driver_name
  938. _apply_vehicle_snapshot_to_plan_transport(
  939. db,
  940. tenant_no=tenant_no,
  941. plan=plan,
  942. transport=transport,
  943. vehicle=vehicle,
  944. driver_id=driver_id,
  945. )
  946. plan.transport_status = "arranged"
  947. plan.follow_status = "arranged"
  948. plan.freight_amount = freight_amount or 0
  949. return transport
  950. PLAN_SELECTABLE_CONTRACT_STATUSES = ("approved", "executing")
  951. PLAN_CONTRACT_SIDE_SUPPLIER = "supplier"
  952. PLAN_CONTRACT_SIDE_DEMANDER = "demander"
  953. def assert_contract_selectable_for_plan(
  954. db: Session,
  955. contract_id: str,
  956. tenant_no: str,
  957. *,
  958. expected_customer_id: Optional[str] = None,
  959. side_label: str = "合同",
  960. ):
  961. from app.models.crm_models import CrmContract
  962. contract = (
  963. db.query(CrmContract)
  964. .filter(
  965. CrmContract.id == contract_id,
  966. CrmContract.tenant_no == tenant_no,
  967. CrmContract.del_flag == 0,
  968. )
  969. .first()
  970. )
  971. if not contract:
  972. raise HTTPException(status_code=404, detail=f"{side_label}不存在")
  973. if contract.status not in PLAN_SELECTABLE_CONTRACT_STATUSES:
  974. raise HTTPException(status_code=422, detail=f"仅已审批或执行中的合同可关联运输计划({side_label})")
  975. if expected_customer_id and contract.customer_id != expected_customer_id:
  976. raise HTTPException(status_code=422, detail=f"{side_label}不属于所选客户")
  977. return contract
  978. def _normalize_contract_links(
  979. links: Optional[list[PlanContractLink]],
  980. ) -> list[PlanContractLink]:
  981. if not links:
  982. return []
  983. seen: set[str] = set()
  984. out: list[PlanContractLink] = []
  985. for link in links:
  986. cid = (link.contract_id or "").strip()
  987. if not cid or cid in seen:
  988. continue
  989. seen.add(cid)
  990. out.append(link)
  991. return out
  992. def validate_plan_party_contracts(
  993. db: Session,
  994. tenant_no: str,
  995. *,
  996. supplier_id: Optional[str],
  997. supplier_name: Optional[str],
  998. customer_id: Optional[str],
  999. customer_name: Optional[str],
  1000. supplier_contracts: Optional[list[PlanContractLink]],
  1001. demander_contracts: Optional[list[PlanContractLink]],
  1002. require_products: Optional[list] = None,
  1003. ) -> tuple[list[PlanContractLink], list[PlanContractLink]]:
  1004. if not (supplier_id or "").strip() or not (supplier_name or "").strip():
  1005. raise HTTPException(status_code=422, detail="请选择供方客户")
  1006. if not (customer_id or "").strip() or not (customer_name or "").strip():
  1007. raise HTTPException(status_code=422, detail="请选择需方客户")
  1008. if require_products is not None and not require_products:
  1009. raise HTTPException(status_code=422, detail="请至少保留一条产品明细")
  1010. supplier_links = _normalize_contract_links(supplier_contracts)
  1011. demander_links = _normalize_contract_links(demander_contracts)
  1012. if not is_internal_customer_name(supplier_name) and not supplier_links:
  1013. raise HTTPException(status_code=422, detail="请选择供方合同")
  1014. if not is_internal_customer_name(customer_name) and not demander_links:
  1015. raise HTTPException(status_code=422, detail="请选择需方合同")
  1016. resolved_supplier: list[PlanContractLink] = []
  1017. for link in supplier_links:
  1018. contract = assert_contract_selectable_for_plan(
  1019. db,
  1020. link.contract_id,
  1021. tenant_no,
  1022. expected_customer_id=supplier_id,
  1023. side_label="供方合同",
  1024. )
  1025. resolved_supplier.append(
  1026. PlanContractLink(contractId=contract.id, contractNo=contract.contract_no or link.contract_no)
  1027. )
  1028. resolved_demander: list[PlanContractLink] = []
  1029. for link in demander_links:
  1030. contract = assert_contract_selectable_for_plan(
  1031. db,
  1032. link.contract_id,
  1033. tenant_no,
  1034. expected_customer_id=customer_id,
  1035. side_label="需方合同",
  1036. )
  1037. resolved_demander.append(
  1038. PlanContractLink(contractId=contract.id, contractNo=contract.contract_no or link.contract_no)
  1039. )
  1040. return resolved_supplier, resolved_demander
  1041. def replace_plan_contracts(
  1042. db: Session,
  1043. *,
  1044. tenant_no: str,
  1045. plan_id: str,
  1046. supplier_contracts: list[PlanContractLink],
  1047. demander_contracts: list[PlanContractLink],
  1048. user_id: Optional[str],
  1049. dept_id: Optional[str],
  1050. ) -> None:
  1051. now = datetime.now()
  1052. db.query(PlPlanContract).filter(
  1053. PlPlanContract.plan_id == plan_id,
  1054. PlPlanContract.tenant_no == tenant_no,
  1055. PlPlanContract.del_flag == 0,
  1056. ).update(
  1057. {"del_flag": 1, "update_by": user_id, "update_time": now},
  1058. synchronize_session=False,
  1059. )
  1060. rows: list[tuple[str, list[PlanContractLink]]] = [
  1061. (PLAN_CONTRACT_SIDE_SUPPLIER, supplier_contracts),
  1062. (PLAN_CONTRACT_SIDE_DEMANDER, demander_contracts),
  1063. ]
  1064. order_num = 0
  1065. for side, links in rows:
  1066. for link in links:
  1067. order_num += 1
  1068. db.add(
  1069. PlPlanContract(
  1070. id=generate_uuid(),
  1071. plan_id=plan_id,
  1072. side=side,
  1073. contract_id=link.contract_id,
  1074. contract_no=link.contract_no,
  1075. tenant_no=tenant_no,
  1076. order_num=order_num,
  1077. create_by=user_id,
  1078. create_dept_id=dept_id or "",
  1079. create_time=now,
  1080. update_by=user_id,
  1081. update_time=now,
  1082. del_flag=0,
  1083. )
  1084. )
  1085. def load_plan_contracts_grouped(
  1086. db: Session,
  1087. tenant_no: str,
  1088. plan_ids: list[str],
  1089. ) -> dict[str, tuple[list[PlanContractLink], list[PlanContractLink]]]:
  1090. if not plan_ids:
  1091. return {}
  1092. rows = (
  1093. db.query(PlPlanContract)
  1094. .filter(
  1095. PlPlanContract.tenant_no == tenant_no,
  1096. PlPlanContract.plan_id.in_(plan_ids),
  1097. PlPlanContract.del_flag == 0,
  1098. )
  1099. .order_by(PlPlanContract.order_num.asc(), PlPlanContract.create_time.asc())
  1100. .all()
  1101. )
  1102. buckets: dict[str, list[list[PlanContractLink]]] = {
  1103. pid: [[], []] for pid in plan_ids
  1104. }
  1105. for row in rows:
  1106. bucket = buckets.setdefault(row.plan_id, [[], []])
  1107. link = PlanContractLink(contractId=row.contract_id, contractNo=row.contract_no)
  1108. if row.side == PLAN_CONTRACT_SIDE_SUPPLIER:
  1109. bucket[0].append(link)
  1110. elif row.side == PLAN_CONTRACT_SIDE_DEMANDER:
  1111. bucket[1].append(link)
  1112. return {pid: (lists[0], lists[1]) for pid, lists in buckets.items()}
  1113. def validate_create_full(payload: PlanCreateFull) -> None:
  1114. if not (payload.supplier_id or "").strip() or not (payload.supplier_name or "").strip():
  1115. raise HTTPException(status_code=422, detail="请选择供方客户")
  1116. if not (payload.customer_id or "").strip() or not (payload.customer_name or "").strip():
  1117. raise HTTPException(status_code=422, detail="请选择需方客户")
  1118. if not payload.products:
  1119. raise HTTPException(status_code=422, detail="请至少保留一条产品明细")
  1120. supplier_links = _normalize_contract_links(payload.supplier_contracts)
  1121. demander_links = _normalize_contract_links(payload.demander_contracts)
  1122. if not is_internal_customer_name(payload.supplier_name) and not supplier_links:
  1123. raise HTTPException(status_code=422, detail="请选择供方合同")
  1124. if not is_internal_customer_name(payload.customer_name) and not demander_links:
  1125. raise HTTPException(status_code=422, detail="请选择需方合同")
  1126. transport = payload.transport
  1127. if not transport or not transport.vehicle_id:
  1128. return
  1129. if payload.has_freight:
  1130. if transport.freight_type == "by_weight":
  1131. if not transport.freight_price or Decimal(str(transport.freight_price)) <= 0:
  1132. raise HTTPException(status_code=422, detail="按重量计费时请填写运输单价")
  1133. qty = transport.transport_qty or sum_product_actual_quantities(payload.products)
  1134. if qty <= 0:
  1135. raise HTTPException(status_code=422, detail="运输量必须大于0,请填写产品实际数量或运输量")
  1136. elif transport.freight_type == "by_trip":
  1137. if not transport.freight_amount or Decimal(str(transport.freight_amount)) <= 0:
  1138. raise HTTPException(status_code=422, detail="按趟计费时请填写运费金额")
  1139. def calc_plan_total_amount(products: list[PlanProductCreate]) -> Decimal:
  1140. total = Decimal("0")
  1141. for prod in products:
  1142. if prod.tax_subtotal is not None:
  1143. total += Decimal(str(prod.tax_subtotal))
  1144. return total
  1145. _LOCKED_SETTLEMENT_STATUSES = ("approving", "approved", "confirmed")
  1146. def transport_has_locked_freight_settlement(
  1147. db: Session, tenant_no: str, transport_id: Optional[str]
  1148. ) -> bool:
  1149. if not transport_id:
  1150. return False
  1151. return (
  1152. db.query(PlFreightSettlementTransport.id)
  1153. .join(
  1154. PlFreightSettlement,
  1155. PlFreightSettlement.id == PlFreightSettlementTransport.settlement_id,
  1156. )
  1157. .filter(
  1158. PlFreightSettlementTransport.transport_id == transport_id,
  1159. PlFreightSettlementTransport.tenant_no == tenant_no,
  1160. PlFreightSettlementTransport.del_flag == 0,
  1161. PlFreightSettlement.tenant_no == tenant_no,
  1162. PlFreightSettlement.del_flag == 0,
  1163. PlFreightSettlement.settlement_status.in_(_LOCKED_SETTLEMENT_STATUSES),
  1164. )
  1165. .first()
  1166. is not None
  1167. )
  1168. def sync_archive_qty_from_io_order(
  1169. db: Session,
  1170. *,
  1171. tenant_no: str,
  1172. plan: PlPlan,
  1173. items: list[PlanArchiveProductItem],
  1174. ) -> list[PlanArchiveProductItem]:
  1175. """若存在入库/出库单,用明细 actual_qty 覆盖归档提交中的实际数量(权威来源)。"""
  1176. from app.models.pl_models import PlInboundOrder, PlInboundOrderItem, PlOutboundOrder, PlOutboundOrderItem
  1177. if plan.plan_type not in ("inbound", "outbound"):
  1178. return items
  1179. header_m = PlInboundOrder if plan.plan_type == "inbound" else PlOutboundOrder
  1180. item_m = PlInboundOrderItem if plan.plan_type == "inbound" else PlOutboundOrderItem
  1181. header = (
  1182. db.query(header_m)
  1183. .filter(
  1184. header_m.tenant_no == tenant_no,
  1185. header_m.plan_id == plan.id,
  1186. header_m.status != "voided",
  1187. header_m.del_flag == 0,
  1188. )
  1189. .order_by(header_m.create_time.desc())
  1190. .first()
  1191. )
  1192. if not header:
  1193. return items
  1194. io_items = (
  1195. db.query(item_m)
  1196. .filter(item_m.order_id == header.id, item_m.del_flag == 0)
  1197. .all()
  1198. )
  1199. qty_by_plan_product = {
  1200. row.plan_product_id: Decimal(str(row.actual_qty))
  1201. for row in io_items
  1202. if row.actual_qty is not None
  1203. }
  1204. if not qty_by_plan_product:
  1205. return items
  1206. out: list[PlanArchiveProductItem] = []
  1207. for item in items:
  1208. qty = qty_by_plan_product.get(item.id)
  1209. if qty is not None:
  1210. out.append(PlanArchiveProductItem(id=item.id, actual_qty=qty))
  1211. else:
  1212. out.append(item)
  1213. return out
  1214. def apply_archive_product_quantities(
  1215. db: Session,
  1216. *,
  1217. tenant_no: str,
  1218. plan: PlPlan,
  1219. items: list[PlanArchiveProductItem],
  1220. user_id: str,
  1221. sync_from_io_order: bool = True,
  1222. ) -> None:
  1223. if sync_from_io_order:
  1224. items = sync_archive_qty_from_io_order(db, tenant_no=tenant_no, plan=plan, items=items)
  1225. product_map = {
  1226. row.id: row
  1227. for row in db.query(PlPlanProduct).filter(
  1228. PlPlanProduct.plan_id == plan.id,
  1229. PlPlanProduct.tenant_no == tenant_no,
  1230. PlPlanProduct.del_flag == 0,
  1231. ).all()
  1232. }
  1233. for item in items:
  1234. prod = product_map.get(item.id)
  1235. if not prod:
  1236. raise HTTPException(status_code=400, detail=f"产品明细不存在: {item.id}")
  1237. actual = Decimal(str(item.actual_qty))
  1238. prod.actual_qty = actual
  1239. price_in = prod.inbound_price or prod.tax_price
  1240. price_out = prod.outbound_price or prod.tax_price
  1241. if plan.plan_type == "inbound":
  1242. prod.inbound_qty = actual
  1243. if price_in is not None:
  1244. prod.inbound_amount = actual * Decimal(str(price_in))
  1245. elif plan.plan_type == "outbound":
  1246. prod.outbound_qty = actual
  1247. if price_out is not None:
  1248. prod.outbound_amount = actual * Decimal(str(price_out))
  1249. prod.update_by = user_id
  1250. brand_rows = []
  1251. if item.brand_qtys:
  1252. brand_rows = [
  1253. {
  1254. "brandId": bq.brand_id or "",
  1255. "brandName": bq.brand_name,
  1256. "qty": bq.qty,
  1257. }
  1258. for bq in item.brand_qtys
  1259. ]
  1260. try:
  1261. validate_brand_qty_rows(
  1262. brand_mode=prod.brand_mode,
  1263. required_brand_id=prod.brand_id,
  1264. actual_qty=actual,
  1265. rows=brand_rows,
  1266. )
  1267. except ValueError as exc:
  1268. raise HTTPException(status_code=400, detail=str(exc)) from exc
  1269. persist_plan_product_brand_qtys(
  1270. db,
  1271. tenant_no=tenant_no,
  1272. plan_id=plan.id,
  1273. plan_product_id=prod.id,
  1274. brand_mode=prod.brand_mode,
  1275. rows=brand_rows,
  1276. user_id=user_id,
  1277. dept_id=getattr(prod, "create_dept_id", "") or "",
  1278. )
  1279. def apply_archive_freight(
  1280. db: Session,
  1281. *,
  1282. tenant_no: str,
  1283. plan: PlPlan,
  1284. freight: PlanArchiveFreightBlock,
  1285. user_id: str,
  1286. ) -> None:
  1287. has_freight = int(freight.has_freight if freight.has_freight is not None else plan.has_freight or 0)
  1288. plan.has_freight = has_freight
  1289. plan.update_by = user_id
  1290. if has_freight != 1:
  1291. plan.freight_amount = Decimal("0")
  1292. if not plan.transport_id:
  1293. return
  1294. if transport_has_locked_freight_settlement(db, tenant_no, plan.transport_id):
  1295. raise HTTPException(status_code=400, detail="运输单已纳入运费结算,无法修改运费")
  1296. transport = (
  1297. db.query(PlTransport)
  1298. .filter(
  1299. PlTransport.id == plan.transport_id,
  1300. PlTransport.tenant_no == tenant_no,
  1301. PlTransport.del_flag == 0,
  1302. )
  1303. .first()
  1304. )
  1305. if not transport:
  1306. raise HTTPException(status_code=400, detail="关联运输单不存在")
  1307. transport.freight_amount = Decimal("0")
  1308. transport.freight_price = None
  1309. transport.transport_qty = None
  1310. transport.update_by = user_id
  1311. sync_draft_settlement_freight_for_transport(
  1312. db,
  1313. tenant_no=tenant_no,
  1314. transport_id=plan.transport_id,
  1315. freight_amount=Decimal("0"),
  1316. user_id=user_id,
  1317. )
  1318. return
  1319. if not plan.transport_id:
  1320. raise HTTPException(status_code=400, detail="计划有运输费用但未安排运输,请先安排运输")
  1321. if transport_has_locked_freight_settlement(db, tenant_no, plan.transport_id):
  1322. raise HTTPException(status_code=400, detail="运输单已纳入运费结算,无法修改运费")
  1323. transport = (
  1324. db.query(PlTransport)
  1325. .filter(
  1326. PlTransport.id == plan.transport_id,
  1327. PlTransport.tenant_no == tenant_no,
  1328. PlTransport.del_flag == 0,
  1329. )
  1330. .first()
  1331. )
  1332. if not transport:
  1333. raise HTTPException(status_code=400, detail="关联运输单不存在")
  1334. transport.distance_km = freight.distance_km
  1335. transport.freight_type = freight.freight_type or "by_weight"
  1336. transport.freight_price = freight.freight_price
  1337. amount = freight.freight_amount or Decimal("0")
  1338. transport.freight_amount = amount
  1339. freight_type = freight.freight_type or transport.freight_type or "by_weight"
  1340. plan_products = (
  1341. db.query(PlPlanProduct)
  1342. .filter(
  1343. PlPlanProduct.plan_id == plan.id,
  1344. PlPlanProduct.tenant_no == tenant_no,
  1345. PlPlanProduct.del_flag == 0,
  1346. )
  1347. .all()
  1348. )
  1349. if freight.transport_qty is not None:
  1350. transport.transport_qty = freight.transport_qty
  1351. elif freight_type == "by_weight":
  1352. transport.transport_qty = sum_plan_product_actual_quantities(plan_products)
  1353. if freight_type == "by_weight" and (transport.transport_qty or 0) <= 0:
  1354. raise HTTPException(status_code=400, detail="运输量必须大于0,请先填写产品实际数量")
  1355. if freight.has_invoice is not None:
  1356. transport.has_invoice = freight.has_invoice
  1357. transport.update_by = user_id
  1358. plan.freight_amount = amount
  1359. sync_draft_settlement_freight_for_transport(
  1360. db,
  1361. tenant_no=tenant_no,
  1362. transport_id=plan.transport_id,
  1363. freight_amount=amount,
  1364. user_id=user_id,
  1365. )
  1366. def record_plan_change(
  1367. db: Session,
  1368. *,
  1369. tenant_no: str,
  1370. plan_id: str,
  1371. change_type: str,
  1372. change_content: str,
  1373. user_id: str,
  1374. user_name: Optional[str],
  1375. dept_id: str,
  1376. ) -> None:
  1377. db.add(
  1378. PlPlanChange(
  1379. id=generate_uuid(),
  1380. tenant_no=tenant_no,
  1381. plan_id=plan_id,
  1382. change_type=change_type,
  1383. change_content=change_content,
  1384. change_user_id=user_id,
  1385. change_user_name=user_name,
  1386. create_by=user_id,
  1387. create_dept_id=dept_id,
  1388. )
  1389. )
  1390. def apply_plan_driver_feedback(
  1391. db: Session,
  1392. *,
  1393. tenant_no: str,
  1394. plan: PlPlan,
  1395. payload: PlanDriverFeedbackUpdateBody,
  1396. user_id: str,
  1397. ) -> PlTransport:
  1398. if plan.follow_status == "cancelled":
  1399. raise HTTPException(status_code=400, detail="已作废的计划不能编辑司机回传")
  1400. if not plan.transport_id:
  1401. raise HTTPException(status_code=400, detail="计划未安排运输,无法编辑司机回传")
  1402. transport = (
  1403. db.query(PlTransport)
  1404. .filter(
  1405. PlTransport.id == plan.transport_id,
  1406. PlTransport.tenant_no == tenant_no,
  1407. PlTransport.del_flag == 0,
  1408. )
  1409. .first()
  1410. )
  1411. if not transport:
  1412. raise HTTPException(status_code=404, detail="运输单不存在")
  1413. if payload.products is not None:
  1414. products_payload = [
  1415. {
  1416. "planProductId": item.plan_product_id,
  1417. "actualQty": item.actual_qty,
  1418. "brandQtys": [b.model_dump(by_alias=True) for b in (item.brand_qtys or [])],
  1419. }
  1420. for item in payload.products
  1421. ]
  1422. try:
  1423. persist_transport_product_feedback(
  1424. db,
  1425. tenant_no=tenant_no,
  1426. transport=transport,
  1427. plan_id=plan.id,
  1428. products=products_payload,
  1429. user_id=user_id,
  1430. enforce_brand=True,
  1431. )
  1432. except ValueError as exc:
  1433. raise HTTPException(status_code=400, detail=str(exc)) from exc
  1434. elif payload.feedback_qty is not None:
  1435. transport.feedback_qty = payload.feedback_qty
  1436. if payload.feedback_remark is not None:
  1437. transport.feedback_remark = payload.feedback_remark.strip() if payload.feedback_remark else None
  1438. if payload.feedback_images is not None:
  1439. transport.feedback_images = json.dumps([item for item in payload.feedback_images if item])
  1440. transport.feedback_time = payload.feedback_time or datetime.now()
  1441. transport.update_by = user_id
  1442. return transport
  1443. def apply_plan_site_feedback(
  1444. db: Session,
  1445. *,
  1446. tenant_no: str,
  1447. plan: PlPlan,
  1448. link_id: str,
  1449. payload: PlanSiteFeedbackUpdateBody,
  1450. user_id: str,
  1451. ) -> PlTransportSiteLink:
  1452. if plan.follow_status == "cancelled":
  1453. raise HTTPException(status_code=400, detail="已作废的计划不能编辑现场回传")
  1454. link = (
  1455. db.query(PlTransportSiteLink)
  1456. .filter(
  1457. PlTransportSiteLink.id == link_id,
  1458. PlTransportSiteLink.plan_id == plan.id,
  1459. PlTransportSiteLink.tenant_no == tenant_no,
  1460. PlTransportSiteLink.del_flag == 0,
  1461. )
  1462. .first()
  1463. )
  1464. if not link:
  1465. raise HTTPException(status_code=404, detail="现场链接不存在")
  1466. if payload.site_feedback_remark is not None:
  1467. link.site_feedback_remark = (
  1468. payload.site_feedback_remark.strip() if payload.site_feedback_remark else None
  1469. )
  1470. if payload.site_feedback_images is not None:
  1471. link.site_feedback_images = json.dumps([item for item in payload.site_feedback_images if item])
  1472. link.site_feedback_time = payload.site_feedback_time or datetime.now()
  1473. link.update_by = user_id
  1474. return link