runbooks 0.7.9__py3-none-any.whl → 0.9.0__py3-none-any.whl

runbooks/inventory/rich_inventory_display.py

@@ -20,23 +20,23 @@ Version: 0.7.8
 from datetime import datetime
 from typing import Any, Dict, List, Optional
 
-from rich.console import Console
-from rich.progress import Progress, SpinnerColumn, TextColumn, BarColumn, TaskProgressColumn
-from rich.table import Table
 from rich import box
+from rich.console import Console
 from rich.panel import Panel
+from rich.progress import BarColumn, Progress, SpinnerColumn, TaskProgressColumn, TextColumn
+from rich.table import Table
 from rich.text import Text
 from rich.tree import Tree
 
 from runbooks.common.rich_utils import (
+    STATUS_INDICATORS,
     console,
-    create_table,
+    create_panel,
     create_progress_bar,
+    create_table,
+    print_info,
     print_status,
     print_success,
-    print_info,
-    create_panel,
-    STATUS_INDICATORS
 )
 
 
@@ -55,14 +55,10 @@ def display_inventory_header(operation: str, profile: str, accounts: int, region
 
 [yellow]Profile:[/yellow] {profile}
 [yellow]Scope:[/yellow] {accounts} accounts × {regions} regions = {accounts * regions} total operations
-[yellow]Started:[/yellow] {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
+[yellow]Started:[/yellow] {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
 """
-    
-    console.print(create_panel(
-        header_text.strip(),
-        title="📊 Inventory Operation",
-        border_style="cyan"
-    ))
+
+    console.print(create_panel(header_text.strip(), title="📊 Inventory Operation", border_style="cyan"))
 
 
 def create_inventory_progress(total_operations: int, operation_name: str = "Scanning Resources") -> Progress:
@@ -84,14 +80,13 @@ def create_inventory_progress(total_operations: int, operation_name: str = "Scan
         TextColumn("•"),
         TextColumn("[blue]{task.completed}/{task.total} operations"),
         console=console,
-        transient=False
+        transient=False,
     )
 
 
-def display_ec2_inventory_results(instances: List[Dict[str, Any]], 
-                                accounts: int, 
-                                regions: int, 
-                                timing_info: Optional[Dict] = None) -> None:
+def display_ec2_inventory_results(
+    instances: List[Dict[str, Any]], accounts: int, regions: int, timing_info: Optional[Dict] = None
+) -> None:
     """
     Display EC2 inventory results in a professional Rich table format.
 
@@ -103,27 +98,27 @@ def display_ec2_inventory_results(instances: List[Dict[str, Any]],
     """
     # Summary panel first
     total_instances = len(instances)
-    
+
     # Count by state
     state_counts = {}
     for instance in instances:
         state = instance.get("State", {}).get("Name", "unknown")
         state_counts[state] = state_counts.get(state, 0) + 1
-    
+
     # Create status breakdown
     status_text = ""
     for state, count in sorted(state_counts.items()):
         status_indicator = {
             "running": "🟢",
-            "stopped": "🔴", 
+            "stopped": "🔴",
             "stopping": "🟡",
             "starting": "🟡",
             "terminated": "⚫",
-            "terminating": "🟡"
+            "terminating": "🟡",
         }.get(state, "⚪")
-        
+
         status_text += f"{status_indicator} {state.title()}: {count}\n"
-    
+
     summary_content = f"""
 [bold cyan]EC2 Inventory Summary[/bold cyan]
 
@@ -134,13 +129,9 @@ def display_ec2_inventory_results(instances: List[Dict[str, Any]],
 [bold yellow]Instance States:[/bold yellow]
 {status_text.strip()}
 """
-    
-    console.print(create_panel(
-        summary_content.strip(),
-        title="📊 Discovery Results",
-        border_style="green"
-    ))
-    
+
+    console.print(create_panel(summary_content.strip(), title="📊 Discovery Results", border_style="green"))
+
     # Detailed results table if instances found
     if instances:
         # Group instances by account for better organization
@@ -150,7 +141,7 @@ def display_ec2_inventory_results(instances: List[Dict[str, Any]],
             if account_id not in instances_by_account:
                 instances_by_account[account_id] = []
             instances_by_account[account_id].append(instance)
-        
+
         # Create detailed table
         table = create_table(
             title="🖥️ EC2 Instance Details",
@@ -160,26 +151,26 @@ def display_ec2_inventory_results(instances: List[Dict[str, Any]],
                 {"name": "Instance ID", "style": "magenta"},
                 {"name": "Type", "style": "blue"},
                 {"name": "State", "style": "green"},
-                {"name": "Name", "style": "white"}
+                {"name": "Name", "style": "white"},
             ],
-            box_style=box.ROUNDED
+            box_style=box.ROUNDED,
         )
-        
+
         # Add rows (show first 50 instances to avoid overwhelming output)
         displayed_count = 0
         for account_id in sorted(instances_by_account.keys()):
             account_instances = instances_by_account[account_id][:10]  # Max 10 per account
-            
+
             for instance in account_instances:
                 if displayed_count >= 50:  # Overall limit
                     break
-                    
+
                 # Extract instance information
                 instance_id = instance.get("InstanceId", "N/A")
                 instance_type = instance.get("InstanceType", "N/A")
                 state = instance.get("State", {}).get("Name", "unknown")
                 region = instance.get("Region", "N/A")
-                
+
                 # Get instance name from tags
                 instance_name = "N/A"
                 tags = instance.get("Tags", [])
@@ -187,7 +178,7 @@ def display_ec2_inventory_results(instances: List[Dict[str, Any]],
                     if tag.get("Key") == "Name":
                         instance_name = tag.get("Value", "N/A")
                         break
-                
+
                 # Style state with appropriate color
                 state_styled = {
                     "running": "[green]🟢 Running[/green]",
@@ -195,38 +186,37 @@ def display_ec2_inventory_results(instances: List[Dict[str, Any]],
                     "stopping": "[yellow]🟡 Stopping[/yellow]",
                     "starting": "[yellow]🟡 Starting[/yellow]",
                     "terminated": "[dim]⚫ Terminated[/dim]",
-                    "terminating": "[yellow]🟡 Terminating[/yellow]"
+                    "terminating": "[yellow]🟡 Terminating[/yellow]",
                 }.get(state, f"[white]⚪ {state.title()}[/white]")
-                
+
                 table.add_row(
                     account_id[:12],  # Truncate account ID
                     region,
                     instance_id,
                     instance_type,
                     state_styled,
-                    instance_name[:20] if instance_name != "N/A" else "N/A"  # Truncate long names
+                    instance_name[:20] if instance_name != "N/A" else "N/A",  # Truncate long names
                 )
-                
+
                 displayed_count += 1
-        
+
         console.print(table)
-        
+
         # Show truncation message if needed
         if total_instances > 50:
             console.print(f"\n[dim]Showing first 50 instances. Total found: {total_instances}[/dim]")
-    
+
     # Timing information
     if timing_info:
         execution_time = timing_info.get("total_time", 0)
         print_success(f"✅ Inventory scan completed in {execution_time:.2f} seconds")
-    
+
     print_info("💡 Use --output json or --output csv to export complete results")
 
 
-def display_generic_inventory_results(resource_type: str,
-                                    resources: List[Dict[str, Any]], 
-                                    accounts: int, 
-                                    regions: int) -> None:
+def display_generic_inventory_results(
+    resource_type: str, resources: List[Dict[str, Any]], accounts: int, regions: int
+) -> None:
     """
     Display generic inventory results for any resource type.
 
@@ -237,7 +227,7 @@ def display_generic_inventory_results(resource_type: str,
         regions: Number of regions scanned
     """
     total_resources = len(resources)
-    
+
     # Resource type icons
     resource_icons = {
         "rds": "🗄️",
@@ -247,11 +237,11 @@ def display_generic_inventory_results(resource_type: str,
         "iam": "👤",
         "cloudformation": "📚",
         "ssm": "🔑",
-        "route53": "🌍"
+        "route53": "🌍",
     }
-    
+
     icon = resource_icons.get(resource_type.lower(), "📦")
-    
+
     summary_content = f"""
 [bold cyan]{resource_type.upper()} Inventory Summary[/bold cyan]
 
@@ -259,13 +249,9 @@ def display_generic_inventory_results(resource_type: str,
 [green]Accounts Scanned:[/green] {accounts}
 [green]Regions Scanned:[/green] {regions}
 """
-    
-    console.print(create_panel(
-        summary_content.strip(),
-        title=f"{icon} Discovery Results",
-        border_style="green"
-    ))
-    
+
+    console.print(create_panel(summary_content.strip(), title=f"{icon} Discovery Results", border_style="green"))
+
     if total_resources > 0:
         print_success(f"✅ Found {total_resources} {resource_type} resources across {accounts} accounts")
     else:
@@ -281,24 +267,18 @@ def display_inventory_error(error_message: str, suggestions: Optional[List[str]]
         suggestions: Optional list of suggestions for resolution
     """
     error_content = f"[bold red]❌ Inventory Operation Failed[/bold red]\n\n{error_message}"
-    
+
     if suggestions:
         error_content += "\n\n[yellow]💡 Suggestions:[/yellow]\n"
         for suggestion in suggestions:
             error_content += f"  • {suggestion}\n"
-    
-    console.print(create_panel(
-        error_content.strip(),
-        title="🚨 Error",
-        border_style="red",
-        padding=1
-    ))
-
-
-def display_multi_resource_summary(resource_counts: Dict[str, int], 
-                                 accounts: int, 
-                                 regions: int,
-                                 execution_time: float) -> None:
+
+    console.print(create_panel(error_content.strip(), title="🚨 Error", border_style="red", padding=1))
+
+
+def display_multi_resource_summary(
+    resource_counts: Dict[str, int], accounts: int, regions: int, execution_time: float
+) -> None:
     """
     Display summary for multi-resource inventory operations.
 
@@ -314,22 +294,18 @@ def display_multi_resource_summary(resource_counts: Dict[str, int],
         columns=[
             {"name": "Resource Type", "style": "cyan"},
             {"name": "Count", "style": "green", "justify": "right"},
-            {"name": "Status", "style": "yellow"}
-        ]
+            {"name": "Status", "style": "yellow"},
+        ],
     )
-    
+
     total_resources = 0
     for resource_type, count in sorted(resource_counts.items()):
         status = "✅ Found" if count > 0 else "⚪ None"
-        table.add_row(
-            resource_type.title(),
-            str(count),
-            status
-        )
+        table.add_row(resource_type.title(), str(count), status)
         total_resources += count
-    
+
     console.print(table)
-    
+
     # Overall summary
     summary_text = f"""
 [bold green]Overall Summary[/bold green]
@@ -340,12 +316,8 @@ def display_multi_resource_summary(resource_counts: Dict[str, int],
 [cyan]Execution Time:[/cyan] {execution_time:.2f} seconds
 [cyan]Performance:[/cyan] {(accounts * regions) / execution_time:.1f} operations/second
 """
-    
-    console.print(create_panel(
-        summary_text.strip(),
-        title="🎯 Inventory Complete",
-        border_style="bright_blue"
-    ))
+
+    console.print(create_panel(summary_text.strip(), title="🎯 Inventory Complete", border_style="bright_blue"))
 
 
 def display_account_tree(accounts_data: Dict[str, Dict]) -> None:
@@ -356,38 +328,33 @@ def display_account_tree(accounts_data: Dict[str, Dict]) -> None:
         accounts_data: Nested dictionary of account -> resource data
     """
     tree = Tree("🏢 [bold cyan]AWS Organization Structure[/bold cyan]")
-    
+
     for account_id, account_data in accounts_data.items():
         account_name = account_data.get("account_name", "Unknown")
         account_branch = tree.add(f"📊 [yellow]Account: {account_id}[/yellow] ({account_name})")
-        
+
         # Add regions
         regions = account_data.get("regions", {})
         for region, region_data in regions.items():
             region_branch = account_branch.add(f"🌍 [green]Region: {region}[/green]")
-            
+
             # Add resource counts
             for resource_type, count in region_data.get("resource_counts", {}).items():
                 if count > 0:
-                    icon = {
-                        "ec2": "🖥️",
-                        "rds": "🗄️", 
-                        "s3": "🪣",
-                        "lambda": "⚡"
-                    }.get(resource_type, "📦")
-                    
+                    icon = {"ec2": "🖥️", "rds": "🗄️", "s3": "🪣", "lambda": "⚡"}.get(resource_type, "📦")
+
                     region_branch.add(f"{icon} {resource_type.upper()}: [bold]{count}[/bold] resources")
-    
+
     console.print(tree)
 
 
 # Export public functions
 __all__ = [
     "display_inventory_header",
-    "create_inventory_progress", 
+    "create_inventory_progress",
     "display_ec2_inventory_results",
     "display_generic_inventory_results",
     "display_inventory_error",
     "display_multi_resource_summary",
-    "display_account_tree"
-]
+    "display_account_tree",
+]

runbooks/inventory/run_on_multi_accounts.py

@@ -192,7 +192,9 @@ for account_num in Accounts:
     )
     response = check_account_access(aws_acct, account_num, pAccessRole)
     if response["Success"]:
-        console.print(f"[green]✅ Account {account_num} was successfully connected via role {pAccessRole} from {aws_acct.acct_number}[/green]")
+        console.print(
+            f"[green]✅ Account {account_num} was successfully connected via role {pAccessRole} from {aws_acct.acct_number}[/green]"
+        )
         """
 		Put more commands here... Or you can write functions that represent your commands and call them from here.
 		"""
@@ -201,11 +203,13 @@ for account_num in Accounts:
         username = "Paul"
         user_response = participant_user(tgt_aws_access, username=username)
     else:
-        console.print(Panel(
-            f"Access Role {pAccessRole} failed to connect to {account_num} from {aws_acct.acct_number}",
-            title=f"[red]❌ Connection Error: {response['ErrorMessage']}[/red]",
-            border_style="red"
-        ))
+        console.print(
+            Panel(
+                f"Access Role {pAccessRole} failed to connect to {account_num} from {aws_acct.acct_number}",
+                title=f"[red]❌ Connection Error: {response['ErrorMessage']}[/red]",
+                border_style="red",
+            )
+        )
 
     # Display access keys with Rich formatting
     credentials_info = [
@@ -213,14 +217,10 @@ for account_num in Accounts:
         f"[cyan]User:[/cyan] {user_response['User']} (created/confirmed in account {user_response['AccountId']})",
         f"[cyan]Password:[/cyan] {user_response['Password']}",
         f"[cyan]Access Key ID:[/cyan] {user_response['AccessKeyId']}",
-        f"[cyan]Secret Access Key:[/cyan] {user_response['SecretAccessKey']}"
+        f"[cyan]Secret Access Key:[/cyan] {user_response['SecretAccessKey']}",
     ]
-    
-    console.print(Panel(
-        "\n".join(credentials_info),
-        title="[blue]🔑 Account Credentials[/blue]",
-        border_style="blue"
-    ))
+
+    console.print(Panel("\n".join(credentials_info), title="[blue]🔑 Account Credentials[/blue]", border_style="blue"))
 
 console.print("")
 console.print("[green]✅ Thanks for using this script...[/green]")