fix(opentelemetry): Gate third-party imports

Author: alexander-alderman-webb

sentry_sdk/integrations/opentelemetry/consts.py

@@ -1,5 +1,9 @@
-from opentelemetry.context import create_key
+from sentry_sdk.integrations import DidNotEnable
 
+try:
+    from opentelemetry.context import create_key
+except ImportError:
+    raise DidNotEnable("opentelemetry not installed")
 
 SENTRY_TRACE_KEY = create_key("sentry-trace")
 SENTRY_BAGGAGE_KEY = create_key("sentry-baggage")

sentry_sdk/integrations/opentelemetry/propagator.py

@@ -1,23 +1,4 @@
-from opentelemetry import trace
-from opentelemetry.context import (
-    Context,
-    get_current,
-    set_value,
-)
-from opentelemetry.propagators.textmap import (
-    CarrierT,
-    Getter,
-    Setter,
-    TextMapPropagator,
-    default_getter,
-    default_setter,
-)
-from opentelemetry.trace import (
-    NonRecordingSpan,
-    SpanContext,
-    TraceFlags,
-)
-
+from sentry_sdk.integrations import DidNotEnable
 from sentry_sdk.integrations.opentelemetry.consts import (
     SENTRY_BAGGAGE_KEY,
     SENTRY_TRACE_KEY,
@@ -31,6 +12,29 @@
 )
 from sentry_sdk.tracing_utils import Baggage, extract_sentrytrace_data
 
+try:
+    from opentelemetry import trace
+    from opentelemetry.context import (
+        Context,
+        get_current,
+        set_value,
+    )
+    from opentelemetry.propagators.textmap import (
+        CarrierT,
+        Getter,
+        Setter,
+        TextMapPropagator,
+        default_getter,
+        default_setter,
+    )
+    from opentelemetry.trace import (
+        NonRecordingSpan,
+        SpanContext,
+        TraceFlags,
+    )
+except ImportError:
+    raise DidNotEnable("opentelemetry not installed")
+
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:

sentry_sdk/integrations/opentelemetry/span_processor.py

@@ -2,21 +2,9 @@
 from time import time
 from typing import TYPE_CHECKING, cast
 
-from opentelemetry.context import get_value
-from opentelemetry.sdk.trace import SpanProcessor, ReadableSpan as OTelSpan
-from opentelemetry.semconv.trace import SpanAttributes
-from opentelemetry.trace import (
-    format_span_id,
-    format_trace_id,
-    get_current_span,
-    SpanKind,
-)
-from opentelemetry.trace.span import (
-    INVALID_SPAN_ID,
-    INVALID_TRACE_ID,
-)
 from sentry_sdk import get_client, start_transaction
 from sentry_sdk.consts import INSTRUMENTER, SPANSTATUS
+from sentry_sdk.integrations import DidNotEnable
 from sentry_sdk.integrations.opentelemetry.consts import (
     SENTRY_BAGGAGE_KEY,
     SENTRY_TRACE_KEY,
@@ -26,6 +14,23 @@
 
 from urllib3.util import parse_url as urlparse
 
+try:
+    from opentelemetry.context import get_value
+    from opentelemetry.sdk.trace import SpanProcessor, ReadableSpan as OTelSpan
+    from opentelemetry.semconv.trace import SpanAttributes
+    from opentelemetry.trace import (
+        format_span_id,
+        format_trace_id,
+        get_current_span,
+        SpanKind,
+    )
+    from opentelemetry.trace.span import (
+        INVALID_SPAN_ID,
+        INVALID_TRACE_ID,
+    )
+except ImportError:
+    raise DidNotEnable("opentelemetry not installed")
+
 if TYPE_CHECKING:
     from typing import Any, Optional, Union
     from opentelemetry import context as context_api

tests/test_shadowed_module.py

@@ -32,7 +32,6 @@ def pytest_generate_tests(metafunc):
                 "clickhouse_driver",
                 "grpc",
                 "litellm",
-                "opentelemetry",
                 "pure_eval",
                 "ray",
                 "trytond",