claude-mpm 3.9.11__py3-none-any.whl → 4.0.4__py3-none-any.whl

claude_mpm/core/lazy.py

@@ -15,18 +15,19 @@ import asyncio
 import functools
 import threading
 import time
-from typing import Any, Callable, Dict, Optional, Type, TypeVar, Generic
-from datetime import datetime
 from dataclasses import dataclass, field
+from datetime import datetime
+from typing import Any, Callable, Dict, Generic, Optional, Type, TypeVar
 
 from ..core.logger import get_logger
 
-T = TypeVar('T')
+T = TypeVar("T")
 
 
 @dataclass
 class LazyMetrics:
     """Metrics for lazy loading performance."""
+
     created_at: datetime = field(default_factory=datetime.now)
     first_access: Optional[datetime] = None
     initialization_time: float = 0.0
@@ -37,13 +38,13 @@ class LazyMetrics:
 
 class LazyService(Generic[T]):
     """Lazy loading wrapper for expensive services.
-    
+
     WHY this design:
     - Services are only initialized when first accessed
     - Thread-safe initialization ensures single instance
     - Transparent proxy pattern preserves service interface
     - Metrics tracking for performance monitoring
-    
+
     Example:
         # Wrap expensive service
         lazy_analyzer = LazyService(
@@ -51,31 +52,31 @@ class LazyService(Generic[T]):
             init_args=(),
             init_kwargs={'config': config}
         )
-        
+
         # Service is not initialized yet
         assert not lazy_analyzer.is_initialized
-        
+
         # First access triggers initialization
         result = lazy_analyzer.analyze()  # Initializes here
-        
+
         # Subsequent accesses use cached instance
         result2 = lazy_analyzer.analyze()  # No initialization
     """
-    
+
     def __init__(
         self,
         service_class: Type[T],
         init_args: tuple = (),
         init_kwargs: dict = None,
         name: Optional[str] = None,
-        eager: bool = False
+        eager: bool = False,
     ):
         """Initialize lazy service wrapper.
-        
+
         Args:
             service_class: The class to lazily instantiate
             init_args: Positional arguments for initialization
-            init_kwargs: Keyword arguments for initialization  
+            init_kwargs: Keyword arguments for initialization
             name: Optional name for logging
             eager: If True, initialize immediately (for testing)
         """
@@ -84,92 +85,93 @@ class LazyService(Generic[T]):
         self._init_kwargs = init_kwargs or {}
         self._name = name or service_class.__name__
         self._eager = eager
-        
+
         self._instance: Optional[T] = None
         self._lock = threading.RLock()
         self._metrics = LazyMetrics()
         self._logger = get_logger(f"lazy.{self._name}")
-        
+
         # Initialize immediately if eager mode
         if eager:
             self._ensure_initialized()
-    
+
     @property
     def is_initialized(self) -> bool:
         """Check if service has been initialized."""
         return self._instance is not None
-    
+
     @property
     def metrics(self) -> LazyMetrics:
         """Get lazy loading metrics."""
         return self._metrics
-    
+
     def _ensure_initialized(self) -> T:
         """Ensure service is initialized, creating if necessary.
-        
+
         Thread-safe initialization with metrics tracking.
         """
         if self._instance is not None:
             return self._instance
-        
+
         with self._lock:
             # Double-check pattern for thread safety
             if self._instance is not None:
                 return self._instance
-            
+
             # Track initialization
             start_time = time.time()
             if self._metrics.first_access is None:
                 self._metrics.first_access = datetime.now()
-            
+
             try:
                 self._logger.debug(f"Initializing lazy service: {self._name}")
-                
+
                 # Create the actual service instance
                 self._instance = self._service_class(
-                    *self._init_args,
-                    **self._init_kwargs
+                    *self._init_args, **self._init_kwargs
                 )
-                
+
                 # Update metrics
                 self._metrics.initialization_time = time.time() - start_time
                 self._metrics.is_initialized = True
-                
+
                 self._logger.info(
                     f"Lazy service {self._name} initialized in "
                     f"{self._metrics.initialization_time:.2f}s"
                 )
-                
+
                 return self._instance
-                
+
             except Exception as e:
                 self._metrics.initialization_error = e
                 self._logger.error(f"Failed to initialize {self._name}: {e}")
                 raise
-    
+
     def __getattr__(self, name: str) -> Any:
         """Proxy attribute access to the underlying service.
-        
+
         This is called when an attribute is not found on LazyService itself.
         It ensures initialization and forwards the request.
         """
         # Avoid recursion for internal attributes
