| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551 |
- """运输计划管理 - 基础数据 API(承运商、车辆、人员)"""
- from datetime import datetime
- from typing import Any, List, Optional
- from fastapi import APIRouter, Depends, HTTPException, Query, Request, status
- from pydantic import BaseModel, Field
- from sqlalchemy import case, func, or_
- from sqlalchemy.orm import Session
- from app.utils.excel_export import (
- EXPORT_MAX_ROWS,
- ExportColumn,
- ExportValidationError,
- export_rows_to_xlsx_response,
- filter_export_columns,
- )
- from app.core.database import get_db
- from app.core.db_auth import DbUserContext, get_db_current_user
- from app.core.delete_guard import assert_row_deletable_by_creator, user_info_from_context
- from app.core.tenant_context import get_tenant_no
- from app.core.utils import generate_uuid
- from app.models.pl_models import (
- PlCarrier,
- PlPersonnel,
- PlVehicle,
- PlVehicleDriver,
- PlVehiclePersonnel,
- )
- from app.schemas.pl import (
- CarrierAuditReject,
- CarrierCreate,
- CarrierListResponse,
- CarrierRead,
- CarrierUpdate,
- carrier_to_read,
- CertRead,
- DriverCreate,
- DriverListResponse,
- DriverRead,
- DriverUpdate,
- PersonnelCreate,
- PersonnelDetailRead,
- PersonnelListResponse,
- PersonnelRead,
- PersonnelUpdate,
- VehicleCreate,
- VehicleDetailRead,
- VehicleDriverCreate,
- VehicleListResponse,
- VehicleRead,
- VehicleUpdate,
- )
- from app.services.pl_cert_service import (
- certs_to_read,
- list_owner_certs,
- list_owner_certs_batch,
- nearest_expire_date,
- save_owner_certs,
- )
- from app.services.pl_cert_required_service import (
- validate_personnel_required_certs,
- validate_vehicle_save_certs,
- )
- from app.services.pl_carrier_service import carrier_eligible_for_transport, normalize_carrier_urls
- from app.services.pl_personnel_service import (
- load_vehicle_personnel_rows,
- load_vehicle_personnel_summaries,
- save_vehicle_personnel,
- )
- from app.services.pl_personnel_user_bind import assert_sys_user_bind_available
- from app.schemas.pl.paste_recognize import PasteCommitBody, PasteCommitResult
- from app.services.pl_paste_recognize_service import commit_paste_recognize
- router = APIRouter(prefix="", tags=["运输计划-基础数据"])
- def _merge_query(request: Request, **known) -> dict:
- out = {k: v for k, v in known.items() if v is not None and v != ""}
- for k, v in request.query_params.multi_items():
- if k in ("page", "pageSize"):
- continue
- if k not in out and v not in (None, ""):
- out[k] = v
- return out
- def _param_get(params: dict, *keys: str) -> Any:
- for key in keys:
- if key in params and params[key] not in (None, ""):
- return params[key]
- return None
- def _body_to_params(body: BaseModel) -> dict:
- data = body.model_dump(by_alias=True, exclude_none=True)
- for key in ("ids", "columns"):
- data.pop(key, None)
- extra = getattr(body, "model_extra", None) or {}
- for key, value in extra.items():
- if key not in data and value not in (None, ""):
- data[key] = value
- return data
- # =============================================================================
- # 承运商 API
- # =============================================================================
- @router.get("/carrier", response_model=CarrierListResponse)
- def list_carriers(
- request: Request,
- q: Optional[str] = Query(default=None, description="模糊搜索"),
- status: Optional[str] = Query(default=None, description="状态"),
- audit_status: Optional[str] = Query(default=None, alias="auditStatus", description="审核状态"),
- page: int = Query(default=1, ge=1),
- page_size: int = Query(default=20, ge=1, le=500, alias="pageSize"),
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> CarrierListResponse:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- params = _merge_query(request, q=q, status=status, auditStatus=audit_status)
- query = db.query(PlCarrier).filter(
- PlCarrier.tenant_no == tenant_no,
- PlCarrier.del_flag == 0,
- )
- audit = _param_get(params, "auditStatus", "audit_status")
- if audit:
- query = query.filter(PlCarrier.audit_status == audit)
- qv = _param_get(params, "q")
- if qv:
- query = query.filter(
- or_(
- func.lower(PlCarrier.carrier_name).contains(str(qv).lower()),
- func.lower(PlCarrier.contact_name).contains(str(qv).lower()),
- PlCarrier.contact_phone.contains(str(qv)),
- )
- )
- from app.services.pl.carrier_list_filters import build_carrier_column_filters
- column_filters = build_carrier_column_filters(
- carrier_name=_param_get(params, "carrierName", "carrier_name"),
- contact_name=_param_get(params, "contactName", "contact_name"),
- contact_phone=_param_get(params, "contactPhone", "contact_phone"),
- status=_param_get(params, "status"),
- status_in=_param_get(params, "statusIn", "status_in"),
- )
- if column_filters:
- query = query.filter(*column_filters)
- # 承运商为物流主数据,列表不做 create_by 数据范围过滤
- query = query.order_by(
- case((PlCarrier.order_num.is_(None), 1), else_=0).asc(),
- PlCarrier.create_time.desc(),
- )
- total = query.count()
- offset = (page - 1) * page_size
- rows = query.offset(offset).limit(page_size).all()
- return CarrierListResponse(
- total=total,
- list=[carrier_to_read(row) for row in rows],
- )
- CARRIER_EXPORT_COLUMNS = [
- ExportColumn(title="承运商名称", field="carrierName"),
- ExportColumn(title="联系人", field="contactName"),
- ExportColumn(title="联系电话", field="contactPhone"),
- ExportColumn(title="合作状态", field="statusLabel"),
- ExportColumn(title="资质提示", field="qualificationHint"),
- ExportColumn(title="协议有效期", field="transportAgreementExpireDate"),
- ExportColumn(title="备注", field="remark"),
- ]
- class CarrierExportBody(BaseModel):
- q: Optional[str] = None
- status: Optional[str] = None
- audit_status: Optional[str] = Field(default=None, alias="auditStatus")
- ids: Optional[List[str]] = None
- columns: Optional[List[str]] = None
- model_config = {"populate_by_name": True, "extra": "allow"}
- @router.post("/carrier/export")
- def export_carriers(
- body: CarrierExportBody,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ):
- """导出承运商列表(当前筛选全量,或 ids 已选)。"""
- try:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- params = _body_to_params(body)
- query = db.query(PlCarrier).filter(
- PlCarrier.tenant_no == tenant_no,
- PlCarrier.del_flag == 0,
- )
- audit = _param_get(params, "auditStatus", "audit_status")
- if audit:
- query = query.filter(PlCarrier.audit_status == audit)
- qv = _param_get(params, "q")
- if qv:
- query = query.filter(
- or_(
- func.lower(PlCarrier.carrier_name).contains(str(qv).lower()),
- func.lower(PlCarrier.contact_name).contains(str(qv).lower()),
- PlCarrier.contact_phone.contains(str(qv)),
- )
- )
- from app.services.pl.carrier_list_filters import build_carrier_column_filters
- column_filters = build_carrier_column_filters(
- carrier_name=_param_get(params, "carrierName", "carrier_name"),
- contact_name=_param_get(params, "contactName", "contact_name"),
- contact_phone=_param_get(params, "contactPhone", "contact_phone"),
- status=_param_get(params, "status"),
- status_in=_param_get(params, "statusIn", "status_in"),
- )
- if column_filters:
- query = query.filter(*column_filters)
- if body.ids:
- query = query.filter(PlCarrier.id.in_(body.ids))
- query = query.order_by(
- case((PlCarrier.order_num.is_(None), 1), else_=0).asc(),
- PlCarrier.create_time.desc(),
- )
- rows = query.limit(EXPORT_MAX_ROWS + 1).all()
- items = []
- for row in rows:
- data = carrier_to_read(row).model_dump(by_alias=True)
- data["statusLabel"] = "合作中" if str(data.get("status")) == "1" else "已终止"
- if data.get("transportAgreementExpired"):
- data["qualificationHint"] = "协议已过期"
- elif data.get("qualificationComplete") is False:
- data["qualificationHint"] = "资质不全"
- else:
- data["qualificationHint"] = "资质齐全"
- items.append(data)
- requested = None
- if body.columns:
- requested = []
- for key in body.columns:
- if key in (
- "qualificationAttachments",
- "qualificationAttachment",
- "photoUrls",
- ):
- continue
- if key == "status":
- requested.append("statusLabel")
- elif key == "qualificationHint":
- requested.append("qualificationHint")
- else:
- requested.append(key)
- cols = filter_export_columns(CARRIER_EXPORT_COLUMNS, requested)
- date_str = datetime.now().strftime("%Y%m%d")
- return export_rows_to_xlsx_response(
- db,
- tenant_no=tenant_no,
- sheet_title="承运商列表",
- columns=cols,
- rows=items,
- filename_stem=f"承运商列表_{date_str}",
- )
- except ExportValidationError as e:
- raise HTTPException(status_code=400, detail=str(e)) from e
- @router.post("/carrier", response_model=CarrierRead, status_code=status.HTTP_201_CREATED)
- def create_carrier(
- payload: CarrierCreate,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> CarrierRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- carrier = PlCarrier(
- id=generate_uuid(),
- tenant_no=tenant_no,
- carrier_name=payload.carrier_name,
- contact_name=payload.contact_name,
- contact_phone=payload.contact_phone,
- status=payload.status,
- business_license_urls=normalize_carrier_urls(payload.business_license_urls),
- road_transport_license_urls=normalize_carrier_urls(payload.road_transport_license_urls),
- transport_agreement_urls=normalize_carrier_urls(payload.transport_agreement_urls),
- transport_agreement_expire_date=payload.transport_agreement_expire_date,
- audit_status="pending",
- remark=payload.remark,
- create_by=user.id,
- create_dept_id=user.dept_id,
- )
- db.add(carrier)
- db.commit()
- db.refresh(carrier)
- return carrier_to_read(carrier)
- @router.get("/carrier/{id}", response_model=CarrierRead)
- def get_carrier(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> CarrierRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- carrier = _get_carrier(db, tenant_no, id, user)
- return carrier_to_read(carrier)
- @router.put("/carrier/{id}", response_model=CarrierRead)
- def update_carrier(
- id: str,
- payload: CarrierUpdate,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> CarrierRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- carrier = _get_carrier(db, tenant_no, id, user)
- payload_data = payload.model_dump(exclude_unset=True)
- for key in (
- "business_license_urls",
- "road_transport_license_urls",
- "transport_agreement_urls",
- ):
- if key in payload_data:
- payload_data[key] = normalize_carrier_urls(payload_data[key])
- for key, value in payload_data.items():
- setattr(carrier, key, value)
- carrier.update_by = user.id
- db.commit()
- db.refresh(carrier)
- return carrier_to_read(carrier)
- @router.delete("/carrier/{id}", status_code=status.HTTP_204_NO_CONTENT)
- def delete_carrier(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> None:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- carrier = _get_carrier(db, tenant_no, id, user)
- assert_row_deletable_by_creator(carrier, user_info_from_context(user))
- carrier.del_flag = 1
- carrier.update_by = user.id
- db.commit()
- @router.post("/carrier/{id}/approve", response_model=CarrierRead)
- def approve_carrier(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> CarrierRead:
- raise HTTPException(status_code=400, detail="承运商审核流程已停用")
- @router.post("/carrier/{id}/reject", response_model=CarrierRead)
- def reject_carrier(
- id: str,
- payload: CarrierAuditReject,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> CarrierRead:
- raise HTTPException(status_code=400, detail="承运商审核流程已停用")
- def _get_carrier(db: Session, tenant_no: str, carrier_id: str, user: DbUserContext) -> PlCarrier:
- carrier = (
- db.query(PlCarrier)
- .filter(
- PlCarrier.id == carrier_id,
- PlCarrier.tenant_no == tenant_no,
- PlCarrier.del_flag == 0,
- )
- .first()
- )
- if not carrier:
- raise HTTPException(status_code=404, detail="承运商不存在")
- return carrier
- # =============================================================================
- # 车辆 API
- # =============================================================================
- def _vehicle_to_read(
- vehicle: PlVehicle,
- summary: Optional[str] = None,
- default_driver_name: Optional[str] = None,
- *,
- cert_list: Optional[list[CertRead]] = None,
- ) -> VehicleRead:
- data = VehicleRead.model_validate(vehicle)
- if summary is not None:
- data.personnel_summary = summary
- if default_driver_name is not None:
- data.default_driver_name = default_driver_name
- if cert_list is not None:
- data.cert_list = cert_list
- return data
- def _vehicle_to_detail(
- db: Session, tenant_no: str, vehicle: PlVehicle
- ) -> VehicleDetailRead:
- summaries = load_vehicle_personnel_summaries(db, tenant_no, [vehicle.id])
- summary, default_name = summaries.get(vehicle.id, ("", None))
- personnel_list = load_vehicle_personnel_rows(db, tenant_no, vehicle.id)
- cert_list = certs_to_read(
- list_owner_certs(db, tenant_no, "vehicle", vehicle.id)
- )
- base = _vehicle_to_read(vehicle, summary, default_name)
- return VehicleDetailRead(
- **base.model_dump(by_alias=False, exclude={"cert_list"}),
- personnel_list=personnel_list,
- cert_list=cert_list,
- )
- @router.get("/vehicle", response_model=VehicleListResponse)
- def list_vehicles(
- request: Request,
- q: Optional[str] = Query(default=None, description="模糊搜索车牌号"),
- vehicle_type: Optional[str] = Query(default=None, alias="vehicleType", description="车辆类型"),
- carrier_id: Optional[str] = Query(default=None, alias="carrierId", description="承运商ID"),
- vehicle_status: Optional[str] = Query(default=None, alias="vehicleStatus", description="车辆状态"),
- exclude_ineligible_carrier: Optional[bool] = Query(
- default=False,
- alias="excludeIneligibleCarrier",
- description="排除资质不全或协议过期的承运商车辆",
- ),
- page: int = Query(default=1, ge=1),
- page_size: int = Query(default=20, ge=1, le=500, alias="pageSize"),
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> VehicleListResponse:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- params = _merge_query(
- request,
- q=q,
- vehicleType=vehicle_type,
- carrierId=carrier_id,
- vehicleStatus=vehicle_status,
- )
- query = db.query(PlVehicle).filter(
- PlVehicle.tenant_no == tenant_no,
- PlVehicle.del_flag == 0,
- )
- vt = _param_get(params, "vehicleType", "vehicle_type")
- if vt:
- query = query.filter(PlVehicle.vehicle_type == vt)
- cid = _param_get(params, "carrierId", "carrier_id")
- if cid:
- query = query.filter(PlVehicle.carrier_id == cid)
- qv = _param_get(params, "q")
- if qv:
- query = query.filter(PlVehicle.license_plate.contains(str(qv)))
- from app.services.pl.vehicle_list_filters import build_vehicle_column_filters
- column_filters = build_vehicle_column_filters(
- license_plate=_param_get(params, "licensePlate", "license_plate"),
- vehicle_status=_param_get(params, "vehicleStatus", "vehicle_status"),
- vehicle_status_in=_param_get(params, "vehicleStatusIn", "vehicle_status_in"),
- )
- if column_filters:
- query = query.filter(*column_filters)
- # 车辆为主数据,列表不做 create_by 数据范围过滤
- query = query.order_by(
- case((PlVehicle.order_num.is_(None), 1), else_=0).asc(),
- PlVehicle.create_time.desc(),
- )
- if exclude_ineligible_carrier:
- vehicles = query.all()
- carrier_ids = {
- v.carrier_id for v in vehicles if v.vehicle_type == "carrier" and v.carrier_id
- }
- carrier_map: dict[str, PlCarrier] = {}
- if carrier_ids:
- carriers = (
- db.query(PlCarrier)
- .filter(
- PlCarrier.id.in_(carrier_ids),
- PlCarrier.tenant_no == tenant_no,
- PlCarrier.del_flag == 0,
- )
- .all()
- )
- carrier_map = {c.id: c for c in carriers}
- vehicles = [
- v
- for v in vehicles
- if v.vehicle_type != "carrier"
- or not v.carrier_id
- or (
- v.carrier_id in carrier_map
- and carrier_eligible_for_transport(carrier_map[v.carrier_id])
- )
- ]
- summaries = load_vehicle_personnel_summaries(
- db, tenant_no, [v.id for v in vehicles]
- )
- certs_by_owner = list_owner_certs_batch(
- db, tenant_no, "vehicle", [v.id for v in vehicles]
- )
- items = [
- _vehicle_to_read(
- v,
- *summaries.get(v.id, ("", None)),
- cert_list=certs_to_read(certs_by_owner.get(v.id, [])),
- )
- for v in vehicles
- ]
- return VehicleListResponse(total=len(items), list=items)
- total = query.count()
- offset = (page - 1) * page_size
- vehicles = query.offset(offset).limit(page_size).all()
- summaries = load_vehicle_personnel_summaries(
- db, tenant_no, [v.id for v in vehicles]
- )
- certs_by_owner = list_owner_certs_batch(
- db, tenant_no, "vehicle", [v.id for v in vehicles]
- )
- return VehicleListResponse(
- total=total,
- list=[
- _vehicle_to_read(
- v,
- *summaries.get(v.id, ("", None)),
- cert_list=certs_to_read(certs_by_owner.get(v.id, [])),
- )
- for v in vehicles
- ],
- )
- VEHICLE_EXPORT_COLUMNS = [
- ExportColumn(title="车牌号", field="licensePlate"),
- ExportColumn(title="关联人员", field="personnelSummary"),
- ExportColumn(title="车辆类型", field="vehicleTypeLabel"),
- ExportColumn(title="车辆状态", field="vehicleStatusLabel"),
- ExportColumn(title="载重(吨)", field="loadCapacity"),
- ExportColumn(title="承运商", field="carrierName"),
- ExportColumn(title="备注", field="remark"),
- ]
- _VEHICLE_TYPE_LABEL = {"self": "自有车", "carrier": "承运商车"}
- _VEHICLE_STATUS_LABEL = {"1": "启用", "0": "停用", "idle": "空闲", "busy": "在途", "maintain": "维保"}
- class VehicleExportBody(BaseModel):
- q: Optional[str] = None
- vehicle_type: Optional[str] = Field(default=None, alias="vehicleType")
- carrier_id: Optional[str] = Field(default=None, alias="carrierId")
- vehicle_status: Optional[str] = Field(default=None, alias="vehicleStatus")
- ids: Optional[List[str]] = None
- columns: Optional[List[str]] = None
- model_config = {"populate_by_name": True, "extra": "allow"}
- @router.post("/vehicle/export")
- def export_vehicles(
- body: VehicleExportBody,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ):
- """导出车辆列表(当前筛选全量,或 ids 已选)。"""
- try:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- params = _body_to_params(body)
- query = db.query(PlVehicle).filter(
- PlVehicle.tenant_no == tenant_no,
- PlVehicle.del_flag == 0,
- )
- vt = _param_get(params, "vehicleType", "vehicle_type")
- if vt:
- query = query.filter(PlVehicle.vehicle_type == vt)
- cid = _param_get(params, "carrierId", "carrier_id")
- if cid:
- query = query.filter(PlVehicle.carrier_id == cid)
- qv = _param_get(params, "q")
- if qv:
- query = query.filter(PlVehicle.license_plate.contains(str(qv)))
- from app.services.pl.vehicle_list_filters import build_vehicle_column_filters
- column_filters = build_vehicle_column_filters(
- license_plate=_param_get(params, "licensePlate", "license_plate"),
- vehicle_status=_param_get(params, "vehicleStatus", "vehicle_status"),
- vehicle_status_in=_param_get(params, "vehicleStatusIn", "vehicle_status_in"),
- )
- if column_filters:
- query = query.filter(*column_filters)
- if body.ids:
- query = query.filter(PlVehicle.id.in_(body.ids))
- query = query.order_by(
- case((PlVehicle.order_num.is_(None), 1), else_=0).asc(),
- PlVehicle.create_time.desc(),
- )
- vehicles = query.limit(EXPORT_MAX_ROWS + 1).all()
- summaries = load_vehicle_personnel_summaries(db, tenant_no, [v.id for v in vehicles])
- items = []
- for v in vehicles:
- data = _vehicle_to_read(v, *summaries.get(v.id, ("", None))).model_dump(by_alias=True)
- vt = str(data.get("vehicleType") or "")
- data["vehicleTypeLabel"] = _VEHICLE_TYPE_LABEL.get(vt, vt)
- vs = str(data.get("vehicleStatus") or "")
- data["vehicleStatusLabel"] = _VEHICLE_STATUS_LABEL.get(vs, vs)
- items.append(data)
- requested = None
- if body.columns:
- requested = []
- for key in body.columns:
- if key in ("photoUrls", "certAttachments", "certList"):
- continue
- if key == "vehicleType":
- requested.append("vehicleTypeLabel")
- elif key == "vehicleStatus":
- requested.append("vehicleStatusLabel")
- else:
- requested.append(key)
- cols = filter_export_columns(VEHICLE_EXPORT_COLUMNS, requested)
- date_str = datetime.now().strftime("%Y%m%d")
- return export_rows_to_xlsx_response(
- db,
- tenant_no=tenant_no,
- sheet_title="车辆列表",
- columns=cols,
- rows=items,
- filename_stem=f"车辆列表_{date_str}",
- )
- except ExportValidationError as e:
- raise HTTPException(status_code=400, detail=str(e)) from e
- @router.post("/vehicle", response_model=VehicleDetailRead, status_code=status.HTTP_201_CREATED)
- def create_vehicle(
- payload: VehicleCreate,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> VehicleDetailRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- existing = (
- db.query(PlVehicle)
- .filter(
- PlVehicle.license_plate == payload.license_plate,
- PlVehicle.del_flag == 0,
- )
- .first()
- )
- if existing:
- raise HTTPException(status_code=400, detail="车牌号已存在")
- validate_vehicle_save_certs(
- db,
- tenant_no,
- vehicle_type=payload.vehicle_type,
- trailer_no=payload.trailer_no,
- cert_rows=payload.cert_list,
- personnel_rows=payload.personnel_list,
- )
- vehicle = PlVehicle(
- id=generate_uuid(),
- tenant_no=tenant_no,
- vehicle_type=payload.vehicle_type,
- carrier_id=payload.carrier_id,
- carrier_name=payload.carrier_name,
- license_plate=payload.license_plate,
- trailer_no=payload.trailer_no,
- load_capacity=payload.load_capacity,
- vehicle_status=payload.vehicle_status,
- photo_urls=payload.photo_urls,
- remark=payload.remark,
- create_by=user.id,
- create_dept_id=user.dept_id,
- )
- db.add(vehicle)
- db.flush()
- try:
- save_vehicle_personnel(
- db,
- vehicle.id,
- payload.personnel_list,
- tenant_no,
- user_id=user.id,
- dept_id=user.dept_id,
- vehicle_type=payload.vehicle_type,
- )
- save_owner_certs(
- db,
- "vehicle",
- vehicle.id,
- payload.cert_list,
- tenant_no,
- user_id=user.id,
- dept_id=user.dept_id,
- )
- db.commit()
- except HTTPException:
- db.rollback()
- raise
- except Exception:
- db.rollback()
- raise
- db.refresh(vehicle)
- return _vehicle_to_detail(db, tenant_no, vehicle)
- @router.get("/vehicle/{id}", response_model=VehicleDetailRead)
- def get_vehicle(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> VehicleDetailRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- vehicle = _get_vehicle(db, tenant_no, id, user)
- return _vehicle_to_detail(db, tenant_no, vehicle)
- @router.put("/vehicle/{id}", response_model=VehicleDetailRead)
- def update_vehicle(
- id: str,
- payload: VehicleUpdate,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> VehicleDetailRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- vehicle = _get_vehicle(db, tenant_no, id, user)
- personnel_list = payload.personnel_list
- cert_list = payload.cert_list
- data = payload.model_dump(
- exclude_unset=True, exclude={"personnel_list", "cert_list"}
- )
- effective_vehicle_type = data.get("vehicle_type", vehicle.vehicle_type)
- effective_trailer_no = data.get("trailer_no", vehicle.trailer_no)
- effective_cert_rows = (
- cert_list
- if cert_list is not None
- else list_owner_certs(db, tenant_no, "vehicle", vehicle.id)
- )
- effective_personnel_rows = personnel_list
- if effective_personnel_rows is None:
- effective_personnel_rows = load_vehicle_personnel_rows(db, tenant_no, vehicle.id)
- validate_vehicle_save_certs(
- db,
- tenant_no,
- vehicle_type=effective_vehicle_type,
- trailer_no=effective_trailer_no,
- cert_rows=effective_cert_rows,
- personnel_rows=effective_personnel_rows,
- )
- for key, value in data.items():
- setattr(vehicle, key, value)
- vehicle.update_by = user.id
- try:
- if personnel_list is not None:
- save_vehicle_personnel(
- db,
- vehicle.id,
- personnel_list,
- tenant_no,
- user_id=user.id,
- dept_id=user.dept_id,
- vehicle_type=effective_vehicle_type,
- )
- if cert_list is not None:
- save_owner_certs(
- db,
- "vehicle",
- vehicle.id,
- cert_list,
- tenant_no,
- user_id=user.id,
- dept_id=user.dept_id,
- )
- db.commit()
- except HTTPException:
- db.rollback()
- raise
- except Exception:
- db.rollback()
- raise
- db.refresh(vehicle)
- return _vehicle_to_detail(db, tenant_no, vehicle)
- @router.delete("/vehicle/{id}", status_code=status.HTTP_204_NO_CONTENT)
- def delete_vehicle(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> None:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- vehicle = _get_vehicle(db, tenant_no, id, user)
- assert_row_deletable_by_creator(vehicle, user_info_from_context(user))
- vehicle.del_flag = 1
- vehicle.update_by = user.id
- db.commit()
- def _get_vehicle(db: Session, tenant_no: str, vehicle_id: str, user: DbUserContext) -> PlVehicle:
- vehicle = (
- db.query(PlVehicle)
- .filter(
- PlVehicle.id == vehicle_id,
- PlVehicle.tenant_no == tenant_no,
- PlVehicle.del_flag == 0,
- )
- .first()
- )
- if not vehicle:
- raise HTTPException(status_code=404, detail="车辆不存在")
- return vehicle
- # =============================================================================
- # 人员 API
- # =============================================================================
- def _personnel_to_read(
- db: Session,
- tenant_no: str,
- personnel: PlPersonnel,
- *,
- cert_list: Optional[list[CertRead]] = None,
- ) -> PersonnelRead:
- nearest = nearest_expire_date(db, tenant_no, "personnel", personnel.id)
- data = PersonnelRead.model_validate(personnel)
- data.nearest_expire_date = nearest
- if cert_list is not None:
- data.cert_list = cert_list
- return data
- def _assert_personnel_sys_user_bind(
- db: Session,
- tenant_no: str,
- *,
- personnel_id: str,
- sys_user_id: Optional[str],
- ) -> None:
- uid = (sys_user_id or "").strip()
- if not uid:
- return
- existing = (
- db.query(PlPersonnel)
- .filter(
- PlPersonnel.tenant_no == tenant_no,
- PlPersonnel.sys_user_id == uid,
- PlPersonnel.del_flag == 0,
- )
- .all()
- )
- try:
- assert_sys_user_bind_available(
- existing=existing,
- personnel_id=personnel_id,
- sys_user_id=uid,
- )
- except ValueError as exc:
- raise HTTPException(status_code=400, detail=str(exc)) from exc
- def _personnel_to_detail(
- db: Session, tenant_no: str, personnel: PlPersonnel
- ) -> PersonnelDetailRead:
- base = _personnel_to_read(db, tenant_no, personnel)
- cert_list = certs_to_read(
- list_owner_certs(db, tenant_no, "personnel", personnel.id)
- )
- return PersonnelDetailRead(
- **base.model_dump(by_alias=False, exclude={"cert_list"}),
- cert_list=cert_list,
- )
- @router.get("/personnel", response_model=PersonnelListResponse)
- def list_personnel(
- request: Request,
- q: Optional[str] = Query(default=None, description="模糊搜索姓名/手机号"),
- status: Optional[str] = Query(default=None, description="状态"),
- personnel_affiliation: Optional[str] = Query(
- default=None, alias="personnelAffiliation", description="人员归属 self/carrier"
- ),
- personnel_type: Optional[str] = Query(
- default=None, alias="personnelType", description="人员类型 driver/escort"
- ),
- page: int = Query(default=1, ge=1),
- page_size: int = Query(default=20, ge=1, le=500, alias="pageSize"),
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> PersonnelListResponse:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- params = _merge_query(
- request,
- q=q,
- status=status,
- personnelAffiliation=personnel_affiliation,
- personnelType=personnel_type,
- )
- query = db.query(PlPersonnel).filter(
- PlPersonnel.tenant_no == tenant_no,
- PlPersonnel.del_flag == 0,
- )
- affiliation = _param_get(params, "personnelAffiliation", "personnel_affiliation")
- if affiliation:
- query = query.filter(PlPersonnel.personnel_affiliation == affiliation)
- qv = _param_get(params, "q")
- if qv:
- query = query.filter(
- or_(
- func.lower(PlPersonnel.personnel_name).contains(str(qv).lower()),
- PlPersonnel.phone_no.contains(str(qv)),
- )
- )
- from app.services.pl.personnel_list_filters import build_personnel_column_filters
- column_filters = build_personnel_column_filters(
- personnel_name=_param_get(params, "personnelName", "personnel_name"),
- phone_no=_param_get(params, "phoneNo", "phone_no"),
- status=_param_get(params, "status"),
- status_in=_param_get(params, "statusIn", "status_in"),
- personnel_type=_param_get(params, "personnelType", "personnel_type"),
- personnel_type_in=_param_get(params, "personnelTypeIn", "personnel_type_in"),
- )
- if column_filters:
- query = query.filter(*column_filters)
- # 人员为主数据,列表不做 create_by 数据范围过滤
- query = query.order_by(
- case((PlPersonnel.order_num.is_(None), 1), else_=0).asc(),
- PlPersonnel.create_time.desc(),
- )
- total = query.count()
- offset = (page - 1) * page_size
- rows = query.offset(offset).limit(page_size).all()
- personnel_ids = [p.id for p in rows]
- certs_by_owner = list_owner_certs_batch(db, tenant_no, "personnel", personnel_ids)
- return PersonnelListResponse(
- total=total,
- list=[
- _personnel_to_read(
- db,
- tenant_no,
- p,
- cert_list=certs_to_read(certs_by_owner.get(p.id, [])),
- )
- for p in rows
- ],
- )
- PERSONNEL_EXPORT_COLUMNS = [
- ExportColumn(title="姓名", field="personnelName"),
- ExportColumn(title="人员归属", field="personnelAffiliationLabel"),
- ExportColumn(title="人员类型", field="personnelTypeLabel"),
- ExportColumn(title="手机号", field="phoneNo"),
- ExportColumn(title="年龄", field="age"),
- ExportColumn(title="身份证号", field="idCardNo"),
- ExportColumn(title="状态", field="statusLabel"),
- ExportColumn(title="最近到期", field="nearestExpireDate"),
- ExportColumn(title="备注", field="remark"),
- ]
- _AFFILIATION_LABEL = {"self": "自有", "carrier": "承运商"}
- _PERSONNEL_TYPE_LABEL = {"driver": "司机", "escort": "押运员"}
- class PersonnelExportBody(BaseModel):
- q: Optional[str] = None
- status: Optional[str] = None
- personnel_affiliation: Optional[str] = Field(default=None, alias="personnelAffiliation")
- personnel_type: Optional[str] = Field(default=None, alias="personnelType")
- ids: Optional[List[str]] = None
- columns: Optional[List[str]] = None
- model_config = {"populate_by_name": True, "extra": "allow"}
- @router.post("/personnel/export")
- def export_personnel(
- body: PersonnelExportBody,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ):
- """导出人员列表(当前筛选全量,或 ids 已选)。"""
- try:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- params = _body_to_params(body)
- query = db.query(PlPersonnel).filter(
- PlPersonnel.tenant_no == tenant_no,
- PlPersonnel.del_flag == 0,
- )
- affiliation = _param_get(params, "personnelAffiliation", "personnel_affiliation")
- if affiliation:
- query = query.filter(PlPersonnel.personnel_affiliation == affiliation)
- qv = _param_get(params, "q")
- if qv:
- query = query.filter(
- or_(
- func.lower(PlPersonnel.personnel_name).contains(str(qv).lower()),
- PlPersonnel.phone_no.contains(str(qv)),
- )
- )
- from app.services.pl.personnel_list_filters import build_personnel_column_filters
- column_filters = build_personnel_column_filters(
- personnel_name=_param_get(params, "personnelName", "personnel_name"),
- phone_no=_param_get(params, "phoneNo", "phone_no"),
- status=_param_get(params, "status"),
- status_in=_param_get(params, "statusIn", "status_in"),
- personnel_type=_param_get(params, "personnelType", "personnel_type"),
- personnel_type_in=_param_get(params, "personnelTypeIn", "personnel_type_in"),
- )
- if column_filters:
- query = query.filter(*column_filters)
- if body.ids:
- query = query.filter(PlPersonnel.id.in_(body.ids))
- query = query.order_by(
- case((PlPersonnel.order_num.is_(None), 1), else_=0).asc(),
- PlPersonnel.create_time.desc(),
- )
- rows = query.limit(EXPORT_MAX_ROWS + 1).all()
- items = []
- for p in rows:
- data = _personnel_to_read(db, tenant_no, p).model_dump(by_alias=True)
- af = str(data.get("personnelAffiliation") or "")
- data["personnelAffiliationLabel"] = _AFFILIATION_LABEL.get(af, af)
- pt = str(data.get("personnelType") or "")
- data["personnelTypeLabel"] = _PERSONNEL_TYPE_LABEL.get(pt, pt)
- st = str(data.get("status") or "")
- data["statusLabel"] = "启用" if st == "1" else ("停用" if st == "0" else st)
- items.append(data)
- requested = None
- if body.columns:
- requested = []
- for key in body.columns:
- if key in ("photoUrls", "certAttachments", "certList"):
- continue
- if key == "personnelAffiliation":
- requested.append("personnelAffiliationLabel")
- elif key == "personnelType":
- requested.append("personnelTypeLabel")
- elif key == "status":
- requested.append("statusLabel")
- else:
- requested.append(key)
- cols = filter_export_columns(PERSONNEL_EXPORT_COLUMNS, requested)
- date_str = datetime.now().strftime("%Y%m%d")
- return export_rows_to_xlsx_response(
- db,
- tenant_no=tenant_no,
- sheet_title="人员列表",
- columns=cols,
- rows=items,
- filename_stem=f"人员列表_{date_str}",
- )
- except ExportValidationError as e:
- raise HTTPException(status_code=400, detail=str(e)) from e
- @router.post("/personnel", response_model=PersonnelDetailRead, status_code=status.HTTP_201_CREATED)
- def create_personnel(
- payload: PersonnelCreate,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> PersonnelDetailRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- validate_personnel_required_certs(
- payload.cert_list,
- payload.personnel_type,
- personnel_affiliation=payload.personnel_affiliation,
- subject_label=f"「{payload.personnel_name}」",
- )
- personnel_id = generate_uuid()
- _assert_personnel_sys_user_bind(
- db,
- tenant_no,
- personnel_id=personnel_id,
- sys_user_id=payload.sys_user_id,
- )
- personnel = PlPersonnel(
- id=personnel_id,
- tenant_no=tenant_no,
- personnel_name=payload.personnel_name,
- phone_no=payload.phone_no,
- age=payload.age,
- id_card_no=payload.id_card_no,
- personnel_type=payload.personnel_type,
- personnel_affiliation=payload.personnel_affiliation,
- status=payload.status,
- photo_urls=payload.photo_urls,
- sys_user_id=(payload.sys_user_id or "").strip() or None,
- remark=payload.remark,
- create_by=user.id,
- create_dept_id=user.dept_id,
- )
- db.add(personnel)
- db.flush()
- try:
- save_owner_certs(
- db,
- "personnel",
- personnel.id,
- payload.cert_list,
- tenant_no,
- user_id=user.id,
- dept_id=user.dept_id,
- )
- db.commit()
- except Exception:
- db.rollback()
- raise
- db.refresh(personnel)
- return _personnel_to_detail(db, tenant_no, personnel)
- @router.get("/personnel/{id}", response_model=PersonnelDetailRead)
- def get_personnel(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> PersonnelDetailRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- personnel = _get_personnel(db, tenant_no, id, user)
- return _personnel_to_detail(db, tenant_no, personnel)
- @router.put("/personnel/{id}", response_model=PersonnelDetailRead)
- def update_personnel(
- id: str,
- payload: PersonnelUpdate,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> PersonnelDetailRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- personnel = _get_personnel(db, tenant_no, id, user)
- cert_list = payload.cert_list
- data = payload.model_dump(exclude_unset=True, exclude={"cert_list"})
- effective_personnel_type = data.get("personnel_type", personnel.personnel_type)
- effective_personnel_affiliation = data.get(
- "personnel_affiliation", personnel.personnel_affiliation
- )
- effective_cert_rows = (
- cert_list
- if cert_list is not None
- else list_owner_certs(db, tenant_no, "personnel", personnel.id)
- )
- subject_name = data.get("personnel_name", personnel.personnel_name)
- validate_personnel_required_certs(
- effective_cert_rows,
- effective_personnel_type,
- personnel_affiliation=effective_personnel_affiliation,
- subject_label=f"「{subject_name}」",
- )
- if "sys_user_id" in data:
- raw_uid = data.get("sys_user_id")
- data["sys_user_id"] = (raw_uid or "").strip() or None
- _assert_personnel_sys_user_bind(
- db,
- tenant_no,
- personnel_id=personnel.id,
- sys_user_id=data["sys_user_id"],
- )
- for key, value in data.items():
- setattr(personnel, key, value)
- personnel.update_by = user.id
- try:
- if cert_list is not None:
- save_owner_certs(
- db,
- "personnel",
- personnel.id,
- cert_list,
- tenant_no,
- user_id=user.id,
- dept_id=user.dept_id,
- )
- db.commit()
- except Exception:
- db.rollback()
- raise
- db.refresh(personnel)
- return _personnel_to_detail(db, tenant_no, personnel)
- @router.delete("/personnel/{id}", status_code=status.HTTP_204_NO_CONTENT)
- def delete_personnel(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> None:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- personnel = _get_personnel(db, tenant_no, id, user)
- assert_row_deletable_by_creator(personnel, user_info_from_context(user))
- personnel.del_flag = 1
- personnel.update_by = user.id
- db.commit()
- def _get_personnel(db: Session, tenant_no: str, personnel_id: str, user: DbUserContext) -> PlPersonnel:
- personnel = (
- db.query(PlPersonnel)
- .filter(
- PlPersonnel.id == personnel_id,
- PlPersonnel.tenant_no == tenant_no,
- PlPersonnel.del_flag == 0,
- )
- .first()
- )
- if not personnel:
- raise HTTPException(status_code=404, detail="人员不存在")
- return personnel
- # =============================================================================
- # 司机 API(兼容别名,内部使用 pl_personnel)
- # =============================================================================
- def _personnel_to_driver_read(personnel: PlPersonnel) -> DriverRead:
- return DriverRead(
- id=personnel.id,
- driver_name=personnel.personnel_name,
- phone_no=personnel.phone_no,
- id_card_no=personnel.id_card_no,
- tenant_no=personnel.tenant_no,
- order_num=personnel.order_num,
- remark=personnel.remark,
- create_by=personnel.create_by,
- create_dept_id=personnel.create_dept_id,
- create_time=personnel.create_time,
- update_by=personnel.update_by,
- update_time=personnel.update_time,
- del_flag=personnel.del_flag,
- )
- @router.get("/driver", response_model=DriverListResponse)
- def list_drivers(
- q: Optional[str] = Query(default=None, description="模糊搜索姓名/手机号"),
- page: int = Query(default=1, ge=1),
- page_size: int = Query(default=20, ge=1, le=500, alias="pageSize"),
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> DriverListResponse:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- query = db.query(PlPersonnel).filter(
- PlPersonnel.tenant_no == tenant_no,
- PlPersonnel.del_flag == 0,
- )
- if q:
- query = query.filter(
- or_(
- func.lower(PlPersonnel.personnel_name).contains(q.lower()),
- PlPersonnel.phone_no.contains(q),
- )
- )
- # 司机列表为主数据选项,不做 create_by 数据范围过滤
- query = query.order_by(
- case((PlPersonnel.order_num.is_(None), 1), else_=0).asc(),
- PlPersonnel.create_time.desc(),
- )
- total = query.count()
- offset = (page - 1) * page_size
- rows = query.offset(offset).limit(page_size).all()
- return DriverListResponse(
- total=total,
- list=[_personnel_to_driver_read(p) for p in rows],
- )
- DRIVER_EXPORT_COLUMNS = [
- ExportColumn(title="司机姓名", field="driverName"),
- ExportColumn(title="手机号", field="phoneNo"),
- ExportColumn(title="身份证号", field="idCardNo"),
- ExportColumn(title="备注", field="remark"),
- ]
- class DriverExportBody(BaseModel):
- q: Optional[str] = None
- ids: Optional[List[str]] = None
- columns: Optional[List[str]] = None
- @router.post("/driver/export")
- def export_drivers(
- body: DriverExportBody,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ):
- """导出司机列表(当前筛选全量,或 ids 已选)。"""
- try:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- query = db.query(PlPersonnel).filter(
- PlPersonnel.tenant_no == tenant_no,
- PlPersonnel.del_flag == 0,
- )
- if body.q:
- query = query.filter(
- or_(
- func.lower(PlPersonnel.personnel_name).contains(body.q.lower()),
- PlPersonnel.phone_no.contains(body.q),
- )
- )
- if body.ids:
- query = query.filter(PlPersonnel.id.in_(body.ids))
- query = query.order_by(
- case((PlPersonnel.order_num.is_(None), 1), else_=0).asc(),
- PlPersonnel.create_time.desc(),
- )
- rows = query.limit(EXPORT_MAX_ROWS + 1).all()
- items = [_personnel_to_driver_read(p).model_dump(by_alias=True) for p in rows]
- cols = filter_export_columns(DRIVER_EXPORT_COLUMNS, body.columns)
- date_str = datetime.now().strftime("%Y%m%d")
- return export_rows_to_xlsx_response(
- db,
- tenant_no=tenant_no,
- sheet_title="司机列表",
- columns=cols,
- rows=items,
- filename_stem=f"司机列表_{date_str}",
- )
- except ExportValidationError as e:
- raise HTTPException(status_code=400, detail=str(e)) from e
- @router.post("/driver", response_model=DriverRead, status_code=status.HTTP_201_CREATED)
- def create_driver(
- payload: DriverCreate,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> DriverRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- personnel = PlPersonnel(
- id=generate_uuid(),
- tenant_no=tenant_no,
- personnel_name=payload.driver_name,
- phone_no=payload.phone_no,
- id_card_no=payload.id_card_no,
- status="1",
- remark=payload.remark,
- create_by=user.id,
- create_dept_id=user.dept_id,
- )
- db.add(personnel)
- db.commit()
- db.refresh(personnel)
- return _personnel_to_driver_read(personnel)
- @router.get("/driver/{id}", response_model=DriverRead)
- def get_driver(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> DriverRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- personnel = _get_personnel(db, tenant_no, id, user)
- return _personnel_to_driver_read(personnel)
- @router.put("/driver/{id}", response_model=DriverRead)
- def update_driver(
- id: str,
- payload: DriverUpdate,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> DriverRead:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- personnel = _get_personnel(db, tenant_no, id, user)
- if payload.driver_name is not None:
- personnel.personnel_name = payload.driver_name
- if payload.phone_no is not None:
- personnel.phone_no = payload.phone_no
- if payload.id_card_no is not None:
- personnel.id_card_no = payload.id_card_no
- if payload.remark is not None:
- personnel.remark = payload.remark
- personnel.update_by = user.id
- db.commit()
- db.refresh(personnel)
- return _personnel_to_driver_read(personnel)
- @router.delete("/driver/{id}", status_code=status.HTTP_204_NO_CONTENT)
- def delete_driver(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> None:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- personnel = _get_personnel(db, tenant_no, id, user)
- assert_row_deletable_by_creator(personnel, user_info_from_context(user))
- personnel.del_flag = 1
- personnel.update_by = user.id
- db.commit()
- # =============================================================================
- # 车辆-司机绑定 API(兼容,写入 pl_vehicle_personnel)
- # =============================================================================
- @router.post("/vehicle-driver", status_code=status.HTTP_201_CREATED)
- def bind_vehicle_driver(
- payload: VehicleDriverCreate,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> dict:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- _get_vehicle(db, tenant_no, payload.vehicle_id, user)
- _get_personnel(db, tenant_no, payload.driver_id, user)
- if payload.is_default == 1:
- db.query(PlVehiclePersonnel).filter(
- PlVehiclePersonnel.vehicle_id == payload.vehicle_id,
- PlVehiclePersonnel.role == "driver",
- PlVehiclePersonnel.del_flag == 0,
- ).update({"is_default": 0})
- vd = PlVehiclePersonnel(
- id=generate_uuid(),
- tenant_no=tenant_no,
- vehicle_id=payload.vehicle_id,
- personnel_id=payload.driver_id,
- role="driver",
- is_default=payload.is_default,
- create_by=user.id,
- create_dept_id=user.dept_id,
- )
- db.add(vd)
- db.commit()
- return {"id": vd.id}
- @router.delete("/vehicle-driver/{id}", status_code=status.HTTP_204_NO_CONTENT)
- def unbind_vehicle_driver(
- id: str,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> None:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- vd = (
- db.query(PlVehiclePersonnel)
- .filter(
- PlVehiclePersonnel.id == id,
- PlVehiclePersonnel.tenant_no == tenant_no,
- PlVehiclePersonnel.del_flag == 0,
- )
- .first()
- )
- if not vd:
- vd = (
- db.query(PlVehicleDriver)
- .filter(
- PlVehicleDriver.id == id,
- PlVehicleDriver.tenant_no == tenant_no,
- PlVehicleDriver.del_flag == 0,
- )
- .first()
- )
- if vd:
- assert_row_deletable_by_creator(vd, user_info_from_context(user))
- vd.del_flag = 1
- db.commit()
- @router.post("/paste-recognize/commit", response_model=PasteCommitResult)
- def paste_recognize_commit(
- payload: PasteCommitBody,
- db: Session = Depends(get_db),
- user: DbUserContext = Depends(get_db_current_user),
- ) -> PasteCommitResult:
- tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
- try:
- result = commit_paste_recognize(
- db,
- tenant_no,
- payload,
- user_id=user.id,
- dept_id=user.dept_id,
- )
- db.commit()
- return result
- except HTTPException:
- db.rollback()
- raise
- except Exception:
- db.rollback()
- raise
|