weighing.py 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702
  1. from typing import Any, List, Optional, Tuple
  2. from datetime import datetime
  3. from fastapi import APIRouter, Body, Depends, HTTPException, Query, Request, status
  4. from fastapi.responses import JSONResponse
  5. from pydantic import BaseModel, ConfigDict, Field
  6. from sqlalchemy import and_, func, or_
  7. from sqlalchemy.orm import Session
  8. from app.core.database import get_db
  9. from app.core.db_auth import DbUserContext, get_db_current_user
  10. from app.core.delete_guard import assert_row_deletable_by_creator, user_info_from_context
  11. from app.core.tenant_context import get_tenant_no
  12. from app.core.utils import generate_uuid
  13. from app.services.db_data_scope import apply_db_scope, assert_db_row_in_scope
  14. from app.models.db_models import DbInboundOutbound, DbPurchasePlan, DbWeighingSlip
  15. from app.schemas.db.inventory import InboundOutboundRead
  16. from app.schemas.db.plans import PurchasePlanRead
  17. from app.schemas.db.weighing import (
  18. WeighingSlipCreate,
  19. WeighingSlipListResponse,
  20. WeighingSlipRead,
  21. WeighingSlipUpdate,
  22. )
  23. from app.schemas.pl.cancel_reason import VoidReasonBody, normalize_optional_reason
  24. from app.services.db.mappers import (
  25. inventory_date_to_api,
  26. inventory_to_read,
  27. load_user_display_name_map,
  28. parse_datetime_str,
  29. plan_to_read,
  30. weighing_to_read,
  31. )
  32. from app.services.db.list_sort import (
  33. order_weighing,
  34. parse_sort_params,
  35. weighing_needs_inventory_join,
  36. )
  37. from app.services.db.weighing_list_filters import build_weighing_column_filters
  38. from app.services.db.weighing_logic import process_weighing_business_logic
  39. from app.services.db.product_helpers import (
  40. apply_product_snapshot_to_weighing,
  41. resolve_crm_product,
  42. )
  43. from app.utils.excel_export import (
  44. EXPORT_MAX_ROWS,
  45. ExportColumn,
  46. ExportValidationError,
  47. export_rows_to_xlsx_response,
  48. filter_export_columns,
  49. )
  50. router = APIRouter(prefix="/weighing", tags=["weighing"])
  51. _WEIGHING_REMARK_FIELD = "remarks"
  52. def _base_query(db: Session, tenant_no: str):
  53. return db.query(DbWeighingSlip).filter(
  54. DbWeighingSlip.tenant_no == tenant_no,
  55. DbWeighingSlip.del_flag == 0,
  56. DbWeighingSlip.is_voided == 0,
  57. )
  58. def _merge_query(request: Request, **known) -> dict:
  59. out = {k: v for k, v in known.items() if v is not None and v != ""}
  60. for k, v in request.query_params.multi_items():
  61. if k in ("page", "pageSize"):
  62. continue
  63. if k not in out and v not in (None, ""):
  64. out[k] = v
  65. return out
  66. def _param_get(params: dict, *keys: str) -> Any:
  67. for key in keys:
  68. if key in params and params[key] not in (None, ""):
  69. return params[key]
  70. return None
  71. def _filters_from_params(params: dict) -> list:
  72. return build_weighing_column_filters(
  73. plan_no=_param_get(params, "planNo", "plan_no"),
  74. plan_no_in=_param_get(params, "planNoIn", "plan_no_in"),
  75. batch_no=_param_get(params, "batchNo", "batch_no"),
  76. batch_no_in=_param_get(params, "batchNoIn", "batch_no_in"),
  77. vehicle_no=_param_get(params, "vehicleNo", "vehicle_no"),
  78. vehicle_no_in=_param_get(params, "vehicleNoIn", "vehicle_no_in"),
  79. cargo_name=_param_get(params, "cargoName", "cargo_name"),
  80. cargo_name_in=_param_get(params, "cargoNameIn", "cargo_name_in"),
  81. supplier=_param_get(params, "supplier"),
  82. supplier_in=_param_get(params, "supplierIn", "supplier_in"),
  83. consignee=_param_get(params, "consignee"),
  84. consignee_in=_param_get(params, "consigneeIn", "consignee_in"),
  85. business_type=_param_get(params, "businessType", "business_type"),
  86. business_type_in=_param_get(params, "businessTypeIn", "business_type_in"),
  87. date_from=_param_get(params, "dateFrom", "date_from"),
  88. date_to=_param_get(params, "dateTo", "date_to"),
  89. net_weight_op=_param_get(params, "netWeightOp", "net_weight_op"),
  90. net_weight=_param_get(params, "netWeight", "net_weight"),
  91. net_weight_min=_param_get(params, "netWeightMin", "net_weight_min"),
  92. net_weight_max=_param_get(params, "netWeightMax", "net_weight_max"),
  93. gross_weight_op=_param_get(params, "grossWeightOp", "gross_weight_op"),
  94. gross_weight=_param_get(params, "grossWeight", "gross_weight"),
  95. gross_weight_min=_param_get(params, "grossWeightMin", "gross_weight_min"),
  96. gross_weight_max=_param_get(params, "grossWeightMax", "gross_weight_max"),
  97. )
  98. def _body_to_params(body: "WeighingExportBody") -> dict:
  99. data = body.model_dump(by_alias=True, exclude_none=True)
  100. data.pop("ids", None)
  101. data.pop("columns", None)
  102. extra = getattr(body, "model_extra", None) or {}
  103. for key, value in extra.items():
  104. if key not in data and value not in (None, ""):
  105. data[key] = value
  106. return data
  107. def _weighing_query(db: Session, user: DbUserContext, params: dict):
  108. tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
  109. query = _base_query(db, tenant_no)
  110. q = _param_get(params, "q")
  111. if q:
  112. keyword = str(q).lower()
  113. query = query.filter(
  114. or_(
  115. func.lower(DbWeighingSlip.vehicle_no).contains(keyword),
  116. func.lower(DbWeighingSlip.plan_no).contains(keyword),
  117. func.lower(DbWeighingSlip.batch_no).contains(keyword),
  118. func.lower(DbWeighingSlip.cargo_name).contains(keyword),
  119. func.lower(DbWeighingSlip.supplier).contains(keyword),
  120. func.lower(DbWeighingSlip.consignee).contains(keyword),
  121. )
  122. )
  123. column_filters = _filters_from_params(params)
  124. if column_filters:
  125. query = query.filter(*column_filters)
  126. query = apply_db_scope(query, DbWeighingSlip, user, db, tenant_no)
  127. sf, so = parse_sort_params(params)
  128. if weighing_needs_inventory_join(sf, so):
  129. query = query.outerjoin(
  130. DbInboundOutbound,
  131. and_(
  132. DbWeighingSlip.inbound_outbound_id == DbInboundOutbound.id,
  133. DbInboundOutbound.del_flag == 0,
  134. ),
  135. )
  136. query = query.order_by(
  137. order_weighing(sf, so, default=DbWeighingSlip.gross_weight_time.desc()),
  138. DbWeighingSlip.id.desc(),
  139. )
  140. return query
  141. def _map_weighings(db: Session, weighings: list) -> list[WeighingSlipRead]:
  142. inventory_ids = [w.inbound_outbound_id for w in weighings if w.inbound_outbound_id]
  143. inventory_date_map: dict[str, str] = {}
  144. if inventory_ids:
  145. inventories = (
  146. db.query(DbInboundOutbound)
  147. .filter(
  148. DbInboundOutbound.id.in_(inventory_ids),
  149. DbInboundOutbound.del_flag == 0,
  150. )
  151. .all()
  152. )
  153. for inventory in inventories:
  154. inventory_date_map[inventory.id] = inventory_date_to_api(inventory.date)
  155. user_name_map = load_user_display_name_map(db, [w.create_by for w in weighings])
  156. return [
  157. weighing_to_read(
  158. weighing,
  159. inventory_date=inventory_date_map.get(weighing.inbound_outbound_id or ""),
  160. user_name_map=user_name_map,
  161. )
  162. for weighing in weighings
  163. ]
  164. def _fetch_weighing(
  165. db: Session,
  166. user: DbUserContext,
  167. params: dict,
  168. *,
  169. page: Optional[int] = None,
  170. page_size: Optional[int] = None,
  171. ) -> Tuple[list[WeighingSlipRead], int]:
  172. query = _weighing_query(db, user, params)
  173. if page is not None and page_size is not None:
  174. total = query.order_by(None).count()
  175. weighings = query.offset((page - 1) * page_size).limit(page_size).all()
  176. return _map_weighings(db, weighings), total
  177. weighings = query.all()
  178. items = _map_weighings(db, weighings)
  179. return items, len(items)
  180. def _get_weighing(
  181. db: Session, tenant_no: str, weighing_id: str, user: DbUserContext
  182. ) -> DbWeighingSlip:
  183. row = (
  184. db.query(DbWeighingSlip)
  185. .filter(
  186. DbWeighingSlip.id == weighing_id,
  187. DbWeighingSlip.tenant_no == tenant_no,
  188. DbWeighingSlip.del_flag == 0,
  189. )
  190. .one_or_none()
  191. )
  192. if row is None:
  193. raise HTTPException(
  194. status_code=status.HTTP_404_NOT_FOUND,
  195. detail=f"Weighing {weighing_id} not found",
  196. )
  197. assert_db_row_in_scope(db, row, DbWeighingSlip, user, tenant_no)
  198. return row
  199. def _apply_weighing_product_update(
  200. db: Session, tenant_no: str, row: DbWeighingSlip, payload_dict: dict
  201. ) -> None:
  202. if "cargo_name" in payload_dict and "product_id" not in payload_dict:
  203. raise HTTPException(
  204. status_code=status.HTTP_400_BAD_REQUEST,
  205. detail="请从 CRM 重新选择产品",
  206. )
  207. product_id = payload_dict.pop("product_id", None)
  208. payload_dict.pop("cargo_name", None)
  209. payload_dict.pop("specification", None)
  210. if product_id is not None:
  211. snapshot = resolve_crm_product(db, tenant_no, product_id)
  212. apply_product_snapshot_to_weighing(row, snapshot)
  213. def _apply_weighing_update(row: DbWeighingSlip, payload_dict: dict) -> None:
  214. for key, value in payload_dict.items():
  215. if key == _WEIGHING_REMARK_FIELD:
  216. if isinstance(value, str):
  217. value = value.strip()
  218. row.remark = value
  219. continue
  220. if key in {"gross_weight_time", "tare_weight_time"}:
  221. if value is None or (isinstance(value, str) and not str(value).strip()):
  222. if key == "tare_weight_time":
  223. row.tare_weight_time = None
  224. else:
  225. parsed = parse_datetime_str(str(value))
  226. if parsed:
  227. setattr(row, key, parsed)
  228. continue
  229. if isinstance(value, str):
  230. value = value.strip()
  231. setattr(row, key, value)
  232. @router.get("/")
  233. def list_weighing(
  234. request: Request,
  235. q: Optional[str] = Query(default=None, description="模糊搜索关键词"),
  236. date_from: Optional[str] = Query(default=None, alias="dateFrom", description="开始日期"),
  237. date_to: Optional[str] = Query(default=None, alias="dateTo", description="结束日期"),
  238. plan_no: Optional[str] = Query(default=None, alias="planNo", description="计划号"),
  239. batch_no: Optional[str] = Query(default=None, alias="batchNo", description="出入库单号"),
  240. vehicle_no: Optional[str] = Query(default=None, alias="vehicleNo", description="车牌号"),
  241. cargo_name: Optional[str] = Query(default=None, alias="cargoName", description="货物名称"),
  242. supplier: Optional[str] = Query(default=None, description="供货单位"),
  243. consignee: Optional[str] = Query(default=None, description="收货单位"),
  244. business_type: Optional[str] = Query(default=None, alias="businessType", description="业务类型"),
  245. page: Optional[int] = Query(default=None, ge=1, description="页码;传入则返回分页结构"),
  246. page_size: Optional[int] = Query(
  247. default=None, ge=1, le=500, alias="pageSize", description="每页条数"
  248. ),
  249. db: Session = Depends(get_db),
  250. user: DbUserContext = Depends(get_db_current_user),
  251. ):
  252. params = _merge_query(
  253. request,
  254. q=q,
  255. dateFrom=date_from,
  256. dateTo=date_to,
  257. planNo=plan_no,
  258. batchNo=batch_no,
  259. vehicleNo=vehicle_no,
  260. cargoName=cargo_name,
  261. supplier=supplier,
  262. consignee=consignee,
  263. businessType=business_type,
  264. )
  265. # 管理端未传 page:仍返回全量数组(前端本地分页);小程序传 page/pageSize 走服务端分页
  266. paged = page is not None
  267. size = page_size if page_size is not None else 20
  268. page_num = page if page is not None else 1
  269. items, total = _fetch_weighing(
  270. db,
  271. user,
  272. params,
  273. page=page_num if paged else None,
  274. page_size=size if paged else None,
  275. )
  276. if paged:
  277. body = WeighingSlipListResponse(
  278. items=items, total=total, page=page_num, pageSize=size
  279. )
  280. return JSONResponse(content=body.model_dump(by_alias=True))
  281. return JSONResponse(
  282. content=[item.model_dump(by_alias=True) for item in items]
  283. )
  284. class WeighingColumnFacetBody(BaseModel):
  285. """地磅单列头可选值(去重;可附带列表上下文)"""
  286. model_config = ConfigDict(populate_by_name=True, extra="allow")
  287. field: str = Field(..., description="planNo | batchNo | vehicleNo | cargoName | supplier | consignee")
  288. keyword: Optional[str] = Field(default=None, description="可选值关键字")
  289. limit: Optional[int] = Field(default=500, ge=1, le=500, description="返回条数上限")
  290. _WEIGHING_FACET_COLUMNS = {
  291. "planNo": DbWeighingSlip.plan_no,
  292. "batchNo": DbWeighingSlip.batch_no,
  293. "vehicleNo": DbWeighingSlip.vehicle_no,
  294. "cargoName": DbWeighingSlip.cargo_name,
  295. "supplier": DbWeighingSlip.supplier,
  296. "consignee": DbWeighingSlip.consignee,
  297. }
  298. @router.post("/column-facet-values")
  299. def weighing_column_facet_values(
  300. body: WeighingColumnFacetBody,
  301. db: Session = Depends(get_db),
  302. user: DbUserContext = Depends(get_db_current_user),
  303. ):
  304. """地磅单列头可选值:在列表上下文(排除本列)下去重;未作废单据。"""
  305. from app.services.column_facet import list_column_facet_values
  306. from app.services.pl.facet_context import strip_simple_facet_self_params
  307. column = _WEIGHING_FACET_COLUMNS.get(body.field)
  308. if column is None:
  309. raise HTTPException(status_code=400, detail="不支持的列")
  310. tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
  311. params = strip_simple_facet_self_params(_body_to_params(body), body.field)
  312. extra_filters = [DbWeighingSlip.is_voided == 0]
  313. extra_filters.extend(_filters_from_params(params))
  314. qv = _param_get(params, "q")
  315. if qv:
  316. keyword = str(qv).lower()
  317. extra_filters.append(
  318. or_(
  319. func.lower(DbWeighingSlip.vehicle_no).contains(keyword),
  320. func.lower(DbWeighingSlip.plan_no).contains(keyword),
  321. func.lower(DbWeighingSlip.batch_no).contains(keyword),
  322. func.lower(DbWeighingSlip.cargo_name).contains(keyword),
  323. func.lower(DbWeighingSlip.supplier).contains(keyword),
  324. func.lower(DbWeighingSlip.consignee).contains(keyword),
  325. )
  326. )
  327. rows = list_column_facet_values(
  328. db,
  329. DbWeighingSlip,
  330. column,
  331. tenant_no=tenant_no,
  332. keyword=body.keyword,
  333. limit=body.limit if body.limit is not None else 500,
  334. apply_scope=lambda q: apply_db_scope(q, DbWeighingSlip, user, db, tenant_no),
  335. extra_filters=extra_filters,
  336. )
  337. return {"list": rows}
  338. WEIGHING_EXPORT_COLUMNS = [
  339. ExportColumn(title="日期", field="inventoryDate"),
  340. ExportColumn(title="计划单号", field="planNo"),
  341. ExportColumn(title="出入库单号", field="batchNo"),
  342. ExportColumn(title="车牌号", field="vehicleNo"),
  343. ExportColumn(title="货物名称", field="cargoName"),
  344. ExportColumn(title="规格", field="specification"),
  345. ExportColumn(title="业务类型", field="businessType"),
  346. ExportColumn(title="单据状态", field="docStatus"),
  347. ExportColumn(title="供货单位", field="supplier"),
  348. ExportColumn(title="收货单位", field="consignee"),
  349. ExportColumn(title="本次量(吨)", field="currentQuantity"),
  350. ExportColumn(title="净重(吨)", field="netWeight"),
  351. ExportColumn(title="毛重(kg)", field="grossWeight"),
  352. ExportColumn(title="毛重时间", field="grossWeightTime"),
  353. ExportColumn(title="皮重(kg)", field="tareWeight"),
  354. ExportColumn(title="皮重时间", field="tareWeightTime"),
  355. ExportColumn(title="扣杂(kg)", field="deduction"),
  356. ExportColumn(title="创建人", field="createByName"),
  357. ]
  358. class WeighingExportBody(BaseModel):
  359. q: Optional[str] = None
  360. date_from: Optional[str] = Field(default=None, alias="dateFrom")
  361. date_to: Optional[str] = Field(default=None, alias="dateTo")
  362. plan_no: Optional[str] = Field(default=None, alias="planNo")
  363. batch_no: Optional[str] = Field(default=None, alias="batchNo")
  364. vehicle_no: Optional[str] = Field(default=None, alias="vehicleNo")
  365. cargo_name: Optional[str] = Field(default=None, alias="cargoName")
  366. supplier: Optional[str] = None
  367. consignee: Optional[str] = None
  368. business_type: Optional[str] = Field(default=None, alias="businessType")
  369. sort_field: Optional[str] = Field(default=None, alias="sortField")
  370. sort_order: Optional[str] = Field(default=None, alias="sortOrder")
  371. ids: Optional[List[str]] = None
  372. columns: Optional[List[str]] = None
  373. model_config = {"populate_by_name": True, "extra": "allow"}
  374. def _kg_to_ton(value) -> float:
  375. try:
  376. return round(float(value or 0) / 1000, 2)
  377. except (TypeError, ValueError):
  378. return 0.0
  379. @router.post("/export")
  380. def export_weighing(
  381. body: WeighingExportBody,
  382. db: Session = Depends(get_db),
  383. user: DbUserContext = Depends(get_db_current_user),
  384. ):
  385. """导出地磅单列表(当前筛选全量,或 ids 已选)。"""
  386. try:
  387. items, _total = _fetch_weighing(db, user, _body_to_params(body))
  388. if body.ids:
  389. id_set = set(body.ids)
  390. items = [x for x in items if x.id in id_set]
  391. if len(items) > EXPORT_MAX_ROWS:
  392. raise ExportValidationError(f"导出超过 {EXPORT_MAX_ROWS} 条,请缩小筛选范围")
  393. rows = []
  394. for item in items:
  395. data = item.model_dump(by_alias=True)
  396. data["docStatus"] = "已作废" if data.get("isVoided") == 1 else "正常"
  397. data["currentQuantity"] = _kg_to_ton(data.get("currentQuantity"))
  398. data["netWeight"] = _kg_to_ton(data.get("netWeight"))
  399. if not data.get("inventoryDate"):
  400. gwt = data.get("grossWeightTime") or ""
  401. data["inventoryDate"] = str(gwt)[:10] if gwt else ""
  402. if not data.get("createByName"):
  403. data["createByName"] = data.get("createBy") or ""
  404. rows.append(data)
  405. cols = filter_export_columns(WEIGHING_EXPORT_COLUMNS, body.columns)
  406. date_str = datetime.now().strftime("%Y%m%d")
  407. tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
  408. return export_rows_to_xlsx_response(
  409. db,
  410. tenant_no=tenant_no,
  411. sheet_title="地磅单列表",
  412. columns=cols,
  413. rows=rows,
  414. filename_stem=f"地磅单列表_{date_str}",
  415. )
  416. except ExportValidationError as e:
  417. raise HTTPException(status_code=400, detail=str(e)) from e
  418. @router.post("/", response_model=WeighingSlipRead, status_code=status.HTTP_201_CREATED)
  419. def create_weighing(
  420. payload: WeighingSlipCreate,
  421. db: Session = Depends(get_db),
  422. user: DbUserContext = Depends(get_db_current_user),
  423. ) -> WeighingSlipRead:
  424. tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
  425. inbound_outbound_id = payload.inbound_outbound_id
  426. if inbound_outbound_id:
  427. existing = (
  428. db.query(DbWeighingSlip)
  429. .filter(
  430. DbWeighingSlip.inbound_outbound_id == inbound_outbound_id,
  431. DbWeighingSlip.is_voided == 0,
  432. DbWeighingSlip.del_flag == 0,
  433. )
  434. .first()
  435. )
  436. if existing:
  437. raise HTTPException(
  438. status_code=status.HTTP_400_BAD_REQUEST,
  439. detail="该出入库单已经存在关联的地磅单,无法重复创建",
  440. )
  441. planned_quantity = payload.planned_quantity
  442. current_quantity = payload.current_quantity
  443. purchase_plan_id = payload.purchase_plan_id
  444. batch_no = payload.batch_no.strip() if payload.batch_no else None
  445. if inbound_outbound_id:
  446. inventory = (
  447. db.query(DbInboundOutbound)
  448. .filter(
  449. DbInboundOutbound.id == inbound_outbound_id,
  450. DbInboundOutbound.tenant_no == tenant_no,
  451. DbInboundOutbound.del_flag == 0,
  452. )
  453. .one_or_none()
  454. )
  455. if inventory:
  456. if current_quantity is None:
  457. current_quantity = inventory.quantity * 1000
  458. if purchase_plan_id is None and inventory.purchase_plan_id:
  459. purchase_plan_id = inventory.purchase_plan_id
  460. if purchase_plan_id:
  461. plan = (
  462. db.query(DbPurchasePlan)
  463. .filter(DbPurchasePlan.id == purchase_plan_id)
  464. .one_or_none()
  465. )
  466. if plan and planned_quantity is None:
  467. planned_quantity = plan.planned_quantity * 1000
  468. if not batch_no and inventory.batch_no:
  469. batch_no = inventory.batch_no
  470. gross_time = parse_datetime_str(payload.gross_weight_time) if payload.gross_weight_time else None
  471. tare_time = (
  472. parse_datetime_str(payload.tare_weight_time)
  473. if payload.tare_weight_time and payload.tare_weight_time.strip()
  474. else None
  475. )
  476. snapshot = resolve_crm_product(db, tenant_no, payload.product_id)
  477. row = DbWeighingSlip(
  478. id=generate_uuid(),
  479. tenant_no=tenant_no,
  480. vehicle_no=payload.vehicle_no.strip(),
  481. plan_no=payload.plan_no.strip() if payload.plan_no else None,
  482. business_type=payload.business_type,
  483. batch_no=batch_no,
  484. supplier=payload.supplier.strip(),
  485. consignee=payload.consignee.strip(),
  486. gross_weight=payload.gross_weight or 0,
  487. gross_weight_time=gross_time or datetime.now(),
  488. tare_weight=payload.tare_weight or 0,
  489. tare_weight_time=tare_time,
  490. deduction=payload.deduction or 0,
  491. net_weight=payload.net_weight or 0,
  492. planned_quantity=planned_quantity,
  493. executed_quantity=payload.executed_quantity,
  494. purchase_plan_id=purchase_plan_id,
  495. inbound_outbound_id=inbound_outbound_id,
  496. weighing_point=payload.weighing_point or "山峡化工",
  497. remark=payload.remarks,
  498. current_quantity=current_quantity,
  499. is_voided=0,
  500. create_by=user.id,
  501. create_dept_id=user.dept_id,
  502. )
  503. apply_product_snapshot_to_weighing(row, snapshot)
  504. db.add(row)
  505. db.commit()
  506. db.refresh(row)
  507. process_weighing_business_logic(db, row)
  508. db.refresh(row)
  509. return weighing_to_read(row, db=db)
  510. @router.get("/{id}", response_model=WeighingSlipRead)
  511. def get_weighing(
  512. id: str,
  513. db: Session = Depends(get_db),
  514. user: DbUserContext = Depends(get_db_current_user),
  515. ) -> WeighingSlipRead:
  516. tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
  517. row = _get_weighing(db, tenant_no, id, user)
  518. return weighing_to_read(row, db=db)
  519. @router.patch("/{id}", response_model=WeighingSlipRead)
  520. def update_weighing(
  521. id: str,
  522. payload: WeighingSlipUpdate,
  523. db: Session = Depends(get_db),
  524. user: DbUserContext = Depends(get_db_current_user),
  525. ) -> WeighingSlipRead:
  526. tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
  527. row = _get_weighing(db, tenant_no, id, user)
  528. if row.is_voided == 1:
  529. raise HTTPException(
  530. status_code=status.HTTP_400_BAD_REQUEST,
  531. detail="Cannot update voided weighing record",
  532. )
  533. payload_dict = payload.model_dump(exclude_unset=True)
  534. _apply_weighing_product_update(db, tenant_no, row, payload_dict)
  535. _apply_weighing_update(row, payload_dict)
  536. row.update_by = user.id
  537. db.commit()
  538. db.refresh(row)
  539. process_weighing_business_logic(db, row)
  540. db.refresh(row)
  541. return weighing_to_read(row, db=db)
  542. @router.post("/{id}/void", status_code=status.HTTP_204_NO_CONTENT)
  543. def void_weighing(
  544. id: str,
  545. body: Optional[VoidReasonBody] = Body(default=None),
  546. db: Session = Depends(get_db),
  547. user: DbUserContext = Depends(get_db_current_user),
  548. ) -> None:
  549. tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
  550. row = _get_weighing(db, tenant_no, id, user)
  551. if row.is_voided == 1:
  552. raise HTTPException(
  553. status_code=status.HTTP_400_BAD_REQUEST,
  554. detail="Weighing record is already voided",
  555. )
  556. now = datetime.now()
  557. row.is_voided = 1
  558. row.void_reason = normalize_optional_reason(body.void_reason if body else None)
  559. row.void_time = now
  560. row.void_by = user.id
  561. row.update_by = user.id
  562. db.commit()
  563. process_weighing_business_logic(db, row)
  564. @router.delete("/{id}", status_code=status.HTTP_204_NO_CONTENT)
  565. def delete_weighing(
  566. id: str,
  567. db: Session = Depends(get_db),
  568. user: DbUserContext = Depends(get_db_current_user),
  569. ) -> None:
  570. tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
  571. row = _get_weighing(db, tenant_no, id, user)
  572. assert_row_deletable_by_creator(row, user_info_from_context(user))
  573. row.is_voided = 1
  574. row.update_by = user.id
  575. db.commit()
  576. process_weighing_business_logic(db, row)
  577. class WeighingRelatedData(BaseModel):
  578. plan: Optional[PurchasePlanRead]
  579. inventory: Optional[InboundOutboundRead]
  580. @router.get("/{id}/related", response_model=WeighingRelatedData)
  581. def get_weighing_related(
  582. id: str,
  583. db: Session = Depends(get_db),
  584. user: DbUserContext = Depends(get_db_current_user),
  585. ) -> WeighingRelatedData:
  586. tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
  587. row = _get_weighing(db, tenant_no, id, user)
  588. plan = None
  589. if row.purchase_plan_id:
  590. plan = (
  591. db.query(DbPurchasePlan)
  592. .filter(
  593. DbPurchasePlan.id == row.purchase_plan_id,
  594. DbPurchasePlan.del_flag == 0,
  595. )
  596. .one_or_none()
  597. )
  598. elif row.plan_no:
  599. plan = (
  600. db.query(DbPurchasePlan)
  601. .filter(
  602. DbPurchasePlan.plan_no == row.plan_no,
  603. DbPurchasePlan.tenant_no == tenant_no,
  604. DbPurchasePlan.del_flag == 0,
  605. )
  606. .limit(1)
  607. .first()
  608. )
  609. inventory = None
  610. if row.inbound_outbound_id:
  611. inventory = (
  612. db.query(DbInboundOutbound)
  613. .filter(
  614. DbInboundOutbound.id == row.inbound_outbound_id,
  615. DbInboundOutbound.del_flag == 0,
  616. )
  617. .one_or_none()
  618. )
  619. return WeighingRelatedData(
  620. plan=plan_to_read(plan, db=db) if plan else None,
  621. inventory=inventory_to_read(inventory, db=db) if inventory else None,
  622. )