50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
from db_connect import DatabaseConnect
|
|
|
|
from flask import Flask, abort, session
|
|
from jinja2 import Template
|
|
import plotly.express as px
|
|
from plotly.subplots import make_subplots
|
|
import plotly.graph_objects as go
|
|
|
|
import logging
|
|
from os import sep
|
|
|
|
app = Flask(__name__)
|
|
db = DatabaseConnect()
|
|
|
|
@app.route("/")
|
|
def index():
|
|
avail_rooms = db.view_valid_rooms(1)
|
|
outtext = "<h1>Available rooms</h1><ul>"
|
|
if len(avail_rooms) == 0:
|
|
outtext += "<li>You have no rooms you can view</li>"
|
|
else:
|
|
for room in avail_rooms:
|
|
outtext += f"<li><a href=/room/{room[1]}>{room[0]}</li>"
|
|
outtext += "</ul>"
|
|
return outtext
|
|
|
|
@app.route('/room/<room_name>')
|
|
def room_page(room_name=None):
|
|
if not db.user_has_room_perms(1,room_name):
|
|
abort(403)
|
|
|
|
sensor_list = db.get_sensors_in_room(room_name)
|
|
fig = make_subplots(rows=1, cols=len(sensor_list))
|
|
for idx, sensorID in enumerate(sensor_list):
|
|
lst = [x for x in db.get_sensor_data(sensorID)]
|
|
fig.add_trace(
|
|
go.Scatter(
|
|
x = [x[0] for x in lst],
|
|
y = [x[1] for x in lst],
|
|
name=db.get_sensor_type(sensorID)),
|
|
row = 1,
|
|
col = idx + 1
|
|
)
|
|
fig.update_layout(title_text=f"Available Devices in room {room_name}:")
|
|
|
|
template_path = f'web{sep}room_template.html'
|
|
px_jinja_data = {"fig":fig.to_html(full_html=False)}
|
|
with open(template_path,'r') as template_file:
|
|
j2_template = Template(template_file.read())
|
|
return j2_template.render(px_jinja_data) |