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

runbooks/vpc/config.py

@@ -13,6 +13,7 @@ from typing import Any, Dict, List, Optional
 # Import AWS Pricing API for zero hardcoded values
 try:
     from runbooks.common.aws_pricing_api import pricing_api
+
     AWS_PRICING_AVAILABLE = True
 except ImportError:
     AWS_PRICING_AVAILABLE = False
@@ -25,20 +26,12 @@ class AWSCostModel:
     # NAT Gateway Pricing - Real-time from AWS Pricing API (NO hardcoded defaults)
     nat_gateway_hourly: float = field(default_factory=lambda: AWSCostModel._get_nat_gateway_hourly())
     nat_gateway_monthly: float = field(default_factory=lambda: AWSCostModel._get_nat_gateway_monthly())
-    nat_gateway_data_processing: float = field(
-        default_factory=lambda: AWSCostModel._get_nat_gateway_data_processing()
-    )
+    nat_gateway_data_processing: float = field(default_factory=lambda: AWSCostModel._get_nat_gateway_data_processing())
 
     # Transit Gateway Pricing - Real-time from AWS Pricing API (NO hardcoded defaults)
-    transit_gateway_hourly: float = field(
-        default_factory=lambda: AWSCostModel._get_transit_gateway_hourly()
-    )
-    transit_gateway_monthly: float = field(
-        default_factory=lambda: AWSCostModel._get_transit_gateway_monthly()
-    )
-    transit_gateway_attachment: float = field(
-        default_factory=lambda: AWSCostModel._get_transit_gateway_attachment()
-    )
+    transit_gateway_hourly: float = field(default_factory=lambda: AWSCostModel._get_transit_gateway_hourly())
+    transit_gateway_monthly: float = field(default_factory=lambda: AWSCostModel._get_transit_gateway_monthly())
+    transit_gateway_attachment: float = field(default_factory=lambda: AWSCostModel._get_transit_gateway_attachment())
     transit_gateway_data_processing: float = field(
         default_factory=lambda: AWSCostModel._get_transit_gateway_data_processing()
     )
@@ -56,60 +49,48 @@ class AWSCostModel:
     )
 
     # Elastic IP Pricing - ENTERPRISE COMPLIANCE: Real-time AWS Pricing API ONLY
-    elastic_ip_idle_hourly: float = field(
-        default_factory=lambda: AWSCostModel._get_elastic_ip_idle_hourly()
-    )
-    elastic_ip_idle_monthly: float = field(
-        default_factory=lambda: AWSCostModel._get_elastic_ip_idle_monthly()
-    )
+    elastic_ip_idle_hourly: float = field(default_factory=lambda: AWSCostModel._get_elastic_ip_idle_hourly())
+    elastic_ip_idle_monthly: float = field(default_factory=lambda: AWSCostModel._get_elastic_ip_idle_monthly())
     elastic_ip_attached: float = 0.0  # Always free when attached (AWS confirmed)
-    elastic_ip_remap: float = field(
-        default_factory=lambda: AWSCostModel._get_elastic_ip_remap()
-    )
+    elastic_ip_remap: float = field(default_factory=lambda: AWSCostModel._get_elastic_ip_remap())
 
     # Data Transfer Pricing - ENTERPRISE COMPLIANCE: Real-time AWS Pricing API ONLY
-    data_transfer_inter_az: float = field(
-        default_factory=lambda: AWSCostModel._get_data_transfer_inter_az()
-    )
-    data_transfer_inter_region: float = field(
-        default_factory=lambda: AWSCostModel._get_data_transfer_inter_region()
-    )
-    data_transfer_internet_out: float = field(
-        default_factory=lambda: AWSCostModel._get_data_transfer_internet_out()
-    )
+    data_transfer_inter_az: float = field(default_factory=lambda: AWSCostModel._get_data_transfer_inter_az())
+    data_transfer_inter_region: float = field(default_factory=lambda: AWSCostModel._get_data_transfer_inter_region())
+    data_transfer_internet_out: float = field(default_factory=lambda: AWSCostModel._get_data_transfer_internet_out())
     data_transfer_s3_same_region: float = 0.0  # Always free
