From 3dad927bf7059557dd66a11fffc56b98ab2cc24c Mon Sep 17 00:00:00 2001 From: Andreas Tsouchlos Date: Thu, 4 Jan 2024 05:12:02 +0100 Subject: [PATCH] Implement category assignment --- banking_breakdown/ui.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/banking_breakdown/ui.py b/banking_breakdown/ui.py index e046d11..9d680a7 100644 --- a/banking_breakdown/ui.py +++ b/banking_breakdown/ui.py @@ -283,7 +283,14 @@ class MainWindow(QMainWindow): self._list_widget.takeItem(self._list_widget.row(selected_item)) def _assign_category(self, category: str): - print(category) + indexes = self._table_view.selectionModel().selectedRows() + + row_indices = [self._table_view.model().mapToSource(index).row() + for index in indexes] + + df = self.get_statement_data() + df.loc[row_indices, 'category'] = category + self.set_statement_data(df) def _handle_apply_click(self): category = self._list_widget.selectedItems()[0].text()