From 2e08147b57f2f12652d98399a764c15a53b96683 Mon Sep 17 00:00:00 2001
From: Jordan Laser <jordan.science.laser@gmail.com>
Date: Mon, 4 Dec 2023 10:49:27 -0700
Subject: [PATCH] datastream config generator

---
 python/configure-datastream.py | 71 ++++++++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 python/configure-datastream.py

diff --git a/python/configure-datastream.py b/python/configure-datastream.py
new file mode 100644
index 00000000..66523755
--- /dev/null
+++ b/python/configure-datastream.py
@@ -0,0 +1,71 @@
+import argparse, json, os, datetime
+from pathlib import Path
+
+def create_nwmurl_conf(conf, out_dir):
+    nwm_conf = conf['nwmurl']
+    nwm_conf['start_date'] = conf['globals']['start_date']
+    nwm_conf['end_date']   = conf['globals']['end_date']
+    conf_name = 'conf_nwmurl.json'
+    conf_path = Path(out_dir,conf_name)
+    with open(conf_path,'w') as fp:
+        json.dump(nwm_conf, fp)
+
+    return conf_path
+
+def create_fp_conf(conf, out_dir):
+    fp_conf = conf['forcingprocessor']
+    fp_conf['forcing']['start_date'] = conf['globals']['start_date']
+    fp_conf['forcing']['end_date']   = conf['globals']['end_date']
+    conf_name = 'conf_fp.json'
+    conf_path = Path(out_dir,conf_name)
+    with open(conf_path,'w') as fp:
+        json.dump(fp_conf, fp)
+
+    return conf_path
+
+def create_confs(conf):
+    if conf['globals']['start_date'] == "DAILY":
+        date = datetime.datetime.now()
+        date_tomorrow = date + datetime.timedelta(days=1)
+        date = date.strftime('%Y%m%d')
+        date_tomorrow = date_tomorrow.strftime('%Y%m%d')
+        hourminute = '0000'
+        
+        conf['globals']['start_date'] = date + hourminute
+        conf['globals']['end_date']   = date_tomorrow + hourminute
+
+    if "relative_to" in conf['globals'].keys():
+        out_dir = Path(conf['globals']['relative_to'],conf['globals']['data_dir'])
+    else:
+        out_dir = conf['globals']['data_dir']
+    out_dir = Path(out_dir,'datastream-configs')
+
+    create_nwmurl_conf(conf,out_dir)
+    create_fp_conf(conf,out_dir)
+
+    conf_name = 'conf_datastream.json'
+    conf_path = Path(out_dir,conf_name)
+    with open(conf_path,'w') as fp:
+        json.dump(conf, fp)
+
+    print(f'\ndatastream configs have been generated and placed here\n{out_dir}\n')
+
+if __name__ == "__main__":
+    # Take in user config
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        dest="infile", type=str, help="A json containing user inputs to run ngen-datastream"
+    )
+    args = parser.parse_args()
+
+    if args.infile[0] == '{':
+        conf = json.loads(args.infile)
+    else:
+        if 's3' in args.infile:
+            os.system(f'wget {args.infile}')
+            filename = args.infile.split('/')[-1]
+            conf = json.load(open(filename))
+        else:
+            conf = json.load(open(args.infile))
+
+    create_confs(conf)