Adists: Prepare for scenarios update (commit code is not working).

src/Model/InitialConditionsAdisTS/InitialConditionsAdisTSSpec.py

@@ -56,25 +56,28 @@ class ICAdisTSSpec(SQLSubModel):
         ICAdisTSSpec._id_cnt = max(ICAdisTSSpec._id_cnt + 1, self.id)
 
     @classmethod
-    def _db_create(cls, execute):
-        execute("""
-            CREATE TABLE initial_conditions_spec(
-              id INTEGER NOT NULL PRIMARY KEY,
-              ic_default INTEGER NOT NULL,
-              name TEXT NOT NULL,
-              reach INTEGER NOT NULL,
-              start_rk REAL NOT NULL,
-              end_rk REAL NOT NULL,
-              concentration REAL NOT NULL,
-              eg REAL NOT NULL,
-              em REAL NOT NULL,
-              ed REAL NOT NULL,
-              rate REAL NOT NULL,
-              enabled BOOLEAN NOT NULL,
-              FOREIGN KEY(ic_default) REFERENCES initial_conditions_adists(id),
-              FOREIGN KEY(reach) REFERENCES river_reach(id)
-            )
-            """)
+    def _db_create(cls, execute, ext=""):
+        execute(f"""
+          CREATE TABLE initial_conditions_spec{ext}(
+            {cls.create_db_add_pamhyr_id()},
+            ic_default INTEGER NOT NULL,
+            name TEXT NOT NULL,
+            reach INTEGER NOT NULL,
+            start_rk REAL NOT NULL,
+            end_rk REAL NOT NULL,
+            concentration REAL NOT NULL,
+            eg REAL NOT NULL,
+            em REAL NOT NULL,
+            ed REAL NOT NULL,
+            rate REAL NOT NULL,
+            enabled BOOLEAN NOT NULL,
+            {Scenario.create_db_add_scenario()},
+            {Scenario.create_db_add_scenario_fk()},
+            FOREIGN KEY(ic_default)
+              REFERENCES initial_conditions_adists(pamhyr_id),
+            FOREIGN KEY(reach) REFERENCES river_reach(pamhyr_id)
+          )
+        """)
 
         return cls._create_submodel(execute)
 
@@ -85,8 +88,35 @@ class ICAdisTSSpec(SQLSubModel):
             if int(release) < 6:
                 cls._db_create(execute)
 
+        if major == "0" and int(minor) < 2:
+            cls._db_update_to_0_2_0(execute, data)
+
         return True
 
+    @classmethod
+    def _db_update_to_0_2_0(cls, execute, data):
+        table = "initial_conditions_spec"
+        reachs = data['id2pid']['river_reach']
+
+        cls.update_db_add_pamhyr_id(execute, table, data)
+        Scenario.update_db_add_scenario(execute, table)
+
+        cls._db_create_lca(execute, ext="_tmp")
+
+        execute(
+            f"INSERT INTO {table}_tmp " +
+            "(pamhyr_id, ic_default, name, reach, start_rk, end_rk, " +
+            "concentration, eg, em, ed, rate, enabled, scenario) " +
+            "SELECT pamhyr_id, ic_default, name, reach, start_rk, end_rk, " +
+            "concentration, eg, em, ed, rate, enabled, scenario " +
+            f"FROM {table}"
+        )
+
+        execute(f"DROP TABLE {table}")
+        execute(f"ALTER TABLE {table}_tmp RENAME TO {table}")
+
+        cls._db_update_to_0_2_0_set_reach_pid(execute, table, reachs)
+
     @classmethod
     def _db_load(cls, execute, data=None):
         new = []