From 65c1163a556ce6d925f61654727c7965339260d9 Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr>
Date: Wed, 29 Nov 2023 16:41:05 +0100
Subject: [PATCH] Reservoir: Allow to select only internal node.

---
 src/View/Reservoir/Table.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/View/Reservoir/Table.py b/src/View/Reservoir/Table.py
index 66cdae82..14362050 100644
--- a/src/View/Reservoir/Table.py
+++ b/src/View/Reservoir/Table.py
@@ -56,9 +56,22 @@ class ComboBoxDelegate(QItemDelegate):
     def createEditor(self, parent, option, index):
         self.editor = QComboBox(parent)
 
+        nodes = list(
+            map(
+                lambda n: n.name,
+                filter(
+                    lambda n: not (
+                        self._data.is_upstream_node(n)
+                        or self._data.is_downstream_node(n)
+                    ),
+                    self._data.nodes()
+                )
+            )
+        )
+
         self.editor.addItems(
             [_translate("Reservoir", "Not associated")] +
-            self._data.nodes_names()
+            nodes
         )
 
         self.editor.setCurrentText(index.data(Qt.DisplayRole))
-- 
GitLab