azpaddypy 0.1.9__py3-none-any.whl → 0.2.0__py3-none-any.whl

azpaddypy/mgmt/logging.py

@@ -21,6 +21,33 @@ class AzureLogger:
     using Azure Monitor OpenTelemetry integration with advanced tracing capabilities.
     """
 
+    _instances = {}
+    _tracer = None
+
+    @classmethod
+    def get_instance(
+        cls,
+        service_name: str,
+        service_version: str = "1.0.0",
+        connection_string: Optional[str] = None,
+        log_level: int = logging.INFO,
+        enable_console_logging: bool = True,
+        custom_resource_attributes: Optional[Dict[str, str]] = None,
+    ):
+        """
+        Get or create a singleton AzureLogger instance for a given service_name.
+        """
+        if service_name not in cls._instances:
+            cls._instances[service_name] = cls(
+                service_name=service_name,
+                service_version=service_version,
+                connection_string=connection_string,
+                log_level=log_level,
+                enable_console_logging=enable_console_logging,
+                custom_resource_attributes=custom_resource_attributes,
+            )
+        return cls._instances[service_name]
+
     def __init__(
         self,
         service_name: str,
@@ -30,17 +57,26 @@ class AzureLogger:
         enable_console_logging: bool = True,
         custom_resource_attributes: Optional[Dict[str, str]] = None,
     ):
-        """
-        Initialize the Azure Logger with OpenTelemetry tracing
+        # On first initialization, clean up all lingering loggers, tracers, and meters
+        if not AzureLogger._instances:
+            # Remove all handlers from all loggers except the one being initialized
+            for name, logger in logging.root.manager.loggerDict.items():
+                if isinstance(logger, logging.Logger) and name != service_name:
+                    logger.handlers.clear()
+            # Reset OpenTelemetry tracer and meter providers
+            try:
+                from opentelemetry import trace, metrics
+                from opentelemetry.sdk.trace import TracerProvider
+                from opentelemetry.sdk.metrics import MeterProvider
 
