UpdateFoodTruck_13.py

from PyQt5 import QtCore, QtWidgets
from Helpers import *
from Middle import *
from CheckableComboBox import CheckableComboBox

class UpdateFoodTruckWindow(QtWidgets.QWidget):
    toManageFoodTruck = QtCore.pyqtSignal()
    toUpdateFoodTruck = QtCore.pyqtSignal(str, str, list)
    
    def __init__(self, user, name, station, staffList):
        self.user = user
        self.truckName = name
        self.givenStation = station
        QtWidgets.QWidget.__init__(self)
        self.setWindowTitle('Window')
        self.nameTextbox = buildTextbox(False, name)
        nameLayout = buildLayout('H', [buildLabel("Name"), self.nameTextbox])
        
        stations = getStationNames()
        if station != '' : 
            stations[0] = station

        self.stationCombobox = buildComboBox(stations)
        stationLayout = buildLayout('H', [buildLabel("Station"), self.stationCombobox])

        availableStaff = viewFoodTruckAvailableStaff(self.user.username, name)
        
        if staffList == None:
            currentStaff = viewFoodTruckStaff(name)
            staffList = []
            for i in range(len(availableStaff)):
                if availableStaff[i] in currentStaff:
                    staffList.append(i)

        self.staff = buildList(availableStaff, staffList)
        # TODO autofill menu when opened
        staffLayout = buildLayout('H', [buildLabel("Assigned Staff"), self.staff])
        menuList = viewFoodTruckMenu(name)+ self.user.menuItems
        grid = buildGrid(["Food", "Price"], self.formatForGrid(menuList))

        self.foodCombobox = buildComboBox(getFoods())
        self.priceTextbox = buildTextbox()
        addButton = buildButton("+", self.addFood)
        newItemLayout = buildLayout('H', [buildLabel("Food"), self.foodCombobox, buildLabel("Price"), self.priceTextbox, addButton])

        backButton = buildButton("Back", self.back)
        updateButton = buildButton("Update", self.update)
        buttonLayout = buildLayout('H', [backButton, updateButton])

        layout = buildLayout('V', [buildLabel("Update Food Truck"), nameLayout, stationLayout, staffLayout, buildLabel("\n Menu Item"), grid, newItemLayout, buttonLayout])

        self.setLayout(layout)

    def formatForGrid(self, rows):
        newList = []
        for row in rows:
            foodName = buildLabel(row[0])
            price = buildLabel(str(row[1]))
            newList.append((foodName, price))
        return newList

    def addFood(self):
        for i in (self.user.menuItems + viewFoodTruckMenu(self.truckName)):
            if i[0] == self.foodCombobox.currentText():
                return

        if self.priceTextbox.text() == "":
            return
        try:
            price = float(self.priceTextbox.text())
            if price < 0:
                return
            fprice = '{:.2f}'.format(price)

            staffItems = self.staff.selectedItems()
            selectedIndices = []
            for item in staffItems:
                selectedIndices.append(self.staff.indexFromItem(item).row())

            self.user.menuItems.append((self.foodCombobox.currentText(), fprice))
            self.toUpdateFoodTruck.emit(self.nameTextbox.text(), self.stationCombobox.currentText(), selectedIndices)
        except ValueError:
            return
    
    def back(self):
        self.user.menuItems = []
        self.user.selectedTruck = None
        self.toManageFoodTruck.emit()

    def update(self):
        staffList = self.staff.selectedItems()
        for i in range(len(staffList)):
            staffList[i] = staffList[i].text()
        currentStaff = viewFoodTruckStaff(self.truckName)

        if self.nameTextbox.text() and self.stationCombobox.currentText() and len(staffList) != 0:
            if (self.stationCombobox.currentText() != self.givenStation) :
                if not updateFoodTruckStation(self.truckName, self.stationCombobox.currentText()):
                    QtWidgets.QMessageBox.about(self, "Create Error", "Station at full capacity.")
                    return

            for staff in staffList:
                assignStaff(self.truckName, staff)
            for staff in currentStaff:
                if staff not in staffList:
                    assignStaff(None, staff)
            for item in self.user.menuItems:
                addMenuItem(self.truckName, item[1], item[0])
            
            self.back()