Skip to content

Commit d0f7cfa

Browse files
committed
fix: failing postgresql tests
1 parent 7d4c530 commit d0f7cfa

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

debug_toolbar/panels/sql/tracking.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -57,11 +57,10 @@ def cursor(*args, **kwargs):
5757
def chunked_cursor(*args, **kwargs):
5858
# prevent double wrapping
5959
# solves https://github.com/jazzband/django-debug-toolbar/issues/1239
60-
if hasattr(connection._djdt_cursor, "__wrapped__"):
61-
return connection._djdt_cursor(*args, **kwargs)
62-
return state.Wrapper(
63-
connection._djdt_chunked_cursor(*args, **kwargs), connection, panel
64-
)
60+
cursor = connection._djdt_chunked_cursor(*args, **kwargs)
61+
if not isinstance(cursor, NormalCursorWrapper):
62+
return state.Wrapper(cursor, connection, panel)
63+
return cursor
6564

6665
connection.cursor = cursor
6766
connection.chunked_cursor = chunked_cursor

0 commit comments

Comments
 (0)