修改--批量同步数据时,添加插入message库

app/api/v1/endpoints/events.py

@@ -253,6 +253,7 @@ async def get_messages_handle_robot(
 # 批量处理告警数据--传递messageId List
 @router.post("/events/handleOcrAlerts", response_model=BaseResponse)
 async def handle_ocr_alerts(
+    number:Optional[str] = None,
     db: AsyncSession = Depends(get_db),
     messageIdList: List[int] = Body(...)
 ):
@@ -268,6 +269,7 @@ async def handle_ocr_alerts(
 # 批量处理告警数据--直接将 handle=0的置为1
 @router.get("/events/handleOcrAlerts", response_model=BaseResponse)
 async def handle_ocr_alerts_get(
+    number:Optional[str] = None,
     db: AsyncSession = Depends(get_db)
 ):
     """_summary_
@@ -278,7 +280,7 @@ async def handle_ocr_alerts_get(
     Returns:
         _type_: _description_
     """
-    flag = await event.handle_ocr_alerts_get(db)
+    flag = await event.handle_ocr_alerts_get(db, number)
     if flag:
         return BaseResponse(code=200, msg="success")
     return BaseResponse(code=500, msg="fail to update data")

app/crud/event.py

@@ -325,17 +325,22 @@ class CRUDEvent(CRUDBase[Event, EventUpdate, EventUpdate]):
         
     async def handle_ocr_alerts_get(
         self,
-        db: AsyncSession
+        db: AsyncSession,
+        number: Optional[str] = None
     ):
         try:
-            update_stmt = (
-                update(Message)
-                .where(Message.handle == "0")
-                .values(
-                    handle = "1",
-                    updateTime=datetime.now()
+            
+            if number:
+                update_stmt = ()
+            else:
+                update_stmt = (
+                    update(Message)
+                    .where(Message.handle == "0")
+                    .values(
+                        handle = "1",
+                        updateTime=datetime.now()
+                    )
                 )
-            )
             
             # 执行更新
             await db.execute(

app/services/event_sync_service.py

@@ -207,6 +207,9 @@ class EventSyncService:
                 # 保存ocr温度信息
                 await self._ocr_images(session, image_list)
                 
+                # 保存告警消息
+                await self._save_message(session, detail)
+                
 
     async def _update_event(self, session: AsyncSession, detail: Dict[str, Any]):
         """更新事件信息"""

run_sync.py

@@ -7,5 +7,5 @@ from app.services.event_sync_service import run_sync, run_sync_event
 
 if __name__ == "__main__":
     print("启动事件同步服务...")
-    # asyncio.run(run_sync()) 
-    asyncio.run(run_sync_event("f3a4951618334c7dafef03ad5345a3c1"))
+    asyncio.run(run_sync()) 
+    # asyncio.run(run_sync_event("f3a4951618334c7dafef03ad5345a3c1"))