问题描述:小白学习python的第N天,继续练习。做一个模拟购票的小程序,没有用数据库和文件来存储数据,只是能够单词选择。
# hzh 每天进步一点点
# 2022/5/13 17:24
import colorama
import time
import os
import prettytable as pt
colorama.init(autoreset=True)
class rcolors:
OKRED = '\033[1;31m'
END = '\033[0m'
class Bcolors:
OKRED = '\033[1;32m'
END = '\033[0m'
def xunhuan():
dict_ticker = {'K1177': ['北京南-台北东', '18:05', '18:39', '00:34'],
'T8917': ['武汉南-天津西', '18:15', '18:49', '00:34'],
'G6023': ['拉萨南-重庆西', '18:20', '19:19', '00:59'],
'G203': ['纽约西-旧金山', '18:35', '19:09', '00:34'],
}
print(rcolors.OKRED + '车次\t出发站-到达站\t\t出发时间\t到达时间\t历时时长' + rcolors.OKRED)
for item in dict_ticker:
print(Bcolors.OKRED + item, end='\t')
for i in dict_ticker[item]:
print(Bcolors.OKRED + i, end='\t\t')
print()
train_no = input('请输入要购买的车次:')
persons = input('请输入乘车人:')
flag = True
for j in dict_ticker: #输出字典的key,作为输入高铁班次做对比
if train_no == j:
flag = True
s = f'您已购买了{train_no}次列车,'
s_info = dict_ticker[train_no]
s += s_info[0] + '' + s_info[1] + '开,' # 获取列表中的对应索引的值
print('正在查询余票.')
row_num = 13
show_ticket(row_num)
chose_num = input('请选择您的座位号,如13,5表示13排5座位(13,5):')
try:
row, column = chose_num.split(',')
order_ticket(row_num, row, column)
time.sleep(2)
print('正在出票中,请稍等...')
time.sleep(3)
print(Bcolors.OKRED + f'座位号:{row}排{column}列,{s}请{persons}尽快取走纸质车票,祝您旅途平安。【铁路客服】' + Bcolors.END)
break
except:
print('输入格式有异常,如13,5表示13排5座位13,5,请重新购买车次')
xunhuan()
else:
flag = False
# print('没有查询到该车次信息,请重新输入:')
if not flag:
print(rcolors.OKRED+'未查询到该车次信息,或者输入有误,请重新输入'+rcolors.END)
xunhuan()
def shuru():
print(rcolors.OKRED+'-----欢迎进入北京市列车购票系统-----'+rcolors.END)
print('请使用账号(admin) 密码(123456)来登录购票系统')
user = input('请输入用户名:')
passwd = input('请输入密码:')
if user=='admin' and passwd == '123456':
xunhuan()
else:
print('您输入的用户名或者密码不正确,请重新输入')
shuru()
#显示坐席
def show_ticket(row_num):
tb = pt.PrettyTable()
tb.field_names = ['行号','座位1','座位2','座位3','座位4','座位5']
for i in range(row_num):
lst = [f'第{i+1}排','有票','有票','有票','有票','有票']
tb.add_row(lst)
print(tb)
#购票
def order_ticket(row_num,row,column):
tb = pt.PrettyTable()
tb.field_names = ['行号','座位1','座位2','座位3','座位4','座位5']
for i in range(row_num):
if int(row) == i+1:
lst = [f'第{i + 1}排', '有票', '有票', '有票', '有票', '有票']
lst[int(column)] = '已售'
tb.add_row(lst)
else:
lst = [f'第{i + 1}排', '有票', '有票', '有票', '有票', '有票']
tb.add_row(lst)
print(tb)
if __name__ == '__main__':
shuru()
os.system('pause')