Prepared for SP 4-6.

[edumed.git] / curriculum / templatetags / curriculum_tags.py

diff --git a/curriculum/templatetags/curriculum_tags.py b/curriculum/templatetags/curriculum_tags.py
index 00959ee..4e02e04 100755 (executable)
--- a/curriculum/templatetags/curriculum_tags.py
+++ b/curriculum/templatetags/curriculum_tags.py
@@ -1,24 +1,26 @@
 from django import template
 from django.utils.datastructures import SortedDict
-from ..models import Competence, Curriculum
+from ..models import Competence, Curriculum, CurriculumCourse
+from catalogue.models import Lesson
 
 register = template.Library()
 
 
 @register.inclusion_tag("curriculum/snippets/competence.html")
-def competence(texts):
+def competence(texts, level):
     try:
         comps = [Competence.from_text(text) for text in texts]
     except:
         return {'texts': texts}
     return {
         'comps': comps,
+        'level': level,
     }
 
 @register.inclusion_tag("curriculum/snippets/curriculum.html")
 def curriculum(identifiers):
     try:
-        currs = [Curriculum.objects.get(identifier=identifier)
+        currs = [Curriculum.objects.get(identifier__iexact=identifier.replace(' ', ''))
                     for identifier in identifiers]
     except Curriculum.DoesNotExist:
         return {'identifiers': identifiers}
@@ -41,7 +43,46 @@ def curriculum(identifiers):
 @register.filter
 def url_for_level(comp, level):
     try:
-        return comp.for_level(level).get_absolute_url()
+        return comp.url_for_level(level)
     except:
         return comp.get_absolute_url()
 
+
+@register.inclusion_tag("curriculum/snippets/course_box.html")
+def course_box(course):
+    lessons = SortedDict()
+    for lesson in course.lesson_set.all():
+        if lesson.level not in lessons:
+            newdict = SortedDict()
+            newdict['synthetic'] = []
+            newdict['course'] = []
+            lessons[lesson.level] = newdict
+        if lesson.type not in lessons[lesson.level]:
+            lessons[lesson.level][lesson.type] = []
+        lessons[lesson.level][lesson.type].append(lesson)
+    return {
+        "course": course,
+        "lessons": lessons,
+    }
+
+@register.inclusion_tag("curriculum/snippets/course_boxes.html")
+def course_boxes():
+    return {'object_list': CurriculumCourse.objects.all()}
+
+@register.inclusion_tag("curriculum/snippets/course_boxes_toc.html")
+def course_boxes_toc(accusative=False):
+    last = None, None
+    object_list = []
+    for l in Lesson.curriculum_courses.through.objects.all().order_by(
+            'lesson__level', 'curriculumcourse'):
+        level, course = l.lesson.level, l.curriculumcourse
+        if (level, course) == last:
+            continue
+        if level != last[0]:
+            object_list.append((level, []))
+        object_list[-1][1].append(course)
+        last = (level, course)
+    return {
+        'object_list': object_list,
+        'accusative': accusative,
+    }