from flask import flash from flask_admin.contrib.sqla.fields import QuerySelectField, QuerySelectMultipleField from flask_security import current_user from sqlalchemy import and_ from wtforms import BooleanField, Form, RadioField, SelectField, StringField from wtforms.fields import DateField from wtforms.validators import NumberRange from advlabdb import assistantSpace, db from advlabdb.customClasses import SecureAssistantModelView from advlabdb.exceptions import DataBaseException, ModelViewException from advlabdb.models import ( Appointment, Assistant, Experiment, ExperimentMark, Group, GroupExperiment, Part, PartStudent, Program, Role, Semester, SemesterExperiment, Student, User, ) from advlabdb.utils import initActiveSemesterMenuLinks, userActiveSemester class AssistantAppointmentView(SecureAssistantModelView): column_list = [ "date", "special", "group_experiment.semester_experiment.experiment", "group_experiment.group", "group_experiment.group.part_students", ] column_labels = { "group_experiment.semester_experiment.experiment": "Experiment", "group_experiment.group": "Group", "group_experiment.group.part_students": "Students", } column_editable_list = [ "date", ] def queryFilter(self): return and_( Appointment.group_experiment.has( GroupExperiment.semester_experiment.has(SemesterExperiment.semester == userActiveSemester()) ), Appointment.assistant == current_user.assistant, ) class AssistantExperimentMarkView(SecureAssistantModelView): column_list = [ "oral_mark", "protocol_mark", "final_experiment_mark", "group_experiment.semester_experiment.experiment", "part_student.student.first_name", "part_student.student.last_name", "part_student.group", "part_student.student.uni_email", "part_student.student.contact_email", "part_student.part", "assistant", "edited_by_admin", ] column_labels = { "group_experiment.semester_experiment.experiment": "Experiment", "part_student.student.first_name": "First Name", "part_student.student.last_name": "Last Name", "part_student.group": "Group", "part_student.student.uni_email": "Uni Email", "part_student.student.contact_email": "Contact Email", "part_student.part": "Part", } column_descriptions = { "oral_mark": "Between 0 and 15", "protocol_mark": "Between 0 and 15", "final_experiment_mark": "Calculated automatically with oral and protocol marks and weightings", "part_student.student.contact_email": "The preferred contact email address if entered by the student", "assistant": "The assistant who assigned the mark", "edited_by_admin": "If the mark was edited by an admin", } column_editable_list = [ "oral_mark", "protocol_mark", ] form_args = { "oral_mark": {"validators": [NumberRange(0, 15)]}, "protocol_mark": {"validators": [NumberRange(0, 15)]}, } column_default_sort = [("oral_mark", False), ("protocol_mark", False)] def queryFilter(self): return ExperimentMark.group_experiment.has( GroupExperiment.semester_experiment.has( and_( SemesterExperiment.semester == userActiveSemester(), SemesterExperiment.assistants.any(Assistant.user == current_user), ) ) ) def update_model(self, form, model): if (form.oral_mark and form.oral_mark.data != model.oral_mark) or ( form.protocol_mark and form.protocol_mark.data != model.protocol_mark ): model.assistant = current_user.assistant ret = super().update_model(form, model) model.part_student.checkThenSetFinalPartMark() return ret else: # Nothing changed return True class AssistantUserView(SecureAssistantModelView): column_sortable_list = [] column_list = [ "email", "phone_number", "mobile_phone_number", "room", "building", "assistant.semester_experiments", ] column_labels = { "assistant.semester_experiments": "Semester Experiments", } column_editable_list = [ "phone_number", "mobile_phone_number", "room", "building", ] def queryFilter(self): return User.id == current_user.id assistantSpace.add_view( AssistantAppointmentView(Appointment, db.session, endpoint="assistant_appointment", url="appointment") ) assistantSpace.add_view( AssistantExperimentMarkView(ExperimentMark, db.session, endpoint="assistant_experimentmark", url="experimentmark") ) assistantSpace.add_view(AssistantUserView(User, db.session, endpoint="assistant_user", url="user")) initActiveSemesterMenuLinks(assistantSpace)