ref(asyncio): Use new scopes API (getsentry#2895)

sentrivana · web-flow · commit ccb311fd026d · 2024-03-22T14:36:28.000+01:00
diff --git a/sentry_sdk/integrations/asyncio.py b/sentry_sdk/integrations/asyncio.py
@@ -1,7 +1,7 @@
 import sys
 
+import sentry_sdk
 from sentry_sdk.consts import OP
-from sentry_sdk.hub import Hub
 from sentry_sdk.integrations import Integration, DidNotEnable
 from sentry_sdk._types import TYPE_CHECKING
 from sentry_sdk.utils import event_from_exception, reraise
@@ -41,15 +41,16 @@ def _sentry_task_factory(loop, coro, **kwargs):
 
             async def _coro_creating_hub_and_span():
                 # type: () -> Any
-                hub = Hub(Hub.current)
                 result = None
 
-                with hub:
-                    with hub.start_span(op=OP.FUNCTION, description=get_name(coro)):
+                with sentry_sdk.isolation_scope():
+                    with sentry_sdk.start_span(
+                        op=OP.FUNCTION, description=get_name(coro)
+                    ):
                         try:
                             result = await coro
                         except Exception:
-                            reraise(*_capture_exception(hub))
+                            reraise(*_capture_exception())
 
                 return result
 
@@ -76,21 +77,20 @@ async def _coro_creating_hub_and_span():
         pass
 
 
-def _capture_exception(hub):
-    # type: (Hub) -> ExcInfo
+def _capture_exception():
+    # type: () -> ExcInfo
     exc_info = sys.exc_info()
 
-    integration = hub.get_integration(AsyncioIntegration)
-    if integration is not None:
-        # If an integration is there, a client has to be there.
-        client = hub.client  # type: Any
+    client = sentry_sdk.get_client()
 
+    integration = client.get_integration(AsyncioIntegration)
+    if integration is not None:
         event, hint = event_from_exception(
             exc_info,
             client_options=client.options,
             mechanism={"type": "asyncio", "handled": False},
         )
-        hub.capture_event(event, hint=hint)
+        sentry_sdk.capture_event(event, hint=hint)
 
     return exc_info
 
