fix no parent_context

sentrivana · sentrivana · commit 4a4ca4a253ae · 2024-06-21T15:53:54.000+02:00
diff --git a/sentry_sdk/integrations/opentelemetry/span_processor.py b/sentry_sdk/integrations/opentelemetry/span_processor.py
@@ -130,14 +130,11 @@ def on_start(self, otel_span, parent_context=None):
         if self._is_sentry_span(otel_span):
             return
 
-        if parent_context is None:
-            return
-
         trace_data = self._get_trace_data(otel_span, parent_context)
 
         parent_span_id = trace_data["parent_span_id"]
         sentry_parent_span = (
-            self.otel_span_map.get(parent_span_id, None) if parent_span_id else None
+            self.otel_span_map.get(parent_span_id) if parent_span_id else None
         )
 
         start_timestamp = None
@@ -261,7 +258,7 @@ def _get_otel_context(self, otel_span):
         return ctx
 
     def _get_trace_data(self, otel_span, parent_context):
-        # type: (OTelSpan, context_api.Context) -> dict[str, Any]
+        # type: (OTelSpan, Optional[context_api.Context]) -> dict[str, Any]
         """
         Extracts tracing information from one OTel span and its parent OTel context.
         """
@@ -279,14 +276,17 @@ def _get_trace_data(self, otel_span, parent_context):
         )
         trace_data["parent_span_id"] = parent_span_id
 
-        sentry_trace_data = get_value(SENTRY_TRACE_KEY, parent_context)
-        sentry_trace_data = cast("dict[str, Union[str, bool, None]]", sentry_trace_data)
-        trace_data["parent_sampled"] = (
-            sentry_trace_data["parent_sampled"] if sentry_trace_data else None
-        )
+        if parent_context is not None:
+            sentry_trace_data = get_value(SENTRY_TRACE_KEY, parent_context)
+            sentry_trace_data = cast(
+                "dict[str, Union[str, bool, None]]", sentry_trace_data
+            )
+            trace_data["parent_sampled"] = (
+                sentry_trace_data["parent_sampled"] if sentry_trace_data else None
+            )
 
-        baggage = get_value(SENTRY_BAGGAGE_KEY, parent_context)
-        trace_data["baggage"] = baggage
+            baggage = get_value(SENTRY_BAGGAGE_KEY, parent_context)
+            trace_data["baggage"] = baggage
 
         return trace_data
 
