tools/update: move update_host into class host

Instead of an external function, update should
live inside Host class.

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

Author: olivierh-pro

lib/host.py

@@ -450,6 +450,23 @@ def yum_update(self, enablerepos: List[str] = []):
 
         return self.ssh(base_command)
 
+    def update(self, enablerepos: list[str] = []):
+        """Updates current host.
+
+        An helper function that wraps update tasks on current host.
+
+        :param list[str] enablerepos:
+            Repositories to enable when updating.
+        """
+        logging.info(f"[{self}] Updating...")
+
+        self.yum_clean_metadata()
+        self.yum_update(enablerepos=enablerepos)
+        # Everything's ok, just reboot
+        self.reboot(verify=True)
+
+        logging.info(f"[{self}] Updated successfully!")
+
     def restart_toolstack(self, verify=False):
         logging.info("Restart toolstack on host %s" % self)
         self.ssh(['xe-toolstack-restart'])

lib/tools/tasks/update.py

@@ -4,7 +4,6 @@
 """
 from concurrent.futures import ThreadPoolExecutor
 
-from lib.host import Host
 from lib.pool import Pool
 
 from .. import logger
@@ -25,24 +24,4 @@ def update_all(inventory: dict) -> None:
 
     with ThreadPoolExecutor() as executor:
         for p in pools:
-            executor.submit(update_host, p.master, inventory[p.master.hostname_or_ip]["enablerepos"])
-
-
-def update_host(host: Host, enablerepos: list[str] = []):
-    """Updates the target host.
-
-    An helper function that wraps update tasks on specific host.
-
-    :param :py:class:`lib.host.Host` host:
-        Target host to update.
-    :param list[str] enablerepos:
-        Repositories to enable when updating.
-    """
-    logger.info(f"[{host}] Updating...")
-
-    host.yum_clean_metadata()
-    host.yum_update(enablerepos=enablerepos)
-    # Everything's ok, just reboot
-    host.reboot(verify=True)
-
-    logger.info(f"[{host}] Updated successfully!")
+            executor.submit(p.master.update, inventory[p.master.hostname_or_ip]["enablerepos"])