mypy

Author: sentrivana

sentry_sdk/integrations/django/tasks.py

@@ -8,7 +8,7 @@
     # django.tasks were added in Django 6.0
     from django.tasks.base import Task
 except ImportError:
-    Task = None
+    Task = None  # type: ignore[misc, assignment]
 
 from typing import TYPE_CHECKING
 
@@ -22,7 +22,7 @@ def patch_tasks() -> None:
 
     old_task_enqueue = Task.enqueue
 
-    @wraps(old_task_enqueue)
+    @wraps(old_task_enqueue)  # type: ignore[arg-type]
     def _sentry_enqueue(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
         from sentry_sdk.integrations.django import DjangoIntegration
 
@@ -37,4 +37,4 @@ def _sentry_enqueue(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
         ):
             return old_task_enqueue(self, *args, **kwargs)
 
-    Task.enqueue = _sentry_enqueue
+    Task.enqueue = _sentry_enqueue  # type: ignore[method-assign]

sentry_sdk/integrations/httpx.py

@@ -22,6 +22,8 @@
 if TYPE_CHECKING:
     from typing import Any, Optional, Union
 
+    from sentry_sdk._types import Attributes
+
 
 try:
     from httpx import AsyncClient, Client, Request, Response
@@ -61,7 +63,7 @@ def send(self: "Client", request: "Request", **kwargs: "Any") -> "Response":
 
         span_ctx: "Optional[Union[Span, StreamedSpan]]" = None
         if span_streaming:
-            attributes = {
+            attributes: "Attributes" = {
                 "sentry.op": OP.HTTP_CLIENT,
                 "sentry.origin": HttpxIntegration.origin,
                 SPANDATA.HTTP_METHOD: request.method,
@@ -145,7 +147,7 @@ async def send(
 
         span_ctx: "Optional[Union[Span, StreamedSpan]]" = None
         if span_streaming:
-            attributes = {
+            attributes: "Attributes" = {
                 "sentry.op": OP.HTTP_CLIENT,
                 "sentry.origin": HttpxIntegration.origin,
                 SPANDATA.HTTP_METHOD: request.method,

sentry_sdk/integrations/rust_tracing.py

@@ -32,7 +32,7 @@
 
 import json
 from enum import Enum, auto
-from typing import Any, Callable, Dict, Optional
+from typing import Any, Callable, Dict, Optional, Union
 
 import sentry_sdk
 from sentry_sdk.integrations import Integration
@@ -184,7 +184,9 @@ def on_event(self, event: str, sentry_span: "Span") -> None:
         elif event_type == EventTypeMapping.Event:
             process_event(deserialized_event)
 
-    def on_new_span(self, attrs: str, span_id: str) -> "Optional[Span]":
+    def on_new_span(
+        self, attrs: str, span_id: str
+    ) -> "Union[Span, StreamedSpan, None]":
         attrs = json.loads(attrs)
         metadata = attrs.get("metadata", {})
 
@@ -206,6 +208,7 @@ def on_new_span(self, attrs: str, span_id: str) -> "Optional[Span]":
 
         client = sentry_sdk.get_client()
 
+        sentry_span: "Union[Span, StreamedSpan]"
         if has_span_streaming_enabled(client.options):
             sentry_span = sentry_sdk.traces.start_span(
                 name=sentry_span_name,