python - 创建我自己的错误消息？但是哪里？

Question

我已经创建了这个计算器，但是现在我想通过创建诸如“请仅输入数字”之类的错误消息来改进它。但是我应该在哪里添加这个呢？

这是我的代码:

def menu():
    print "Welcome to the calculator"
    print "The options are 1) Addition 2)Subtraction 3)Multiplication 4)Division 5)Exit        Calculator"
    return input("Choose your option")

def add(a,b):
    print a+b

def sub(a,b):
    print a-b

def mult(a,b):
    print a*b

def div(a,b):
    print a/b

loop=1
choice=0
while loop==1:
    choice=menu()
    if choice==1:
        add(input("Enter first number"),input("Enter second number"))
    elif choice==2:
         sub(input("Enter first number"),input("Enter second number"))
    elif choice==3:
          mult(input("Enter first number"),input("Enter second number"))
    elif choice==4:
          div(input("Enter first number"),input("Enter second number"))

    elif choice==5:
                loop=0
                exit()

谢谢:-)

Answer 1

基本思想是使用raw_input而不是输入。这给了我们一个来自用户的字符串，我们测试它是否是一个数字。如果这不是一个数字，我们会再次询问。

def myInput(message):

    # loop forever / until we have a valid input
    while True:

        # ask user for input
        user_input = raw_input(message)

        # check if its a number
        try:
            result = float(user_input)

            # Valid input, return the result
            return result

        except ValueError:
            # It couldn't be converted to a number, ask again
            print "Please only enter numbers"

            # Repeat the loop / ask again
            continue

然后在您当前使用 input('some message') 的地方使用 myInput('some message')。