adding commands to setup

dms_datastore/download_cimis.py

@@ -360,7 +360,7 @@ def cache_to_pkl(self, dfstations):
             print(dfstations[dfstations["Station Number"].isin(failed_stations)])
 
 
-if __name__ == "__main__":
+def main():
     password = os.environ.get("CIMIS_PASSWORD", default="xxx")
     cx = CIMIS(password=password)
     dfcat = cx.get_stations_info()
@@ -383,3 +383,7 @@ def cache_to_pkl(self, dfstations):
         except Exception as e:
             print(f"Error: {e}")
             continue
+
+
+if __name__ == "__main__":
+    main()

dms_datastore/download_ucdipm.py

@@ -17,7 +17,9 @@ def get_weather_data(start_date, end_date, stnKey=281):
 
 
 # argparse for calling get_weather_data from command line
-if __name__ == "__main__":
+
+
+def main():
     import argparse
 
     parser = argparse.ArgumentParser()
@@ -27,3 +29,7 @@ def get_weather_data(start_date, end_date, stnKey=281):
     args = parser.parse_args()
     df = get_weather_data(args.start_date, args.end_date)
     df.to_csv(f"ucdipm_{args.stnKey}.csv")
+
+
+if __name__ == "__main__":
+    main()

setup.py

@@ -67,6 +67,8 @@
             "download_des=dms_datastore.download_des:main",
             "download_ncro=dms_datastore.download_ncro:main",
             "download_mokelumne=dms_datastore.download_mokelumne:main",
+            "download_ucdipm=dms_datastore.download_ucdipm:main",
+            "download_cimis=dms_datastore.download_cimis:main",
             "compare_directories=dms_datastore.compare_directories:main",
             "populate_repo=dms_datastore.populate_repo:main",
             "station_info=dms_datastore.station_info:main",