Nicer displaying of advanced lessons.

[edumed.git] / catalogue / models.py

diff --git a/catalogue/models.py b/catalogue/models.py
index e5ee3a9..46338c4 100644 (file)
--- a/catalogue/models.py
+++ b/catalogue/models.py
@@ -56,10 +56,9 @@ class Section(models.Model):
 
         return section
 
-
-    def syntetic_lesson(self):
+    def syntetic_lesson(self, level):
         try:
-            return self.lesson_set.filter(type='synthetic')[0]
+            return self.lesson_set.filter(type='synthetic', level=level)[0]
         except IndexError:
             return None
 
@@ -166,6 +165,9 @@ class Lesson(models.Model):
             "%s%s.zip" % (self.slug, "_student" if student else ""),
             ContentFile(buff.getvalue()))
 
+    def get_syntetic(self):
+        return self.section.syntetic_lesson(self.level)
+
 
 class Attachment(models.Model):
     slug = models.CharField(max_length=255)