mirror of
https://codeberg.org/Mo8it/AdvLabDB.git
synced 2024-09-19 18:31:16 +00:00
113 lines
3.2 KiB
Python
113 lines
3.2 KiB
Python
"""
|
|
Functions dependent on advlabdb.models.
|
|
"""
|
|
|
|
from functools import cache
|
|
from os import environ
|
|
|
|
from flask import flash
|
|
from flask_admin.menu import MenuLink
|
|
from flask_security import current_user
|
|
from wtforms.fields import BooleanField, IntegerField, SelectField, StringField
|
|
from wtforms.validators import DataRequired, NumberRange, Optional
|
|
|
|
from . import app
|
|
from .models import MAX_MARK, MIN_MARK, Semester
|
|
|
|
|
|
def initActiveSemesterMenuLinks(space):
|
|
with app.app_context():
|
|
try:
|
|
semesters = Semester.sortedSemestersStartingWithNewest()
|
|
for semester in semesters:
|
|
space.add_link(
|
|
MenuLink(
|
|
name=str(semester),
|
|
# Using SERVER_NAME because of missing request context
|
|
url=environ["SERVER_NAME"] + "/set_semester" + "?semester_id=" + str(semester.id),
|
|
category="Active semester",
|
|
)
|
|
)
|
|
except Exception:
|
|
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=environ["SERVER_NAME"] + "/logout"))
|
|
|
|
|
|
def active_semester_str():
|
|
active_semester = current_user.active_semester
|
|
active_semester_str = str(active_semester)
|
|
if active_semester != Semester.lastSemester():
|
|
flash(f"You are in the old semester {active_semester_str}!", "warning")
|
|
|
|
return active_semester_str
|
|
|
|
|
|
def mark_field(mark_type: str):
|
|
return IntegerField(
|
|
mark_type + " Mark",
|
|
validators=[Optional(), NumberRange(MIN_MARK, MAX_MARK)],
|
|
description=f"Between {MIN_MARK} and {MAX_MARK}.",
|
|
)
|
|
|
|
|
|
@cache
|
|
def selection_mark_field_choices():
|
|
choices = [(mark, str(mark)) for mark in range(MAX_MARK, MIN_MARK - 1, -1)]
|
|
choices.insert(0, (-1, "Not set yet"))
|
|
|
|
return choices
|
|
|
|
|
|
def selection_mark_field(mark_type: str, default):
|
|
choices = selection_mark_field_choices()
|
|
|
|
if default is None:
|
|
default = -1
|
|
|
|
return SelectField(
|
|
mark_type + " Mark",
|
|
default=default,
|
|
choices=choices,
|
|
validators=[DataRequired()],
|
|
render_kw={"class": "form-control", "style": "width:auto;"},
|
|
)
|
|
|
|
|
|
def parse_selection_mark_field(field):
|
|
data = int(field.data)
|
|
if data == -1:
|
|
return None
|
|
|
|
return data
|
|
|
|
|
|
def user_info_fields():
|
|
phone_number = StringField(
|
|
"Phone Number",
|
|
validators=[Optional()],
|
|
)
|
|
mobile_phone_number = StringField(
|
|
"Mobile Phone Number",
|
|
validators=[Optional()],
|
|
)
|
|
|
|
building = StringField(
|
|
"Building",
|
|
validators=[Optional()],
|
|
)
|
|
room = StringField(
|
|
"Room",
|
|
validators=[Optional()],
|
|
)
|
|
|
|
return phone_number, mobile_phone_number, building, room
|
|
|
|
|
|
def generate_new_password_field():
|
|
return BooleanField(
|
|
"Generate new random password. For security reasons, it is not possible to manually enter a password. Please use a password manager like Bitwarden or KeepassXC to save the randomly generated password.",
|
|
default=False,
|
|
)
|