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

runbooks/operate/networking_cost_heatmap.py

@@ -43,10 +43,19 @@ class NetworkingCostHeatMapConfig:
     """Configuration for networking cost heat map generation"""
 
     # AWS Profiles (READ-ONLY) - Universal environment support using proven profile pattern
-    billing_profile: str = field(default_factory=lambda: os.getenv("AWS_BILLING_PROFILE") or os.getenv("BILLING_PROFILE", "default"))
-    centralized_ops_profile: str = field(default_factory=lambda: os.getenv("AWS_CENTRALISED_OPS_PROFILE") or os.getenv("CENTRALISED_OPS_PROFILE", "default"))
-    single_account_profile: str = field(default_factory=lambda: os.getenv("AWS_SINGLE_ACCOUNT_PROFILE") or os.getenv("SINGLE_AWS_PROFILE", "default"))
-    management_profile: str = field(default_factory=lambda: os.getenv("AWS_MANAGEMENT_PROFILE") or os.getenv("MANAGEMENT_PROFILE", "default"))
+    billing_profile: str = field(
+        default_factory=lambda: os.getenv("AWS_BILLING_PROFILE") or os.getenv("BILLING_PROFILE", "default")
+    )
+    centralized_ops_profile: str = field(
+        default_factory=lambda: os.getenv("AWS_CENTRALISED_OPS_PROFILE")
+        or os.getenv("CENTRALISED_OPS_PROFILE", "default")
+    )
+    single_account_profile: str = field(
+        default_factory=lambda: os.getenv("AWS_SINGLE_ACCOUNT_PROFILE") or os.getenv("SINGLE_AWS_PROFILE", "default")
+    )
+    management_profile: str = field(
+        default_factory=lambda: os.getenv("AWS_MANAGEMENT_PROFILE") or os.getenv("MANAGEMENT_PROFILE", "default")
+    )
 
     # Analysis parameters
     regions: List[str] = field(
@@ -238,10 +247,22 @@ class NetworkingCostHeatMapOperation(BaseOperation):
 
         # Account categories for realistic distribution - dynamic from environment
         account_categories = {
-            "production": {"count": int(os.getenv("AWS_PROD_ACCOUNTS", "15")), "cost_multiplier": float(os.getenv("PROD_COST_MULTIPLIER", "5.0"))},
-            "staging": {"count": int(os.getenv("AWS_STAGING_ACCOUNTS", "15")), "cost_multiplier": float(os.getenv("STAGING_COST_MULTIPLIER", "2.0"))},
-            "development": {"count": int(os.getenv("AWS_DEV_ACCOUNTS", "20")), "cost_multiplier": float(os.getenv("DEV_COST_MULTIPLIER", "1.0"))},
-            "sandbox": {"count": int(os.getenv("AWS_SANDBOX_ACCOUNTS", "10")), "cost_multiplier": float(os.getenv("SANDBOX_COST_MULTIPLIER", "0.3"))},
+            "production": {
+                "count": int(os.getenv("AWS_PROD_ACCOUNTS", "15")),
+                "cost_multiplier": float(os.getenv("PROD_COST_MULTIPLIER", "5.0")),
+            },
+            "staging": {
+                "count": int(os.getenv("AWS_STAGING_ACCOUNTS", "15")),
+                "cost_multiplier": float(os.getenv("STAGING_COST_MULTIPLIER", "2.0")),
+            },
+            "development": {
+                "count": int(os.getenv("AWS_DEV_ACCOUNTS", "20")),
+                "cost_multiplier": float(os.getenv("DEV_COST_MULTIPLIER", "1.0")),
+            },
+            "sandbox": {
+                "count": int(os.getenv("AWS_SANDBOX_ACCOUNTS", "10")),
+                "cost_multiplier": float(os.getenv("SANDBOX_COST_MULTIPLIER", "0.3")),
+            },
         }
 
         # Generate aggregated heat map