Adds abstract class to manage flavors

Author: DjDCH

updatewrapper/__init__.py

@@ -5,7 +5,7 @@
 from updatewrapper.host import Host
 from updatewrapper.utils.display import ask_yes_no, print_banner, print_info, print_notice, print_success, print_warning, spinner_list
 from updatewrapper.utils.file import get_config_file, get_hosts, get_logfile, save_output
-from updatewrapper.wrapper import Wrapper
+from updatewrapper.flavor import get_flavor_wrapper
 
 
 def wrap(hosts, out_dir):
@@ -21,7 +21,7 @@ def wrap(hosts, out_dir):
     print()
     for host in hosts:
         try:
-            wrapper = Wrapper(host)
+            wrapper = get_flavor_wrapper(host)
 
             print_info('BEGIN host %s' % host.addr)
             host.ask_passwords()

updatewrapper/flavor/__init__.py

@@ -0,0 +1,37 @@
+import abc
+import importlib
+
+
+def get_flavor_wrapper(host):
+    flavor = host.flavor if host.flavor in ['debian', 'redhat'] else None
+
+    if flavor is None:
+        raise ValueError('Invalid flavor')
+
+    Wrapper = getattr(importlib.import_module('updatewrapper.flavor.' + flavor), flavor.capitalize())
+
+    return Wrapper(host)
+
+
+class FlavorBase:
+    __metaclass__ = abc.ABCMeta
+
+    def __init__(self, host):
+        self.host = host
+
+    @abc.abstractmethod
+    def update_cache(self):
+        pass
+
+    @abc.abstractmethod
+    def check_update(self):
+        pass
+
+    @abc.abstractmethod
+    def has_update(self, returncode, stdout, stderr):
+        pass
+
+    @abc.abstractmethod
+    def perform_update(self):
+        pass
+

updatewrapper/flavor/debian.py

@@ -0,0 +1,17 @@
+from ansible.module_utils._text import to_bytes
+
+from updatewrapper.flavor import FlavorBase
+
+
+class Debian(FlavorBase):
+    def update_cache(self):
+        return self.host.run('apt-get update')
+
+    def check_update(self):
+        return self.host.run('apt-get --show-upgraded --assume-no upgrade')
+
+    def has_update(self, returncode, stdout, stderr):
+        return to_bytes('The following packages will be upgraded') in stdout
+
+    def perform_update(self):
+        return self.host.run('apt-get --show-upgraded --assume-yes upgrade')

updatewrapper/flavor/redhat.py

@@ -0,0 +1,15 @@
+from updatewrapper.flavor import FlavorBase
+
+
+class Redhat(FlavorBase):
+    def update_cache(self):
+        return self.host.run('yum clean expire-cache')
+
+    def check_update(self):
+        return self.host.run('yum check-update')
+
+    def has_update(self, returncode, stdout, stderr):
+        return returncode == 100
+
+    def perform_update(self):
+        return self.host.run('yum --assumeyes update')