tools/update: use inventory instead of flat args

Flat args in `update_all` have been replaced with
a new inventory object.

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

Author: olivierh-pro

lib/tools/cli.py

@@ -8,10 +8,16 @@
 
 from lib.common import HostAddress
 from lib.tools import logger
+from lib.tools.inventory import into_inventory, load_inventory
 from lib.tools.tasks.update import update_all
 
 def _command_update(args):
-    update_all(args.hosts, args.repos)
+    if args.inventory:
+        inventory = load_inventory(args.inventory)
+    else:
+        inventory = into_inventory(args.hosts, args.repos)
+
+    update_all(inventory)
 
 
 def cli():

lib/tools/tasks/update.py

@@ -4,31 +4,28 @@
 """
 from concurrent.futures import ThreadPoolExecutor
 
-from lib.common import HostAddress
 from lib.host import Host
 from lib.pool import Pool
 
 from .. import logger
 
-def update_all(master_hosts: list[HostAddress], enablerepos: list[str]) -> None:
+def update_all(inventory: dict) -> None:
     """Updates all master (primary) hosts.
 
     .. note:: Host must be a master
 
         Throws error if hosts are not master (primary).
 
-    :param :py:class:`list[lib.common.HostAddress]` master_hosts:
-        A list of hosts to update.
-    :param list[str] enablerepos:
-        Repositories to enable when updating.
+    :param dict inventory:
+        Each host (key) holds its own config data (values).
     """
-    logger.debug(f"[{master_hosts}] enablerepos: {enablerepos}")
+    logger.debug(f"Inventory: {inventory}")
     # init related pools
-    pools = [Pool(h) for h in master_hosts]
+    pools = [Pool(h) for h in inventory]
 
     with ThreadPoolExecutor() as executor:
         for p in pools:
-            executor.submit(update_host, p.master, enablerepos)
+            executor.submit(update_host, p.master, inventory[p.master.hostname_or_ip]["enablerepos"])
 
 
 def update_host(host: Host, enablerepos: list[str] = []):