Enable running on command line

Author: HansKimDWR

dms_datastore/dropbox_data.py

@@ -5,6 +5,7 @@
 from dms_datastore.read_ts import read_ts, infer_freq_robust
 from dms_datastore.write_ts import write_ts_csv
 from dms_datastore.dstore_config import station_dbase
+import argparse
 
 # Global variable to store cached data
 _cached_spec = None
@@ -167,6 +168,14 @@ def dropbox_data(spec_fname):
     spec = get_spec(spec_fname)
     get_data(spec)
 
-
-if __name__ == "__name__" or True:
-    dropbox_data("dropbox_spec.yaml")
+def create_arg_parser():
+    parser = argparse.ArgumentParser("Reads unformatted data files and writes to formatted csv files according to dropbox specification.")
+    parser.add_argument('--input',default=False,help=".yaml file with dropbox specification")
+    return parser
+
+def main():
+    parser = create_arg_parser()
+    args = parser.parse_args()
+    if not args.input:
+        raise ValueError("input yaml file required")
+    dropbox_data(args.input)

setup.py

@@ -81,7 +81,8 @@
             "usgs_multi=dms_datastore.usgs_multi:main",
             "delete_from_filelist=dms_datastore.delete_from_filelist:main",
             "data_cache=dms_datastore.caching:main",
-            "merge_files=dms_datastore.merge_files:main"
+            "merge_files=dms_datastore.merge_files:main",
+            "dropbox_data=dms_datastore.dropbox_data:main"
         ]
     },
 )