背景:
中午快餐,菜单吃了个遍,天天纠结于不知道点啥菜。
想起读书考试时,丢纸团选答案,于是用python写个随机点菜脚本玩玩。
功能:
菜单为Excel,一个Sheet页为一个分类,每行显示每道菜的编号、菜名、价格,如下图所示。
1、程序启动,程序提示选择分类,按输入的数字进入相应分类所在Sheet页(目前分类写死,只有3类)。
2、 程序获取Sheet页最大行数,以此为上限,随机生成一个1到最大行数之间的数字。
3、程序根据生成的随机数,读出该行的菜名、价格。
4、程序总计可随机选菜3次,3次后退出,每次提示是否继续,是则继续,否则程序立即退出。
5、若所选分类所在Sheet页总行数小于3,不能满足3次随机,则程序提示并退出。
6、输入时,程序检查输入合法性,并给出相应提示。
Python 2.7.5代码:
# coding=utf-8
import sys
import xlrd
import random
#打开菜单excel
workbook = xlrd.open_workbook('D:\\menu.xls')
#计数器为0
counters = 0
#默认继续
YesOrNo = 'y'
while 1:
#判断计数器,最多能选择3次
if counters<3:
#判断继续还是退出,默认是继续的
if YesOrNo=='y':
choose_num = raw_input('选择分类(1、小炒类|2、面食类|3、其他类):')
#判断选择的分类是否在三大类这中
if choose_num in ['1','2','3']:
#根据选择的分类,进入到相应Sheet页
excel_sheet = workbook.sheet_by_index(int(choose_num)-1)
#获取Sheet页的行数和列数
nrows_num = excel_sheet.nrows
ncols_num = excel_sheet.ncols
print
if nrows_num>3:
#生成一个随机数
random_num = random.randint(1,nrows_num-1)
#根据生成的随机数,获取到该行的菜名
name = excel_sheet.row(random_num)[1].value
#根据生成的随机数,获取到该行的价格
price = excel_sheet.row(random_num)[2].value
#输出获取到的分类、菜名、价格
print '-----第',counters+1,'次随机点菜结果:-----'.decode('utf-8')
print ' 分类:',excel_sheet.name
print ' 菜名:',name
print ' 价格:'.decode('utf-8'),price,'元'.decode('utf-8')
#计数器+1
counters+=1
print
#提示是否继续?将输入内容英文小写化
YesOrNo = raw_input('是否继续?Y或N(不分大小写):').lower()
print
else:
print '该分类行数小于3,不支持3次随机,请修改!!!'
break
#判断输入的分类是否是1、2、3
else:
print ' 只能输入1、2、3,请重新输入!'
print
#判断是否继续输入n,则退出
elif YesOrNo=='n':
print '你选择了退出!'
break
#判断是否继续输入的值,非Y或N则继续要求输入
else :
print '只能输入Y或N,Try Again:'
print
YesOrNo = raw_input('是否继续?Y或N(不分大小写):').lower()
print
#判断计数器达到3次,程序退出
else:
print
print '你已经选了三次,还不能确定要吃啥?你直接吃粑粑去吧!!!'
break
测试:
1、正常流,输入分类,打印出随机得到的结果,计数器达3次,程序退出:
2、校验‘分类’输入非法(中文、英文、分类之外数字、浮点数)时,给出相应提示:
3、 校验‘是否继续’输入非法(中文、非Y或N之外的英文、浮点数)时,给出相应提示,输入N或n,则退出 :
4、行数小于3时,不能满足3次随机,给出相应提示,程序退出:
5、重复选择同一分类,结果如下(此处存在随机数重复的Bug,有待改进):
Bug: