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

"""运输计划管理 - 盘存统计 API"""
from datetime import date
from decimal import Decimal
from typing import Optional

from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy import func
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.services.pl_data_scope import apply_pl_scope, assert_pl_row_in_scope
from app.core.tenant_context import get_tenant_no
from app.core.utils import generate_uuid
from app.models.pl_models import (
    PlDocumentReminder,
    PlInventory,
    PlInventoryTransaction,
    PlInventoryWarning,
)
from app.schemas.pl.inventory import (
    DocumentReminderRead,
    DocumentReminderStatsRead,
    DocumentReminderUpdate,
    InventoryLedgerQuery,
    InventoryLedgerResponse,
    InventoryRead,
    InventorySummaryItem,
    InventorySummaryResponse,
    InventoryTransactionQuery,
    InventoryTransactionRead,
    InventoryTransactionResponse,
    InventoryWarningRead,
    InventoryWarningThreshold,
    InventoryWarningUpdate,
    TransportStatsItem,
    TransportStatsResponse,
)
from app.services.pl_cert_service import scan_cert_expiry

router = APIRouter(prefix="/inventory", tags=["运输计划-盘存统计"])


# =============================================================================
# 库存台账
# =============================================================================


@router.get("/ledger", response_model=InventoryLedgerResponse)
def get_inventory_ledger(
    product_name: Optional[str] = Query(default=None, alias="productName", description="产品名称"),
    page: int = Query(default=1, ge=1),
    page_size: int = Query(default=20, ge=1, le=100, alias="pageSize"),
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
) -> InventoryLedgerResponse:
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    query = db.query(PlInventory).filter(
        PlInventory.tenant_no == tenant_no,
        PlInventory.del_flag == 0,
    )

    if product_name:
        query = query.filter(PlInventory.product_name.contains(product_name))

    query = apply_pl_scope(query, PlInventory, user, db, tenant_no)

    total = query.count()
    offset = (page - 1) * page_size
    items = query.order_by(PlInventory.product_name).offset(offset).limit(page_size).all()

    return InventoryLedgerResponse(total=total, list=items)


# =============================================================================
# 库存预警阈值设置
# =============================================================================


@router.post("/warning-threshold")
def set_warning_threshold(
    payload: InventoryWarningThreshold,
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
) -> dict:
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())

    inventory = (
        db.query(PlInventory)
        .filter(
            PlInventory.product_id == payload.product_id,
            PlInventory.tenant_no == tenant_no,
            PlInventory.del_flag == 0,
        )
        .first()
    )

    if inventory:
        assert_pl_row_in_scope(db, inventory, PlInventory, user, tenant_no)
        inventory.low_stock_warning = payload.low_stock_warning
        inventory.high_stock_warning = payload.high_stock_warning
    else:
        # 创建库存记录
        inventory = PlInventory(
            id=generate_uuid(),
            tenant_no=tenant_no,
            product_id=payload.product_id,
            product_no="",
            product_name="",
            unit="吨",
            stock_qty=0,
            stock_amount=0,
            low_stock_warning=payload.low_stock_warning,
            high_stock_warning=payload.high_stock_warning,
            create_by=user.id,
            create_dept_id=user.dept_id,
        )
        db.add(inventory)

    db.commit()
    return {"success": True}


# =============================================================================
# 库存预警列表
# =============================================================================


@router.get("/warnings", response_model=list[InventoryWarningRead])
def list_warnings(
    status: Optional[str] = Query(default=None, description="状态"),
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
) -> list[InventoryWarningRead]:
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    query = db.query(PlInventoryWarning).filter(
        PlInventoryWarning.tenant_no == tenant_no,
        PlInventoryWarning.del_flag == 0,
    )

    if status:
        query = query.filter(PlInventoryWarning.status == status)

    query = apply_pl_scope(query, PlInventoryWarning, user, db, tenant_no)
    return query.order_by(PlInventoryWarning.create_time.desc()).all()


@router.put("/warnings/{id}", response_model=InventoryWarningRead)
def update_warning(
    id: str,
    payload: InventoryWarningUpdate,
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
) -> InventoryWarningRead:
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())

    warning = (
        db.query(PlInventoryWarning)
        .filter(
            PlInventoryWarning.id == id,
            PlInventoryWarning.tenant_no == tenant_no,
            PlInventoryWarning.del_flag == 0,
        )
        .first()
    )

    if not warning:
        raise HTTPException(status_code=404, detail="预警不存在")
    assert_pl_row_in_scope(db, warning, PlInventoryWarning, user, tenant_no)

    if payload.status == "resolved":
        warning.status = "resolved"
        from datetime import datetime

        warning.resolve_time = datetime.now()
        warning.resolve_remark = payload.resolve_remark

    db.commit()
    db.refresh(warning)
    return warning


# =============================================================================
# 出入库明细
# =============================================================================


