hestia-earth-utils 0.16.17__tar.gz → 0.16.19__tar.gz

{hestia_earth_utils-0.16.17 → hestia_earth_utils-0.16.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hestia_earth_utils
-Version: 0.16.17
+Version: 0.16.19
 Summary: HESTIA's utils library
 Home-page: https://gitlab.com/hestia-earth/hestia-utils
 Author: HESTIA Team

{hestia_earth_utils-0.16.17 → hestia_earth_utils-0.16.19}/hestia_earth/utils/pipeline.py

@@ -1,3 +1,4 @@
+import os
 from os.path import join
 import json
 import numpy as np
@@ -375,3 +376,29 @@ def get_related_nodes_data(
         }
         for node in related_nodes
     ]
+
+
+def start_instances(multiple_instances: bool = False):
+    import boto3
+
+    client = boto3.client("ec2")
+    instance_ids = os.getenv("START_INSTANCE_ID").split(",")
+    ids = instance_ids if multiple_instances else [instance_ids[0]]
+    try:
+        for instance_id in ids:
+            response = client.describe_instances(InstanceIds=[instance_id])
+            state = response["Reservations"][0]["Instances"][0]["State"]["Name"]
+            print(f"Checking {instance_id}: Current state is '{state}'")
+
+            if state in ["pending", "running"]:
+                print(f" -> Instance {instance_id} is already active. Skipping.")
+                continue
+
+            elif state == "stopped":
+                print(f" -> Instance {instance_id} is stopped. Starting it now...")
+                client.start_instances(InstanceIds=[instance_id])
+                break
+    except client.exceptions.ClientError as e:
+        print(f"Error starting instances: {str(e)}")
+
+    print("Could not start any instance")

hestia_earth_utils-0.16.19/hestia_earth/utils/version.py

@@ -0,0 +1 @@
+VERSION = "0.16.19"

{hestia_earth_utils-0.16.17 → hestia_earth_utils-0.16.19}/hestia_earth_utils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hestia_earth_utils
-Version: 0.16.17
+Version: 0.16.19
 Summary: HESTIA's utils library
 Home-page: https://gitlab.com/hestia-earth/hestia-utils
 Author: HESTIA Team

{hestia_earth_utils-0.16.17 → hestia_earth_utils-0.16.19}/tests/test_calculation_status.py

@@ -13,18 +13,18 @@ def test_emissions_with_status():
 
     result = _emissions_with_status(nodes[0])
     assert result == {
-        "emissions-total": 202,
+        "emissions-total": 192,
         "emissions-complete": 55,
         "emissions-incomplete": 1,
-        "emissions-missing": 146,
+        "emissions-missing": 136,
         "emissions": result["emissions"],  # ignore
     }
 
     result = _emissions_with_status(nodes[1])
     assert result == {
-        "emissions-total": 202,
+        "emissions-total": 192,
         "emissions-complete": 0,
         "emissions-incomplete": 13,
-        "emissions-missing": 189,
+        "emissions-missing": 179,
         "emissions": result["emissions"],  # ignore
     }

hestia_earth_utils-0.16.17/hestia_earth/utils/version.py

@@ -1 +0,0 @@
-VERSION = "0.16.17"