Files
zkxw/backend/app/crud/production_material_prep.py
T
2026-06-15 17:17:26 +08:00

1307 lines
44 KiB
Python

"""配置试剂配制单 CRUD"""
from __future__ import annotations
from datetime import date, datetime
from decimal import Decimal
from typing import Any, Dict, List, Optional, Tuple
from sqlalchemy import and_, func, or_
from sqlalchemy.orm import Session
from app.core.utils import generate_uuid, parse_datetime
from app.crud.production_material import _dec, _fmt_dt, _is_enabled_status
from app.crud.production_material_stock_batch import (
_create_batch_and_usage,
_get_material_for_update,
_parse_date,
_parse_expire_date,
_require_str,
_stock_out_batch,
stock_in_no_exists,
_optional_str,
)
from app.crud.production_material_usage import list_available_requisitions_for_prep
from app.models.models import SysUser
from app.models.production_material import (
BizProductionMaterial,
BizProductionMaterialPrep,
BizProductionMaterialPrepLine,
BizProductionMaterialStockBatch,
BizProductionMaterialUsage,
)
PREP_STATUSES = ("draft", "completed", "cancelled")
def _status_label(status: str) -> str:
if status == "completed":
return "已完成"
if status == "cancelled":
return "已作废"
return "草稿"
def _fmt_date(d: Optional[date]) -> str:
return d.strftime("%Y-%m-%d") if d else ""
def _batch_map_for_prep_lines(
db: Session, batch_ids: List[str]
) -> Dict[str, BizProductionMaterialStockBatch]:
ids = [x for x in batch_ids if x]
if not ids:
return {}
rows = (
db.query(BizProductionMaterialStockBatch)
.filter(BizProductionMaterialStockBatch.id.in_(ids), BizProductionMaterialStockBatch.del_flag == 0)
.all()
)
return {r.id: r for r in rows}
def _user_nick_name_map(db: Session, user_ids: List[str]) -> Dict[str, str]:
ids = list({x for x in user_ids if x})
if not ids:
return {}
rows = db.query(SysUser.id, SysUser.nick_name).filter(SysUser.id.in_(ids)).all()
return {u.id: (u.nick_name or "").strip() for u in rows}
def _generate_prep_no(db: Session, tenant_no: str) -> str:
day = datetime.now().strftime("%Y%m%d")
like = f"PZ-{day}-%"
cnt = (
db.query(func.count(BizProductionMaterialPrep.id))
.filter(
BizProductionMaterialPrep.tenant_no == tenant_no,
BizProductionMaterialPrep.prep_no.like(like),
)
.scalar()
)
return f"PZ-{day}-{(cnt or 0) + 1:03d}"
def _generate_stock_in_no(db: Session, tenant_no: str) -> str:
day = datetime.now().strftime("%Y%m%d")
like = f"RK{day}%"
cnt = (
db.query(func.count(BizProductionMaterialPrep.id))
.filter(
BizProductionMaterialPrep.tenant_no == tenant_no,
BizProductionMaterialPrep.stock_in_no.like(like),
BizProductionMaterialPrep.del_flag == 0,
)
.scalar()
)
sno = f"RK{day}{(cnt or 0) + 1:04d}"
if stock_in_no_exists(db, sno, tenant_no):
sno = f"RK{day}{(cnt or 0) + 1:04d}P"
return sno
def _line_to_dict(row: BizProductionMaterialPrepLine) -> Dict[str, Any]:
return {
"id": row.id,
"prepId": row.prep_id,
"componentMaterialId": row.component_material_id,
"componentMaterialType": row.component_material_type or "",
"componentMaterialCode": row.component_material_code or "",
"componentMaterialName": row.component_material_name or "",
"requisitionUsageId": (row.requisition_usage_id or "").strip(),
"actualQty": _dec(row.actual_qty),
"usageUnit": row.usage_unit or "",
"orderNum": row.order_num if row.order_num is not None else 0,
"remark": row.remark or "",
}
def _header_to_dict(
row: BizProductionMaterialPrep,
*,
create_by_name: str = "",
line_count: int = 0,
) -> Dict[str, Any]:
return {
"id": row.id,
"prepNo": row.prep_no or "",
"configuredMaterialId": row.configured_material_id,
"configuredMaterialCode": row.configured_material_code or "",
"configuredMaterialName": row.configured_material_name or "",
"prepTime": _fmt_dt(row.prep_time),
"outputQty": _dec(row.output_qty),
"outputUnit": row.output_unit or "",
"batchNo": row.batch_no or "",
"storageLocation": row.storage_location or "",
"expireDate": _fmt_date(row.expire_date),
"stockInNo": row.stock_in_no or "",
"outputBatchId": row.output_batch_id or "",
"status": row.status or "draft",
"statusLabel": _status_label(row.status or "draft"),
"lineCount": line_count,
"remark": row.remark or "",
"createBy": row.create_by or "",
"createByName": create_by_name or row.create_by or "",
"createTime": _fmt_dt(row.create_time),
"updateTime": _fmt_dt(row.update_time),
}
def _get_prep_row(db: Session, prep_id: str, tenant_no: str) -> BizProductionMaterialPrep:
row = (
db.query(BizProductionMaterialPrep)
.filter(
BizProductionMaterialPrep.id == prep_id,
BizProductionMaterialPrep.tenant_no == tenant_no,
BizProductionMaterialPrep.del_flag == 0,
)
.first()
)
if not row:
raise ValueError("配制单不存在")
return row
def _require_draft(row: BizProductionMaterialPrep) -> None:
if (row.status or "").strip() != "draft":
raise ValueError("仅草稿状态的配制单可编辑")
def _validate_requisition(
db: Session,
req_id: str,
*,
tenant_no: str,
component_material_id: str,
seen: set,
) -> BizProductionMaterialUsage:
if not req_id:
raise ValueError("请选择子物料领用记录")
if req_id in seen:
raise ValueError("配制明细中存在重复的领用记录")
seen.add(req_id)
req = (
db.query(BizProductionMaterialUsage)
.filter(
BizProductionMaterialUsage.id == req_id,
BizProductionMaterialUsage.tenant_no == tenant_no,
BizProductionMaterialUsage.del_flag == 0,
BizProductionMaterialUsage.source_type == "manual_out",
BizProductionMaterialUsage.change_type == "use",
)
.first()
)
if not req:
raise ValueError("领用记录不存在或已失效")
if (req.material_id or "").strip() != component_material_id:
raise ValueError("领用记录与子物料不匹配")
return req
def _parse_prep_lines(
db: Session,
lines: List[Dict[str, Any]],
*,
tenant_no: str,
) -> List[Dict[str, Any]]:
if not lines:
raise ValueError("请至少添加一条配制明细")
parsed: List[Dict[str, Any]] = []
seen_req: set = set()
for idx, it in enumerate(lines):
comp_id = (
it.get("componentMaterialId") or it.get("component_material_id") or ""
).strip()
if not comp_id:
raise ValueError("请选择子物料")
comp_type = (
it.get("componentMaterialType") or it.get("component_material_type") or ""
).strip().lower()
if comp_type not in ("reagent", "consumable"):
raise ValueError("子物料类型无效")
comp = (
db.query(BizProductionMaterial)
.filter(
BizProductionMaterial.id == comp_id,
BizProductionMaterial.tenant_no == tenant_no,
BizProductionMaterial.del_flag == 0,
)
.first()
)
if not comp:
raise ValueError("子物料不存在")
req_id = (
it.get("requisitionUsageId") or it.get("requisition_usage_id") or ""
).strip()
req = _validate_requisition(
db, req_id, tenant_no=tenant_no, component_material_id=comp_id, seen=seen_req
)
qty_raw = it.get("actualQty") if it.get("actualQty") is not None else it.get("actual_qty")
if qty_raw is None or Decimal(str(qty_raw)) <= 0:
raise ValueError("实际用量必须大于0")
actual_qty = Decimal(str(qty_raw))
usage_unit = (
it.get("usageUnit") or it.get("usage_unit") or comp.unit or ""
).strip()[:32]
parsed.append(
{
"idx": idx,
"component": comp,
"component_type": comp_type,
"requisition": req,
"actual_qty": actual_qty,
"usage_unit": usage_unit,
"remark": (it.get("remark") or "").strip()[:512] or None,
}
)
return parsed
def list_preps(
db: Session,
tenant_no: str,
*,
page: int = 1,
page_size: int = 10,
keyword: Optional[str] = None,
status: Optional[str] = None,
configured_material_id: Optional[str] = None,
) -> Tuple[List[Dict[str, Any]], int]:
q0 = db.query(BizProductionMaterialPrep).filter(
BizProductionMaterialPrep.tenant_no == tenant_no,
BizProductionMaterialPrep.del_flag == 0,
)
st = (status or "").strip().lower()
if st in PREP_STATUSES:
q0 = q0.filter(BizProductionMaterialPrep.status == st)
cmid = (configured_material_id or "").strip()
if cmid:
q0 = q0.filter(BizProductionMaterialPrep.configured_material_id == cmid)
kw = (keyword or "").strip()
if kw:
like = f"%{kw}%"
q0 = q0.filter(
or_(
BizProductionMaterialPrep.prep_no.like(like),
BizProductionMaterialPrep.configured_material_code.like(like),
BizProductionMaterialPrep.configured_material_name.like(like),
BizProductionMaterialPrep.batch_no.like(like),
BizProductionMaterialPrep.stock_in_no.like(like),
)
)
total = q0.count()
rows = (
q0.order_by(BizProductionMaterialPrep.prep_time.desc(), BizProductionMaterialPrep.create_time.desc())
.offset((page - 1) * page_size)
.limit(page_size)
.all()
)
prep_ids = [r.id for r in rows]
line_counts: Dict[str, int] = {}
if prep_ids:
cnt_rows = (
db.query(
BizProductionMaterialPrepLine.prep_id,
func.count(BizProductionMaterialPrepLine.id),
)
.filter(
BizProductionMaterialPrepLine.prep_id.in_(prep_ids),
BizProductionMaterialPrepLine.del_flag == 0,
)
.group_by(BizProductionMaterialPrepLine.prep_id)
.all()
)
line_counts = {pid: int(c) for pid, c in cnt_rows}
name_map = _user_nick_name_map(db, [r.create_by for r in rows if r.create_by])
items = [
_header_to_dict(
r,
create_by_name=name_map.get(r.create_by or "", ""),
line_count=line_counts.get(r.id, 0),
)
for r in rows
]
return items, total
def get_prep_detail(db: Session, prep_id: str, tenant_no: str) -> Dict[str, Any]:
row = _get_prep_row(db, prep_id, tenant_no)
lines = (
db.query(BizProductionMaterialPrepLine)
.filter(
BizProductionMaterialPrepLine.prep_id == prep_id,
BizProductionMaterialPrepLine.tenant_no == tenant_no,
BizProductionMaterialPrepLine.del_flag == 0,
)
.order_by(
BizProductionMaterialPrepLine.order_num.asc(),
BizProductionMaterialPrepLine.create_time.asc(),
)
.all()
)
name_map = _user_nick_name_map(db, [row.create_by] if row.create_by else [])
result = _header_to_dict(
row,
create_by_name=name_map.get(row.create_by or "", ""),
line_count=len(lines),
)
req_ids = list(
{(ln.requisition_usage_id or "").strip() for ln in lines if (ln.requisition_usage_id or "").strip()}
)
req_map: Dict[str, BizProductionMaterialUsage] = {}
if req_ids:
reqs = (
db.query(BizProductionMaterialUsage)
.filter(
BizProductionMaterialUsage.id.in_(req_ids),
BizProductionMaterialUsage.tenant_no == tenant_no,
BizProductionMaterialUsage.del_flag == 0,
)
.all()
)
req_map = {r.id: r for r in reqs}
req_bmap = _batch_map_for_prep_lines(db, [r.batch_id for r in req_map.values() if r.batch_id])
enriched_lines: List[Dict[str, Any]] = []
for ln in lines:
item = _line_to_dict(ln)
req = req_map.get((ln.requisition_usage_id or "").strip())
if req:
batch = req_bmap.get(req.batch_id or "")
item["batchNo"] = req.batch_no or ""
item["stockInNo"] = req.stock_in_no or ""
item["catalogNo"] = req.catalog_no or ""
item["manufacturer"] = req.manufacturer or ""
item["storageLocation"] = (req.storage_location or "").strip() or (
(batch.storage_location or "") if batch else ""
)
item["requisitionQty"] = abs(float(req.change_qty or 0))
if batch:
item["specModel"] = batch.spec_model or ""
item["stockInDate"] = _fmt_date(batch.stock_in_date)
item["expireDate"] = _fmt_date(batch.expire_date)
enriched_lines.append(item)
result["lines"] = enriched_lines
batch_id = (row.output_batch_id or "").strip()
if batch_id:
batch = (
db.query(BizProductionMaterialStockBatch)
.filter(
BizProductionMaterialStockBatch.id == batch_id,
BizProductionMaterialStockBatch.tenant_no == tenant_no,
BizProductionMaterialStockBatch.del_flag == 0,
)
.first()
)
if batch:
result["specModel"] = batch.spec_model or ""
result["catalogNo"] = batch.catalog_no or ""
result["manufacturer"] = batch.manufacturer or ""
result["stockInDate"] = _fmt_date(batch.stock_in_date)
result["unitPrice"] = None
result["amount"] = None
return result
def create_prep(
db: Session,
*,
data: Dict[str, Any],
tenant_no: str,
user_id: Optional[str] = None,
dept_id: str = "",
) -> Dict[str, Any]:
"""创建配制单(草稿):用于旧版流程。"""
configured_id = (
data.get("configuredMaterialId") or data.get("configured_material_id") or ""
).strip()
if not configured_id:
raise ValueError("请选择配置试剂")
configured = (
db.query(BizProductionMaterial)
.filter(
BizProductionMaterial.id == configured_id,
BizProductionMaterial.tenant_no == tenant_no,
BizProductionMaterial.del_flag == 0,
)
.first()
)
if not configured:
raise ValueError("配置试剂不存在")
if (configured.material_type or "").strip() != "configured_reagent":
raise ValueError("所选物料不是配置试剂")
prep_time_raw = data.get("prepTime") or data.get("prep_time")
prep_time = parse_datetime(prep_time_raw) if prep_time_raw else datetime.now()
output_qty_raw = data.get("outputQty") if data.get("outputQty") is not None else data.get("output_qty")
if output_qty_raw is None or Decimal(str(output_qty_raw)) <= 0:
raise ValueError("产出数量必须大于0")
output_qty = Decimal(str(output_qty_raw))
batch_no = _require_str(data.get("batchNo") or data.get("batch_no"), "产出批号")
expire_raw = data.get("expireDate") or data.get("expire_date")
expire_date = _parse_expire_date(expire_raw)
lines_raw = data.get("lines") or []
parsed_lines = _parse_prep_lines(db, lines_raw, tenant_no=tenant_no)
now = datetime.now()
prep_id = generate_uuid()
prep_no = _generate_prep_no(db, tenant_no)
prep = BizProductionMaterialPrep(
id=prep_id,
prep_no=prep_no,
configured_material_id=configured_id,
configured_material_code=configured.material_code,
configured_material_name=configured.material_name,
prep_time=prep_time,
output_qty=output_qty,
output_unit=(data.get("outputUnit") or data.get("output_unit") or configured.unit or "").strip()[:32] or None,
batch_no=batch_no,
storage_location=_optional_str(data.get("storageLocation") or data.get("storage_location")),
expire_date=expire_date,
status="draft",
tenant_no=tenant_no,
remark=(data.get("remark") or "").strip()[:512] or None,
create_by=user_id,
create_dept_id=dept_id or configured.create_dept_id or "",
create_time=now,
update_by=user_id,
update_time=now,
del_flag=0,
)
db.add(prep)
for pl in parsed_lines:
comp = pl["component"]
db.add(
BizProductionMaterialPrepLine(
id=generate_uuid(),
prep_id=prep_id,
component_material_id=comp.id,
component_material_type=pl["component_type"],
component_material_code=comp.material_code,
component_material_name=comp.material_name,
requisition_usage_id=pl["requisition"].id,
actual_qty=pl["actual_qty"],
usage_unit=pl["usage_unit"] or None,
tenant_no=tenant_no,
order_num=pl["idx"],
remark=pl["remark"],
create_by=user_id,
create_dept_id=dept_id or configured.create_dept_id or "",
create_time=now,
update_by=user_id,
update_time=now,
del_flag=0,
)
)
db.commit()
return get_prep_detail(db, prep_id, tenant_no)
def create_quick_prep(
db: Session,
*,
data: Dict[str, Any],
tenant_no: str,
user_id: Optional[str] = None,
dept_id: str = "",
) -> Dict[str, Any]:
"""一次性配制:创建即完成(扣减子物料 → 入库配置试剂 → 自动出库给操作人)。"""
configured_name = (data.get("configuredMaterialName") or data.get("configured_material_name") or "").strip()
if not configured_name:
raise ValueError("请输入配置试剂名称")
output_qty_raw = data.get("outputQty") if data.get("outputQty") is not None else data.get("output_qty")
if output_qty_raw is None or Decimal(str(output_qty_raw)) <= 0:
raise ValueError("产出数量必须大于0")
output_qty = Decimal(str(output_qty_raw))
lines_raw = data.get("lines") or []
if not lines_raw:
raise ValueError("请至少添加一条子物料")
# 校验并解析子物料
parsed_lines = _parse_quick_prep_lines(db, lines_raw, tenant_no=tenant_no)
now = datetime.now()
prep_id = generate_uuid()
prep_no = _generate_prep_no(db, tenant_no)
stock_in_no_raw = (data.get("stockInNo") or data.get("stock_in_no") or "").strip()
if stock_in_no_raw:
if stock_in_no_exists(db, stock_in_no_raw, tenant_no):
raise ValueError("入库编号已存在")
stock_in_no = stock_in_no_raw
else:
stock_in_no = _generate_stock_in_no(db, tenant_no)
# 配置试剂为自配物料,不核算单价与金额(库表 NOT NULL,存 0;接口层按类型屏蔽展示)
unit_price_out = Decimal("0")
amount_out = Decimal("0")
spec_model = _optional_str(data.get("specModel") or data.get("spec_model"))
output_unit = _optional_str(data.get("outputUnit") or data.get("output_unit")) or "瓶"
# 自动创建或获取配置试剂物料
configured_material = _get_or_create_configured_material(
db,
configured_name,
tenant_no,
user_id,
dept_id,
spec_model=spec_model,
unit=output_unit,
)
output_unit = output_unit or configured_material.unit or "瓶"
# 生成批号
batch_no_raw = (data.get("batchNo") or data.get("batch_no") or "").strip()
batch_no = batch_no_raw if batch_no_raw else prep_no
catalog_no_raw = (data.get("catalogNo") or data.get("catalog_no") or "").strip()
catalog_no = catalog_no_raw if catalog_no_raw else (configured_material.material_code or batch_no or "—")
manufacturer = (data.get("manufacturer") or "").strip() or "自配"
# 入库日期(业务日期,仅日期)与变动记录时间(含时分秒)分离,对齐普通入库逻辑
stock_in_date_raw = data.get("stockInDate") or data.get("stock_in_date")
stock_in_date = _parse_date(stock_in_date_raw) if stock_in_date_raw else now.date()
prep_time_raw = (
data.get("prepTime")
or data.get("prep_time")
or data.get("stockInTime")
or data.get("stock_in_time")
)
operate_time = parse_datetime(prep_time_raw) if prep_time_raw else now
try:
# 1. 创建配制单(状态直接为 completed)
expire_date = _parse_expire_date(data.get("expireDate") or data.get("expire_date") or "")
if not expire_date:
expire_date = datetime(now.year + 1, now.month, now.day).date()
prep = BizProductionMaterialPrep(
id=prep_id,
prep_no=prep_no,
configured_material_id=configured_material.id,
configured_material_code=configured_material.material_code,
configured_material_name=configured_name,
prep_time=operate_time,
output_qty=output_qty,
output_unit=output_unit,
batch_no=batch_no,
storage_location=_optional_str(data.get("storageLocation") or data.get("storage_location")),
expire_date=expire_date,
stock_in_no=stock_in_no,
status="completed",
tenant_no=tenant_no,
remark=(data.get("remark") or "").strip()[:512] or None,
create_by=user_id,
create_dept_id=dept_id or configured_material.create_dept_id or "",
create_time=now,
update_by=user_id,
update_time=now,
del_flag=0,
)
db.add(prep)
# 2. 创建配制明细并扣减子物料
for pl in parsed_lines:
comp = pl["component"]
req = pl["requisition"]
actual_qty = pl["actual_qty"]
# 创建配制明细行
db.add(
BizProductionMaterialPrepLine(
id=generate_uuid(),
prep_id=prep_id,
component_material_id=comp.id,
component_material_type=pl["component_type"],
component_material_code=comp.material_code,
component_material_name=comp.material_name,
requisition_usage_id=req.id if req else None,
actual_qty=actual_qty,
usage_unit=pl["usage_unit"] or None,
tenant_no=tenant_no,
order_num=pl["idx"],
remark=pl.get("remark"),
create_by=user_id,
create_dept_id=dept_id or configured_material.create_dept_id or "",
create_time=now,
update_by=user_id,
update_time=now,
del_flag=0,
)
)
# 扣减子物料库存(写入 prep_consume 记录)
_write_quick_prep_consume_usage(
db,
prep=prep,
material=comp,
requisition=req,
actual_qty=actual_qty,
usage_unit=pl["usage_unit"] or None,
tenant_no=tenant_no,
user_id=user_id,
dept_id=dept_id or configured_material.create_dept_id or "",
operate_time=operate_time,
)
# 3. 创建配置试剂入库批次
catalog_no = catalog_no if catalog_no else (configured_material.material_code or batch_no or "—")
batch = _create_batch_and_usage(
db,
material=configured_material,
stock_in_no=stock_in_no,
stock_in_date=stock_in_date,
expire_date=expire_date,
catalog_no=catalog_no,
batch_no=batch_no,
manufacturer=manufacturer,
storage_location=_optional_str(data.get("storageLocation") or data.get("storage_location")),
unit_price=unit_price_out,
amount=amount_out,
qty=output_qty,
remark=f"配制单 {prep_no}",
tenant_no=tenant_no,
user_id=user_id,
operate_time=operate_time,
commit=False,
source_type="prep_in",
prep_id=prep.id,
)
prep.output_batch_id = batch.id
db.add(prep)
# 4. 自动出库给当前操作人
_create_auto_stock_out(
db,
batch=batch,
output_qty=output_qty,
tenant_no=tenant_no,
user_id=user_id,
dept_id=dept_id or configured_material.create_dept_id or "",
operate_time=operate_time,
)
db.commit()
except Exception:
db.rollback()
raise
return get_prep_detail(db, prep_id, tenant_no)
def _parse_quick_prep_lines(
db: Session,
lines: List[Dict[str, Any]],
*,
tenant_no: str,
) -> List[Dict[str, Any]]:
"""解析一次性配制的子物料(按领用记录)。"""
if not lines:
raise ValueError("请至少添加一条子物料")
parsed: List[Dict[str, Any]] = []
seen_req: set = set()
for idx, it in enumerate(lines):
req_id = (
it.get("requisitionUsageId")
or it.get("requisition_usage_id")
or ""
).strip()
if not req_id:
raise ValueError("请选择领用记录")
comp_id = (it.get("componentMaterialId") or it.get("component_material_id") or "").strip()
if not comp_id:
raise ValueError("子物料不能为空")
comp_type = (it.get("componentMaterialType") or it.get("component_material_type") or "").strip().lower()
if comp_type not in ("reagent", "consumable"):
raise ValueError("子物料类型无效")
qty_raw = it.get("consumptionQty") if it.get("consumptionQty") is not None else it.get("consumption_qty")
if qty_raw is None or Decimal(str(qty_raw)) <= 0:
raise ValueError("消耗数量必须大于0")
consumption_qty = Decimal(str(qty_raw))
if req_id in seen_req:
raise ValueError("同一领用记录不可重复使用")
seen_req.add(req_id)
req = (
db.query(BizProductionMaterialUsage)
.filter(
BizProductionMaterialUsage.id == req_id,
BizProductionMaterialUsage.tenant_no == tenant_no,
BizProductionMaterialUsage.del_flag == 0,
BizProductionMaterialUsage.source_type == "manual_out",
BizProductionMaterialUsage.change_type == "use",
)
.first()
)
if not req:
raise ValueError("领用记录不存在或已失效")
if (req.requisition_status or "open") == "closed":
raise ValueError("领用记录已标记为使用完,不可用于配制")
comp = (
db.query(BizProductionMaterial)
.filter(
BizProductionMaterial.id == comp_id,
BizProductionMaterial.tenant_no == tenant_no,
BizProductionMaterial.del_flag == 0,
)
.first()
)
if not comp:
raise ValueError("子物料不存在")
if (req.material_id or "").strip() != comp.id:
raise ValueError("领用记录与子物料不匹配")
req_qty = abs(req.change_qty or Decimal("0"))
if consumption_qty > req_qty:
raise ValueError(
f"子物料「{comp.material_name}」消耗数量不能超过领用数量 {req_qty}"
)
batch = None
if req.batch_id:
batch = (
db.query(BizProductionMaterialStockBatch)
.filter(
BizProductionMaterialStockBatch.id == req.batch_id,
BizProductionMaterialStockBatch.tenant_no == tenant_no,
BizProductionMaterialStockBatch.del_flag == 0,
)
.first()
)
usage_unit = (it.get("usageUnit") or it.get("usage_unit") or req.usage_unit or comp.unit or "").strip()[:32]
parsed.append({
"idx": idx,
"component": comp,
"component_type": comp_type,
"batch": batch,
"requisition": req,
"actual_qty": consumption_qty,
"usage_unit": usage_unit,
"remark": (it.get("remark") or "").strip()[:512] or None,
})
return parsed
def _get_or_create_configured_material(
db: Session,
name: str,
tenant_no: str,
user_id: Optional[str],
dept_id: str,
*,
spec_model: Optional[str] = None,
unit: Optional[str] = None,
) -> BizProductionMaterial:
"""获取或创建配置试剂物料(按名称匹配)。"""
# 先查找是否存在
existing = (
db.query(BizProductionMaterial)
.filter(
BizProductionMaterial.material_name == name,
BizProductionMaterial.material_type == "configured_reagent",
BizProductionMaterial.tenant_no == tenant_no,
BizProductionMaterial.del_flag == 0,
)
.first()
)
if existing:
if not _is_enabled_status(existing.status):
raise ValueError(f"配置试剂「{name}」已停用,请启用后再配制")
return existing
# 自动生成编码
day = datetime.now().strftime("%Y%m%d")
like = f"PZR-{day}-%"
cnt = (
db.query(func.count(BizProductionMaterial.id))
.filter(
BizProductionMaterial.tenant_no == tenant_no,
BizProductionMaterial.material_code.like(like),
BizProductionMaterial.del_flag == 0,
)
.scalar()
)
code = f"PZR-{day}-{(cnt or 0) + 1:03d}"
now = datetime.now()
mat = BizProductionMaterial(
id=generate_uuid(),
material_type="configured_reagent",
material_code=code,
material_name=name,
spec_model=spec_model,
unit=unit or "瓶",
stock_qty=Decimal("0"),
warning_qty=Decimal("0"),
status="1",
tenant_no=tenant_no,
create_by=user_id,
create_dept_id=dept_id,
create_time=now,
update_by=user_id,
update_time=now,
del_flag=0,
)
db.add(mat)
return mat
def _write_quick_prep_consume_usage(
db: Session,
*,
prep: BizProductionMaterialPrep,
material: BizProductionMaterial,
requisition: Optional[BizProductionMaterialUsage],
actual_qty: Decimal,
usage_unit: Optional[str],
tenant_no: str,
user_id: Optional[str],
dept_id: str,
operate_time: datetime,
) -> None:
"""写入子物料消耗记录(关联领用记录,不重复扣减批次库存)。"""
if not requisition:
raise ValueError("子物料必须关联领用记录")
stock_qty = material.stock_qty if material.stock_qty is not None else Decimal("0")
unit_price = requisition.unit_price if requisition else None
line_amount = Decimal("0")
if unit_price is not None and actual_qty > 0:
line_amount = (Decimal(str(unit_price)) * actual_qty).quantize(Decimal("0.01"))
usage = BizProductionMaterialUsage(
id=generate_uuid(),
material_id=material.id,
material_type=material.material_type,
material_code=material.material_code,
material_name=material.material_name,
batch_id=requisition.batch_id if requisition else None,
stock_in_no=requisition.stock_in_no if requisition else None,
catalog_no=requisition.catalog_no if requisition else None,
batch_no=requisition.batch_no if requisition else None,
manufacturer=requisition.manufacturer if requisition else None,
storage_location=requisition.storage_location if requisition else None,
unit_price=unit_price,
amount=line_amount if line_amount > 0 else (requisition.amount if requisition else None),
change_type="use",
change_qty=-actual_qty,
usage_unit=usage_unit,
stock_before=stock_qty,
stock_after=stock_qty, # 汇总库存暂不扣减,按批次扣减
parent_usage_id=requisition.id if requisition else None,
prep_id=prep.id,
source_type="prep_consume",
tenant_no=tenant_no,
remark=f"配制单 {prep.prep_no}",
create_by=user_id,
create_dept_id=dept_id,
create_time=operate_time,
update_by=user_id,
update_time=operate_time,
del_flag=0,
)
db.add(usage)
def _create_auto_stock_out(
db: Session,
*,
batch,
output_qty: Decimal,
tenant_no: str,
user_id: Optional[str],
dept_id: str,
operate_time: datetime,
) -> None:
"""自动出库配置试剂给当前操作人(写入领用记录)。"""
if not user_id:
return
_stock_out_batch(
db,
batch.id,
qty=output_qty,
receiver=user_id,
purpose="配制自用",
remark="配制产出自动出库",
usage_unit=batch.unit,
tenant_no=tenant_no,
user_id=user_id,
operate_time=operate_time,
)
def update_prep(
db: Session,
prep_id: str,
*,
data: Dict[str, Any],
tenant_no: str,
user_id: Optional[str] = None,
dept_id: str = "",
) -> Dict[str, Any]:
prep = _get_prep_row(db, prep_id, tenant_no)
_require_draft(prep)
now = datetime.now()
if data.get("prepTime") or data.get("prep_time"):
prep.prep_time = parse_datetime(data.get("prepTime") or data.get("prep_time"))
if data.get("outputQty") is not None or data.get("output_qty") is not None:
output_qty_raw = data.get("outputQty") if data.get("outputQty") is not None else data.get("output_qty")
if Decimal(str(output_qty_raw)) <= 0:
raise ValueError("产出数量必须大于0")
prep.output_qty = Decimal(str(output_qty_raw))
if data.get("outputUnit") is not None or data.get("output_unit") is not None:
prep.output_unit = (
data.get("outputUnit") or data.get("output_unit") or ""
).strip()[:32] or None
if data.get("batchNo") or data.get("batch_no"):
prep.batch_no = _require_str(data.get("batchNo") or data.get("batch_no"), "产出批号")
if data.get("storageLocation") is not None or data.get("storage_location") is not None:
prep.storage_location = _optional_str(
data.get("storageLocation") if data.get("storageLocation") is not None else data.get("storage_location")
)
if data.get("expireDate") or data.get("expire_date"):
prep.expire_date = _parse_expire_date(data.get("expireDate") or data.get("expire_date"))
if data.get("remark") is not None:
prep.remark = (data.get("remark") or "").strip()[:512] or None
lines_raw = data.get("lines")
if lines_raw is not None:
parsed_lines = _parse_prep_lines(db, lines_raw, tenant_no=tenant_no)
existing = (
db.query(BizProductionMaterialPrepLine)
.filter(
BizProductionMaterialPrepLine.prep_id == prep_id,
BizProductionMaterialPrepLine.del_flag == 0,
)
.all()
)
for ln in existing:
ln.del_flag = 1
ln.update_by = user_id
ln.update_time = now
db.add(ln)
for pl in parsed_lines:
comp = pl["component"]
db.add(
BizProductionMaterialPrepLine(
id=generate_uuid(),
prep_id=prep_id,
component_material_id=comp.id,
component_material_type=pl["component_type"],
component_material_code=comp.material_code,
component_material_name=comp.material_name,
requisition_usage_id=pl["requisition"].id,
actual_qty=pl["actual_qty"],
usage_unit=pl["usage_unit"] or None,
tenant_no=tenant_no,
order_num=pl["idx"],
remark=pl["remark"],
create_by=user_id,
create_dept_id=dept_id or prep.create_dept_id or "",
create_time=now,
update_by=user_id,
update_time=now,
del_flag=0,
)
)
prep.update_by = user_id
prep.update_time = now
db.add(prep)
db.commit()
return get_prep_detail(db, prep_id, tenant_no)
def _write_prep_consume_usage(
db: Session,
*,
prep: BizProductionMaterialPrep,
line: BizProductionMaterialPrepLine,
req: BizProductionMaterialUsage,
tenant_no: str,
user_id: Optional[str],
dept_id: str,
operate_time: datetime,
) -> Decimal:
material = (
db.query(BizProductionMaterial)
.filter(
BizProductionMaterial.id == line.component_material_id,
BizProductionMaterial.tenant_no == tenant_no,
BizProductionMaterial.del_flag == 0,
)
.first()
)
if not material:
raise ValueError("子物料不存在")
actual_qty = line.actual_qty or Decimal("0")
stock_qty = material.stock_qty if material.stock_qty is not None else Decimal("0")
unit_price = req.unit_price
line_amount = Decimal("0")
if unit_price is not None and actual_qty > 0:
line_amount = (Decimal(str(unit_price)) * actual_qty).quantize(Decimal("0.01"))
usage = BizProductionMaterialUsage(
id=generate_uuid(),
material_id=material.id,
material_type=material.material_type,
material_code=material.material_code,
material_name=material.material_name,
batch_id=req.batch_id,
stock_in_no=req.stock_in_no,
catalog_no=req.catalog_no,
batch_no=req.batch_no,
manufacturer=req.manufacturer,
storage_location=req.storage_location,
unit_price=req.unit_price,
amount=line_amount if line_amount > 0 else req.amount,
change_type="use",
change_qty=-actual_qty,
usage_unit=line.usage_unit,
stock_before=stock_qty,
stock_after=stock_qty,
parent_usage_id=req.id,
prep_id=prep.id,
source_type="prep_consume",
tenant_no=tenant_no,
remark=f"配制单 {prep.prep_no}",
create_by=user_id,
create_dept_id=dept_id,
create_time=operate_time,
update_by=user_id,
update_time=operate_time,
del_flag=0,
)
db.add(usage)
return line_amount
def complete_prep(
db: Session,
prep_id: str,
*,
tenant_no: str,
user_id: Optional[str] = None,
dept_id: str = "",
) -> Dict[str, Any]:
prep = _get_prep_row(db, prep_id, tenant_no)
if (prep.status or "").strip() != "draft":
raise ValueError("仅草稿状态的配制单可确认配制")
lines = (
db.query(BizProductionMaterialPrepLine)
.filter(
BizProductionMaterialPrepLine.prep_id == prep_id,
BizProductionMaterialPrepLine.del_flag == 0,
)
.order_by(BizProductionMaterialPrepLine.order_num.asc())
.all()
)
if not lines:
raise ValueError("配制单无明细,无法确认")
configured = _get_material_for_update(db, prep.configured_material_id, tenant_no)
if (configured.material_type or "").strip() != "configured_reagent":
raise ValueError("配置试剂目录无效")
if not _is_enabled_status(configured.status):
raise ValueError(f"配置试剂「{configured.material_name}」已停用")
operate_time = datetime.now()
total_amount = Decimal("0")
try:
req_map: Dict[str, BizProductionMaterialUsage] = {}
for line in lines:
req_id = (line.requisition_usage_id or "").strip()
if not req_id:
raise ValueError("配制明细缺少领用记录")
if req_id not in req_map:
req = (
db.query(BizProductionMaterialUsage)
.filter(
BizProductionMaterialUsage.id == req_id,
BizProductionMaterialUsage.tenant_no == tenant_no,
BizProductionMaterialUsage.del_flag == 0,
BizProductionMaterialUsage.source_type == "manual_out",
)
.first()
)
if not req:
raise ValueError("领用记录不存在或已失效")
req_map[req_id] = req
amt = _write_prep_consume_usage(
db,
prep=prep,
line=line,
req=req_map[req_id],
tenant_no=tenant_no,
user_id=user_id,
dept_id=dept_id or prep.create_dept_id or "",
operate_time=operate_time,
)
total_amount += amt
stock_in_no = _generate_stock_in_no(db, tenant_no)
output_qty = prep.output_qty or Decimal("0")
if output_qty <= 0:
raise ValueError("产出数量必须大于0")
unit_price = Decimal("0")
amount = total_amount
if output_qty > 0 and total_amount > 0:
unit_price = (total_amount / output_qty).quantize(Decimal("0.01"))
amount = (unit_price * output_qty).quantize(Decimal("0.01"))
stock_in_date = prep.prep_time.date() if prep.prep_time else operate_time.date()
catalog_no = (configured.material_code or prep.batch_no or "—").strip() or "—"
batch = _create_batch_and_usage(
db,
material=configured,
stock_in_no=stock_in_no,
stock_in_date=stock_in_date,
expire_date=prep.expire_date,
catalog_no=catalog_no,
batch_no=prep.batch_no,
manufacturer="自配",
storage_location=prep.storage_location,
unit_price=unit_price,
amount=amount,
qty=output_qty,
remark=f"配制单 {prep.prep_no}",
tenant_no=tenant_no,
user_id=user_id,
operate_time=operate_time,
commit=False,
source_type="prep_in",
prep_id=prep.id,
)
prep.status = "completed"
prep.stock_in_no = stock_in_no
prep.output_batch_id = batch.id
prep.update_by = user_id
prep.update_time = operate_time
db.add(prep)
db.commit()
except Exception:
db.rollback()
raise
return get_prep_detail(db, prep_id, tenant_no)
def cancel_prep(
db: Session,
prep_id: str,
*,
tenant_no: str,
user_id: Optional[str] = None,
) -> None:
prep = _get_prep_row(db, prep_id, tenant_no)
_require_draft(prep)
now = datetime.now()
prep.status = "cancelled"
prep.update_by = user_id
prep.update_time = now
db.add(prep)
db.commit()
def delete_prep(
db: Session,
prep_id: str,
*,
tenant_no: str,
user_id: Optional[str] = None,
) -> None:
prep = _get_prep_row(db, prep_id, tenant_no)
_require_draft(prep)
now = datetime.now()
prep.del_flag = 1
prep.update_by = user_id
prep.update_time = now
db.add(prep)
lines = (
db.query(BizProductionMaterialPrepLine)
.filter(
BizProductionMaterialPrepLine.prep_id == prep_id,
BizProductionMaterialPrepLine.del_flag == 0,
)
.all()
)
for ln in lines:
ln.del_flag = 1
ln.update_by = user_id
ln.update_time = now
db.add(ln)
db.commit()
def list_available_material_batches(
db: Session,
tenant_no: str,
*,
page: int = 1,
page_size: int = 50,
keyword: Optional[str] = None,
material_type: Optional[str] = None,
user_info=None,
) -> Tuple[List[Dict[str, Any]], int]:
"""查询可用于配制的领用记录(有剩余、未标记已用完)。"""
return list_available_requisitions_for_prep(
db,
tenant_no,
page=page,
page_size=page_size,
keyword=keyword,
material_type=material_type,
user_info=user_info,
)