add option to disable contact form after given datetime

[edumed.git] / comment / views.py

diff --git a/comment/views.py b/comment/views.py
index 1471029..40db4b0 100644 (file)
--- a/comment/views.py
+++ b/comment/views.py
@@ -1,26 +1,27 @@
+# -*- coding: utf-8 -*-
 from django.views.generic import ListView, DetailView
 from django.conf import settings
 from django.utils.translation import get_language
 
 from django.core.urlresolvers import reverse
 
-from .models import CommentDocument
+from .models import CommentDocument as CommentDocumentModel
 
 
 class CommentDocumentList(ListView):
-    model = CommentDocument
+    model = CommentDocumentModel
 
-    def get_queryset(self, **kwargs):
-        return super(CommentDocumentList, self).get_queryset(**kwargs).filter(language_code = get_language())
+    def get_queryset(self):
+        return super(CommentDocumentList, self).get_queryset().filter(language_code=get_language())
 
     def get_context_data(self, **kwargs):
         context = super(CommentDocumentList, self).get_context_data(**kwargs)
-        context['form_href'] = reverse('contact_form', kwargs = dict(form_tag = 'mil'))
+        context['form_href'] = reverse('contact_form', kwargs={'form_tag': 'mil'})
         return context
 
 
 class CommentDocument(DetailView):
-    model = CommentDocument
+    model = CommentDocumentModel
 
     def get_context_data(self, **kwargs):
         context = super(CommentDocument, self).get_context_data(**kwargs)