-        if name.startswith('_'):
-            raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'")
-        
+        if name.startswith("_"):
+            raise AttributeError(
+                f"'{type(self).__name__}' object has no attribute '{name}'"
+            )
+
         # Initialize if needed
         instance = self._ensure_initialized()
         self._metrics.access_count += 1
-        
+
         # Forward attribute access
         return getattr(instance, name)
-    
+
     def __call__(self, *args, **kwargs) -> Any:
         """Make LazyService callable if underlying service is."""
         instance = self._ensure_initialized()
         self._metrics.access_count += 1
         return instance(*args, **kwargs)
-    
+
     def __repr__(self) -> str:
         """String representation of lazy service."""
         status = "initialized" if self.is_initialized else "pending"
@@ -178,69 +180,69 @@ class LazyService(Generic[T]):
 
 class LazyServiceRegistry:
     """Registry for managing lazy services.
-    
+
     WHY registry pattern:
     - Central management of all lazy services
     - Bulk initialization for testing
     - Performance metrics aggregation
     - Service dependency resolution
     """
-    
+
     def __init__(self):
         self._services: Dict[str, LazyService] = {}
         self._logger = get_logger("lazy_registry")
         self._lock = threading.Lock()
-    
+
     def register(
         self,
         name: str,
         service_class: Type,
         init_args: tuple = (),
         init_kwargs: dict = None,
-        eager: bool = False
+        eager: bool = False,
     ) -> LazyService:
         """Register a lazy service.
-        
+
         Args:
             name: Unique name for the service
             service_class: The service class to wrap
             init_args: Initialization arguments
             init_kwargs: Initialization keyword arguments
             eager: Whether to initialize immediately
-        
+
         Returns:
             The registered LazyService instance
         """
         with self._lock:
             if name in self._services:
                 self._logger.warning(f"Overwriting existing service: {name}")
-            
+
             service = LazyService(
                 service_class=service_class,
                 init_args=init_args,
                 init_kwargs=init_kwargs,
                 name=name,
-                eager=eager
+                eager=eager,
             )
-            
+
             self._services[name] = service
             self._logger.debug(f"Registered lazy service: {name}")
             return service
-    
+
     def get(self, name: str) -> Optional[LazyService]:
         """Get a registered service by name."""
         return self._services.get(name)
-    
+
     def initialize_all(self) -> Dict[str, float]:
         """Initialize all registered services.
-        
+
         Useful for testing or preloading.
-        
+
         Returns:
             Dictionary mapping service names to initialization times
         """
         init_times = {}
-        
+
         for name, service in self._services.items():
             if not service.is_initialized:
                 try:
@@ -249,13 +251,13 @@ class LazyServiceRegistry:
                 except Exception as e:
                     self._logger.error(f"Failed to initialize {name}: {e}")
                     init_times[name] = -1.0
-        
+
         return init_times
-    
+
     def get_metrics(self) -> Dict[str, Dict[str, Any]]:
         """Get metrics for all registered services."""
         metrics = {}
-        
+
         for name, service in self._services.items():
             m = service.metrics
             metrics[name] = {
@@ -263,11 +265,11 @@ class LazyServiceRegistry:
                 "first_access": m.first_access.isoformat() if m.first_access else None,
                 "initialization_time": m.initialization_time,
                 "access_count": m.access_count,
-                "has_error": m.initialization_error is not None
+                "has_error": m.initialization_error is not None,
             }
-        
+
         return metrics
-    
+
     def reset(self):
         """Clear all registered services."""
         with self._lock:
@@ -283,37 +285,37 @@ def lazy_load(
     service_class: Type,
     name: Optional[str] = None,
     init_args: tuple = (),
-    init_kwargs: dict = None
+    init_kwargs: dict = None,
 ) -> LazyService:
     """Convenience function to create and register a lazy service.
-    
+
     Args:
         service_class: The service class to wrap
         name: Optional name (defaults to class name)
         init_args: Initialization arguments
         init_kwargs: Initialization keyword arguments
-    
+
     Returns:
         Registered LazyService instance
-    
+
     Example:
         # Create lazy service
         analyzer = lazy_load(
             ProjectAnalyzer,
             init_kwargs={'config': config}
         )
-        
+
         # Use like normal service
         result = analyzer.analyze_project()
     """
     if name is None:
         name = service_class.__name__
