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

runbooks/common/rich_utils.py

@@ -13,6 +13,7 @@ Features:
 - Error/warning/success message formatting
 - Tree displays for hierarchical data
 - Layout templates for complex displays
+- Test mode support to prevent I/O conflicts with Click CliRunner
 
 Author: CloudOps Runbooks Team
 Version: 0.7.8
@@ -20,6 +21,9 @@ Version: 0.7.8
 
 import csv
 import json
+import os
+import re
+import sys
 import tempfile
 from datetime import datetime
 from io import StringIO
@@ -27,7 +31,6 @@ from typing import Any, Dict, List, Optional, Union
 
 from rich import box
 from rich.columns import Columns
-from rich.console import Console
 from rich.layout import Layout
 from rich.markdown import Markdown
 from rich.panel import Panel
@@ -35,11 +38,58 @@ from rich.progress import BarColumn, Progress, SpinnerColumn, TaskProgressColumn
 from rich.rule import Rule
 from rich.style import Style
 from rich.syntax import Syntax
-from rich.table import Table
+from rich.table import Table as RichTable
 from rich.text import Text
 from rich.theme import Theme
 from rich.tree import Tree
 
+# Test Mode Support: Disable Rich Console in test environments to prevent I/O conflicts
+# Issue: Rich Console writes to StringIO buffer that Click CliRunner closes, causing ValueError
+# Solution: Use plain print() in test mode (RUNBOOKS_TEST_MODE=1), Rich Console in production
+USE_RICH = os.getenv("RUNBOOKS_TEST_MODE") != "1"
+
+if USE_RICH:
+    from rich.console import Console as RichConsole
+
+    Console = RichConsole
+    Table = RichTable
+else:
+    # Mock Rich Console for testing - plain text output compatible with Click CliRunner
+    class MockConsole:
+        """Mock console that prints to stdout without Rich formatting."""
+
+        def print(self, *args, **kwargs):
+            """Mock print that outputs plain text to stdout."""
+            if args:
+                # Extract text content from Rich markup if present
+                text = str(args[0]) if args else ""
+                # Remove Rich markup tags for plain output
+                text = re.sub(r"\[.*?\]", "", text)
+                print(text, file=sys.stdout)
+
+        def __enter__(self):
+            return self
+
+        def __exit__(self, *args):
+            pass
+
+    class MockTable:
+        """Mock table for testing - minimal implementation."""
+
+        def __init__(self, *args, **kwargs):
+            self.title = kwargs.get("title", "")
+            self.columns = []
+            self.rows = []
+
+        def add_column(self, header, **kwargs):
+            self.columns.append(header)
+
+        def add_row(self, *args):
+            self.rows.append(args)
+
+    Console = MockConsole
+    Table = MockTable
+
 # CloudOps Custom Theme
 CLOUDOPS_THEME = Theme(
     {
@@ -59,8 +109,11 @@ CLOUDOPS_THEME = Theme(
     }
 )
 
-# Initialize console with custom theme
-console = Console(theme=CLOUDOPS_THEME)
+# Initialize console with custom theme (test-aware via USE_RICH flag)
+if USE_RICH:
+    console = Console(theme=CLOUDOPS_THEME)
+else:
+    console = Console()  # MockConsole instance
 
 # Status indicators
 STATUS_INDICATORS = {
@@ -109,6 +162,7 @@ def print_header(title: str, version: Optional[str] = None) -> None:
     """
     if version is None:
         from runbooks import __version__
+
         version = __version__
 
     header_text = Text()
@@ -124,7 +178,10 @@ def print_header(title: str, version: Optional[str] = None) -> None:
 def print_banner() -> None:
     """Print a clean, minimal CloudOps Runbooks banner."""
     from runbooks import __version__
-    console.print(f"\n[header]CloudOps Runbooks[/header] [subheader]Enterprise AWS Automation Platform[/subheader] [dim]v{__version__}[/dim]")
+
+    console.print(
+        f"\n[header]CloudOps Runbooks[/header] [subheader]Enterprise AWS Automation Platform[/subheader] [dim]v{__version__}[/dim]"
+    )
     console.print()
 
 
@@ -431,7 +488,9 @@ def create_display_profile_name(profile_name: str, max_length: int = 25, context
     return f"{profile_name[: max_length - 3]}..."
 
 
-def format_profile_name(profile_name: str, style: str = "cyan", display_max_length: int = 25, secure_logging: bool = True) -> Text:
+def format_profile_name(
+    profile_name: str, style: str = "cyan", display_max_length: int = 25, secure_logging: bool = True
+) -> Text:
     """
     Format profile name with consistent styling, intelligent truncation, and security enhancements.
 
@@ -449,7 +508,7 @@ def format_profile_name(profile_name: str, style: str = "cyan", display_max_leng
 
     Returns:
         Rich Text object with formatted profile name
-        
+
     Security Note:
         When secure_logging=True, account IDs are masked in display to prevent
         account enumeration while maintaining profile identification.
@@ -458,13 +517,14 @@ def format_profile_name(profile_name: str, style: str = "cyan", display_max_leng
     if secure_logging:
         try:
             from runbooks.common.aws_utils import AWSProfileSanitizer
+
             display_profile = AWSProfileSanitizer.sanitize_profile_name(profile_name)
         except ImportError:
             # Fallback to original profile if aws_utils not available
             display_profile = profile_name
     else:
         display_profile = profile_name
-    
+
     display_name = create_display_profile_name(display_profile, display_max_length)
 
     text = Text()
@@ -476,7 +536,7 @@ def format_profile_name(profile_name: str, style: str = "cyan", display_max_leng
     else:
         # Full name - normal style
         text.append(display_name, style=style)
-        
+
     # Add security indicator for sanitized profiles
     if secure_logging and "***masked***" in display_name:
         text.append(" 🔒", style="dim yellow")
@@ -623,21 +683,21 @@ def format_workspaces_analysis(workspaces_data: Dict[str, Any], target_savings:
     Args:
         workspaces_data: Dictionary containing WorkSpaces cost and utilization data
         target_savings: Annual savings target (default: $12,518)
-        
+
     Returns:
         Rich Panel with formatted WorkSpaces analysis
     """
-    current_cost = workspaces_data.get('monthly_cost', 0)
-    unused_count = workspaces_data.get('unused_count', 0)
-    total_count = workspaces_data.get('total_count', 0)
-    optimization_potential = workspaces_data.get('optimization_potential', 0)
-    
+    current_cost = workspaces_data.get("monthly_cost", 0)
+    unused_count = workspaces_data.get("unused_count", 0)
+    total_count = workspaces_data.get("total_count", 0)
+    optimization_potential = workspaces_data.get("optimization_potential", 0)
+
     annual_savings = optimization_potential * 12
     target_achievement = min(100, (annual_savings / target_savings) * 100) if target_savings > 0 else 0
-    
+
     status = "🎯 TARGET ACHIEVABLE" if target_achievement >= 90 else "⚠️ TARGET REQUIRES EXPANDED SCOPE"
     status_style = "bright_green" if target_achievement >= 90 else "yellow"
-    
+
     content = f"""💼 [bold]Manager's Priority #1: WorkSpaces Cleanup Analysis[/bold]
 
 📊 Current State:
@@ -658,35 +718,38 @@ def format_workspaces_analysis(workspaces_data: Dict[str, Any], target_savings:
 
 [{status_style}]{status}[/]"""
 
-    return Panel(content, title="[bright_cyan]WorkSpaces Cost Optimization[/bright_cyan]",
-                border_style="bright_green" if target_achievement >= 90 else "yellow")
+    return Panel(
+        content,
+        title="[bright_cyan]WorkSpaces Cost Optimization[/bright_cyan]",
+        border_style="bright_green" if target_achievement >= 90 else "yellow",
+    )
 
 
 def format_nat_gateway_optimization(nat_data: Dict[str, Any], target_completion: int = 95) -> Panel:
     """
     Format NAT Gateway optimization analysis for manager's completion target.
-    
+
     Manager's requirement to increase NAT Gateway optimization from 75% to 95% completion.
-    
+
     Args:
         nat_data: Dictionary containing NAT Gateway configuration and cost data
         target_completion: Completion target percentage (default: 95% from manager's priority)
-        
+
     Returns:
         Rich Panel with formatted NAT Gateway optimization analysis
     """
-    total_gateways = nat_data.get('total', 0)
-    active_gateways = nat_data.get('active', 0)
-    monthly_cost = nat_data.get('monthly_cost', 0)
-    optimization_ready = nat_data.get('optimization_ready', 0)
-    
+    total_gateways = nat_data.get("total", 0)
+    active_gateways = nat_data.get("active", 0)
+    monthly_cost = nat_data.get("monthly_cost", 0)
+    optimization_ready = nat_data.get("optimization_ready", 0)
+
     current_completion = 75  # Manager specified current state
     optimization_potential = monthly_cost * 0.75  # 75% can be optimized
     annual_savings = optimization_potential * 12
-    
+
     completion_gap = target_completion - current_completion
     status = "🎯 READY FOR 95% TARGET" if active_gateways > 0 else "❌ NO OPTIMIZATION OPPORTUNITIES"
-    
+
     content = f"""🌐 [bold]Manager's Priority #2: NAT Gateway Optimization[/bold]
 
 🔍 Current Infrastructure:
@@ -711,45 +774,46 @@ def format_nat_gateway_optimization(nat_data: Dict[str, Any], target_completion:
 
 [bright_green]{status}[/bright_green]"""
 
-    return Panel(content, title="[bright_cyan]Manager's Priority #2: NAT Gateway Optimization[/bright_cyan]",
-                border_style="cyan")
+    return Panel(
+        content, title="[bright_cyan]Manager's Priority #2: NAT Gateway Optimization[/bright_cyan]", border_style="cyan"
+    )
 
 
 def format_rds_optimization_analysis(rds_data: Dict[str, Any], savings_range: Dict[str, int] = None) -> Panel:
     """
     Format RDS Multi-AZ optimization analysis for manager's FinOps-23 scenario.
-    
+
     Manager's requirement for measurable range annual savings through RDS manual snapshot cleanup
     and Multi-AZ configuration review.
-    
+
     Args:
         rds_data: Dictionary containing RDS instance and snapshot data
         savings_range: Dict with 'min' and 'max' annual savings (default: {'min': 5000, 'max': 24000})
-        
+
     Returns:
         Rich Panel with formatted RDS optimization analysis
     """
     if savings_range is None:
-        savings_range = {'min': 5000, 'max': 24000}
-    
-    total_instances = rds_data.get('total', 0)
-    multi_az_instances = rds_data.get('multi_az_instances', 0)
-    manual_snapshots = rds_data.get('manual_snapshots', 0)
-    snapshot_storage_gb = rds_data.get('snapshot_storage_gb', 0)
-    
+        savings_range = {"min": 5000, "max": 24000}
+
+    total_instances = rds_data.get("total", 0)
+    multi_az_instances = rds_data.get("multi_az_instances", 0)
+    manual_snapshots = rds_data.get("manual_snapshots", 0)
+    snapshot_storage_gb = rds_data.get("snapshot_storage_gb", 0)
+
     # Calculate savings potential
     snapshot_savings = snapshot_storage_gb * 0.095 * 12  # $0.095/GB/month
     multi_az_savings = multi_az_instances * 1000 * 12  # ~$1K/month per instance
     total_savings = snapshot_savings + multi_az_savings
-    
-    savings_min = savings_range['min']
-    savings_max = savings_range['max']
-    
+
+    savings_min = savings_range["min"]
+    savings_max = savings_range["max"]
+
     # Check if we're within manager's target range
     within_range = savings_min <= total_savings <= savings_max
     range_status = "✅ WITHIN TARGET RANGE" if within_range else "📊 ANALYSIS PENDING"
     range_style = "bright_green" if within_range else "yellow"
-    
+
     content = f"""🗄️ [bold]Manager's Priority #3: RDS Cost Optimization[/bold]
 
 📊 Current RDS Environment:
@@ -775,42 +839,45 @@ def format_rds_optimization_analysis(rds_data: Dict[str, Any], savings_range: Di
 
 [{range_style}]{range_status}[/]"""
 
-    return Panel(content, title="[bright_cyan]FinOps-23: RDS Multi-AZ & Snapshot Optimization[/bright_cyan]",
-                border_style="bright_green" if within_range else "yellow")
+    return Panel(
+        content,
+        title="[bright_cyan]FinOps-23: RDS Multi-AZ & Snapshot Optimization[/bright_cyan]",
+        border_style="bright_green" if within_range else "yellow",
+    )
 
 
 def format_manager_business_summary(all_scenarios_data: Dict[str, Any]) -> Panel:
     """
     Format executive summary panel for manager's complete AWSO business case.
-    
+
     Combines all three manager priorities into executive-ready decision package:
     - FinOps-24: WorkSpaces cleanup ($12,518)
     - Manager Priority #2: NAT Gateway optimization (95% completion)
     - FinOps-23: RDS optimization (measurable range range)
-    
+
     Args:
         all_scenarios_data: Dictionary containing data from all three scenarios
-        
+
     Returns:
         Rich Panel with complete executive summary
     """
-    workspaces = all_scenarios_data.get('workspaces', {})
-    nat_gateway = all_scenarios_data.get('nat_gateway', {})
-    rds = all_scenarios_data.get('rds', {})
-    
+    workspaces = all_scenarios_data.get("workspaces", {})
+    nat_gateway = all_scenarios_data.get("nat_gateway", {})
+    rds = all_scenarios_data.get("rds", {})
+
     # Calculate totals
-    workspaces_annual = workspaces.get('optimization_potential', 0) * 12
-    nat_annual = nat_gateway.get('monthly_cost', 0) * 0.75 * 12
-    rds_annual = rds.get('total_savings', 15000)  # Mid-range estimate
-    
+    workspaces_annual = workspaces.get("optimization_potential", 0) * 12
+    nat_annual = nat_gateway.get("monthly_cost", 0) * 0.75 * 12
+    rds_annual = rds.get("total_savings", 15000)  # Mid-range estimate
+
     total_min_savings = workspaces_annual + nat_annual + 5000
     total_max_savings = workspaces_annual + nat_annual + 24000
-    
+
     # Overall assessment
     overall_confidence = 85  # Weighted average of individual confidences
     payback_months = 2.4  # Quick payback period
     roi_percentage = 567  # Strong ROI
-    
+
     content = f"""🏆 [bold]MANAGER'S AWSO BUSINESS CASE - EXECUTIVE SUMMARY[/bold]
 
 💼 Three Strategic Priorities:
@@ -837,8 +904,12 @@ def format_manager_business_summary(all_scenarios_data: Dict[str, Any]) -> Panel
 
 🎯 [bold]RECOMMENDATION: APPROVED FOR IMPLEMENTATION[/bold]"""
 
-    return Panel(content, title="[bright_green]🏆 MANAGER'S AWSO BUSINESS CASE - DECISION PACKAGE[/bright_green]",
-                border_style="bright_green", padding=(1, 2))
+    return Panel(
+        content,
+        title="[bright_green]🏆 MANAGER'S AWSO BUSINESS CASE - DECISION PACKAGE[/bright_green]",
+        border_style="bright_green",
+        padding=(1, 2),
+    )
 
 
 # Export all public functions and constants
@@ -891,18 +962,18 @@ __all__ = [
 def create_dual_metric_display(unblended_total: float, amortized_total: float, variance_pct: float) -> Columns:
     """
     Create dual-metric cost display with technical and financial perspectives.
-    
+
     Args:
         unblended_total: Technical total (UnblendedCost)
-        amortized_total: Financial total (AmortizedCost)  
+        amortized_total: Financial total (AmortizedCost)
         variance_pct: Variance percentage between metrics
-        
+
     Returns:
         Rich Columns object with dual-metric display
     """
     from rich.columns import Columns
     from rich.panel import Panel
-    
+
     # Technical perspective (UnblendedCost)
     tech_content = Text()
     tech_content.append("🔧 Technical Analysis\n", style="bright_blue bold")
@@ -913,14 +984,9 @@ def create_dual_metric_display(unblended_total: float, amortized_total: float, v
     tech_content.append("Resource optimization\n", style="white")
     tech_content.append("Audience: ", style="bright_blue")
     tech_content.append("DevOps, SRE, Tech teams", style="white")
-    
-    tech_panel = Panel(
-        tech_content,
-        title="🔧 Technical Perspective",
-        border_style="bright_blue",
-        padding=(1, 2)
-    )
-    
+
+    tech_panel = Panel(tech_content, title="🔧 Technical Perspective", border_style="bright_blue", padding=(1, 2))
+
     # Financial perspective (AmortizedCost)
     financial_content = Text()
     financial_content.append("📊 Financial Reporting\n", style="bright_green bold")
@@ -931,14 +997,11 @@ def create_dual_metric_display(unblended_total: float, amortized_total: float, v
     financial_content.append("Budget planning\n", style="white")
     financial_content.append("Audience: ", style="bright_green")
     financial_content.append("Finance, Executives", style="white")
-    
+
     financial_panel = Panel(
-        financial_content,
-        title="📊 Financial Perspective", 
-        border_style="bright_green",
-        padding=(1, 2)
+        financial_content, title="📊 Financial Perspective", border_style="bright_green", padding=(1, 2)
     )
-    
+
     return Columns([tech_panel, financial_panel])
 
 
@@ -978,6 +1041,7 @@ def format_metric_variance(variance: float, variance_pct: float) -> Text:
 # UNIVERSAL FORMAT EXPORT FUNCTIONS
 # ===========================
 
+
 def export_data(data: Any, format_type: str, output_file: Optional[str] = None, title: Optional[str] = None) -> str:
     """
     Universal data export function supporting multiple output formats.
@@ -999,7 +1063,7 @@ def export_data(data: Any, format_type: str, output_file: Optional[str] = None,
     format_type = format_type.lower().strip()
 
     # Handle table display (default Rich behavior)
-    if format_type == 'table':
+    if format_type == "table":
         if isinstance(data, Table):
             # Capture Rich table output
             with console.capture() as capture:
@@ -1009,26 +1073,26 @@ def export_data(data: Any, format_type: str, output_file: Optional[str] = None,
             # Convert data to table format
             output = _convert_to_table_string(data, title)
 
-    elif format_type == 'csv':
+    elif format_type == "csv":
         output = export_to_csv(data, title)
 
-    elif format_type == 'json':
+    elif format_type == "json":
         output = export_to_json(data, title)
 
-    elif format_type == 'markdown':
+    elif format_type == "markdown":
         output = export_to_markdown(data, title)
 
-    elif format_type == 'pdf':
+    elif format_type == "pdf":
         output = export_to_pdf(data, title, output_file)
 
     else:
-        supported_formats = ['table', 'csv', 'json', 'markdown', 'pdf']
+        supported_formats = ["table", "csv", "json", "markdown", "pdf"]
         raise ValueError(f"Unsupported format: {format_type}. Supported formats: {supported_formats}")
 
     # Write to file if specified
-    if output_file and format_type != 'pdf':  # PDF handles its own file writing
+    if output_file and format_type != "pdf":  # PDF handles its own file writing
         try:
-            with open(output_file, 'w', encoding='utf-8') as f:
+            with open(output_file, "w", encoding="utf-8") as f:
                 f.write(output)
             print_success(f"Output saved to: {output_file}")
         except IOError as e:
@@ -1078,14 +1142,14 @@ def export_to_csv(data: Any, title: Optional[str] = None) -> str:
     elif isinstance(data, dict):
         # Dictionary - convert to key-value pairs
         writer = csv.writer(output)
-        writer.writerow(['Key', 'Value'])
+        writer.writerow(["Key", "Value"])
         for key, value in data.items():
             writer.writerow([key, value])
 
     else:
         # Fallback for other types
         writer = csv.writer(output)
-        writer.writerow(['Data'])
+        writer.writerow(["Data"])
         writer.writerow([str(data)])
 
     return output.getvalue()
@@ -1105,7 +1169,7 @@ def export_to_json(data: Any, title: Optional[str] = None) -> str:
     # Prepare data for JSON serialization
     if isinstance(data, Table):
         json_data = _extract_table_data_as_dict(data)
-    elif hasattr(data, '__dict__'):
+    elif hasattr(data, "__dict__"):
         # Object with attributes
         json_data = data.__dict__
     else:
@@ -1115,12 +1179,8 @@ def export_to_json(data: Any, title: Optional[str] = None) -> str:
     # Add metadata if title provided
     if title:
         output_data = {
-            "metadata": {
-                "title": title,
-                "generated": datetime.now().isoformat(),
-                "format": "json"
-            },
-            "data": json_data
+            "metadata": {"title": title, "generated": datetime.now().isoformat(), "format": "json"},
+            "data": json_data,
         }
     else:
         output_data = json_data
@@ -1217,13 +1277,11 @@ def export_to_pdf(data: Any, title: Optional[str] = None, output_file: Optional[
         from reportlab.lib.units import inch
         from reportlab.platypus import SimpleDocTemplate, Table as RLTable, TableStyle, Paragraph, Spacer
     except ImportError:
-        raise ImportError(
-            "PDF export requires reportlab. Install with: pip install reportlab"
-        )
+        raise ImportError("PDF export requires reportlab. Install with: pip install reportlab")
 
     if not output_file:
         # Generate temporary file if none provided
-        output_file = tempfile.mktemp(suffix='.pdf')
+        output_file = tempfile.mktemp(suffix=".pdf")
 
     # Create PDF document
     doc = SimpleDocTemplate(output_file, pagesize=A4)
@@ -1233,18 +1291,14 @@ def export_to_pdf(data: Any, title: Optional[str] = None, output_file: Optional[
     # Add title
     if title:
         title_style = ParagraphStyle(
-            'CustomTitle',
-            parent=styles['Heading1'],
-            fontSize=16,
-            textColor=colors.darkblue,
-            spaceAfter=12
+            "CustomTitle", parent=styles["Heading1"], fontSize=16, textColor=colors.darkblue, spaceAfter=12
         )
         story.append(Paragraph(title, title_style))
         story.append(Spacer(1, 12))
 
     # Add generation info
     info_text = f"Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
-    story.append(Paragraph(info_text, styles['Normal']))
+    story.append(Paragraph(info_text, styles["Normal"]))
     story.append(Spacer(1, 12))
 
     # Handle different data types
@@ -1254,16 +1308,20 @@ def export_to_pdf(data: Any, title: Optional[str] = None, output_file: Optional[
         if table_data:
             # Create ReportLab table
             rl_table = RLTable(table_data)
-            rl_table.setStyle(TableStyle([
-                ('BACKGROUND', (0, 0), (-1, 0), colors.lightblue),
-                ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
-                ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
-                ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
-                ('FONTSIZE', (0, 0), (-1, 0), 12),
-                ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
-                ('BACKGROUND', (0, 1), (-1, -1), colors.beige),
-                ('GRID', (0, 0), (-1, -1), 1, colors.black)
-            ]))
+            rl_table.setStyle(
+                TableStyle(
+                    [
+                        ("BACKGROUND", (0, 0), (-1, 0), colors.lightblue),
+                        ("TEXTCOLOR", (0, 0), (-1, 0), colors.whitesmoke),
+                        ("ALIGN", (0, 0), (-1, -1), "CENTER"),
+                        ("FONTNAME", (0, 0), (-1, 0), "Helvetica-Bold"),
+                        ("FONTSIZE", (0, 0), (-1, 0), 12),
+                        ("BOTTOMPADDING", (0, 0), (-1, 0), 12),
+                        ("BACKGROUND", (0, 1), (-1, -1), colors.beige),
+                        ("GRID", (0, 0), (-1, -1), 1, colors.black),
+                    ]
+                )
+            )
             story.append(rl_table)
 
     elif isinstance(data, (list, dict)):
@@ -1275,26 +1333,30 @@ def export_to_pdf(data: Any, title: Optional[str] = None, output_file: Optional[
             table_data = [headers] + rows
 
             rl_table = RLTable(table_data)
-            rl_table.setStyle(TableStyle([
-                ('BACKGROUND', (0, 0), (-1, 0), colors.lightblue),
-                ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
-                ('ALIGN', (0, 0), (-1, -1), 'LEFT'),
-                ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
-                ('FONTSIZE', (0, 0), (-1, 0), 10),
-                ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
-                ('BACKGROUND', (0, 1), (-1, -1), colors.beige),
-                ('GRID', (0, 0), (-1, -1), 1, colors.black)
-            ]))
+            rl_table.setStyle(
+                TableStyle(
+                    [
+                        ("BACKGROUND", (0, 0), (-1, 0), colors.lightblue),
+                        ("TEXTCOLOR", (0, 0), (-1, 0), colors.whitesmoke),
+                        ("ALIGN", (0, 0), (-1, -1), "LEFT"),
+                        ("FONTNAME", (0, 0), (-1, 0), "Helvetica-Bold"),
+                        ("FONTSIZE", (0, 0), (-1, 0), 10),
+                        ("BOTTOMPADDING", (0, 0), (-1, 0), 12),
+                        ("BACKGROUND", (0, 1), (-1, -1), colors.beige),
+                        ("GRID", (0, 0), (-1, -1), 1, colors.black),
+                    ]
+                )
+            )
             story.append(rl_table)
         else:
             # Convert to readable text
             text_content = json.dumps(data, indent=2, default=str, ensure_ascii=False)
-            for line in text_content.split('\n'):
-                story.append(Paragraph(line, styles['Code']))
+            for line in text_content.split("\n"):
+                story.append(Paragraph(line, styles["Code"]))
 
     else:
         # Other data types
-        story.append(Paragraph(str(data), styles['Normal']))
+        story.append(Paragraph(str(data), styles["Normal"]))
 
     # Build PDF
     doc.build(story)
@@ -1336,11 +1398,7 @@ def _extract_table_data_as_dict(table: Table) -> Dict[str, Any]:
     headers = table_data[0]
     rows = table_data[1:]
 
-    return {
-        "headers": headers,
-        "rows": rows,
-        "row_count": len(rows)
-    }
+    return {"headers": headers, "rows": rows, "row_count": len(rows)}
 
 
 def _convert_to_table_string(data: Any, title: Optional[str] = None) -> str:
@@ -1372,7 +1430,9 @@ def _write_csv_data(output: StringIO, csv_data: List[List[str]]) -> None:
         writer.writerows(csv_data)
 
 
-def handle_output_format(data: Any, output_format: str = 'table', output_file: Optional[str] = None, title: Optional[str] = None):
+def handle_output_format(
+    data: Any, output_format: str = "table", output_file: Optional[str] = None, title: Optional[str] = None
+):
     """
     Handle output formatting for CLI commands - unified interface for all modules.
 
@@ -1400,7 +1460,7 @@ def handle_output_format(data: Any, output_format: str = 'table', output_file: O
         handle_output_format(data, output_format='pdf', output_file='report.pdf', title='AWS Resources Report')
     """
     try:
-        if output_format == 'table':
+        if output_format == "table":
             # Default Rich table display - just print to console
             if isinstance(data, Table):
                 console.print(data)
@@ -1438,10 +1498,10 @@ def handle_output_format(data: Any, output_format: str = 'table', output_file: O
             output = export_data(data, output_format, output_file, title)
 
             # If no output file specified, print to console for non-table formats
-            if not output_file and output_format != 'pdf':
-                if output_format == 'json':
+            if not output_file and output_format != "pdf":
+                if output_format == "json":
                     print_json(json.loads(output))
-                elif output_format == 'markdown':
+                elif output_format == "markdown":
                     print_markdown(output)
                 else:
                     console.print(output)