[Kimchi-devel] [PATCH][Kimchi 1/6] Add function get_bootorder_node

[Ramon Medeiros]

This function retrieves ElementTree nodes instead of xml

Signed-off-by: Ramon Medeiros <ramonn at linux.vnet.ibm.com>
---
 xmlutils/bootorder.py | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/xmlutils/bootorder.py b/xmlutils/bootorder.py
index 9e444bd..6d92bd6 100644
--- a/xmlutils/bootorder.py
+++ b/xmlutils/bootorder.py
@@ -21,6 +21,17 @@ import lxml.etree as ET
 from lxml.builder import E
 
 
+def get_bootorder_node(boot_order=None):
+    if boot_order is None:
+        boot_order = ['hd', 'cdrom', 'network']
+
+    boot_nodes = []
+    for device in boot_order:
+        boot_nodes.append(E.boot(dev=device))
+
+    return boot_nodes
+
+
 def get_bootorder_xml(boot_order=None):
     """
     Returns the XML for boot order. The default return includes the following:
@@ -31,12 +42,8 @@ def get_bootorder_xml(boot_order=None):
 
     To a different boot order, specify the order by a list as argument.
     """
-    if boot_order is None:
-        boot_order = ['hd', 'cdrom', 'network']
-
     boot_xml = ''
-    for device in boot_order:
-        boot = E.boot(dev=device)
-        boot_xml += ET.tostring(boot, encoding='utf-8', pretty_print=True)
+    for device in get_bootorder_node(boot_order):
+        boot_xml += ET.tostring(device, encoding='utf-8', pretty_print=True)
 
     return boot_xml
-- 
2.5.5