0.8.1 vm_ha_setup

Author: lexxai

configs/scenarios_configs.yaml

@@ -17,17 +17,20 @@ Scenarios:
       tags: ["tag1", "dot-{vm_dot_ip}","ip-{vm_ip}"]
       replications:
         - node: "c04"
-          schedule: "*/30"
-          rate: 50.0
-          comment: "Replication to c04"
-          disable: False
-        - node: "c05"
-        - node: "c07"
-          schedule: "*/30"
+#          schedule: "*/30"
+#          rate: 50.0
+#          comment: "Replication to c04"
+#          disable: False
+#        - node: "c05"
+#        - node: "c07"
+#          schedule: "*/30"
       ha:
         group: "gr-02f-04-05-07"
         nodes: "c02:100,c04,c05,c07"
         overwrite: True
+        resource:
+          create: True
+          overwrite: True
 
 
 #  CloneTemplateVM-3:

src/cluster_tasks/scenarios/clone_template_vm_async.py

@@ -217,4 +217,23 @@ async def vm_ha_setup(self, proxmox_tasks):
             if nodes:
                 overwrite = self.ha.get("overwrite", False)
                 logger.info(f"HA Group '{group}' creating with nodes '{nodes}'")
-                await proxmox_tasks.ha_group_create(group, nodes, overwrite=overwrite)
+                result = await proxmox_tasks.ha_group_create(
+                    group, nodes, overwrite=overwrite
+                )
+                if not result:
+                    raise Exception(
+                        f"Failed to create HA Group for VM {self.destination_vm_id}"
+                    )
+            resource = self.ha.get("resource", {})
+            if resource.get("create", False):
+                overwrite = resource.get("overwrite", False)
+                logger.info(
+                    f"HA Resource for '{self.destination_vm_id}' with group '{group}' creating ..."
+                )
+                result = await proxmox_tasks.ha_resources_create(
+                    self.destination_vm_id, group, overwrite=overwrite
+                )
+                if not result:
+                    raise Exception(
+                        f"Failed to create HA Resource for VM {self.destination_vm_id}"
+                    )

src/cluster_tasks/scenarios/clone_template_vm_sync.py

@@ -203,4 +203,23 @@ def vm_ha_setup(self, proxmox_tasks):
             if nodes:
                 overwrite = self.ha.get("overwrite", False)
                 logger.info(f"HA Group '{group}' creating with nodes '{nodes}'")
-                proxmox_tasks.ha_group_create(group, nodes, overwrite=overwrite)
+                result = proxmox_tasks.ha_group_create(
+                    group, nodes, overwrite=overwrite
+                )
+                if not result:
+                    raise Exception(
+                        f"Failed to create HA Group for VM {self.destination_vm_id}"
+                    )
+            resource = self.ha.get("resource", {})
+            if resource.get("create", False):
+                overwrite = resource.get("overwrite", False)
+                logger.info(
+                    f"HA Resource for '{self.destination_vm_id}' with group '{group}' creating ..."
+                )
+                result = proxmox_tasks.ha_resources_create(
+                    self.destination_vm_id, group, overwrite=overwrite
+                )
+                if not result:
+                    raise Exception(
+                        f"Failed to create HA Resource for VM {self.destination_vm_id}"
+                    )

src/cluster_tasks/tasks/proxmox_tasks_sync.py

@@ -402,9 +402,7 @@ def ha_resources_create(
                 return result.get("success") if result else False
         data["sid"] = sid
         logger.info(f"VM {vid_id} creating resource ...")
-        result = self.api.cluster.ha.resources.post(
-            data=data, filter_keys="_raw_"
-        )
+        result = self.api.cluster.ha.resources.post(data=data, filter_keys="_raw_")
         return result.get("success") if result else False
 
     def ha_resources_delete(self, vid_id: int, type_resource: str = "vm"):