239 lines
7.8 KiB
Python
239 lines
7.8 KiB
Python
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)
|