一、设计要求
设计开发一个公交信息查询系统，提供公交的站点查询、线路查询、换乘查询功能
二、程序功能模块设计
主界面模块：完成系统菜单的显示
功能选择模块：根据用户的选择，完成其相应的功能
路线查询模块：根据用户提供的线路，查询改线路所经过的站点
站点查询的模块：根据用户提供的站点，查询经过该站点的所有线路
后台管理员的模块：后台管理员可以进行一些操作
后台管理员的账号系统管理的模块：注册后台管理员的账号和密码
换乘信息的模块：根据用户提供的起点和终点，帮助用户找到最便捷的线路
==代码分布==
主函数
管理员账号管理界面的函数
用户登录的函数
后台管理员界面的函数
定义修改路线的函数
判断是否存在路线的函数
线路查询的函数
判断站点是否存在的函数
站点查询的函数
检查添加线路是否存在的函数
添加线路的函数
清楚所有线路的函数
展示所有线路的函数
删除线路的函数
判断是否存在线路的函数
展示特定线路中的站点的函数
换乘信息函数
判断线路直达的函数
存储数据的函数
断站点所在的线路的函数
import os
import pickle
import csv
with open('公交.csv','r',encoding='utf-8')as file_obj:
reader = csv.reader(file_obj)
for r in reader:
print(r)
def main(): # 主函数
while True:
print('-' * 30)
print('------菜单------')
print('1.线路查询')
print('2.站点查询')
print('3.换乘信息')
print('4.后台管理员界面')
print('5.管理员的账号管理界面')
print('6.退出程序')
print('-' * 30)
num = int(input('请选择你的操作：\n'))
if num == 1:
line_find()
continue
if num == 2:
site_find()
continue
if num == 3:
change_line()
continue
if num == 5:
manager_account()
continue
if num == 6:
break
if num == 4:
manager()
continue
else:
print('请重新做出你的选择！！')
def manager_account(): # 管理员账号管理界面的函数
def new_():
file_path = 'D:\公交查询系统\info.pkl'
a = input('请设置你的账号：')
b = input('请设置入你的密码：')
w = input('请再次输入你的的密码：')
d = {
'user': a,
'password': b
}
if b == w:
if os.path.exists(file_path) and os.path.getsize(file_path):
o = open('info.pkl', 'rb')
m = pickle.load(o)
with open('info.pkl', 'ab') as fp:
pickle.dump(d, fp)
print('您已经注册成功！！')
else:
with open('info.pkl', 'wb') as fp:
pickle.dump(d, fp)
print('您已经注册成功！！')
else:
print('对不起，你两次输入的密码不一致，请您重新输入！！')
def xiaohui():
h = open('info.pkl', 'w+') # 清空文件里的内容。。。
h.truncate()
print('你已经销毁成功。。。')
while True:
print('-----管理员的账号管理界面------\n'
'1.注册\n'
'2.注销原有的所有账户\n'
'3.退出管理员的账号管理界面')
choice = input('请做出你的选择：\n')
if choice == '1':
new_()
if choice == '2':
xiaohui()
elif choice == '3':
print('-' * 35)
print('------菜单------')
print('1.线路查询')
print('2.站点查询')
print('3.换乘信息')
print('4.后台管理员界面')
print('5.管理员的账号管理界面')
print('6.退出程序')
print('-' * 35)
break
else:
print('请重新做出你的选择')
def login(f): # 用户登录的函数
def inner(*args, **kwargs):
with open('info.pkl', 'rb') as fp:
q = input('请输入你的的账号：')
w = input('请输入你的的密码：')
while True:
try:
data = pickle.load(fp)
if q == data['user'] and w == data['password']:
print('您已经登录成功！！！')
res = f(*args, **kwargs)
return res
except:
print('对不起，你登录失败，不可以让您进入后台管理员界面')
break
return inner
# 装饰器用于检验用户登录
@login
def manager(): # 后台管理员界面的函数
while True:
print('-' * 30)
print('-----后台管理员界面-----')
print('1.添加线路')
print('2.删除线路')
print('3.修改线路')
print('4.查看所有线路')
print('5.删除所有线路')
print('6.退出')
print('-' * 30)
num = input('请做出你的选择:\n')
if num == '3':
revise_line()
continue
if num == '1':
add_line()
continue
if num == '6':
print('-' * 25)
print('------菜单------')
print('1.线路查询')
print('2.站点查询')
print('3.换乘信息')
print('4.后台管理员界面')
print('5.管理员的账号管理界面')
print('6.退出程序')
print('-' * 25)
break
if num == '5':
clear_line_all()
continue
if num == '4':
show_line_all()
continue
if num == '2':
num = input('请输入你要删除的线路：\n').strip()
clear_line(num)
continue
else:
print('请重新做出你的选择...')
def revise_line(): # 定义修改线路的函数
num = input('请输入您要修改的线路：').strip()
if check_add_line(num):
show_line_all()
tag = input('请确认是否继续修改线路：(1为是，0为否)\n')
if tag == '1':
file_path = 'D:\公交查询系统\line_info.pkl'
a = input('请再次输入要修改的线路：').strip()
clear_line(a)
b = input('请输入经过的站点：')
d = {
a: {b}
}
if os.path.exists(file_path) and os.path.getsize(file_path):
with open('line_info.pkl', 'ab') as f:
pickle.dump(d, f)
print('您已经修改线路成功！！')
else:
with open('line_info.pkl', 'wb') as fp:
pickle.dump(d, fp)
print("您已经修改线路成功了！！！！")
else:
print('您要修改的线路不存在，需要创建一个新的线路。。。')
add_line()
def line_check(num): # 判断是否存在线路的函数
with open('line_info.pkl', 'rb') as fp:
while True:
try:
data = pickle.load(fp)
if num in data:
print(f'{num}号线的路线经过的站点为{data[num]}')
return True
except:
break
def line_find(): # 线路查询的函数
num = input('请输入你要查询的线路：\n').strip()
if line_check(num):
pass
else:
print('您要查找的线路不存在....')
def site_check(): # 判断站点是否存在的函数
res = []
with open('line_info.pkl', 'rb') as fp:
num = input('请输入你要查询的站点名：\n').strip()
while True:
try:
data = pickle.load(fp)
# 遍历字典
for key, value in data.items():
ls = str(value)
if num in ls:
res.append(f'经过{num}站点的线路为{key}号线')
continue
except:
break
if res:
print(res)
return True
else:
return False
def site_find(): # 站点查询的函数
if site_check():
pass
else:
print('您要查询的站点不存在.....')
def check_add_line(num): # 检查添加线路是否存在的函数
with open('line_info.pkl', 'rb') as fp:
while True:
try:
data = pickle.load(fp)
# 遍历字典
for key, value in data.items():
if num in key:
return True
except:
break
def add_line(): # 添加线路的函数
file_path = 'D:\公交查询系统\line_info.pkl'
a = input('请输入添加的线路：').strip()
b = input('请输入经过的站点：')
d = {
a: {b}
}
if os.path.exists(file_path) and os.path.getsize(file_path):
if check_add_line(a):
print('您要添加的线路已经存在了......')
else:
with open('line_info.pkl', 'ab') as f:
pickle.dump(d, f)
print('您已经添加线路成功！！')
else:
with open('line_info.pkl', 'wb') as fp:
pickle.dump(d, fp)
print('您已经添加线路成功了！！')
def clear_line_all(): # 清除所有线路的函数
h = open('line_info.pkl', 'w+') # 清空文件里的内容。。。
h.truncate()
print('你已经销毁成功。。。')
def show_line_all(): # 展示所有线路的函数
with open('line_info.pkl', 'rb') as fp:
while True:
try:
data = pickle.load(fp)
# 遍历字典
for key, value in data.items():
print(f'{key}线路经过的站点是{value}')
except:
break
def clear_line(num): # 删除线路的函数
with open('line_info.pkl', 'rb') as fp:
ls = []
while True:
try:
data = pickle.load(fp)
# 遍历字典
for key, value in data.items():
if num in key:
print(f'{num}线路已经删除成功！！')
pass
else:
ls.append(data)
except:
break
save_data(ls)
def line_check(num): # 判断是否存在线路的函数
with open('line_info.pkl', 'rb') as fp:
while True:
try:
data = pickle.load(fp)
if num in data:
print(f'{num}号线的路线经过的站点为{data[num]}')
return True
except:
break
def show_site(num): # 展示特定线路中的站点的函数
with open('line_info.pkl', 'rb') as fp:
while True:
try:
data = pickle.load(fp)
if num in data:
return data[num]
except:
break
def change_line(): # 换乘信息的函数
begin = input('请输入你的起点站点：\n').strip()
end = input('请输入你的终点站点：\n').strip()
if direct_line(begin, end):
pass
else:
ls = [] # 存储经过起点的线路的全部站点
ls_ = [] # 存储经过终点的线路的全部站点
tag = True
data = check_site(begin, end)
# print(type(data[0]),type(data[1]))
print(f'经过{begin}站点的线路为：{data[0]}，经过{end}站点的线路为{data[1]}') # 判断经过起始点和终点的对应线路
for i in range(len(list(data)[0])):
ls.append(show_site(data[0][i]))
print(f"{line_check(str(data[0][i]))}")
for j in range(len(list(data)[1])):
ls_.append(show_site(data[1][j]))
print(f"{line_check(str(data[1][j]))}")
for i in ls: # i为集合的形式
for a in list(i):
for b in str(a).split(): # 切割字符串
if b in str(ls_): # 判断是否存在相同的站点
tag = False
print(f"您可以通过从{data[0]}号线的{b}站转到{data[1]}号线来到达目的地！！")
if tag:
print('对不起，无法通过换乘来到达目的地')
def direct_line(begin, end): # 判断线路直达的函数
with open('line_info.pkl', 'rb') as fp:
while True:
try:
data = pickle.load(fp)
# 遍历字典
for key, value in data.items():
ls = str(value)
if begin in ls and end in ls:
print(f'您可以通过{key}号线路直接到达')
return True
except:
break
def save_data(ls): # 存储数据的函数
with open('line_info.pkl', 'wb') as fp:
for data in ls:
pickle.dump(data, fp)
def check_site(begin, end): # 判断站点所在的线路的函数。
with open('line_info.pkl', 'rb') as fp:
ls = []
ls_ = []
while True:
try:
data = pickle.load(fp)
# 遍历字典
for key, value in data.items():
ls1 = str(value)
if begin in ls1:
ls.append(key)
if end in ls1:
ls_.append(key)
except:
break
return ls, ls_
if __name__ == 'main':
main()
main()