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