Compare commits

..

No commits in common. "0346080e16e653ae354f39b526d8d036834ad5d3" and "0e1e3665a63a74d9145dd9c7562ea1740f34f04a" have entirely different histories.

5 changed files with 22 additions and 51 deletions

View File

@ -193,9 +193,9 @@ class OutputRKAdists(SQLSubModel):
"output_rk_adists(pamhyr_id, deleted, " +
"reach, rk, title, scenario) " +
"VALUES (" +
f"{self.id}, {self._db_format(self.is_deleted())}" +
f"{self.id}, {self._db_format({self.is_deleted()})}" +
f"{self._reach}, {self._rk}, " +
f"'{self._db_format(self._title)}', " +
f"'{self._db_format(self._title)}'" +
f"{self._status.scenario_id}" +
")"
)

View File

@ -547,12 +547,12 @@ class River(Graph):
data['loaded_pid'] = set()
new._Pollutants = PollutantsList._db_load(execute, data)
new._Output_rk_adists = OutputRKAdistsList._db_load(
execute, data
)
new._Pollutants = PollutantsList._db_load(execute, data)
new._InitialConditionsAdisTS = InitialConditionsAdisTSList._db_load(
execute, data)

View File

@ -93,10 +93,21 @@ class Scenario(SQLSubModel):
@classmethod
def _db_update(cls, execute, version, data=None):
major, minor, release = version.strip().split(".")
if major == minor == "0":
if int(release) < 12:
cls._db_create(execute)
cls._db_add_default(execute)
if major == "0" and int(minor) <= 2:
cls._db_create(execute)
cls._db_add_default(execute)
if major == "0" and minor == "1":
if int(release) < 2:
execute(
"ALTER TABLE scenario " +
"ADD COLUMN x REAL NOT NULL DEFAULT 1000"
)
execute(
f"ALTER TABLE scenario " +
"ADD COLUMN y REAL NOT NULL DEFAULT 1000"
)
return True

View File

@ -166,44 +166,6 @@ class StudyScenarioTestCase(unittest.TestCase):
study.reload_from_scenario(study.scenarios[nid])
self.assertEqual(len(study.river.additional_files), 2)
def test_scenario_river_nodes(self):
study = Study.new("foo", "bar")
study.river.additional_files.new(0)
self.assertNotEqual(study.river, None)
dir = tempfile.mkdtemp()
f = os.path.join(dir, "foo.pamhyr")
study.filename = f
study.save()
# Add nodes
n0 = study.river.add_node()
n1 = study.river.add_node(x=1.0, y=0.0)
n2 = study.river.add_node(x=0.0, y=1.0)
self.assertEqual(study.river.nodes_counts(), 3)
study.save()
# New scenario
new = study.new_scenario_from_current()
nid = new.id
n3 = study.river.add_node(x=0.0, y=1.0)
study.save()
self.assertEqual(study.river.nodes_counts(), 4)
study.reload_from_scenario(study.scenarios[0])
self.assertEqual(study.river.nodes_counts(), 3)
study.reload_from_scenario(study.scenarios[nid])
self.assertEqual(study.river.nodes_counts(), 4)
study.save()
class RiverTestCase(unittest.TestCase):
def test_create_river(self):

View File

@ -65,13 +65,11 @@ class ComboBoxDelegate(QItemDelegate):
.get(index.row()) \
.reach
if reach_id is not None:
reach = next(
filter(
lambda edge: edge.id == reach_id, self._data.edges()
)
)
reach = next(filter(
lambda edge: edge.id == reach_id, self._data.edges()
))
if reach_id is not None:
val = list(
map(
lambda rk: str(rk), reach.reach.get_rk()