shanxia2.0/backend/app/api/pl/site_plan.py

"""小程序/门户登录态 · 运输计划单（现场记录/回传）"""
from __future__ import annotations

from typing import Optional

from fastapi import APIRouter, Depends, File, Form, HTTPException, Query, UploadFile
from sqlalchemy.orm import Session

from app.core.database import get_db
from app.core.db_auth import DbUserContext, get_db_current_user
from app.core.response import error_response, success_response
from app.core.tenant_context import get_tenant_no
from app.schemas.pl.transport import PublicTransportSiteStopFeedback
from app.services import pl_site_plan_service as site_svc

router = APIRouter(prefix="/site-plan", tags=["运输计划-现场计划单"])


@router.get("/history")
def get_site_plan_history(
    date: str = Query(..., description="筛选日 YYYY-MM-DD"),
    q: Optional[str] = Query(None, description="车牌号/产品名称模糊"),
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
):
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    page, err = site_svc.build_site_history_page(
        db, tenant_no=tenant_no, date_str=date, keyword=q
    )
    if err:
        return error_response(400, err)
    return success_response(page.model_dump(by_alias=True))


@router.get("/calendar-counts")
def get_site_plan_calendar_counts(
    month: str = Query(..., description="月份 YYYY-MM"),
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
):
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    counts, err = site_svc.build_site_calendar_counts(
        db, tenant_no=tenant_no, month_str=month
    )
    if err:
        return error_response(400, err)
    return success_response({"month": month, "counts": counts or {}})


@router.get("/{plan_id}")
def get_site_plan_detail(
    plan_id: str,
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
):
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    page, err = site_svc.build_site_page_for_plan(
        db, tenant_no=tenant_no, plan_id=plan_id
    )
    if err or page is None:
        raise HTTPException(status_code=404, detail=err or "计划不存在或无现场链接")
    return success_response(page.model_dump(by_alias=True))


@router.post("/{plan_id}/feedback")
def save_site_plan_feedback(
    plan_id: str,
    payload: PublicTransportSiteStopFeedback,
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
):
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    ok, msg = site_svc.save_site_feedback_for_plan(
        db,
        tenant_no=tenant_no,
        plan_id=plan_id,
        stop_type=payload.stop_type,
        stop_id=payload.stop_id,
        quantity=payload.quantity,
        attachment_urls=payload.attachment_urls,
        remark=payload.remark,
    )
    if not ok:
        return error_response(400, msg or "保存失败")
    db.commit()
    return success_response({"success": True, "message": "保存成功"})


@router.post("/{plan_id}/attachments")
def upload_site_plan_attachment(
    plan_id: str,
    file: UploadFile = File(...),
    folder: Optional[str] = Form(""),
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
):
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    data, err = site_svc.upload_site_attachment_for_plan(
        db,
        tenant_no=tenant_no,
        plan_id=plan_id,
        file=file,
        folder=folder or "",
    )
    if err:
        code, msg = err
        return error_response(code, msg)
    return success_response(data)