From a525d8a2b0f8a351896f5fa4f70a05f9b8382a7a Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <contact@parouby.fr>
Date: Wed, 13 Aug 2025 16:23:46 +0200
Subject: [PATCH] Adists: Scenario: Minor fix.

---
 src/Model/D90AdisTS/D90AdisTS.py                            | 6 +++---
 src/Model/DIFAdisTS/DIFAdisTS.py                            | 2 +-
 .../InitialConditionsAdisTS/InitialConditionsAdisTS.py      | 6 ++++--
 .../LateralContributionsAdisTS/LateralContributionAdisTS.py | 3 +++
 src/Model/River.py                                          | 2 +-
 src/Model/Study.py                                          | 6 +++---
 6 files changed, 15 insertions(+), 10 deletions(-)

diff --git a/src/Model/D90AdisTS/D90AdisTS.py b/src/Model/D90AdisTS/D90AdisTS.py
index 86b59802..abde7494 100644
--- a/src/Model/D90AdisTS/D90AdisTS.py
+++ b/src/Model/D90AdisTS/D90AdisTS.py
@@ -101,7 +101,7 @@ class D90AdisTS(SQLSubModel):
         new = []
 
         table = execute(
-            "SELECT id, name, d90, enabled " +
+            "SELECT pamhyr_id, name, d90, enabled " +
             "FROM d90_adists"
         )
 
@@ -129,7 +129,7 @@ class D90AdisTS(SQLSubModel):
         return new
 
     def _db_save(self, execute, data=None):
-        execute(f"DELETE FROM d90_adists WHERE id = {self.id}")
+        execute(f"DELETE FROM d90_adists WHERE pamhyr_id = {self.id}")
 
         d90 = -1.
         if self.d90 is not None:
@@ -138,7 +138,7 @@ class D90AdisTS(SQLSubModel):
         sql = (
             "INSERT INTO " +
             "d90_adists(" +
-            "id, name, d90, enabled" +
+            "pamhyr_id, name, d90, enabled" +
             ") " +
             "VALUES (" +
             f"{self.id}, '{self._db_format(self._name)}', " +
diff --git a/src/Model/DIFAdisTS/DIFAdisTS.py b/src/Model/DIFAdisTS/DIFAdisTS.py
index b6ad0bbc..e5bc5d31 100644
--- a/src/Model/DIFAdisTS/DIFAdisTS.py
+++ b/src/Model/DIFAdisTS/DIFAdisTS.py
@@ -107,7 +107,7 @@ class DIFAdisTS(SQLSubModel):
         new = []
 
         table = execute(
-            "SELECT id, name, method, dif, b, c, enabled " +
+            "SELECT pamhyr_id, name, method, dif, b, c, enabled " +
             "FROM dif_adists"
         )
 
diff --git a/src/Model/InitialConditionsAdisTS/InitialConditionsAdisTS.py b/src/Model/InitialConditionsAdisTS/InitialConditionsAdisTS.py
index 094d2807..0f418523 100644
--- a/src/Model/InitialConditionsAdisTS/InitialConditionsAdisTS.py
+++ b/src/Model/InitialConditionsAdisTS/InitialConditionsAdisTS.py
@@ -99,8 +99,10 @@ class InitialConditionsAdisTS(SQLSubModel):
 
         execute(
             f"INSERT INTO {table}_tmp " +
-            "(pamhyr_id, pollutant, name, concentration, eg, em, ed, scenario) " +
-            "SELECT pamhyr_id, pollutant, name, concentration, eg, em, ed, scenario) " +
+            "(pamhyr_id, pollutant, name, concentration, " +
+            "eg, em, ed, scenario) " +
+            "SELECT pamhyr_id, pollutant, name, concentration, " +
+            "eg, em, ed, scenario) " +
             f"FROM {table}"
         )
 
diff --git a/src/Model/LateralContributionsAdisTS/LateralContributionAdisTS.py b/src/Model/LateralContributionsAdisTS/LateralContributionAdisTS.py
index cfd02aa5..299b119e 100644
--- a/src/Model/LateralContributionsAdisTS/LateralContributionAdisTS.py
+++ b/src/Model/LateralContributionsAdisTS/LateralContributionAdisTS.py
@@ -153,6 +153,9 @@ class LateralContributionAdisTS(SQLSubModel):
         scenario = data["scenario"]
         loaded = data['loaded_pid']
 
+        if scenario is None:
+            return new
+
         table = execute(
             "SELECT pamhyr_id, pollutant, reach, begin_rk, end_rk " +
             "FROM lateral_contribution_adists " +
diff --git a/src/Model/River.py b/src/Model/River.py
index b51d61b6..1bc54e2d 100644
--- a/src/Model/River.py
+++ b/src/Model/River.py
@@ -100,7 +100,7 @@ class RiverNode(Node):
     def _db_update(cls, execute, version, data=None):
         major, minor, release = version.strip().split(".")
 
-        if major == minor == "0":
+        if major == "0" and int(minor) < 2:
             cls._db_update_to_0_2_0(execute, data=data)
 
         if major == "0" and minor == "1":
diff --git a/src/Model/Study.py b/src/Model/Study.py
index 97395099..318eebce 100644
--- a/src/Model/Study.py
+++ b/src/Model/Study.py
@@ -277,10 +277,10 @@ class Study(SQLModel):
                 "INSERT INTO info VALUES ('study_release', '0')"
             )
 
-        if major == "0" and int(minor) <= 1:
+        if major == "0" and int(minor) <= 2:
             self._add_into_info_if_not_exists('current_scenario', '0')
 
-        if major == "0" and int(minor) < 1:
+        if major == "0" and int(minor) < 2:
             # Need to temporary disable the sqlite foreign keys
             # checking to update db dans change the table id fk to
             # table pamhyr_id fk
@@ -290,7 +290,7 @@ class Study(SQLModel):
 
         ok = self._update_submodel(version[0], data={})
 
-        if major == "0" and int(minor) < 1:
+        if major == "0" and int(minor) < 2:
             # Reactivate foreign keys checking
             self.execute(
                 "PRAGMA foreign_keys = ON;"