while 1: menu = input( "Hello, How are you? Enter 1 to log in , 2 to sign up, 3 to exit: " ) menu = int(menu) if menu == 1: y = check() user_id = y.getUser_id() username = y.getusername() driver_id = y.getdriver_id() role = sign_in() if role == 'Dispatcher': dispatcher = Dispatcher(user_id, conn, cursor) dispatcher.dispatcher_menu() elif role == 'Account Manager': accountManager = AccountManager(username, conn, cursor) accountManager.menu() elif role == 'Driver': driver = Driver(driver_id, conn, cursor) driver.driver_menu() elif role == 'Supervisor': supervisor = Supervisor(user_id, conn, cursor) supervisor.supervisor_menu() elif menu == 2: sign_up() elif menu == 3: sys.exit() # print(role)
import sqlite3 from AccountManager import AccountManager if __name__ == "__main__": db_file_path = "waste_management.db" conn = sqlite3.connect(db_file_path) cursor = conn.cursor() manager00 = AccountManager(23451, conn, cursor) manager01 = AccountManager(23452, conn, cursor) manager00.menu()