113 lines
3.8 KiB
Python
113 lines
3.8 KiB
Python
|
from fastapi import APIRouter, Query, Path
|
|||
|
|
from mysql.connector import Error as MySQLError
|
|||
|
|
|
|||
|
|
from ds.db import db
|
|||
|
|
from encryption.encrypt_decorator import encrypt_response
|
|||
|
|
from schema.device_action_schema import (
|
|||
|
|
DeviceActionResponse,
|
|||
|
|
DeviceActionListResponse
|
|||
|
|
)
|
|||
|
|
from schema.response_schema import APIResponse
|
|||
|
|
|
|||
|
|
# 路由配置
|
|||
|
|
router = APIRouter(
|
|||
|
|
prefix="/api/device/actions",
|
|||
|
|
tags=["设备操作记录"]
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
@router.get("/list", response_model=APIResponse, summary="分页查询设备操作记录")
|
|||
|
|
@encrypt_response()
|
|||
|
|
async def get_device_action_list(
|
|||
|
|
page: int = Query(1, ge=1, description="页码、默认1"),
|
|||
|
|
page_size: int = Query(10, ge=1, le=100, description="每页条数、1-100"),
|
|||
|
|
client_ip: str = Query(None, description="按客户端IP筛选"),
|
|||
|
|
action: int = Query(None, ge=0, le=1, description="按状态筛选（0=离线、1=上线）")
|
|||
|
|
):
|
|||
|
|
conn = None
|
|||
|
|
cursor = None
|
|||
|
|
try:
|
|||
|
|
conn = db.get_connection()
|
|||
|
|
cursor = conn.cursor(dictionary=True)
|
|||
|
|
# 构建筛选条件（参数化查询、避免注入）
|
|||
|
|
where_clause = []
|
|||
|
|
params = []
|
|||
|
|
if client_ip:
|
|||
|
|
where_clause.append("client_ip = %s")
|
|||
|
|
params.append(client_ip)
|
|||
|
|
if action is not None:
|
|||
|
|
where_clause.append("action = %s")
|
|||
|
|
params.append(action)
|
|||
|
|
|
|||
|
|
# 查询总记录数（用于返回 total）
|
|||
|
|
count_sql = "SELECT COUNT(*) AS total FROM device_action"
|
|||
|
|
if where_clause:
|
|||
|
|
count_sql += " WHERE " + " AND ".join(where_clause)
|
|||
|
|
cursor.execute(count_sql, params)
|
|||
|
|
total = cursor.fetchone()["total"]
|
|||
|
|
|
|||
|
|
# 分页查询记录（按创建时间倒序、确保最新记录在前）
|
|||
|
|
offset = (page - 1) * page_size
|
|||
|
|
list_sql = "SELECT * FROM device_action"
|
|||
|
|
if where_clause:
|
|||
|
|
list_sql += " WHERE " + " AND ".join(where_clause)
|
|||
|
|
list_sql += " ORDER BY created_at DESC LIMIT %s OFFSET %s"
|
|||
|
|
params.extend([page_size, offset])
|
|||
|
|
|
|||
|
|
cursor.execute(list_sql, params)
|
|||
|
|
action_list = cursor.fetchall()
|
|||
|
|
|
|||
|
|
# 仅返回 total + device_actions
|
|||
|
|
return APIResponse(
|
|||
|
|
code=200,
|
|||
|
|
message="查询成功",
|
|||
|
|
data=DeviceActionListResponse(
|
|||
|
|
total=total,
|
|||
|
|
device_actions=[DeviceActionResponse(**item) for item in action_list]
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
except MySQLError as e:
|
|||
|
|
raise Exception(f"查询记录失败: {str(e)}") from e
|
|||
|
|
finally:
|
|||
|
|
db.close_connection(conn, cursor)
|
|||
|
|
|
|||
|
|
@router.get("/{client_ip}", response_model=APIResponse, summary="根据IP查询设备操作记录")
|
|||
|
|
@encrypt_response()
|
|||
|
|
async def get_device_actions_by_ip(
|
|||
|
|
client_ip: str = Path(..., description="客户端IP地址")
|
|||
|
|
):
|
|||
|
|
conn = None
|
|||
|
|
cursor = None
|
|||
|
|
try:
|
|||
|
|
conn = db.get_connection()
|
|||
|
|
cursor = conn.cursor(dictionary=True)
|
|||
|
|
|
|||
|
|
# 1. 查询总记录数
|
|||
|
|
count_sql = "SELECT COUNT(*) AS total FROM device_action WHERE client_ip = %s"
|
|||
|
|
cursor.execute(count_sql, (client_ip,))
|
|||
|
|
total = cursor.fetchone()["total"]
|
|||
|
|
|
|||
|
|
# 2. 查询该IP的所有记录（按创建时间倒序）
|
|||
|
|
list_sql = """
|
|||
|
|
SELECT * FROM device_action
|
|||
|
|
WHERE client_ip = %s
|
|||
|
|
ORDER BY created_at DESC
|
|||
|
|
"""
|
|||
|
|
cursor.execute(list_sql, (client_ip,))
|
|||
|
|
action_list = cursor.fetchall()
|
|||
|
|
|
|||
|
|
# 3. 返回结果
|
|||
|
|
return APIResponse(
|
|||
|
|
code=200,
|
|||
|
|
message="查询成功",
|
|||
|
|
data=DeviceActionListResponse(
|
|||
|
|
total=total,
|
|||
|
|
device_actions=[DeviceActionResponse(**item) for item in action_list]
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
except MySQLError as e:
|
|||
|
|
raise Exception(f"查询记录失败: {str(e)}") from e
|
|||
|
|
finally:
|
|||
|
|
db.close_connection(conn, cursor)
|