fix: Fix deprecation warning in sqlalchemy integration (getsentry#612)

untitaker · web-flow · commit ff969c9bc87c · 2020-01-29T19:41:13.000+01:00
Fix getsentry#611
diff --git a/sentry_sdk/integrations/sqlalchemy.py b/sentry_sdk/integrations/sqlalchemy.py
@@ -25,7 +25,7 @@ def setup_once():
 
         listen(Engine, "before_cursor_execute", _before_cursor_execute)
         listen(Engine, "after_cursor_execute", _after_cursor_execute)
-        listen(Engine, "dbapi_error", _dbapi_error)
+        listen(Engine, "handle_error", _handle_error)
 
 
 def _before_cursor_execute(
@@ -63,8 +63,9 @@ def _after_cursor_execute(conn, cursor, statement, *args):
         ctx_mgr.__exit__(None, None, None)
 
 
-def _dbapi_error(conn, *args):
+def _handle_error(context, *args):
     # type: (Any, *Any) -> None
+    conn = context.connection
     span = getattr(conn, "_sentry_sql_span", None)  # type: Optional[Span]
 
     if span is not None:
