如何在Python中简化elif

时间:2018-01-11 10:33:14

标签: python-3.x

我没有很多python的经验,但我认为这可以简单得多。如果有任何可用的相同结果。使用字典映射到函数而不是所有那些elif吗?

    choice = input("Select an option: ")

    if choice == "1":
        try:
            new_contact = create_contact()
        except NotAPhoneNumberException:
            print("The phone number entered is invalid, creation aborted!")
        else:
            contacts[new_contact['name']] = new_contact
            save_contacts(contacts, filename)

    elif choice == "2":
        print_contact()

    elif choice == "3":
        search = input("Please enter name (case sensitive): ")
        try:
            print_contact(contacts[search])
        except KeyError:
            print("Contact not found")


    elif choice == "0":
        print("Ending Phone Book.\nHave a nice day!")
        break

    else:
        print("Invalid Input! Try again.")

0 个答案:

没有答案