Skip to content
Snippets Groups Projects
Commit b79455bf authored by Alvise de'Faveri's avatar Alvise de'Faveri
Browse files

Add correct formatting when writing on the XML.

parent 74c21196
Branches
No related tags found
No related merge requests found
...@@ -33,9 +33,10 @@ def craftXmlMsgField(msg_field_data): ...@@ -33,9 +33,10 @@ def craftXmlMsgField(msg_field_data):
msg_field.text=msg_field_data.description msg_field.text=msg_field_data.description
return msg_field return msg_field
def save_xml(tree):
def save_xml(root): xmlstr = prettify(tree.getroot())
root.write(FILENAME, encoding='utf8') with open(FILENAME, "w") as f:
f.write(xmlstr)
def deleteMsgFromXml(msg): def deleteMsgFromXml(msg):
tree = xml.parse(FILENAME) tree = xml.parse(FILENAME)
...@@ -55,7 +56,6 @@ def deleteEnumFromXml(enum): ...@@ -55,7 +56,6 @@ def deleteEnumFromXml(enum):
mav_enums.remove(mav_enum) mav_enums.remove(mav_enum)
save_xml(tree) save_xml(tree)
def addMsgToXml(msg): def addMsgToXml(msg):
tree = xml.parse(FILENAME) tree = xml.parse(FILENAME)
mav_root = tree.getroot() mav_root = tree.getroot()
...@@ -120,8 +120,9 @@ def prettify(elem): ...@@ -120,8 +120,9 @@ def prettify(elem):
"""Return a pretty-printed XML string for the Element. """Return a pretty-printed XML string for the Element.
""" """
rough_string = xml.tostring(elem, 'utf-8') rough_string = xml.tostring(elem, 'utf-8')
rough_string = rough_string.replace('\r', '').replace('\n', '')
reparsed = minidom.parseString(rough_string) reparsed = minidom.parseString(rough_string)
return reparsed.toprettyxml(indent="\t") return '\n'.join([line for line in reparsed.toprettyxml(indent=' ').split('\n') if line.strip()])
def get_attr_if_exist(node,att_name): def get_attr_if_exist(node,att_name):
if att_name in node.attrib: if att_name in node.attrib:
...@@ -147,7 +148,7 @@ def get_enum(): ...@@ -147,7 +148,7 @@ def get_enum():
entrylist = [] entrylist = []
xmlEnumEntry = elem.findall("entry") xmlEnumEntry = elem.findall("entry")
for entry in xmlEnumEntry: for entry in xmlEnumEntry:
#print prettify(entry) #-print prettify(entry)
name =entry.get("name") name =entry.get("name")
desc = get_text_if_exist(entry,"description") desc = get_text_if_exist(entry,"description")
value = entry.get("value") value = entry.get("value")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment