Merge pull request #2 from nhm-usgs/newopt

Newopt merged into master

.gitignore

@@ -1,4 +1,5 @@
 Data
+Output/
 Output/*.nc
 Output/*.txt
 pkg/__pycache__

notebooks/.gitignore

@@ -0,0 +1,5 @@
+*.nc
+*.txt
+*.gz
+*.png
+*.cbh

pkg/climate_etl.py

@@ -0,0 +1,128 @@
+from fponhm import FpoNHM
+import sys, getopt
+from pathlib import Path
+import datetime
+import argparse
+
+def valid_date(s):
+    try:
+        return datetime.datetime.strptime(s, "%Y-%m-%d")
+    except ValueError:
+        msg = "Not a valid date: '{0}'.".format(s)
+        raise argparse.ArgumentTypeError(msg)
+
+def main():
+    numdays = None
+    startdate = None
+    enddate = None
+    idir = None
+    odir = None
+    wght_file = None
+    extract_type = None
+    file_prefix = None
+
+    my_parser = argparse.ArgumentParser(prog='climate_etl',
+                                    description='map gridded climate data to polygon using zonal area weighted mean')
+
+    # my_parser.add_argument('-s', '--startdate', type=valid_date,
+    #                        help='option: start date of retrieval (YYYY-MM-DD); requires -e (end_date)',
+    #                        metavar='start_date',
+    #                        default=None)
+    #
+    # my_parser.add_argument('-e', '--enddate', type=valid_date,
+    #                        help='option: end date of retrieval (YYYY-MM-DD); requires -s (start_date)',
+    #                        metavar='end_date',
+    #                        default=None)
+    my_parser.add_argument('-t', '--extract_type', type=str,
+                           help='extract method: (days) or (date)', metavar='input_path',
+                           default=None, required=True, choices=['days', 'date'])
+    my_parser.add_argument('-p', '--period', type=valid_date,
+                           help='option: start date and end date of retrieval (YYYY-MM-DD)',
+                           metavar='date',
+                           default=None,
+                           nargs=2)
+
+    my_parser.add_argument('-d', '--days', type=int,
+                           help='option: number of days to retrieve; if specified take precedence over -s & -e option',
+                           metavar='numdays', default=None)
+
+    my_parser.add_argument('-f', '--file_prefix', type=str,
+                           help='option: prefix for output files',
+                           metavar='file_prefix', default='')
+
+    my_parser.add_argument('-i', '--inpath',type=str,
+                           help='input path (location of shapefiles)', metavar='input_path',
+                           default=None, required=True)
+
+    my_parser.add_argument('-o', '--outpath', type=str,
+                           help='Output path (location of netcdf by shapefile output)', metavar='output_path',
+                           default=None, required=True)
+
+    my_parser.add_argument('-w', '--weightsfile', type=str,
+                           help='path/weight.csv - path to weight file', metavar='weight_file',
+                           default=None, required=True)
+
+    args = my_parser.parse_args()
+
+    if all(i is not None for i  in [args.period, args.days]):
+        my_parser.error('Either the --days or --period option must be specified not both')
+    if all(i is None for i in [args.period, args.days]):
+        my_parser.error('Either the --days or --period option must be specified')
+    # if args.period is not None:
+    #     startdate = args.period[0]
+    #     enddate = args.period[1]
+    if args.extract_type is not None:
+        extract_type = args.extract_type
+        if args.extract_type == 'days':
+            if args.days is not None:
+                numdays = args.days
+            else:
+                my_parser.error('if -t --extract_type == days then -d must be specified')
+        elif args.extract_type == 'date':
+            if args.period is not None:
+                startdate = args.period[0]
+                enddate = args.period[1]
+                if startdate >= enddate:
+                    my_parser.error('when using -p the first date must occur before the second')
+            else:
+                my_parser.error('if -t --extract_type == dates then -p must be specified')
+
+    if args.outpath is not None:
+        odir = args.outpath
+    if args.inpath is not None:
+        idir = args.inpath
+    if args.weightsfile is not None:
+        wght_file = args.weightsfile
+    if args.file_prefix is not None:
+        file_prefix = args.file_prefix
+    test = 0
+
+
+    print('starting Script')
+    #numdays = 2
+    fp = FpoNHM()
+    print('instantiated')
+    #initialize(self, iptpath, optpath, weights_file, type=None, days=None, start_date=None, end_date=None)
+    ready = fp.initialize(idir, odir, wght_file, extract_type, numdays, startdate, enddate, file_prefix)
+    if ready:
+        print('initalized\n')
+        print('running')
+        fp.run_weights()
+        print('finished running')
+        fp.finalize()
+        print('finalized')
+    else:
+        if extract_type == 'days':
+            print('Gridmet not updated continue with numdays -1')
+            fp.setNumdays(numdays-1)
+            print('initalized\n')
+            print('running')
+            fp.run_weights()
+            print('finished running')
+            fp.finalize()
+            print('finalized')
+        else:
+            print('error: extract did not return period specified')
+
+if __name__ == "__main__":
+    main()