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

Add category column if it does not exist

Browse Source
This commit is contained in:
Andreas Tsouchlos 2024-01-04 03:19:30 +01:00
parent 8a0d7f748f
commit 0816ebd26b
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
banking_breakdown/ui.py
View File

@ -121,6 +121,9 @@ class MainWindow(QMainWindow):
self._warnings.append((layout, warningIcon, label)) self._warnings.append((layout, warningIcon, label))
def set_statement_data(self, df: pd.DataFrame): def set_statement_data(self, df: pd.DataFrame):
if 'category' not in df.columns:
df['category'] = [' '] * len(df.index)
model = PandasModel(df) model = PandasModel(df)
proxyModel = QSortFilterProxyModel(self) proxyModel = QSortFilterProxyModel(self)
proxyModel.setSourceModel(model) proxyModel.setSourceModel(model)