-    
+
     return _registry.register(
         name=name,
         service_class=service_class,
         init_args=init_args,
-        init_kwargs=init_kwargs
+        init_kwargs=init_kwargs,
     )
 
 
@@ -334,13 +336,13 @@ def initialize_all_services() -> Dict[str, float]:
 
 class lazy_property:
     """Decorator for lazy property initialization.
-    
+
     WHY decorator pattern:
     - Pythonic approach for lazy attributes
     - Caches expensive computations
     - Thread-safe initialization
     - Transparent to callers
-    
+
     Example:
         class MyService:
             @lazy_property
@@ -348,28 +350,28 @@ class lazy_property:
                 # This only runs once
                 return self._compute_expensive_data()
     """
-    
+
     def __init__(self, func: Callable) -> None:
         self.func = func
         self.lock = threading.RLock()
         functools.update_wrapper(self, func)
-    
+
     def __get__(self, obj: Any, objtype: Type = None) -> Any:
         if obj is None:
             return self
-        
+
         # Check if already cached
         val = obj.__dict__.get(self.func.__name__)
         if val is not None:
             return val
-        
+
         # Thread-safe initialization
         with self.lock:
             # Double-check pattern
             val = obj.__dict__.get(self.func.__name__)
             if val is not None:
                 return val
-            
+
             # Compute and cache
             val = self.func(obj)
             obj.__dict__[self.func.__name__] = val
@@ -378,84 +380,82 @@ class lazy_property:
 
 class AsyncLazyService(Generic[T]):
     """Async version of LazyService for async services.
-    
+
     Supports services that require async initialization.
     """
-    
+
     def __init__(
         self,
         service_class: Type[T],
         init_args: tuple = (),
         init_kwargs: dict = None,
-        name: Optional[str] = None
+        name: Optional[str] = None,
     ):
         self._service_class = service_class
         self._init_args = init_args
         self._init_kwargs = init_kwargs or {}
         self._name = name or service_class.__name__
-        
+
         self._instance: Optional[T] = None
         self._lock = asyncio.Lock()
         self._metrics = LazyMetrics()
         self._logger = get_logger(f"async_lazy.{self._name}")
-    
+
     @property
     def is_initialized(self) -> bool:
         """Check if service has been initialized."""
         return self._instance is not None
-    
+
     async def _ensure_initialized(self) -> T:
         """Ensure service is initialized asynchronously."""
         if self._instance is not None:
             return self._instance
-        
+
         async with self._lock:
             if self._instance is not None:
                 return self._instance
-            
+
             start_time = time.time()
             if self._metrics.first_access is None:
                 self._metrics.first_access = datetime.now()
-            
+
             try:
                 self._logger.debug(f"Async initializing: {self._name}")
-                
+
                 # Handle both async and sync initialization
                 if asyncio.iscoroutinefunction(self._service_class):
                     self._instance = await self._service_class(
-                        *self._init_args,
-                        **self._init_kwargs
+                        *self._init_args, **self._init_kwargs
                     )
                 else:
                     # Run sync initialization in executor
                     loop = asyncio.get_event_loop()
                     self._instance = await loop.run_in_executor(
-                        None,
-                        self._service_class,
-                        *self._init_args,
-                        **self._init_kwargs
+                        None, self._service_class, *self._init_args, **self._init_kwargs
                     )
-                
+
                 self._metrics.initialization_time = time.time() - start_time
                 self._metrics.is_initialized = True
-                
+
                 self._logger.info(
                     f"Async service {self._name} initialized in "
                     f"{self._metrics.initialization_time:.2f}s"
                 )
-                
+
                 return self._instance
-                
+
             except Exception as e:
                 self._metrics.initialization_error = e
                 self._logger.error(f"Failed to initialize {self._name}: {e}")
                 raise
-    
+
     def __getattr__(self, name: str) -> Any:
         """Proxy attribute access to the underlying service."""
-        if name.startswith('_'):
-            raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'")
-        
+        if name.startswith("_"):
+            raise AttributeError(
+                f"'{type(self).__name__}' object has no attribute '{name}'"
+            )
+
         async def async_wrapper(*args, **kwargs):
             instance = await self._ensure_initialized()
             self._metrics.access_count += 1
@@ -463,5 +463,5 @@ class AsyncLazyService(Generic[T]):
             if asyncio.iscoroutinefunction(attr):
                 return await attr(*args, **kwargs)
             return attr(*args, **kwargs)
-        
-        return async_wrapper
+
+        return async_wrapper