fix for non-sectioned lessons nav

[edumed.git] / catalogue / views.py

diff --git a/catalogue/views.py b/catalogue/views.py
index 60f00ef..273bfa3 100644 (file)
--- a/catalogue/views.py
+++ b/catalogue/views.py
@@ -1 +1,26 @@
-# Create your views here.
+import os.path
+from django.conf import settings
+from django.views.generic import DetailView, ListView
+from .models import Lesson, Section
+
+
+class SectionView(ListView):
+    model = Section
+
+    def get_context_data(self, **kwargs):
+        context = super(SectionView, self).get_context_data(**kwargs)
+        context['appendix'] = Lesson.objects.filter(type='appendix')
+        context['package_url'] = os.path.join(settings.MEDIA_URL, settings.CATALOGUE_PACKAGE)
+        context['package_student_url'] = os.path.join(settings.MEDIA_URL, settings.CATALOGUE_PACKAGE_STUDENT)
+        return context
+
+
+class LessonView(DetailView):
+    model = Lesson
+
+    def get_template_names(self):
+        return [
+            'catalogue/lesson/%s/lesson_detail.html' % self.object.type,
+            'catalogue/lesson/lesson_detail.html',
+        ]
+