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

runbooks/remediation/rds_snapshot_list.py

@@ -16,8 +16,14 @@ from botocore.exceptions import ClientError
 
 from .commons import display_aws_account_info, get_client, write_to_csv
 from ..common.rich_utils import (
-    console, print_header, print_success, print_error, print_warning,
-    create_table, create_progress_bar, format_cost
+    console,
+    print_header,
+    print_success,
+    print_error,
+    print_warning,
+    create_table,
+    create_progress_bar,
+    format_cost,
 )
 
 logger = logging.getLogger(__name__)
@@ -36,16 +42,17 @@ def calculate_snapshot_age(create_time):
 def estimate_snapshot_cost(allocated_storage, storage_type="gp2", days_old=1):
     """
     Estimate monthly snapshot storage cost with enhanced accuracy.
-    
+
     JIRA FinOps-23: Enhanced cost estimation for measurable range annual savings target
     Based on AWS RDS snapshot pricing: https://aws.amazon.com/rds/pricing/
     """
     # Real-time RDS Snapshot cost from AWS Pricing API - NO hardcoded defaults
     from runbooks.common.aws_pricing_api import pricing_api
+
     # Get region from caller context or default to us-east-1
-    region = os.getenv('AWS_DEFAULT_REGION', 'us-east-1')
+    region = os.getenv("AWS_DEFAULT_REGION", "us-east-1")
     snapshot_cost_per_gb_month = pricing_api.get_rds_snapshot_cost_per_gb(region)
-    
+
     # Calculate base monthly cost
     monthly_cost = allocated_storage * snapshot_cost_per_gb_month
 
@@ -59,28 +66,26 @@ def estimate_snapshot_cost(allocated_storage, storage_type="gp2", days_old=1):
 def calculate_manual_snapshot_savings(snapshots_data: List[Dict]) -> Dict[str, float]:
     """
     Calculate potential savings from manual snapshot cleanup.
-    
+
     JIRA FinOps-23: Focuses on 89 manual snapshots for cost optimization
     """
     manual_snapshots = [s for s in snapshots_data if s.get("SnapshotType", "").lower() == "manual"]
-    
+
     # Calculate costs by age groups
     old_manual_snapshots = [s for s in manual_snapshots if s.get("AgeDays", 0) >= 90]  # 3+ months old
     very_old_manual_snapshots = [s for s in manual_snapshots if s.get("AgeDays", 0) >= 180]  # 6+ months old
-    
+
     total_manual_cost = sum(s.get("EstimatedMonthlyCost", 0) for s in manual_snapshots)
     old_manual_cost = sum(s.get("EstimatedMonthlyCost", 0) for s in old_manual_snapshots)
     very_old_manual_cost = sum(s.get("EstimatedMonthlyCost", 0) for s in very_old_manual_snapshots)
