airlabs-manage/backend/routers/dashboard.py

"""仪表盘 + 结算路由"""
from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session
from sqlalchemy import func as sa_func
from datetime import date, timedelta
from database import get_db
from models import (
    User, Project, Submission, AIToolCost,
    ProjectStatus, ProjectType, WorkType, PhaseGroup
)
from auth import get_current_user, require_permission
from calculations import (
    calc_project_settlement, calc_waste_for_project,
    calc_labor_cost_for_project, calc_ai_tool_cost_for_project,
    calc_outsource_cost_for_project, calc_overhead_cost_for_project,
    calc_management_cost_for_project, calc_team_efficiency
)

router = APIRouter(prefix="/api", tags=["仪表盘与结算"])


@router.get("/dashboard")
def get_dashboard(
    db: Session = Depends(get_db),
    current_user: User = Depends(require_permission("dashboard:view"))
):
    """全局仪表盘数据"""
    # 项目概览
    active = db.query(Project).filter(Project.status == ProjectStatus.IN_PROGRESS).all()
    completed = db.query(Project).filter(Project.status == ProjectStatus.COMPLETED).all()
    abandoned = db.query(Project).filter(Project.status == ProjectStatus.ABANDONED).all()

    # 当月日期范围
    today = date.today()
    month_start = today.replace(day=1)

    # 本月人力成本（简化：统计本月所有有提交的人的日成本）
    monthly_submitters = db.query(Submission.user_id, Submission.submit_date).filter(
        Submission.submit_date >= month_start,
        Submission.submit_date <= today,
    ).distinct().all()

    monthly_labor = 0.0
    processed_user_dates = set()
    for uid, d in monthly_submitters:
        key = (uid, d)
        if key not in processed_user_dates:
            processed_user_dates.add(key)
            user = db.query(User).filter(User.id == uid).first()
            if user:
                monthly_labor += user.daily_cost

    # 本月 AI 工具成本
    monthly_ai = db.query(sa_func.sum(AIToolCost.amount)).filter(
        AIToolCost.record_date >= month_start,
        AIToolCost.record_date <= today,
    ).scalar() or 0

    # 本月总产出秒数（只算中期动画制作）
    monthly_secs = db.query(sa_func.sum(Submission.total_seconds)).filter(
        Submission.submit_date >= month_start,
        Submission.submit_date <= today,
        Submission.total_seconds > 0,
        Submission.project_phase == PhaseGroup.PRODUCTION,
    ).scalar() or 0

    # 活跃人数
    active_users = db.query(Submission.user_id).filter(
        Submission.submit_date >= month_start,
    ).distinct().count()
    working_days = max(1, (today - month_start).days + 1)
    avg_daily = round(monthly_secs / max(1, active_users) / working_days, 1)

    # 各项目摘要（进度只算中期动画制作）
    project_summaries = []
    for p in active:
        waste = calc_waste_for_project(p.id, db)
        # 只用中期制作产出算进度，不把后期算进去
        production_secs = db.query(sa_func.sum(Submission.total_seconds)).filter(
            Submission.project_id == p.id,
            Submission.total_seconds > 0,
            Submission.project_phase == PhaseGroup.PRODUCTION,
        ).scalar() or 0
        target = p.target_total_seconds
        progress = round(production_secs / target * 100, 1) if target > 0 else 0
        is_overdue = (
            p.estimated_completion_date and today > p.estimated_completion_date
        )
        project_summaries.append({
            "id": p.id,
            "name": p.name,
            "project_type": p.project_type.value if hasattr(p.project_type, 'value') else p.project_type,
            "progress_percent": progress,
            "target_seconds": target,
            "submitted_seconds": round(production_secs, 1),
            "waste_rate": waste.get("waste_rate", 0),
            "waste_hours": waste.get("total_waste_hours", 0),
            "is_overdue": bool(is_overdue),
            "estimated_completion_date": str(p.estimated_completion_date) if p.estimated_completion_date else None,
        })

    # 损耗排行（含废弃项目，废弃项目全部产出记为损耗）
    waste_ranking = []
    total_waste_seconds_all = 0.0
    total_waste_hours_all = 0.0
    total_target_seconds_all = 0.0
    for p in active + completed + abandoned:
        w = calc_waste_for_project(p.id, db)
        total_waste_seconds_all += w.get("total_waste_seconds", 0)
        total_waste_hours_all += w.get("total_waste_hours", 0)
        total_target_seconds_all += p.target_total_seconds or 0
        if w.get("total_waste_seconds", 0) > 0 or w.get("total_waste_hours", 0) > 0:
            waste_ranking.append({
                "project_id": p.id,
                "project_name": p.name,
                "waste_seconds": w["total_waste_seconds"],
                "waste_hours": w.get("total_waste_hours", 0),
                "waste_rate": w["waste_rate"],
            })
    waste_ranking.sort(key=lambda x: x["waste_rate"], reverse=True)
    total_waste_rate = round(total_waste_seconds_all / total_target_seconds_all * 100, 1) if total_target_seconds_all > 0 else 0

    # 已结算项目
    settled = []
    for p in completed:
        settlement = calc_project_settlement(p.id, db)
        settled.append({
            "project_id": p.id,
            "project_name": p.name,
            "project_type": settlement.get("project_type", ""),
            "total_cost": settlement.get("total_cost", 0),
            "contract_amount": settlement.get("contract_amount"),
            "profit_loss": settlement.get("profit_loss"),
        })

    # ── 图表数据：近30天每日产出趋势（只算中期动画制作） ──
    daily_trend = []
    for i in range(29, -1, -1):
        d = today - timedelta(days=i)
        day_secs = db.query(sa_func.sum(Submission.total_seconds)).filter(
            Submission.submit_date == d,
            Submission.total_seconds > 0,
            Submission.project_phase == PhaseGroup.PRODUCTION,
        ).scalar() or 0
        daily_trend.append({
            "date": str(d),
            "seconds": round(day_secs, 1),
        })

    # ── 图表数据：成本构成 ──
    total_labor_all = 0.0
    total_ai_all = 0.0
    total_outsource_all = 0.0
    total_overhead_all = 0.0
    total_management_all = 0.0
    for p in active + completed + abandoned:
        total_labor_all += calc_labor_cost_for_project(p.id, db)
        total_ai_all += calc_ai_tool_cost_for_project(p.id, db)
        total_outsource_all += calc_outsource_cost_for_project(p.id, db)
        total_overhead_all += calc_overhead_cost_for_project(p.id, db)
        total_management_all += calc_management_cost_for_project(p.id, db)

    cost_breakdown = [
        {"name": "人力成本", "value": round(total_labor_all, 0)},
        {"name": "AI工具", "value": round(total_ai_all, 0)},
        {"name": "外包", "value": round(total_outsource_all, 0)},
        {"name": "固定开支", "value": round(total_overhead_all, 0)},
        {"name": "管理成本", "value": round(total_management_all, 0)},
    ]

    # ── 图表数据：各项目产出对比（进行中项目，只算中期动画制作） ──
    project_comparison = []
    for p in active:
        total_secs_p = db.query(sa_func.sum(Submission.total_seconds)).filter(
            Submission.project_id == p.id,
            Submission.total_seconds > 0,
            Submission.project_phase == PhaseGroup.PRODUCTION,
        ).scalar() or 0
        project_comparison.append({
            "name": p.name,
            "submitted": round(total_secs_p, 0),
            "target": p.target_total_seconds,
        })

    # ── 盈利概览 ──
    total_contract = 0.0
    total_cost_completed = 0.0
    for s in settled:
        if s.get("contract_amount"):
            total_contract += s["contract_amount"]
        total_cost_completed += s.get("total_cost", 0)
    total_profit = total_contract - total_cost_completed
    profit_rate = round(total_profit / total_contract * 100, 1) if total_contract > 0 else 0

    # 进行中项目的合同额和当前成本
    in_progress_contract = 0.0
    in_progress_cost = 0.0
    for p in active:
        if p.contract_amount:
            in_progress_contract += p.contract_amount
        in_progress_cost += calc_labor_cost_for_project(p.id, db) + calc_ai_tool_cost_for_project(p.id, db) + calc_outsource_cost_for_project(p.id, db) + calc_overhead_cost_for_project(p.id, db) + calc_management_cost_for_project(p.id, db)

    # 每个项目的盈亏（用于柱状图）
    profit_by_project = []
    for s in settled:
        if s.get("contract_amount") and s["contract_amount"] > 0:
            profit_by_project.append({
                "name": s["project_name"],
                "profit": round((s.get("contract_amount", 0) or 0) - s.get("total_cost", 0), 0),
            })

    profitability = {
        "total_contract": round(total_contract, 0),
        "total_cost": round(total_cost_completed, 0),
        "total_profit": round(total_profit, 0),
        "profit_rate": profit_rate,
        "in_progress_contract": round(in_progress_contract, 0),
        "in_progress_cost": round(in_progress_cost, 0),
        "profit_by_project": profit_by_project,
    }

    # ── 风险预警 ──
    from services.report_service import analyze_project_risks
    risk_alerts = analyze_project_risks(db)

    # ── 今日提交情况（排除豁免角色） ──
    from models import Role
    exempt_role_ids = set(
        r.id for r in db.query(Role).filter(Role.exempt_submission == 1).all()
    )
    all_active_users = [
        u for u in db.query(User).filter(User.is_active == 1).all()
        if u.role_id not in exempt_role_ids
    ]
    submitted_user_ids = set(
        uid for (uid,) in db.query(Submission.user_id).filter(
            Submission.submit_date == today,
        ).distinct().all()
    )
    submitted_users = []
    not_submitted_users = []
    for u in all_active_users:
        info = {"id": u.id, "name": u.name}
        if u.id in submitted_user_ids:
            hours = db.query(sa_func.sum(Submission.hours_spent)).filter(
                Submission.user_id == u.id,
                Submission.submit_date == today,
            ).scalar() or 0
            info["hours"] = round(hours, 1)
            submitted_users.append(info)
        else:
            not_submitted_users.append(info)
    daily_attendance = {
        "total": len(all_active_users),
        "submitted_count": len(submitted_users),
        "not_submitted_count": len(not_submitted_users),
        "submitted": submitted_users,
        "not_submitted": not_submitted_users,
    }

    return {
        "active_projects": len(active),
        "completed_projects": len(completed),
        "monthly_labor_cost": round(monthly_labor, 2),
        "monthly_ai_tool_cost": round(monthly_ai, 2),
        "monthly_total_seconds": round(monthly_secs, 1),
        "avg_daily_seconds_per_person": avg_daily,
        "projects": project_summaries,
        "total_waste_seconds": round(total_waste_seconds_all, 0),
        "total_waste_hours": round(total_waste_hours_all, 0),
        "total_waste_rate": total_waste_rate,
        "waste_ranking": waste_ranking,
        "settled_projects": settled,
        "profitability": profitability,
        "risk_alerts": risk_alerts,
        "daily_attendance": daily_attendance,
        # 图表数据
        "daily_trend": daily_trend,
        "cost_breakdown": cost_breakdown,
        "project_comparison": project_comparison,
    }


@router.get("/projects/{project_id}/settlement")
def get_settlement(
    project_id: int,
    db: Session = Depends(get_db),
    current_user: User = Depends(require_permission("settlement:view"))
):
    """项目结算报告"""
    project = db.query(Project).filter(Project.id == project_id).first()
    if not project:
        raise HTTPException(status_code=404, detail="项目不存在")
    return calc_project_settlement(project_id, db)


@router.get("/projects/{project_id}/efficiency")
def get_efficiency(
    project_id: int,
    db: Session = Depends(get_db),
    current_user: User = Depends(require_permission("efficiency:view"))
):
    """项目团队效率数据"""
    return calc_team_efficiency(project_id, db)