Print settings (#137)

E3SM-Project · Sep 29, 2021 · 9d814ff · 9d814ff
1 parent ade8fb7
commit 9d814ff
Show file tree

Hide file tree

Showing 7 changed files with 49 additions and 0 deletions.
diff --git a/zppy/amwg.py b/zppy/amwg.py
@@ -1,4 +1,5 @@
 import os
+import pprint
 
 import jinja2
 
@@ -42,6 +43,7 @@ def amwg(config, scriptDir):
             c["prefix"] = prefix
             scriptFile = os.path.join(scriptDir, "%s.csh" % (prefix))
             statusFile = os.path.join(scriptDir, "%s.status" % (prefix))
+            settingsFile = os.path.join(scriptDir, "%s.settings" % (prefix))
             skip = checkStatus(statusFile)
             if skip:
                 continue
@@ -58,6 +60,11 @@ def amwg(config, scriptDir):
                 ),
             ]
 
+            with open(settingsFile, "w") as sf:
+                p = pprint.PrettyPrinter(indent=2, stream=sf)
+                p.pprint(c)
+                p.pprint(s)
+
             if not c["dry_run"]:
                 # Submit job
                 jobid = submitScript(

diff --git a/zppy/climo.py b/zppy/climo.py
@@ -1,4 +1,5 @@
 import os
+import pprint
 import re
 
 import jinja2
@@ -62,6 +63,7 @@ def climo(config, scriptDir):
             c["prefix"] = prefix
             scriptFile = os.path.join(scriptDir, "%s.bash" % (prefix))
             statusFile = os.path.join(scriptDir, "%s.status" % (prefix))
+            settingsFile = os.path.join(scriptDir, "%s.settings" % (prefix))
             skip = checkStatus(statusFile)
             if skip:
                 continue
@@ -70,6 +72,11 @@ def climo(config, scriptDir):
             with open(scriptFile, "w") as f:
                 f.write(template.render(**c))
 
+            with open(settingsFile, "w") as sf:
+                p = pprint.PrettyPrinter(indent=2, stream=sf)
+                p.pprint(c)
+                p.pprint(s)
+
             if not c["dry_run"]:
                 # Submit job
                 jobid = submitScript(scriptFile)

diff --git a/zppy/e3sm_diags.py b/zppy/e3sm_diags.py
@@ -1,4 +1,5 @@
 import os
+import pprint
 
 import jinja2
 
@@ -46,6 +47,7 @@ def e3sm_diags(config, scriptDir):
             c["prefix"] = prefix
             scriptFile = os.path.join(scriptDir, "%s.bash" % (prefix))
             statusFile = os.path.join(scriptDir, "%s.status" % (prefix))
+            settingsFile = os.path.join(scriptDir, "%s.settings" % (prefix))
             skip = checkStatus(statusFile)
             if skip:
                 continue
@@ -90,6 +92,11 @@ def e3sm_diags(config, scriptDir):
                             )
                         )
 
+            with open(settingsFile, "w") as sf:
+                p = pprint.PrettyPrinter(indent=2, stream=sf)
+                p.pprint(c)
+                p.pprint(s)
+
             if not c["dry_run"]:
                 # Submit job
                 jobid = submitScript(

diff --git a/zppy/e3sm_diags_vs_model.py b/zppy/e3sm_diags_vs_model.py
@@ -1,4 +1,5 @@
 import os
+import pprint
 
 import jinja2
 
@@ -52,6 +53,7 @@ def e3sm_diags_vs_model(config, scriptDir):
             c["prefix"] = prefix
             scriptFile = os.path.join(scriptDir, "%s.bash" % (prefix))
             statusFile = os.path.join(scriptDir, "%s.status" % (prefix))
+            settingsFile = os.path.join(scriptDir, "%s.settings" % (prefix))
             skip = checkStatus(statusFile)
             if skip:
                 continue
@@ -68,6 +70,11 @@ def e3sm_diags_vs_model(config, scriptDir):
                 ),
             ]
 
+            with open(settingsFile, "w") as sf:
+                p = pprint.PrettyPrinter(indent=2, stream=sf)
+                p.pprint(c)
+                p.pprint(s)
+
             if not c["dry_run"]:
                 # Submit job
                 jobid = submitScript(

diff --git a/zppy/global_time_series.py b/zppy/global_time_series.py
@@ -1,4 +1,5 @@
 import os
+import pprint
 
 import jinja2
 
@@ -36,6 +37,7 @@ def global_time_series(config, scriptDir):
             c["prefix"] = prefix
             scriptFile = os.path.join(scriptDir, "%s.bash" % (prefix))
             statusFile = os.path.join(scriptDir, "%s.status" % (prefix))
+            settingsFile = os.path.join(scriptDir, "%s.settings" % (prefix))
             skip = checkStatus(statusFile)
             if skip:
                 continue
@@ -92,6 +94,11 @@ def global_time_series(config, scriptDir):
                     )
                 )
 
+            with open(settingsFile, "w") as sf:
+                p = pprint.PrettyPrinter(indent=2, stream=sf)
+                p.pprint(c)
+                p.pprint(s)
+
             if not c["dry_run"]:
                 # Submit job
                 jobid = submitScript(

diff --git a/zppy/mpas_analysis.py b/zppy/mpas_analysis.py
@@ -1,4 +1,5 @@
 import os
+import pprint
 
 import jinja2
 
@@ -76,6 +77,7 @@ def mpas_analysis(config, scriptDir):
             c["prefix"] = prefix
             scriptFile = os.path.join(scriptDir, "%s.bash" % (prefix))
             statusFile = os.path.join(scriptDir, "%s.status" % (prefix))
+            settingsFile = os.path.join(scriptDir, "%s.settings" % (prefix))
 
             # Check if we can skip because it completed successfully before
             skip = checkStatus(statusFile)
@@ -88,6 +90,11 @@ def mpas_analysis(config, scriptDir):
             with open(scriptFile, "w") as f:
                 f.write(template.render(**c))
 
+            with open(settingsFile, "w") as sf:
+                p = pprint.PrettyPrinter(indent=2, stream=sf)
+                p.pprint(c)
+                p.pprint(s)
+
             if not c["dry_run"]:
                 # Submit job
                 jobid = submitScript(scriptFile, dependFiles=dependencies)

diff --git a/zppy/ts.py b/zppy/ts.py
@@ -1,4 +1,5 @@
 import os
+import pprint
 import re
 
 import jinja2
@@ -70,6 +71,7 @@ def ts(config, scriptDir):
             c["prefix"] = prefix
             scriptFile = os.path.join(scriptDir, "%s.bash" % (prefix))
             statusFile = os.path.join(scriptDir, "%s.status" % (prefix))
+            settingsFile = os.path.join(scriptDir, "%s.settings" % (prefix))
             skip = checkStatus(statusFile)
             if skip:
                 continue
@@ -78,6 +80,11 @@ def ts(config, scriptDir):
             with open(scriptFile, "w") as f:
                 f.write(template.render(**c))
 
+            with open(settingsFile, "w") as sf:
+                p = pprint.PrettyPrinter(indent=2, stream=sf)
+                p.pprint(c)
+                p.pprint(s)
+
             if not c["dry_run"]:
                 # Submit job
                 jobid = submitScript(scriptFile)