输入序号查找假日

# 以下代码为提示框架
# 请在...处使用一行或多行代码替换
#
# 注意：提示框架代码可以任意修改，以完成程序功能为准

fi = open('PY301-vacations.csv', 'r')
ls = []
li = {}
for line in fi:
    ls.append(line.strip("\n").split(','))
for line in ls:
    li[line[0]] = line[0]
fi.close()
ss = input('请输入节假日序号:').split(' ')
while True:
    for s in ss:
        flag = li.get(str(s), 0)
        if flag == False:
            print("输入节假日编号有误！")
        else:
            for line in ls:
                if s == line[0]:
                    print("{}({})假期是{}月{}日至{}月{}日之间".format(line[1], line[0], line[2][:-2], line[2][-2:], line[3][:-2], line[3][-2:]))
    ss = input('请输入节假日序号:').split(' ')