video/service/device_service.py

import json

from fastapi import APIRouter, Query, HTTPException,Request
from mysql.connector import Error as MySQLError

from ds.db import db
from schema.device_schema import DeviceCreateRequest, DeviceResponse, DeviceListResponse
from schema.response_schema import APIResponse

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


# ------------------------------
# 内部工具方法 - 通过客户端IP增加设备报警次数
# ------------------------------
def increment_alarm_count_by_ip(client_ip: str) -> bool:
    """
    通过客户端IP增加设备的报警次数（内部服务方法）

    :param client_ip: 客户端IP地址
    :return: 操作是否成功
    """
    if not client_ip:
        raise ValueError("客户端IP不能为空")

    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        # 检查设备是否存在
        cursor.execute("SELECT id FROM devices WHERE client_ip = %s", (client_ip,))
        if not cursor.fetchone():
            raise ValueError(f"客户端IP为 {client_ip} 的设备不存在")

        # 报警次数加1、并更新时间戳
        update_query = """
            UPDATE devices 
            SET alarm_count = alarm_count + 1, 
                updated_at = CURRENT_TIMESTAMP 
            WHERE client_ip = %s
        """
        cursor.execute(update_query, (client_ip,))
        conn.commit()

        return True
    except MySQLError as e:
        if conn:
            conn.rollback()
        raise Exception(f"更新报警次数失败: {str(e)}") from e
    finally:
        db.close_connection(conn, cursor)


# ------------------------------
# 内部工具方法 - 通过客户端IP更新设备在线状态
# ------------------------------
def update_online_status_by_ip(client_ip: str, online_status: int) -> bool:
    """
    通过客户端IP更新设备的在线状态（内部服务方法）

    :param client_ip: 客户端IP地址
    :param online_status: 在线状态（1-在线、0-离线）
    :return: 操作是否成功
    """
    if not client_ip:
        raise ValueError("客户端IP不能为空")

    # 验证状态值有效性
    if online_status not in (0, 1):
        raise ValueError("在线状态必须是0（离线）或1（在线）")

    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        # 检查设备是否存在
        cursor.execute("SELECT id FROM devices WHERE client_ip = %s", (client_ip,))
        if not cursor.fetchone():
            raise ValueError(f"客户端IP为 {client_ip} 的设备不存在")

        # 更新在线状态和时间戳
        update_query = """
            UPDATE devices 
            SET device_online_status = %s, 
                updated_at = CURRENT_TIMESTAMP 
            WHERE client_ip = %s
        """
        cursor.execute(update_query, (online_status, client_ip))
        conn.commit()

        return True
    except MySQLError as e:
        if conn:
            conn.rollback()
        raise Exception(f"更新设备在线状态失败: {str(e)}") from e
    finally:
        db.close_connection(conn, cursor)


# ------------------------------
# 原有接口保持不变
# ------------------------------
@router.post("/add", response_model=APIResponse, summary="创建设备信息")
async def create_device(device_data: DeviceCreateRequest, request: Request):  # 注入Request对象
    # 原有代码保持不变
    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        cursor.execute("SELECT * FROM devices WHERE client_ip = %s", (device_data.ip,))
        existing_device = cursor.fetchone()
        if existing_device:
            # 更新设备状态为在线
            update_online_status_by_ip(client_ip=device_data.ip, online_status=1)
            # 返回信息
            return APIResponse(
                code=200,
                message=f"设备IP {device_data.ip} 已存在、返回已有设备信息",
                data=DeviceResponse(** existing_device)
            )

        # 直接使用注入的request对象获取用户代理
        user_agent = request.headers.get("User-Agent", "").lower()

        if user_agent == "default":
            device_type = device_data.params.get("os") if (
                    device_data.params and isinstance(device_data.params, dict)) else "unknown"
        elif "windows" in user_agent:
            device_type = "windows"
        elif "android" in user_agent:
            device_type = "android"
        elif "linux" in user_agent:
            device_type = "linux"
        else:
            device_type = "unknown"

        device_params_json = json.dumps(device_data.params) if device_data.params else None

        insert_query = """
            INSERT INTO devices 
            (client_ip, hostname, device_online_status, device_type, alarm_count, params)
            VALUES (%s, %s, %s, %s, %s, %s)
        """
        cursor.execute(insert_query, (
            device_data.ip,
            device_data.hostname,
            1,
            device_type,
            0,
            device_params_json
        ))
        conn.commit()

        device_id = cursor.lastrowid
        cursor.execute("SELECT * FROM devices WHERE id = %s", (device_id,))
        new_device = cursor.fetchone()

        return APIResponse(
            code=200,
            message="设备创建成功",
            data=DeviceResponse(**new_device)
        )

    except MySQLError as e:
        if conn:
            conn.rollback()
        raise Exception(f"创建设备失败: {str(e)}") from e
    except json.JSONDecodeError as e:
        raise Exception(f"设备详细信息JSON序列化失败: {str(e)}") from e
    except Exception as e:
        if conn:
            conn.rollback()
        raise e
    finally:
        db.close_connection(conn, cursor)

@router.get("/", response_model=APIResponse, summary="获取设备列表（支持筛选分页）")
async def get_device_list(
        page: int = Query(1, ge=1, description="页码、默认第1页"),
        page_size: int = Query(10, ge=1, le=100, description="每页条数、1-100之间"),
        device_type: str = Query(None, description="按设备类型筛选"),
        online_status: int = Query(None, ge=0, le=1, description="按在线状态筛选")
):
    # 原有代码保持不变
    conn = None
    cursor = None
    try:
        conn = db.get_connection()
        cursor = conn.cursor(dictionary=True)

        where_clause = []
        params = []

        if device_type:
            where_clause.append("device_type = %s")
            params.append(device_type)
        if online_status is not None:
            where_clause.append("device_online_status = %s")
            params.append(online_status)

        count_query = "SELECT COUNT(*) AS total FROM devices"
        if where_clause:
            count_query += " WHERE " + " AND ".join(where_clause)
        cursor.execute(count_query, params)
        total = cursor.fetchone()["total"]

        offset = (page - 1) * page_size
        list_query = "SELECT * FROM devices"
        if where_clause:
            list_query += " WHERE " + " AND ".join(where_clause)
        list_query += " ORDER BY id DESC LIMIT %s OFFSET %s"
        params.extend([page_size, offset])

        cursor.execute(list_query, params)
        device_list = cursor.fetchall()

        return APIResponse(
            code=200,
            message="获取设备列表成功",
            data=DeviceListResponse(
                total=total,
                devices=[DeviceResponse(**device) for device in device_list]
            )
        )

    except MySQLError as e:
        raise Exception(f"获取设备列表失败: {str(e)}") from e
    finally:
        db.close_connection(conn, cursor)