import mariadb
from PySide6.QtWidgets import QHBoxLayout, QCheckBox

from database import KontorDB


class KontorModelConfig:

    def __init__(self, kontor_db: KontorDB, main_window, table_name: str, table):
        self.header = {}
        self.filter = {}
        self.main_window = main_window
        self._table_name = table_name
        self._table = table
        self.kontor_db = kontor_db
        self.get_table_config()

    def get_table_config(self):
        self.header = self.kontor_db.get_column_meta_data(self._table_name)
        self.filter = self.kontor_db.get_filters(self._table_name)

    def get_filter(self) -> str:
        filter_rule = ""
        # print(self.filter["download"].isChecked())
        for column, filter_info in self.filter.items():
            # print(column, filter_info)
            if filter_info['widget'].isChecked():
                # print(column, filter_info, filter_rule, len(filter_rule))
                if len(filter_rule) < 1:
                    filter_rule += "WHERE "
                if len(filter_rule) > 8:
                    filter_rule += " AND "
                filter_rule += f"{column} is true"
        # print(f"{filter_rule=}")
        return filter_rule

    def filters(self) -> dict:
        _filters = {}
        # print(self.filter["download"].isChecked())
        for column, filter_info in self.filter.items():
            # print(column, filter_info)
            if filter_info['widget'].isChecked():
                _filters[column] = True
        # print(f"{filter_rule=}")
        return _filters

    def get_data(self) -> list:
        # data = self.kontor_db.get_data(self._table_name, self.header, self.get_filter())
        # data.clear()
        data = self.kontor_db.data(self._table, self.header, self.filters())
        # print(f"KontorModelConfig.get_data: {len(data)}")
        # comics = self.kontor_db.session.query(Comic).all()
        # print(f'{len(comics)} Comics loaded')
        return data

    def get_filter_layout(self) -> QHBoxLayout:
        filter_layout = QHBoxLayout()
        for column, filter_info in self.filter.items():
            filter_checkbox = QCheckBox()
            filter_checkbox.setText(filter_info['label'])
            filter_checkbox.checkStateChanged.connect(self.main_window.refresh)
            self.filter[column]['widget'] = filter_checkbox
            filter_layout.addWidget(filter_checkbox)
        filter_layout.addStretch()
        return filter_layout