warpzone-sdk 15.0.0.dev1__tar.gz → 15.0.0.dev2__tar.gz

{warpzone_sdk-15.0.0.dev1 → warpzone_sdk-15.0.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: warpzone-sdk
-Version: 15.0.0.dev1
+Version: 15.0.0.dev2
 Summary: The main objective of this package is to centralize logic used to interact with Azure Functions, Azure Service Bus and Azure Table Storage
 Author: Team Enigma
 Author-email: enigma@energinet.dk

{warpzone_sdk-15.0.0.dev1 → warpzone_sdk-15.0.0.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "warpzone-sdk"
-version = "15.0.0.dev1"
+version = "15.0.0.dev2"
 description = "The main objective of this package is to centralize logic used to interact with Azure Functions, Azure Service Bus and Azure Table Storage"
 authors = [{ name = "Team Enigma", email = "enigma@energinet.dk" }]
 requires-python = ">=3.10"

{warpzone_sdk-15.0.0.dev1 → warpzone_sdk-15.0.0.dev2}/warpzone/function/monitor.py

@@ -1,15 +1,25 @@
-import asyncio
+import inspect
 from contextlib import contextmanager
 from typing import Callable
 
 import azure.functions as func
 from azure.monitor.opentelemetry import configure_azure_monitor
+from opentelemetry import trace
 
 from warpzone.function.types import SingleArgumentCallable
 from warpzone.monitor import logs, traces
 
+# Configure Azure Monitor first
 configure_azure_monitor()
 
+# Apply trace filtering to suppress all Azure SDK traces except Service Bus
+tracer_provider = trace.get_tracer_provider()
+if hasattr(tracer_provider, "_active_span_processor"):
+    # Wrap the existing span processor with our filter
+    original_processor = tracer_provider._active_span_processor
+    filtered_processor = traces.AzureSDKTraceFilter(original_processor)
+    tracer_provider._active_span_processor = filtered_processor
+
 SUBJECT_IDENTIFIER = "<Subject>"
 
 tracer = traces.get_tracer(__name__)
@@ -54,7 +64,7 @@ def monitor(main: SingleArgumentCallable) -> Callable:
             result = main(arg)
             return result
 
-    if asyncio.iscoroutinefunction(main):
+    if inspect.iscoroutinefunction(main):
         return wrapper_async
     else:
         return wrapper

warpzone_sdk-15.0.0.dev2/warpzone/monitor/traces.py

@@ -0,0 +1,99 @@
+from contextlib import contextmanager
+
+from azure.core.settings import settings
+from opentelemetry import context, trace
+from opentelemetry.sdk.trace import ReadableSpan, SpanProcessor
+from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
+
+settings.tracing_implementation = "opentelemetry"
+
+
+class AzureSDKTraceFilter(SpanProcessor):
+    """Custom SpanProcessor to filter out Azure SDK traces except Service Bus.
+
+    It drops spans from Azure SDK libraries except Service Bus messages,
+    preventing them from being exported to Azure Monitor.
+    """
+
+    def __init__(self, wrapped_processor: SpanProcessor):
+        """Initialize with the actual processor to wrap.
+
+        Args:
+            wrapped_processor: The underlying processor (e.g., BatchSpanProcessor)
+        """
+        self.wrapped_processor = wrapped_processor
+
+    def on_start(
+        self, span: ReadableSpan, parent_context: context.Context = None
+    ) -> None:
+        """Called when a span is started."""
+        self.wrapped_processor.on_start(span, parent_context)
+
+    def on_end(self, span: ReadableSpan) -> None:
+        """Called when a span is ended. Filter based on span attributes."""
+        # Check if service bus span - always allow
+        if "servicebus.message" in span.name.lower():
+            self.wrapped_processor.on_end(span)
+            return
+
+        # Check if this is an Azure SDK span we want to suppress
+        instrumentation_scope = span.instrumentation_scope
+        if instrumentation_scope and instrumentation_scope.name:
+            # Suppress spans from Azure SDK libraries
+            if instrumentation_scope.name.startswith("azure."):
+                return  # Drop this span
+
+        # Pass through all other spans
+        self.wrapped_processor.on_end(span)
+
+    def shutdown(self) -> None:
+        """Shutdown the wrapped processor."""
+        self.wrapped_processor.shutdown()
+
+    def force_flush(self, timeout_millis: int = 30000) -> bool:
+        """Force flush the wrapped processor."""
+        return self.wrapped_processor.force_flush(timeout_millis)
+
+
+@contextmanager
+def set_trace_context(trace_parent: str, trace_state: str = ""):
+    """Context manager for setting the trace context
+
+    Args:
+        trace_parent (str): Trace parent ID
+        trace_state (str, optional): Trace state. Defaults to "".
+    """
+    carrier = {"traceparent": trace_parent, "tracestate": trace_state}
+    ctx = TraceContextTextMapPropagator().extract(carrier=carrier)
+
+    token = context.attach(ctx)  # attach context before run
+    try:
+        yield
+    finally:
+        context.detach(token)  # detach context after run
+
+
+def get_tracer(name: str):
+    tracer = trace.get_tracer(name)
+    return tracer
+
+
+def get_current_diagnostic_id() -> str:
+    """Gets diagnostic id from current span
+
+    The diagnostic id is a concatenation of operation-id and parent-id
+
+    Returns:
+        str: diagnostic id
+    """
+    span = trace.get_current_span()
+
+    if not span.is_recording():
+        return ""
+
+    operation_id = "{:016x}".format(span.context.trace_id)
+    parent_id = "{:016x}".format(span.context.span_id)
+
+    diagnostic_id = f"00-{operation_id}-{parent_id}-01"
+
+    return diagnostic_id

warpzone_sdk-15.0.0.dev1/warpzone/monitor/traces.py

@@ -1,51 +0,0 @@
-from contextlib import contextmanager
-
-from azure.core.settings import settings
-from opentelemetry import context, trace
-from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
-
-settings.tracing_implementation = "opentelemetry"
-
-
-@contextmanager
-def set_trace_context(trace_parent: str, trace_state: str = ""):
-    """Context manager for setting the trace context
-
-    Args:
-        trace_parent (str): Trace parent ID
-        trace_state (str, optional): Trace state. Defaults to "".
-    """
-    carrier = {"traceparent": trace_parent, "tracestate": trace_state}
-    ctx = TraceContextTextMapPropagator().extract(carrier=carrier)
-
-    token = context.attach(ctx)  # attach context before run
-    try:
-        yield
-    finally:
-        context.detach(token)  # detach context after run
-
-
-def get_tracer(name: str):
-    tracer = trace.get_tracer(name)
-    return tracer
-
-
-def get_current_diagnostic_id() -> str:
-    """Gets diagnostic id from current span
-
-    The diagnostic id is a concatenation of operation-id and parent-id
-
-    Returns:
-        str: diagnostic id
-    """
-    span = trace.get_current_span()
-
-    if not span.is_recording():
-        return ""
-
-    operation_id = "{:016x}".format(span.context.trace_id)
-    parent_id = "{:016x}".format(span.context.span_id)
-
-    diagnostic_id = f"00-{operation_id}-{parent_id}-01"
-
-    return diagnostic_id