Files
ProjectIOT/application/dashboard/views.py
2025-06-03 11:21:18 +02:00

76 lines
2.2 KiB
Python

from flask import (
Blueprint,
render_template,
request,
jsonify,
flash,
send_from_directory,
)
from application.dashboard.models import AllowedPlate, LoggedItem
from application import db
from application.dashboard.forms import npForm
from flask_login import login_required
dash_blueprint = Blueprint("dash", __name__, template_folder="templates")
@dash_blueprint.route("/dashboard")
@login_required
def dashboard():
Plates = AllowedPlate.query.all()
logs = (
LoggedItem.query.order_by(LoggedItem.dateLogged.desc()).limit(50).all()
)
return render_template("dashboard.html", plates=Plates, logs=logs)
@dash_blueprint.route("/add", methods=["GET", "POST"])
@login_required
def add():
Plates = AllowedPlate.query.all()
form = npForm()
if form.validate_on_submit():
plate = form.numberplate.data
if plate: # To prevent red lines in VSCode
# Check if number plate already exists
if AllowedPlate.query.filter_by(plate=plate).first():
flash("Numberplate is already registered")
else: # NP does not exist
ap = AllowedPlate(plate=plate)
db.session.add(ap)
db.session.commit()
flash("Numberplate succesfully added")
# Update the list on the page with JavaScript
if request.headers.get("X-Requested-With") == "XMLHttpRequest":
plates = AllowedPlate.query.order_by(AllowedPlate.id).all()
return jsonify(
{
"plates": [
{"id": p.id, "plate": p.plate} for p in plates
]
}
)
# form wasn't valid
return render_template(
"add.html", form=npForm(formdata=None), plates=Plates
)
@dash_blueprint.route("/live", methods=["GET"])
@login_required
def live():
return render_template("live.html")
@dash_blueprint.route("/live_image", methods=["GET"])
@login_required
def live_image():
return send_from_directory(
"static/live",
"feed.png",
as_attachment=False,
conditional=True,
)