Move get_for_coursebook into schema

aleksis/apps/alsijil/models.py

@@ -568,46 +568,6 @@ class Documentation(CalendarEvent):
 
         return (docs, dummies)
 
-    @classmethod
-    def get_for_coursebook(
-        cls,
-        own: bool,
-        date_start: datetime,
-        date_end: datetime,
-        request: HttpRequest,
-        obj_type: Optional[str] = None,
-        obj_id: Optional[str] = None,
-        incomplete: Optional[bool] = False,
-    ) -> list:
-        """Get all the documentations for an object and a time frame.
-
-        obj_type may be one of TEACHER, GROUP, ROOM, COURSE
-        """
-
-        # Find all LessonEvents for all Lessons of this Course in this date range
-        event_params = {
-            "own": own,
-        }
-        if obj_type is not None and obj_id is not None:
-            event_params.update(
-                {
-                    "type": obj_type,
-                    "id": obj_id,
-                }
-            )
-
-        events = LessonEvent.get_single_events(
-            date_start,
-            date_end,
-            request,
-            event_params,
-            with_reference_object=True,
-        )
-
-        # Lookup or create documentations and return them all.
-        docs, dummies = cls.get_documentations_for_events(events, incomplete)
-        return docs + dummies
-
 
 class ParticipationStatus(CalendarEvent):
     """A participation or absence record about a single person.

aleksis/apps/alsijil/schema/__init__.py

@@ -63,9 +63,6 @@ class Query(graphene.ObjectType):
         incomplete=False,
         **kwargs,
     ):
-        datetime_start = datetime.combine(date_start, datetime.min.time())
-        datetime_end = datetime.combine(date_end, datetime.max.time())
-
         if (
             (
                 obj_type == "COURSE"
@@ -88,10 +85,30 @@ class Query(graphene.ObjectType):
         ):
             raise PermissionDenied()
 
-        return Documentation.get_for_coursebook(
-            own, datetime_start, datetime_end, info.context, obj_type, obj_id, incomplete
+        # Find all LessonEvents for all Lessons of this Course in this date range
+        event_params = {
+            "own": own,
+        }
+        if obj_type is not None and obj_id is not None:
+            event_params.update(
+                {
+                    "type": obj_type,
+                    "id": obj_id,
+                }
+            )
+
+        events = LessonEvent.get_single_events(
+            datetime.combine(date_start, datetime.min.time()),
+            datetime.combine(date_end, datetime.max.time()),
+            info.context,
+            event_params,
+            with_reference_object=True,
         )
 
+        # Lookup or create documentations and return them all.
+        docs, dummies = Documentation.get_documentations_for_events(events, incomplete)
+        return docs + dummies
+
     @staticmethod
     def resolve_groups_by_person(root, info, person=None):
         if person: