use type method instead of isinstance

Author: ericapisani

sentry_sdk/integrations/fastapi.py

@@ -1,19 +1,19 @@
 import sys
 from copy import deepcopy
 from functools import wraps
+from typing import TYPE_CHECKING
 
 import sentry_sdk
 from sentry_sdk.integrations import DidNotEnable
 from sentry_sdk.scope import should_send_default_pii
-from sentry_sdk.traces import _is_sampled_streamed_span
+from sentry_sdk.traces import _is_streamed_span
 from sentry_sdk.tracing import SOURCE_FOR_STYLE, TransactionSource
 from sentry_sdk.tracing_utils import has_span_streaming_enabled
 from sentry_sdk.utils import transaction_from_function
 
-from typing import TYPE_CHECKING
-
 if TYPE_CHECKING:
     from typing import Any, Callable, Dict
+
     from sentry_sdk._types import Event
 
 try:
@@ -95,7 +95,7 @@ def _sentry_call(*args: "Any", **kwargs: "Any") -> "Any":
                 if has_span_streaming_enabled(client.options):
                     current_span = current_scope.streamed_span
 
-                    if _is_sampled_streamed_span(current_span):
+                    if _is_streamed_span(current_span):
                         segment = current_span._segment
                         segment._update_active_thread()
 

sentry_sdk/integrations/starlette.py

@@ -1,17 +1,18 @@
 import functools
 import json
-import warnings
 import sys
+import warnings
 from collections.abc import Set
 from copy import deepcopy
 from json import JSONDecodeError
+from typing import TYPE_CHECKING
 
 import sentry_sdk
 from sentry_sdk.consts import OP
 from sentry_sdk.integrations import (
+    _DEFAULT_FAILED_REQUEST_STATUS_CODES,
     DidNotEnable,
     Integration,
-    _DEFAULT_FAILED_REQUEST_STATUS_CODES,
 )
 from sentry_sdk.integrations._wsgi_common import (
     DEFAULT_HTTP_METHODS_TO_CAPTURE,
@@ -21,7 +22,7 @@
 )
 from sentry_sdk.integrations.asgi import SentryAsgiMiddleware
 from sentry_sdk.scope import should_send_default_pii
-from sentry_sdk.traces import _get_current_streamed_span, _is_sampled_streamed_span
+from sentry_sdk.traces import _get_current_streamed_span, _is_streamed_span
 from sentry_sdk.tracing import (
     SOURCE_FOR_STYLE,
     TransactionSource,
@@ -36,8 +37,6 @@
     transaction_from_function,
 )
 
-from typing import TYPE_CHECKING
-
 if TYPE_CHECKING:
     from typing import Any, Awaitable, Callable, Container, Dict, Optional, Tuple, Union
 
@@ -54,7 +53,8 @@
     )
     from starlette.requests import Request  # type: ignore
     from starlette.routing import Match  # type: ignore
-    from starlette.types import ASGIApp, Receive, Scope as StarletteScope, Send  # type: ignore
+    from starlette.types import ASGIApp, Receive, Send  # type: ignore
+    from starlette.types import Scope as StarletteScope
 except ImportError:
     raise DidNotEnable("Starlette is not installed")
 
@@ -255,7 +255,7 @@ def _set_request_body_data_on_streaming_segment(
     info: "Optional[Dict[str, Any]]",
 ) -> None:
     current_span = _get_current_streamed_span()
-    if info and "data" in info and _is_sampled_streamed_span(current_span):
+    if info and "data" in info and _is_streamed_span(current_span):
         with capture_internal_exceptions():
             current_span._segment.set_attribute(
                 "http.request.body.data",
@@ -552,7 +552,7 @@ def _sentry_sync_func(*args: "Any", **kwargs: "Any") -> "Any":
                 if span_streaming:
                     current_span = current_scope.streamed_span
 
-                    if _is_sampled_streamed_span(current_span):
+                    if _is_streamed_span(current_span):
                         current_span._segment._update_active_thread()
                 elif current_scope.transaction is not None:
                     current_scope.transaction.update_active_thread()

sentry_sdk/scope.py

@@ -1,13 +1,14 @@
 import os
 import sys
 import warnings
-from copy import copy, deepcopy
 from collections import deque
 from contextlib import contextmanager
-from enum import Enum
+from copy import copy, deepcopy
 from datetime import datetime, timezone
+from enum import Enum
 from functools import wraps
 from itertools import chain
+from typing import TYPE_CHECKING, cast
 
 import sentry_sdk
 from sentry_sdk._types import AnnotatedValue
@@ -18,27 +19,19 @@
     INSTRUMENTER,
     SPANDATA,
 )
-from sentry_sdk.feature_flags import FlagBuffer, DEFAULT_FLAG_CAPACITY
+from sentry_sdk.feature_flags import DEFAULT_FLAG_CAPACITY, FlagBuffer
 from sentry_sdk.profiler.continuous_profiler import (
     get_profiler_id,
     try_autostart_continuous_profiler,
     try_profile_lifecycle_trace_start,
 )
 from sentry_sdk.profiler.transaction_profiler import Profile
 from sentry_sdk.session import Session
