mirror of
https://codeberg.org/Mo8it/AdvLabDB.git
synced 2024-11-08 21:21:06 +00:00
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
from flask import flash, url_for
|
|
from flask_admin.menu import MenuLink
|
|
from flask_security import current_user
|
|
|
|
from . import app, db
|
|
from .models import Semester
|
|
|
|
|
|
def flashRandomPassword(password):
|
|
flash(f"Random password: {password}", category="warning")
|
|
|
|
|
|
def sortedSemestersStartingWithNewest():
|
|
return Semester.query.order_by(Semester.year.desc()).order_by(Semester.label.desc())
|
|
|
|
|
|
def userActiveSemester(flashWarning=False):
|
|
lastSemesterId = sortedSemestersStartingWithNewest().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 = db.session.get(Semester, current_user.active_semester_id)
|
|
if activeSemester is not None:
|
|
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 db.session.get(Semester, current_user.active_semester_id)
|
|
|
|
|
|
def setUserActiveSemester(semesterId):
|
|
if current_user.active_semester_id != semesterId:
|
|
semester = db.session.get(Semester, semesterId)
|
|
if semester is not None:
|
|
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():
|
|
try:
|
|
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",
|
|
)
|
|
)
|
|
except Exception as ex:
|
|
print(
|
|
"ERROR: The Semester table does not exist yet! Therefore, menu links could not be generated. You can ignore this error if you are just initializing the database."
|
|
)
|
|
else:
|
|
space.add_link(MenuLink(name="Logout", url=url_for("security.logout")))
|
|
|
|
|
|
def reportBadAttempt(message):
|
|
print("BAD ATTEMPT:", message) # TODO: Log
|