import typing
from functools import partial

import pandas as pd
from PyQt6.QtGui import QPixmap
from PyQt6.QtWidgets import QMainWindow, QTableView, QHBoxLayout, QLabel
from PyQt6 import uic, QtGui, QtCore
from PyQt6.QtCore import Qt

from banking_breakdown.ui.main_window import Ui_MainWindow


class PandasModel(QtCore.QAbstractTableModel):
    def __init__(self, df: pd.DataFrame, parent=None):
        QtCore.QAbstractTableModel.__init__(self, parent)
        self._data = df

        self._horizontalHeaders = [''] * len(df.columns)

        for i, column in enumerate(df.columns):
            self.setHeaderData(i, Qt.Orientation.Horizontal, column)

    def setHeaderData(self, section, orientation, data,
                      role=Qt.ItemDataRole.EditRole):

        if ((orientation == Qt.Orientation.Horizontal)
                and ((role == Qt.ItemDataRole.DisplayRole)
                     or (role == Qt.ItemDataRole.EditRole))):

            self._horizontalHeaders[section] = data
            return True
        else:
            return super().setHeaderData(section, orientation, data, role)

    def headerData(self, section, orientation,
                   role=Qt.ItemDataRole.DisplayRole):

        if (orientation == Qt.Orientation.Horizontal
                and role == Qt.ItemDataRole.DisplayRole):

            return self._horizontalHeaders[section]
        else:
            return super().headerData(section, orientation, role)

    def rowCount(self, parent=None):
        return len(self._data.values)

    def columnCount(self, parent=None):
        return self._data.columns.size

    def data(self, index, role=Qt.ItemDataRole.DisplayRole):
        if not index.isValid():
            return QtCore.QVariant()
        if role != Qt.ItemDataRole.DisplayRole:
            return QtCore.QVariant()

        item = self._data.iloc[index.row()].iloc[index.column()]

        if type(item) is pd.Timestamp:
            return QtCore.QVariant(item.strftime('%Y-%m-%d'))
        else:
            return QtCore.QVariant(str(item))


class GeneratedWindowWrapper(QMainWindow):
    def __init__(self):
        super(GeneratedWindowWrapper, self).__init__()

        self._window = Ui_MainWindow()
        self._window.setupUi(self)

        # Set window behavior

        self._window.statementTableView.setSelectionBehavior(
            QTableView.SelectionBehavior.SelectRows)

        # Populate categories

        self._category_model = QtGui.QStandardItemModel()
        self._window.categoryListView.setModel(self._category_model)

        # Onclick handlers

        self._window.createCategoryButton.clicked.connect(
            self._handle_create_click)

    def add_categories(self, categories: typing.Sequence[str]):
        for category in categories:
            item = QtGui.QStandardItem(category)
            self._category_model.appendRow(item)

    def set_statement_data(self, df: pd.DataFrame):
        model = PandasModel(df)
        self._window.statementTableView.setModel(model)

    def _handle_create_click(self):
        self.add_categories(['asdf'])

    def _handle_delete_click(self):
        pass

    def _handle_apply_click(self):
        pass

    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._window.warningLayout.addLayout(layout)