-from sentry_sdk.tracing_utils import (
-    Baggage,
-    has_tracing_enabled,
-    has_span_streaming_enabled,
-    is_ignored_span,
-    _make_sampling_decision,
-    PropagationContext,
-)
 from sentry_sdk.traces import (
     _DEFAULT_PARENT_SPAN,
-    StreamedSpan,
     NoOpStreamedSpan,
-    _is_sampled_streamed_span,
+    StreamedSpan,
+    _is_streamed_span,
 )
 from sentry_sdk.tracing import (
     BAGGAGE_HEADER_NAME,
@@ -47,40 +40,48 @@
     Span,
     Transaction,
 )
+from sentry_sdk.tracing_utils import (
+    Baggage,
+    PropagationContext,
+    _make_sampling_decision,
+    has_span_streaming_enabled,
+    has_tracing_enabled,
+    is_ignored_span,
+)
 from sentry_sdk.utils import (
+    ContextVar,
     capture_internal_exception,
     capture_internal_exceptions,
-    ContextVar,
     datetime_from_isoformat,
     disable_capture_event,
     event_from_exception,
     exc_info_from_error,
     format_attribute,
-    logger,
     has_logs_enabled,
     has_metrics_enabled,
+    logger,
 )
 
-from typing import TYPE_CHECKING, cast
-
 if TYPE_CHECKING:
     from collections.abc import Mapping
-
-    from typing import Any
-    from typing import Callable
-    from typing import Deque
-    from typing import Dict
-    from typing import Generator
-    from typing import Iterator
-    from typing import List
-    from typing import Optional
-    from typing import ParamSpec
-    from typing import Tuple
-    from typing import TypeVar
-    from typing import Union
+    from typing import (
+        Any,
+        Callable,
+        Deque,
+        Dict,
+        Generator,
+        Iterator,
+        List,
+        Optional,
+        ParamSpec,
+        Tuple,
+        TypeVar,
+        Union,
+    )
 
     from typing_extensions import Unpack
 
+    import sentry_sdk
     from sentry_sdk._types import (
         Attributes,
         AttributeValue,
@@ -97,11 +98,8 @@
         SamplingContext,
         Type,
     )
-
     from sentry_sdk.tracing import TransactionKwargs
 
-    import sentry_sdk
-
     P = ParamSpec("P")
     R = TypeVar("R")
 
@@ -590,7 +588,7 @@ def get_traceparent(self, *args: "Any", **kwargs: "Any") -> "Optional[str]":
 
         span_streaming = has_span_streaming_enabled(client.options)
         # If we have an active span, return traceparent from there
-        if span_streaming and _is_sampled_streamed_span(self.streamed_span):
+        if span_streaming and _is_streamed_span(self.streamed_span):
             return self.streamed_span._to_traceparent()
         elif not span_streaming and self.span is not None:
             return self.span._to_traceparent()
@@ -610,7 +608,7 @@ def get_baggage(self, *args: "Any", **kwargs: "Any") -> "Optional[Baggage]":
 
         span_streaming = has_span_streaming_enabled(client.options)
         # If we have an active span, return baggage from there
-        if span_streaming and _is_sampled_streamed_span(self.streamed_span):
+        if span_streaming and _is_streamed_span(self.streamed_span):
             return self.streamed_span._to_baggage()
         elif not span_streaming and self.span is not None:
             return self.span._to_baggage()
@@ -915,7 +913,7 @@ def streamed_span(self, span: "Optional[StreamedSpan]") -> None:
 
         # Also set _transaction and _transaction_info in streaming mode as this
         # is used for populating events and linking them to segments
-        if _is_sampled_streamed_span(span) and span._is_segment():
+        if _is_streamed_span(span) and span._is_segment():
             self._transaction = span.name
             if span._attributes.get("sentry.span.source"):
                 self._transaction_info["source"] = str(

sentry_sdk/traces.py

@@ -41,6 +41,7 @@
         TypeVar,
         Union,
     )
+
     from sentry_sdk._types import Attributes, AttributeValue
     from sentry_sdk.profiler.continuous_profiler import ContinuousProfile
 
@@ -766,9 +767,9 @@ def make_db_query(sql):
         return decorator
 
 
-def _is_sampled_streamed_span(span: "Any") -> "TypeGuard[StreamedSpan]":
-    """Returns True if span is a StreamedSpan that was sampled (not a NoOpStreamedSpan)."""
-    return isinstance(span, StreamedSpan) and not isinstance(span, NoOpStreamedSpan)
+def _is_streamed_span(span: "Any") -> "TypeGuard[StreamedSpan]":
+    """Returns True if span is a StreamedSpan (not a NoOpStreamedSpan)."""
+    return type(span) is StreamedSpan
 
 
 def _get_current_streamed_span(