AdvLabDB/advlabdb/utils.py

from random import choice
from string import ascii_letters, digits

from flask import flash, url_for
from flask_admin.menu import MenuLink
from flask_security import current_user

from advlabdb import app, db
from advlabdb.models import Semester


def randomPassword():
    return "".join(choice(ascii_letters + digits) for i in range(12))


def flashRandomPassword(password):
    flash(f"Random password: {password}", category="warning")


def userActiveSemester(flashWarning=False):
    lastSemesterId = (
        Semester.query.order_by(Semester.year.desc()).order_by(Semester.label.desc()).first().id
    )  # "WS" > "SS"
    if current_user.active_semester_id is None:
        current_user.active_semester_id = lastSemesterId
        db.session.commit()
    elif current_user.active_semester_id != lastSemesterId:
        activeSemester = Semester.query.get(current_user.active_semester_id)
        if activeSemester:
            if flashWarning:
                flash(f"You are in the old semester {activeSemester.repr()}!", "warning")
        else:
            current_user.active_semester_id = lastSemesterId
            db.session.commit()
            flash("Semester changed!", "warning")

    return Semester.query.get(current_user.active_semester_id)


def setUserActiveSemester(semesterId):
    if current_user.active_semester_id != semesterId:
        semester = Semester.query.get(semesterId)
        if semester:
            try:
                current_user.active_semester_id = semesterId
                db.session.commit()
            except Exception as ex:
                flash(str(ex), "error")

                db.session.rollback()
            else:
                flash(f"Active semester changed to {semester.repr()}!", "warning")
        else:
            flash(f"No semester with ID {semesterId}!", "error")


def initActiveSemesterMenuLinks(space):
    with app.app_context():
        semesters = Semester.query.order_by(Semester.id.desc())
        for semester in semesters:
            space.add_link(
                MenuLink(
                    name=semester.repr(),
                    url=url_for("set_semester") + "?semester_id=" + str(semester.id),
                    category="Active semester",
                )
            )

        space.add_link(MenuLink(name="Logout", url=url_for("security.logout")))


def reportBadAttempt(message):
    print("BAD ATTEMPT:", message)  # TODO: Log