Merge pull request #468 from xcp-ng/ohu/update-tools/snapshot-3

tools/update - create snapshot for nested hosts

Author: olivierh-pro

lib/tools/tasks/snapshot.py

@@ -0,0 +1,28 @@
+"""Snapshot tasks.
+"""
+from concurrent.futures import ThreadPoolExecutor
+from datetime import date
+
+from lib.host import Host
+from lib.vm import VM
+
+from .. import logger
+
+def create_snapshots(host: Host, vm_uuids: list[str]):
+    """Create a snapshot for a list of VMs in host.
+
+    :param host:
+        The target hosting pool which hosts the VMs.
+    :param vm_uuids:
+        uuids of target VMs.
+    """
+    # init VMs list
+    vms = [VM(uuid, host) for uuid in vm_uuids]
+
+    snapshot_name = f"utd-{date.today().strftime('%Y%m%d')}"
+
+    logger.debug(f"[{host}] Create snapshot '{snapshot_name}' for VMs: {vm_uuids}")
+
+    with ThreadPoolExecutor() as executor:
+        for vm in vms:
+            executor.submit(vm.snapshot, None, snapshot_name)

lib/tools/tasks/update.py

@@ -6,8 +6,10 @@
 
 from concurrent.futures import ThreadPoolExecutor
 
+from lib.host import Host
 from lib.pool import NotAMasterHostError, Pool
 from lib.tools.inventory import Inventory
+from lib.tools.tasks.snapshot import create_snapshots
 
 from .. import logger
 
@@ -27,10 +29,18 @@ def update_pools(inventory: Inventory) -> None:
     inventory_hosts = inventory["hosts"]
     # init related pools
     pools: list[Pool] = []
+    nested_hosts: dict[str, list[Host]] = {}
     for host in inventory_hosts:
         try:
             p = Pool(host)
             pools.append(p)
+            hosting_pool = inventory_hosts[host]["hosting_pool"]
+            if hosting_pool is not None:
+                # we assume all hosts are nested, not only master
+                if nested_hosts.get(hosting_pool) is not None:
+                    nested_hosts[hosting_pool].extend(p.hosts)
+                else:
+                    nested_hosts[hosting_pool] = p.hosts
         except NotAMasterHostError:
             logger.warning(f"[{host}] Skipping: not a master host")
 
@@ -47,3 +57,9 @@ def update_pools(inventory: Inventory) -> None:
                 # repos are the same as for the master host
                 repos = inventory_hosts[p.master.hostname_or_ip]["repositories"]
                 executor.submit(other_host.update, repos)
+
+    # Snapshot creation
+    for hosting_pool, nested in nested_hosts.items():
+        pool = Pool(hosting_pool) # mandatory for getting an host instance
+        vm_uuids = [h.get_system_uuid() for h in nested]
+        create_snapshots(pool.master, vm_uuids)