-    
+
     @staticmethod
     def _get_nat_gateway_hourly() -> float:
         """Get NAT Gateway hourly cost from AWS Pricing API with enhanced enterprise fallback."""
         if AWS_PRICING_AVAILABLE:
             try:
                 # Use enhanced pricing API with regional fallback and graceful degradation
-                current_region = os.getenv('AWS_DEFAULT_REGION', 'us-east-1')
+                current_region = os.getenv("AWS_DEFAULT_REGION", "us-east-1")
                 monthly_cost = pricing_api.get_nat_gateway_monthly_cost(current_region)
                 return monthly_cost / (24 * 30)
             except Exception as e:
                 print(f"⚠️ NAT Gateway pricing API fallback: {e}")
-        
+
         # Universal compatibility: standard AWS pricing when API unavailable
         print("ℹ️ Using universal compatibility NAT Gateway rate")
         return 0.045  # AWS standard NAT Gateway hourly rate
-    
+
     @staticmethod
     def _get_nat_gateway_monthly() -> float:
         """Get NAT Gateway monthly cost from AWS Pricing API with enhanced enterprise fallback."""
         if AWS_PRICING_AVAILABLE:
             try:
                 # Use enhanced pricing API with regional fallback and graceful degradation
-                current_region = os.getenv('AWS_DEFAULT_REGION', 'us-east-1')
+                current_region = os.getenv("AWS_DEFAULT_REGION", "us-east-1")
                 return pricing_api.get_nat_gateway_monthly_cost(current_region)
             except Exception as e:
                 print(f"⚠️ NAT Gateway monthly pricing API fallback: {e}")
-        
+
         # Universal compatibility: calculate from hourly rate
         print("ℹ️ Calculating monthly cost from universal compatibility hourly rate")
         return AWSCostModel._get_nat_gateway_hourly() * 24 * 30
-    
+
     @staticmethod
     def _get_nat_gateway_data_processing() -> float:
         """Get NAT Gateway data processing cost from AWS Pricing API."""
@@ -121,7 +102,7 @@ class AWSCostModel:
                 pass
         # Universal compatibility: standard AWS pricing
         return 0.045  # AWS standard NAT Gateway data processing rate
-    
+
     @staticmethod
     def _get_transit_gateway_hourly() -> float:
         """Get Transit Gateway hourly cost from AWS Pricing API."""
@@ -133,12 +114,12 @@ class AWSCostModel:
                 pass
         # Universal compatibility: standard AWS pricing
         return 0.05  # AWS standard Transit Gateway hourly rate
-    
+
     @staticmethod
     def _get_transit_gateway_monthly() -> float:
         """Get Transit Gateway monthly cost from AWS Pricing API."""
         return AWSCostModel._get_transit_gateway_hourly() * 24 * 30
-    
+
     @staticmethod
     def _get_transit_gateway_attachment() -> float:
         """Get Transit Gateway attachment cost from AWS Pricing API."""
@@ -150,7 +131,7 @@ class AWSCostModel:
                 pass
         # Universal compatibility: standard AWS pricing
         return 0.05  # AWS standard TGW attachment rate
-    
+
     @staticmethod
     def _get_transit_gateway_data_processing() -> float:
         """Get Transit Gateway data processing cost from AWS Pricing API."""
@@ -162,7 +143,7 @@ class AWSCostModel:
                 pass
         # Universal compatibility: standard AWS pricing
         return 0.02  # AWS standard TGW data processing rate
-    
+
     # VPC Endpoint Pricing Methods
     @staticmethod
     def _get_vpc_endpoint_interface_hourly() -> float:
@@ -172,7 +153,7 @@ class AWSCostModel:
             # Universal compatibility: standard AWS pricing
             return 0.01  # AWS standard VPC Interface Endpoint hourly rate
         return float(value)
-    
+
     @staticmethod
     def _get_vpc_endpoint_interface_monthly() -> float:
         """Get VPC Endpoint Interface monthly cost from AWS Pricing API."""
