python火车余票查询（s1）

import datetime
#车站仅限西安、北京（pinyin）
print('---欢迎使用火车票查询系统---')
print('本系统可查询未来7天内火车票余票信息')
#输入查询条件
from_station = input('请输入出发车站（xian）：')
to_station = input('请输入到达车站（beijing）：')
from_date = input('请输入出发日期（2018-01-01）：')
#判断输入合法性
valid_from_station = False
valid_to_station = False
valid_from_date = False
while valid_from_station==False and valid_to_station==False and valid_from_date==False:
    #判断输入出发车站合法性
    if from_station == 'xian' or from_station == 'beijing':
        valid_from_station = True

    #判断输入到达车站合法性    
    if to_station == 'xian' or to_station == 'beijing':
        valid_to_station = True

    #判断输入出发时间合法性
    today = datetime.date.today()
    year = int(from_date[:4])
    month = int(from_date[5:7])
    day = int(from_date[8:])
    qday = datetime.date(year,month,day)
    lastday = today + datetime.timedelta(days=7)
    if today <= qday and qday <= lastday:
        valid_from_date = True
        
    if valid_from_station == False:
        from_station = input('请重新输入出发车站（xian）：')
    if valid_to_station == False:
        to_station = input('请重新输入到达车站（beijing）：')
    if valid_from_date == False:
        from_date = input('请重新输入出发日期（2018-01-01）：')

#执行查询
print('正在查询.......')

#显示查询结果
print('你的出发站：', from_station)
print('你的到达站：', to_station)
print('你的出发时间：', from_date)