-    
+
     return {
         "total_manual_snapshots": len(manual_snapshots),
         "total_manual_monthly_cost": total_manual_cost,
         "total_manual_annual_cost": total_manual_cost * 12,
-        
         "old_manual_snapshots": len(old_manual_snapshots),  # 90+ days
         "old_manual_monthly_savings": old_manual_cost,
         "old_manual_annual_savings": old_manual_cost * 12,
-        
         "very_old_manual_snapshots": len(very_old_manual_snapshots),  # 180+ days
         "very_old_manual_monthly_savings": very_old_manual_cost,
         "very_old_manual_annual_savings": very_old_manual_cost * 12,
@@ -96,15 +101,17 @@ def calculate_manual_snapshot_savings(snapshots_data: List[Dict]) -> Dict[str, f
 @click.option("--older-than", default=90, help="Focus on snapshots older than X days")
 @click.option("--calculate-savings", is_flag=True, help="Calculate detailed cost savings analysis")
 @click.option("--analyze", is_flag=True, help="Perform comprehensive cost analysis")
-def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type, manual_only, older_than, calculate_savings, analyze):
+def get_rds_snapshot_details(
+    output_file, old_days, include_cost, snapshot_type, manual_only, older_than, calculate_savings, analyze
+):
     """
     Analyze RDS snapshots for lifecycle management and cost optimization.
-    
+
     JIRA FinOps-23: Enhanced RDS snapshots analysis for measurable range annual savings
     Focus on 89 manual snapshots causing storage costs and operational clutter
     """
     print_header("RDS Snapshot Cost Optimization Analysis", "latest version")
-    
+
     account_info = display_aws_account_info()
     console.print(f"[cyan]Analyzing RDS snapshots in {account_info}[/cyan]")
 
@@ -126,19 +133,25 @@ def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type,
         if manual_only:
             original_count = len(snapshots)
             snapshots = [s for s in snapshots if s.get("SnapshotType", "").lower() == "manual"]
-            console.print(f"[dim]JIRA FinOps-23 Filter: {len(snapshots)} manual snapshots (from {original_count} total)[/dim]")
-            
+            console.print(
+                f"[dim]JIRA FinOps-23 Filter: {len(snapshots)} manual snapshots (from {original_count} total)[/dim]"
+            )
+
         if snapshot_type:
             original_count = len(snapshots)
             snapshots = [s for s in snapshots if s.get("SnapshotType", "").lower() == snapshot_type.lower()]
             console.print(f"[dim]Filtered to {len(snapshots)} snapshots of type '{snapshot_type}'[/dim]")
-            
+
         if older_than > 0:
             now = datetime.now(tz=timezone.utc)
             threshold_date = now - timedelta(days=older_than)
             original_count = len(snapshots)
-            snapshots = [s for s in snapshots if s.get("SnapshotCreateTime") and s["SnapshotCreateTime"] < threshold_date]
-            console.print(f"[dim]Age filter: {len(snapshots)} snapshots older than {older_than} days (from {original_count})[/dim]")
+            snapshots = [
+                s for s in snapshots if s.get("SnapshotCreateTime") and s["SnapshotCreateTime"] < threshold_date
+            ]
+            console.print(
+                f"[dim]Age filter: {len(snapshots)} snapshots older than {older_than} days (from {original_count})[/dim]"
+            )
 
         data = []
         old_snapshots = []
@@ -148,11 +161,8 @@ def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type,
         total_estimated_cost = 0
 
         with create_progress_bar() as progress:
-            task_id = progress.add_task(
-                f"Analyzing {len(snapshots)} snapshots...",
-                total=len(snapshots)
-            )
-            
+            task_id = progress.add_task(f"Analyzing {len(snapshots)} snapshots...", total=len(snapshots))
+
             for i, snapshot in enumerate(snapshots, 1):
                 snapshot_id = snapshot["DBSnapshotIdentifier"]
 
@@ -183,7 +193,7 @@ def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type,
                     if allocated_storage > 0:
                         estimated_cost = estimate_snapshot_cost(allocated_storage, storage_type, age_days)
                         total_estimated_cost += estimated_cost
-                
+
                 snapshot_data["EstimatedMonthlyCost"] = estimated_cost
                 snapshot_data["EstimatedAnnualCost"] = estimated_cost * 12
 
@@ -234,12 +244,12 @@ def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type,
                 "projected_annual_savings": 0.0,
                 "old_manual_snapshots": 0,
                 "old_manual_monthly_savings": 0.0,
-                "old_manual_annual_savings": 0.0
+                "old_manual_annual_savings": 0.0,
             }
-        
+
         # Create comprehensive summary table with Rich CLI
         print_header("RDS Snapshot Analysis Summary")
-        
+
         summary_table = create_table(
             title="RDS Snapshot Cost Analysis - JIRA FinOps-23",
             columns=[
@@ -247,8 +257,8 @@ def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type,
                 {"header": "Count", "style": "green bold"},
                 {"header": "Storage (GB)", "style": "yellow"},
                 {"header": "Monthly Cost", "style": "red"},
-                {"header": "Annual Cost", "style": "red bold"}
-            ]
+                {"header": "Annual Cost", "style": "red bold"},
+            ],
         )
 
         # Basic metrics
@@ -257,31 +267,31 @@ def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type,
             str(len(data)),
             str(total_storage),
             format_cost(total_estimated_cost) if (include_cost or calculate_savings or analyze) else "N/A",
-            format_cost(total_estimated_cost * 12) if (include_cost or calculate_savings or analyze) else "N/A"
+            format_cost(total_estimated_cost * 12) if (include_cost or calculate_savings or analyze) else "N/A",
         )
-        
+
         summary_table.add_row(
             "Manual Snapshots",
             str(len(manual_snapshots)),
             str(sum(s["AllocatedStorage"] for s in data if s["SnapshotType"].lower() == "manual")),
             format_cost(savings_analysis["total_manual_monthly_cost"]) if (calculate_savings or analyze) else "N/A",
-            format_cost(savings_analysis["total_manual_annual_cost"]) if (calculate_savings or analyze) else "N/A"
+            format_cost(savings_analysis["total_manual_annual_cost"]) if (calculate_savings or analyze) else "N/A",
         )
-        
+
         summary_table.add_row(
-            "Automated Snapshots", 
+            "Automated Snapshots",
             str(len(automated_snapshots)),
             str(sum(s["AllocatedStorage"] for s in data if s["SnapshotType"].lower() == "automated")),
             "Retention Policy",
-            "Retention Policy"
+            "Retention Policy",
         )
-        
+
         summary_table.add_row(
             f"Old Snapshots (>{old_days} days)",
             str(len(old_snapshots)),
             str(sum(s["AllocatedStorage"] for s in data if s["IsOld"])),
             "Mixed Types",
-            "Mixed Types"
+            "Mixed Types",
         )
 
         # JIRA FinOps-23 specific analysis
