Fix #76 - psycopg2 cursors not working when using Dict cursors

dynatrace-oss · Jun 22, 2023 · 49091a0 · 49091a0
1 parent 7c90d89
commit 49091a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autodynatrace/wrappers/psycopg2/wrapper.py b/autodynatrace/wrappers/psycopg2/wrapper.py
@@ -11,7 +11,7 @@ def instrument():
     def parse_dsn(dsn):
         return {c.split("=")[0]: c.split("=")[1] for c in dsn.split() if "=" in c}
 
-    class DynatraceCursor(psycopg2.extensions.cursor):
+    class DynatraceCursor(psycopg2.extra.DictCursorBase):
         def __init__(self, *args, **kwargs):
             self._dynatrace_db_info = kwargs.pop("dynatrace_db_info", None)
             super(DynatraceCursor, self).__init__(*args, **kwargs)