Changes in remit/util/shortcuts.py [f7dd5e7:21360c8]

File:

• remit/util/shortcuts.py (modified) (2 diffs)

remit/util/shortcuts.py

@@ -2 +2 @@
 from django.template import RequestContext, Template
 from django.template.loader import get_template
+from django.views.generic.base import TemplateView
+from django.views.generic.list import ListView
 
 def get_403_response(request, errmsg=None, **extra_context):
@@ -8 +10 @@
     page = tmpl.render(ctx, )
     return HttpResponseForbidden(page)
+
+
+class ListViewWithContext(ListView):
+    extra_context = {}
+
+    # I believe .queryset will work out-of-the-box
+
+    def get_context_data(self, **kwargs):
+        context = super(ListViewWithContext,self).get_context_data(**kwargs)
+        context.update(self.extra_context)
+        return context
+
+
+class TemplateViewWithContext(TemplateView):
+    extra_context = {}
+
+    def get_context_data(self, **kwargs):
+        context = super(TemplateViewWithContext,self).get_context_data(**kwargs)
+        print context
+        context.update(self.extra_context)