diff options
| -rw-r--r-- | debug_toolbar/panels/sql/tracking.py | 6 |
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() |