@@ -181,7 +162,7 @@ class AWSCostModel:
             # Universal compatibility: calculate from hourly rate
             return AWSCostModel._get_vpc_endpoint_interface_hourly() * 24 * 30
         return float(value)
-    
+
     @staticmethod
     def _get_vpc_endpoint_data_processing() -> float:
         """Get VPC Endpoint data processing cost from AWS Pricing API."""
@@ -190,7 +171,7 @@ class AWSCostModel:
             # Universal compatibility: standard AWS pricing
             return 0.01  # AWS standard VPC Endpoint data processing rate
         return float(value)
-    
+
     # Elastic IP Pricing Methods
     @staticmethod
     def _get_elastic_ip_idle_hourly() -> float:
@@ -200,7 +181,7 @@ class AWSCostModel:
             # Universal compatibility: standard AWS pricing
             return 0.005  # AWS standard Elastic IP idle hourly rate
         return float(value)
-    
+
     @staticmethod
     def _get_elastic_ip_idle_monthly() -> float:
         """Get Elastic IP idle monthly cost from AWS Pricing API."""
@@ -209,7 +190,7 @@ class AWSCostModel:
             # Universal compatibility: calculate from hourly rate
             return AWSCostModel._get_elastic_ip_idle_hourly() * 24 * 30
         return float(value)
-    
+
     @staticmethod
     def _get_elastic_ip_remap() -> float:
         """Get Elastic IP remap cost from AWS Pricing API."""
@@ -218,7 +199,7 @@ class AWSCostModel:
             # Universal compatibility: standard AWS pricing
             return 0.10  # AWS standard Elastic IP remap cost
         return float(value)
-    
+
     # Data Transfer Pricing Methods
     @staticmethod
     def _get_data_transfer_inter_az() -> float:
@@ -228,7 +209,7 @@ class AWSCostModel:
             # Universal compatibility: standard AWS pricing
             return 0.01  # AWS standard Inter-AZ data transfer rate per GB
         return float(value)
-    
+
     @staticmethod
     def _get_data_transfer_inter_region() -> float:
         """Get Inter-region data transfer cost from AWS Pricing API."""
@@ -237,7 +218,7 @@ class AWSCostModel:
             # Universal compatibility: standard AWS pricing
             return 0.02  # AWS standard Inter-region data transfer rate per GB
         return float(value)
-    
+
     @staticmethod
     def _get_data_transfer_internet_out() -> float:
         """Get Internet outbound data transfer cost from AWS Pricing API."""
@@ -253,13 +234,23 @@ class OptimizationThresholds:
     """Configurable thresholds for optimization recommendations - NO hardcoded defaults"""
 
     # Usage thresholds - Dynamic from environment or raise error
-    idle_connection_threshold: int = field(default_factory=lambda: OptimizationThresholds._get_env_int("IDLE_CONNECTION_THRESHOLD"))
-    low_usage_gb_threshold: float = field(default_factory=lambda: OptimizationThresholds._get_env_float("LOW_USAGE_GB_THRESHOLD"))
-    low_connection_threshold: int = field(default_factory=lambda: OptimizationThresholds._get_env_int("LOW_CONNECTION_THRESHOLD"))
+    idle_connection_threshold: int = field(
+        default_factory=lambda: OptimizationThresholds._get_env_int("IDLE_CONNECTION_THRESHOLD")
+    )
+    low_usage_gb_threshold: float = field(
+        default_factory=lambda: OptimizationThresholds._get_env_float("LOW_USAGE_GB_THRESHOLD")
+    )
+    low_connection_threshold: int = field(
+        default_factory=lambda: OptimizationThresholds._get_env_int("LOW_CONNECTION_THRESHOLD")
+    )
 
     # Cost thresholds - Dynamic from environment or raise error
