From 352c0beb2777e22d720869f9ccad69a09bf1f0e7 Mon Sep 17 00:00:00 2001
From: Theophile Terraz <theophile.terraz@inrae.fr>
Date: Mon, 11 Dec 2023 15:29:02 +0100
Subject: [PATCH] add short dict and getter in HS
---
src/Model/HydraulicStructures/Basic/Types.py | 5 +++++
src/Model/HydraulicStructures/HydraulicStructures.py | 3 +++
2 files changed, 8 insertions(+)
diff --git a/src/Model/HydraulicStructures/Basic/Types.py b/src/Model/HydraulicStructures/Basic/Types.py
index d82998ad..a976aaa3 100644
--- a/src/Model/HydraulicStructures/Basic/Types.py
+++ b/src/Model/HydraulicStructures/Basic/Types.py
@@ -35,3 +35,8 @@ class NotDefined(BasicHS):
BHSValue("bar", float, 42.0),
BHSValue("baz", int, 13),
]
+
+
+BHS_types = {
+ "ND": NotDefined,
+}
diff --git a/src/Model/HydraulicStructures/HydraulicStructures.py b/src/Model/HydraulicStructures/HydraulicStructures.py
index 5799925b..6d1a4734 100644
--- a/src/Model/HydraulicStructures/HydraulicStructures.py
+++ b/src/Model/HydraulicStructures/HydraulicStructures.py
@@ -233,6 +233,9 @@ class HydraulicStructure(SQLSubModel):
def basic_structures(self):
return self._data.copy()
+ def basic_structure(self, index: int):
+ return self._data[index]
+
def add(self, index: int):
value = NotDefined(status=self._status)
self._data.insert(index, value)
--
GitLab