Changeset 1434 for trunk

Timestamp:

04/04/18 12:51:05 (6 years ago)

Author:

sdipsl

Message:

• add a function to handle very specific exception that affect LIM3 and dr2xml generated files.

File:

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

trunk/libIGCM/libIGCM_post/xios_parser.py

@@ -207 +207 @@
             traceback.print_exc(file=sys.stdout)
             return
- 
+
+def modifyDR2XML(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:
+        write=False
+        # 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='field'):
+            if elem.attrib.get('id') == 'CMIP6_siconc_Scaltypesi':
+                write=True
+                elem.set('freq_op', '2ts')
+                elem.set('freq_offset', '0ts')
+                if args.verbosity >= 2: print elem.tag, elem.attrib
+
+        if write:
+            # 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__':
 
@@ -243 +278 @@
         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 "modifyDR2XML" command
+        parser_check = subparsers.add_parser('modifyDR2XML', help='Fine tune dr2xml files for CMIP6 production')
+        parser_check.add_argument('--file', nargs='+', required=True, help='XIOS xml dr2xml_file type')
+        parser_check.set_defaults(func=modifyDR2XML)
         
         # create the parser for the "showtime" command