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

runbooks/inventory/vpc_flow_analyzer.py

@@ -609,7 +609,7 @@ class VPCFlowAnalyzer(BaseInventory):
                 if i != j:  # Cross-AZ traffic
                     # REMOVED: Random traffic simulation violates enterprise standards
                     # TODO: Calculate actual cross-AZ traffic from VPC Flow Logs
-                    
+
                     traffic_bytes = 500000 * time_range_hours  # Deterministic baseline
                     az_pair = f"{source_az}-to-{dest_az}"
 

runbooks/main.py

@@ -22,45 +22,46 @@ Enterprise Features Preserved:
 - Safety controls and audit trails
 """
 
+import os
 import sys
 from datetime import datetime
 from pathlib import Path
 from typing import Optional
-import os
 
 import click
 from loguru import logger
 
 try:
     from rich.console import Console
-    from rich.table import Table
     from rich.markup import escape
+    from rich.table import Table
+
     _HAS_RICH = True
 except ImportError:
     _HAS_RICH = False
+
     # Fallback console implementation
     class Console:
         def print(self, *args, **kwargs):
             output = " ".join(str(arg) for arg in args)
             print(output)
 
+
 import boto3
 
 from runbooks import __version__
 from runbooks.cli.registry import DRYCommandRegistry
-from runbooks.common.performance_monitor import get_performance_benchmark
-
-# Initialize Rich console
-console = Console()
-
-# Common decorators and options
 from runbooks.common.decorators import (
     common_aws_options,
-    common_output_options,
     common_filter_options,
+    common_output_options,
+    error_handler,
     performance_timing,
-    error_handler
 )
+from runbooks.common.performance_monitor import get_performance_benchmark
+
+# Initialize Rich console
+console = Console()
 
 
 def display_banner():
@@ -81,9 +82,12 @@ def display_banner():
 @click.group()
 @click.version_option(version=__version__)
 @click.option("--debug", is_flag=True, help="Enable debug logging")
-@click.option("--log-level",
-              type=click.Choice(['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL']),
-              default='INFO', help="Set logging level")
+@click.option(
+    "--log-level",
+    type=click.Choice(["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]),
+    default="INFO",
+    help="Set logging level",
+)
 @click.option("--json-output", is_flag=True, help="Enable structured JSON output for programmatic use")
 @common_aws_options
 @click.option("--config", type=click.Path(), help="Configuration file path")
@@ -120,22 +124,28 @@ def main(ctx, debug, log_level, json_output, profile, region, dry_run, config):
 
     # Configure logging
     if debug:
-        log_level = 'DEBUG'
+        log_level = "DEBUG"
 
     logger.remove()
-    logger.add(sys.stderr, level=log_level, format="<green>{time}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>")
+    logger.add(
+        sys.stderr,
+        level=log_level,
+        format="<green>{time}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
+    )
 
     # Store global options in context
-    ctx.obj.update({
-        'debug': debug,
-        'log_level': log_level,
-        'json_output': json_output,
-        'profile': profile,
-        'region': region,
-        'dry_run': dry_run,
-        'config': config,
-        'start_time': datetime.now()
-    })
+    ctx.obj.update(
+        {
+            "debug": debug,
+            "log_level": log_level,
+            "json_output": json_output,
+            "profile": profile,
+            "region": region,
+            "dry_run": dry_run,
+            "config": config,
+            "start_time": datetime.now(),
+        }
+    )
 
     # Display banner for interactive use
     if not json_output and ctx.invoked_subcommand is None:
@@ -167,18 +177,19 @@ except Exception as e:
 @click.pass_context
 def version(ctx):
     """Display version information and performance metrics."""
-    performance_data = get_performance_benchmark()
+    performance_data = get_performance_benchmark("main")
 
     version_info = {
         "version": __version__,
         "python_version": sys.version.split()[0],
         "platform": sys.platform,
         "commands_loaded": len(DRYCommandRegistry.list_commands()),
-        "performance": performance_data
+        "performance": performance_data,
     }
 
-    if ctx.obj.get('json_output'):
+    if ctx.obj.get("json_output"):
         import json
+
         click.echo(json.dumps(version_info, indent=2))
     else:
         console.print(f"[bold]CloudOps Runbooks v{__version__}[/bold]")
@@ -187,23 +198,27 @@ def version(ctx):
         console.print(f"Commands: {version_info['commands_loaded']} groups loaded")
 
         if performance_data:
-            console.print(f"Performance: {performance_data.get('avg_execution_time', 'N/A')}s avg execution")
+            avg_time = getattr(performance_data, "avg_execution_time", "N/A")
+            console.print(f"Performance: {avg_time}s avg execution")
 
 
 @main.command()
-@click.option("--format", "output_format", type=click.Choice(['table', 'json', 'yaml']),
-              default='table', help="Output format")
+@click.option(
+    "--format", "output_format", type=click.Choice(["table", "json", "yaml"]), default="table", help="Output format"
+)
 @click.pass_context
 def list_commands(ctx, output_format):
     """List all available commands and their descriptions."""
     commands = DRYCommandRegistry.register_commands()
 
-    if output_format == 'json':
+    if output_format == "json":
         import json
+
         command_data = {name: cmd.get_short_help_str() for name, cmd in commands.items()}
         click.echo(json.dumps(command_data, indent=2))
-    elif output_format == 'yaml':
+    elif output_format == "yaml":
         import yaml
+
         command_data = {name: cmd.get_short_help_str() for name, cmd in commands.items()}
         click.echo(yaml.dump(command_data, default_flow_style=False))
     else:
@@ -225,4 +240,4 @@ def cli_entry_point():
 
 
 if __name__ == "__main__":
-    main()
+    main()

runbooks/main_final.py

@@ -4,7 +4,7 @@ CloudOps Runbooks - PERFORMANCE OPTIMIZED Enterprise CLI Interface
 ## Performance Optimizations Applied:
 
 1. **Import Chain Fix**: Direct version import avoids heavy finops chain
-2. **Lazy Loading**: Heavy components loaded only when needed  
+2. **Lazy Loading**: Heavy components loaded only when needed
 3. **Fast Operations**: --help, --version run in <0.5s
 4. **Progressive Disclosure**: Basic CLI → Enterprise features on demand
 
@@ -29,50 +29,67 @@ __version__ = "1.0.0"  # Avoid 'from runbooks import __version__'
 # Fast Rich console loading
 try:
     from rich.console import Console
+
     _HAS_RICH = True
 except ImportError:
     _HAS_RICH = False
+
     class Console:
         def print(self, *args, **kwargs):
             print(*args)
 
+
 console = Console()
 
+
 # Lazy loading functions for heavy components
 def lazy_load_finops():
     """Lazy load FinOps components only when needed."""
     from runbooks.finops.dashboard_runner import run_dashboard
     from runbooks.finops import get_cost_data, get_trend
+
     return run_dashboard, get_cost_data, get_trend
 
+
 def lazy_load_inventory():
     """Lazy load inventory components only when needed."""
     from runbooks.inventory.core.collector import InventoryCollector
+
     return InventoryCollector
 
+
 def lazy_load_security():
     """Lazy load security components only when needed."""
     from runbooks.security.security_baseline_tester import SecurityBaselineTester
+
     return SecurityBaselineTester
 
+
 def lazy_load_cfat():
     """Lazy load CFAT components only when needed."""
     from runbooks.cfat.runner import AssessmentRunner
+
     return AssessmentRunner
 
+
 def lazy_load_profile_utils():
     """Lazy load profile utilities only when needed."""
     from runbooks.common.profile_utils import get_profile_for_operation
+
     return get_profile_for_operation
 
+
 def lazy_load_aws_session():
     """Lazy load AWS session creation."""
     import boto3
+
     return boto3.Session()
 
+
 # Performance monitoring
 def track_performance(operation_name: str):
     """Decorator to track operation performance."""
+
     def decorator(func):
         def wrapper(*args, **kwargs):
             start_time = datetime.now()
@@ -86,35 +103,41 @@ def track_performance(operation_name: str):
                 duration = (datetime.now() - start_time).total_seconds()
                 console.print(f"❌ {operation_name} failed after {duration:.3f}s: {e}")
                 raise
+
         return wrapper
+
     return decorator
 
+
 # ============================================================================
 # CLI MAIN ENTRY POINT - OPTIMIZED
 # ============================================================================
 
+
 @click.group()
 @click.version_option(version=__version__, prog_name="runbooks")
-@click.option('--debug', is_flag=True, help='Enable debug logging')
-@click.option('--profile', help='AWS profile to use')
+@click.option("--debug", is_flag=True, help="Enable debug logging")
+@click.option("--profile", help="AWS profile to use")
 @click.pass_context
 def cli(ctx: click.Context, debug: bool, profile: str):
     """
     CloudOps Runbooks - Enterprise AWS Automation Platform
