diff --git a/src/Model/Tools/PamhyrDB.py b/src/Model/Tools/PamhyrDB.py index 4dc7035e..24c521b9 100644 --- a/src/Model/Tools/PamhyrDB.py +++ b/src/Model/Tools/PamhyrDB.py @@ -202,6 +202,24 @@ class SQLSubModel(PamhyrID): self._owner_scenario = self._status.scenario_id self._status.modified() + + def set_owner_scenario_if(self, predicate): + """Set study status the object owner_scenario to current + scenario if predicate(scenario_id) is true + + Args: + predicate: Function predicate on owner_scenario id + + Returns: + Nothing + """ + if self._status is None: + return + + if predicate(self._owner_scenario): + self._owner_scenario = self._status.scenario_id + self._status.modified() + def is_deleted(self): """This object is deleted?