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

runbooks/common/date_utils.py

@@ -22,94 +22,82 @@ def get_current_year() -> int:
 def get_current_month_period() -> Dict[str, str]:
     """
     Generate current month's start and end dates for AWS API calls.
-    
+
     Returns:
         Dict with 'Start' and 'End' keys in YYYY-MM-DD format
     """
     now = datetime.now()
-    start_date = now.replace(day=1).strftime('%Y-%m-%d')
-    
+    start_date = now.replace(day=1).strftime("%Y-%m-%d")
+
     # Get last day of current month
     if now.month == 12:
         next_month = now.replace(year=now.year + 1, month=1, day=1)
     else:
         next_month = now.replace(month=now.month + 1, day=1)
-    
-    end_date = (next_month - timedelta(days=1)).strftime('%Y-%m-%d')
-    
-    return {
-        'Start': start_date,
-        'End': end_date
-    }
+
+    end_date = (next_month - timedelta(days=1)).strftime("%Y-%m-%d")
+
+    return {"Start": start_date, "End": end_date}
 
 
 def get_previous_month_period() -> Dict[str, str]:
     """
     Generate previous month's start and end dates for AWS API calls.
-    
+
     Returns:
         Dict with 'Start' and 'End' keys in YYYY-MM-DD format
     """
     now = datetime.now()
-    
+
     # Get first day of previous month
     if now.month == 1:
         prev_month = now.replace(year=now.year - 1, month=12, day=1)
     else:
         prev_month = now.replace(month=now.month - 1, day=1)
-    
-    start_date = prev_month.strftime('%Y-%m-%d')
-    
-    # Get last day of previous month  
-    end_date = (now.replace(day=1) - timedelta(days=1)).strftime('%Y-%m-%d')
-    
-    return {
-        'Start': start_date,
-        'End': end_date
-    }
+
+    start_date = prev_month.strftime("%Y-%m-%d")
+
+    # Get last day of previous month
+    end_date = (now.replace(day=1) - timedelta(days=1)).strftime("%Y-%m-%d")
+
+    return {"Start": start_date, "End": end_date}
 
 
 def get_test_date_period(days_back: int = 30) -> Dict[str, str]:
     """
     Generate test date periods for consistent test data.
-    
+
     Args:
         days_back: Number of days back from today for start date
-        
+
     Returns:
         Dict with 'Start' and 'End' keys in YYYY-MM-DD format
     """
-    end_date = datetime.now().strftime('%Y-%m-%d')
-    start_date = (datetime.now() - timedelta(days=days_back)).strftime('%Y-%m-%d')
-    
-    return {
-        'Start': start_date,
-        'End': end_date
-    }
+    end_date = datetime.now().strftime("%Y-%m-%d")
+    start_date = (datetime.now() - timedelta(days=days_back)).strftime("%Y-%m-%d")
+
+    return {"Start": start_date, "End": end_date}
 
 
 def get_aws_cli_example_period() -> Tuple[str, str]:
     """
     Generate example date period for AWS CLI documentation.
     Uses yesterday and today to ensure valid time range.
-    
+
     Returns:
         Tuple of (start_date, end_date) in YYYY-MM-DD format
     """
     today = datetime.now()
     yesterday = today - timedelta(days=1)
-    
-    return (
-        yesterday.strftime('%Y-%m-%d'),
-        today.strftime('%Y-%m-%d')
-    )
+
+    return (yesterday.strftime("%Y-%m-%d"), today.strftime("%Y-%m-%d"))
 
 
 def get_collection_timestamp() -> str:
     """
     Generate collection timestamp for test data.
-    
+
     Returns:
         ISO format timestamp string
     """
-    return datetime.now().isoformat()
+    return datetime.now().isoformat()

runbooks/common/decorators.py

