runbooks 1.1.4__py3-none-any.whl → 1.1.5__py3-none-any.whl

runbooks/common/module_cli_base.py

@@ -21,12 +21,18 @@ from rich.panel import Panel
 from rich.table import Table
 
 from runbooks.common.rich_utils import (
-    console, print_header, print_success, print_error, print_warning, print_info,
-    create_table, create_progress_bar, format_cost, STATUS_INDICATORS
-)
-from runbooks.common.cli_decorators import (
-    common_aws_options, rich_output_options, enterprise_safety_options
+    console,
+    print_header,
+    print_success,
+    print_error,
+    print_warning,
+    print_info,
+    create_table,
+    create_progress_bar,
+    format_cost,
+    STATUS_INDICATORS,
 )
+from runbooks.common.cli_decorators import common_aws_options, rich_output_options, enterprise_safety_options
 from runbooks.common.error_handling import handle_aws_errors, handle_validation_errors
 from runbooks.common.profile_utils import get_profile_for_operation, validate_profile_access_decorator
 from runbooks.common.business_logic import BusinessMetrics, OptimizationResult, UniversalBusinessLogic
@@ -35,6 +41,7 @@ from runbooks.common.business_logic import BusinessMetrics, OptimizationResult,
 @dataclass
 class ModuleConfig:
     """Configuration for runbooks module CLI."""
+
     name: str
     version: str
     description: str
@@ -69,11 +76,7 @@ class ModuleCLIBase(abc.ABC):
         """Print standardized module header with Rich formatting."""
         print_header(self.config.name, self.config.version)
         self.console.print(
-            Panel(
-                f"[cyan]{self.config.description}[/cyan]",
-                title=f"🚀 {self.config.name} Module",
-                border_style="blue"
-            )
+            Panel(f"[cyan]{self.config.description}[/cyan]", title=f"🚀 {self.config.name} Module", border_style="blue")
         )
 
     def validate_prerequisites(self, profile: Optional[str] = None) -> bool:
@@ -88,15 +91,12 @@ class ModuleCLIBase(abc.ABC):
         """
         try:
             # Validate AWS profile access
-            selected_profile = get_profile_for_operation(
-                self.config.primary_operation_type,
-                profile
-            )
+            selected_profile = get_profile_for_operation(self.config.primary_operation_type, profile)
 
             if not validate_profile_access(selected_profile, self.config.name):
                 return False
 
-            self._session_data['validated_profile'] = selected_profile
+            self._session_data["validated_profile"] = selected_profile
             return True
 
         except Exception as e:
@@ -125,8 +125,7 @@ class ModuleCLIBase(abc.ABC):
         """
         pass
 
-    def create_results_table(self, title: str, data: List[Dict[str, Any]],
-                           columns: List[str]) -> Table:
+    def create_results_table(self, title: str, data: List[Dict[str, Any]], columns: List[str]) -> Table:
         """
         Create standardized results table with Rich formatting.
 
@@ -138,18 +137,19 @@ class ModuleCLIBase(abc.ABC):
         Returns:
             Configured Rich Table object
         """
-        table = create_table(title=title, columns=[
-            {"name": col, "style": "cyan" if "name" in col.lower() else "default"}
-            for col in columns
-        ])
+        table = create_table(
+            title=title,
+            columns=[{"name": col, "style": "cyan" if "name" in col.lower() else "default"} for col in columns],
+        )
 
         for row in data:
             table.add_row(*[str(row.get(col, "N/A")) for col in columns])
 
         return table
 
-    def export_results(self, results: Dict[str, Any], format_type: str = "json",
-                      output_path: Optional[str] = None) -> bool:
+    def export_results(
+        self, results: Dict[str, Any], format_type: str = "json", output_path: Optional[str] = None
+    ) -> bool:
         """
         Export results in specified format.
 
@@ -166,9 +166,7 @@ class ModuleCLIBase(abc.ABC):
             return False
 
         try:
-            export_handler = self.business_logic.create_export_handler(
-                format_type, output_path
-            )
+            export_handler = self.business_logic.create_export_handler(format_type, output_path)
 
             success = export_handler.export_data(results)
 
@@ -205,6 +203,7 @@ class ModuleCLIBase(abc.ABC):
         Returns:
             Decorated click command function
         """
