コード例 #1
0
def main_menu():
    try:
        user = User()
        while True:
            # input choice
            print("1. BDO Login")
            print("2. GPM Login")
            print("3. Member Login")
            print("Press Any Other key to Exit......")
            choice = int(input("Enter choice: "))

            if choice == 1:
                os.system('clear')
                # redirect to login page
                user_access = user.bdo_login(db)
                if user_access[0] == 'true':
                    bdo_id = user_access[1]
                    # redirect to display_panel of bdo
                    display_panel_bdo(bdo_id, db)
                else:
                    print("Wrong Credentials, login again!")
                    main_menu()

            elif choice == 2:
                os.system('clear')
                # initialize customer instance
                user_access = user.gpm_login(db)
                if user_access[0] == 'true':
                    gpm_id = user_access[1]
                    # redirect to dashboard
                    display_panel_gpm(gpm_id, db)
                else:
                    print("Wrong Credentials, login again!")
                    main_menu()

            elif choice == 3:
                os.system('clear')
                # initialize customer instance
                user_access = user.member_login(db)
                if user_access[0] == 'true':
                    member_id = user_access[1]
                    # redirect to dashboard
                    display_panel_member(member_id, db)
                else:
                    print("Wrong Credentials, login again!")
                    main_menu()
            else:
                os.system('clear')
                print("Thank You for Using This Application")
                time.sleep(2)
                sys.exit()
    except Error as se:
        db.conn.rollback()
        print("Something went wrong: {}".format(se))
    except ValueError as ve:
        print("Invalid Choice, Please enter a valid number")
    except Exception as e:
        print("exception handled: {}".format(e))