From c20e83fc946c8898d40b0210f71286575891ae94 Mon Sep 17 00:00:00 2001
From: bkonkle
Date: Thu, 15 Sep 2011 10:09:53 -0500
Subject: Correct some issues with the stats methods

---
 debug_toolbar/panels/__init__.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'debug_toolbar')

diff --git a/debug_toolbar/panels/__init__.py b/debug_toolbar/panels/__init__.py
index 584ddb7..baebf02 100644
--- a/debug_toolbar/panels/__init__.py
+++ b/debug_toolbar/panels/__init__.py
@@ -41,10 +41,14 @@ class DebugPanel(object):
         raise NotImplementedError
     
     def record_stats(self, stats):
-        self.toolbar.stats[self.slug].update(stats)
+        panel_stats = self.toolbar.stats.get(self.slug)
+        if panel_stats:
+            panel_stats.update(stats)
+        else:
+            self.toolbar.stats[self.slug] = stats
     
     def get_stats(self):
-        return self.toolbar.stats[self.slug]
+        return self.toolbar.stats.get(self.slug, {})
     
     # Standard middleware methods
     def process_request(self, request):
-- 
cgit v1.2.3