azpaddypy 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

{azpaddypy-0.1.0.dist-info → azpaddypy-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azpaddypy
-Version: 0.1.0
+Version: 0.1.1
 Summary: Add your description here
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

azpaddypy-0.1.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+azpaddypy-0.1.1.dist-info/licenses/LICENSE,sha256=hQ6t0g2QaewGCQICHqTckBFbMVakGmoyTAzDpmEYV4c,1089
+mgmt/__init__.py,sha256=5-0eZuJMZlCONZNJ5hEvWXfvLIM36mg7FsEVs32bY_A,183
+mgmt/logging.py,sha256=jQ1jIkdSf7p44_oBJZD3rPodtwkhY5Y9rwRNafwXgcI,21061
+azpaddypy-0.1.1.dist-info/METADATA,sha256=Zpx4YyLu-lUev_RmVmIjXERbXWzzIqIM-sgF02snlLI,256
+azpaddypy-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+azpaddypy-0.1.1.dist-info/top_level.txt,sha256=Xqg3gg53XEnWtdRarYe2ubJO0NDuer-pkr8RN2y7G-c,5
+azpaddypy-0.1.1.dist-info/RECORD,,

azpaddypy-0.1.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+mgmt

{azpaddypy/mgmt → mgmt}/logging.py

@@ -10,6 +10,8 @@ from opentelemetry import trace
 from opentelemetry.trace import Status, StatusCode, Span
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.resources import Resource
+from opentelemetry import baggage
+from opentelemetry.context import Context
 
 
 class AzureLogger:
@@ -110,6 +112,40 @@ class AzureLogger:
         """Get current correlation ID"""
         return self._correlation_id
 
+    def set_baggage(self, key: str, value: str) -> Context:
+        """
+        Set a baggage item in the current context
+
+        Args:
+            key: Baggage key
+            value: Baggage value
+
+        Returns:
+            Updated context with the new baggage item
+        """
+        return baggage.set_baggage(key, value)
+
+    def get_baggage(self, key: str) -> Optional[str]:
+        """
+        Get a baggage item from the current context
+
+        Args:
+            key: Baggage key
+
+        Returns:
+            Baggage value if exists, None otherwise
+        """
+        return baggage.get_baggage(key)
+
+    def get_all_baggage(self) -> Dict[str, str]:
+        """
+        Get all baggage items from the current context
+
+        Returns:
+            Dictionary of all baggage items
+        """
+        return dict(baggage.get_all())
+
     def _enhance_extra(self, extra: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
         """Enhance log extra data with correlation and service info"""
         enhanced_extra = {
@@ -128,6 +164,11 @@ class AzureLogger:
             enhanced_extra["trace_id"] = format(span_context.trace_id, "032x")
             enhanced_extra["span_id"] = format(span_context.span_id, "016x")
 
+        # Add baggage items
+        baggage_items = self.get_all_baggage()
+        if baggage_items:
+            enhanced_extra["baggage"] = baggage_items
+
         if extra:
             enhanced_extra.update(extra)
 

azpaddypy-0.1.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-azpaddypy/mgmt/__init__.py,sha256=5-0eZuJMZlCONZNJ5hEvWXfvLIM36mg7FsEVs32bY_A,183
-azpaddypy/mgmt/logging.py,sha256=bpxGz674_1_Re5LzC1b6vslImeUQXSPTdBfHbt81jI0,19942
-azpaddypy-0.1.0.dist-info/licenses/LICENSE,sha256=hQ6t0g2QaewGCQICHqTckBFbMVakGmoyTAzDpmEYV4c,1089
-azpaddypy-0.1.0.dist-info/METADATA,sha256=IYtfH4emxwpYEuSY1vHij7D4ZeLpYlNPvzaOiS6hcaQ,256
-azpaddypy-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-azpaddypy-0.1.0.dist-info/top_level.txt,sha256=hsDuboDhT61320ML8X479ezSTwT3rrlDWz1_Z45B2cs,10
-azpaddypy-0.1.0.dist-info/RECORD,,

azpaddypy-0.1.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-azpaddypy