Add Canadian Meterological Center Model Support

README.md

@@ -29,6 +29,7 @@ currently packaged with adapters for pulling and converting `.grib` data from:
 - [CEDA Atmospheric Archive](https://catalogue.ceda.ac.uk)
 - [ECMWF MARS API](https://apps.ecmwf.int/mars-catalogue)
 - [DWD's ICON Model from the Opendata API](https://opendata.dwd.de)
+- [CMC's GDPS Model from the Opendata API](https://dd.weather.gc.ca/)
 
 Similarly, the service can write to multiple sinks:
 

src/nwp_consumer/internal/config/env.py

@@ -90,6 +90,13 @@ class ICONEnv(EnvParser):
     ICON_PARAMETER_GROUP: str = "default"
 
 
+class CMCEnv(EnvParser):
+    """Config for CMC API."""
+
+    CMC_MODEL: str = "gdps"
+    CMC_HOURS: int = 240
+    CMC_PARAMETER_GROUP: str = "full"
+
 # --- Outputs environment variables --- #
 
 

src/nwp_consumer/internal/inputs/__init__.py

@@ -1,8 +1,9 @@
-__all__ = ["ceda", "metoffice", "ecmwf", "icon"]
+__all__ = ["ceda", "metoffice", "ecmwf", "icon", "cmc"]
 
 from . import (
     ceda, 
     metoffice, 
     ecmwf, 
-    icon
+    icon,
+    cmc,
 )

src/nwp_consumer/internal/inputs/cmc/__init__.py

@@ -0,0 +1,4 @@
+__all__ = ["Client"]
+
+from .client import Client
+

src/nwp_consumer/internal/inputs/cmc/_consts.py

@@ -0,0 +1,62 @@
+"""Defines all parameters available from GDPS."""
+
+
+GDPS_VARIABLES = [
+    "ALBDO",
+    "ABSV",
+    "CWAT",
+    "TSOIL",
+    "SOILVIC",
+    "SOILM",
+    "SFCWRO",
+    "CAPE",
+    "CIN",
+    "ACPCP",
+    "DLWRF",
+    "DSWRF",
+    "HGT",
+    "FPRATE",
+    "IPRATE",
+    "PCPNTYPE",
+    "LHTFL",
+    "NLWRS",
+    "NSWRS",
+    "PRATE",
+    "PRES",
+    "RH",
+    "SKINT",
+    "SDEN",
+    "SNOD",
+    "SPRATE",
+    "SPFH",
+    "TMP",
+    "TCDC",
+    "APCP",
+    "ULWRF",
+    "VVEL",
+    "GUST",
+    "UGRD",
+    "VGRD",
+]
+
+GEPS_VARIABLES = [
+    "CAPE",
+    "CIN",
+    "HGT",
+    "ICETK",
+    "PRES",
+    "PRMSL",
+    "PWAT",
+    "RH",
+    "SCWRO",
+    "SNOD",
+    "SPFH",
+    "TCDC",
+    "TMP",
+    "TSOIL",
+    "UGRD",
+    "VGRD",
+    "WEASD",
+    "WIND",
+    "VVEL"
+    ]

src/nwp_consumer/internal/inputs/cmc/_models.py

@@ -0,0 +1,22 @@
+import datetime as dt
+
+from nwp_consumer import internal
+
+
+class CMCFileInfo(internal.FileInfoModel):
+    def __init__(
+        self, it: dt.datetime, filename: str, currentURL: str, step: int,
+    ) -> "CMCFileInfo":
+        self._it = it
+        self._filename = filename
+        self._url = currentURL
+        self.step = step
+
+    def filename(self) -> str:
+        return self._filename
+
+    def filepath(self) -> str:
+        return self._url + "/" + self._filename
+
+    def it(self) -> dt.datetime:
+        return self._it