Files
shanxia2.0/backend/app/api/pilot_delivery.py
T
2026-05-25 15:14:06 +08:00

235 lines
7.2 KiB
Python

"""中试服务交付管理 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))