SL: Fix save for scenario.

src/Model/Geometry/ProfileXYZ.py

@@ -339,6 +339,16 @@ class ProfileXYZ(Profile, SQLSubModel):
 
         return ok
 
+    def _data_traversal(self,
+                        predicate=lambda obj, data: True,
+                        modifier=lambda obj, data: None,
+                        data={}):
+        if predicate(self, data):
+            modifier(self, data)
+
+        for p in self._points:
+            p._data_traversal(predicate, modifier, data)
+
     @classmethod
     def from_data(cls, header, data):
         profile = None

src/Model/Geometry/Reach.py

@@ -99,6 +99,13 @@ class Reach(SQLSubModel):
 
         return ok
 
+    def _data_traversal(self,
+                        predicate=lambda obj, data: True,
+                        modifier=lambda obj, data: None,
+                        data={}):
+        for p in self._profiles:
+            p._data_traversal(predicate, modifier, data)
+
     def profile(self, i):
         """Returns profile at index i
 

src/Model/LateralContribution/LateralContribution.py

@@ -439,7 +439,6 @@ class LateralContribution(SQLSubModel):
 
         return True
 
-
     def _data_traversal(self,
                         predicate=lambda obj, data: True,
                         modifier=lambda obj, data: None,

src/Model/River.py

@@ -190,6 +190,13 @@ class RiverNode(Node):
 
         return True
 
+    def _data_traversal(self,
+                        predicate=lambda obj, data: True,
+                        modifier=lambda obj, data: None,
+                        data={}):
+        if predicate(self, data):
+            modifier(self, data)
+
     def is_deleted(self):
         return self._deleted
 
@@ -387,6 +394,16 @@ class RiverReach(Edge):
         objs = [self._reach, self._frictions]
         return self._save_submodel(execute, objs, data)
 
+    def _data_traversal(self,
+                        predicate=lambda obj, data: True,
+                        modifier=lambda obj, data: None,
+                        data={}):
+        if predicate(self, data):
+            modifier(self, data)
+
+        self._reach._data_traversal(predicate, modifier, data)
+        self._frictions._data_traversal(predicate, modifier, data)
+
     def is_deleted(self):
         return self._deleted
 

src/Model/SedimentLayer/SedimentLayer.py

@@ -40,6 +40,7 @@ class Layer(SQLSubModel):
             owner_scenario=owner_scenario
         )
 
+        self._sl = sl
         self._name = name
         self._type = type
 
@@ -58,6 +59,7 @@ class Layer(SQLSubModel):
     @name.setter
     def name(self, name):
         self._name = name
+        self.modified()
 
     @property
     def type(self):
@@ -66,6 +68,7 @@ class Layer(SQLSubModel):
     @type.setter
     def type(self, type):
         self._type = type
+        self.modified()
 
     @property
     def height(self):
@@ -74,6 +77,7 @@ class Layer(SQLSubModel):
     @height.setter
     def height(self, height):
         self._height = float(height)
+        self.modified()
 
     @property
     def d50(self):
@@ -82,6 +86,7 @@ class Layer(SQLSubModel):
     @d50.setter
     def d50(self, d50):
         self._d50 = float(d50)
+        self.modified()
 
     @property
     def sigma(self):
@@ -90,6 +95,7 @@ class Layer(SQLSubModel):
     @sigma.setter
     def sigma(self, sigma):
         self._sigma = float(sigma)
+        self.modified()
 
     @property
     def critical_constraint(self):
@@ -98,6 +104,7 @@ class Layer(SQLSubModel):
     @critical_constraint.setter
     def critical_constraint(self, critical_constraint):
         self._critical_constraint = float(critical_constraint)
+        self.modified()
 
     @classmethod
     def _db_create(cls, execute, ext=""):
@@ -199,7 +206,7 @@ class Layer(SQLSubModel):
             "SELECT pamhyr_id, deleted, name, type, height, " +
             "d50, sigma, critical_constraint, scenario " +
             "FROM sedimentary_layer_layer " +
-            f"WHERE sl = {sl} " +
+            f"WHERE sl = {sl.pamhyr_id} " +
             f"AND scenario = {scenario.id} " +
             f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))}) " +
             "ORDER BY ind ASC"
@@ -263,6 +270,10 @@ class Layer(SQLSubModel):
 
         return True
 
+    def modified(self):
+        super(Layer, self).modified()
+
+        self._sl.modified()
 
 class SedimentLayer(SQLSubModel):
     _sub_classes = [Layer]
@@ -433,7 +444,7 @@ class SedimentLayer(SQLSubModel):
             if deleted:
                 sl.set_as_deleted()
 
-            data["sl"] = sl.id
+            data["sl"] = sl
             sl._layers = Layer._db_load(execute, data)
 
             loaded.add(pid)

src/Model/Study.py

@@ -514,13 +514,12 @@ class Study(SQLModel):
 
         new.name = self.status.scenario.name + " (copy)"
 
+        self.status.scenario = new
         self.river._data_traversal(
             predicate=lambda obj, data: obj._owner_scenario == source,
             modifier=lambda obj, data: obj.set_owner_scenario(),
             data={}
         )
 
-        self.status.scenario = new
-
         self.status.set_as_editable()
         return new