-        Args:
-            service_name: Name of your service/application
-            service_version: Version of your service
-            connection_string: Application Insights connection string
-            log_level: Logging level (default: INFO)
-            enable_console_logging: Enable console output for local development
-            custom_resource_attributes: Additional resource attributes
-        """
+                trace.set_tracer_provider(TracerProvider())
+                metrics.set_meter_provider(MeterProvider())
+            except Exception as e:
+                print(f"Warning: Failed to reset OpenTelemetry providers: {e}")
+        # Prevent re-initialization if already exists
+        if hasattr(self, "_initialized") and self._initialized:
+            return
+        self._initialized = True
         self.service_name = service_name
         self.service_version = service_version
         self.connection_string = connection_string or os.getenv(
@@ -75,19 +111,19 @@ class AzureLogger:
                 "Warning: No Application Insights connection string found. Telemetry disabled."
             )
 
-        # Set up logger without instrumentor
+        # Set up logger without instrumentation
         self.logger = logging.getLogger(service_name)
         self.logger.setLevel(log_level)
 
-        # Clear existing handlers to avoid duplicates
-        self.logger.handlers.clear()
-
-        # Add console handler for local development
-        if enable_console_logging:
-            self._setup_console_handler()
+        # Only add handlers if none exist
+        if not self.logger.handlers:
+            if enable_console_logging:
+                self._setup_console_handler()
 
-        # Get tracer for manual span creation
-        self.tracer = trace.get_tracer(__name__)
+        # Get tracer for manual span creation (singleton per class)
+        if AzureLogger._tracer is None:
+            AzureLogger._tracer = trace.get_tracer(__name__)
+        self.tracer = AzureLogger._tracer
 
         # Initialize correlation context
         self._correlation_id = None
@@ -174,7 +210,7 @@ class AzureLogger:
             enhanced_extra.update(extra)
 
         return enhanced_extra
-    
+
     def exception(self, message: str, extra: Optional[Dict[str, Any]] = None):
         """Log debug message with enhanced context"""
         self.logger.exception(message, extra=self._enhance_extra(extra))
@@ -606,6 +642,36 @@ class AzureLogger:
             except Exception as e:
                 self.warning(f"Failed to flush telemetry: {e}")
 
+    async def flush_async(self):
+        """Asynchronously flush all pending telemetry data"""
+        if self._telemetry_enabled:
+            try:
+                from opentelemetry.sdk.trace import TracerProvider
+
+                tracer_provider = trace.get_tracer_provider()
+                if hasattr(tracer_provider, "force_flush"):
+                    await asyncio.to_thread(
+                        tracer_provider.force_flush, timeout_millis=5000
+                    )
+            except Exception as e:
+                self.warning(f"Failed to async flush telemetry: {e}")
+
+    def add_handler(self, handler: logging.Handler):
+        """Add a logging handler to the logger."""
+        self.logger.addHandler(handler)
+
+    def remove_handler(self, handler: logging.Handler):
+        """Remove a logging handler from the logger."""
+        self.logger.removeHandler(handler)
+
+    def list_handlers(self) -> list:
+        """List all handlers attached to the logger."""
+        return self.logger.handlers
+
+    def clear_handlers(self):
+        """Remove all handlers from the logger."""
+        self.logger.handlers.clear()
+
 
 # Factory functions for easy instantiation
 def create_app_logger(

{azpaddypy-0.1.9.dist-info → azpaddypy-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azpaddypy
-Version: 0.1.9
+Version: 0.2.0
 Summary: Add your description here
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.11

azpaddypy-0.2.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+azpaddypy/mgmt/__init__.py,sha256=5-0eZuJMZlCONZNJ5hEvWXfvLIM36mg7FsEVs32bY_A,183
+azpaddypy/mgmt/logging.py,sha256=UYlep7e3TddWo1EtJliGg45Zv-22iSbIYdC8Iz1STKU,28462
+azpaddypy/test_function/__init__.py,sha256=0NjUl36wvUWV79GpRwBFkgkBaC6uDZsTdaSVOIHMFEU,3481
+azpaddypy/test_function/function_app.py,sha256=6nX54-iq0L1l_hZpD6E744-j79oLxdaldFyWDCpwH7c,3867
+azpaddypy-0.2.0.dist-info/licenses/LICENSE,sha256=hQ6t0g2QaewGCQICHqTckBFbMVakGmoyTAzDpmEYV4c,1089
+azpaddypy-0.2.0.dist-info/METADATA,sha256=EDK4QI8b__iy5AX6PmYxKsUmegQStO_y9WUzQiAZ5zw,304
+azpaddypy-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+azpaddypy-0.2.0.dist-info/top_level.txt,sha256=hsDuboDhT61320ML8X479ezSTwT3rrlDWz1_Z45B2cs,10
+azpaddypy-0.2.0.dist-info/RECORD,,

azpaddypy-0.1.9.dist-info/RECORD

@@ -1,9 +0,0 @@
-azpaddypy/mgmt/__init__.py,sha256=5-0eZuJMZlCONZNJ5hEvWXfvLIM36mg7FsEVs32bY_A,183
-azpaddypy/mgmt/logging.py,sha256=TCOjec_DSvN4wqqzGihn24XXjUZ2NpFm2cG3UxYUWSA,25666
-azpaddypy/test_function/__init__.py,sha256=0NjUl36wvUWV79GpRwBFkgkBaC6uDZsTdaSVOIHMFEU,3481
-azpaddypy/test_function/function_app.py,sha256=6nX54-iq0L1l_hZpD6E744-j79oLxdaldFyWDCpwH7c,3867
-azpaddypy-0.1.9.dist-info/licenses/LICENSE,sha256=hQ6t0g2QaewGCQICHqTckBFbMVakGmoyTAzDpmEYV4c,1089
-azpaddypy-0.1.9.dist-info/METADATA,sha256=srjAa16qE_vCvXvKucLGUCC00EnxPB7CIU4L80_WWwQ,304
-azpaddypy-0.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-azpaddypy-0.1.9.dist-info/top_level.txt,sha256=hsDuboDhT61320ML8X479ezSTwT3rrlDWz1_Z45B2cs,10
-azpaddypy-0.1.9.dist-info/RECORD,,