import xarray as xr
from glob import glob

paths = glob("/lsdf/kit/imk-tro/projects/MOD/Gruppe_Voigt/TRACMIP_ESGFCOPY/*/%s/%s*/%s/*/*/*/*/*/*"
             % ("ECHAM63", "aquaControl", "mon"))

for version in set([i.split("/")[-2] for i in paths]):

  ver_paths = [i for i in paths if version in i]

  merged = xr.open_dataset(ver_paths[0], chunks={})

  for path in ver_paths:

      # flatten height coordinate
      ds = xr.open_dataset(path)
      var = [i for i in ds.data_vars if "bnds" not in i][0]
      if "height" in ds.coords:
          ds[var].attrs["height"] = "%s %s" % (ds.height.values, ds.height.units)
      merged[var] = ds[var].chunk({"time" : "auto"})

  # drop height from overall dataset
  if "height" in merged.coords:
      merged = merged.drop("height")

  # convert to zarr
  merged.to_zarr("temp", mode="w", consolidated=True)