an.tsouchlos/banking-breakdown
1
0
Fork 0
Code Issues 2 Pull Requests 1 Packages Projects Releases Wiki Activity

Preserve sort on data reload

Browse Source
This commit is contained in:
Andreas Tsouchlos 2024-01-04 15:37:28 +01:00
parent 0fdaff0fa2
commit 8c2fa9296d
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
banking_breakdown/ui.py
View File

@ -116,14 +116,15 @@ class MainWindow(QMainWindow):
self._list_widget.itemSelectionChanged.connect( self._list_widget.itemSelectionChanged.connect(
self._handle_list_selection_changed) self._handle_list_selection_changed)
self._proxyModel = QSortFilterProxyModel(self)
self._table_view.setModel(self._proxyModel)
def set_statement_data(self, df: pd.DataFrame): def set_statement_data(self, df: pd.DataFrame):
if 'category' not in df.columns: if 'category' not in df.columns:
df['category'] = [' '] * len(df.index) df['category'] = [' '] * len(df.index)
model = PandasModel(df) model = PandasModel(df)
proxyModel = QSortFilterProxyModel(self) self._proxyModel.setSourceModel(model)
proxyModel.setSourceModel(model)
self._table_view.setModel(proxyModel)
if len(df.columns) < 10: # Experimentally determined threshold if len(df.columns) < 10: # Experimentally determined threshold
# Properly resize columns (takes longer) # Properly resize columns (takes longer)