check test still fails

sentry_sdk/integrations/aiohttp.py

@@ -1,5 +1,6 @@
 import sys
 import weakref
+from functools import wraps
 
 import sentry_sdk
 from sentry_sdk.api import continue_trace
@@ -156,11 +157,14 @@ async def sentry_app_handle(self, request, *args, **kwargs):
 
         old_urldispatcher_resolve = UrlDispatcher.resolve
 
+        @wraps(old_urldispatcher_resolve)
         async def sentry_urldispatcher_resolve(self, request):
             # type: (UrlDispatcher, Request) -> UrlMappingMatchInfo
             rv = await old_urldispatcher_resolve(self, request)
 
             integration = sentry_sdk.get_client().get_integration(AioHttpIntegration)
+            if integration is None:
+                return rv
 
             name = None
 

sentry_sdk/integrations/anthropic.py

@@ -7,7 +7,6 @@
 from sentry_sdk.scope import should_send_default_pii
 from sentry_sdk.utils import (
     capture_internal_exceptions,
-    ensure_integration_enabled,
     event_from_exception,
     package_version,
 )
@@ -78,10 +77,11 @@ def _calculate_token_usage(result, span):
 def _wrap_message_create(f):
     # type: (Any) -> Any
     @wraps(f)
-    @ensure_integration_enabled(AnthropicIntegration, f)
     def _sentry_patched_create(*args, **kwargs):
         # type: (*Any, **Any) -> Any
-        if "messages" not in kwargs:
+        integration = sentry_sdk.get_client().get_integration(AnthropicIntegration)
+
+        if integration is None or "messages" not in kwargs:
             return f(*args, **kwargs)
 
         try:
@@ -106,8 +106,6 @@ def _sentry_patched_create(*args, **kwargs):
             span.__exit__(None, None, None)
             raise exc from None
 
-        integration = sentry_sdk.get_client().get_integration(AnthropicIntegration)
-
         with capture_internal_exceptions():
             span.set_data(SPANDATA.AI_MODEL_ID, model)
             span.set_data(SPANDATA.AI_STREAMING, False)

sentry_sdk/integrations/atexit.py

@@ -5,8 +5,6 @@
 import sentry_sdk
 from sentry_sdk.utils import logger
 from sentry_sdk.integrations import Integration
-from sentry_sdk.utils import ensure_integration_enabled
-
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
@@ -44,13 +42,16 @@ def __init__(self, callback=None):
     def setup_once():
         # type: () -> None
         @atexit.register
-        @ensure_integration_enabled(AtexitIntegration)
         def _shutdown():
             # type: () -> None
-            logger.debug("atexit: got shutdown signal")
             client = sentry_sdk.get_client()
             integration = client.get_integration(AtexitIntegration)
 
+            if integration is None:
+                return
+
+            logger.debug("atexit: got shutdown signal")
             logger.debug("atexit: shutting down client")
             sentry_sdk.get_isolation_scope().end_session()
+
             client.close(callback=integration.callback)

sentry_sdk/integrations/aws_lambda.py

@@ -1,3 +1,4 @@
+import functools
 import json
 import re
 import sys
@@ -70,7 +71,7 @@ def sentry_init_error(*args, **kwargs):
 
 def _wrap_handler(handler):
     # type: (F) -> F
-    @ensure_integration_enabled(AwsLambdaIntegration, handler)
+    @functools.wraps(handler)
     def sentry_handler(aws_event, aws_context, *args, **kwargs):
         # type: (Any, Any, *Any, **Any) -> Any
 
@@ -84,6 +85,12 @@ def sentry_handler(aws_event, aws_context, *args, **kwargs):
         # will be the same for all events in the list, since they're all hitting
         # the lambda in the same request.)
 
+        client = sentry_sdk.get_client()
+        integration = client.get_integration(AwsLambdaIntegration)
+
+        if integration is None:
+            return handler(aws_event, aws_context, *args, **kwargs)
+
         if isinstance(aws_event, list) and len(aws_event) >= 1:
             request_data = aws_event[0]
             batch_size = len(aws_event)
@@ -97,9 +104,6 @@ def sentry_handler(aws_event, aws_context, *args, **kwargs):
             # this is empty
             request_data = {}
 
-        client = sentry_sdk.get_client()
-        integration = client.get_integration(AwsLambdaIntegration)
-
         configured_time = aws_context.get_remaining_time_in_millis()
 
         with sentry_sdk.isolation_scope() as scope: