From 8c2fa9296d660cf3bf1c5a24640c8a2822294036 Mon Sep 17 00:00:00 2001 From: Andreas Tsouchlos Date: Thu, 4 Jan 2024 15:37:28 +0100 Subject: [PATCH] Preserve sort on data reload --- banking_breakdown/ui.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/banking_breakdown/ui.py b/banking_breakdown/ui.py index fc68ad7..88ae675 100644 --- a/banking_breakdown/ui.py +++ b/banking_breakdown/ui.py @@ -116,14 +116,15 @@ class MainWindow(QMainWindow): self._list_widget.itemSelectionChanged.connect( self._handle_list_selection_changed) + self._proxyModel = QSortFilterProxyModel(self) + self._table_view.setModel(self._proxyModel) + def set_statement_data(self, df: pd.DataFrame): if 'category' not in df.columns: df['category'] = [' '] * len(df.index) model = PandasModel(df) - proxyModel = QSortFilterProxyModel(self) - proxyModel.setSourceModel(model) - self._table_view.setModel(proxyModel) + self._proxyModel.setSourceModel(model) if len(df.columns) < 10: # Experimentally determined threshold # Properly resize columns (takes longer)