Merge pull request #665 from deNBI/refactor/metadata_write_token

refactor(Metadata):renamed metadata server token to metadata_write_to…
deNBI · Jan 6, 2025 · c18fd67 · c18fd67
2 parents f5443e1 + 78fff05
commit c18fd67
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/simple_vm_client/metadata_connector/metadata_connector.py b/simple_vm_client/metadata_connector/metadata_connector.py
@@ -50,15 +50,15 @@ def load_config_yml(self, config_file: str) -> None:
 
     def load_env_config(self):
         required_keys = [
-            "METADATA_SERVER_TOKEN",
+            "METADATA_WRITE_TOKEN",
         ]
         missing_keys = [key for key in required_keys if key not in os.environ]
         if missing_keys:
             missing_keys_str = ", ".join(missing_keys)
             logger.error(
                 f"MetadataServer missing keys {missing_keys_str} not provided in env!"
             )
-        self.METADATA_SERVER_TOKEN = os.environ.get("METADATA_SERVER_TOKEN")
+        self.METADATA_WRITE_TOKEN = os.environ.get("METADATA_WRITE_TOKEN")
         logger.info("Metadata Environment loaded")
 
     def remove_metadata(self, ip: str):
@@ -74,8 +74,8 @@ def remove_metadata(self, ip: str):
                 remove_metadata_url,
                 timeout=(30, 30),
                 headers={
-                    "X-Auth-Token": self.METADATA_SERVER_TOKEN,
-                }
+                    "X-Auth-Token": self.METADATA_WRITE_TOKEN,
+                },
             )
             response.raise_for_status()
             logger.info(f"Metadata removed successfully for {ip}")
@@ -102,9 +102,9 @@ def set_metadata(self, ip: str, metadata: VirtualMachineServerMetadata):
                 data=serialized_data,
                 timeout=(30, 30),
                 headers={
-                    "X-Auth-Token": self.METADATA_SERVER_TOKEN,
+                    "X-Auth-Token": self.METADATA_WRITE_TOKEN,
                     "Content-Type": "application/json",
-                }
+                },
             )
             response.raise_for_status()
             logger.info(f"Metadata set successfully for {ip}")
@@ -121,10 +121,7 @@ def is_metadata_server_available(self):
         health_url = urljoin(self.METADATA_BASE_URL, "health")
 
         try:
-            response = requests.get(
-                health_url,
-                timeout=(30, 30)
-            )
+            response = requests.get(health_url, timeout=(30, 30))
             response.raise_for_status()
             logger.info(f"Metadata Health Check --- {response.json()}")
             return True