finnal

FastApi/fastApi.py

@@ -0,0 +1,116 @@
+#本程序用于启用fastApi,用于与摄像头的数据交互建立连接
+
+from fastapi import FastAPI, File, UploadFile, Form, WebSocket, WebSocketDisconnect
+from fastapi.responses import JSONResponse
+import os, shutil, subprocess, json
+
+app = FastAPI()
+
+VIDEO_SAVE_PATH = "/mnt/save/video"
+IMAGE_SAVE_PATH = "/mnt/save/warning"
+MODBUS_BIN_PATH = "/home/orangepi/RKApp/ModBus/modbus"
+
+os.makedirs(VIDEO_SAVE_PATH, exist_ok=True)
+os.makedirs(IMAGE_SAVE_PATH, exist_ok=True)
+
+
+@app.websocket("/ws/distance")
+@app.websocket("/ws/distance/")
+async def websocket_distance(websocket: WebSocket):
+    await websocket.accept()
+    print("✅ WebSocket 客户端已连接")
+    try:
+        while True:
+            data = await websocket.receive_text()
+            try:
+                msg = json.loads(data)
+                distance = msg.get("distance")
+                ts = msg.get("ts")
+                print(f"收到距离: {distance}, 时间戳: {ts}")
+
+                # # 写入日志，单独try，避免异常影响后续流程
+                # try:
+                #     now = datetime.now()
+                #     log_line = f"{now.strftime('%Y-%m-%d %H:%M:%S.%f')[:-3]} distance={distance} ts={ts}\n"
+                #     with open("/home/orangepi/Opencv/time.txt", "a") as f:
+                #         f.write(log_line)
+                # except Exception as log_e:
+                #     print(f"日志写入失败: {log_e}")
+                #调用 Modbus 可执行文件
+                result = subprocess.run(
+                    [MODBUS_BIN_PATH, str(int(distance))],
+                    capture_output=True, text=True, timeout=5
+                )
+                if result.returncode != 0:
+                    # 发送错误信息给客户端
+                    await websocket.send_json({
+                        "error": f"modbus 程序执行失败: {result.stderr.strip()}"
+                    })
+                    continue
+
+                # 正常发送结果
+                await websocket.send_json({
+                    "distance": distance,
+                    "modbus_output": result.stdout.strip()
+                })
+
+            except json.JSONDecodeError:
+                await websocket.send_text("invalid JSON")
+            except Exception as e:
+                await websocket.send_text(f"server error: {e}")
+
+    except WebSocketDisconnect:
+        print("⚠️ WebSocket 客户端断开连接")
+    except Exception as e:
+        print("❌ WebSocket 处理出错:", e)
+
+@app.post("/upload_video/")
+async def upload_video(
+        video: UploadFile = File(..., description="上传的视频（.mp4）")
+):
+    if not video.filename.lower().endswith('.mp4'):
+        return JSONResponse(status_code=400, content={"error": "视频必须为.mp4格式"})
+    video_path = os.path.join(VIDEO_SAVE_PATH, video.filename)
+    with open(video_path, "wb") as vid_file:
+        shutil.copyfileobj(video.file, vid_file)
+    return {"video_saved_to": video_path}
+
+
+@app.post("/upload_image/")
+async def upload_image(
+        image: UploadFile = File(..., description="上传的图片（.jpg）")
+):
+    if not image.filename.lower().endswith('.jpg'):
+        return JSONResponse(status_code=400, content={"error": "图片必须为.jpg格式"})
+    image_path = os.path.join(IMAGE_SAVE_PATH, image.filename)
+    with open(image_path, "wb") as img_file:
+        shutil.copyfileobj(image.file, img_file)
+    return {"image_saved_to": image_path}
+
+
+# @app.post("/upload_distance/")
+# async def upload_distance(
+#         distance: float = Form(..., description="距离")
+# ):
+#     # 调用本地 modbus 程序，将距离作为参数传递
+#     try:
+#         #写入日志精确到毫秒
+#         # now = datetime.now()
+#         # log_line = f"{now.strftime('%Y-%m-%d %H:%M:%S.%f')[:-3]} distance={distance} ts={ts}\n"
+#         # with open("/home/orangepi/Opencv/time.txt", "a") as f:
+#         #     f.write(log_line)
+#         result = subprocess.run(
+#             [MODBUS_BIN_PATH, str(int(distance))],
+#             capture_output=True,
+#             text=True,
+#             timeout=5
+#         )
+#         if result.returncode != 0:
+#             return JSONResponse(status_code=500, content={"error": f"modbus程序执行失败: {result.stderr}"})
+#         return {
+#             "distance": distance,
+#             "modbus_output": result.stdout.strip()
+#         }
+#     except Exception as e:
+#         return JSONResponse(status_code=500, content={"error": str(e)})
+