45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
"""Workload endpoints for skill domain."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from ....database import get_db
|
|
from ....dependencies import get_current_user, get_tenant_id
|
|
from ....models import User
|
|
from ....services import workload_service
|
|
from .shared import ensure_self_or_admin
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/workload/team")
|
|
async def get_team_workload_board(
|
|
tenant_id: int = Depends(get_tenant_id),
|
|
current_user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
_ = current_user
|
|
data = await workload_service.get_team_workload_board(db, tenant_id)
|
|
return {"items": data, "total": len(data)}
|
|
|
|
|
|
@router.get("/workload/users/{user_id}", response_model=dict)
|
|
async def get_user_personal_workload(
|
|
user_id: int,
|
|
tenant_id: int = Depends(get_tenant_id),
|
|
current_user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
ensure_self_or_admin(
|
|
current_user=current_user,
|
|
target_user_id=user_id,
|
|
detail="仅可查询自己的负载",
|
|
)
|
|
|
|
data = await workload_service.get_user_personal_workload(db, tenant_id, user_id)
|
|
if not data:
|
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="无负载数据")
|
|
return data
|