From ccb55ff4e11c03ff3d1350c78d52d1a5e36b5ddc Mon Sep 17 00:00:00 2001 From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr> Date: Tue, 12 Dec 2023 10:22:52 +0100 Subject: [PATCH] HS: Fix pep8 and minor change. --- .../BasicHydraulicStructures/Table.py | 12 ++++---- .../BasicHydraulicStructures/Translate.py | 2 +- .../BasicHydraulicStructures/UndoCommand.py | 3 +- .../BasicHydraulicStructures/Window.py | 28 ++++++++++--------- src/View/HydraulicStructures/Window.py | 4 ++- 5 files changed, 25 insertions(+), 24 deletions(-) diff --git a/src/View/HydraulicStructures/BasicHydraulicStructures/Table.py b/src/View/HydraulicStructures/BasicHydraulicStructures/Table.py index 2bea72d2..18d7b854 100644 --- a/src/View/HydraulicStructures/BasicHydraulicStructures/Table.py +++ b/src/View/HydraulicStructures/BasicHydraulicStructures/Table.py @@ -61,15 +61,13 @@ class ComboBoxDelegate(QItemDelegate): def createEditor(self, parent, option, index): self.editor = QComboBox(parent) - lst = list( map( - lambda k: self._long_types[k], BHS_types.keys() + lambda k: self._long_types[k], + BHS_types.keys() ) ) - self.editor.addItems( - lst - ) + self.editor.addItems(lst) self.editor.setCurrentText(index.data(Qt.DisplayRole)) return self.editor @@ -159,8 +157,8 @@ class TableModel(PamhyrTableModel): question = QMessageBox(self._parent) question.setWindowTitle(self._trad['msg_type_change_title']) - question.setText(self._trad['msg_type_change_msg']) - question.setStandardButtons(QMessageBox.Cancel | QMessageBox.Ok ) + question.setText(self._trad['msg_type_change_text']) + question.setStandardButtons(QMessageBox.Cancel | QMessageBox.Ok) question.setIcon(QMessageBox.Question) res = question.exec() diff --git a/src/View/HydraulicStructures/BasicHydraulicStructures/Translate.py b/src/View/HydraulicStructures/BasicHydraulicStructures/Translate.py index 787d4cdb..42757cb6 100644 --- a/src/View/HydraulicStructures/BasicHydraulicStructures/Translate.py +++ b/src/View/HydraulicStructures/BasicHydraulicStructures/Translate.py @@ -32,7 +32,7 @@ class BasicHydraulicStructuresTranslate(PamhyrTranslate): "Change hydraulic structure type" ) - self._dict['msg_type_change_msg'] = _translate( + self._dict['msg_type_change_text'] = _translate( "BasicHydraulicStructures", "Do you want to change the hydraulic structure type and reset \ hydraulic structure values?" diff --git a/src/View/HydraulicStructures/BasicHydraulicStructures/UndoCommand.py b/src/View/HydraulicStructures/BasicHydraulicStructures/UndoCommand.py index 5448d530..97dc7bf8 100644 --- a/src/View/HydraulicStructures/BasicHydraulicStructures/UndoCommand.py +++ b/src/View/HydraulicStructures/BasicHydraulicStructures/UndoCommand.py @@ -126,8 +126,7 @@ class PasteCommand(QUndoCommand): self._bhs.reverse() def undo(self): - self._hs.delete_i(range(self._row, self._row + len(self._bhs)) - ) + self._hs.delete_i(range(self._row, self._row + len(self._bhs))) def redo(self): for r in self._bhs: diff --git a/src/View/HydraulicStructures/BasicHydraulicStructures/Window.py b/src/View/HydraulicStructures/BasicHydraulicStructures/Window.py index e31a1c91..b13d7218 100644 --- a/src/View/HydraulicStructures/BasicHydraulicStructures/Window.py +++ b/src/View/HydraulicStructures/BasicHydraulicStructures/Window.py @@ -45,7 +45,9 @@ from View.HydraulicStructures.BasicHydraulicStructures.Table import ( ) from View.Network.GraphWidget import GraphWidget -from View.HydraulicStructures.BasicHydraulicStructures.Translate import BasicHydraulicStructuresTranslate +from View.HydraulicStructures.BasicHydraulicStructures.Translate import ( + BasicHydraulicStructuresTranslate +) _translate = QCoreApplication.translate @@ -150,25 +152,25 @@ class BasicHydraulicStructuresWindow(PamhyrWindow): table = self.find(QTableView, "tableView") table.selectionModel()\ - .selectionChanged\ - .connect(self.update) + .selectionChanged\ + .connect(self.update) self._table.layoutChanged.connect(self.update) def index_selected(self): table = self.find(QTableView, "tableView") - r = table.selectionModel()\ - .selectedRows() - if len(r)>0: + r = table.selectionModel().selectedRows() + + if len(r) > 0: return r[0] else: return None def index_selected_row(self): table = self.find(QTableView, "tableView") - r = table.selectionModel()\ - .selectedRows() - if len(r)>0: + r = table.selectionModel().selectedRows() + + if len(r) > 0: return r[0].row() else: return None @@ -187,6 +189,7 @@ class BasicHydraulicStructuresWindow(PamhyrWindow): def add(self): rows = self.index_selected_rows() + if len(self._hs) == 0 or len(rows) == 0: self._table.add(0) else: @@ -194,6 +197,7 @@ class BasicHydraulicStructuresWindow(PamhyrWindow): def delete(self): rows = self.index_selected_rows() + if len(rows) == 0: return @@ -213,6 +217,7 @@ class BasicHydraulicStructuresWindow(PamhyrWindow): def _set_checkbox_state(self): row = self.index_selected_row() + if row is None: self._checkbox.setEnabled(False) self._checkbox.setChecked(True) @@ -222,6 +227,7 @@ class BasicHydraulicStructuresWindow(PamhyrWindow): def _set_basic_structure_state(self): row = self.index_selected_row() + if row is not None: self._table.enabled( row, @@ -230,7 +236,3 @@ class BasicHydraulicStructuresWindow(PamhyrWindow): def update(self): self._set_checkbox_state() - self._update_clear_plot() - - def _update_clear_plot(self): - rows = self.index_selected_rows() diff --git a/src/View/HydraulicStructures/Window.py b/src/View/HydraulicStructures/Window.py index 7c139dd3..2a071906 100644 --- a/src/View/HydraulicStructures/Window.py +++ b/src/View/HydraulicStructures/Window.py @@ -48,7 +48,9 @@ from View.HydraulicStructures.Table import ( from View.Network.GraphWidget import GraphWidget from View.HydraulicStructures.Translate import HydraulicStructuresTranslate -from View.HydraulicStructures.BasicHydraulicStructures.Window import BasicHydraulicStructuresWindow +from View.HydraulicStructures.BasicHydraulicStructures.Window import ( + BasicHydraulicStructuresWindow +) _translate = QCoreApplication.translate -- GitLab