289 lines
11 KiB
Python
289 lines
11 KiB
Python
# -*- coding: utf-8 -*-
|
|
import wx
|
|
import wx.lib.buttons
|
|
import wx.lib.platebtn
|
|
from plot import *
|
|
from database import *
|
|
from settings import *
|
|
import flask as fla
|
|
if settings.enableRFID:
|
|
import rfid
|
|
|
|
|
|
class MainWindow(wx.Frame):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(MainWindow, self).__init__(*args, **kwargs)
|
|
self.SetSize((480, 320))
|
|
self.SetTitle('Baroness Control')
|
|
|
|
self.panelStart = PanelStart(self)
|
|
self.panelDrinks = PanelDrinks(self)
|
|
self.panelUsers = PanelUsers(self)
|
|
self.panelThanks = PanelThanks(self)
|
|
self.panelSorry = PanelSorry(self)
|
|
self.panelRFID = PanelRFID(self)
|
|
|
|
if settings.enableRFID:
|
|
self.rfid = rfid.RFID(self.on_rfid)
|
|
|
|
self.user = User()
|
|
self.rfidid = ""
|
|
self.drinkl = ""
|
|
|
|
self.active = 0
|
|
self.switchPanels()
|
|
|
|
self.Show()
|
|
|
|
def onStart(self, e):
|
|
self.active = 2
|
|
self.switchPanels()
|
|
|
|
def delayExit(self, e=None):
|
|
wx.CallLater(5000, self.onExit)
|
|
|
|
def getUser(self):
|
|
return self.user.id
|
|
|
|
def onExit(self, e=None):
|
|
self.active = 0
|
|
self.switchPanels()
|
|
|
|
def onUser(self, e):
|
|
longn = e.GetEventObject().GetLabelText()
|
|
if not longn:
|
|
return
|
|
for user in get_users():
|
|
if user.longname == longn:
|
|
self.user = user
|
|
if self.user.isblack:
|
|
self.active = 4 #Sorry Bro
|
|
else:
|
|
self.active = 1 #Drinks
|
|
self.switchPanels()
|
|
|
|
def on_rfid(self, rfidid):
|
|
if self.active != 0:
|
|
return
|
|
self.rfidid = rfidid
|
|
self.user = get_user_by_rfid(rfidid)
|
|
if self.user is None:
|
|
self.active = 5 #Screen showing RFID ID
|
|
elif self.user.isblack:
|
|
self.active = 4 #Sorry Bro
|
|
else:
|
|
self.active = 1 #Drinks
|
|
logging.info("switching panels: " + str(self.active))
|
|
#has to be called from main thread!!
|
|
wx.CallAfter(self.switchPanels)
|
|
|
|
def onProduct(self, e):
|
|
self.active = 3
|
|
logging.info(self.user.longname + ' consumes')
|
|
self.drinkl = e.GetEventObject().GetLabelText()
|
|
drink = get_product_by_name(self.drinkl.split('\n')[0]).id
|
|
with app.app_context():
|
|
for i in range(0, int(self.panelDrinks.GetAmount())):
|
|
add_consume(self.user.name, drink)
|
|
plot_all_thread(self.user)
|
|
self.switchPanels()
|
|
|
|
def switchPanels(self):
|
|
active = self.active
|
|
self.panelStart.Hide()
|
|
self.panelDrinks.Hide()
|
|
self.panelUsers.Hide()
|
|
self.panelThanks.Hide()
|
|
self.panelSorry.Hide()
|
|
self.panelRFID.Hide()
|
|
if active == 0:
|
|
if settings.enableRFID:
|
|
self.rfid.start()
|
|
self.panelStart.Show()
|
|
elif active == 1:
|
|
if not settings.onlyOneDrink:
|
|
self.panelDrinks.l_amount.SetLabel("%02d" % 1)
|
|
self.panelDrinks.l_user.SetLabel(self.user.longname)
|
|
self.panelDrinks.Show()
|
|
elif active == 2:
|
|
self.panelUsers.Show()
|
|
elif active == 3:
|
|
self.panelThanks.label_1.SetLabel(self.user.longname + "\n" + "%02d x " % int(self.panelDrinks.GetAmount()) + self.drinkl.split('\n')[0])
|
|
self.panelThanks.label_1.Wrap(340)
|
|
try:
|
|
self.panelThanks.bitmap_2.SetBitmap(wx.Bitmap("./app/static/product_%s.png" % self.drinkl.split('\n')[0], wx.BITMAP_TYPE_ANY))
|
|
except:
|
|
logging.error("no picture for drink: " + self.drinkl.split('\n')[0])
|
|
self.panelThanks.Show()
|
|
self.delayExit()
|
|
elif active == 4:
|
|
self.panelSorry.label_1.SetLabel(self.user.longname)
|
|
self.panelSorry.Show()
|
|
else: #active == 5:
|
|
self.panelRFID.label_1.SetLabel(self.rfidid)
|
|
self.panelRFID.Show()
|
|
|
|
|
|
class PanelStart (wx.Panel):
|
|
|
|
def __init__(self, parent):
|
|
wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
|
|
self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/start.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
|
|
if not settings.hideGuiList:
|
|
self.Bind(wx.EVT_LEFT_DOWN, parent.onStart)
|
|
self.bitmap_1.Bind(wx.EVT_LEFT_DOWN, parent.onStart)
|
|
|
|
|
|
class PanelThanks (wx.Panel):
|
|
|
|
def __init__(self, parent):
|
|
wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
|
|
|
|
self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/thanks.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
|
|
self.bitmap_2 = wx.StaticBitmap(self, wx.ID_ANY, wx.NullBitmap, pos=(10, 10))
|
|
|
|
self.label_1 = wx.StaticText(self, wx.ID_ANY, 'bla blub', pos=(120, 50), size=(340, 100))
|
|
self.label_1.SetFont(wx.Font(25, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
self.label_1.SetForegroundColour("white")
|
|
|
|
|
|
class PanelSorry (wx.Panel):
|
|
|
|
def __init__(self, parent):
|
|
wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
|
|
self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/sorry.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
|
|
self.bitmap_1.Bind(wx.EVT_LEFT_DOWN, parent.onExit)
|
|
self.label_1 = wx.StaticText(self, wx.ID_ANY, 'bla blub', pos=(100,100))
|
|
self.label_1.SetFont(wx.Font(30, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
|
|
|
|
class PanelRFID (wx.Panel):
|
|
|
|
def __init__(self, parent):
|
|
wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
|
|
self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/rfid.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
|
|
self.bitmap_1.Bind(wx.EVT_LEFT_DOWN, parent.onExit)
|
|
self.label_1 = wx.StaticText(self, wx.ID_ANY, 'bla blub', pos=(100,100), size=(100,220))
|
|
self.label_1.SetFont(wx.Font(30, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
|
|
|
|
class PanelDrinks (wx.Panel):
|
|
|
|
def __init__(self, parent):
|
|
wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
|
|
|
|
self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/products.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
|
|
|
|
products = get_products()
|
|
buttonids = []
|
|
i = 0
|
|
for product in products:
|
|
if product.isshown:
|
|
#480x320
|
|
self.but = wx.Button(self, id=wx.ID_ANY, label=product.name + u"\n" + u"%0.2f" % product.price, pos=((i%4)*120, int(i/4)*120), size=(120, 120))
|
|
self.but.SetBackgroundColour((255-((i+int(i/4))*20 % 40), (160+((i+int(i/4))*50 % 100)), 0))
|
|
#self.but.SetForegroundColour("#006699")
|
|
self.but.SetFont(wx.Font(23, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
self.but.Bind(wx.EVT_LEFT_DOWN, parent.onProduct, id=self.but.Id)
|
|
i += 1
|
|
|
|
if not settings.onlyOneDrink:
|
|
self.b_less = wx.Button(self, id = wx.ID_ANY, label=u"-", pos=(0,250), size=(120, 70))
|
|
self.b_less.SetFont(wx.Font(60, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
self.b_less.Bind(wx.EVT_LEFT_DOWN, self.onLess, id=self.b_less.Id)
|
|
|
|
self.l_amount = wx.StaticText(self, wx.ID_ANY, "%2d" % 0, pos=(137, 245), style=wx.ALIGN_CENTER)
|
|
self.l_amount.SetFont(wx.Font(50, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Sans"))
|
|
|
|
self.b_more = wx.Button(self, id = wx.ID_ANY, label=u"+", pos=(240,250), size=(120, 70))
|
|
self.b_more.SetFont(wx.Font(60, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
self.b_more.Bind(wx.EVT_LEFT_DOWN, self.onMore, id=self.b_more.Id)
|
|
|
|
self.b_exit = wx.Button(self, id = wx.ID_ANY, label=u"x", pos=(360,250), size=(120, 70))
|
|
self.b_exit.SetFont(wx.Font(30, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
self.b_exit.Bind(wx.EVT_LEFT_DOWN, parent.onExit, id=self.b_exit.Id)
|
|
|
|
if i <= 4:
|
|
self.l_user = wx.StaticText(self, wx.ID_ANY, "todo", pos=(20,170), style=wx.ALIGN_CENTER)
|
|
else:
|
|
self.l_user = wx.StaticText(self, wx.ID_ANY, "todo", pos=((i%4)*120+20,170), style=wx.ALIGN_CENTER)
|
|
self.l_user.SetFont(wx.Font(25, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
|
|
def onMore(self,e, id=-1):
|
|
self.amount = int(self.l_amount.GetLabelText()) + 1
|
|
if self.amount <= settings.drinkLimit:
|
|
self.l_amount.SetLabel("%02d" % self.amount)
|
|
self.Layout()
|
|
|
|
def onLess(self,e, id=-1):
|
|
self.amount = int(self.l_amount.GetLabelText()) - 1
|
|
if self.amount > 0:
|
|
self.l_amount.SetLabel( "%02d" % self.amount)
|
|
self.Layout()
|
|
|
|
def GetAmount(self):
|
|
return int(self.l_amount.GetLabelText())
|
|
|
|
class PanelUsers (wx.Panel):
|
|
|
|
def __init__(self, parent):
|
|
wx.Panel.__init__(self, parent, id=wx.ID_ANY, pos=(0, 0), size=(480, 320))
|
|
self.parent = parent
|
|
self.bitmap_1 = wx.StaticBitmap(self, wx.ID_ANY, wx.Bitmap("./gui/users.png", wx.BITMAP_TYPE_ANY), pos=(0, 0))
|
|
|
|
users = get_users()
|
|
self.names = list()
|
|
for user in users:
|
|
if user.isshown and not user.onlyrfid:
|
|
self.names.append(user.longname)
|
|
|
|
self.but_names = list()
|
|
self.position = 0
|
|
self.lines = 4
|
|
self.items = len(self.names)
|
|
|
|
for i in range(self.lines):
|
|
#480x320
|
|
self.but = wx.lib.platebtn.PlateButton(self, label=self.names[self.position+i] if self.position+i < len(self.names) else "", pos=(3, 3+i*80),
|
|
style=wx.BU_EXACTFIT | wx.lib.platebtn.PB_STYLE_SQUARE)
|
|
self.but.SetSize((392, 74))
|
|
|
|
self.but.SetBackgroundColour((255-(i*20 % 40), (160+(i*50 % 100)), 0))
|
|
#self.but.SetForegroundColour("#006699")
|
|
self.but.SetPressColor(wx.Colour(255,255,255,0))
|
|
self.but.SetFont(wx.Font(25, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
self.but.Bind(wx.EVT_LEFT_DOWN, parent.onUser, id=self.but.Id)
|
|
self.but_names.append(self.but)
|
|
|
|
self.b_up = wx.Button(self, id=wx.ID_ANY, label=u"\u25B2", pos=(400, 0), size=(80, 80))
|
|
self.b_up.SetFont(wx.Font(20, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
self.b_up.Bind(wx.EVT_LEFT_DOWN, self.on_up, id=self.b_up.Id)
|
|
|
|
self.b_down = wx.Button(self, id=wx.ID_ANY, label=u"\u25BC", pos=(400, 240), size=(80, 80))
|
|
self.b_down.SetFont(wx.Font(20, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
self.b_down.Bind(wx.EVT_LEFT_DOWN, self.on_down, id=self.b_down.Id)
|
|
|
|
self.b_exit = wx.Button(self, id=wx.ID_ANY, label="X", pos=(400, 120), size=(80, 80))
|
|
self.b_exit.SetFont(wx.Font(20, wx.DEFAULT, wx.NORMAL, wx.NORMAL, 0, "Humor Sans"))
|
|
self.b_exit.Bind(wx.EVT_LEFT_DOWN, parent.onExit, id=self.b_exit.Id)
|
|
self.Layout()
|
|
|
|
def on_down(self,e , id=-1):
|
|
if self.position+self.lines >= self.items:
|
|
return
|
|
self.position += self.lines
|
|
for i, button in enumerate(self.but_names):
|
|
button.SetLabel(self.names[self.position+i] if self.position+i < self.items else "")
|
|
button.Hide()
|
|
button.Show()
|
|
|
|
def on_up(self,e , id=-1):
|
|
if self.position-self.lines < 0:
|
|
return
|
|
self.position -= self.lines
|
|
for i, button in enumerate(self.but_names):
|
|
button.SetLabel(self.names[self.position+i] if self.position+i < self.items else "")
|
|
button.Hide()
|
|
button.Show()
|