Look at the Dialect for SQLAlchemy, and small cleanup

Author: ericapisani

sentry_sdk/integrations/asyncpg.py

@@ -199,6 +199,7 @@ async def _inner(*args: "Any", **kwargs: "Any") -> "T":
 
 def _set_db_data(span: "Span", conn: "Any") -> None:
     span.set_data(SPANDATA.DB_SYSTEM, "postgresql")
+    span.set_data(SPANDATA.DB_DRIVER_NAME, "asyncpg")
 
     addr = conn._addr
     if addr:
@@ -215,5 +216,3 @@ def _set_db_data(span: "Span", conn: "Any") -> None:
     user = conn._params.user
     if user:
         span.set_data(SPANDATA.DB_USER, user)
-
-    span.set_data(SPANDATA.DB_DRIVER_NAME, "asyncpg")

sentry_sdk/integrations/sqlalchemy.py

@@ -138,7 +138,7 @@ def _set_db_data(span: "Span", conn: "Any") -> None:
         span.set_data(SPANDATA.DB_SYSTEM, db_system)
 
     try:
-        driver = conn.engine.driver
+        driver = conn.dialect.driver
         if driver:
             span.set_data(SPANDATA.DB_DRIVER_NAME, driver)
     except Exception: