1
0
Fork 0
mirror of https://codeberg.org/Mo8it/AdvLabDB.git synced 2024-09-19 18:31:16 +00:00
AdvLabDB/advlabdb/utils.py

49 lines
1.6 KiB
Python

from random import choice
from string import ascii_letters, digits
from flask import flash
from flask_security import current_user
from advlabdb import db
from advlabdb.models import Semester
def randomPassword():
return "".join(choice(ascii_letters + digits) for i in range(12))
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")