diff --git a/src/View/Geometry/Profile/Window.py b/src/View/Geometry/Profile/Window.py index 90fe1f44b054efea646e54ad4ebe4a0116a67405..e4112f409f74684f428d62d58bb141c9d841c777 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 2fccde41ab74c8754038ca2c51cb162b9dc132aa..af27773f0051da081f4612c641c79ab748bc4c0a 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