Adds Red Hat (yum) support with Wrapper class

Author: DjDCH

updatewrapper/__init__.py

@@ -2,11 +2,10 @@
 import os
 import sys
 
-from ansible.module_utils._text import to_bytes
-
 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
 
 
 def wrap(hosts, out_dir):
@@ -22,22 +21,23 @@ def wrap(hosts, out_dir):
     print()
     for host in hosts:
         try:
+            wrapper = Wrapper(host)
+
             print_info('BEGIN host %s' % host.addr)
             host.ask_passwords()
 
-            print_success('Updating APT index files')
-            host.run('apt-get update')
+            print_success('Updating index cache')
+            wrapper.update_cache()
 
             print_success('Listing available package upgrades')
-            returncode, stdout, stderr = host.run('apt-get --show-upgraded --assume-no upgrade')
+            returns = wrapper.check_update()
 
-            if to_bytes('The following packages will be upgraded') in stdout:
+            if wrapper.has_update(*returns):
                 print_warning('Some packages need to be upgraded')
 
                 if ask_yes_no('Do you want to continue?'):
-
                     print_success('Installing available package upgrades')
-                    returncode, stdout, stderr = host.run('apt-get --show-upgraded --assume-yes upgrade')
+                    returncode, stdout, stderr = wrapper.perform_update()
 
                     logfile = get_logfile(host.name, out_dir)
                     save_output(logfile, stdout)

updatewrapper/host.py

@@ -19,13 +19,14 @@
 
 
 class Host:
-    def __init__(self, name='localhost', addr='localhost', port='22', user='root', sudo=False, sudo_pass=False):
+    def __init__(self, name='localhost', addr='localhost', port='22', user='root', sudo=False, sudo_pass=False, flavor='debian'):
         self.name = addr if name is 'localhost' else name
         self.addr = name if addr is 'localhost' else addr
         self.port = port
         self.user = user
         self.sudo = sudo or sudo_pass
         self.sudo_pass = sudo_pass
+        self.flavor = flavor
 
         self._passwords = None
 

updatewrapper/wrapper.py

@@ -0,0 +1,40 @@
+from ansible.module_utils._text import to_bytes
+
+
+class Wrapper:
+    def __init__(self, host):
+        self.host = host
+        self.flavor = host.flavor if host.flavor in ['debian', 'redhat'] else None
+
+        if self.flavor is None:
+            raise ValueError('Invalid flavor')
+
+    def is_debian(self):
+        return self.flavor == 'debian'
+
+    def is_redhat(self):
+        return self.flavor == 'redhat'
+
+    def update_cache(self):
+        if self.is_debian():
+            return self.host.run('apt-get update')
+        elif self.is_redhat():
+            return self.host.run('yum clean expire-cache')
+
+    def check_update(self):
+        if self.is_debian():
+            return self.host.run('apt-get --show-upgraded --assume-no upgrade')
+        elif self.is_redhat():
+            return self.host.run('yum check-update')
+
+    def has_update(self, returncode, stdout, stderr):
+        if self.is_debian():
+            return to_bytes('The following packages will be upgraded') in stdout
+        elif self.is_redhat():
+            return returncode == 100
+
+    def perform_update(self):
+        if self.is_debian():
+            return self.host.run('apt-get --show-upgraded --assume-yes upgrade')
+        elif self.is_redhat():
+            return self.host.run('yum --assumeyes update')