WeatherParameters: New data in model+db to setup default parameters

src/Model/WeatherParameters/WeatherParametersList.py

@@ -22,7 +22,10 @@ from tools import trace, timer
 from Model.Tools.PamhyrListExt import PamhyrModelList
 from Model.Except import NotImplementedMethodeError
 
-from Model.WeatherParameters.WeatherParameters import WeatherParameters
+from Model.WeatherParameters.WeatherParameters import (
+    WeatherParameters,
+    WeatherParametersDefault,
+)
 from Model.WeatherParameters.WeatherParametersTypes import (
     NotDefined, WeatherParam, AirTemperature, SpecificHumidity,
     GlobalRadiation, ReferenceWindSpeed, GroundwaterTemperature,
@@ -34,6 +37,14 @@ class WeatherParametersList(PamhyrModelList):
     _sub_classes = [
         WeatherParameters,
     ]
+    _default_types = [
+        "AT", "SH", "GR", "RWS", "GT", "GFR", "ALB", "SC", "CCF"
+    ]
+
+    def __init__(self, status=None):
+        super(WeatherParametersList, self).__init__(status=status)
+        self._default = []
+        self._init_missing_default_values()
 
     @classmethod
     def _db_load(cls, execute, data=None):
@@ -46,6 +57,13 @@ class WeatherParametersList(PamhyrModelList):
             execute, data
         )
 
+        data['loaded_pid'] = set()
+        new._default = WeatherParametersDefault._db_load(
+            execute, data
+        )
+        new._init_missing_default_values()
+        new._sort_default_values()
+
         return new
 
     def _db_save(self, execute, data=None):
@@ -57,6 +75,10 @@ class WeatherParametersList(PamhyrModelList):
             "DELETE FROM weather_parameters_data " +
             f"WHERE scenario = {self._status.scenario_id}"
         )
+        execute(
+            "DELETE FROM weather_parameters_default " +
+            f"WHERE scenario = {self._status.scenario_id}"
+        )
 
         if data is None:
             data = {}
@@ -64,8 +86,41 @@ class WeatherParametersList(PamhyrModelList):
         for wp in self._lst:
             wp._db_save(execute, data=data)
 
+        for wp_default in self._default:
+            wp_default._db_save(execute, data=data)
+
         return True
 
+    def _data_traversal(self,
+                        predicate=lambda obj, data: True,
+                        modifier=lambda obj, data: None,
+                        data={}):
+        if predicate(self, data):
+            modifier(self, data)
+
+        for el in self._lst:
+            el._data_traversal(predicate, modifier, data)
+
+        for el in self._default:
+            el._data_traversal(predicate, modifier, data)
+
+    def _init_missing_default_values(self):
+        existing = [wp_default.type for wp_default in self.default]
+        for type in self._default_types:
+            if type not in existing:
+                self._default.append(
+                    WeatherParametersDefault(type=type, status=self._status)
+                )
+
+        self._sort_default_values()
+
+    def _sort_default_values(self):
+        self._default.sort(
+            key=lambda wp_default: self._default_types.index(wp_default.type)
+            if wp_default.type in self._default_types
+            else len(self._default_types)
+        )
+
     def new(self, index, type="ND"):
         if type == "AT":
             n = AirTemperature(status=self._status)
@@ -94,3 +149,18 @@ class WeatherParametersList(PamhyrModelList):
     @property
     def Weather_Parameters_List(self):
         return self.lst
+
+    @property
+    def default(self):
+        return list(
+            filter(
+                lambda el: not el.is_deleted(),
+                self._default
+            )
+        )
+
+    def default_for_type(self, type):
+        for wp_default in self.default:
+            if wp_default.type == type:
+                return wp_default
+        return None