shanxia2.0/backend/app/api/bx/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.core.bx_auth import BxUserContext, get_bx_current_user
from app.core.database import get_db
from app.core.tenant_context import get_tenant_no
from app.models.bx_models import BxReimburseApply, BxWorkflowDef
from app.schemas.bx.common import ApiResponse, PagedData
from app.schemas.bx.reimburse import (
    ReimburseApplyCreate,
    ReimburseApplyOut,
    ReimburseApplyUpdate,
    ReimburseApprovalCreate,
    ReimburseApprovalOut,
    ReimburseApproverPendingStatsOut,
    ReimburseMyStatsOut,
)
from app.services.bx.expense_item import ExpenseItemService
from app.services.bx.reimburse import ReimburseService
from app.utils.bx_codegen import generate_apply_code, generate_title
from app.utils.bx_date_range import append_create_time_filters

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


def _tenant(db: Session, user: BxUserContext) -> str:
    return get_tenant_no(db, user.to_tenant_user_dict())


def _build_details_data(db: Session, tenant_no: str, body_details) -> List[dict]:
    details_data: List[dict] = []
    for d in body_details:
        item = ExpenseItemService.get_by_id(db, d.expense_item_id, tenant_no, user=user)
        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,
            }
        )
    return details_data


@router.get("/generate-title", response_model=ApiResponse[str])
def get_generated_title(
    reimburseDate: Optional[str] = None,
    excludeApplyId: Optional[str] = None,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    exclude_id = str(excludeApplyId).strip() if excludeApplyId else None
    title = generate_title(db, user.name, tenant_no, reimburseDate, exclude_id)
    return ApiResponse(data=title)


@router.get("/", response_model=ApiResponse[PagedData[ReimburseApplyOut]])
def list_reimburses(
    page: int = 1,
    pageSize: int = 10,
    status: Optional[str] = None,
    keyword: str = "",
    expense_id: Optional[str] = None,
    expenseName: Optional[str] = None,
    applicantId: Optional[str] = None,
    applyDate: Optional[str] = None,
    applyDateFrom: Optional[str] = None,
    applyDateTo: Optional[str] = None,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    filters = []
    if status:
        filters.append(BxReimburseApply.status == status)
    if keyword:
        filters.append(
            or_(
                BxReimburseApply.apply_code.contains(keyword),
                BxReimburseApply.applicant_name.contains(keyword),
                BxReimburseApply.title.contains(keyword),
            )
        )
    if expense_id:
        filters.append(BxReimburseApply.expense_id == str(expense_id).strip())
    expense_filter = ReimburseService.build_expense_name_condition(db, tenant_no, expenseName)
    if expense_filter is not None:
        filters.append(expense_filter)
    if applicantId:
        filters.append(BxReimburseApply.applicant_id == str(applicantId).strip())
    append_create_time_filters(
        filters,
        BxReimburseApply,
        apply_date=applyDate,
        apply_date_from=applyDateFrom,
        apply_date_to=applyDateTo,
    )
    rows, total = ReimburseService.get_paged(db, tenant_no, page=page, page_size=pageSize, filters=filters, user=user)
    out_rows = ReimburseService.out_rows_with_attachment_counts(db, tenant_no, list(rows))
    return ApiResponse(data=PagedData(rows=out_rows, total=total, page=page, page_size=pageSize))


@router.get("/my", response_model=ApiResponse[PagedData[ReimburseApplyOut]])
def my_reimburses(
    page: int = 1,
    pageSize: int = 10,
    status: Optional[str] = None,
    keyword: str = "",
    expenseName: Optional[str] = None,
    applyDate: Optional[str] = None,
    applyDateFrom: Optional[str] = None,
    applyDateTo: Optional[str] = None,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    filters = [
        or_(
            BxReimburseApply.applicant_id == user.id,
            BxReimburseApply.create_by == user.id,
        )
    ]
    if status:
        filters.append(BxReimburseApply.status == status)
    if keyword:
        filters.append(
            or_(
                BxReimburseApply.apply_code.contains(keyword),
                BxReimburseApply.applicant_name.contains(keyword),
                BxReimburseApply.title.contains(keyword),
            )
        )
    expense_filter = ReimburseService.build_expense_name_condition(db, tenant_no, expenseName)
    if expense_filter is not None:
        filters.append(expense_filter)
    append_create_time_filters(
        filters,
        BxReimburseApply,
        apply_date=applyDate,
        apply_date_from=applyDateFrom,
        apply_date_to=applyDateTo,
    )
    rows, total = ReimburseService.get_paged(db, tenant_no, page=page, page_size=pageSize, filters=filters, user=user)
    out_rows = ReimburseService.out_rows_with_attachment_counts(db, tenant_no, list(rows))
    return ApiResponse(data=PagedData(rows=out_rows, total=total, page=page, page_size=pageSize))


@router.get("/my/stats", response_model=ApiResponse[ReimburseMyStatsOut])
def my_reimburse_stats(
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    data = ReimburseService.my_apply_stats(db, user, tenant_no)
    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),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    return ApiResponse(data=ReimburseService.global_status_counts(db, tenant_no, user=user))


@router.get("/pending-approval", response_model=ApiResponse[PagedData[ReimburseApplyOut]])
def pending_approval(
    page: int = 1,
    pageSize: int = 10,
    keyword: str = "",
    applicantId: Optional[str] = None,
    expenseName: Optional[str] = None,
    applyDate: Optional[str] = None,
    applyDateFrom: Optional[str] = None,
    applyDateTo: Optional[str] = None,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    applicant_id = str(applicantId).strip() if applicantId else None
    rows, total = ReimburseService.pending_approval(
        db,
        user,
        tenant_no,
        page=page,
        page_size=pageSize,
        keyword=keyword,
        applicant_id=applicant_id,
        expense_name=expenseName,
        apply_date=applyDate,
        apply_date_from=applyDateFrom,
        apply_date_to=applyDateTo,
    )
    out_rows = ReimburseService.out_rows_with_attachment_counts(db, tenant_no, list(rows))
    return ApiResponse(data=PagedData(rows=out_rows, total=total, page=page, page_size=pageSize))


@router.get("/pending-approval/stats", response_model=ApiResponse[ReimburseApproverPendingStatsOut])
def pending_approval_stats(
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    data = ReimburseService.pending_approval_stats(db, user, tenant_no)
    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[str] = None,
    expenseName: Optional[str] = None,
    applyDate: Optional[str] = None,
    applyDateFrom: Optional[str] = None,
    applyDateTo: Optional[str] = None,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    applicant_id = str(applicantId).strip() if applicantId else None
    rows, total = ReimburseService.approved_by_me(
        db,
        user,
        tenant_no,
        page=page,
        page_size=pageSize,
        keyword=keyword,
        applicant_id=applicant_id,
        expense_name=expenseName,
        apply_date=applyDate,
        apply_date_from=applyDateFrom,
        apply_date_to=applyDateTo,
    )
    out_rows = ReimburseService.out_rows_with_attachment_counts(db, tenant_no, list(rows))
    return ApiResponse(data=PagedData(rows=out_rows, total=total, page=page, page_size=pageSize))


@router.get("/{reimburse_id}", response_model=ApiResponse[ReimburseApplyOut])
def get_reimburse(
    reimburse_id: str,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    reimburse = ReimburseService.get_by_id(db, reimburse_id, tenant_no, user=user)
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    out = ReimburseService.build_apply_out_with_details(db, tenant_no, reimburse)
    return ApiResponse(data=out)


@router.get("/{reimburse_id}/approvals", response_model=ApiResponse[List[ReimburseApprovalOut]])
def get_approvals(
    reimburse_id: str,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    if not ReimburseService.get_by_id(db, reimburse_id, tenant_no, user=user):
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    approvals = ReimburseService.get_approvals(db, reimburse_id, tenant_no, user=user)
    return ApiResponse(data=[ReimburseApprovalOut.model_validate(a) for a in approvals])


@router.get("/{reimburse_id}/approval-history", response_model=ApiResponse[List[ReimburseApprovalOut]])
def get_approval_history(
    reimburse_id: str,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    if not ReimburseService.get_by_id(db, reimburse_id, tenant_no, user=user):
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    history = ReimburseService.get_approval_history(db, reimburse_id, tenant_no, user=user)
    return ApiResponse(data=[ReimburseApprovalOut.model_validate(a) for a in history])


@router.post("/", response_model=ApiResponse[ReimburseApplyOut])
def create_reimburse(
    body: ReimburseApplyCreate,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    details_data = _build_details_data(db, tenant_no, body.details)
    data = {
        "apply_code": generate_apply_code(db, "BX", tenant_no),
        "status": "draft",
        "remark": body.remark,
        "reimburse_date": body.reimburse_date,
    }
    if body.title:
        data["title"] = body.title
    wf = (
        db.query(BxWorkflowDef)
        .filter(
            BxWorkflowDef.biz_type == "reimburse",
            BxWorkflowDef.tenant_no == tenant_no,
            BxWorkflowDef.del_flag == 0,
        )
        .first()
    )
    if wf:
        data["wf_id"] = wf.id
    reimburse = ReimburseService.create_with_details(db, data, details_data, user=user)
    db.commit()
    db.refresh(reimburse)
    out = ReimburseService.build_apply_out_with_details(db, tenant_no, reimburse)
    return ApiResponse(data=out)


@router.put("/{reimburse_id}", response_model=ApiResponse[ReimburseApplyOut])
def update_reimburse(
    reimburse_id: str,
    body: ReimburseApplyUpdate,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    details_data = None
    if body.details is not None:
        details_data = _build_details_data(db, tenant_no, body.details)
    main_data = {}
    if body.remark is not None:
        main_data["remark"] = body.remark
    if body.reimburse_date is not None:
        main_data["reimburse_date"] = body.reimburse_date
    if body.title is not None:
        main_data["title"] = body.title
    reimburse = ReimburseService.update_with_details(
        db, reimburse_id, main_data, details_data, user=user
    )
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    db.commit()
    db.refresh(reimburse)
    out = ReimburseService.build_apply_out_with_details(db, tenant_no, reimburse)
    return ApiResponse(data=out)


@router.post("/{reimburse_id}/submit", response_model=ApiResponse[ReimburseApplyOut])
def submit_reimburse(
    reimburse_id: str,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    reimburse = ReimburseService.submit(db, reimburse_id, user)
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    db.commit()
    db.refresh(reimburse)
    return ApiResponse(data=ReimburseService.out_one_with_attachment_counts(db, tenant_no, reimburse))


@router.post("/{reimburse_id}/approve", response_model=ApiResponse[ReimburseApplyOut])
def approve_reimburse(
    reimburse_id: str,
    body: ReimburseApprovalCreate,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    reimburse = ReimburseService.approve(db, reimburse_id, user, body.approval_opinion or "")
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在或无权审批")
    db.commit()
    db.refresh(reimburse)
    return ApiResponse(data=ReimburseService.out_one_with_attachment_counts(db, tenant_no, reimburse))


@router.post("/{reimburse_id}/reject", response_model=ApiResponse[ReimburseApplyOut])
def reject_reimburse(
    reimburse_id: str,
    body: ReimburseApprovalCreate,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    reimburse = ReimburseService.reject(db, reimburse_id, user, body.approval_opinion or "")
    if not reimburse:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在或无权审批")
    db.commit()
    db.refresh(reimburse)
    return ApiResponse(data=ReimburseService.out_one_with_attachment_counts(db, tenant_no, reimburse))


@router.delete("/{reimburse_id}", response_model=ApiResponse[None])
def delete_reimburse(
    reimburse_id: str,
    db: Session = Depends(get_db),
    user: BxUserContext = Depends(get_bx_current_user),
):
    tenant_no = _tenant(db, user)
    if not ReimburseService.get_by_id(db, reimburse_id, tenant_no, user=user):
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="报销单不存在")
    ReimburseService.soft_delete(db, reimburse_id, user=user)
    db.commit()
    return ApiResponse(data=None)