diff --git a/aleksis/apps/alsijil/util/alsijil_helpers.py b/aleksis/apps/alsijil/util/alsijil_helpers.py index f44a90944d8b1e8d6360012b7b818a2f77fd4359..0ea834badf5b646c971ce437a38778021c9a7328 100644 --- a/aleksis/apps/alsijil/util/alsijil_helpers.py +++ b/aleksis/apps/alsijil/util/alsijil_helpers.py @@ -76,12 +76,12 @@ def annotate_documentations( instances = instances.order_by("period__weekday", "period__period") else: instances = instances.order_by("period_from__weekday", "period_from__period") - args = {Event: "event", LessonPeriod: "lesson_period", ExtraLesson: "extra_lesson"} + instances = instances.annotate( has_documentation=Exists( LessonDocumentation.objects.filter( ~Q(topic__exact=""), week=wanted_week.week, year=wanted_week.year, - ).filter(**{args[klass]: OuterRef("pk")}) + ).filter(**{klass.label_: OuterRef("pk")}) ) )