import os
from PyQt5.QtWidgets import QGridLayout, QPushButton

DIR_SHARE = os.path.abspath(
    os.path.normpath(os.path.join(__file__, '..', '..', 'share')))

__all__ = ['ManageTables', 'MixButton']

STATES = {
    'available': 'rgb(215, 215, 215)',
    'occupied': 'rgb(246, 211, 70)',
    'reserved': 'rgba(80, 190, 220, 0.8)'
}

button_style = """
    min-height: 40px;
    min-width: 100px;
"""

button_style_1 = """
    font: bold 30px;
    min-width: 100px;
    text-align: center;
"""


class MixButton(QPushButton):
    def __init__(self, value, activate):
        super(MixButton, self).__init__()
        self.name = value['name']
        self.table_id = value['id']
        self.set_data(value)
        self.setText(self.name)
        self.activate = activate
        self.clicked.connect(self.activate_method)
        self.show()

    def activate_method(self):
        self.activate(self.table_id)

    def set_data(self, record):
        self.state = record['state']
        self.sale_id = record['sale']
        color = STATES[self.state]
        new_style_ = f"""
            min-height: 90px;
            min-width: 100px;
            background-color: {color};
            font-weight: bold;
            border-width: 0px;
            border-radius: 15px;
        """
        self.setStyleSheet(new_style_)


class ManageTables(QGridLayout):

    def __init__(self, parent, tables):
        super(ManageTables, self).__init__()
        self.setHorizontalSpacing(1)
        self.setVerticalSpacing(1)
        self.parent = parent
        columns = 6
        rows = int(len(tables) / columns) + 1
        self.buttons = {}
        positions = [(i, j) for i in range(rows) for j in range(columns)]
        for position, value in zip(positions, tables):
            button = MixButton(value, self.activate_method)
            self.buttons[button.table_id] = button
            self.addWidget(button, *position)

    def update_records(self, records):
        for rec in records:
            button = self.buttons[rec['id']]
            button.set_data(rec)

    def activate_method(self, table_id):
        button = self.buttons[table_id]
        if button.state == 'available':
            res = self.parent.action_assign_table(table_id, button.name)
            if res['to_add']:
                button.set_data(res['to_add'])
            if res['to_drop']:
                button = self.buttons[res['to_drop']['id']]
                button.set_data(res['to_drop'])
        else:
            self.parent.action_see_table(table_id)