Add BreakManager for prefetching

aleksis/apps/chronos/managers.py

@@ -199,6 +199,14 @@ class ExtraLessonManager(CurrentSiteManager):
         )
 
 
+class BreakManager(CurrentSiteManager):
+    """Manager adding specific methods to breaks."""
+
+    def get_queryset(self):
+        """Ensure all related data is loaded as well."""
+        return super().get_queryset().select_related("before_period", "after_period")
+
+
 class WeekQuerySetMixin:
     def annotate_week(self, week: Union[CalendarWeek]):
         """Annotate all lessons in the QuerySet with the number of the provided calendar week."""

aleksis/apps/chronos/models.py

@@ -23,6 +23,7 @@ from django_global_request.middleware import get_request
 
 from aleksis.apps.chronos.managers import (
     AbsenceQuerySet,
+    BreakManager,
     CurrentSiteManager,
     EventQuerySet,
     ExtraLessonQuerySet,
@@ -750,6 +751,8 @@ class SupervisionArea(ExtensibleModel):
 
 
 class Break(ValidityRangeRelatedExtensibleModel):
+    objects = BreakManager()
+
     short_name = models.CharField(verbose_name=_("Short name"), max_length=255)
     name = models.CharField(verbose_name=_("Long name"), max_length=255)