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