from models.toy_model import ToyModel
from views.catalog_view import CatalogView


class CatalogController:
    def __init__(self, database, full_name, role, auth):
        self.database = database
        self.auth = auth
        self.role = role
        self.model = ToyModel(database)

        age_groups = self.model.get_age_groups()
        suppliers = self.model.get_suppliers()

        self.view = CatalogView(full_name, role, age_groups, suppliers)

        self.view.refresh_button.clicked.connect(self.load_data)
        self.view.sort_button.clicked.connect(self.sort_by_price)
        self.view.logout_button.clicked.connect(self.logout)

        self.load_data()

    def show(self):
        self.view.show()

    def load_data(self):
        toys = self.model.get_all()
        self.view.load_data(toys)

    def sort_by_price(self):
        toys = self.model.sort_by_price()
        self.view.load_data(toys)

    def logout(self):
        self.view.close()
        self.auth.show()