Let the annoying commence.

[edumed.git] / catalogue / views.py

diff --git a/catalogue/views.py b/catalogue/views.py
index 273bfa3..0ce7454 100644 (file)
--- a/catalogue/views.py
+++ b/catalogue/views.py
@@ -2,16 +2,17 @@ import os.path
 from django.conf import settings
 from django.views.generic import DetailView, ListView
 from .models import Lesson, Section
+from curriculum.models import Level
+from publishers.models import Publisher
 
 
-class SectionView(ListView):
-    model = Section
+class LessonListView(ListView):
+    queryset = Level.objects.exclude(lesson=None)
+    template_name = "catalogue/lesson_list.html"
 
     def get_context_data(self, **kwargs):
-        context = super(SectionView, self).get_context_data(**kwargs)
+        context = super(LessonListView, 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
 
 
@@ -23,4 +24,12 @@ class LessonView(DetailView):
             'catalogue/lesson/%s/lesson_detail.html' % self.object.type,
             'catalogue/lesson/lesson_detail.html',
         ]
-        
+
+    def get_context_data(self, **kwargs):
+        context = super(LessonView, self).get_context_data(**kwargs)
+        try:
+            context['publisher'] = Publisher.objects.get(
+                name=context['object'].dc.get('publisher', '').strip())
+        except:
+            pass
+        return context