我正在学习一些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!"
为什么不让用户一次性输入整个日期,并尝试验证它?
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
的有效输入。
如果你想接受几种格式,只需列出你想接受的格式字符串,然后在输入中一个接一个地尝试,直到找到一个有效的。但是要注意使用过载的问题。
为了验证电话号码,我只需要使用regexp。如果您以前从未使用过regexp,这里有一个很好的python regexp howto。地址是非常自由的形式,所以我认为除了限制长度和进行基本的安全检查之外,我不会去验证它们,尤其是如果你接受国际地址。
但总的来说,如果有一个python模块,你应该试着根据输入创建一个实例并捕捉错误,就像我在上面的例子中为时间模块所做的那样。
甚至不要尝试验证名称。为什么不呢?看这篇文章。:)
也许像colander这样的框架在这里可能会有所帮助:
http://docs.pylonsproject.org/projects/colander/en/latest/?awesome