resource pool system

app/portal/pool.py

@@ -0,0 +1,78 @@
+from datetime import datetime
+
+from flask import render_template, url_for, flash, redirect, Response, Blueprint
+from flask.typing import ResponseReturnValue
+from flask_wtf import FlaskForm
+import sqlalchemy
+from wtforms import StringField, SubmitField
+from wtforms.validators import DataRequired
+
+from app.extensions import db
+from app.models.base import Pool
+
+bp = Blueprint("pool", __name__)
+
+
+class NewPoolForm(FlaskForm):  # type: ignore
+    group_name = StringField("Short Name", validators=[DataRequired()])
+    description = StringField("Description", validators=[DataRequired()])
+    submit = SubmitField('Save Changes', render_kw={"class": "btn btn-success"})
+
+
+class EditPoolForm(FlaskForm):  # type: ignore
+    description = StringField("Description", validators=[DataRequired()])
+    submit = SubmitField('Save Changes', render_kw={"class": "btn btn-success"})
+
+
+@bp.route("/list")
+def pool_list() -> ResponseReturnValue:
+    pools = Pool.query.order_by(Pool.pool_name).all()
+    return render_template("list.html.j2",
+                           section="pool",
+                           title="Resource Pools",
+                           item="pool",
+                           items=pools,
+                           new_link=url_for("portal.pool.pool_new"))
+
+
+@bp.route("/new", methods=['GET', 'POST'])
+def pool_new() -> ResponseReturnValue:
+    form = NewPoolForm()
+    if form.validate_on_submit():
+        pool = Pool()
+        pool.pool_name = form.group_name.data
+        pool.description = form.description.data
+        pool.created = datetime.utcnow()
+        pool.updated = datetime.utcnow()
+        try:
+            db.session.add(pool)
+            db.session.commit()
+            flash(f"Created new pool {pool.pool_name}.", "success")
+            return redirect(url_for("portal.pool.pool_edit", pool_id=pool.id))
+        except sqlalchemy.exc.SQLAlchemyError:
+            flash("Failed to create new pool.", "danger")
+            return redirect(url_for("portal.pool.pool_list"))
+    return render_template("new.html.j2", section="pool", form=form)
+
+
+@bp.route('/edit/<pool_id>', methods=['GET', 'POST'])
+def pool_edit(pool_id: int) -> ResponseReturnValue:
+    pool = Pool.query.filter(Pool.id == pool_id).first()
+    if pool is None:
+        return Response(render_template("error.html.j2",
+                                        section="pool",
+                                        header="404 Pool Not Found",
+                                        message="The requested pool could not be found."),
+                        status=404)
+    form = EditPoolForm(description=pool.description)
+    if form.validate_on_submit():
+        pool.description = form.description.data
+        pool.updated = datetime.utcnow()
+        try:
+            db.session.commit()
+            flash("Saved changes to pool.", "success")
+        except sqlalchemy.exc.SQLAlchemyError:
+            flash("An error occurred saving the changes to the pool.", "danger")
+    return render_template("pool.html.j2",
+                           section="pool",
+                           pool=pool, form=form)