from flask import flash, request, url_for from flask_admin.contrib.sqla.filters import BaseSQLAFilter from flask_admin.menu import MenuLink from flask_security import hash_password from wtforms import BooleanField, SelectField, TextField from wtforms.validators import DataRequired, Email from advlabdb import admin, app, db, user_datastore from advlabdb.configUtils import getConfig from advlabdb.customClasses import SecureModelView from advlabdb.models import ( Appointment, Assistant, Experiment, ExperimentMark, Group, GroupExperiment, Part, PartExperiment, PartStudent, Role, Semester, Student, User, ) from advlabdb.utils import ( partFromLabelInUserActiveSemester, randomPassword, setActiveSemester, userActiveSemester, ) class UserModelView(SecureModelView): column_list = ["email", "active", "roles", "assistant"] column_searchable_list = ["email"] column_filters = ["active"] form_columns = ["email", "active", "roles"] form_args = { "email": {"validators": [Email()]}, "active": {"default": True}, "roles": {"validators": [DataRequired(message="A role is required!")]}, } def create_model(self, form): password = randomPassword() passwordHash = hash_password(password) email = form.email.data.lower() roles = [role.name for role in form.roles.data] if "admin" in roles: flash("You have registered a new admin!", "danger") model = user_datastore.create_user(email=email, password=passwordHash, roles=roles) db.session.commit() flash( f"{email} registered with roles: {', '.join([role.name for role in form.roles.data])}.", category="success" ) flash(f"Random password: {password}", category="warning") return model class RoleModelView(SecureModelView): column_exclude_list = ["update_datetime"] class SemesterModelView(SecureModelView): column_list = ["label", "parts"] form_columns = ["label", "create_parts"] form_extra_fields = { "create_parts": BooleanField("Create parts:" + ", ".join(getConfig("partsLabels")) + ".", default=True) } def after_model_change(self, form, model, is_created): if form.create_parts.data: if is_created == False and model.parts != []: flash("This semester already has parts!", "danger") return model.createParts() if is_created: admin.add_link( MenuLink( name=model.label, url=url_for("set_semester") + "?semester_id=" + str(model.id), category="Active semester", ) ) setActiveSemester(model.id) class PartModelView(SecureModelView): can_view_details = True column_details_list = ["label", "semester", "part_experiments", "part_students", "groups"] form_columns = ["label", "semester"] partsLabels = getConfig("partsLabels") form_choices = {"label": list(zip(partsLabels, partsLabels))} class StudentModelView(SecureModelView): can_view_details = True column_list = ["student_number", "first_name", "last_name", "uni_email", "contact_email", "part_students"] column_details_list = column_list + ["bachelor_thesis", "bachelor_thesis_work_group", "note"] column_searchable_list = ["student_number", "uni_email", "contact_email", "first_name", "last_name"] form_columns = column_details_list + ["new_part_student_part", "new_part_student_group_number"] form_args = { "uni_email": {"validators": [Email()]}, "contact_email": {"validators": [Email()]}, } partChoices = ["-"] + getConfig("partsLabels") form_extra_fields = { "new_part_student_part": SelectField( "Part", choices=list(zip(partChoices, partChoices)), default=partChoices[0] ), "new_part_student_group_number": TextField("Group number"), } def validate_form(self, form): if request.method == "POST": partLabel = form.new_part_student_part.data groupNumber = form.new_part_student_group_number.data if (partLabel != self.partChoices[0] and groupNumber == "") or ( partLabel == self.partChoices[0] and groupNumber != "" ): flash("You have to assign both part and group if you want to add a part student!", "danger") return False if partLabel != self.partChoices[0] and not partFromLabelInUserActiveSemester(partLabel): flash(f"Part {partLabel} is not created in {str(userActiveSemester())} yet!", "danger") return False if groupNumber != "": message = "The group number has to be an integer > 0 !" try: groupNumber = int(groupNumber) except: flash(message, "danger") return False if groupNumber < 1: flash(message, "danger") return False return super().validate_form(form) def after_model_change(self, form, model, is_created): partLabel = form.new_part_student_part.data print("\nLL\n") if partLabel != self.partChoices[0]: groupNumber = int(form.new_part_student_group_number.data) part = partFromLabelInUserActiveSemester(partLabel) group = Group.query.filter(Group.number == groupNumber, Group.part == part).first() if group is None: group = Group(number=groupNumber, part=part) db.session.add(group) flash(f"Added the new group with number {str(groupNumber)} in part {str(part)}.", "success") partStudent = PartStudent(student=model, part=part, group=group) db.session.add(partStudent) db.session.commit() flash("Added part student.", "success") class PartFilter(BaseSQLAFilter): def apply(self, query, value, alias=None): return query.filter(self.column == partFromLabelInUserActiveSemester(value).id) def operation(self): return "equals" def validate(self, value): if partFromLabelInUserActiveSemester(value): return True else: flash(f"Part {value} not found in your active semester {userActiveSemester()}!", "danger") return False class PartStudentModelView(SecureModelView): partsLabels = getConfig("partsLabels") column_filters = [PartFilter(PartStudent.part_id, "Part", options=list(zip(partsLabels, partsLabels)))] form_excluded_columns = ["experiment_marks"] class GroupModelView(SecureModelView): partsLabels = getConfig("partsLabels") column_filters = [PartFilter(Group.part_id, "Part", options=list(zip(partsLabels, partsLabels)))] def validate_form(self, form): if request.method == "POST": if Group.query.filter(Group.number == form.number.data, Group.part == form.part.data).first(): lastTakenGroupNumber = ( Group.query.filter(Group.part == form.part.data).order_by(Group.number)[-1].number ) flash( f"Group number taken in this part! Last group number taken in this part is {lastTakenGroupNumber}.", "warning", ) return False return super().validate_form(form) admin.add_view(StudentModelView(Student, db.session)) admin.add_view(PartStudentModelView(PartStudent, db.session)) admin.add_view(GroupModelView(Group, db.session)) admin.add_view(SecureModelView(GroupExperiment, db.session)) admin.add_view(SecureModelView(Experiment, db.session)) admin.add_view(SecureModelView(PartExperiment, db.session)) admin.add_view(SecureModelView(Assistant, db.session)) admin.add_view(SecureModelView(Appointment, db.session)) admin.add_view(PartModelView(Part, db.session)) admin.add_view(SemesterModelView(Semester, db.session)) admin.add_view(SecureModelView(ExperimentMark, db.session)) admin.add_view(UserModelView(User, db.session)) admin.add_view(RoleModelView(Role, db.session)) with app.app_context(): semesters = Semester.query.all()[::-1] for semester in semesters: admin.add_link( MenuLink( name=semester.label, url=url_for("set_semester") + "?semester_id=" + str(semester.id), category="Active semester", ) )