Spaces:
Sleeping
Sleeping
File size: 2,872 Bytes
b52caaa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
from flask import Blueprint, render_template, jsonify, request, Response
from models import Device, db # Assuming Device is your database model from models.py
from device_core import Sensor, Camera # Import your Sensor class
from device_routes import SENSORS, CAMERAS # Import your SENSORS list
import json
import plotly
sensor_blueprint = Blueprint('sensor_blueprint', __name__, template_folder='templates')
camera_blueprint = Blueprint('camera_blueprint', __name__, template_folder='templates')
@sensor_blueprint.route('/plot/<sensor_id>')
def generate_plot(sensor_id):
sensor = Sensor.find(SENSORS, sensor_id) # Assuming SENSORS is accessible here
if not sensor:
return jsonify({"error": "Sensor not found"}), 404
fig = sensor.plot(Device, title=sensor.name)
return json.dumps(fig, cls=plotly.utils.PlotlyJSONEncoder)
@sensor_blueprint.route('/postplain/', methods=['POST'])
def post_plain():
data = request.get_json()
if not data:
return jsonify({'message': 'No input data provided'}), 400
device_id = data.get('id')
device_values = data.get('values')
device_units = data.get('units')
if device_values and device_id:
sensor = Sensor.find(SENSORS, device_id)
if sensor:
sensor.update_values(device_values, device_units)
device = Device(
sensor_id=device_id,
name=sensor.name,
time=sensor.time,
)
for i, value in enumerate(sensor.values):
setattr(device, f"value{i + 1}", value)
for i, unit in enumerate(sensor.units):
setattr(device, f"unit{i + 1}", unit)
db.session.add(device)
db.session.commit()
return jsonify({'message': 'Success!'}), 200
else:
return jsonify({'message': 'Sensor not found'}), 404
else:
return jsonify({'message': 'Missing data'}), 400
# -------------- CAMERA ROUTES --------------
@camera_blueprint.route('/video_feed/<camera_id>')
def stream_camera(camera_id):
"""
Stream the video feed from a specific camera.
"""
camera = next((cam for cam in CAMERAS if cam.id == camera_id), None)
if not camera:
return Response("Camera not found or could not be opened", status=404)
return Response(camera.gen_frames(), mimetype='multipart/x-mixed-replace; boundary=frame')
@camera_blueprint.route('/refresh_camera')
def refresh_camera():
success = True
message = "All cameras refreshed successfully."
for cam in CAMERAS:
cam.close_camera()
try:
cam.__init__(cam.id, cam.name, cam.index)
except:
success = False
message = "Error occurred while refreshing cameras."
return jsonify({"success": success, "message": message})
|