From 04eea3de29ec609621ac91687676b6b62a8f8169 Mon Sep 17 00:00:00 2001
From: Simon Meers
Date: Thu, 13 Feb 2014 20:35:10 +1100
Subject: Make NormalCursorWrapper a context manager  -- like
 django.backends.utils.CursorWrapper since Django #17671

---
 debug_toolbar/panels/sql/tracking.py | 6 ++++++
 1 file changed, 6 insertions(+)

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()
-- 
cgit v1.2.3