44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from ds.db import db
|
||
from schema.device_action_schema import DeviceActionCreate, DeviceActionResponse
|
||
from mysql.connector import Error as MySQLError
|
||
|
||
|
||
# 新增设备操作记录
|
||
def add_device_action(client_ip: str, action: int) -> DeviceActionResponse:
|
||
"""
|
||
新增设备操作记录（内部方法、非接口）
|
||
:param action_data: 含client_ip和action（0/1）
|
||
:return: 新增的完整记录
|
||
"""
|
||
conn = None
|
||
cursor = None
|
||
try:
|
||
conn = db.get_connection()
|
||
cursor = conn.cursor(dictionary=True)
|
||
|
||
# 插入SQL（id自增、依赖数据库自动生成）
|
||
insert_query = """
|
||
INSERT INTO device_action
|
||
(client_ip, action, created_at, updated_at)
|
||
VALUES (%s, %s, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
|
||
"""
|
||
cursor.execute(insert_query, (
|
||
client_ip,
|
||
action
|
||
))
|
||
conn.commit()
|
||
|
||
# 获取新增记录（通过自增ID查询）
|
||
new_id = cursor.lastrowid
|
||
cursor.execute("SELECT * FROM device_action WHERE id = %s", (new_id,))
|
||
new_action = cursor.fetchone()
|
||
|
||
return DeviceActionResponse(**new_action)
|
||
|
||
except MySQLError as e:
|
||
if conn:
|
||
conn.rollback()
|
||
raise Exception(f"新增记录失败: {str(e)}") from e
|
||
finally:
|
||
db.close_connection(conn, cursor)
|