完善牧安云哨-后端

bbit_ai/app/db/postgres/sentinel.py

@@ -1,5 +1,6 @@
 from config.minIO import get_temp_url_dict
 from config.pgDb import pg_pool
+from models.SentinelRecordRequest import SentinelRecordRequest
 from utils.MyUtils import format_datetime
 
 
@@ -259,3 +260,55 @@ def delete_sentinel_record_db(id: str) -> int:
             cursor.execute("DELETE FROM sentinel_records WHERE id=%s;", (id,))
             conn.commit()
             return cursor.rowcount
+
+
+def saveSentinelRecord(data: SentinelRecordRequest) -> str:
+    sql = """
+        INSERT INTO sentinel_records (
+            license_plate,
+            license_plate_image,
+            vehicle_type,
+            vehicle_image
+        )
+        VALUES (%s, %s, %s, %s)
+        RETURNING id;
+    """
+
+    with pg_pool.getConn() as conn:
+        with conn.cursor() as cursor:
+            cursor.execute(
+                sql,
+                (
+                    data.LicensePlate,
+                    data.LicensePlateImage,
+                    data.VehicleType,
+                    data.VehicleImage,
+                ),
+            )
+            new_id = cursor.fetchone()[0]
+            conn.commit()
+            return str(new_id)
+
+
+def update_sentinel_record(
+    id: str, livestock_type: str, remark: str, dept_id: str
+) -> bool:
+    """
+    根据 id 更新 sentinel_records 表中的 livestock_type 和 dept_id
+    """
+    sql = """
+        UPDATE sentinel_records
+        SET livestock_type = %s,
+            remark = %s,
+            dept_id = %s,
+            updated_at = now()
+        WHERE id = %s
+        RETURNING id;
+    """
+
+    with pg_pool.getConn() as conn:
+        with conn.cursor() as cursor:
+            cursor.execute(sql, (livestock_type, remark, dept_id, id))
+            record = cursor.fetchone()
+            conn.commit()
+            return record is not None