Fixed unwrap cursor for make it work with DDT. Fixed #181

graphql-python · Jun 1, 2016 · 7df1f8a · 7df1f8a
1 parent b9695c8
commit 7df1f8a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/graphene/contrib/django/debug/sql/tracking.py b/graphene/contrib/django/debug/sql/tracking.py
@@ -47,8 +47,9 @@ def cursor():
 
 def unwrap_cursor(connection):
     if hasattr(connection, '_graphene_cursor'):
+        previous_cursor = connection._graphene_cursor
+        connection.cursor = previous_cursor
         del connection._graphene_cursor
-        del connection.cursor
 
 
 class ExceptionCursorWrapper(object):