From b9b0cbe718c6a1736af1fa86452ffc4af28b64f4 Mon Sep 17 00:00:00 2001
From: Rob Hudson
Date: Thu, 26 Mar 2009 07:29:26 -0700
Subject: New configuration option to provide your own 'show_toolbar' method.

This change allows a custom method to be used in place of the default 'show_toolbar' method on whether the toolbar should be displayed.---
 debug_toolbar/middleware.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'debug_toolbar/middleware.py')

diff --git a/debug_toolbar/middleware.py b/debug_toolbar/middleware.py
index b10ced2..e4d7494 100644
--- a/debug_toolbar/middleware.py
+++ b/debug_toolbar/middleware.py
@@ -37,7 +37,15 @@ class DebugToolbarMiddleware(object):
         self.original_pattern = patterns('', ('', include(self.original_urlconf)),)
         self.override_url = True
 
-    def show_toolbar(self, request):
+        # Set method to use to decide to show toolbar
+        self.show_toolbar = self._show_toolbar # default
+        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
+
+    def _show_toolbar(self, request):
         if not settings.DEBUG:
             return False
         if request.is_ajax() and not \
-- 
cgit v1.2.3