oc-cdtapi 3.21.8__py3-none-any.whl → 3.23.4__py3-none-any.whl

oc_cdtapi/ForemanAPI.py

@@ -2,6 +2,8 @@ import json
 import logging
 import posixpath
 import re
+import time
+from time import sleep
 
 from oc_cdtapi.API import HttpAPI, HttpAPIError
 from collections import namedtuple, defaultdict
@@ -1177,6 +1179,25 @@ class ForemanAPI(HttpAPI):
         logging.debug('Passing to get_host_uuid_v1')
         return self.get_host_uuid_v1(hostname)
 
+    def get_host_disk_size(self, hostname):
+        """
+        :param hostname: str
+        :return: int
+        """
+        logging.debug('Reached get_host_disk_size')
+        response = self.get(posixpath.join("hosts", hostname, "vm_compute_attributes"))
+        data = response.json()
+        try:
+            volumes = data.get("volumes_attributes", {})
+            volume = volumes.get("0") or volumes.get(0)
+            if volume and "size_gb" in volume:
+                return int(volume["size_gb"])
+            logging.error('Could not find size_gb in vm_compute_attributes for [%s]' % hostname)
+            return None
+        except (KeyError, ValueError, TypeError) as e:
+            logging.error('Error parsing disk size for [%s]: %s' % (hostname, e))
+            return None
+
     def get_all_users(self):
         """
         :return: list
@@ -1225,3 +1246,97 @@ class ForemanAPI(HttpAPI):
         pl['host'] = {}
         pl['host']['owner_id'] = owner_id
         self.update_host(hostname, pl)
+
+    def get_job_template_id(self, template_name):
+        """
+        Get template ID by name. Cache results to avoid repeated API calls.
+        :param template_name: str
+        :return template_id: int
+        """
+        logging.debug('Reached get_job_template_id')
+        if not hasattr(self, '_template_cache'):
+            self._template_cache = {}
+
+        if template_name in self._template_cache:
+            return self._template_cache[template_name]
+
+        response = self.get(posixpath.join("job_templates"), params={'per_page': 'all'})
+        templates = response.json()["results"]
+
+        for job in templates:
+            self._template_cache[job["name"]] = job["id"]
+
+        template_id = self._template_cache.get(template_name)
+        if not template_id:
+            raise ForemanAPIError(f"Job template '{template_name}' not found")
+
+        logging.debug(f"Template id for [{template_name}] is [{template_id}]")
+        return template_id
+
+    def send_job_invocation(self, task_name, vm_name, **inputs):
+        """
+        Send a job invocation for a specific task and VM.
+        :param task_name: str
+        :param vm_name: str
+        :param **inputs: kwargs
+        :return job_id
+        """
+        logging.debug('Reached send_job_invocation')
+        task_configs = {
+            "resize_partition": {
+                "template_name": "Run \"cdt-resize-partition\" role CDT",
+                "description": "Partition Resizing"
+            }
+        }
+
+        config = task_configs.get(task_name)
+        if not config:
+            raise ForemanAPIError(f"Unknown task: {task_name}")
+
+        logging.debug(f"config for [{task_name}] is [{config}]")
+
+        template_id = self.get_job_template_id(config["template_name"])
+
+        payload = {
+            "job_invocation": {
+                "job_template_id": template_id,
+                "targeting_type": "static_query",
+                "search_query": f'name = "{vm_name}"',
+                "description_format": config["description"],
+                "inputs": inputs,
+                "ansible": {
+                    "tags": "",
+                    "tags_flag": "include"
+                }
+            }
+        }
+
+        logging.debug(f"About to send job invocations with following request [{payload}]")
+        response = self.post(
+            posixpath.join("job_invocations"),
+            headers=self.headers,
+            json=payload
+        )
+
+        return response.json()["id"]
+
+    def is_job_invocation_success(self, job_id, timeout=300, poll_interval=5):
+        """
+        Get a job status by given id.
+        :param job_id: str
+        :param timeout: int
+        :param poll_interval: int
+        :return status: bool
+        """
+        start_time = time.time()
+        while True:
+            if time.time() - start_time > timeout:
+                raise ForemanAPIError(f"Job {job_id} timed out after {timeout} seconds")
+            response = self.get(posixpath.join("job_invocations", str(job_id)))
+            is_pending = bool(response.json().get("pending", False))
+            if not is_pending:
+                break
+            logging.debug(f"job still pending, sleeping for {poll_interval} second")
+            sleep(poll_interval)
+
+        return bool(response.json().get("succeeded", False))

{oc_cdtapi-3.21.8.dist-info → oc_cdtapi-3.23.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oc-cdtapi
-Version: 3.21.8
+Version: 3.23.4
 Summary: Custom Development python API libraries
 License: Apache2.0
 Requires-Python: >=3.6

{oc_cdtapi-3.21.8.dist-info → oc_cdtapi-3.23.4.dist-info}/RECORD

@@ -3,7 +3,7 @@ oc_cdtapi/Dbsm2API.py,sha256=WDBst1dCAmDVU9d9Ogzwp2nkjtKZ4thU2xG4sAPI_Nk,9963
 oc_cdtapi/DevPIAPI.py,sha256=9WND9ld66eHmC5qoLJq3KDYSoO-pP69UqOQsGZNLZYg,1835
 oc_cdtapi/DmsAPI.py,sha256=eNFdwQLhCbPvHB5SUtP4QcZZtSdjkgt_Cxn3oQ3iJ5s,15605
 oc_cdtapi/DmsGetverAPI.py,sha256=ZPU4HlF59fngKu5mSFhtss3rlBuduffDOSm_y3XWrxk,15556
-oc_cdtapi/ForemanAPI.py,sha256=9r1MSOsubiSxM1sdzF2MHD7IcSwX9Y4UDJknHqQK07g,44255
+oc_cdtapi/ForemanAPI.py,sha256=oL_OdUgRL_7z9te83lmLarInDaKWW1_rjwXaWTZajIY,48401
 oc_cdtapi/JenkinsAPI.py,sha256=lZ8pe3a4eb_6h53JE7QLuzOSlu7Sqatc9PQwWhio9Vg,15748
 oc_cdtapi/NexusAPI.py,sha256=uU12GtHvKlWorFaPAnFcQ5AGEc94MZ5SdmfM2Pw3F7A,26122
 oc_cdtapi/PgAPI.py,sha256=URSz7qu-Ir7AOj0jI3ucTXn2PM-nC96nmPZI746OLjA,14356
@@ -12,9 +12,9 @@ oc_cdtapi/RundeckAPI.py,sha256=O3LmcFaHSz8UqeUyIHTTEMJncDD191Utd-iZaeJay2s,24243
 oc_cdtapi/TestServer.py,sha256=HV97UWg2IK4gOYAp9yaMdwFUWsw9v66MxyZdI3qQctA,2715
 oc_cdtapi/VaultAPI.py,sha256=P-x_PsWe_S0mGUKTCmR1KhUjdfs7GmyaltjGQcnWj_s,2967
 oc_cdtapi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-oc_cdtapi-3.21.8.data/scripts/nexus.py,sha256=4teqZ_KtCSrwHDJVgA7lkreteod4Xt5XJFZNbwb7E6E,6858
-oc_cdtapi-3.21.8.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-oc_cdtapi-3.21.8.dist-info/METADATA,sha256=dZ_OtFV8oU_jNs249Mj4KMEUkAjN2hrPxlkl9wjcD_c,504
-oc_cdtapi-3.21.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-oc_cdtapi-3.21.8.dist-info/top_level.txt,sha256=d4-5-D-0CSeSXYuLCP7-nIFCpjkfmJr-Y_muzds8iVU,10
-oc_cdtapi-3.21.8.dist-info/RECORD,,
+oc_cdtapi-3.23.4.data/scripts/nexus.py,sha256=4teqZ_KtCSrwHDJVgA7lkreteod4Xt5XJFZNbwb7E6E,6858
+oc_cdtapi-3.23.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+oc_cdtapi-3.23.4.dist-info/METADATA,sha256=vm9AmaXHNBtw1DznVmDF9PLEfIDC1xn0bNrxE-mV0uU,504
+oc_cdtapi-3.23.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+oc_cdtapi-3.23.4.dist-info/top_level.txt,sha256=d4-5-D-0CSeSXYuLCP7-nIFCpjkfmJr-Y_muzds8iVU,10
+oc_cdtapi-3.23.4.dist-info/RECORD,,