79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
|
from mysql.connector import Error as MySQLError
|
|||
|
|
|
|||
|
|
from ds.db import db
|
|||
|
|
from schema.device_danger_schema import DeviceDangerCreateRequest, DeviceDangerResponse
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ------------------------------
|
|||
|
|
# 内部工具方法 - 检查设备是否存在（复用设备表逻辑）
|
|||
|
|
# ------------------------------
|
|||
|
|
def check_device_exist(client_ip: str) -> bool:
|
|||
|
|
"""
|
|||
|
|
检查指定IP的设备是否在devices表中存在
|
|||
|
|
|
|||
|
|
:param client_ip: 设备IP地址
|
|||
|
|
:return: 存在返回True、不存在返回False
|
|||
|
|
"""
|
|||
|
|
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,))
|
|||
|
|
return cursor.fetchone() is not None
|
|||
|
|
except MySQLError as e:
|
|||
|
|
raise Exception(f"检查设备存在性失败: {str(e)}") from e
|
|||
|
|
finally:
|
|||
|
|
db.close_connection(conn, cursor)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ------------------------------
|
|||
|
|
# 内部工具方法 - 创建设备危险记录（核心插入逻辑）
|
|||
|
|
# ------------------------------
|
|||
|
|
def create_danger_record(danger_data: DeviceDangerCreateRequest) -> DeviceDangerResponse:
|
|||
|
|
"""
|
|||
|
|
内部工具方法：向device_danger表插入新的危险记录
|
|||
|
|
|
|||
|
|
:param danger_data: 危险记录创建请求数据
|
|||
|
|
:return: 创建成功的危险记录模型对象
|
|||
|
|
"""
|
|||
|
|
# 先检查设备是否存在
|
|||
|
|
if not check_device_exist(danger_data.client_ip):
|
|||
|
|
raise ValueError(f"IP为 {danger_data.client_ip} 的设备不存在、无法创建危险记录")
|
|||
|
|
|
|||
|
|
conn = None
|
|||
|
|
cursor = None
|
|||
|
|
try:
|
|||
|
|
conn = db.get_connection()
|
|||
|
|
cursor = conn.cursor(dictionary=True)
|
|||
|
|
|
|||
|
|
# 插入危险记录（id自增、时间自动填充）
|
|||
|
|
insert_query = """
|
|||
|
|
INSERT INTO device_danger
|
|||
|
|
(client_ip, type, result, created_at, updated_at)
|
|||
|
|
VALUES (%s, %s, %s, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
|
|||
|
|
"""
|
|||
|
|
cursor.execute(insert_query, (
|
|||
|
|
danger_data.client_ip,
|
|||
|
|
danger_data.type,
|
|||
|
|
danger_data.result
|
|||
|
|
))
|
|||
|
|
conn.commit()
|
|||
|
|
|
|||
|
|
# 获取刚创建的记录（用自增ID查询）
|
|||
|
|
danger_id = cursor.lastrowid
|
|||
|
|
cursor.execute("SELECT * FROM device_danger WHERE id = %s", (danger_id,))
|
|||
|
|
new_danger = cursor.fetchone()
|
|||
|
|
|
|||
|
|
return DeviceDangerResponse(**new_danger)
|
|||
|
|
except MySQLError as e:
|
|||
|
|
if conn:
|
|||
|
|
conn.rollback()
|
|||
|
|
raise Exception(f"插入危险记录失败: {str(e)}") from e
|
|||
|
|
finally:
|
|||
|
|
db.close_connection(conn, cursor)
|