zkxw/backend/app/api/pilot_delivery.py

"""中试服务交付管理 API（交付信息、交付物料、交付方式）"""
from typing import List, Optional

from fastapi import APIRouter, Depends, Query
from pydantic import BaseModel
from sqlalchemy.orm import Session

from app.core.database import get_db
from app.core.menu_guard import require_admin_or_super_admin
from app.core.response import error_response, success_response
from app.crud import pilot_delivery
from app.models.models import SysUser

router = APIRouter(prefix="/web/admin", tags=["中试服务交付管理"])


def _get_tenant_no(db: Session, user: dict) -> str:
    user_id = user.get("user_id")
    if not user_id:
        raise PermissionError("用户身份信息缺失，无法确定租户")
    sys_user = db.query(SysUser).filter(SysUser.id == user_id).first()
    if not sys_user or not sys_user.tenant_no:
        raise PermissionError("用户租户信息缺失")
    return sys_user.tenant_no


# ---------------------------------------------------------------------------
# Pydantic Schemas
# ---------------------------------------------------------------------------


class IdBody(BaseModel):
    id: str


class DeliverySaveBody(BaseModel):
    remark: Optional[str] = None


class DeliveryMaterialItem(BaseModel):
    material_name: str
    qty: Optional[float] = None
    unit_id: Optional[str] = None
    remark: Optional[str] = None


class DeliveryMaterialSaveBody(BaseModel):
    delivery_id: str
    items: List[DeliveryMaterialItem]


class DeliveryMethodSaveBody(BaseModel):
    delivery_id: str
    delivery_mode: str
    delivery_attachments: Optional[str] = None
    ship_time: Optional[str] = None
    handover_temp: Optional[str] = None
    pickup_person: Optional[str] = None
    pickup_phone: Optional[str] = None
    sender_name: Optional[str] = None
    express_company_id: Optional[str] = None
    waybill_no: Optional[str] = None
    express_fee: Optional[float] = None
    eta_time: Optional[str] = None
    remark: Optional[str] = None


class DeliveryCreateBody(BaseModel):
    infusion_schedule_id: str
    remark: Optional[str] = None


# ---------------------------------------------------------------------------
# 1. 交付信息 CRUD
# ---------------------------------------------------------------------------


@router.get("/pilot-delivery/detail")
async def get_by_schedule(
    infusion_schedule_id: str = Query(...),
    user: dict = Depends(require_admin_or_super_admin),
    db: Session = Depends(get_db),
):
    tenant_no = _get_tenant_no(db, user)
    result = pilot_delivery.get_delivery_by_schedule(
        db, schedule_id=infusion_schedule_id, tenant_no=tenant_no
    )
    if result is None:
        return success_response(None)
    return success_response(result)


@router.get("/pilot-delivery/list-by-application")
async def list_by_application(
    application_id: str = Query(...),
    user: dict = Depends(require_admin_or_super_admin),
    db: Session = Depends(get_db),
):
    tenant_no = _get_tenant_no(db, user)
    result = pilot_delivery.list_deliveries_by_application(
        db, application_id=application_id, tenant_no=tenant_no
    )
    return success_response(result)


@router.post("/pilot-delivery/create")
async def create_delivery(
    body: DeliveryCreateBody,
    user: dict = Depends(require_admin_or_super_admin),
    db: Session = Depends(get_db),
):
    try:
        tenant_no = _get_tenant_no(db, user)
        data = {"remark": body.remark}
        result = pilot_delivery.create_delivery(
            db,
            schedule_id=body.infusion_schedule_id,
            data=data,
            tenant_no=tenant_no,
            user_id=user.get("user_id"),
            dept_id=user.get("region_id", ""),
        )
        return success_response(result, message="创建成功")
    except ValueError as e:
        return error_response(400, str(e))


@router.delete("/pilot-delivery/delete")
async def soft_delete(
    body: IdBody,
    user: dict = Depends(require_admin_or_super_admin),
    db: Session = Depends(get_db),
):
    try:
        tenant_no = _get_tenant_no(db, user)
        pilot_delivery.soft_delete_delivery(
            db, delivery_id=body.id, tenant_no=tenant_no
        )
        return success_response(None, message="删除成功")
    except ValueError as e:
        return error_response(400, str(e))


# ---------------------------------------------------------------------------
# 2. 交付物料
# ---------------------------------------------------------------------------


@router.get("/pilot-delivery/materials")
async def list_materials(
    delivery_id: str = Query(...),
    user: dict = Depends(require_admin_or_super_admin),
    db: Session = Depends(get_db),
):
    tenant_no = _get_tenant_no(db, user)
    result = pilot_delivery.list_delivery_materials(
        db, delivery_id=delivery_id, tenant_no=tenant_no
    )
    return success_response(result)


@router.post("/pilot-delivery/materials/save")
async def save_materials(
    body: DeliveryMaterialSaveBody,
    user: dict = Depends(require_admin_or_super_admin),
    db: Session = Depends(get_db),
):
    try:
        tenant_no = _get_tenant_no(db, user)
        items_json = [item.dict() for item in body.items]
        result = pilot_delivery.save_delivery_materials(
            db,
            delivery_id=body.delivery_id,
            items_json=items_json,
            tenant_no=tenant_no,
            user_id=user.get("user_id"),
        )
        return success_response(result, message="保存成功")
    except ValueError as e:
        return error_response(400, str(e))


# ---------------------------------------------------------------------------
# 3. 交付方式
# ---------------------------------------------------------------------------


@router.get("/pilot-delivery/method")
async def get_method(
    delivery_id: str = Query(...),
    user: dict = Depends(require_admin_or_super_admin),
    db: Session = Depends(get_db),
):
    tenant_no = _get_tenant_no(db, user)
    result = pilot_delivery.get_delivery_method(
        db, delivery_id=delivery_id, tenant_no=tenant_no
    )
    if result is None:
        return success_response(None)
    return success_response(result)


@router.post("/pilot-delivery/method/save")
async def save_method(
    body: DeliveryMethodSaveBody,
    user: dict = Depends(require_admin_or_super_admin),
    db: Session = Depends(get_db),
):
    try:
        tenant_no = _get_tenant_no(db, user)
        data = {
            "delivery_mode": body.delivery_mode,
            "delivery_attachments": body.delivery_attachments,
            "ship_time": body.ship_time,
            "handover_temp": body.handover_temp,
            "pickup_person": body.pickup_person,
            "pickup_phone": body.pickup_phone,
            "sender_name": body.sender_name,
            "express_company_id": body.express_company_id,
            "waybill_no": body.waybill_no,
            "express_fee": body.express_fee,
            "eta_time": body.eta_time,
            "remark": body.remark,
        }
        result = pilot_delivery.save_delivery_method(
            db,
            delivery_id=body.delivery_id,
            data=data,
            tenant_no=tenant_no,
            user_id=user.get("user_id"),
        )
        return success_response(result, message="保存成功")
    except ValueError as e:
        return error_response(400, str(e))