73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
from flask import render_template, redirect
|
|
|
|
from estusshots import app
|
|
from estusshots import forms, models, orm
|
|
from estusshots.util import authorize
|
|
from estusshots.orm import Drink
|
|
|
|
|
|
@app.route("/drink")
|
|
@authorize
|
|
def drink_list():
|
|
db = orm.new_session()
|
|
drinks = db.query(Drink).order_by(Drink.name).all()
|
|
model = {
|
|
"drinks": drinks,
|
|
"columns": [("name", "Drink Name"), ("vol", "Alcohol %")],
|
|
"controls": [("edit", "Edit")],
|
|
}
|
|
return render_template("drink_list.html", model=model)
|
|
|
|
|
|
@app.route("/drink/<drink_id>/edit", methods=["GET"])
|
|
@authorize
|
|
def drink_edit(drink_id: int):
|
|
db = orm.new_session()
|
|
drink = db.query(Drink).filter(Drink.id == drink_id).first()
|
|
form = forms.DrinkForm()
|
|
form.drink_id.data = drink.id
|
|
form.name.data = drink.name
|
|
form.vol.data = drink.vol
|
|
|
|
model = models.GenericFormModel(
|
|
page_title="Edit Drink",
|
|
form_title=f'Edit Drink "{drink.name}"',
|
|
post_url="/drink/save",
|
|
)
|
|
|
|
return render_template("generic_form.html", model=model, form=form)
|
|
|
|
|
|
@app.route("/drink/new", methods=["GET"])
|
|
@authorize
|
|
def new_drink():
|
|
form = forms.DrinkForm()
|
|
model = models.GenericFormModel(
|
|
page_title="New Drink",
|
|
form_title=f"Create a new Drink",
|
|
post_url="/drink/save",
|
|
)
|
|
return render_template("generic_form.html", model=model, form=form)
|
|
|
|
|
|
@app.route("/drink/save", methods=["POST"])
|
|
@authorize
|
|
def drink_save():
|
|
form = forms.DrinkForm()
|
|
if form.validate_on_submit():
|
|
drink_id = int(form.drink_id.data) if form.drink_id.data else None
|
|
db = orm.new_session()
|
|
if drink_id:
|
|
drink = db.query(Drink).filter(Drink.id == drink_id).first()
|
|
else:
|
|
drink = Drink()
|
|
db.add(drink)
|
|
drink.populate_from_form(form)
|
|
err = db.commit()
|
|
return redirect("/drink")
|
|
|
|
model = models.GenericFormModel(
|
|
page_title="Drinks", form_title="Edit Drink", post_url="/drink/save"
|
|
)
|
|
return render_template("generic_form.html", model=model, form=form)
|