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

src/Model/WeatherParameters/WeatherParameters.py

@@ -154,6 +154,93 @@ class Data(SQLSubModel):
         self._data[key] = self._types[key](value)
 
 
+class WeatherParametersDefault(SQLSubModel):
+    def __init__(self, type: str, value: float = 0.0,
+                 id: int = -1, status=None, owner_scenario=-1):
+        super(WeatherParametersDefault, self).__init__(
+            id=id, status=status,
+            owner_scenario=owner_scenario
+        )
+
+        self._type = type
+        self._value = float(value)
+
+    @classmethod
+    def _db_load(cls, execute, data=None):
+        new = []
+        status = data['status']
+        scenario = data["scenario"]
+        loaded = data['loaded_pid']
+
+        if scenario is None:
+            return new
+
+        table = execute(
+            "SELECT pamhyr_id, deleted, type, value, scenario " +
+            "FROM weather_parameters_default " +
+            f"WHERE scenario = {scenario.id} " +
+            f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))}) "
+        )
+
+        for row in table:
+            it = iter(row)
+
+            pid = next(it)
+            deleted = next(it)
+            type = next(it)
+            value = next(it)
+            owner_scenario = next(it)
+
+            wp_default = cls(
+                type=type,
+                value=value,
+                id=pid,
+                status=status,
+                owner_scenario=owner_scenario
+            )
+            if deleted:
+                wp_default.set_as_deleted()
+
+            loaded.add(pid)
+            new.append(wp_default)
+
+        data["scenario"] = scenario.parent
+        new += cls._db_load(execute, data)
+        data["scenario"] = scenario
+
+        return new
+
+    def _db_save(self, execute, data=None):
+        if not self.must_be_saved():
+            return True
+
+        execute(
+            "INSERT INTO " +
+            "weather_parameters_default(" +
+            "pamhyr_id, deleted, type, value, scenario) " +
+            "VALUES (" +
+            f"{self.id}, {self._db_format(self.is_deleted())}, " +
+            f"'{self._db_format(self._type)}', {self._value}, " +
+            f"{self._status.scenario_id}" +
+            ")"
+        )
+
+        return True
+
+    @property
+    def type(self):
+        return self._type
+
+    @property
+    def value(self):
+        return self._value
+
+    @value.setter
+    def value(self, value):
+        self._value = float(value)
+        self.modified()
+
+
 class WeatherParameters(SQLSubModel):
     _sub_classes = [Data]
 
@@ -196,6 +283,18 @@ class WeatherParameters(SQLSubModel):
           )
         """)
 
+        execute(f"""
+          CREATE TABLE weather_parameters_default{ext}(
+            {cls.create_db_add_pamhyr_id()},
+            deleted BOOLEAN NOT NULL DEFAULT FALSE,
+            type TEXT NOT NULL,
+            value FLOAT NOT NULL,
+            {Scenario.create_db_add_scenario()},
+            {Scenario.create_db_add_scenario_fk()},
+            primary KEY(pamhyr_id, scenario)
+          )
+        """)
+
         if ext == "_tmp":
             return True
 
@@ -245,6 +344,19 @@ class WeatherParameters(SQLSubModel):
                 cls._db_create(execute)
                 created = True
 
+        if not cls.is_table_exists(execute, "weather_parameters_default"):
+            execute(f"""
+              CREATE TABLE weather_parameters_default(
+                {cls.create_db_add_pamhyr_id()},
+                deleted BOOLEAN NOT NULL DEFAULT FALSE,
+                type TEXT NOT NULL,
+                value FLOAT NOT NULL,
+                {Scenario.create_db_add_scenario()},
+                {Scenario.create_db_add_scenario_fk()},
+                primary KEY(pamhyr_id, scenario)
+              )
+            """)
+
         if not created:
             return cls._update_submodel(execute, version, data)