@@ -289,20 +299,30 @@ def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type,
             summary_table.add_row(
                 f"🎯 Manual >{older_than}d (Cleanup Target)",
                 str(savings_analysis["old_manual_snapshots"]),
-                str(sum(s["AllocatedStorage"] for s in data if s["AgeDays"] >= older_than and s["SnapshotType"].lower() == "manual")),
+                str(
+                    sum(
+                        s["AllocatedStorage"]
+                        for s in data
+                        if s["AgeDays"] >= older_than and s["SnapshotType"].lower() == "manual"
+                    )
+                ),
                 format_cost(savings_analysis["old_manual_monthly_savings"]),
-                format_cost(savings_analysis["old_manual_annual_savings"])
+                format_cost(savings_analysis["old_manual_annual_savings"]),
             )
 
         console.print(summary_table)
 
         # Cleanup recommendations with Rich CLI
         cleanup_candidates = [s for s in data if s["IsOld"] and s["SnapshotType"].lower() == "manual"]
-        high_priority_candidates = [s for s in data if s["AgeDays"] >= older_than and s["SnapshotType"].lower() == "manual"]
-        
+        high_priority_candidates = [
+            s for s in data if s["AgeDays"] >= older_than and s["SnapshotType"].lower() == "manual"
+        ]
+
         if high_priority_candidates:
-            print_warning(f"🎯 JIRA FinOps-23: {len(high_priority_candidates)} high-priority manual snapshots (>{older_than} days):")
-            
+            print_warning(
+                f"🎯 JIRA FinOps-23: {len(high_priority_candidates)} high-priority manual snapshots (>{older_than} days):"
+            )
+
             # Create detailed cleanup candidates table
             cleanup_table = create_table(
                 title=f"High-Priority Manual Snapshots (>{older_than} days old)",
@@ -312,25 +332,25 @@ def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type,
                     {"header": "Age (Days)", "style": "yellow"},
                     {"header": "Size (GB)", "style": "green"},
                     {"header": "Monthly Cost", "style": "red"},
-                    {"header": "Engine", "style": "magenta"}
-                ]
+                    {"header": "Engine", "style": "magenta"},
+                ],
             )
-            
+
             for snap in high_priority_candidates[:15]:  # Show first 15 for readability
                 cleanup_table.add_row(
-                    snap['DBSnapshotIdentifier'],
-                    snap['DBInstanceIdentifier'],
-                    str(snap['AgeDays']),
-                    str(snap['AllocatedStorage']),
-                    format_cost(snap['EstimatedMonthlyCost']) if snap['EstimatedMonthlyCost'] > 0 else "N/A",
-                    snap['Engine']
+                    snap["DBSnapshotIdentifier"],
+                    snap["DBInstanceIdentifier"],
+                    str(snap["AgeDays"]),
+                    str(snap["AllocatedStorage"]),
+                    format_cost(snap["EstimatedMonthlyCost"]) if snap["EstimatedMonthlyCost"] > 0 else "N/A",
+                    snap["Engine"],
                 )
-            
+
             console.print(cleanup_table)
-            
+
             if len(high_priority_candidates) > 15:
                 console.print(f"[dim]... and {len(high_priority_candidates) - 15} more high-priority snapshots[/dim]")
-                
+
         elif cleanup_candidates:
             print_warning(f"⚠ {len(cleanup_candidates)} old manual snapshots for review (>{old_days} days)")
         else:
@@ -341,15 +361,21 @@ def get_rds_snapshot_details(output_file, old_days, include_cost, snapshot_type,
             target_min_annual = 5000.0
             target_max_annual = 24000.0
             actual_savings = savings_analysis["old_manual_annual_savings"]
-            
+
             if actual_savings >= target_min_annual:
                 if actual_savings <= target_max_annual:
-                    print_success(f"🎯 Target Achievement: ${actual_savings:,.0f} within JIRA FinOps-23 range (${target_min_annual:,.0f}-${target_max_annual:,.0f})")
+                    print_success(
+                        f"🎯 Target Achievement: ${actual_savings:,.0f} within JIRA FinOps-23 range (${target_min_annual:,.0f}-${target_max_annual:,.0f})"
+                    )
                 else:
-                    print_success(f"🎯 Target Exceeded: ${actual_savings:,.0f} exceeds JIRA FinOps-23 maximum target (${target_max_annual:,.0f})")
+                    print_success(
+                        f"🎯 Target Exceeded: ${actual_savings:,.0f} exceeds JIRA FinOps-23 maximum target (${target_max_annual:,.0f})"
+                    )
             else:
                 percentage = (actual_savings / target_min_annual) * 100
-                print_warning(f"📊 Analysis: ${actual_savings:,.0f} is {percentage:.1f}% of JIRA FinOps-23 minimum target (${target_min_annual:,.0f})")
+                print_warning(
+                    f"📊 Analysis: ${actual_savings:,.0f} is {percentage:.1f}% of JIRA FinOps-23 minimum target (${target_min_annual:,.0f})"
+                )
 
         # Encryption status
         encrypted_count = sum(1 for s in data if s["Encrypted"])