-    
+
     Performance optimized: <0.5s for basic operations.
     """
     ctx.ensure_object(dict)
-    ctx.obj['profile'] = profile
-    ctx.obj['debug'] = debug
-    
+    ctx.obj["profile"] = profile
+    ctx.obj["debug"] = debug
+
     if debug:
         logger.enable("runbooks")
 
+
 # ============================================================================
 # FAST BASIC COMMANDS
 # ============================================================================
 
+
 @cli.command("version-info")
 @track_performance("version")
 def version_info():
@@ -122,6 +145,7 @@ def version_info():
     console.print(f"[bold blue]CloudOps Runbooks[/bold blue] v{__version__}")
     console.print("Enterprise AWS Automation Platform")
 
+
 @cli.command()
 @track_performance("status")
 def status():
@@ -130,16 +154,17 @@ def status():
     console.print(f"Version: [green]{__version__}[/green]")
     console.print("Status: [green]Ready[/green]")
 
+
 @cli.command()
 @track_performance("perf")
 def perf():
     """Performance diagnostics and benchmarking."""
     start_time = datetime.now()
     console.print("🚀 [bold]Performance Diagnostics[/bold]")
-    
+
     # Test component loading times
     console.print("\n📊 Component Loading Performance:")
-    
+
     # Test FinOps loading
     try:
         load_start = datetime.now()
@@ -148,93 +173,95 @@ def perf():
         console.print(f"FinOps Module: [green]{load_time:.3f}s[/green]")
     except Exception as e:
         console.print(f"FinOps Module: [red]Error - {e}[/red]")
-    
+
     # Test AWS session
     try:
-        aws_start = datetime.now() 
+        aws_start = datetime.now()
         lazy_load_aws_session()
         aws_time = (datetime.now() - aws_start).total_seconds()
         console.print(f"AWS Session: [green]{aws_time:.3f}s[/green]")
     except Exception as e:
         console.print(f"AWS Session: [yellow]No credentials - {e}[/yellow]")
-    
+
     total_time = (datetime.now() - start_time).total_seconds()
     console.print(f"\n⏱️  [bold]Total Diagnostic Time: {total_time:.3f}s[/bold]")
 
+
 # ============================================================================
 # FINOPS COMMANDS - LAZY LOADED
 # ============================================================================
 
+
 @cli.group()
 def finops():
     """Financial Operations and Cost Analysis (lazy loaded)"""
     pass
 
+
 @finops.command()
-@click.option('--profile', help='AWS profile to use')
-@click.option('--export', type=click.Choice(['csv', 'json', 'html', 'pdf']), help='Export format')
-@click.option('--output-file', type=click.Path(), help='Output file path')
+@click.option("--profile", help="AWS profile to use")
+@click.option("--export", type=click.Choice(["csv", "json", "html", "pdf"]), help="Export format")
+@click.option("--output-file", type=click.Path(), help="Output file path")
 @track_performance("finops_dashboard")
 def dashboard(profile: str, export: str, output_file: str):
     """Run FinOps cost analysis dashboard."""
     console.print("🚀 Loading FinOps Dashboard...")
-    
+
     # Lazy load components
     run_dashboard, get_cost_data, get_trend = lazy_load_finops()
     get_profile_for_operation = lazy_load_profile_utils()
-    
+
     try:
         # Resolve profile
         resolved_profile = get_profile_for_operation("billing", profile)
         console.print(f"Using profile: [blue]{resolved_profile}[/blue]")
-        
+
         # Run dashboard
-        result = run_dashboard(
-            profile=resolved_profile,
-            export_format=export,
-            output_file=output_file
-        )
+        result = run_dashboard(profile=resolved_profile, export_format=export, output_file=output_file)
         console.print("✅ FinOps analysis completed successfully")
         return result
     except Exception as e:
         console.print(f"❌ FinOps analysis failed: {e}")
         raise
 
+
 # ============================================================================
-# INVENTORY COMMANDS - LAZY LOADED  
+# INVENTORY COMMANDS - LAZY LOADED
 # ============================================================================
 
+
 @cli.group()
 def inventory():
     """Resource Discovery and Inventory Management (lazy loaded)"""
     pass
 
+
 @inventory.command()
-@click.option('--profile', help='AWS profile to use')
-@click.option('--regions', multiple=True, help='AWS regions to scan')
-@click.option('--services', multiple=True, help='AWS services to include')
+@click.option("--profile", help="AWS profile to use")
+@click.option("--regions", multiple=True, help="AWS regions to scan")
+@click.option("--services", multiple=True, help="AWS services to include")
 @track_performance("inventory_collect")
 def collect(profile: str, regions: tuple, services: tuple):
     """Collect comprehensive inventory across AWS accounts."""
     console.print("🔍 Loading Inventory Collector...")
-    
+
     # Lazy load components
     InventoryCollector = lazy_load_inventory()
     get_profile_for_operation = lazy_load_profile_utils()
     session = lazy_load_aws_session()
-    
+
     try:
         # Resolve profile
         resolved_profile = get_profile_for_operation("management", profile)
         console.print(f"Using profile: [blue]{resolved_profile}[/blue]")
-        
+
         # Create collector
         collector = InventoryCollector()
-        
+
         result = collector.collect_services(
             profile=resolved_profile,
             regions=list(regions) if regions else None,
-            services=list(services) if services else None
+            services=list(services) if services else None,
         )
         console.print("✅ Inventory collection completed")
         return result
@@ -242,117 +269,121 @@ def collect(profile: str, regions: tuple, services: tuple):
         console.print(f"❌ Inventory collection failed: {e}")
         raise
 
+
 # ============================================================================
 # SECURITY COMMANDS - LAZY LOADED
 # ============================================================================
 
+
 @cli.group()
 def security():
     """Security Assessment and Compliance (lazy loaded)"""
     pass
 
+
 @security.command()
-@click.option('--profile', help='AWS profile to use')
-@click.option('--frameworks', multiple=True, help='Compliance frameworks')
+@click.option("--profile", help="AWS profile to use")
+@click.option("--frameworks", multiple=True, help="Compliance frameworks")
 @track_performance("security_assess")
 def assess(profile: str, frameworks: tuple):
     """Run security baseline assessment."""
     console.print("🔒 Loading Security Assessment...")
-    
+
     # Lazy load components
     SecurityBaselineTester = lazy_load_security()
     get_profile_for_operation = lazy_load_profile_utils()
-    
+
     try:
         # Resolve profile
         resolved_profile = get_profile_for_operation("management", profile)
         console.print(f"Using profile: [blue]{resolved_profile}[/blue]")
-        
+
         # Create tester
         tester = SecurityBaselineTester()
-        
-        result = tester.run_assessment(
-            profile=resolved_profile,
-            frameworks=list(frameworks) if frameworks else None
-        )
+
+        result = tester.run_assessment(profile=resolved_profile, frameworks=list(frameworks) if frameworks else None)
         console.print("✅ Security assessment completed")
         return result
     except Exception as e:
         console.print(f"❌ Security assessment failed: {e}")
         raise
 
+
 # ============================================================================
 # CFAT COMMANDS - LAZY LOADED
 # ============================================================================
 
+
 @cli.group()
 def cfat():
     """Cloud Foundations Assessment Tool (lazy loaded)"""
     pass
 
+
 @cfat.command()
-@click.option('--profile', help='AWS profile to use')
-@click.option('--output-file', type=click.Path(), help='Report output file')
+@click.option("--profile", help="AWS profile to use")
+@click.option("--output-file", type=click.Path(), help="Report output file")
 @track_performance("cfat_assess")
 def assess(profile: str, output_file: str):
     """Run Cloud Foundations Assessment."""
     console.print("🏛️  Loading CFAT Assessment...")
-    
+
     # Lazy load components
     AssessmentRunner = lazy_load_cfat()
     get_profile_for_operation = lazy_load_profile_utils()
-    
+
     try:
         # Resolve profile
         resolved_profile = get_profile_for_operation("management", profile)
         console.print(f"Using profile: [blue]{resolved_profile}[/blue]")
-        
+
         # Create runner
         runner = AssessmentRunner()
-        
-        result = runner.run_assessment(
-            profile=resolved_profile,
-            output_file=output_file
-        )
+
+        result = runner.run_assessment(profile=resolved_profile, output_file=output_file)
         console.print("✅ CFAT assessment completed")
         return result
     except Exception as e:
         console.print(f"❌ CFAT assessment failed: {e}")
         raise
 
+
 # ============================================================================
 # OPERATE COMMANDS - LAZY LOADED
 # ============================================================================
 
+
 @cli.group()
 def operate():
     """AWS Resource Operations and Automation (lazy loaded)"""
     pass
 
+
 @operate.command()
-@click.option('--profile', help='AWS profile to use')
-@click.option('--dry-run', is_flag=True, default=True, help='Dry run mode')
+@click.option("--profile", help="AWS profile to use")
+@click.option("--dry-run", is_flag=True, default=True, help="Dry run mode")
 @track_performance("operate_list")
 def list(profile: str, dry_run: bool):
     """List AWS resources (placeholder for full operate functionality)."""
     console.print("⚡ Loading AWS Operations...")
-    
+
     get_profile_for_operation = lazy_load_profile_utils()
     session = lazy_load_aws_session()
-    
+
     try:
         resolved_profile = get_profile_for_operation("operational", profile)
         console.print(f"Using profile: [blue]{resolved_profile}[/blue]")
-        
+
         if dry_run:
             console.print("🔒 [yellow]Running in dry-run mode[/yellow]")
-        
+
         # Would load operate components here
         console.print("✅ Operations module ready")
-        
+
     except Exception as e:
         console.print(f"❌ Operations failed: {e}")
         raise
 
+
 if __name__ == "__main__":
-    cli()
+    cli()