shanxia/bx/bx-backend/app/routers/reimburse.py

from typing import List, Optional
from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy import or_
from sqlalchemy.orm import Session

from app.database import get_db
from app.core.deps import get_current_user
from app.models.user import User
from app.models.reimburse import ReimburseApply
from app.schemas.common import ApiResponse, PagedData
from app.schemas.reimburse import (
    ReimburseApplyCreate, ReimburseApplyUpdate, ReimburseApplyOut,
    ReimburseApprovalCreate, ReimburseApprovalOut,
    ReimburseMyStatsOut, ReimburseApproverPendingStatsOut,
)
from app.services.reimburse import ReimburseService
from app.services.expense_item import ExpenseItemService
from app.models.workflow import WorkflowDef
from app.utils.codegen import generate_apply_code, generate_title
from app.utils.date_range import parse_create_time_day_bounds

router = APIRouter(prefix="/api/reimburses", tags=["报销管理"])


def _build_out_with_details(reimburse: ReimburseApply, db: Session) -> ReimburseApplyOut:
    return ReimburseService.build_apply_out_with_details(db, reimburse)


@router.get("/generate-title", response_model=ApiResponse[str])
def get_generated_title(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
    title = generate_title(db, current_user.real_name or current_user.username)
    return ApiResponse(data=title)


@router.get("/", response_model=ApiResponse[PagedData[ReimburseApplyOut]])
def list_reimburses(
    page: int = 1,
    pageSize: int = 10,
    status: str | None = None,
    keyword: str = "",
    expense_id: int | None = None,
    applicantId: Optional[int] = None,
    applyDate: Optional[str] = None,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    filters = [ReimburseApply.del_flag == 0]
    if status:
        filters.append(ReimburseApply.status == status)
    if keyword:
        filters.append(or_(ReimburseApply.apply_code.contains(keyword), ReimburseApply.applicant_name.contains(keyword)))
    if expense_id:
        filters.append(ReimburseApply.expense_id == expense_id)
    if applicantId is not None:
        filters.append(ReimburseApply.applicant_id == applicantId)
    bounds = parse_create_time_day_bounds(applyDate)
    if bounds:
        d0, d1 = bounds
        filters.append(ReimburseApply.create_time >= d0)
        filters.append(ReimburseApply.create_time < d1)
    rows, total = ReimburseService.get_paged(db, page=page, page_size=pageSize, filters=filters)
    out_rows = ReimburseService.out_rows_with_attachment_counts(db, rows)
    paged_data = PagedData(rows=out_rows, total=total, page=page, page_size=pageSize)
    return ApiResponse(data=paged_data)


@router.get("/my", response_model=ApiResponse[PagedData[ReimburseApplyOut]])
def my_reimburses(
    page: int = 1,
    pageSize: int = 10,
    status: str | None = None,
    keyword: str = "",
    applyDate: Optional[str] = None,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    filters = [ReimburseApply.del_flag == 0, ReimburseApply.applicant_id == current_user.id]
    if status:
        filters.append(ReimburseApply.status == status)
    if keyword:
        filters.append(or_(ReimburseApply.apply_code.contains(keyword), ReimburseApply.applicant_name.contains(keyword)))
    bounds = parse_create_time_day_bounds(applyDate)
    if bounds:
        d0, d1 = bounds
        filters.append(ReimburseApply.create_time >= d0)
        filters.append(ReimburseApply.create_time < d1)
    rows, total = ReimburseService.get_paged(db, page=page, page_size=pageSize, filters=filters)
    out_rows = ReimburseService.out_rows_with_attachment_counts(db, rows)
    paged_data = PagedData(rows=out_rows, total=total, page=page, page_size=pageSize)
    return ApiResponse(data=paged_data)


@router.get("/my/stats", response_model=ApiResponse[ReimburseMyStatsOut])
def my_reimburse_stats(
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    data = ReimburseService.my_apply_stats(db, current_user.id)
    return ApiResponse(data=ReimburseMyStatsOut.model_validate(data))


@router.get("/stats/status-counts", response_model=ApiResponse[dict])
def reimburse_status_counts(
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    """全库报销单各状态数量（与列表页「全部单据」汇总一致，不受分页影响）。"""
    return ApiResponse(data=ReimburseService.global_status_counts(db))


@router.get("/pending-approval", response_model=ApiResponse[PagedData[ReimburseApplyOut]])
def pending_approval(
    page: int = 1,
    pageSize: int = 10,
    keyword: str = "",
    applicantId: Optional[int] = None,
    applyDate: Optional[str] = None,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    rows, total = ReimburseService.pending_approval(
        db,
        current_user,
        page=page,
        page_size=pageSize,
        keyword=keyword,
        applicant_id=applicantId,
        apply_date=applyDate,
    )
    out_rows = ReimburseService.out_rows_with_attachment_counts(db, rows)
    paged_data = PagedData(rows=out_rows, total=total, page=page, page_size=pageSize)
    return ApiResponse(data=paged_data)


@router.get("/pending-approval/stats", response_model=ApiResponse[ReimburseApproverPendingStatsOut])
def pending_approval_stats(
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    data = ReimburseService.pending_approval_stats(db, current_user)
    return ApiResponse(data=ReimburseApproverPendingStatsOut.model_validate(data))


@router.get("/approved-by-me", response_model=ApiResponse[PagedData[ReimburseApplyOut]])
def approved_by_me(
    page: int = 1,
    pageSize: int = 10,
    keyword: str = "",
    applicantId: Optional[int] = None,
    applyDate: Optional[str] = None,
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
):
    rows, total = ReimburseService.approved_by_me(
        db,
        current_user,
        page=page,
        page_size=pageSize,
        keyword=keyword,
        applicant_id=applicantId,
        apply_date=applyDate,
    )
    out_rows = ReimburseService.out_rows_with_attachment_counts(db, rows)
    paged_data = PagedData(rows=out_rows, total=total, page=page, page_size=pageSize)
    return ApiResponse(data=paged_data)


@router.get("/{reimburse_id}", response_model=ApiResponse[ReimburseApplyOut])
def get_reimburse(reimburse_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
    reimburse = ReimburseService.get_by_id(db, reimburse_id)
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    out = _build_out_with_details(reimburse, db)
    return ApiResponse(data=out)


@router.get("/{reimburse_id}/approvals", response_model=ApiResponse[List[ReimburseApprovalOut]])
def get_approvals(reimburse_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
    approvals = ReimburseService.get_approvals(db, reimburse_id)
    return ApiResponse(data=[ReimburseApprovalOut.model_validate(a) for a in approvals])


@router.post("/", response_model=ApiResponse[ReimburseApplyOut])
def create_reimburse(body: ReimburseApplyCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
    details_data = []
    for d in body.details:
        item = ExpenseItemService.get_by_id(db, d.expense_item_id)
        if not item:
            raise HTTPException(
                status_code=status.HTTP_400_BAD_REQUEST,
                detail=f"费用项目 {d.expense_item_id} 不存在",
            )
        details_data.append({
            "expense_id": d.expense_item_id,
            "expense_name": item.item_name,
            "amount": d.amount,
            "remark": d.remark,
        })

    data = {
        "apply_code": generate_apply_code(db, "BX"),
        "status": "draft",
        "remark": body.remark,
    }
    if body.title:
        data["title"] = body.title
    wf = db.query(WorkflowDef).filter(WorkflowDef.biz_type == "reimburse", WorkflowDef.del_flag == 0).first()
    if wf:
        data["wf_id"] = wf.id

    reimburse = ReimburseService.create_with_details(db, data, details_data, current_user)
    out = _build_out_with_details(reimburse, db)
    return ApiResponse(data=out)


@router.put("/{reimburse_id}", response_model=ApiResponse[ReimburseApplyOut])
def update_reimburse(reimburse_id: int, body: ReimburseApplyUpdate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
    details_data = None
    if body.details is not None:
        details_data = []
        for d in body.details:
            item = ExpenseItemService.get_by_id(db, d.expense_item_id)
            if not item:
                raise HTTPException(
                    status_code=status.HTTP_400_BAD_REQUEST,
                    detail=f"费用项目 {d.expense_item_id} 不存在",
                )
            details_data.append({
                "expense_id": d.expense_item_id,
                "expense_name": item.item_name,
                "amount": d.amount,
                "remark": d.remark,
            })

    main_data = {}
    if body.remark is not None:
        main_data["remark"] = body.remark

    reimburse = ReimburseService.update_with_details(db, reimburse_id, main_data, details_data, current_user)
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    out = _build_out_with_details(reimburse, db)
    return ApiResponse(data=out)


@router.post("/{reimburse_id}/submit", response_model=ApiResponse[ReimburseApplyOut])
def submit_reimburse(reimburse_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
    reimburse = ReimburseService.submit(db, reimburse_id, current_user)
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    return ApiResponse(data=ReimburseService.out_one_with_attachment_counts(db, reimburse))


@router.post("/{reimburse_id}/approve", response_model=ApiResponse[ReimburseApplyOut])
def approve_reimburse(reimburse_id: int, body: ReimburseApprovalCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
    reimburse = ReimburseService.approve(db, reimburse_id, current_user, body.approval_opinion or "")
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    return ApiResponse(data=ReimburseService.out_one_with_attachment_counts(db, reimburse))


@router.post("/{reimburse_id}/reject", response_model=ApiResponse[ReimburseApplyOut])
def reject_reimburse(reimburse_id: int, body: ReimburseApprovalCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
    reimburse = ReimburseService.reject(db, reimburse_id, current_user, body.approval_opinion or "")
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    return ApiResponse(data=ReimburseService.out_one_with_attachment_counts(db, reimburse))


@router.delete("/{reimburse_id}", response_model=ApiResponse[None])
def delete_reimburse(reimburse_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
    ReimburseService.soft_delete(db, reimburse_id)
    return ApiResponse(data=None)