refactoring: Minor fix in geometry.

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()]
             )
         )
 

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