From 804350a646291e95e271366d3e83e32479839ac7 Mon Sep 17 00:00:00 2001
From: Dominik George <dominik.george@teckids.org>
Date: Thu, 7 Nov 2019 22:11:24 +0100
Subject: [PATCH] Add model extensions for a person's lesson periods.

---
 biscuit/apps/chronos/model_extensions.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/biscuit/apps/chronos/model_extensions.py b/biscuit/apps/chronos/model_extensions.py
index 25acbd54..b4a6d4ad 100644
--- a/biscuit/apps/chronos/model_extensions.py
+++ b/biscuit/apps/chronos/model_extensions.py
@@ -1,8 +1,18 @@
 from biscuit.core.models import Person
 
-from .models import Lesson
+from .models import Lesson, LessonPeriod
 
 
 @Person.property
 def lessons_as_participant(self):
     return Lesson.objects.filter(groups__members=self)
+
+
+@Person.property
+def lesson_periods_as_participant(self):
+    return LessonPeriod.objects.filter(lesson__groups__members=self)
+
+
+@Person.property
+def lesson_periods_as_teacher(self):
+    return LessonPeriod.objects.filter(lesson__teachers=self)
-- 
GitLab