From dc235442f6601db27ca7ce81d7e37aaf42ec9f6c Mon Sep 17 00:00:00 2001
From: Jonathan Weth <git@jonathanweth.de>
Date: Mon, 25 Nov 2024 18:09:05 +0100
Subject: [PATCH] Add query optimizer and remove debug print

---
 aleksis/apps/chronos/schema/__init__.py | 9 ++++++---
 aleksis/apps/chronos/util/build.py      | 1 -
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/aleksis/apps/chronos/schema/__init__.py b/aleksis/apps/chronos/schema/__init__.py
index d2afca66..898de1e3 100644
--- a/aleksis/apps/chronos/schema/__init__.py
+++ b/aleksis/apps/chronos/schema/__init__.py
@@ -1,4 +1,5 @@
 import graphene
+import graphene_django_optimizer
 from graphene_django import DjangoObjectType
 
 from aleksis.core.models import Group, Person, Room
@@ -241,13 +242,15 @@ class Query(graphene.ObjectType):
     timetable_days = graphene.List(graphene.Int)
 
     def resolve_timetable_teachers(self, info, **kwargs):
-        return get_teachers(info.context.user)
+        return graphene_django_optimizer.query(get_teachers(info.context.user), info)
 
     def resolve_timetable_groups(self, info, **kwargs):
-        return get_groups(info.context.user, request=info.context)
+        return graphene_django_optimizer.query(
+            get_groups(info.context.user, request=info.context), info
+        )
 
     def resolve_timetable_rooms(self, info, **kwargs):
-        return get_rooms(info.context.user)
+        return graphene_django_optimizer.query(get_rooms(info.context.user), info)
 
     def resolve_available_timetables(self, info, **kwargs):
         all_timetables = []
diff --git a/aleksis/apps/chronos/util/build.py b/aleksis/apps/chronos/util/build.py
index 66168ce2..a84e4800 100644
--- a/aleksis/apps/chronos/util/build.py
+++ b/aleksis/apps/chronos/util/build.py
@@ -37,7 +37,6 @@ def build_substitutions_list(wanted_day: date) -> tuple[list[dict], set[Person],
         params={"amending": True},
         with_reference_object=True,
     )
-    print(supervision_events)
 
     for supervision_event in supervision_events:
         affected_teachers.update(supervision_event["REFERENCE_OBJECT"].teachers.all())
-- 
GitLab