Implement clearing of category upon deletion

2024-01-04 05:17:32 +01:00 · 2024-01-04 05:17:32 +01:00 · e2bf90d4d8
commit e2bf90d4d8
parent 3dad927bf7
1 changed files with 2 additions and 0 deletions
--- a/banking_breakdown/ui.py
+++ b/banking_breakdown/ui.py
@ -281,6 +281,8 @@ class MainWindow(QMainWindow):

        if button == QMessageBox.StandardButton.Yes:
            self._list_widget.takeItem(self._list_widget.row(selected_item))
+            df = self.get_statement_data()
+            df.loc[df['category'] == selected_item.text(), 'category'] = ' '

    def _assign_category(self, category: str):
        indexes = self._table_view.selectionModel().selectedRows()