diff --git a/banking_breakdown/ui.py b/banking_breakdown/ui.py index ce1d486..164b785 100644 --- a/banking_breakdown/ui.py +++ b/banking_breakdown/ui.py @@ -102,6 +102,8 @@ class MainWindow(QMainWindow): self._create_button.clicked.connect(self._handle_create_click) self._delete_button.clicked.connect(self._handle_delete_click) self._apply_button.clicked.connect(self._handle_apply_click) + self._list_widget.itemActivated.connect(self._handle_item_double_click) + self._action_save.triggered.connect(self._handle_save) header = self._table_view.horizontalHeader() @@ -204,8 +206,12 @@ class MainWindow(QMainWindow): model = self._table_view.horizontalHeader().model() column_text = model.headerData(column, Qt.Orientation.Horizontal) - new_name, _ = QInputDialog.getText(self, "Rename column", "New name:", - text=column_text) + new_name, flag = QInputDialog.getText(self, "Rename column", + "New name:", + text=column_text) + + if not flag: + return if (new_name != column_text) and (new_name != ''): df = self.get_statement_data() @@ -231,11 +237,15 @@ class MainWindow(QMainWindow): column_text = model.headerData(column, Qt.Orientation.Horizontal) df = self.get_statement_data() + columns = [column for column in df.columns if column != column_text] - other_name, _ = QInputDialog.getItem(self, "Switch column position", - f"Switch position of colum" - f" '{column_text}' with:", - df.columns, editable=False) + other_name, flag = QInputDialog.getItem(self, "Switch column position", + f"Switch position of colum" + f" '{column_text}' with:", + columns, editable=False) + + if not flag: + return column_titles = list(df.columns) index1, index2 = column_titles.index(column_text), column_titles.index( @@ -250,10 +260,13 @@ class MainWindow(QMainWindow): self.add_categories(['asdf']) def _handle_delete_click(self): - print(self._list_widget.selectedIndexes()) + print(self._list_widget.selectedItems()[0].text()) def _handle_apply_click(self): - pass + print(self._list_widget.selectedItems()[0].text()) + + def _handle_item_double_click(self, item): + print(item.text()) def _handle_save(self): filename, _ = QFileDialog.getSaveFileName(self, 'Save File')