@router.get("/transactions", response_model=InventoryTransactionResponse)
def list_transactions(
    product_id: Optional[str] = Query(default=None, alias="productId", description="产品ID"),
    transaction_type: Optional[str] = Query(default=None, alias="transactionType", description="事务类型"),
    date_from: Optional[date] = Query(default=None, alias="dateFrom", description="开始日期"),
    date_to: Optional[date] = Query(default=None, alias="dateTo", description="结束日期"),
    page: int = Query(default=1, ge=1),
    page_size: int = Query(default=20, ge=1, le=100, alias="pageSize"),
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
) -> InventoryTransactionResponse:
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    query = db.query(PlInventoryTransaction).filter(
        PlInventoryTransaction.tenant_no == tenant_no,
        PlInventoryTransaction.del_flag == 0,
    )

    if product_id:
        query = query.filter(PlInventoryTransaction.product_id == product_id)

    if transaction_type:
        query = query.filter(PlInventoryTransaction.transaction_type == transaction_type)

    if date_from:
        query = query.filter(PlInventoryTransaction.create_time >= date_from)

    if date_to:
        query = query.filter(PlInventoryTransaction.create_time <= date_to)

    query = apply_pl_scope(query, PlInventoryTransaction, user, db, tenant_no)

    total = query.count()
    offset = (page - 1) * page_size
    items = query.order_by(PlInventoryTransaction.create_time.desc()).offset(offset).limit(page_size).all()

    return InventoryTransactionResponse(total=total, list=items)


# =============================================================================
# 出入库汇总
# =============================================================================


@router.get("/summary", response_model=InventorySummaryResponse)
def get_summary(
    group_by: str = Query(default="month", description="汇总维度 month/week"),
    date_from: date = Query(..., alias="dateFrom", description="开始日期"),
    date_to: date = Query(..., alias="dateTo", description="结束日期"),
    product_id: Optional[str] = Query(default=None, alias="productId", description="产品ID"),
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
) -> InventorySummaryResponse:
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())

    query = db.query(PlInventoryTransaction).filter(
        PlInventoryTransaction.tenant_no == tenant_no,
        PlInventoryTransaction.del_flag == 0,
        PlInventoryTransaction.create_time >= date_from,
        PlInventoryTransaction.create_time <= date_to,
    )
    query = apply_pl_scope(query, PlInventoryTransaction, user, db, tenant_no)

    if product_id:
        query = query.filter(PlInventoryTransaction.product_id == product_id)

    transactions = query.all()

    # 按产品分组汇总
    summary_map = {}
    for t in transactions:
        key = t.product_id

        if key not in summary_map:
            summary_map[key] = {
                "product_id": t.product_id,
                "product_name": t.product_name,
                "inbound_qty": Decimal("0"),
                "outbound_qty": Decimal("0"),
            }

        if t.transaction_type == "inbound":
            summary_map[key]["inbound_qty"] += t.qty
        else:
            summary_map[key]["outbound_qty"] += abs(t.qty)

    items = []
    for key, data in summary_map.items():
        net = data["inbound_qty"] - data["outbound_qty"]
        items.append(InventorySummaryItem(
            period="汇总",
            product_id=data["product_id"],
            product_name=data["product_name"],
            inbound_qty=data["inbound_qty"],
            outbound_qty=data["outbound_qty"],
            net_change=net,
        ))

    return InventorySummaryResponse(list=items)


# =============================================================================
# 证件到期提醒
# =============================================================================


@router.get("/document-reminders/stats", response_model=DocumentReminderStatsRead)
def document_reminder_stats(
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
) -> DocumentReminderStatsRead:
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    scan_cert_expiry(db, tenant_no)
    db.commit()

    query = db.query(PlDocumentReminder).filter(
        PlDocumentReminder.tenant_no == tenant_no,
        PlDocumentReminder.del_flag == 0,
        PlDocumentReminder.status == "pending",
        PlDocumentReminder.days_remaining <= 0,
    )
    query = apply_pl_scope(query, PlDocumentReminder, user, db, tenant_no)
    expired_pending_count = query.count()
    return DocumentReminderStatsRead(expired_pending_count=expired_pending_count)


@router.get("/document-reminders", response_model=list[DocumentReminderRead])
def list_document_reminders(
    status: Optional[str] = Query(default=None, description="状态"),
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
) -> list[DocumentReminderRead]:
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())
    scan_cert_expiry(db, tenant_no)
    db.commit()

    query = db.query(PlDocumentReminder).filter(
        PlDocumentReminder.tenant_no == tenant_no,
        PlDocumentReminder.del_flag == 0,
    )

    if status:
        query = query.filter(PlDocumentReminder.status == status)

    query = apply_pl_scope(query, PlDocumentReminder, user, db, tenant_no)
    return query.order_by(PlDocumentReminder.expire_date.asc()).all()


@router.put("/document-reminders/{id}", response_model=DocumentReminderRead)
def update_document_reminder(
    id: str,
    payload: DocumentReminderUpdate,
    db: Session = Depends(get_db),
    user: DbUserContext = Depends(get_db_current_user),
) -> DocumentReminderRead:
    tenant_no = get_tenant_no(db, user.to_tenant_user_dict())

    reminder = (
        db.query(PlDocumentReminder)
        .filter(
            PlDocumentReminder.id == id,
            PlDocumentReminder.tenant_no == tenant_no,
            PlDocumentReminder.del_flag == 0,
        )
        .first()
    )

    if not reminder:
        raise HTTPException(status_code=404, detail="提醒不存在")
    assert_pl_row_in_scope(db, reminder, PlDocumentReminder, user, tenant_no)

    reminder.status = payload.status
    if payload.remark:
        reminder.remark = payload.remark

    db.commit()
    db.refresh(reminder)
    return reminder


# =============================================================================
# 生成证件到期提醒（定时任务调用）
# =============================================================================


@router.post("/check-document-expiry")
def check_document_expiry(db: Session = Depends(get_db)) -> dict:
    """检查证件到期，生成提醒（供定时任务调用）"""
    created_count = scan_cert_expiry(db)
    db.commit()
    return {"created_count": created_count}