majuna/app/portal/group.py

from datetime import datetime, timezone

import sqlalchemy
from flask import Blueprint, Response, flash, redirect, render_template, url_for
from flask.typing import ResponseReturnValue
from flask_wtf import FlaskForm
from wtforms import BooleanField, StringField, SubmitField
from wtforms.validators import DataRequired

from app.extensions import db
from app.models.base import Group

bp = Blueprint("group", __name__)


class NewGroupForm(FlaskForm):  # type: ignore
    group_name = StringField("Short Name", validators=[DataRequired()])
    description = StringField("Description", validators=[DataRequired()])
    eotk = BooleanField("Deploy EOTK instances?")
    submit = SubmitField("Save Changes", render_kw={"class": "btn btn-success"})


class EditGroupForm(FlaskForm):  # type: ignore
    description = StringField("Description", validators=[DataRequired()])
    eotk = BooleanField("Deploy EOTK instances?")
    submit = SubmitField("Save Changes", render_kw={"class": "btn btn-success"})


@bp.route("/list")
def group_list() -> ResponseReturnValue:
    groups = Group.query.order_by(Group.group_name).all()
    return render_template(
        "list.html.j2",
        section="group",
        title="Groups",
        item="group",
        items=groups,
        new_link=url_for("portal.group.group_new"),
    )


@bp.route("/new", methods=["GET", "POST"])
def group_new() -> ResponseReturnValue:
    form = NewGroupForm()
    if form.validate_on_submit():
        group = Group()
        group.group_name = form.group_name.data
        group.description = form.description.data
        group.eotk = form.eotk.data
        group.added = datetime.now(tz=timezone.utc)
        group.updated = datetime.now(tz=timezone.utc)
        try:
            db.session.add(group)
            db.session.commit()
            flash(f"Created new group {group.group_name}.", "success")
            return redirect(url_for("portal.group.group_edit", group_id=group.id))
        except sqlalchemy.exc.SQLAlchemyError:
            flash("Failed to create new group.", "danger")
            return redirect(url_for("portal.group.group_list"))
    return render_template("new.html.j2", section="group", form=form)


@bp.route("/edit/<group_id>", methods=["GET", "POST"])
def group_edit(group_id: int) -> ResponseReturnValue:
    group = Group.query.filter(Group.id == group_id).first()
    if group is None:
        return Response(
            render_template(
                "error.html.j2",
                section="group",
                header="404 Group Not Found",
                message="The requested group could not be found.",
            ),
            status=404,
        )
    form = EditGroupForm(description=group.description, eotk=group.eotk)
    if form.validate_on_submit():
        group.description = form.description.data
        group.eotk = form.eotk.data
        group.updated = datetime.now(tz=timezone.utc)
        try:
            db.session.commit()
            flash("Saved changes to group.", "success")
        except sqlalchemy.exc.SQLAlchemyError:
            flash("An error occurred saving the changes to the group.", "danger")
    return render_template("group.html.j2", section="group", group=group, form=form)