Add prefetch option for get_el_by_pk

aleksis/apps/chronos/util/chronos_helpers.py

@@ -16,9 +16,15 @@ def get_el_by_pk(
     year: Optional[int] = None,
     week: Optional[int] = None,
     regular: Optional[str] = None,
+    prefetch: bool = False,
 ):
     if type_ == TimetableType.GROUP.value:
-        return get_object_or_404(Group, pk=pk)
+        return get_object_or_404(
+            Group.objects.prefetch_related("owners", "parent_groups")
+            if prefetch
+            else Group,
+            pk=pk,
+        )
     elif type_ == TimetableType.TEACHER.value:
         return get_object_or_404(Person, pk=pk)
     elif type_ == TimetableType.ROOM.value:

aleksis/apps/chronos/views.py

@@ -116,7 +116,7 @@ def timetable(
 
     is_smart = regular != "regular"
 
-    el = get_el_by_pk(request, type_, pk)
+    el = get_el_by_pk(request, type_, pk, prefetch=True)
 
     if type(el) == HttpResponseNotFound:
         return HttpResponseNotFound()