AdvLabDB/advlabdb/routes.py

from flask import flash, redirect, render_template, request, url_for
from flask_security import auth_required, current_user

from advlabdb import app, db
from advlabdb.models import Semester
from advlabdb.utils import setActiveSemester, titleToTemplate, userActiveSemester


@app.context_processor
def util_processor():
    def semesterDropDownItems():
        semesters = Semester.query.all()
        items = []
        for semester in semesters:
            items.append(
                '<li><a class="dropdown-item" href="'
                + url_for("set_semester")
                + "?semester_id="
                + str(semester.id)
                + '">'
                + semester.label
                + "</a></li>"
            )
        return items

    def navbarItems(title):
        adminPages = []
        assistantPages = []

        if current_user.is_authenticated:
            roleNames = [role.name for role in current_user.roles]
            if "assistant" in roleNames:
                pages = assistantPages
                if "admin" in roleNames:
                    for p in adminPages:
                        if p not in pages:
                            pages.append(p)
            elif "admin" in roleNames:
                pages = adminPages

            pages.append("Logout")
        else:
            pages = ["Login"]

        items = []
        for page in pages:
            if page == title:
                active = " active"
            else:
                active = ""

            items.append('<a class="nav-link' + active + '" href="' + titleToTemplate(page) + '">' + page + "</a>")
        return items

    return dict(
        semesterDropDownItems=semesterDropDownItems,
        userActiveSemester=userActiveSemester,
        navbarItems=navbarItems,
        current_user=current_user,
    )


@app.route("/")
def index():
    if current_user.has_role("admin"):
        return redirect(url_for("index") + "admin")
    elif current_user.has_role("assistant"):
        return redirect(url_for("index") + "assistant")
    else:
        return render_template("index.html")


"""
@app.route("/assistants")
@roles_required("admin")
def assistants():
    headerAndDataList = [["First name", "row.first_name"],
                         ["Last name", "row.last_name"],
                         ["Email", "row.email"],
                         ["Phone number", "row.phone_number"],
                         ["Mobile phone number", "row.mobile_phone_number"],
                         ["Room", "row.room"],
                         ["Building", "row.building"],
                         ["User", "row.user"],
                         ["Appointments", "row.appointments"],
                         ["Experiment marks", "row.experiment_marks"]]

    table = makeTable(headerAndDataList=headerAndDataList,
                      rows=Assistant.query.filter(Assistant.user_id != None).all())

    return render_template("assistants.html",
                           table=table,
                           )


@app.route("/experiments")
@roles_required("admin")
def experiments():
    semester = Semester.query.get(current_user.active_semester_id)
    parts = semester.parts
    tables = []
    tablesLabels = []

    headerAndDataList = [["Number", "row.experiment.number"],
                         ["Name", "row.experiment.name"],
                         ["Assistants", "row.assistants"],
                         ["Groups with this ex.", "[gEx.group.number for gEx in row.group_experiments]"]]

    for part in parts:
        tablesLabels.append("Part " + part.label + ":")
        tables.append(makeTable(headerAndDataList=headerAndDataList,
                                rows=part.part_experiments))

    return render_template("experiments.html",
                           tables=tables,
                           tablesLabels=tablesLabels,
                           )


@app.route("/appointments")
@roles_required("admin")
def appointments():
    return render_template("appointments.html",
                           )


@app.route("/groups")
@roles_required("admin")
def groups():
    semester = Semester.query.get(current_user.active_semester_id)
    parts = semester.parts
    tables = []
    tablesLabels = []

    headerAndDataList = [["Number", "row.number"],
                         ["Students",
                          "[ps.student.first_name + ' ' + ps.student.last_name for ps in row.part_students]"],
                         ["Experiments (appointments)",
                          "[str(gx.part_experiment.experiment.number) + ' (' + str([appointmentDate(a.date) for a in gx.appointments]) + ')' for gx in row.group_experiments]"]]

    for part in parts:
        tablesLabels.append("Part " + part.label + ":")
        tables.append(makeTable(headerAndDataList=headerAndDataList,
                                rows=part.groups))

    return render_template("groups.html",
                           tables=tables,
                           tablesLabels=tablesLabels,
                           )


@app.route("/deactivate_users", methods=["GET"])
@roles_required("admin")
def deactivate_users():
    usersEmails = json.loads(request.args.get("json"))
    deactivatedUsersEmails = []
    for email in usersEmails:
        user = User.query.filter(User.email == email).first()
        if user == current_user:
            flash("You have tried to deactivate yourself as user!", "danger")
            continue
        if user_datastore.deactivate_user(user):
            deactivatedUsersEmails.append(email)
            db.session.commit()
    deactivatedUsersEmailsLen = len(deactivatedUsersEmails)
    if deactivatedUsersEmailsLen == 0:
        flash(f"No users deactivated!", "warning")
    else:
        flash(f"{deactivatedUsersEmailsLen} Users with emails {deactivatedUsersEmails} deactivated!", "success")
    return redirect(url_for("users"))
"""


@app.route("/set_semester", methods=["GET"])
@auth_required()
def set_semester():
    setActiveSemester(request.args.get("semester_id"))
    ref = request.referrer
    red = ref if ref else url_for("index")
    return redirect(red)