tools/inventory: move hosts into subkey hosts

To prepare the snapshot creation and the new data
in inventory at root level, every host must be
moved into a subkey 'hosts'

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

Author: olivierh-pro

lib/tools/inventory.py

@@ -13,36 +13,41 @@ class HostConfig(TypedDict):
     repositories: list[str]
 
 
-Inventory: TypeAlias = dict[HostAddress, HostConfig]
+HostConfigs: TypeAlias = dict[HostAddress, HostConfig]
+
+class Inventory(TypedDict):
+    hosts: HostConfigs
 
 
 def load_inventory(inventory_path: Path) -> Inventory:
     """Create an inventory object from loaded inventory file."""
-    inventory: Inventory = {}
-
     with open(inventory_path, "rb") as f:
         data = tomllib.load(f)
 
     all = data.get("all", {})
     hosts = data.get("hosts", [])
 
-    for server, config in hosts.items():
+    inventory_hosts: HostConfigs = {}
+    for h, config in hosts.items():
         repos = config.get("repositories", [])
         host: HostConfig = {"repositories": repos or all.get("repositories", [])}
-        inventory[server] = host
+        inventory_hosts[h] = host
 
-    return inventory
+    return {
+        "hosts": inventory_hosts,
+    }
 
 
 def into_inventory(hosts: list[HostAddress], repositories: list[str]) -> Inventory:
     """Create an inventory object from arguments.
 
     Basically, it is used as compatibility when we don't want inventory from file.
     """
-    inventory: Inventory = {}
-
+    inventory_hosts: HostConfigs = {}
     for h in hosts:
         host: HostConfig = {"repositories": repositories or []}
-        inventory[h] = host
+        inventory_hosts[h] = host
 
-    return inventory
+    return {
+        "hosts": inventory_hosts,
+    }

lib/tools/tasks/update.py

@@ -7,10 +7,11 @@
 from concurrent.futures import ThreadPoolExecutor
 
 from lib.pool import NotAMasterHostError, Pool
+from lib.tools.inventory import Inventory
 
 from .. import logger
 
-def update_pools(inventory: dict) -> None:
+def update_pools(inventory: Inventory) -> None:
     """Updates hosts in pool(s).
 
     .. note::
@@ -23,24 +24,25 @@ def update_pools(inventory: dict) -> None:
         Each host (key) holds its own config data (values, eg: `enablerepos`).
     """
     logger.debug(f"Inventory: {inventory}")
+    inventory_hosts = inventory["hosts"]
     # init related pools
-    pools = []
-    for h in inventory:
+    pools: list[Pool] = []
+    for host in inventory_hosts:
         try:
-            p = Pool(h)
+            p = Pool(host)
             pools.append(p)
         except NotAMasterHostError:
-            logger.warning(f"[{h}] Skipping: not a master host")
+            logger.warning(f"[{host}] Skipping: not a master host")
 
     with ThreadPoolExecutor() as executor:
         for p in pools:
-            executor.submit(p.master.update, inventory[p.master.hostname_or_ip]["enablerepos"])
+            executor.submit(p.master.update, inventory_hosts[p.master.hostname_or_ip]["enablerepos"])
 
     # secondary hosts
     with ThreadPoolExecutor() as executor:
         for p in pools:
             # omit first item because it is a primary (master)
-            for h in p.hosts[1:]:
+            for secondary in p.hosts[1:]:
                 # repos are the same as the primary (master)
-                repos = inventory[p.master.hostname_or_ip]["enablerepos"]
-                executor.submit(h.update, repos)
+                repos = inventory_hosts[p.master.hostname_or_ip]["enablerepos"]
+                executor.submit(secondary.update, repos)