Merge pull request #59 from ecmwf/feat/restructure

Feat/restructure

Author: Oisin-M

hat/__init__.py

@@ -1,4 +1 @@
-try:
-    from ._version import __version__
-except ImportError:
-    __version__ = ""
+from ._version import __version__

hat/cli.py

@@ -0,0 +1,21 @@
+import yaml
+import argparse
+
+from .compute_hydrostats.stat_calc import stat_calc
+from .extract_timeseries.extractor import extractor
+from .station_mapping.mapper import mapper
+
+def commandlineify(func):
+    def wrapper():
+        parser = argparse.ArgumentParser(description="Run tool with YAML config")
+        parser.add_argument("config", help="Path to the YAML config file")
+        args = parser.parse_args()
+        confpath = args.config
+        with open(confpath, "r") as file:
+            config = yaml.safe_load(file)
+        func(config)
+    return wrapper
+
+mapper_cli = commandlineify(mapper)
+extractor_cli = commandlineify(extractor)
+stat_calc_cli = commandlineify(stat_calc)

hat/compute_hydrostats/stat_calc.py

@@ -6,7 +6,7 @@
 
 
 def load_da(ds_config):
-    ds = ekd.from_source(*ds_config["datasource"]).to_xarray(xarray_open_mfdataset_kwargs={"chunks": {"time": "auto"}})
+    ds = ekd.from_source(*ds_config["source"]).to_xarray(xarray_open_mfdataset_kwargs={"chunks": {"time": "auto"}})
     var_name = find_main_var(ds, 2)
     da = ds[var_name]
     return da
@@ -45,5 +45,6 @@ def stat_calc(config):
         func = getattr(stats, stat)
         stat_dict[stat] = func(sim_da, obs_da, new_coords.get("t", "time"))
     ds = xr.Dataset(stat_dict)
-
+    if config["output"].get("file", None) is not None:
+        ds.to_netcdf(config["output"]["file"])
     return ds