117 lines
4.4 KiB
Python
117 lines
4.4 KiB
Python
#本程序用于启用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)})
|
||
|