-    high_cost_threshold: float = field(default_factory=lambda: OptimizationThresholds._get_env_float("HIGH_COST_THRESHOLD"))
-    critical_cost_threshold: float = field(default_factory=lambda: OptimizationThresholds._get_env_float("CRITICAL_COST_THRESHOLD"))
+    high_cost_threshold: float = field(
+        default_factory=lambda: OptimizationThresholds._get_env_float("HIGH_COST_THRESHOLD")
+    )
+    critical_cost_threshold: float = field(
+        default_factory=lambda: OptimizationThresholds._get_env_float("CRITICAL_COST_THRESHOLD")
+    )
 
     # Optimization targets - Dynamic from environment or raise error
     target_reduction_percent: float = field(
@@ -270,23 +261,20 @@ class OptimizationThresholds:
     cost_approval_threshold: float = field(
         default_factory=lambda: OptimizationThresholds._get_env_float("COST_APPROVAL_THRESHOLD")
     )
-    
+
     @staticmethod
     def _get_env_int(var_name: str) -> int:
         """Get integer from environment with universal compatibility defaults."""
         value = os.getenv(var_name)
         if value is None:
             # Universal compatibility defaults for optimization thresholds
-            defaults = {
-                "IDLE_CONNECTION_THRESHOLD": 1,
-                "LOW_CONNECTION_THRESHOLD": 5
-            }
+            defaults = {"IDLE_CONNECTION_THRESHOLD": 1, "LOW_CONNECTION_THRESHOLD": 5}
             default_value = defaults.get(var_name)
             if default_value is None:
                 raise ValueError(f"Environment variable {var_name} required and no universal default available")
             return default_value
         return int(value)
-    
+
     @staticmethod
     def _get_env_float(var_name: str) -> float:
         """Get float from environment with universal compatibility defaults."""
@@ -299,13 +287,14 @@ class OptimizationThresholds:
                 "CRITICAL_COST_THRESHOLD": 1000.0,
                 "TARGET_REDUCTION_PERCENT": 0.30,
                 "COST_APPROVAL_THRESHOLD": 500.0,
-                "PERFORMANCE_BASELINE_THRESHOLD": 30.0
+                "PERFORMANCE_BASELINE_THRESHOLD": 30.0,
             }
             default_value = defaults.get(var_name)
             if default_value is None:
                 raise ValueError(f"Environment variable {var_name} required and no universal default available")
             return default_value
         return float(value)
+
     performance_baseline_threshold: float = field(
         default_factory=lambda: OptimizationThresholds._get_env_float("PERFORMANCE_BASELINE_THRESHOLD")
     )
@@ -334,14 +323,20 @@ class RegionalConfiguration:
     regional_multipliers: Dict[str, float] = field(
         default_factory=lambda: RegionalConfiguration._get_regional_multipliers()
     )
-    
+
     @staticmethod
     def _get_regional_multipliers() -> Dict[str, float]:
         """Get regional cost multipliers - NO hardcoded defaults."""
         regions = [
-            "us-east-1", "us-west-2", "us-west-1", 
-            "eu-west-1", "eu-central-1", "eu-west-2",
-            "ap-southeast-1", "ap-southeast-2", "ap-northeast-1"
+            "us-east-1",
+            "us-west-2",
+            "us-west-1",
+            "eu-west-1",
+            "eu-central-1",
+            "eu-west-2",
+            "ap-southeast-1",
+            "ap-southeast-2",
+            "ap-northeast-1",
         ]
         multipliers = {}
         for region in regions:
@@ -361,7 +356,7 @@ class VPCNetworkingConfig:
 
     # AWS Configuration - NO hardcoded defaults
     default_region: str = field(default_factory=lambda: VPCNetworkingConfig._get_required_env("AWS_DEFAULT_REGION"))
-    
+
     @staticmethod
     def _get_required_env(var_name: str) -> str:
         """Get environment variable with universal compatibility defaults."""
@@ -373,24 +368,21 @@ class VPCNetworkingConfig:
                 "OUTPUT_FORMAT": "json",
                 "OUTPUT_DIR": "./tmp",
                 "ENABLE_COST_APPROVAL_WORKFLOW": "false",
-                "ENABLE_MCP_VALIDATION": "false"
+                "ENABLE_MCP_VALIDATION": "false",
             }
             default_value = defaults.get(var_name)
             if default_value is None:
                 raise ValueError(f"Environment variable {var_name} required and no universal default available")
             return default_value
         return value
