Added AssistantGroupExperimentView

advlabdb/__init__.py

@@ -54,7 +54,7 @@ user_datastore = SQLAlchemyUserDatastore(db, models.User, models.Role)
 Security(app, user_datastore)
 
 try:
-    from advlabdb import routes, modelViews
+    from advlabdb import routes, adminModelViews, assistantModelViews
 except Exception as ex:
     print(
         "\nYou are probably initializing the database with a script. If not, then you have to worry about not being able to import in __init__.py!\n"

advlabdb/assistantModelViews.py

@@ -0,0 +1,60 @@
+from flask_security import current_user
+
+from advlabdb import assistantSpace, db
+from advlabdb.customClasses import SecureAssistantModelView
+from advlabdb.models import (
+    Appointment,
+    Assistant,
+    Experiment,
+    ExperimentMark,
+    Group,
+    GroupExperiment,
+    Part,
+    SemesterExperiment,
+    PartStudent,
+    Role,
+    Semester,
+    Student,
+    User,
+    Program,
+)
+from advlabdb.utils import userActiveSemester
+from advlabdb.exceptions import ModelViewException, DataBaseException
+
+
+class AssistantGroupExperimentView(SecureAssistantModelView):
+    can_create = False
+    can_edit = False
+    can_delete = False
+    column_display_actions = False
+
+    column_list = ["group", "semester_experiment", "appointments", "experiment_marks"]
+    column_filters = ["group", "semester_experiment.experiment", "appointments"]
+
+    def queryFilter1(self):
+        return GroupExperiment.group_id.in_(
+            [group.id for group in Group.query.filter(Group.semester == userActiveSemester())]
+        )
+
+    def queryFilter2(self):
+        return GroupExperiment.semester_experiment_id.in_(
+            [
+                semesterExperiment.id
+                for semesterExperiment in SemesterExperiment.query.filter(
+                    SemesterExperiment.assistants.any(Assistant.id == current_user.assistant.id)
+                )
+            ]
+        )
+
+    def get_query(self):
+        return super().get_query().filter(self.queryFilter1()).filter(self.queryFilter2())
+
+    def get_count_query(self):
+        return super().get_count_query().filter(self.queryFilter1()).filter(self.queryFilter2())
+
+
+assistantSpace.add_view(
+    AssistantGroupExperimentView(
+        GroupExperiment, db.session, endpoint="assistant_groupexperiment", url="groupexperiment"
+    )
+)