Changeset 4137

Timestamp:

2013-10-25T16:06:26+02:00 (11 years ago)

Author:

andrewryan

Message:

added a verbose option to quick script which shows the updates the script has made to the namelist

File:

• branches/2013/dev_r3987_UKMO4_OBS/NEMOGCM/TOOLS/OBSTOOLS/OOO/ooo/ooo.py (modified) (4 diffs)

branches/2013/dev_r3987_UKMO4_OBS/NEMOGCM/TOOLS/OBSTOOLS/OOO/ooo/ooo.py

@@ -25 +25 @@
     parser.add_argument("--dry-run", action="store_true",
                         help="Flag to test namelist building without submitting.")
+    parser.add_argument("-v", "--verbose", action="store_true",
+                        help="Prints difference between before and after namelists.")
     parser.add_argument("namelist", metavar="NAMELIST", 
                         help="NEMO namelist to edit.")
@@ -32 +34 @@
     args.lead_times = map(int, args.lead_times.split(','))
     return args
+
+def printdiff(text1, text2):
+    # Provides nice text difference summary of namelists
+    import difflib
+    lines1 = text1.splitlines()
+    lines2 = text2.splitlines()
+    d = difflib.Differ()
+    result = list(d.compare(lines1, lines2))
+    text = '\n'.join(result)
+    print text
 
 def main():
@@ -53 +65 @@
     text = nml.reader(args.namelist)
     sublists = nml.namelists(text)
+
+    # Verbose save original text
+    if args.verbose:
+        original_text = text
 
     # namooo
@@ -87 +103 @@
     text = nml.update("namobs", text, data=namobs)
 
+    # Verbose print namelist differences
+    if args.verbose:
+        printdiff(original_text, text)
+
     # pipe text to file
     tmp = args.namelist+".tmp"