Fix types

Author: sl0thentr0py

sentry_sdk/integrations/anthropic.py

@@ -44,9 +44,10 @@
 except ImportError:
     raise DidNotEnable("Anthropic not installed")
 
+from sentry_sdk.tracing import Span
+
 if TYPE_CHECKING:
     from typing import Any, AsyncIterator, Iterator, List, Optional, Union
-    from sentry_sdk.tracing import Span
     from sentry_sdk._types import TextPart
 
 
@@ -551,7 +552,7 @@ def _sentry_patched_create_sync(*args: "Any", **kwargs: "Any") -> "Any":
             return _execute_sync(f, *args, **kwargs)
         finally:
             span = sentry_sdk.get_current_span()
-            if span is not None and span.status == SPANSTATUS.INTERNAL_ERROR:
+            if isinstance(span, Span) and span.status == SPANSTATUS.INTERNAL_ERROR:
                 with capture_internal_exceptions():
                     span.__exit__(None, None, None)
 
@@ -589,7 +590,7 @@ async def _sentry_patched_create_async(*args: "Any", **kwargs: "Any") -> "Any":
             return await _execute_async(f, *args, **kwargs)
         finally:
             span = sentry_sdk.get_current_span()
-            if span is not None and span.status == SPANSTATUS.INTERNAL_ERROR:
+            if isinstance(span, Span) and span.status == SPANSTATUS.INTERNAL_ERROR:
                 with capture_internal_exceptions():
                     span.__exit__(None, None, None)
 

sentry_sdk/integrations/celery/__init__.py

@@ -449,7 +449,7 @@ def _inner(*args: "Any", **kwargs: "Any") -> "Any":
 
                 with capture_internal_exceptions():
                     with task.app.connection() as conn:
-                        span.set_data(
+                        set_on_span(
                             SPANDATA.MESSAGING_SYSTEM,
                             conn.transport.driver_type,
                         )

sentry_sdk/integrations/threading.py

@@ -58,7 +58,7 @@ def setup_once() -> None:
             django_version = None
 
         try:
-            import channels  # type: ignore[import-untyped]
+            import channels  # type: ignore[import-not-found]
 
             channels_version = channels.__version__
         except (ImportError, AttributeError):

sentry_sdk/scope.py

@@ -826,7 +826,7 @@ def set_transaction_name(self, name: str, source: "Optional[str]" = None) -> Non
         self._transaction = name
         if self._span:
             if isinstance(self._span, StreamedSpan):
-                self._span.segment.name = name
+                self._span.segment.set_name(name)
                 if source:
                     self._span.segment.set_source(source)
 

sentry_sdk/traces.py

@@ -288,7 +288,6 @@ def __init__(
         else:
             self._sample_rand = _generate_sample_rand(self.trace_id)
 
-        self._flags: dict[str, bool] = {}
         self._continuous_profile: "Optional[ContinuousProfile]" = None
 
         self._update_active_thread()