在Python中验证用户输入的正确方法

时间:2012-09-14 10:37:31

标签: python validation user-input

我正在通过一些Python教程工作,其中一个不断出现的问题是用户输入,我只是想检查一下我是否正确验证它并且不是很长时间。

我已经编写了下面的代码,只需要询问当天的月份和年份,但如果我需要开始询问地址电话号码等等,这会增长和增长是正常的吗?

def get_input( i ):

    while True:
        # We are checking the day
        if i == 'd':
            try:
                day = int( raw_input( "Please Enter the day: " ) )
                # If the day is not in range reprint
                if day > 0 and day < 32:
                    #Need to account for short months at some point
                    return day
                else:
                    print 'it has to be between 1 and 31'
            except ( ValueError ):
                print "It has to be a number!"
        elif i == 'm':
            # We are checking the month
            month = raw_input( 'Please enter ' +
                              'in words the month: '
                              ).strip().lower()
            if month in months: # use the dict we created
                return month
            else:
                print 'Please check you spelling!'
        elif i == 'y':
            # Now the year
            try:
                year = int( raw_input( "Please Enter the year" +
                                       "pad with 0's if needed: " ) )
                #make we have enough digits and a positive
                if year > 0 and len( year ) == 4:
                    return year
            except ( ValueError, TypeError ):
                    print "It has to be a four digit number!"

2 个答案:

答案 0 :(得分:5)

为什么不让用户一次性输入整个日期,并尝试验证它?

from time import strptime

def get_date():
    while True:
        date = raw_input("Please enter a date in DD/MM/YYYY format: ")
        try:
            parsed = strptime(date, "%d/%m/%Y")
        except ValueError as e:
            print "Could not parse date: {0}".format(e)
        else:
            return parsed[:3]

year, month, day = get_date()

这会捕获29/2/2011之类的错误,但会接受29/2/2012等有效输入。

如果您想接受多种格式,只需列出您想要接受的format strings,然后在输入中逐个尝试,直到找到有效的格式。但要注意usage overloading的问题。

为了验证电话号码,我只需要regexp。如果你从来没有使用过正则表达式,那么有一个很好的python regexp howto here。地址是非常自由的形式,所以我认为除了限制长度和进行基本的安全检查之外,我不打算验证它们,特别是如果你接受国际地址。

但总的来说,如果有一个python模块,你应该尝试根据输入创建一个实例并捕获错误,就像我在上面的例子中为时间模块所做的那样。

甚至不要尝试验证名称。为什么不?请看this文章。 :)

答案 1 :(得分:0)

也许像漏勺这样的框架在这​​里可能会有所帮助:

http://docs.pylonsproject.org/projects/colander/en/latest/?awesome