import traceback

from lib.db import Users
from PyQt5.QtWidgets import QMessageBox
from ui.crud import Crud, Line_Edit_Cell_Widget
from ui.widget import Widget


class Users_Management(Widget):
    def __init__(self):
        super().__init__()

        class Username_Line_Edit_Cell_Widget(Line_Edit_Cell_Widget):
            def parse(self, row_number=None, crud=None):
                data = super().parse(row_number=row_number, crud=crud)
                if data is not None and len(data):
                    return data.upper()
                return None

        class Password_Line_Edit_Cell_Widget(Line_Edit_Cell_Widget):
            def render(self, data, row_number=None, crud=None):
                if data is not None and len(data):
                    data = u"\u2022" * 8
                else:
                    data = None
                self.setText(data)

            def parse(self, row_number=None, crud=None):
                data = self.text()
                if data is not None and len(data):
                    return data
                return None

        class Roles_Line_Edit_Cell_Widget(Line_Edit_Cell_Widget):
            def render(self, data, row_number=None, crud=None):
                super().render(", ".join(data) if data is not None else None, row_number=row_number, crud=crud)

            def parse(self, row_number=None, crud=None):
                return Users.parse_roles(self.text())

        crud_aliases = {
            "id": "Id",
            "username": "Nome utente",
            "password": "Password",
            "roles": "Ruoli",
        }
        self.crud = Crud(
            "users",
            display_name="GESTIONE UTENTI",
            readonly=["id"],
            select=list(crud_aliases.keys()),
            fields_aliases=crud_aliases,
            autocomplete={"archived": False},
            widget_classes={
                "username": Username_Line_Edit_Cell_Widget,
                "password": Password_Line_Edit_Cell_Widget,
                "roles": Roles_Line_Edit_Cell_Widget,
            },
            row_filter=self.row_filter
        )
        self.layout().addWidget(self.crud, 0, 0, -1, -1)

    def row_filter(self, row, row_number, crud):
        if row["password"] is not None and all(map(lambda x: x == u"\u2022", row["password"])):
            row.pop("password", None)
            row.pop("salt", None)
        else:
            try:
                user = Users.generate(username=row["username"], password=row["password"], roles=row["roles"])
            except AssertionError as e:
                self.log.exception(traceback.format_exc())
                crud().set_row_color(row_number, "red")
                QMessageBox.critical(None, "Errore Salvataggio DB", f"Errore alla riga {row_number}:\n{str(e)}")
                return False, None, True
            row.update(user)
        return True, row, False