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

Code cleanup

Browse Source
This commit is contained in:
Andreas Tsouchlos 2024-01-04 03:30:56 +01:00
parent 97e770ab0d
commit ebe5857d37
1 changed files with 36 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
banking_breakdown/ui.py
View File

@ -11,6 +11,11 @@ from PyQt6.QtWidgets import QMainWindow, QPushButton, QHBoxLayout, QLabel, \
QMessageBox, QFileDialog
#
# PandasModel
#
class PandasModel(QtCore.QAbstractTableModel):
def __init__(self, df: pd.DataFrame, parent=None):
QtCore.QAbstractTableModel.__init__(self, parent)
@ -66,6 +71,11 @@ class PandasModel(QtCore.QAbstractTableModel):
return self._data
#
# MainWindow
#
class MainWindow(QMainWindow):
def __init__(self):
super(MainWindow, self).__init__()
@ -98,25 +108,6 @@ class MainWindow(QMainWindow):
header.setContextMenuPolicy(Qt.ContextMenuPolicy.CustomContextMenu)
header.customContextMenuRequested.connect(self._header_right_clicked)
def _add_warning_text(self, text: str):
layout = QHBoxLayout()
warningIcon = QLabel()
pixmap = QPixmap("res/warning.png")
warningIcon.setPixmap(pixmap)
label = QLabel(text)
label.setWordWrap(True)
layout.addWidget(warningIcon)
layout.addWidget(label)
layout.setStretch(0, 0)
layout.setStretch(1, 1)
self._warning_layout.addLayout(layout)
self._warnings.append((layout, warningIcon, label))
def set_statement_data(self, df: pd.DataFrame):
if 'category' not in df.columns:
df['category'] = [' '] * len(df.index)
@ -138,6 +129,32 @@ class MainWindow(QMainWindow):
self._show_warnings(df)
def get_statement_data(self) -> pd.DataFrame:
return self._table_view.model().sourceModel().get_dataframe()
def add_categories(self, categories: typing.Sequence[str]):
for category in categories:
self._list_widget.addItem(category)
def _add_warning_text(self, text: str):
layout = QHBoxLayout()
warningIcon = QLabel()
pixmap = QPixmap("res/warning.png")
warningIcon.setPixmap(pixmap)
label = QLabel(text)
label.setWordWrap(True)
layout.addWidget(warningIcon)
layout.addWidget(label)
layout.setStretch(0, 0)
layout.setStretch(1, 1)
self._warning_layout.addLayout(layout)
self._warnings.append((layout, warningIcon, label))
def _show_warnings(self, df: pd.DataFrame):
for (layout, icon, text) in self._warnings:
icon.hide()
@ -162,13 +179,6 @@ class MainWindow(QMainWindow):
" rename the column containing the balance"
" after each transaction to 'balance'")
def get_statement_data(self) -> pd.DataFrame:
return self._table_view.model().sourceModel().get_dataframe()
def add_categories(self, categories: typing.Sequence[str]):
for category in categories:
self._list_widget.addItem(category)
def _header_right_clicked(self, pos):
context = QMenu(self)