65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
import os
|
|
|
|
from PyQt5.QtWidgets import QGridLayout, QPushButton
|
|
|
|
DIR_SHARE = os.path.abspath(
|
|
os.path.normpath(os.path.join(__file__, '..', '..', 'share')))
|
|
|
|
__all__ = ['ManageTables']
|
|
|
|
STATES = {
|
|
'available': 'rgb(180, 180, 180)',
|
|
'occupied': 'rgb(255, 210, 30)',
|
|
'reserved': 'rgb(150, 30, 0)'
|
|
}
|
|
|
|
|
|
class CallButton(QPushButton):
|
|
|
|
def __init__(self, value, method):
|
|
super(CallButton, self).__init__(value['name'])
|
|
self.setAutoFillBackground(True)
|
|
self.name = value['name']
|
|
self.id = value['id']
|
|
self.state = value['state']
|
|
self.method = method
|
|
self.set_state(value['state'])
|
|
self.clicked.connect(self.handle_click)
|
|
|
|
def handle_click(self):
|
|
if self.state == 'available':
|
|
state = 'occupied'
|
|
else:
|
|
state = 'available'
|
|
|
|
res = self.method(self.id, self.name, self.state, state)
|
|
if not res:
|
|
return
|
|
|
|
self.set_state(state)
|
|
|
|
def set_state(self, state):
|
|
self.state = state
|
|
color = STATES[self.state]
|
|
self.setStyleSheet('background-color: {}; border:none;'.format(color))
|
|
|
|
|
|
class ManageTables(QGridLayout):
|
|
|
|
def __init__(self, parent, tables, method):
|
|
super(ManageTables, self).__init__()
|
|
self.setHorizontalSpacing(1)
|
|
self.setVerticalSpacing(1)
|
|
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 = CallButton(value, method)
|
|
self.buttons[button.id] = button
|
|
self.addWidget(button, *position)
|
|
|
|
def update_table(self, button_id, state):
|
|
button = self.buttons[button_id]
|
|
button.set_state(state)
|