Make NormalCursorWrapper a context manager

 -- like django.backends.utils.CursorWrapper since Django #17671

1 files changed, 6 insertions, 0 deletions

diff --git a/debug_toolbar/panels/sql/tracking.py b/debug_toolbar/panels/sql/tracking.py
index d3de812..68c77e2 100644
--- a/debug_toolbar/panels/sql/tracking.py
+++ b/debug_toolbar/panels/sql/tracking.py
@@ -185,3 +185,9 @@ class NormalCursorWrapper(object):
 
     def __iter__(self):
         return iter(self.cursor)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        self.close()