1
0
Fork 0
This repository has been archived on 2024-02-17. You can view files and clone it, but cannot push or open issues or pull requests.
Kfet-depreciated/app/routes.py

246 lines
6.3 KiB
Python
Raw Normal View History

2020-02-01 16:43:22 +01:00
import datetime
import functools
2020-01-26 21:24:49 +01:00
from flask import render_template, redirect, url_for, flash, request
from flask_login import current_user, login_user, logout_user, login_required
from flask_socketio import emit, disconnect
2020-01-26 21:24:49 +01:00
from werkzeug.urls import url_parse
2020-02-01 16:43:22 +01:00
from app import app, socketio, db
2020-01-26 21:24:49 +01:00
from app.forms import LoginForm
2020-02-01 16:43:22 +01:00
from app.models import User, Command, Plate, Ingredient, Sauce, Drink, Dessert
2020-01-26 21:24:49 +01:00
def authenticated_only(f):
@functools.wraps(f)
def wrapped(*args, **kwargs):
if not current_user.is_authenticated:
disconnect()
else:
return f(*args, **kwargs)
return wrapped
2020-01-26 21:24:49 +01:00
@app.route("/login", methods=["GET", "POST"])
def login():
if current_user.is_authenticated:
return redirect(url_for("index"))
form = LoginForm()
if form.validate_on_submit():
user = User.query.filter_by(username=form.username.data).first()
if user is None or not user.check_password(form.password.data):
flash("Invalid username or password")
return redirect(url_for("login"))
login_user(user, remember=form.remember_me.data)
next_page = request.args.get("next")
if not next_page or url_parse(next_page).netloc != "":
next_page = url_for("index")
return redirect(next_page)
return render_template("login.html", form=form)
@app.route("/logout")
def logout():
logout_user()
return redirect(url_for("index"))
@app.route("/")
@app.route("/index")
def index():
return render_template("test.html")
@app.route("/pc")
@login_required
2020-01-27 00:22:42 +01:00
def pc():
2020-01-26 21:24:49 +01:00
return render_template("pc.html")
2020-01-27 00:22:42 +01:00
@app.route("/stocks")
@login_required
def stocks():
return render_template("stocks.html")
@app.route("/menu")
@login_required
def menu():
return render_template("menu.html")
2020-02-01 16:43:22 +01:00
def command_json(c):
content = " - ".join([s.id for s in c.content])
sauces = " - ".join([s.id for s in c.sauce])
if c.error:
state = "error"
elif c.give:
state = "gave"
elif c.done:
state = "done"
elif c.take:
state = "waiting"
else:
state = "unknown"
return {"id": c.number, "plate": c.plate_id, "content": content, "sauce": sauces, "drink": c.drink_id, "dessert": c.dessert_id, "state": state}
2020-01-27 00:22:42 +01:00
2020-01-26 21:24:49 +01:00
@socketio.on("connect")
@authenticated_only
def connect():
print("New connection")
emit("connect", "ok")
@socketio.on("list command")
@authenticated_only
def lscmd():
2020-02-01 16:43:22 +01:00
commands = []
for c in Command.query.filter_by(date=datetime.datetime.now().date()).all():
commands.append(command_json(c))
emit("list command", {"list": commands})
2020-01-27 00:22:42 +01:00
@socketio.on("add command")
@authenticated_only
2020-01-27 00:22:42 +01:00
def addcmd(json):
2020-02-01 16:43:22 +01:00
c = Command()
try:
c.number = Command.query.filter_by(date=datetime.datetime.now().date()).order_by(Command.number.desc()).first().number+1
except AttributeError:
c.number = 1
if "pc" in json:
try:
c.pc_id = User.query.get(json["pc"]).id
except AttributeError:
c.pc_id = 0
if "sandwitch" in json:
try:
c.sandwitch_id = User.query.get(json["sandwitch"]).id
except AttributeError:
c.sandwitch_id = 0
if "client" in json:
try:
c.client_id = User.query.get(json["client"]).id
except AttributeError:
c.client_id = 0
if "plate" in json:
try:
c.plate_id = Plate.query.get(json["plate"]).id
except AttributeError:
pass
if "content" in json:
for i in json["content"]:
try:
c.content.append(Ingredient.query.get(i))
except AttributeError:
pass
if "sauce" in json:
for s in json["sauce"]:
try:
c.sauce.append(Sauce.guery.get(s))
except AttributeError:
pass
if "drink" in json:
try:
c.drink_id = Drink.query.get(json["drink"]).id
except AttributeError:
pass
if "dessert" in json:
try:
c.dessert_id = Dessert.query.get(json["dessert"]).id
except AttributeError:
pass
db.session.add(c)
db.session.commit()
emit("new command", command_json(c), broadcast=True)
2020-01-26 21:24:49 +01:00
2020-01-27 00:22:42 +01:00
@socketio.on("clear command")
@authenticated_only
2020-01-27 00:22:42 +01:00
def rmcmd(json):
2020-02-01 16:43:22 +01:00
c = Command.query.get(json["id"])
if c:
c.done = None
c.give = None
c.error = False
db.session.commit()
emit("cleared command", {"id": json["id"]}, broadcast=True)
@socketio.on("done command")
@authenticated_only
def donecmd(json):
c = Command.query.get(json["id"])
if c:
c.done = datetime.datetime.now().time()
db.session.commit()
emit("finish command", {"id": json["id"]}, broadcast=True)
2020-01-26 21:24:49 +01:00
2020-01-27 00:22:42 +01:00
@socketio.on("give command")
@authenticated_only
2020-01-27 00:22:42 +01:00
def givecmd(json):
2020-02-01 16:43:22 +01:00
c = Command.query.get(json["id"])
if c:
c.give = datetime.datetime.now().time()
db.session.commit()
emit("gave command", {"id": json["id"]}, broadcast=True)
2020-01-26 21:24:49 +01:00
2020-01-27 00:22:42 +01:00
@socketio.on("error command")
@authenticated_only
2020-01-27 00:22:42 +01:00
def errcmd(json):
2020-02-01 16:43:22 +01:00
c = Command.query.get(json["id"])
if c:
c.error = True
db.session.commit()
emit("glitched command", {"id": json["id"]}, broadcast=True)
2020-02-01 17:40:01 +01:00
@socketio.on("list plate")
@authenticated_only
def lsplate():
plates = []
for p in Plate.query.all():
plates.append({"id": p.id, "name": p.name})
emit("list plate", {"list": plates})
@socketio.on("list ingredient")
@authenticated_only
def lsingredient():
ingredients = []
for p in Ingredient.query.all():
ingredients.append({"id": p.id, "name": p.name})
emit("list ingredient", {"list": ingredients})
@socketio.on("list sauce")
@authenticated_only
def lssauce():
sauces = []
for p in Sauce.query.all():
sauces.append({"id": p.id, "name": p.name})
emit("list sauce", {"list": sauces})
@socketio.on("list drink")
@authenticated_only
def lsdrink():
drinks = []
for p in Drink.query.all():
drinks.append({"id": p.id, "name": p.name})
emit("list drink", {"list": drinks})
@socketio.on("list dessert")
@authenticated_only
def lsdessert():
desserts = []
for p in Dessert.query.all():
desserts.append({"id": p.id, "name": p.name})
emit("list dessert", {"list": desserts})