diff --git a/aleksis/apps/alsijil/model_extensions.py b/aleksis/apps/alsijil/model_extensions.py index fddde6d59a487d2a48bed95c3eb39e27071d43b0..7881eb9f995d8d59be15af19833cb0edda07e096 100644 --- a/aleksis/apps/alsijil/model_extensions.py +++ b/aleksis/apps/alsijil/model_extensions.py @@ -177,10 +177,14 @@ def get_lesson_documentation( if not week: week = self.week # Use all to make effect of prefetched data - for documentation in self.documentations.all(): - if documentation.week == week.week and documentation.year == week.year: - return documentation - return None + doc_filter = filter( + lambda d: d.week == week.week and d.year == week.year, + self.dopycumentations.all(), + ) + try: + return next(doc_filter) + except StopIteration: + return None @LessonPeriod.method