@@ -41,8 +41,13 @@ def common_output_options(f):
     - --format: Output format selection (table, csv, json, markdown, pdf)
     - --output-file: File output destination
     """
-    f = click.option("--format", "output_format", type=click.Choice(['table', 'csv', 'json', 'markdown', 'pdf']),
-                     default='table', help="Output format")(f)
+    f = click.option(
+        "--format",
+        "output_format",
+        type=click.Choice(["table", "csv", "json", "markdown", "pdf"]),
+        default="table",
+        help="Output format",
+    )(f)
     f = click.option("--output-file", type=click.Path(), help="Output file path")(f)
     return f
 
@@ -69,6 +74,7 @@ def performance_timing(f):
     Automatically tracks and reports command execution time for performance
     monitoring and optimization analysis.
     """
+
     @functools.wraps(f)
     def wrapper(*args, **kwargs):
         start_time = time.time()
@@ -98,6 +104,7 @@ def error_handler(f):
     - Consistent error message structure
     - Debug information when enabled
     """
+
     @functools.wraps(f)
     def wrapper(*args, **kwargs):
         try:
@@ -124,14 +131,16 @@ def require_aws_profile(f):
     Validates that the AWS profile exists and is accessible before
     executing commands that require AWS API access.
     """
+
     @functools.wraps(f)
     def wrapper(*args, **kwargs):
         # Get profile from context or kwargs
         ctx = click.get_current_context()
-        profile = ctx.obj.get('profile', 'default')
+        profile = ctx.obj.get("profile", "default")
 
         try:
             import boto3
+
             # Test profile access
             session = boto3.Session(profile_name=profile)
             session.get_credentials()
@@ -154,36 +163,37 @@ def enterprise_audit_trail(f):
     - User context and timestamp
     - Execution results and duration
     """
+
     @functools.wraps(f)
     def wrapper(*args, **kwargs):
         ctx = click.get_current_context()
 
         # Log command execution start
         audit_data = {
-            'command': ctx.command.name,
-            'profile': ctx.obj.get('profile', 'default'),
-            'region': ctx.obj.get('region', 'default'),
-            'dry_run': ctx.obj.get('dry_run', False),
-            'timestamp': time.time()
+            "command": ctx.command.name,
+            "profile": ctx.obj.get("profile", "default"),
+            "region": ctx.obj.get("region", "default"),
+            "dry_run": ctx.obj.get("dry_run", False),
+            "timestamp": time.time(),
         }
 
         try:
             result = f(*args, **kwargs)
-            audit_data['status'] = 'success'
-            audit_data['duration'] = time.time() - audit_data['timestamp']
+            audit_data["status"] = "success"
+            audit_data["duration"] = time.time() - audit_data["timestamp"]
 
             # Log successful execution
-            if ctx.obj.get('debug'):
+            if ctx.obj.get("debug"):
                 console.print(f"[dim]📋 Audit: {audit_data}[/dim]")
 
             return result
         except Exception as e:
-            audit_data['status'] = 'error'
-            audit_data['error'] = str(e)
-            audit_data['duration'] = time.time() - audit_data['timestamp']
+            audit_data["status"] = "error"
+            audit_data["error"] = str(e)
+            audit_data["duration"] = time.time() - audit_data["timestamp"]
 
             # Log failed execution
-            if ctx.obj.get('debug'):
+            if ctx.obj.get("debug"):
                 console.print(f"[dim]📋 Audit: {audit_data}[/dim]")
 
             raise
@@ -201,15 +211,14 @@ def rich_progress(description: str = "Processing"):
     Automatically shows a progress spinner for operations that take time,
     improving user experience for long-running commands.
     """
+
     def decorator(f):
         @functools.wraps(f)
         def wrapper(*args, **kwargs):
             from rich.progress import Progress, SpinnerColumn, TextColumn
 
             with Progress(
-                SpinnerColumn(),
-                TextColumn("[progress.description]{task.description}"),
-                console=console
+                SpinnerColumn(), TextColumn("[progress.description]{task.description}"), console=console
             ) as progress:
                 task = progress.add_task(description, total=None)
 
@@ -222,4 +231,5 @@ def rich_progress(description: str = "Processing"):
                     raise
 
         return wrapper
-    return decorator
+
+    return decorator