Cursor moved from cursor.py to connection.py

elastic · Jun 30, 2021 · 4942b4a · 4942b4a
1 parent 5b8c341
commit 4942b4a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/elasticapm/instrumentation/packages/asyncio/aiopg.py b/elasticapm/instrumentation/packages/asyncio/aiopg.py
@@ -36,7 +36,12 @@
 class AioPGInstrumentation(AsyncAbstractInstrumentedModule):
     name = "aiopg"
 
-    instrument_list = [("aiopg.cursor", "Cursor.execute"), ("aiopg.cursor", "Cursor.callproc")]
+    instrument_list = [
+        ("aiopg.cursor", "Cursor.execute"),
+        ("aiopg.cursor", "Cursor.callproc"),
+        ("aiopg.connection", "Cursor.execute"),
+        ("aiopg.connection", "Cursor.callproc"),
+    ]
 
     async def call(self, module, method, wrapped, instance, args, kwargs):
         if method == "Cursor.execute":