class managercreateevent(QMainWindow, Ui_ManagerCreateEvent):
    def __init__(self, parent= None):
        super(managercreateevent, self).__init__(parent)
        self.setupUi(self)
        self.BackButton.clicked.connect(self.back_button)
        self.createButton.clicked.connect(self.create_button)
        connection = pymysql.connect(host="localhost",
                                    user="root",
                                    password="12345678",
                                    db="beltline")
        cursor = connection.cursor()
        showStaff = f"select concat(user.firstname, " ", user.lastname) from employee join user on employee.username = user.username where employee.employeetype = 'Staff'"
        df = pd.read_sql(showStaff, connection)
        model = PandasModel(df)
        self.tableView.setModel(model)

    def back_button(self):
        self.window = managermanageevent()
        self.close()
        self.window.show()


    def create_button(self):
        name = self.nameEdit.text()
        price = self.priceEdit.text()
        capacity = self.capacityEdit.text()
        minStaffRequired = self.minStaffEdittext()
        startDate = self.startDateEdit.text()
        endDate = self.endDateEdit.text()
        description = self.textBrowser.text()
        connection = pymysql.connect(host="localhost",
                                    user="root",
                                    password="12345678",
                                    db="beltline")
        cursor = connection.cursor()

        userlist = self.tableView.selectionModel().selectedRows()
        for i in userlist:
            index = i.row()

        createQuery = f"select concat(user.firstname, " ", user.lastname) from employee join user on employee.username = user.username where employee.employeetype = 'Staff'"
        cursor.execute(createQuery)
        assignStaff = cursor.fetchall()[index][0]

        # for now, hard-coding the site name as atl beltline center
        createEvent = f"insert into event (Name, StartDate, SiteName, EndDate, Price, Capacity, Description, MinStaffReq) values ('{name}', '{startDate}', 'Atlanta Beltline Center', '{endDate}', '{price}', '{capacity}', '{description}', '{minStaffRequired}')"
        connection.commit()
        self.window = managermanageevent()
        self.close()
        self.window.show()