From fe848acfddc07a36c945573704312c91bdb78300 Mon Sep 17 00:00:00 2001
From: alexhroom <alex.room@btinternet.com>
Date: Tue, 10 Dec 2024 10:40:52 +0000
Subject: [PATCH] added validators for value warnings and unimplemented
 behaviour

---
 RATapi/models.py | 68 +++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 59 insertions(+), 9 deletions(-)

diff --git a/RATapi/models.py b/RATapi/models.py
index a4188d5..970112f 100644
--- a/RATapi/models.py
+++ b/RATapi/models.py
@@ -1,6 +1,7 @@
 """The models module. Contains the pydantic models used by RAT to store project parameters."""
 
 import pathlib
+import warnings
 from itertools import count
 from typing import Any
 
@@ -60,7 +61,7 @@ class Background(RATModel):
         - if type is 'function', this should be the name of a custom function defined in `Project.custom_files`.
     value_1, value_2, ..., value_5 : str
         Values required by the background.
-        - if type is 'constant', this should be blank.
+        - if type is 'constant', all values will be ignored.
         - if type is 'data', value_1 may be the parameter name for an optional offset. Other values are ignored.
         - if type is 'function', these values may be the names of up to 5 parameters which are passed to the function.
 
@@ -75,6 +76,26 @@ class Background(RATModel):
     value_4: str = ""
     value_5: str = ""
 
+    @model_validator(mode="after")
+    def warn_parameters(self):
+        """Raise a warning if the parameters given are not expected for the given type."""
+        if self.type == TypeOptions.Constant:
+            expected_empty_fields = ["value_1", "value_2", "value_3", "value_4", "value_5"]
+        elif self.type == TypeOptions.Data:
+            expected_empty_fields = ["value_2", "value_3", "value_4", "value_5"]
+        else:
+            return self
+
+        non_empty_fields = [v for v in expected_empty_fields if getattr(self, v) != ""]
+        if non_empty_fields:
+            warnings.warn(
+                "The following values are not recognised by this background type and will be ignored: "
+                f"{', '.join(non_empty_fields)}",
+                stacklevel=2,
+            )
+
+        return self
+
 
 class Contrast(RATModel):
     """A group of all of the components of a model.
@@ -515,19 +536,20 @@ class Resolution(RATModel):
     Parameters
     ----------
     name : str
-        The name of the background.
+        The name of the resolution.
     type : TypeOptions
-        The type of background (constant, function or data)
+        The type of resolution: 'constant', 'data', or (NOT YET IMPLEMENTED) 'function'.
     source : str
-        The source data for the background;
+        The source data for the resolution;
         - if type is 'constant', this should be the name of a background parameter.
-        - if type is 'data', this should be the name of a dataset defined in `Project.data`.
-        - if type is 'function', this should be the name of a custom function defined in `Project.custom_files`.
+        - if type is 'data', this should be empty (resolution data is in the contrast data).
+        - if type is 'function' (NOT YET IMPLEMENTED),
+          this should be the name of a custom function defined in `Project.custom_files`.
     value_1, value_2, ..., value_5 : str
         Values required by the background.
-        - if type is 'constant', this should be blank.
-        - if type is 'data', value_1 may be the parameter name for an optional offset. Other values are ignored.
-        - if type is 'function', these values may be the names of up to 5 parameters which are passed to the function.
+        - if type is 'constant' or 'data', all values will be ignored.
+        - if type is 'function' (NOT YET IMPLEMENTED),
+          these values may be the names of up to 5 parameters which are passed to the function.
 
     """
 
@@ -539,3 +561,31 @@ class Resolution(RATModel):
     value_3: str = ""
     value_4: str = ""
     value_5: str = ""
+
+    @field_validator("type")
+    @classmethod
+    def validate_unimplemented_resolutions(cls, type: TypeOptions):
+        """Raise an error if currently unsupported function resolutions are used."""
+        if type == TypeOptions.Function:
+            raise NotImplementedError("Function resolutions are not yet supported.")
+        return type
+
+    @model_validator(mode="after")
+    def warn_parameters(self):
+        """Raise a warning if the parameters given are not expected for the given type."""
+        if self.type == TypeOptions.Constant:
+            expected_empty_fields = ["value_1", "value_2", "value_3", "value_4", "value_5"]
+        elif self.type == TypeOptions.Data:
+            expected_empty_fields = ["source", "value_1", "value_2", "value_3", "value_4", "value_5"]
+        else:
+            return self
+
+        non_empty_fields = [v for v in expected_empty_fields if getattr(self, v) != ""]
+        if non_empty_fields:
+            warnings.warn(
+                "The following values are not recognised by this resolution type and will be ignored: "
+                f"{', '.join(non_empty_fields)}",
+                stacklevel=2,
+            )
+
+        return self