From 9cea2926659ff3ec06955c88cb975f52dd0fac0b Mon Sep 17 00:00:00 2001
From: Michael DeHaan <michael.dehaan@gmail.com>
Date: Sat, 23 Feb 2013 12:25:04 -0500
Subject: [PATCH] Clean up OS family code

---
 library/setup | 55 ++++++++++++---------------------------------------
 1 file changed, 13 insertions(+), 42 deletions(-)

diff --git a/library/setup b/library/setup
index cfcc353ac9..079f2ca9dd 100644
--- a/library/setup
+++ b/library/setup
@@ -131,44 +131,16 @@ class Facts(object):
     def get_distribution_facts(self):
 
         # A list with OS Family members
-        OSFAMILY = {
-                      'RedHat'       : 'RedHat',
-                      'Fedora'       : 'RedHat',
-                      'CentOS'       : 'RedHat',
-                      'Scientific'   : 'RedHat',
-                      'SLC'          : 'RedHat',
-                      'Ascendos'     : 'RedHat',
-                      'CloudLinux'   : 'RedHat',
-                      'PSBM'         : 'RedHat',
-                      'OracleLinux'  : 'RedHat',
-                      'OVS'          : 'RedHat',
-                      'OEL'          : 'RedHat',
-                      'Amazon'       : 'RedHat',
-                      'XenServer'    : 'RedHat',
-                      #
-                      'Ubuntu'       : 'Debian',
-                      'Debian'       : 'Debian',
-                      #
-                      'SLES'         : 'Suse',
-                      'SLED'         : 'Suse',
-                      'OpenSuSE'     : 'Suse',
-                      'SuSE'         : 'Suse',
-                      #
-                      'Gentoo'       : 'Gentoo',
-                      #
-                      'Archlinux'    : 'Archlinux',
-                      #
-                      'Mandriva'     : 'Mandrake',
-                      'Mandrake'     : 'Mandrake',
-              #
-                      'Solaris'      : 'Solaris',
-                      'Nexenta'      : 'Solaris',
-                      'OmniOS'       : 'Solaris',
-                      'OpenIndiana'  : 'Solaris',
-                      'SmartOS'      : 'Solaris',
-                      #
-                      'AIX'          : 'AIX'
-               }
+        OS_FAMILY = dict(
+            RedHat = 'RedHat', Fedora = 'RedHat', CentOS = 'RedHat', Scientific = 'RedHat',
+            SLC = 'RedHat', Ascendos = 'RedHat', CloudLinux = 'RedHat', PSBM = 'RedHat',
+            OracleLinux = 'RedHat', OVS = 'RedHat', OEL = 'RedHat', Amazon = 'RedHat',
+            XenServer = 'RedHat', Ubuntu = 'Debian', Debian = 'Debian', SLES = 'Suse',
+            SLED = 'Suse', OpenSuSE = 'Suse', SuSE = 'Suse', Gentoo = 'Gentoo',
+            Archlinux = 'Archlinux', Mandriva = 'Mandrake', Mandrake = 'Mandrake',
+            Solaris = 'Solaris', Nexenta = 'Solaris', OmniOS = 'Solaris', OpenIndiana = 'Solaris',
+            SmartOS = 'Solaris', AIX = 'AIX'
+        )
 
         if self.facts['system'] == 'AIX':
             self.facts['distribution'] = 'AIX'
@@ -200,10 +172,9 @@ class Facts(object):
                     else:
                         self.facts['distribution'] = name
 
-        # Find the family
-    self.facts['osfamily'] = self.facts['distribution']
-    if self.facts['distribution'] in OSFAMILY:
-       self.facts['osfamily'] = OSFAMILY[self.facts['distribution']]
+        self.facts['os_family'] = self.facts['distribution']
+        if self.facts['distribution'] in OS_FAMILY:
+           self.facts['os_family'] = OS_FAMILY[self.facts['distribution']]
 
     def get_cmdline(self):
         data = get_file_content('/proc/cmdline')