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