---
pykickstart/base.py | 11 +++--------
1 file changed, 3 insertions(+), 8 deletions(-)
diff --git a/pykickstart/base.py b/pykickstart/base.py
index f9cc161..14d407c 100644
--- a/pykickstart/base.py
+++ b/pykickstart/base.py
@@ -335,14 +335,7 @@ class BaseHandler(KickstartObject):
# register all commands
cmdObj = commandClass(commandName = commandName, data =
addons[addon_id])
self.commands[commandName] = cmdObj
-
- # Also, add the object into the _writeOrder dict in the right place.
- if cmdObj.writePriority is not None:
- if not self._writeOrder.has_key(cmdObj.writePriority):
- self._writeOrder[cmdObj.writePriority] = [cmdObj]
- else:
- self._writeOrder[cmdObj.writePriority].append(cmdObj)
-
+ self._addToWriteOrder(cmdObj)
# Prepare the final structures for 3rd party addons
self.addons = AddonRegistry(addons)
@@ -379,7 +372,9 @@ class BaseHandler(KickstartObject):
name = unicode(cmdObj.__class__.__name__).lower()
setattr(self, name.lower(), cmdObj)
+ self._addToWriteOrder(cmdObj)
+ def _addToWriteOrder(self, cmdObj):
# Also, add the object into the _writeOrder dict in the right place.
if cmdObj.writePriority is not None:
if self._writeOrder.has_key(cmdObj.writePriority):
--
1.7.11.7