from PyQt5 import QtCore, QtGui, QtWidgets


class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(800, 600)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.tableWidget = QtWidgets.QTableWidget(self.centralwidget)
        self.tableWidget.setGeometry(QtCore.QRect(60, 100, 551, 331))
        self.tableWidget.setObjectName("tableWidget")
        self.tableWidget.setColumnCount(3)
        self.tableWidget.setRowCount(0)
        item = QtWidgets.QTableWidgetItem()
        self.tableWidget.setHorizontalHeaderItem(0, item)
        item = QtWidgets.QTableWidgetItem()
        self.tableWidget.setHorizontalHeaderItem(1, item)
        item = QtWidgets.QTableWidgetItem()
        self.tableWidget.setHorizontalHeaderItem(2, item)
        self.pushButton = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton.setGeometry(QtCore.QRect(60, 450, 75, 23))
        self.pushButton.setObjectName("pushButton")
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 21))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)

        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
        item = self.tableWidget.horizontalHeaderItem(0)
        item.setText(_translate("MainWindow", "Col 1"))
        item = self.tableWidget.horizontalHeaderItem(1)
        item.setText(_translate("MainWindow", "Col 2"))
        item = self.tableWidget.horizontalHeaderItem(2)
        item.setText(_translate("MainWindow", "Col 3"))
        self.pushButton.setText(_translate("MainWindow", "PushButton"))

class mainWindow(Ui_MainWindow, QtWidgets.QMainWindow):
    def __init__(self, parent=None):
        super(mainWindow, self).__init__(parent)
        self.setupUi(self)
        self.pushButton.clicked.connect(self.populate_table)

    # Combo box creator templates
    def comboBox_populator(self,values):
        combo = QtWidgets.QComboBox()
        for element in values:
            combo.addItem(element)
        return combo

    # Action to do on vehcile change
    def onComboChange(self):
        # selVehNames = self.selVehNames
        self.switchTo = self.comboBoxRows[self.counter][0].currentText()
        if self.switchTo == [] or self.switchTo == '':
            pass
        else:
            if self.switchTo == 'a':
                col1Data3 = ['x','y','z']
            elif self.switchTo == 'b':
                col1Data3 = ['4','5','6']
            elif self.switchTo == 'c':
                col1Data3 = ['21','22','23']
            else:
                pass
            # Update dict with comboboxes
            # col1Data1 = self.comboBoxRows[self.counter][0]
            # col1Data2 = self.comboBoxRows[self.counter][1]
            # self.AllComboBoxes = [col1Data1, col1Data2, col1Data3]

            colDataBox3 = self.comboBox_populator(col1Data3)
            print(col1Data3)
            # self.comboBoxRows.update({self.counter:self.AllComboBoxes})
            self.tableWidget.setCellWidget(self.counter,2,colDataBox3)

    # Populate the table
    def populate_table(self):
        self.AllComboBoxes = []
        self.comboBoxRows = {}

        # Table row information
        rowCountcur = self.tableWidget.currentRow()
        self.tableWidget.insertRow(rowCountcur+1)
        self.counter = rowCountcur + 1

        # General Data
        col1Data1 = ['a','b','c']
        col1Data2 = ['e','f','g']
        col1Data3 = ['x','y','z']

        # Create combobox
        colDataBox1 = self.comboBox_populator(col1Data1)
        colDataBox2 = self.comboBox_populator(col1Data2)
        colDataBox3 = self.comboBox_populator(col1Data3)
        self.AllComboBoxes = [colDataBox1, colDataBox2, colDataBox3]
        self.comboBoxRows.update({self.counter:self.AllComboBoxes})

        # Populate columns
        self.tableWidget.setCellWidget(self.counter,0,self.comboBoxRows[self.counter][0])
        # Populate vehicle list and cell, col 2
        self.tableWidget.setCellWidget(self.counter,1,self.comboBoxRows[self.counter][1])
        # Populate mission order list and cell, col 3
        self.tableWidget.setCellWidget(self.counter,2,self.comboBoxRows[self.counter][2])

        self.comboBoxRows[self.counter][0].currentTextChanged.connect(self.onComboChange)

if __name__ == "__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    w = mainWindow()
    w.show()
    sys.exit(app.exec_())