012-kaopeilian/backend/app/api/v1/endpoints/dashboard.py

"""
数据大屏 API 端点

提供企业级和团队级数据大屏接口
"""

from typing import Optional
from fastapi import APIRouter, Depends, HTTPException, status, Query
from sqlalchemy.ext.asyncio import AsyncSession

from app.core.database import get_db
from app.core.security import get_current_user
from app.models.user import User
from app.services.dashboard_service import DashboardService

router = APIRouter()


@router.get("/enterprise/overview")
async def get_enterprise_overview(
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """
    获取企业级数据概览
    
    需要管理员或企业管理员权限
    """
    if current_user.role not in ["admin", "enterprise_admin", "manager"]:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="需要管理员权限"
        )
    
    service = DashboardService(db)
    data = await service.get_enterprise_overview()
    
    return {
        "code": 200,
        "message": "success",
        "data": data
    }


@router.get("/enterprise/departments")
async def get_department_comparison(
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """
    获取部门/团队学习对比数据
    """
    if current_user.role not in ["admin", "enterprise_admin", "manager"]:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="需要管理员权限"
        )
    
    service = DashboardService(db)
    data = await service.get_department_comparison()
    
    return {
        "code": 200,
        "message": "success",
        "data": data
    }


@router.get("/enterprise/trend")
async def get_learning_trend(
    days: int = Query(7, ge=1, le=30),
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """
    获取学习趋势数据
    """
    if current_user.role not in ["admin", "enterprise_admin", "manager"]:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="需要管理员权限"
        )
    
    service = DashboardService(db)
    data = await service.get_learning_trend(days)
    
    return {
        "code": 200,
        "message": "success",
        "data": data
    }


@router.get("/enterprise/level-distribution")
async def get_level_distribution(
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """
    获取等级分布数据
    """
    if current_user.role not in ["admin", "enterprise_admin", "manager"]:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="需要管理员权限"
        )
    
    service = DashboardService(db)
    data = await service.get_level_distribution()
    
    return {
        "code": 200,
        "message": "success",
        "data": data
    }


@router.get("/enterprise/activities")
async def get_realtime_activities(
    limit: int = Query(20, ge=1, le=100),
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """
    获取实时动态
    """
    if current_user.role not in ["admin", "enterprise_admin", "manager"]:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="需要管理员权限"
        )
    
    service = DashboardService(db)
    data = await service.get_realtime_activities(limit)
    
    return {
        "code": 200,
        "message": "success",
        "data": data
    }


@router.get("/enterprise/course-ranking")
async def get_course_ranking(
    limit: int = Query(10, ge=1, le=50),
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """
    获取课程热度排行
    """
    if current_user.role not in ["admin", "enterprise_admin", "manager"]:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="需要管理员权限"
        )
    
    service = DashboardService(db)
    data = await service.get_course_ranking(limit)
    
    return {
        "code": 200,
        "message": "success",
        "data": data
    }


@router.get("/team")
async def get_team_dashboard(
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """
    获取团队级数据大屏
    
    面向团队负责人，显示其管理团队的数据
    """
    if current_user.role not in ["admin", "enterprise_admin", "manager"]:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="需要团队负责人权限"
        )
    
    service = DashboardService(db)
    data = await service.get_team_dashboard(current_user.id)
    
    return {
        "code": 200,
        "message": "success",
        "data": data
    }


@router.get("/all")
async def get_all_dashboard_data(
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user)
):
    """
    获取完整的大屏数据（一次性获取所有数据）
    
    用于大屏初始化加载
    """
    if current_user.role not in ["admin", "enterprise_admin", "manager"]:
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail="需要管理员权限"
        )
    
    service = DashboardService(db)
    
    # 并行获取所有数据
    overview = await service.get_enterprise_overview()
    departments = await service.get_department_comparison()
    trend = await service.get_learning_trend(7)
    level_dist = await service.get_level_distribution()
    activities = await service.get_realtime_activities(20)
    course_ranking = await service.get_course_ranking(10)
    
    return {
        "code": 200,
        "message": "success",
        "data": {
            "overview": overview,
            "departments": departments,
            "trend": trend,
            "level_distribution": level_dist,
            "activities": activities,
            "course_ranking": course_ranking,
        }
    }