+
         @click.command(name=command_name)
         @common_aws_options
         @rich_output_options
@@ -212,8 +211,7 @@ class ModuleCLIBase(abc.ABC):
         @handle_aws_errors(module_name=self.config.name)
         @handle_validation_errors
         @validate_profile_access_decorator(operation_type=self.config.primary_operation_type)
-        def standardized_command(profile=None, region=None, dry_run=True,
-                                export_format=None, quiet=False, **kwargs):
+        def standardized_command(profile=None, region=None, dry_run=True, export_format=None, quiet=False, **kwargs):
             """Execute standardized module operation with enterprise safety."""
 
             # Module header
@@ -227,19 +225,11 @@ class ModuleCLIBase(abc.ABC):
             # Execute with performance monitoring
             try:
                 with create_progress_bar() as progress:
-                    task = progress.add_task(
-                        f"[cyan]Executing {self.config.name} operation...",
-                        total=100
-                    )
+                    task = progress.add_task(f"[cyan]Executing {self.config.name} operation...", total=100)
 
                     # Execute primary operation
                     results = self.execute_primary_operation(
-                        profile=profile,
-                        region=region,
-                        dry_run=dry_run,
-                        progress=progress,
-                        task=task,
-                        **kwargs
+                        profile=profile, region=region, dry_run=dry_run, progress=progress, task=task, **kwargs
                     )
 
                     progress.update(task, completed=100)
@@ -252,9 +242,7 @@ class ModuleCLIBase(abc.ABC):
                 metrics = self.calculate_business_metrics(results)
 
                 if not quiet and metrics.annual_savings > 0:
-                    print_success(
-                        f"💰 Potential annual savings: {format_cost(metrics.annual_savings)}"
-                    )
+                    print_success(f"💰 Potential annual savings: {format_cost(metrics.annual_savings)}")
 
                 # Export if requested
                 if export_format:
@@ -294,17 +282,13 @@ class AnalysisModuleCLI(ModuleCLIBase):
         if metrics.roi_percentage > 0:
             summary_content.append(f"📊 ROI: {metrics.roi_percentage:.1f}%")
 
-        if hasattr(metrics, 'resources_analyzed'):
+        if hasattr(metrics, "resources_analyzed"):
             summary_content.append(f"🔍 Resources Analyzed: {metrics.resources_analyzed:,}")
 
         summary_content.append(f"⏱️  Execution Time: <{self.config.performance_target_seconds}s target")
         summary_content.append(f"✅ Confidence: {metrics.confidence_level:.1f}%")
 
-        return Panel(
-            "\n".join(summary_content),
-            title="📈 Executive Summary",
-            border_style="green"
-        )
+        return Panel("\n".join(summary_content), title="📈 Executive Summary", border_style="green")
 
 
 class OperationsModuleCLI(ModuleCLIBase):
@@ -330,7 +314,7 @@ class OperationsModuleCLI(ModuleCLIBase):
             True if operation is safe to proceed
         """
         # Always require explicit approval for destructive operations
-        destructive_operations = ['delete', 'terminate', 'remove', 'destroy']
+        destructive_operations = ["delete", "terminate", "remove", "destroy"]
 
         if any(op in operation.lower() for op in destructive_operations):
             print_warning(f"⚠️  Destructive operation requested: {operation}")
@@ -360,19 +344,14 @@ class OperationsModuleCLI(ModuleCLIBase):
             "timestamp": datetime.now().isoformat(),
             "module": self.config.name,
             "operation": operation,
-            "user": self._session_data.get('validated_profile', 'unknown'),
-            "resources_affected": results.get('resources_affected', []),
-            "success": results.get('success', False),
-            "execution_time": results.get('execution_time_seconds', 0)
+            "user": self._session_data.get("validated_profile", "unknown"),
+            "resources_affected": results.get("resources_affected", []),
+            "success": results.get("success", False),
+            "execution_time": results.get("execution_time_seconds", 0),
         }
 
         return json.dumps(audit_entry, indent=2)
 
 
 # Export standardized classes for module implementations
-__all__ = [
-    "ModuleConfig",
-    "ModuleCLIBase",
-    "AnalysisModuleCLI",
-    "OperationsModuleCLI"
-]
+__all__ = ["ModuleConfig", "ModuleCLIBase", "AnalysisModuleCLI", "OperationsModuleCLI"]