311 lines
10 KiB
Python
311 lines
10 KiB
Python
"""运输计划 - 必备证书校验"""
|
||
from __future__ import annotations
|
||
|
||
from typing import Iterable, Optional, Sequence
|
||
|
||
from fastapi import HTTPException
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.models.pl_models import PlPersonnel
|
||
from app.schemas.pl.base_data import CertRow, VehiclePersonnelRow
|
||
from app.services.pl_cert_service import list_owner_certs
|
||
|
||
VEHICLE_TYPE_SELF = "self"
|
||
VEHICLE_TYPE_CARRIER = "carrier"
|
||
|
||
SELF_HEAD_REQUIRED = ("行驶证", "营运证", "保险单")
|
||
SELF_TRAILER_REQUIRED = ("行驶证", "营运证", "保险单", "罐检报告")
|
||
|
||
CARRIER_HEAD_REQUIRED = ("行驶证", "营运证")
|
||
CARRIER_TRAILER_REQUIRED = ("行驶证", "营运证")
|
||
|
||
PERSONNEL_DRIVER_REQUIRED = ("身份证", "驾驶证", "驾驶员从业资格证")
|
||
PERSONNEL_ESCORT_REQUIRED = ("身份证", "押运员从业资格证")
|
||
PERSONNEL_BOTH_REQUIRED = (
|
||
"身份证",
|
||
"驾驶证",
|
||
"驾驶员从业资格证",
|
||
"押运员从业资格证",
|
||
)
|
||
|
||
VEHICLE_PART_HEAD = "head"
|
||
VEHICLE_PART_TRAILER = "trailer"
|
||
|
||
|
||
def match_required_cert_name(actual_name: str | None, required_name: str) -> bool:
|
||
name = (actual_name or "").strip()
|
||
if not name:
|
||
return False
|
||
if required_name == "押运员从业资格证":
|
||
return name == "押运员从业资格证" or name == "押运员证"
|
||
return name == required_name
|
||
|
||
|
||
def _cert_row_complete(cert: CertRow | dict, required_name: str) -> bool:
|
||
if isinstance(cert, dict):
|
||
cert_name = cert.get("cert_name") or cert.get("certName")
|
||
expire_date = cert.get("expire_date") or cert.get("expireDate")
|
||
attachment_urls = cert.get("attachment_urls") or cert.get("attachmentUrls")
|
||
else:
|
||
cert_name = cert.cert_name
|
||
expire_date = cert.expire_date
|
||
attachment_urls = cert.attachment_urls
|
||
|
||
if not match_required_cert_name(cert_name, required_name):
|
||
return False
|
||
if not expire_date:
|
||
return False
|
||
if not (attachment_urls or "").strip():
|
||
return False
|
||
return True
|
||
|
||
|
||
def _find_cert_for_required(
|
||
cert_rows: Sequence[CertRow | dict],
|
||
required_name: str,
|
||
*,
|
||
vehicle_part: str | None = None,
|
||
) -> CertRow | dict | None:
|
||
for cert in cert_rows:
|
||
if isinstance(cert, dict):
|
||
name = cert.get("cert_name") or cert.get("certName")
|
||
part = cert.get("vehicle_part") or cert.get("vehiclePart")
|
||
else:
|
||
name = cert.cert_name
|
||
part = cert.vehicle_part
|
||
if vehicle_part is not None and (part or VEHICLE_PART_HEAD) != vehicle_part:
|
||
continue
|
||
if match_required_cert_name(name, required_name):
|
||
return cert
|
||
return None
|
||
|
||
|
||
def _missing_field_labels(cert: CertRow | dict | None, required_name: str) -> list[str]:
|
||
if cert is None:
|
||
return ["有效期", "附件"]
|
||
missing: list[str] = []
|
||
if isinstance(cert, dict):
|
||
expire_date = cert.get("expire_date") or cert.get("expireDate")
|
||
attachment_urls = cert.get("attachment_urls") or cert.get("attachmentUrls")
|
||
cert_name = cert.get("cert_name") or cert.get("certName")
|
||
else:
|
||
expire_date = cert.expire_date
|
||
attachment_urls = cert.attachment_urls
|
||
cert_name = cert.cert_name
|
||
|
||
if not match_required_cert_name(cert_name, required_name):
|
||
return ["有效期", "附件"]
|
||
if not expire_date:
|
||
missing.append("有效期")
|
||
if not (attachment_urls or "").strip():
|
||
missing.append("附件")
|
||
return missing
|
||
|
||
|
||
def collect_required_cert_errors(
|
||
cert_rows: Sequence[CertRow | dict],
|
||
required_names: Iterable[str],
|
||
*,
|
||
part_label: str = "",
|
||
) -> list[str]:
|
||
errors: list[str] = []
|
||
prefix = f"{part_label}" if part_label else ""
|
||
vehicle_part = None
|
||
if part_label in ("主车", "head"):
|
||
vehicle_part = VEHICLE_PART_HEAD
|
||
elif part_label in ("挂车", "trailer"):
|
||
vehicle_part = VEHICLE_PART_TRAILER
|
||
|
||
for required_name in required_names:
|
||
cert = _find_cert_for_required(
|
||
cert_rows, required_name, vehicle_part=vehicle_part
|
||
)
|
||
missing = _missing_field_labels(cert, required_name)
|
||
if missing:
|
||
label = f"{prefix}「{required_name}」" if prefix else f"「{required_name}」"
|
||
errors.append(f"{label}缺少：{'、'.join(missing)}")
|
||
return errors
|
||
|
||
|
||
def get_required_personnel_cert_names(personnel_type: str | None) -> tuple[str, ...]:
|
||
ptype = (personnel_type or "").strip()
|
||
if ptype == "escort":
|
||
return PERSONNEL_ESCORT_REQUIRED
|
||
if ptype == "driver":
|
||
return PERSONNEL_DRIVER_REQUIRED
|
||
if ptype == "both":
|
||
return PERSONNEL_BOTH_REQUIRED
|
||
return PERSONNEL_BOTH_REQUIRED
|
||
|
||
|
||
def validate_personnel_type_role_compat(personnel_type: str | None, role: str | None) -> None:
|
||
ptype = (personnel_type or "").strip()
|
||
r = (role or "").strip()
|
||
if not ptype or not r:
|
||
return
|
||
if ptype == "driver" and r == "escort":
|
||
raise HTTPException(status_code=400, detail="驾驶员类型人员不可关联为押运员")
|
||
if ptype == "escort" and r == "driver":
|
||
raise HTTPException(status_code=400, detail="押运员类型人员不可关联为驾驶员")
|
||
|
||
|
||
def validate_personnel_required_certs(
|
||
cert_rows: Sequence[CertRow | dict],
|
||
personnel_type: str | None,
|
||
*,
|
||
subject_label: str = "",
|
||
) -> None:
|
||
ptype = (personnel_type or "").strip()
|
||
if not ptype:
|
||
raise HTTPException(status_code=400, detail="人员类型不能为空")
|
||
required = get_required_personnel_cert_names(ptype)
|
||
errors = collect_required_cert_errors(cert_rows, required)
|
||
if errors:
|
||
prefix = f"人员{subject_label}" if subject_label else "人员"
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail=f"必备证书未齐全：{prefix}{'；'.join(errors)}",
|
||
)
|
||
|
||
|
||
def validate_self_vehicle_required_certs(
|
||
cert_rows: Sequence[CertRow | dict],
|
||
*,
|
||
trailer_no: str | None = None,
|
||
) -> None:
|
||
errors: list[str] = []
|
||
errors.extend(
|
||
collect_required_cert_errors(
|
||
cert_rows, SELF_HEAD_REQUIRED, part_label="主车"
|
||
)
|
||
)
|
||
if (trailer_no or "").strip():
|
||
errors.extend(
|
||
collect_required_cert_errors(
|
||
cert_rows, SELF_TRAILER_REQUIRED, part_label="挂车"
|
||
)
|
||
)
|
||
if errors:
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail=f"必备证书未齐全：{'；'.join(errors)}",
|
||
)
|
||
|
||
|
||
def validate_carrier_vehicle_required_certs(
|
||
cert_rows: Sequence[CertRow | dict],
|
||
*,
|
||
trailer_no: str | None = None,
|
||
) -> None:
|
||
errors: list[str] = []
|
||
errors.extend(
|
||
collect_required_cert_errors(
|
||
cert_rows, CARRIER_HEAD_REQUIRED, part_label="主车"
|
||
)
|
||
)
|
||
if (trailer_no or "").strip():
|
||
errors.extend(
|
||
collect_required_cert_errors(
|
||
cert_rows, CARRIER_TRAILER_REQUIRED, part_label="挂车"
|
||
)
|
||
)
|
||
if errors:
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail=f"必备证书未齐全：{'；'.join(errors)}",
|
||
)
|
||
|
||
|
||
def validate_vehicle_personnel_affiliation(
|
||
db: Session,
|
||
tenant_no: str,
|
||
vehicle_type: str | None,
|
||
personnel_rows: Sequence[VehiclePersonnelRow],
|
||
) -> None:
|
||
vtype = (vehicle_type or "").strip()
|
||
if vtype not in (VEHICLE_TYPE_SELF, VEHICLE_TYPE_CARRIER):
|
||
return
|
||
label = "山峡" if vtype == VEHICLE_TYPE_SELF else "非山峡"
|
||
for row in personnel_rows:
|
||
personnel_id = (row.personnel_id or "").strip()
|
||
if not personnel_id:
|
||
continue
|
||
personnel = (
|
||
db.query(PlPersonnel)
|
||
.filter(
|
||
PlPersonnel.id == personnel_id,
|
||
PlPersonnel.tenant_no == tenant_no,
|
||
PlPersonnel.del_flag == 0,
|
||
)
|
||
.first()
|
||
)
|
||
if not personnel:
|
||
continue
|
||
affiliation = (personnel.personnel_affiliation or VEHICLE_TYPE_SELF).strip()
|
||
if affiliation != vtype:
|
||
name = (row.personnel_name or personnel.personnel_name or personnel_id).strip()
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail=f"关联人员「{name}」归属与车辆类型不匹配，{label}车辆仅可关联{label}人员",
|
||
)
|
||
|
||
|
||
def validate_vehicle_personnel_cert_links(
|
||
db: Session,
|
||
tenant_no: str,
|
||
personnel_rows: Sequence[VehiclePersonnelRow],
|
||
) -> None:
|
||
for row in personnel_rows:
|
||
personnel_id = (row.personnel_id or "").strip()
|
||
if not personnel_id:
|
||
continue
|
||
personnel = (
|
||
db.query(PlPersonnel)
|
||
.filter(
|
||
PlPersonnel.id == personnel_id,
|
||
PlPersonnel.tenant_no == tenant_no,
|
||
PlPersonnel.del_flag == 0,
|
||
)
|
||
.first()
|
||
)
|
||
if not personnel:
|
||
continue
|
||
validate_personnel_type_role_compat(personnel.personnel_type, row.role)
|
||
cert_rows = list_owner_certs(db, tenant_no, "personnel", personnel_id)
|
||
role = (row.role or "").strip()
|
||
required = (
|
||
PERSONNEL_DRIVER_REQUIRED
|
||
if role == "driver"
|
||
else PERSONNEL_ESCORT_REQUIRED
|
||
)
|
||
errors = collect_required_cert_errors(cert_rows, required)
|
||
if errors:
|
||
name = (row.personnel_name or personnel.personnel_name or personnel_id).strip()
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail=(
|
||
f"关联人员「{name}」必备证书未齐全：{'；'.join(errors)}"
|
||
),
|
||
)
|
||
|
||
|
||
def validate_vehicle_save_certs(
|
||
db: Session,
|
||
tenant_no: str,
|
||
*,
|
||
vehicle_type: str | None,
|
||
trailer_no: str | None,
|
||
cert_rows: Sequence[CertRow | dict],
|
||
personnel_rows: Sequence[VehiclePersonnelRow] | None = None,
|
||
) -> None:
|
||
vtype = (vehicle_type or "").strip()
|
||
if vtype == VEHICLE_TYPE_SELF:
|
||
validate_self_vehicle_required_certs(cert_rows, trailer_no=trailer_no)
|
||
elif vtype == VEHICLE_TYPE_CARRIER:
|
||
validate_carrier_vehicle_required_certs(cert_rows, trailer_no=trailer_no)
|
||
if vtype in (VEHICLE_TYPE_SELF, VEHICLE_TYPE_CARRIER) and personnel_rows:
|
||
validate_vehicle_personnel_affiliation(
|
||
db, tenant_no, vtype, personnel_rows
|
||
)
|
||
validate_vehicle_personnel_cert_links(db, tenant_no, personnel_rows)
|