339 lines
9.9 KiB
Python
339 lines
9.9 KiB
Python
import re
|
|
|
|
from sqlalchemy import bindparam, text
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.models.finished_goods import FinishedGoodsBox, FinishedGoodsBoxItem
|
|
from app.schemas.box import BoxSaveRequest, BoxUpdateRequest
|
|
|
|
ZONGPAI_PATTERN = re.compile(r"^(\d{2}(B|C|T)\d+|\d{2}(BW|CW)\d{4})$")
|
|
|
|
|
|
class InvalidZongpaiError(Exception):
|
|
pass
|
|
|
|
|
|
class ZongpaiNotFoundError(Exception):
|
|
pass
|
|
|
|
|
|
class DuplicateBoxItemError(Exception):
|
|
def __init__(self, paichan_no: str, box_no: int):
|
|
self.paichan_no = paichan_no
|
|
self.box_no = box_no
|
|
|
|
|
|
class DuplicateZongpaiBindError(Exception):
|
|
def __init__(self, paichan_no: str, box_no: int, zongpai_no: str):
|
|
self.paichan_no = paichan_no
|
|
self.box_no = box_no
|
|
self.zongpai_no = zongpai_no
|
|
|
|
|
|
class BoxItemNotFoundError(Exception):
|
|
pass
|
|
|
|
|
|
class InvalidBoxItemError(Exception):
|
|
pass
|
|
|
|
|
|
def query_erp_info(db: Session, zongpai_no: str) -> dict:
|
|
"""从 ERP 视图查询总排号对应的排产号、工令号和数量。"""
|
|
sql = text(_erp_info_sql(db))
|
|
row = db.execute(sql, {"zongpai_no": zongpai_no}).fetchone()
|
|
if not row:
|
|
raise ZongpaiNotFoundError()
|
|
return {
|
|
"zongpai_no": row[0],
|
|
"paichan_no": row[1],
|
|
"quantity": int(row[2]),
|
|
"work_order_no": row[3],
|
|
}
|
|
|
|
|
|
def _erp_info_sql(db: Session) -> str:
|
|
dialect_name = db.bind.dialect.name if db.bind is not None else ""
|
|
if dialect_name == "postgresql":
|
|
return (
|
|
'SELECT "总排号", "排产号", "数量", "工令号" '
|
|
'FROM "ERPAuto"."vw_productionContractData" '
|
|
'WHERE "总排号" = :zongpai_no '
|
|
"LIMIT 1"
|
|
)
|
|
return (
|
|
"SELECT TOP 1 [总排号], [排产号], [数量], [工令号] "
|
|
"FROM [ERPAuto].[vw_productionContractData] "
|
|
"WHERE [总排号] = :zongpai_no"
|
|
)
|
|
|
|
|
|
def query_erp_item_info_map(db: Session, zongpai_nos: list[str]) -> dict[str, dict]:
|
|
"""批量查询总排号对应的工令号和 ERP 总数量。"""
|
|
if not zongpai_nos:
|
|
return {}
|
|
|
|
dialect_name = db.bind.dialect.name if db.bind is not None else ""
|
|
if dialect_name == "postgresql":
|
|
sql = text(
|
|
'SELECT "总排号", "工令号", "数量" '
|
|
'FROM "ERPAuto"."vw_productionContractData" '
|
|
'WHERE "总排号" IN :zongpai_nos'
|
|
)
|
|
else:
|
|
sql = text(
|
|
"SELECT [总排号], [工令号], [数量] "
|
|
"FROM [ERPAuto].[vw_productionContractData] "
|
|
"WHERE [总排号] IN :zongpai_nos"
|
|
)
|
|
rows = db.execute(
|
|
sql.bindparams(bindparam("zongpai_nos", expanding=True)),
|
|
{"zongpai_nos": sorted(set(zongpai_nos))},
|
|
).fetchall()
|
|
return {
|
|
row[0]: {
|
|
"work_order_no": row[1],
|
|
"total_quantity": int(row[2]) if row[2] is not None else None,
|
|
}
|
|
for row in rows
|
|
}
|
|
|
|
|
|
def get_box_info(db: Session, zongpai_no: str) -> dict:
|
|
"""获取装箱信息：排产号、数量、已有箱号明细。"""
|
|
zongpai_no = zongpai_no.strip().upper()
|
|
if not ZONGPAI_PATTERN.match(zongpai_no):
|
|
raise InvalidZongpaiError()
|
|
erp = query_erp_info(db, zongpai_no)
|
|
paichan_no = erp["paichan_no"]
|
|
quantity = erp["quantity"]
|
|
|
|
boxes = (
|
|
db.query(FinishedGoodsBox)
|
|
.filter(FinishedGoodsBox.paichan_no == paichan_no)
|
|
.order_by(FinishedGoodsBox.box_no)
|
|
.all()
|
|
)
|
|
|
|
box_items = {}
|
|
item_zongpai_nos = []
|
|
for box in boxes:
|
|
items = (
|
|
db.query(FinishedGoodsBoxItem)
|
|
.filter(FinishedGoodsBoxItem.box_id == box.id)
|
|
.all()
|
|
)
|
|
box_items[box.id] = items
|
|
item_zongpai_nos.extend(item.zongpai_no for item in items)
|
|
|
|
erp_item_info_map = query_erp_item_info_map(db, item_zongpai_nos)
|
|
|
|
existing_boxes = []
|
|
current_zongpai_boxes = []
|
|
max_box_no = 0
|
|
for box in boxes:
|
|
items = box_items[box.id]
|
|
for item in items:
|
|
if item.zongpai_no == zongpai_no:
|
|
current_zongpai_boxes.append({
|
|
"box_item_id": item.id,
|
|
"box_no": box.box_no,
|
|
"quantity": int(item.quantity or 0),
|
|
})
|
|
existing_boxes.append({
|
|
"box_no": box.box_no,
|
|
"items": [
|
|
{
|
|
"box_item_id": item.id,
|
|
"zongpai_no": item.zongpai_no,
|
|
"work_order_no": erp_item_info_map.get(
|
|
item.zongpai_no, {}
|
|
).get("work_order_no"),
|
|
"quantity": int(item.quantity or 0),
|
|
"total_quantity": erp_item_info_map.get(
|
|
item.zongpai_no, {}
|
|
).get("total_quantity"),
|
|
}
|
|
for item in items
|
|
],
|
|
})
|
|
if box.box_no > max_box_no:
|
|
max_box_no = box.box_no
|
|
|
|
return {
|
|
"zongpai_no": zongpai_no,
|
|
"paichan_no": paichan_no,
|
|
"work_order_no": erp["work_order_no"],
|
|
"quantity": quantity,
|
|
"current_zongpai_boxes": current_zongpai_boxes,
|
|
"existing_boxes": existing_boxes,
|
|
"max_box_no": max_box_no,
|
|
"suggested_box_no": max_box_no + 1,
|
|
}
|
|
|
|
|
|
def save_box_record(db: Session, req: BoxSaveRequest) -> dict:
|
|
"""保存装箱记录。箱已存在时仅追加明细（多码一箱场景）。"""
|
|
erp = query_erp_info(db, req.zongpai_no)
|
|
paichan_no = erp["paichan_no"]
|
|
|
|
if req.box_mode == "one-to-one":
|
|
existing_box_no = (
|
|
db.query(FinishedGoodsBox.box_no)
|
|
.join(FinishedGoodsBoxItem, FinishedGoodsBox.id == FinishedGoodsBoxItem.box_id)
|
|
.filter(
|
|
FinishedGoodsBox.paichan_no == paichan_no,
|
|
FinishedGoodsBoxItem.zongpai_no == req.zongpai_no,
|
|
)
|
|
.scalar()
|
|
)
|
|
if existing_box_no is not None:
|
|
raise DuplicateZongpaiBindError(paichan_no, existing_box_no, req.zongpai_no)
|
|
|
|
box = (
|
|
db.query(FinishedGoodsBox)
|
|
.filter(
|
|
FinishedGoodsBox.paichan_no == paichan_no,
|
|
FinishedGoodsBox.box_no == req.box_no,
|
|
)
|
|
.first()
|
|
)
|
|
|
|
if box:
|
|
if req.box_mode in ("one-to-one", "one-to-many"):
|
|
raise DuplicateBoxItemError(paichan_no, req.box_no)
|
|
existing_item = (
|
|
db.query(FinishedGoodsBoxItem)
|
|
.filter(
|
|
FinishedGoodsBoxItem.box_id == box.id,
|
|
FinishedGoodsBoxItem.zongpai_no == req.zongpai_no,
|
|
)
|
|
.first()
|
|
)
|
|
if existing_item:
|
|
raise DuplicateBoxItemError(paichan_no, req.box_no)
|
|
else:
|
|
box = FinishedGoodsBox(
|
|
paichan_no=paichan_no,
|
|
box_no=req.box_no,
|
|
)
|
|
db.add(box)
|
|
db.flush()
|
|
|
|
item = FinishedGoodsBoxItem(
|
|
box_id=box.id,
|
|
zongpai_no=req.zongpai_no,
|
|
quantity=req.quantity,
|
|
)
|
|
db.add(item)
|
|
db.commit()
|
|
db.refresh(item)
|
|
|
|
return {
|
|
"box_item_id": item.id,
|
|
"paichan_no": paichan_no,
|
|
"box_no": req.box_no,
|
|
"zongpai_no": req.zongpai_no,
|
|
"quantity": req.quantity,
|
|
"created_at": item.created_at.isoformat(),
|
|
}
|
|
|
|
|
|
def update_box_item(db: Session, box_item_id: int, req: BoxUpdateRequest) -> dict:
|
|
"""更新一码多箱已分配明细的箱号和数量。"""
|
|
item = (
|
|
db.query(FinishedGoodsBoxItem)
|
|
.filter(FinishedGoodsBoxItem.id == box_item_id)
|
|
.first()
|
|
)
|
|
if item is None:
|
|
raise BoxItemNotFoundError()
|
|
|
|
current_box = (
|
|
db.query(FinishedGoodsBox)
|
|
.filter(FinishedGoodsBox.id == item.box_id)
|
|
.first()
|
|
)
|
|
if current_box is None:
|
|
raise InvalidBoxItemError()
|
|
|
|
paichan_no = current_box.paichan_no
|
|
target_box = (
|
|
db.query(FinishedGoodsBox)
|
|
.filter(
|
|
FinishedGoodsBox.paichan_no == paichan_no,
|
|
FinishedGoodsBox.box_no == req.box_no,
|
|
)
|
|
.first()
|
|
)
|
|
|
|
if target_box is not None and target_box.id != current_box.id:
|
|
raise DuplicateBoxItemError(paichan_no, req.box_no)
|
|
|
|
if target_box is None:
|
|
target_box = FinishedGoodsBox(paichan_no=paichan_no, box_no=req.box_no)
|
|
db.add(target_box)
|
|
db.flush()
|
|
|
|
old_box_id = item.box_id
|
|
item.box_id = target_box.id
|
|
item.quantity = req.quantity
|
|
|
|
if old_box_id != target_box.id:
|
|
old_box_has_items = (
|
|
db.query(FinishedGoodsBoxItem.id)
|
|
.filter(
|
|
FinishedGoodsBoxItem.box_id == old_box_id,
|
|
FinishedGoodsBoxItem.id != item.id,
|
|
)
|
|
.first()
|
|
)
|
|
if old_box_has_items is None:
|
|
old_box = (
|
|
db.query(FinishedGoodsBox)
|
|
.filter(FinishedGoodsBox.id == old_box_id)
|
|
.first()
|
|
)
|
|
if old_box is not None:
|
|
db.delete(old_box)
|
|
|
|
db.commit()
|
|
db.refresh(item)
|
|
|
|
return {
|
|
"box_item_id": item.id,
|
|
"paichan_no": paichan_no,
|
|
"box_no": req.box_no,
|
|
"zongpai_no": item.zongpai_no,
|
|
"quantity": int(item.quantity or 0),
|
|
"updated_at": item.created_at.isoformat(),
|
|
}
|
|
|
|
|
|
def delete_box_item(db: Session, box_item_id: int) -> dict:
|
|
"""删除误录的装箱明细；若箱号下无其他明细，同步删除空箱号。"""
|
|
item = (
|
|
db.query(FinishedGoodsBoxItem)
|
|
.filter(FinishedGoodsBoxItem.id == box_item_id)
|
|
.first()
|
|
)
|
|
if item is None:
|
|
raise BoxItemNotFoundError()
|
|
|
|
box_id = item.box_id
|
|
db.delete(item)
|
|
db.flush()
|
|
|
|
remaining = (
|
|
db.query(FinishedGoodsBoxItem.id)
|
|
.filter(FinishedGoodsBoxItem.box_id == box_id)
|
|
.first()
|
|
)
|
|
if remaining is None:
|
|
box = db.query(FinishedGoodsBox).filter(FinishedGoodsBox.id == box_id).first()
|
|
if box is not None:
|
|
db.delete(box)
|
|
|
|
db.commit()
|
|
return {"box_item_id": box_item_id, "deleted": True}
|