Make os_type list_sysprep method return 1 list
authorNikos Skalkotos <skalkoto@grnet.gr>
Wed, 20 Jun 2012 12:17:01 +0000 (15:17 +0300)
committerNikos Skalkotos <skalkoto@grnet.gr>
Wed, 20 Jun 2012 12:17:01 +0000 (15:17 +0300)
The method used to return 2 lists of enabled and disabled sysprep
object. Since sysprep objs contail an `enabled' attribute, it is easy
to determine which of those are enabled and which are not. There is
no need to return 2 seperate lists.

image_creator/os_type/__init__.py

index 6fe1336..e1481d1 100644 (file)
@@ -90,10 +90,7 @@ class OSBase(object):
         objs = [getattr(self, name) for name in dir(self) \
             if not name.startswith('_')]
 
-        enabled = [x for x in objs if self._is_sysprep(x) and x.enabled]
-        disabled = [x for x in objs if self._is_sysprep(x) and not x.enabled]
-
-        return enabled, disabled
+        return [x for x in objs if self._is_sysprep(x)]
 
     def _sysprep_change_status(self, name, status):
 
@@ -123,7 +120,9 @@ class OSBase(object):
     def print_syspreps(self):
         """Print enabled and disabled system preperation operations."""
 
-        enabled, disabled = self.list_syspreps()
+        syspreps = self.list_syspreps()
+        enabled = filter(lambda x: x.enabled, syspreps)
+        disabled = filter(lambda x: not x.enabled, syspreps)
 
         wrapper = textwrap.TextWrapper()
         wrapper.subsequent_indent = '\t'