track/backend/app/api/v1/endpoints/dashboard.py

"""Dashboard API — 上帝视角（全厂数据，无用户过滤）"""
import io
from datetime import datetime
from fastapi import APIRouter, Depends, Query
from fastapi.responses import StreamingResponse
from sqlalchemy.ext.asyncio import AsyncSession
from app.core.database import get_db
from app.services.dashboard_service import (
    get_dashboard_stats, DashboardStats,
    get_wip_tasks, WipTask,
    get_completed_tasks, CompletedTask,
    get_rejected_tasks, RejectedTask,
    get_people_workload, PersonWorkload,
    get_people_history, PersonHistoryRecord,
    search_product_messages, ProductMessageList,
)

router = APIRouter(prefix="/dashboard", tags=["管理看板"])


@router.get("/stats", response_model=DashboardStats)
async def dashboard_stats(
    since: str | None = Query(None, description="起始日期 ISO 如 2026-08-01T00:00:00"),
    until: str | None = Query(None, description="截止日期 ISO"),
    db: AsyncSession = Depends(get_db),
):
    """
    全局统计（上帝视角）。

    时间筛选仅影响 COMPLETED / REJECTED 计数；
    PENDING / WIP / 总数永远返回实时快照。
    """
    since_dt = datetime.fromisoformat(since) if since else None
    until_dt = datetime.fromisoformat(until) if until else None
    return await get_dashboard_stats(db, since=since_dt, until=until_dt)


@router.get("/wip-tasks", response_model=list[WipTask])
async def wip_tasks(
    limit: int = Query(20, ge=1, le=100),
    db: AsyncSession = Depends(get_db),
):
    """在制品看板 — 永远实时的 PENDING/WIP 任务"""
    return await get_wip_tasks(db, limit)


@router.get("/completed-tasks", response_model=list[CompletedTask])
async def completed_tasks(
    since: str | None = Query(None, description="起始日期 ISO 如 2026-08-01T00:00:00"),
    until: str | None = Query(None, description="截止日期 ISO"),
    limit: int = Query(200, ge=1, le=500),
    db: AsyncSession = Depends(get_db),
):
    """流转完成率下钻 — 按时段查询已完成任务明细"""
    since_dt = datetime.fromisoformat(since) if since else None
    until_dt = datetime.fromisoformat(until) if until else None
    return await get_completed_tasks(db, since=since_dt, until=until_dt, limit=limit)


@router.get("/rejected-tasks", response_model=list[RejectedTask])
async def rejected_tasks(
    since: str | None = Query(None, description="起始日期 ISO 如 2026-08-01T00:00:00"),
    until: str | None = Query(None, description="截止日期 ISO"),
    limit: int = Query(200, ge=1, le=500),
    db: AsyncSession = Depends(get_db),
):
    """驳回/返工下钻 — 按时段查询被驳回任务明细（含返工去向）"""
    since_dt = datetime.fromisoformat(since) if since else None
    until_dt = datetime.fromisoformat(until) if until else None
    return await get_rejected_tasks(db, since=since_dt, until=until_dt, limit=limit)


@router.get("/people-workload", response_model=list[PersonWorkload])
async def people_workload(
    db: AsyncSession = Depends(get_db),
):
    """人员负载 — 按负责人聚合当前在制品设备数（独立人员看板）"""
    return await get_people_workload(db)


@router.get("/people-history", response_model=list[PersonHistoryRecord])
async def people_history(
    since: str | None = Query(None, description="起始日期 ISO"),
    until: str | None = Query(None, description="截止日期 ISO"),
    assignee_id: str | None = Query(None, description="负责人ID（精确）"),
    spec_model: str | None = Query(None, description="规格型号（模糊）"),
    product_sn: str | None = Query(None, description="身份证（模糊）"),
    task_name: str | None = Query(None, description="任务名（模糊）"),
    db: AsyncSession = Depends(get_db),
):
    """人员效能与工时台账 — 平铺 Task 明细，多维筛选 + 时间交集"""
    since_dt = datetime.fromisoformat(since) if since else None
    until_dt = datetime.fromisoformat(until) if until else None
    return await get_people_history(
        db, since=since_dt, until=until_dt,
        assignee_id=assignee_id, spec_model=spec_model,
        product_sn=product_sn, task_name=task_name,
    )


@router.get("/people-history/export")
async def export_people_history(
    since: str | None = Query(None, description="起始日期 ISO"),
    until: str | None = Query(None, description="截止日期 ISO"),
    assignee_id: str | None = Query(None, description="负责人ID（精确）"),
    spec_model: str | None = Query(None, description="规格型号（模糊）"),
    product_sn: str | None = Query(None, description="身份证（模糊）"),
    task_name: str | None = Query(None, description="任务名（模糊）"),
    db: AsyncSession = Depends(get_db),
):
    """导出工时台账为 Excel（与查询接口相同筛选条件）"""
    since_dt = datetime.fromisoformat(since) if since else None
    until_dt = datetime.fromisoformat(until) if until else None
    records = await get_people_history(
        db, since=since_dt, until=until_dt,
        assignee_id=assignee_id, spec_model=spec_model,
        product_sn=product_sn, task_name=task_name,
    )

    import csv

    output = io.StringIO()
    writer = csv.writer(output)
    writer.writerow(["状态", "负责人", "身份证", "业务序列号", "产品名称", "规格型号", "任务名", "开始时间", "结束时间", "总耗时(小时)", "最新有效备注"])
    status_label = {"WIP": "进行中", "PENDING": "待接收", "COMPLETED": "已完成"}
    for r in records:
        writer.writerow([
            status_label.get(r.status, r.status),
            r.assignee_name,
            r.product_sn,
            r.external_serial or "",
            r.material_name,
            r.spec_model,
            r.task_name,
            r.received_at or "",
            r.completed_at or "进行中",
            r.duration_hours,
            r.latest_valid_remark or "",
        ])

    data = output.getvalue().encode("utf-8-sig")  # 带 BOM，Excel 正确识别中文
    buf = io.BytesIO(data)
    buf.seek(0)
    return StreamingResponse(
        buf,
        media_type="text/csv; charset=utf-8",
        headers={"Content-Disposition": "attachment; filename=people_history.csv"},
    )


@router.get("/messages", response_model=ProductMessageList)
async def dashboard_messages(
    keyword: str = Query("", description="搜索: SN码/物料名/留言人/内容"),
    skip: int = Query(0, ge=0),
    limit: int = Query(30, ge=1, le=200),
    db: AsyncSession = Depends(get_db),
):
    """
    协同留言搜索（上帝视角 — 全厂所有产品留言）。

    关联 Product 表返回 serial_number + material_name，
    按时间倒序排列。
    """
    return await search_product_messages(db, keyword=keyword, skip=skip, limit=limit)