From eee102e0319a87d7d0c527f0224d62f34fae66ef Mon Sep 17 00:00:00 2001 From: James Addison Date: Mon, 21 Jun 2021 19:06:37 +0100 Subject: [PATCH] Extract common base cursor wrapper class --- debug_toolbar/panels/sql/tracking.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/debug_toolbar/panels/sql/tracking.py b/debug_toolbar/panels/sql/tracking.py index e7da994d9..2ed691344 100644 --- a/debug_toolbar/panels/sql/tracking.py +++ b/debug_toolbar/panels/sql/tracking.py @@ -58,7 +58,7 @@ def chunked_cursor(*args, **kwargs): # prevent double wrapping # solves https://github.com/jazzband/django-debug-toolbar/issues/1239 cursor = connection._djdt_chunked_cursor(*args, **kwargs) - if not isinstance(cursor, state.Wrapper): + if not isinstance(cursor, BaseCursorWrapper): return state.Wrapper(cursor, connection, panel) return cursor @@ -74,7 +74,11 @@ def unwrap_cursor(connection): del connection.chunked_cursor -class ExceptionCursorWrapper: +class BaseCursorWrapper: + pass + + +class ExceptionCursorWrapper(BaseCursorWrapper): """ Wraps a cursor and raises an exception on any operation. Used in Templates panel. @@ -87,7 +91,7 @@ def __getattr__(self, attr): raise SQLQueryTriggered() -class NormalCursorWrapper: +class NormalCursorWrapper(BaseCursorWrapper): """ Wraps a cursor and logs queries. """