From c744797c6fa77872614a14178f7a70519b6afd26 Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr>
Date: Tue, 3 Oct 2023 16:39:31 +0200
Subject: [PATCH] refactoring: Minor fix in geometry.

---
 src/View/Geometry/Profile/Window.py | 12 ++++++++----
 src/View/Geometry/Window.py         |  2 +-
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/View/Geometry/Profile/Window.py b/src/View/Geometry/Profile/Window.py
index 90fe1f44..e4112f40 100644
--- a/src/View/Geometry/Profile/Window.py
+++ b/src/View/Geometry/Profile/Window.py
@@ -145,7 +145,8 @@ class ProfileWindow(PamhyrWindow):
                       .row()
 
     def add(self):
-        if len(self._tablemodel.selectedIndexes()) == 0:
+        table = self.find(QTableView, "tableView")
+        if len(table.selectedIndexes()) == 0:
             self._tablemodel.insert_row(self._tablemodel.rowCount())
         else:
             row = self.index_selected_row()
@@ -153,10 +154,11 @@ class ProfileWindow(PamhyrWindow):
         self.update_plot()
 
     def delete(self):
+        table = self.find(QTableView, "tableView")
         rows = sorted(
             list(
                 set(
-                    [index.row() for index in self._tablemodel.selectedIndexes()]
+                    [index.row() for index in table.selectedIndexes()]
                 )
             )
         )
@@ -184,7 +186,8 @@ class ProfileWindow(PamhyrWindow):
     def move_down(self):
         rows = list(
             set(
-                [index.row() for index in self._tablemodel.selectedIndexes()]
+                [index.row() for index in
+                 self.find(QTableView, "tableView").selectedIndexes()]
             )
         )
 
@@ -197,7 +200,8 @@ class ProfileWindow(PamhyrWindow):
     def move_up(self):
         rows = list(
             set(
-                [index.row() for index in self._tablemodel.selectedIndexes()]
+                [index.row() for index in
+                 self.find(QTableView, "tableView").selectedIndexes()]
             )
         )
 
diff --git a/src/View/Geometry/Window.py b/src/View/Geometry/Window.py
index 2fccde41..af27773f 100644
--- a/src/View/Geometry/Window.py
+++ b/src/View/Geometry/Window.py
@@ -36,7 +36,7 @@ from PyQt5.QtWidgets import (
     QApplication, QMainWindow, QFileDialog, QCheckBox,
     QUndoStack, QShortcut, QTableView, QHeaderView,
     QAction, QSlider, QPushButton, QVBoxLayout,
-    QLabel,
+    QLabel, QAbstractItemView,
 )
 
 from View.Tools.PamhyrWindow import PamhyrWindow
-- 
GitLab