Security_AI_integrated/api/camera.py

from typing import List, Optional

from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session

from db.crud import (
    create_camera,
    delete_camera,
    get_all_cameras,
    get_camera_by_id,
    update_camera,
)
from db.models import get_db
from inference.pipeline import get_pipeline

router = APIRouter(prefix="/api/cameras", tags=["摄像头管理"])


@router.get("", response_model=List[dict])
def list_cameras(
    enabled_only: bool = True,
    db: Session = Depends(get_db),
):
    cameras = get_all_cameras(db, enabled_only=enabled_only)
    return [
        {
            "id": cam.id,
            "name": cam.name,
            "rtsp_url": cam.rtsp_url,
            "enabled": cam.enabled,
            "fps_limit": cam.fps_limit,
            "process_every_n_frames": cam.process_every_n_frames,
            "created_at": cam.created_at.isoformat() if cam.created_at else None,
        }
        for cam in cameras
    ]


@router.get("/{camera_id}", response_model=dict)
def get_camera(camera_id: int, db: Session = Depends(get_db)):
    camera = get_camera_by_id(db, camera_id)
    if not camera:
        raise HTTPException(status_code=404, detail="摄像头不存在")
    return {
        "id": camera.id,
        "name": camera.name,
        "rtsp_url": camera.rtsp_url,
        "enabled": camera.enabled,
        "fps_limit": camera.fps_limit,
        "process_every_n_frames": camera.process_every_n_frames,
        "created_at": camera.created_at.isoformat() if camera.created_at else None,
    }


@router.post("", response_model=dict)
def add_camera(
    name: str,
    rtsp_url: str,
    fps_limit: int = 30,
    process_every_n_frames: int = 3,
    db: Session = Depends(get_db),
):
    camera = create_camera(
        db,
        name=name,
        rtsp_url=rtsp_url,
        fps_limit=fps_limit,
        process_every_n_frames=process_every_n_frames,
    )

    if camera.enabled:
        pipeline = get_pipeline()
        pipeline.add_camera(camera)

    return {
        "id": camera.id,
        "name": camera.name,
        "rtsp_url": camera.rtsp_url,
        "enabled": camera.enabled,
    }


@router.put("/{camera_id}", response_model=dict)
def modify_camera(
    camera_id: int,
    name: Optional[str] = None,
    rtsp_url: Optional[str] = None,
    fps_limit: Optional[int] = None,
    process_every_n_frames: Optional[int] = None,
    enabled: Optional[bool] = None,
    db: Session = Depends(get_db),
):
    camera = update_camera(
        db,
        camera_id=camera_id,
        name=name,
        rtsp_url=rtsp_url,
        fps_limit=fps_limit,
        process_every_n_frames=process_every_n_frames,
        enabled=enabled,
    )
    if not camera:
        raise HTTPException(status_code=404, detail="摄像头不存在")

    pipeline = get_pipeline()
    if enabled is True:
        pipeline.add_camera(camera)
    elif enabled is False:
        pipeline.remove_camera(camera_id)

    return {
        "id": camera.id,
        "name": camera.name,
        "rtsp_url": camera.rtsp_url,
        "enabled": camera.enabled,
    }


@router.delete("/{camera_id}")
def remove_camera(camera_id: int, db: Session = Depends(get_db)):
    pipeline = get_pipeline()
    pipeline.remove_camera(camera_id)

    if not delete_camera(db, camera_id):
        raise HTTPException(status_code=404, detail="摄像头不存在")

    return {"message": "删除成功"}


@router.get("/{camera_id}/status")
def get_camera_status(camera_id: int, db: Session = Depends(get_db)):
    from db.crud import get_camera_status as get_status

    status = get_status(db, camera_id)
    pipeline = get_pipeline()

    stream = pipeline.stream_manager.get_stream(str(camera_id))
    if stream:
        stream_info = stream.get_info()
    else:
        stream_info = None

    if status:
        return {
            "camera_id": camera_id,
            "is_running": status.is_running,
            "fps": status.fps,
            "error_message": status.error_message,
            "last_check_time": status.last_check_time.isoformat() if status.last_check_time else None,
            "stream": stream_info,
        }
    else:
        return {
            "camera_id": camera_id,
            "is_running": False,
            "fps": 0.0,
            "error_message": None,
            "last_check_time": None,
            "stream": stream_info,
        }