warpzone-sdk 15.0.0.dev1__py3-none-any.whl → 15.0.0.dev3__py3-none-any.whl

warpzone/function/monitor.py

@@ -1,14 +1,34 @@
-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()
+# Lazy initialization flag to ensure configure_azure_monitor is called only once
+_monitor_configured = False
+
+
+def _ensure_monitor_configured():
+    """Configure Azure Monitor lazily on first use to ensure proper trace context."""
+    global _monitor_configured
+    if not _monitor_configured:
+        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
+
+        _monitor_configured = True
+
 
 SUBJECT_IDENTIFIER = "<Subject>"
 
@@ -45,16 +65,18 @@ def monitor(main: SingleArgumentCallable) -> Callable:
     """
 
     async def wrapper_async(arg, context: func.Context):
+        _ensure_monitor_configured()
         with run_in_trace_context(context):
             result = await main(arg)
             return result
 
     def wrapper(arg, context: func.Context):
+        _ensure_monitor_configured()
         with run_in_trace_context(context):
             result = main(arg)
             return result
 
-    if asyncio.iscoroutinefunction(main):
+    if inspect.iscoroutinefunction(main):
         return wrapper_async
     else:
         return wrapper

warpzone/monitor/traces.py

@@ -2,11 +2,59 @@ 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

{warpzone_sdk-15.0.0.dev1.dist-info → warpzone_sdk-15.0.0.dev3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: warpzone-sdk
-Version: 15.0.0.dev1
+Version: 15.0.0.dev3
 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.dist-info → warpzone_sdk-15.0.0.dev3.dist-info}/RECORD

@@ -17,7 +17,7 @@ warpzone/function/__init__.py,sha256=rJOZBpWsUgjMc7YtXMJ1rLGm45KB1AhDJ_Y2ISiSISc
 warpzone/function/checks.py,sha256=B9YqThymf16ac_fVAYKilv20ru5v9nwXgHlbxYIaG98,1018
 warpzone/function/functionize.py,sha256=bSV0QvwKbD9Vo3a_8cc1rgV2rzTdMMvidinyXItBfvs,2128
 warpzone/function/integrations.py,sha256=sDt2BTx6a4mVc-33wTITP9XQVPustwj7rkX4urTyOqo,4018
-warpzone/function/monitor.py,sha256=-WnbllgphJs1JRn_UxkLDHALM2UlrtEwy4ORh8eJ-xI,1660
+warpzone/function/monitor.py,sha256=gGWoxrOTaXofds8FJwPwGf58q0XV19Xwj9VRzhr0GM8,2594
 warpzone/function/process.py,sha256=nbUVywM8ChfUwuaqFisgaD98aNRgeZkK4g5sbtuBdRs,2339
 warpzone/function/processors/__init__.py,sha256=DhIdSWLBcIeSO8IJdxPqGIhgwwnkDN6_Xqwy93BCLeA,46
 warpzone/function/processors/dependencies.py,sha256=m17BwdKyQEvzCPxpQZpAW5l1uYRIHWmweDz3XJskmpA,1259
@@ -29,7 +29,7 @@ warpzone/healthchecks/__init__.py,sha256=9gc_Mt2szs8sDSwy0V4l3JZ6d9hX41xTpZCkDP2
 warpzone/healthchecks/model.py,sha256=mM7DnrirLbUpBPPfi82MUPP654D0eOR2_F65TmzsPD0,1187
 warpzone/monitor/__init__.py,sha256=ggI5fIUu-szgC44ICzuOmpYrIoVOKPbsMT3zza9ssD4,87
 warpzone/monitor/logs.py,sha256=V1A2ImqbPo4c-hDr2qWAZlpr2muyVar30eI9mF2_rzE,986
-warpzone/monitor/traces.py,sha256=SQckLC4LCEbgg79ZBnmYcm26IaxJfHDQ9GV7Huh2Uls,1393
+warpzone/monitor/traces.py,sha256=PhhYeZlPc7EOSq9KVsHdKOwEIivDHCV-RwZnDNVL7ic,3244
 warpzone/servicebus/data/__init__.py,sha256=lnc0uiaGLF0qMi_rWhCpRSFvaj0CJEiMCAl6Yqn1ZiA,21
 warpzone/servicebus/data/client.py,sha256=zECS3JwedhYnDk8PntYgIYpBF_uu9YN38KzpPFK7CKs,6511
 warpzone/servicebus/events/__init__.py,sha256=lnc0uiaGLF0qMi_rWhCpRSFvaj0CJEiMCAl6Yqn1ZiA,21
@@ -52,6 +52,6 @@ warpzone/tools/copy.py,sha256=5fddotMZkXZO8avzUbGOhvs0cp8mce95pNpy0oPVjnQ,2596
 warpzone/transform/__init__.py,sha256=ruGa7tl-v4ndlWpULE1jSGU_a4_iRc3V6eyNr5xKP9E,27
 warpzone/transform/data.py,sha256=Abb8PcrgMbbNCJkkIUdtrTHdlY0OfXid387qw1nDpFY,2362
 warpzone/transform/schema.py,sha256=nbSQtDMvXkyqGKuwhuFCF0WsEDsaNyoPYpMKvbsKlv8,2423
-warpzone_sdk-15.0.0.dev1.dist-info/METADATA,sha256=U7Mz9332k6VsVXQitNaurDeed-oD4QBFyEmPaMWJexM,7398
-warpzone_sdk-15.0.0.dev1.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
-warpzone_sdk-15.0.0.dev1.dist-info/RECORD,,
+warpzone_sdk-15.0.0.dev3.dist-info/METADATA,sha256=SbdclS53SblY02qVWfnpicAHmSQJXPqKIfz_gAxtaRM,7398
+warpzone_sdk-15.0.0.dev3.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
+warpzone_sdk-15.0.0.dev3.dist-info/RECORD,,