Merge pull request #25 from KNMI/parameters

Enhance Parameters functionality

Author: PaulVanSchayck

src/edr_pydantic/parameter.py

@@ -41,7 +41,17 @@ def must_not_have_unit_if_observed_property_has_categories(self):
         return self
 
 
-Parameters = RootModel[Dict[str, Parameter]]
+class Parameters(RootModel):
+    root: Dict[str, Parameter]
+
+    def __iter__(self):
+        return iter(self.root)
+
+    def __getitem__(self, key):
+        return self.root[key]
+
+    def get(self, key, default=None):
+        return self.root.get(key, default)
 
 
 class ParameterGroup(EdrBaseModel, extra="allow"):

tests/test_edr.py

@@ -69,3 +69,13 @@ def test_data_type_alias():
         p.model_dump_json(exclude_none=True, by_alias=True)
         == '{"type":"Parameter","data-type":"integer","observedProperty":{"label":"Wind"}}'
     )
+
+
+def test_parameters_root_model():
+    file = Path(__file__).parent.resolve() / "test_data" / "parameter-with-data-type.json"
+    with open(file, "r") as f:
+        parameters = Parameters.model_validate_json(f.read())
+
+    assert parameters["str-parameter"].observedProperty.label == "string parameter"
+    assert parameters.get("int-parameter").observedProperty.label == "int parameter"
+    assert len([p for p in parameters]) == 3