Changeset 1420 for trunk/libIGCM/libIGCM_post/xios_parser.py

Timestamp:

01/22/18 16:46:08 (6 years ago)

Author:

sdipsl

Message:

• optimize the way we drive where dr2xml_${compname}.xml write files

File:

• trunk/libIGCM/libIGCM_post/xios_parser.py (modified) (2 diffs)

trunk/libIGCM/libIGCM_post/xios_parser.py

@@ -175 +175 @@
             print s1
         print '\n'
-    
+   
+def modifyPath(args):
+    """Recursively find tree element with file tag and change the path prefix of the name attribute."""
+
+    # Loop over file_def files
+    for inputFile in args.file:
+        # Read the file_def_xml
+        if args.verbosity >= 1: print '\nReading dr2xml_def_xml=',inputFile
+        try:
+            tree = ET.parse(inputFile)
+        except:
+            print "Parse error. Please fix so that it can be parsed."
+            traceback.print_exc(file=sys.stdout)
+            return
+        root=tree.getroot()
+        if args.verbosity >= 3: print root.tag, root.attrib, '\n'
+
+        # Change file name prefix to point where we want
+        for elem in tree.iter(tag='file'):
+            name=elem.attrib.get('name')
+            elem.set('name', os.path.normpath(args.newPath[0]) + '/' + os.path.normpath(name))
+            if args.verbosity >= 2: print elem.tag, elem.attrib
+
+        # Write out the results
+        if args.verbosity >= 1: print '\nWriting dr2xml_def_xml=','modified.' + inputFile
+        try:
+            tree.write ('modified.' + inputFile)
+        except:
+            print "Write error. Please fix so that it can be parsed."
+            traceback.print_exc(file=sys.stdout)
+            return
+ 
 if __name__ == '__main__':
 
@@ -206 +237 @@
         parser_check.set_defaults(func=check, correction=True)
 
-        # create the parser for the "modify" command
+        # create the parser for the "modifyPath" command
+        parser_check = subparsers.add_parser('modifyPath', help='Will make sure dr2xml write files where it should be')
+        parser_check.add_argument('--newPath', nargs=1, required=True, help='Directory output for file produced from dr2xml_file type')
+        parser_check.add_argument('--file', nargs='+', required=True, help='XIOS xml dr2xml_file type')
+        parser_check.set_defaults(func=modifyPath)
+        
+        # create the parser for the "showtime" command
         parser_check = subparsers.add_parser('showtime', help='Just want to make sure you feel good today')
         parser_check.set_defaults(func=showtime)