More polished PAI for module html injection and added a couple more places to inject...

[osqa.git] / forum / views / decorators.py

diff --git a/forum/views/decorators.py b/forum/views/decorators.py
index db6dec48e9712f194cac483bdf74e5c49a72c6a2..bae9712332e50a4dce652b0b6b1f996d54326c8f 100644 (file)
--- a/forum/views/decorators.py
+++ b/forum/views/decorators.py
@@ -2,11 +2,13 @@ from django.http import HttpResponse, HttpResponseRedirect, Http404
 from django.utils import simplejson\r
 from django.core.paginator import Paginator, EmptyPage\r
 from django.shortcuts import render_to_response\r
+from django.core.urlresolvers import reverse\r
 from django.template import RequestContext\r
 from django.utils.translation import ungettext, ugettext as _\r
+from forum.modules import ui\r
 import logging\r
 \r
-def render(template=None, tab=None):\r
+def render(template=None, tab=None, tab_title='', weight=500, tabbed=True):\r
     def decorator(func):\r
         def decorated(request, *args, **kwargs):\r
             context = func(request, *args, **kwargs)\r
@@ -17,6 +19,10 @@ def render(template=None, tab=None):
             return render_to_response(context.pop('template', template), context,\r
                                       context_instance=RequestContext(request))\r
 \r
+        if tabbed and tab:\r
+            ui.register(ui.PAGE_TOP_TABS,\r
+                        ui.PageTab(tab, tab_title, lambda: reverse(func.__name__), weight=weight))\r
+            \r
         return decorated\r
 \r
     return decorator\r