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

runbooks/finops/optimizer.py

@@ -519,17 +519,17 @@ class CostOptimizer:
             start_time = end_time - timedelta(days=days)
 
             response = cloudwatch.get_metric_statistics(
-                Namespace='AWS/EC2',
-                MetricName='CPUUtilization',
-                Dimensions=[{'Name': 'InstanceId', 'Value': instance_id}],
+                Namespace="AWS/EC2",
+                MetricName="CPUUtilization",
+                Dimensions=[{"Name": "InstanceId", "Value": instance_id}],
                 StartTime=start_time,
                 EndTime=end_time,
                 Period=3600,  # 1 hour
-                Statistics=['Average']
+                Statistics=["Average"],
             )
 
-            if response['Datapoints']:
-                cpu_avg = sum(dp['Average'] for dp in response['Datapoints']) / len(response['Datapoints'])
+            if response["Datapoints"]:
+                cpu_avg = sum(dp["Average"] for dp in response["Datapoints"]) / len(response["Datapoints"])
                 return cpu_avg
             else:
                 console.print(f"[yellow]⚠️ No CPU metrics found for {instance_id}[/yellow]")
@@ -548,17 +548,17 @@ class CostOptimizer:
             start_time = end_time - timedelta(days=days)
 
             response = cloudwatch.get_metric_statistics(
-                Namespace='CWAgent',
-                MetricName='mem_used_percent',
-                Dimensions=[{'Name': 'InstanceId', 'Value': instance_id}],
+                Namespace="CWAgent",
+                MetricName="mem_used_percent",
+                Dimensions=[{"Name": "InstanceId", "Value": instance_id}],
                 StartTime=start_time,
                 EndTime=end_time,
                 Period=3600,
-                Statistics=['Average']
+                Statistics=["Average"],
             )
 
-            if response['Datapoints']:
-                memory_avg = sum(dp['Average'] for dp in response['Datapoints']) / len(response['Datapoints'])
+            if response["Datapoints"]:
+                memory_avg = sum(dp["Average"] for dp in response["Datapoints"]) / len(response["Datapoints"])
                 return memory_avg
             else:
                 return 0.0  # No memory metrics available
@@ -575,17 +575,17 @@ class CostOptimizer:
             start_time = end_time - timedelta(days=days)
 
             response = cloudwatch.get_metric_statistics(
-                Namespace='AWS/EC2',
-                MetricName='NetworkIn',
-                Dimensions=[{'Name': 'InstanceId', 'Value': instance_id}],
+                Namespace="AWS/EC2",
+                MetricName="NetworkIn",
+                Dimensions=[{"Name": "InstanceId", "Value": instance_id}],
                 StartTime=start_time,
                 EndTime=end_time,
                 Period=3600,
-                Statistics=['Average']
+                Statistics=["Average"],
             )
 
-            if response['Datapoints']:
-                network_avg = sum(dp['Average'] for dp in response['Datapoints']) / len(response['Datapoints'])
+            if response["Datapoints"]:
+                network_avg = sum(dp["Average"] for dp in response["Datapoints"]) / len(response["Datapoints"])
                 return network_avg / 1024 / 1024  # Convert to MB
             else:
                 return 0.0
@@ -610,11 +610,7 @@ class CostOptimizer:
             memory_avg = self._get_memory_utilization(cloudwatch, instance_id)
             network_avg = self._get_network_utilization(cloudwatch, instance_id)
 
-            return {
-                "cpu_avg": cpu_avg,
-                "memory_avg": memory_avg,
-                "network_avg": network_avg
-            }
+            return {"cpu_avg": cpu_avg, "memory_avg": memory_avg, "network_avg": network_avg}
         except Exception as e:
             console.print(f"[red]❌ Error analyzing utilization for {instance_id}: {e}[/red]")
             return {"cpu_avg": 0.0, "memory_avg": 0.0, "network_avg": 0.0}