Updated the request_vars panel

Author: bkonkle

debug_toolbar/panels/request_vars.py

@@ -1,6 +1,7 @@
 from django.template.loader import render_to_string
 from django.utils.translation import ugettext_lazy as _
 
+from debug_toolbar.middleware import DebugToolbarMiddleware
 from debug_toolbar.panels import DebugPanel
 from debug_toolbar.utils import get_name_from_obj
 
@@ -10,53 +11,58 @@ class RequestVarsDebugPanel(DebugPanel):
     """
     name = 'RequestVars'
     has_content = True
-
+    
     def __init__(self, *args, **kwargs):
         DebugPanel.__init__(self, *args, **kwargs)
         self.view_func = None
         self.view_args = None
         self.view_kwargs = None
-
+    
     def nav_title(self):
         return _('Request Vars')
-
+    
     def title(self):
         return _('Request Vars')
-
+    
     def url(self):
         return ''
-
+    
     def process_request(self, request):
         self.request = request
-
+    
     def process_view(self, request, view_func, view_args, view_kwargs):
         self.view_func = view_func
         self.view_args = view_args
         self.view_kwargs = view_kwargs
-
-    def content(self):
-        context = self.context.copy()
-
-        context.update({
+    
+    def process_response(self, request, response):
+        self.stats = {
             'get': [(k, self.request.GET.getlist(k)) for k in self.request.GET],
             'post': [(k, self.request.POST.getlist(k)) for k in self.request.POST],
             'cookies': [(k, self.request.COOKIES.get(k)) for k in self.request.COOKIES],
-        })
+        }
+        
         if hasattr(self, 'view_func'):
             if self.view_func is not None:
                 name = get_name_from_obj(self.view_func)
             else:
                 name = '<no view>'
-
-            context.update({
+            
+            self.stats.update({
                 'view_func': name,
                 'view_args': self.view_args,
                 'view_kwargs': self.view_kwargs
             })
-
+        
         if hasattr(self.request, 'session'):
-            context.update({
+            self.stats.update({
                 'session': [(k, self.request.session.get(k)) for k in self.request.session.iterkeys()]
             })
-
+        
+        toolbar = DebugToolbarMiddleware.get_current()
+        toolbar.stats['request_vars'] = self.stats
+    
+    def content(self):
+        context = self.context.copy()
+        context.update(self.stats)
         return render_to_string('debug_toolbar/panels/request_vars.html', context)