Added configurable HTML tag to attach toolbar to

Signed-off-by: Rob Hudson <rob@cogit8.org>

1 files changed, 12 insertions, 1 deletions

diff --git a/debug_toolbar/middleware.py b/debug_toolbar/middleware.py
index 558019c..86f13cb 100644
--- a/debug_toolbar/middleware.py
+++ b/debug_toolbar/middleware.py
@@ -37,12 +37,20 @@ class DebugToolbarMiddleware(object):
 
         # Set method to use to decide to show toolbar
         self.show_toolbar = self._show_toolbar # default
+
+        # The tag to attach the toolbar to
+        self.tag= u'</body>'
+
         if hasattr(settings, 'DEBUG_TOOLBAR_CONFIG'):
             show_toolbar_callback = settings.DEBUG_TOOLBAR_CONFIG.get(
                 'SHOW_TOOLBAR_CALLBACK', None)
             if show_toolbar_callback:
                 self.show_toolbar = show_toolbar_callback
 
+            tag = settings.DEBUG_TOOLBAR_CONFIG.get('TAG', None)
+            if tag:
+                self.tag = u'</' + tag + u'>'
+
     def _show_toolbar(self, request):
         if not request.META.get('REMOTE_ADDR') in settings.INTERNAL_IPS \
                 or request.is_ajax() or not settings.DEBUG:
@@ -83,6 +91,9 @@ class DebugToolbarMiddleware(object):
             for panel in self.debug_toolbars[request].panels:
                 panel.process_response(request, response)
             if response['Content-Type'].split(';')[0] in _HTML_TYPES:
-                response.content = replace_insensitive(smart_unicode(response.content), u'</body>', smart_unicode(self.debug_toolbars[request].render_toolbar() + u'</body>'))
+                response.content = replace_insensitive(
+                    smart_unicode(response.content), 
+                    self.tag,
+                    smart_unicode(self.debug_toolbars[request].render_toolbar() + self.tag))
         del self.debug_toolbars[request]
         return response