[commit] r2660 - trunk/GME/Parser

[GMESRC Repository Notifications]

Author: ksmyth
Date: Fri Jan 27 09:11:04 2017
New Revision: 2660

Log:
xme_id2guid.py: add --sort option for better merging

Modified:
   trunk/GME/Parser/xme_id2guid.py

Modified: trunk/GME/Parser/xme_id2guid.py
==============================================================================
--- trunk/GME/Parser/xme_id2guid.py	Thu Jan 26 10:02:54 2017	(r2659)
+++ trunk/GME/Parser/xme_id2guid.py	Fri Jan 27 09:11:04 2017	(r2660)
@@ -2,7 +2,7 @@
 import os.path
 import sys
 
-def id2guid(filename, output_filename):
+def id2guid(filename, output_filename, sort_elements=False):
     from xml.etree import ElementTree
     xme = ElementTree.parse(filename)
     id_guid_map = {}
@@ -18,18 +18,34 @@
             element.attrib['referred'] = id_guid_map[element.get('referred')]
     for element in xme.iter('set'):
         if element.get('members'):
-            element.attrib['members'] = " ".join([id_guid_map[id] for id in element.get('members').split()])
+            element.attrib['members'] = " ".join(sorted([id_guid_map[id] for id in element.get('members').split()]))
     for element in xme.iter('connpoint'):
         if element.get('target'):
             element.attrib['target'] = id_guid_map[element.get('target')]
         if element.get('refs'):
             element.attrib['refs'] = " ".join([id_guid_map[id] for id in element.get('refs').split()])
-    
+
+    if sort_elements:
+        def sortchildrenby(parent, attr):
+            parent[:] = sorted(parent, key=lambda child: child.get(attr))
+
+        for element in list(xme.iter()):
+            sortchildrenby(element, 'name')
+            sortchildrenby(element, 'guid')
+
     with open(output_filename, 'wb') as output:
         output.write('<!DOCTYPE project SYSTEM "mga2.dtd">\n')
         xme.write(output)
 
 if __name__ == '__main__':
-    input = sys.argv[1]
+    import argparse
+
+    parser = argparse.ArgumentParser(description='Convert GME id-style .xme (mga.dtd) to GUID-style .xme (mga2.dtd).')
+    parser.add_argument('--sort', action='store_true')
+    parser.add_argument('input-xme-file')
+    # parser.add_argument('--output-file')
+
+    args = parser.parse_args()
+    input = vars(args)['input-xme-file']
     output = os.path.splitext(input)[0] + "_guids.xme"
-    id2guid(input, output)
+    id2guid(input, output, sort_elements=args.sort)