desungongpai/app/routers/devices.py

"""
Devices Router - 设备管理接口
API endpoints for device CRUD operations and statistics.
"""

import math

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

from app.database import get_db
from app.schemas import (
    APIResponse,
    DeviceCreate,
    DeviceResponse,
    DeviceUpdate,
    PaginatedList,
)
from app.services import device_service

router = APIRouter(prefix="/api/devices", tags=["Devices / 设备管理"])


@router.get(
    "",
    response_model=APIResponse[PaginatedList[DeviceResponse]],
    summary="获取设备列表 / List devices",
)
async def list_devices(
    page: int = Query(default=1, ge=1, description="页码 / Page number"),
    page_size: int = Query(default=20, ge=1, le=100, description="每页数量 / Items per page"),
    status: str | None = Query(default=None, description="状态过滤 / Status filter (online/offline)"),
    search: str | None = Query(default=None, description="搜索IMEI或名称 / Search by IMEI or name"),
    db: AsyncSession = Depends(get_db),
):
    """
    获取设备列表，支持分页、状态过滤和搜索。
    List devices with pagination, optional status filter, and search.
    """
    devices, total = await device_service.get_devices(
        db, page=page, page_size=page_size, status_filter=status, search=search
    )
    return APIResponse(
        data=PaginatedList(
            items=[DeviceResponse.model_validate(d) for d in devices],
            total=total,
            page=page,
            page_size=page_size,
            total_pages=math.ceil(total / page_size) if total else 0,
        )
    )


@router.get(
    "/stats",
    response_model=APIResponse[dict],
    summary="获取设备统计 / Get device statistics",
)
async def device_stats(db: AsyncSession = Depends(get_db)):
    """
    获取设备统计信息：总数、在线、离线。
    Get device statistics: total, online, offline counts.
    """
    stats = await device_service.get_device_stats(db)
    return APIResponse(data=stats)


@router.get(
    "/imei/{imei}",
    response_model=APIResponse[DeviceResponse],
    summary="按IMEI查询设备 / Get device by IMEI",
)
async def get_device_by_imei(imei: str, db: AsyncSession = Depends(get_db)):
    """
    按IMEI号查询设备信息。
    Get device details by IMEI number.
    """
    device = await device_service.get_device_by_imei(db, imei)
    if device is None:
        raise HTTPException(status_code=404, detail=f"Device with IMEI {imei} not found / 未找到IMEI为{imei}的设备")
    return APIResponse(data=DeviceResponse.model_validate(device))


@router.get(
    "/{device_id}",
    response_model=APIResponse[DeviceResponse],
    summary="获取设备详情 / Get device details",
)
async def get_device(device_id: int, db: AsyncSession = Depends(get_db)):
    """
    按ID获取设备详细信息。
    Get device details by ID.
    """
    device = await device_service.get_device(db, device_id)
    if device is None:
        raise HTTPException(status_code=404, detail=f"Device {device_id} not found / 未找到设备{device_id}")
    return APIResponse(data=DeviceResponse.model_validate(device))


@router.post(
    "",
    response_model=APIResponse[DeviceResponse],
    status_code=201,
    summary="创建设备 / Create device",
)
async def create_device(device_data: DeviceCreate, db: AsyncSession = Depends(get_db)):
    """
    手动注册新设备。
    Manually register a new device.
    """
    # Check for duplicate IMEI
    existing = await device_service.get_device_by_imei(db, device_data.imei)
    if existing is not None:
        raise HTTPException(
            status_code=400,
            detail=f"Device with IMEI {device_data.imei} already exists / IMEI {device_data.imei} 已存在",
        )

    device = await device_service.create_device(db, device_data)
    return APIResponse(data=DeviceResponse.model_validate(device))


@router.put(
    "/{device_id}",
    response_model=APIResponse[DeviceResponse],
    summary="更新设备信息 / Update device",
)
async def update_device(
    device_id: int, device_data: DeviceUpdate, db: AsyncSession = Depends(get_db)
):
    """
    更新设备信息（名称、状态等）。
    Update device information (name, status, etc.).
    """
    device = await device_service.update_device(db, device_id, device_data)
    if device is None:
        raise HTTPException(status_code=404, detail=f"Device {device_id} not found / 未找到设备{device_id}")
    return APIResponse(data=DeviceResponse.model_validate(device))


@router.delete(
    "/{device_id}",
    response_model=APIResponse,
    summary="删除设备 / Delete device",
)
async def delete_device(device_id: int, db: AsyncSession = Depends(get_db)):
    """
    删除设备及其关联数据。
    Delete a device and all associated records.
    """
    deleted = await device_service.delete_device(db, device_id)
    if not deleted:
        raise HTTPException(status_code=404, detail=f"Device {device_id} not found / 未找到设备{device_id}")
    return APIResponse(message="Device deleted successfully / 设备删除成功")