convert helpers/init_host into a pure function

Author: igalic

libiocage/lib/Distribution.py

@@ -50,7 +50,7 @@ class DistributionGenerator:
     def __init__(self, host, zfs=None, logger=None):
         self.logger = libiocage.lib.helpers.init_logger(logger)
         self.zfs = libiocage.lib.helpers.init_zfs(zfs)
-        libiocage.lib.helpers.init_host(self, host)
+        self.host = libiocage.lib.helpers.init_host(host, logger=self.logger)
 
         self.available_releases = None
 

libiocage/lib/Jail.py

@@ -106,8 +106,7 @@ def __init__(self, data={}, zfs=None, host=None, logger=None, new=False):
 
         self.logger = libiocage.lib.helpers.init_logger(logger)
         self.zfs = libiocage.lib.helpers.init_zfs(zfs)
-
-        libiocage.lib.helpers.init_host(self, host)
+        self.host = libiocage.lib.helpers.init_host(host, logger=self.logger)
 
         if isinstance(data, str):
             data = {
@@ -942,8 +941,6 @@ def __dir__(self):
 
 class Jail(JailGenerator):
 
-    _class_host = libiocage.lib.Host.HostGenerator
-
     def start(self, *args, **kwargs):
         return list(JailGenerator.start(self, *args, **kwargs))
 

libiocage/lib/Jails.py

@@ -47,7 +47,7 @@ def __init__(self,
                  zfs=None):
 
         self.logger = libiocage.lib.helpers.init_logger(logger)
-        libiocage.lib.helpers.init_host(self, host)
+        self.host = libiocage.lib.helpers.init_host(host, logger=self.logger)
 
         # this could be wrong..er.. now?:
         self.zfs = libiocage.lib.helpers.init_zfs(zfs)

libiocage/lib/Prompts.py

@@ -27,8 +27,8 @@
 
 class Prompts:
     def __init__(self, host=None, logger=None):
-        self.logger = logger
-        libiocage.lib.helpers.init_host(self, host)
+        self.logger = libiocage.lib.helpers.init_logger(logger)
+        self.host = libiocage.lib.helpers.init_host(host, logger=self.logger)
 
     def release(self):
         default = None

libiocage/lib/Release.py

@@ -58,7 +58,7 @@ def __init__(self, name=None,
 
         self.logger = libiocage.lib.helpers.init_logger(logger)
         self.zfs = libiocage.lib.helpers.init_zfs(zfs)
-        libiocage.lib.helpers.init_host(self, host)
+        self.host = libiocage.lib.helpers.init_host(host, logger=self.logger)
 
         if not libiocage.lib.helpers.validate_name(name):
             raise NameError(f"Invalid 'name' for Release: '{name}'")

libiocage/lib/Releases.py

@@ -27,9 +27,9 @@
 
 class Releases:
     def __init__(self, host=None, zfs=None, logger=None):
-        libiocage.lib.helpers.init_host(self, host)
-        self.logger = logger
-        self.zfs = zfs
+        self.logger = libiocage.lib.helpers.init_logger(logger)
+        self.zfs = libiocage.lib.helpers.init_zfs(zfs)
+        self.host = libiocage.lib.helpers.init_host(host, logger=self.logger)
 
     @property
     def dataset(self):

libiocage/lib/helpers.py

@@ -42,19 +42,14 @@ def get_zfs():
     return libzfs.ZFS(history=True, history_prefix="<iocage>")
 
 
-def init_host(self, host=None):
+def init_host(
+        host: 'libiocage.lib.Host.Host'=None,
+        logger: 'libiocage.lib.Logger.Logger'=None
+) -> 'libiocage.lib.Host.HostGenerator':
     if host:
-        self.host = host
-    else:
-        try:
-            logger = self.logger
-        except:
-            logger = None
-
-        try:
-            self.host = self._class_host(logger=logger)
-        except:
-            self.host = libiocage.lib.Host.HostGenerator(logger=logger)
+        return host
+
+    return libiocage.lib.Host.HostGenerator(logger=logger)
 
 
 def init_datasets(self, datasets=None):