Benchmarks crud

app/admin/main_admin.py

@@ -7,7 +7,7 @@ from flask import (
     current_app,
 )
 from flask_login import current_user, login_required
-from .forms import UserForm, UpdatePasswordForm
+from .forms import UserForm, UpdatePasswordForm, BenchmarkForm
 from ..models import User, Benchmark, db
 
 admin = Blueprint("admin", __name__, template_folder="templates")
@@ -55,17 +55,12 @@ def user_edit(user_id):
 def user_delete(user_id):
     if user_id != current_user.id and not current_user.admin:
         flash("You are not an admin.", "danger")
-        return redirect(url_for(INDEX))
+        return redirect(url_for(current_app.config["INDEX"]))
     user = User.query.filter_by(id=user_id).first()
     form = UserForm(obj=user)
     del form.password
     del form.password2
-    for field in form:
-        if field.type != "SubmitField" and field.type != "CSRFTokenField":
-            if field.type == "SelectField" or field.type == "BooleanField":
-                field.render_kw = {"disabled": True}
-            else:
-                field.render_kw = {"readonly": True}
+    disable_fields(form)
 
     form.benchmark_id.choices = [
         (b.id, b.name) for b in Benchmark.query.order_by("name")
@@ -90,7 +85,7 @@ def user_delete(user_id):
 def user_new():
     if not current_user.admin:
         flash("You are not an admin.", "danger")
-        return redirect(url_for(INDEX))
+        return redirect(url_for(current_app.config["INDEX"]))
     form = UserForm()
     user = User()
     form.user_id = None
@@ -137,3 +132,88 @@ def password(user_id, back):
     return render_template(
         "password.html", form=form, back=destination, user_name=user.username
     )
+
+
+@admin.route("/benchmarks")
+@login_required
+def benchmarks():
+    if not current_user.admin:
+        flash("You are not an admin.", "danger")
+        return redirect(url_for(current_app.config["INDEX"]))
+    benchmarks = Benchmark.query.all()
+    return render_template("benchmarks.html", benchmarks=benchmarks)
+
+
+@admin.route("/benchmark_edit/<int:benchmark_id>", methods=["GET", "POST"])
+@login_required
+def benchmark_edit(benchmark_id):
+    if not current_user.admin:
+        flash("You are not an admin.", "danger")
+        return redirect(url_for(current_app.config["INDEX"]))
+    form = BenchmarkForm(
+        obj=Benchmark.query.filter_by(id=benchmark_id).first()
+    )
+    form.submit.label.text = "Edit Benchmark"
+    if form.validate_on_submit():
+        form.populate_obj(Benchmark.query.filter_by(id=benchmark_id).first())
+        db.session.commit()
+        flash("Benchmark edited successfully.")
+        return redirect(url_for("admin.benchmarks"))
+    return render_template(
+        "benchmark.html",
+        form=form,
+        alert_type="primary",
+        title="Edit Benchmark",
+    )
+
+
+def disable_fields(form):
+    for field in form:
+        if field.type != "SubmitField" and field.type != "CSRFTokenField":
+            if field.type == "SelectField" or field.type == "BooleanField":
+                field.render_kw = {"disabled": True}
+            else:
+                field.render_kw = {"readonly": True}
+
+
+@admin.route("/benchmark_delete/<int:benchmark_id>", methods=["GET", "POST"])
+@login_required
+def benchmark_delete(benchmark_id):
+    if not current_user.admin:
+        flash("You are not an admin.", "danger")
+        return redirect(url_for(current_app.config["INDEX"]))
+    benchmark = Benchmark.query.filter_by(id=benchmark_id).first()
+    form = BenchmarkForm(obj=benchmark)
+    disable_fields(form)
+    form.submit.label.text = "Delete Benchmark"
+    if form.validate_on_submit():
+        flash("Benchmark deleted successfully.")
+        db.session.delete(benchmark)
+        db.session.commit()
+        return redirect(url_for("admin.benchmarks"))
+    return render_template(
+        "benchmark.html",
+        form=form,
+        alert_type="danger",
+        title="Delete Benchmark",
+    )
+
+
+@admin.route("/benchmark_new", methods=["GET", "POST"])
+@login_required
+def benchmark_new():
+    if not current_user.admin:
+        flash("You are not an admin.", "danger")
+        return redirect(url_for(current_app.config["INDEX"]))
+    form = BenchmarkForm()
+    benchmark = Benchmark()
+    form.submit.label.text = "New Benchmark"
+    if form.validate_on_submit():
+        form.populate_obj(benchmark)
+        db.session.add(benchmark)
+        db.session.commit()
+        flash("Benchmark created successfully.")
+        return redirect(url_for("admin.benchmarks"))
+    return render_template(
+        "benchmark.html", form=form, alert_type="info", title="New Benchmark"
+    )