Return paths to existing files from service functions

openclimatefix · Jan 15, 2024 · ef0dc6e · ef0dc6e
1 parent 91888e9
commit ef0dc6e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/nwp_consumer/internal/service/service.py b/src/nwp_consumer/internal/service/service.py
@@ -87,7 +87,7 @@ def DownloadRawDataset(self, *, start: dt.date, end: dt.date) -> list[pathlib.Pa
                 startDate=start.strftime("%Y-%m-%d %H:%M"),
                 endDate=end.strftime("%Y-%m-%d %H:%M"),
             )
-            return []
+            return [self.rawdir / fi.it().strftime(internal.IT_FOLDER_FMTSTR) / fi.filename() for fi in allWantedFileInfos]
         else:
             log.info(
                 event="downloading files",
@@ -145,7 +145,11 @@ def ConvertRawDatasetToZarr(self, *, start: dt.date, end: dt.date) -> list[pathl
                 startDate=start.strftime("%Y/%m/%d %H:%M"),
                 endDate=end.strftime("%Y/%m/%d %H:%M"),
             )
-            return []
+            return [
+                self.zarrdir / it.strftime(f"{internal.ZARR_FMTSTR}.zarr.zip")
+                for it in allInitTimes
+                if start <= it.date() <= end
+            ]
         else:
             log.info(
                 event=f"converting {len(desiredInitTimes)} init times to zarr.",