tools/update: create a snapshot after updating hosts

Signed-off-by: Olivier Hoareau <olivier.hoareau@vates.tech>

Author: olivierh-pro

lib/tools/tasks/update.py

@@ -6,7 +6,9 @@
 
 from concurrent.futures import ThreadPoolExecutor
 
+from lib.host import Host
 from lib.pool import NotAMasterHostError, Pool
+from lib.tools.tasks.snapshot import create_snapshot
 
 from .. import logger
 
@@ -26,7 +28,7 @@ def update_pools(inventory: dict) -> None:
     inventory_hosts = inventory["hosts"]
     # init related pools
     pools = []
-    nested_hosts = []
+    nested_hosts: list[Host] = []
     for h in inventory_hosts:
         try:
             p = Pool(h)
@@ -49,3 +51,9 @@ def update_pools(inventory: dict) -> None:
                 # repos are the same as the primary (master)
                 repos = inventory_hosts[p.master.hostname_or_ip]["enablerepos"]
                 executor.submit(h.update, repos)
+
+    # Snapshot creation
+    # get ids of VMs for parent host
+    vm_uuids = [h.get_system_uuid() for h in nested_hosts]
+    parent_pool = Pool(inventory["parent"]) # mandatory for getting an host instance
+    create_snapshot(parent_pool.master, vm_uuids)