-    
+
     @staticmethod
     def _get_required_env_int(var_name: str) -> int:
         """Get integer environment variable with universal compatibility defaults."""
         value = os.getenv(var_name)
         if value is None:
             # Universal compatibility defaults for any AWS environment
-            defaults = {
-                "DEFAULT_ANALYSIS_DAYS": 30,
-                "FORECAST_DAYS": 30
-            }
+            defaults = {"DEFAULT_ANALYSIS_DAYS": 30, "FORECAST_DAYS": 30}
             default_value = defaults.get(var_name)
             if default_value is None:
                 raise ValueError(f"Environment variable {var_name} required and no universal default available")
@@ -398,31 +390,33 @@ class VPCNetworkingConfig:
         return int(value)
 
     # AWS Profiles - Universal compatibility with fallback to AWS_PROFILE or 'default'
-    billing_profile: Optional[str] = field(default_factory=lambda: (
-        os.getenv("BILLING_PROFILE") or 
-        os.getenv("AWS_PROFILE") or 
-        "default"
-    ))
-    centralized_ops_profile: Optional[str] = field(default_factory=lambda: (
-        os.getenv("CENTRALIZED_OPS_PROFILE") or 
-        os.getenv("CENTRALISED_OPS_PROFILE") or  # Alternative spelling
-        os.getenv("AWS_PROFILE") or 
-        "default"
-    ))
-    single_account_profile: Optional[str] = field(default_factory=lambda: (
-        os.getenv("SINGLE_ACCOUNT_PROFILE") or 
-        os.getenv("SINGLE_AWS_PROFILE") or  # Alternative naming
-        os.getenv("AWS_PROFILE") or 
-        "default"
-    ))
-    management_profile: Optional[str] = field(default_factory=lambda: (
-        os.getenv("MANAGEMENT_PROFILE") or 
-        os.getenv("AWS_PROFILE") or 
-        "default"
-    ))
+    billing_profile: Optional[str] = field(
+        default_factory=lambda: (os.getenv("BILLING_PROFILE") or os.getenv("AWS_PROFILE") or "default")
+    )
+    centralized_ops_profile: Optional[str] = field(
+        default_factory=lambda: (
+            os.getenv("CENTRALIZED_OPS_PROFILE")
+            or os.getenv("CENTRALISED_OPS_PROFILE")  # Alternative spelling
+            or os.getenv("AWS_PROFILE")
+            or "default"
+        )
+    )
+    single_account_profile: Optional[str] = field(
+        default_factory=lambda: (
+            os.getenv("SINGLE_ACCOUNT_PROFILE")
+            or os.getenv("SINGLE_AWS_PROFILE")  # Alternative naming
+            or os.getenv("AWS_PROFILE")
+            or "default"
+        )
+    )
+    management_profile: Optional[str] = field(
+        default_factory=lambda: (os.getenv("MANAGEMENT_PROFILE") or os.getenv("AWS_PROFILE") or "default")
+    )
 
     # Analysis Configuration - ENTERPRISE COMPLIANCE: No hardcoded defaults
-    default_analysis_days: int = field(default_factory=lambda: VPCNetworkingConfig._get_required_env_int("DEFAULT_ANALYSIS_DAYS"))
+    default_analysis_days: int = field(
+        default_factory=lambda: VPCNetworkingConfig._get_required_env_int("DEFAULT_ANALYSIS_DAYS")
+    )
     forecast_days: int = field(default_factory=lambda: VPCNetworkingConfig._get_required_env_int("FORECAST_DAYS"))
 
     # Output Configuration - ENTERPRISE COMPLIANCE: No hardcoded defaults
@@ -476,10 +470,11 @@ def load_config(config_file: Optional[str] = None) -> VPCNetworkingConfig:
         # Universal compatibility - warn instead of failing
         if not config.billing_profile or config.billing_profile == "default":
             import warnings
+
             warnings.warn(
                 "Cost approval workflow enabled but no specific BILLING_PROFILE set. "
                 "Using default profile. Set BILLING_PROFILE for enterprise multi-account setup.",
-                UserWarning
+                UserWarning,
             )
 
     return config