| import?pickle
from?os.path?import?isfile
from?os?import?remove
def?main():
????running?=?True
????while?running:
????????menu()
????????code?=?'9'
????????count?=?0
????????while?code?not?in?[str(i)?for?i?in?range(8)]:
????????????if?count?==?0:
????????????????code?=?input("请输入：")
????????????else:
????????????????code?=?input("输入有误！请重新输入：")
????????????count?+=?1
????????if?code?==?'0':
????????????running?=?False
????????elif?code?==?'1':
????????????insert()
????????elif?code?==?'2':
????????????search()
????????elif?code?==?'3':
????????????delete()
????????elif?code?==?'4':
????????????modify()
????????elif?code?==?'5':
????????????sort()
????????elif?code?==?'6':
????????????total()
????????elif?code?==?'7':
????????????show()
????else:
????????input("感谢您使用学生信息查询系统程序，请按 <Enter> 键退出程序！")
def?menu():
????print('''
????╔———————学生信息管理系统————————╗
????│????????????????????????????????????????????? │
????│?? =============== 功能菜单 ===============?? │
????│????????????????????????????????????????????? │
????│?? 1 录入学生信息???????????????????????????? │
????│?? 2 查找学生信息???????????????????????????? │
????│?? 3 删除学生信息???????????????????????????? │
????│?? 4 修改学生信息???????????????????????????? │
????│?? 5 排序???????????????????????????????????? │
????│?? 6 统计学生总人数?????????????????????????? │
????│?? 7 显示所有学生信息???????????????????????? │
????│?? 0 退出系统???????????????????????????????? │
????│? ==========================================? │
????│? 说明：通过数字选择菜单????????????????????? │
????╚———————————————————————╝
????????''')
def?insert():
????continue_?=?'y'
????while?continue_?==?'y':
????????ID?=?input("请输入 ID（如 1001）：")
????????name?=?input("请输入名字：")
????????chinese?=?input("请输入语文成绩：")
????????while?not?is_float(chinese):
????????????chinese?=?input("输入无效！请重新输入语文成绩：")
????????chinese?=?float(chinese)
????????math?=?input("请输入数学成绩：")
????????while?not?is_float(math):
????????????math?=?input("输入无效！请重新输入数学成绩：")
????????math?=?float(math)
????????english?=?input("请输入英语成绩：")
????????while?not?is_float(english):
????????????english?=?input("输入无效！请重新输入英语成绩：")
????????english?=?float(english)
????????score?=?[{'id':?ID,?'name': name,
??????????????????'chinese': chinese,?'math': math,?'english': english}]
????????if?not?isfile("students.pkl"):
????????????file?=?open("students.pkl",?"ab")
????????????pickle.dump(score,?file)
????????????file.close()
????????else:
????????????file?=?open("students.pkl",?"rb+")
????????????score_list?=?pickle.load(file)
????????????score_list.extend(score)
????????????file.close()
????????????file?=?open("students.pkl",?"wb")
????????????pickle.dump(score_list,?file)
????????????file.close()
????????continue_?=?input("是否继续添加？（y/n）")
????????while?continue_?not?in?['y',?'n']:
????????????continue_?=?input("输入无效！是否继续添加？（y/n）")
????else:
????????print("学生信息录入完毕！")
def?search():
????continue_?=?'y'
????while?continue_?==?'y':
????????if?not?isfile("students.pkl"):
????????????print("没有任何数据！")
????????????return
????????way?=?input("按ID查输入1；按姓名查输入2：")
????????while?way?not?in?['1',?'2']:
????????????way?=?input("输入无效！按ID查输入1；按姓名查输入2：")
????????if?way?==?'1':
????????????students?=?pickle.load(open("students.pkl",?"rb"))
????????????ID?=?input("请输入学生 ID：")
????????????result?=?[]
????????????for?each?in?students:
????????????????if?each['id']?==?ID:
????????????????????result.append(each)
????????else:
????????????students?=?pickle.load(open("students.pkl",?"rb"))
????????????name?=?input("请输入学生姓名：")
????????????result?=?[]
????????????for?each?in?students:
????????????????if?each['name']?==?name:
????????????????????result.append(each)
????????result_str?=?''
????????result_str?+=?'ID'.center(6)
????????result_str?+=?'姓名'.center(12)
????????result_str?+=?'语文'.center(10)
????????result_str?+=?'数学'.center(10)
????????result_str?+=?'英语'.center(10)
????????result_str?+=?'总成绩'.center(11)
????????for?i?in?result:
????????????result_str?+=?"\n"
????????????result_str?+=?i['id'].center(6)
????????????result_str?+=?i['name'].center(12)
????????????result_str?+=?str(i['chinese']).center(12)
????????????result_str?+=?str(i['math']).center(12)
????????????result_str?+=?str(i['english']).center(12)
????????????result_str?+=?str(i['chinese']?+?i['math']?+
??????????????????????????????i['english']).center(13)
????????print(result_str)
????????continue_?=?input("是否继续查询 (y/n)？")
????????while?continue_?not?in?['y',?'n']:
????????????continue_?=?input("输入无效！是否继续查询 (y/n)？")
def?delete():
????if?not?isfile("students.pkl"):
????????print("没有学生信息！@_@")
????????return
????continue_?=?'y'
????while?continue_?==?'y':
????????show()
????????ID?=?input("请输入要删除的学生ID：")
????????students?=?pickle.load(open("students.pkl",?"rb"))
????????result?=?[]
????????for?each?in?students:
????????????if?each['id']?==?ID:
????????????????result.append(each)
????????if?not?result:
????????????print(f"没有找到ID为 {ID} 的学生信息...")
????????????show()
????????else:
????????????for?i?in?result:
????????????????students.remove(i)
????????????if?not?students:
????????????????remove("students.pkl")
????????????????print(f"ID 为 {ID} 的学生已成功删除！不可继续删除！")
????????????????return
????????????else:
????????????????pickle.dump(students,?open("students.pkl",?"wb"))
????????????????print(f"ID 为 {ID} 的学生已成功删除！")
????????continue_?=?input("是否继续删除 (y/n)？")
????????while?continue_?not?in?['y',?'n']:
????????????continue_?=?input("输入无效！是否继续删除 (y/n)？")
def?modify():
????if?not?isfile("students.pkl"):
????????print("没有学生信息！")
????????return
????show()
????continue_?=?'y'
????while?continue_?==?'y':
????????ID?=?input("请输入要修改的学生ID：")
????????students?=?pickle.load(open("students.pkl",?"rb"))
????????result?=?[]
????????for?each?in?students:
????????????if?each['id']?==?ID:
????????????????result.append(each)
????????if?not?result:
????????????print(f"没有找到ID为 {ID} 的学生信息...")
????????????show()
????????elif?len(result) !=?1:
????????????print("ID 有相同的情况！")
????????else:
????????????students.remove(result[0])
????????????name?=?input("请输入名字：")
????????????chinese?=?input("请输入语文成绩：")
????????????while?not?is_float(chinese):
????????????????chinese?=?input("输入无效！请重新输入语文成绩：")
????????????chinese?=?float(chinese)
????????????math?=?input("请输入数学成绩：")
????????????while?not?is_float(math):
????????????????math?=?input("输入无效！请重新输入数学成绩：")
????????????math?=?float(math)
????????????english?=?input("请输入英语成绩：")
????????????while?not?is_float(english):
????????????????english?=?input("输入无效！请重新输入英语成绩：")
????????????english?=?float(english)
????????????score?=?{'id':?ID,?'name': name,
?????????????????????'chinese': chinese,?'math': math,?'english': english}
????????????students.append(score)
????????????pickle.dump(students,?open("students.pkl",?"wb"))
????????continue_?=?input("是否继续修改 (y/n)？")
????????while?continue_?not?in?['y',?'n']:
????????????continue_?=?input("输入无效！是否继续修改 (y/n)？")
def?sort():
????if?not?isfile("students.pkl"):
????????print("没有学生信息！")
????????return
????show()
????reverse?=?input("请选择（0升序；1降序）：")
????while?reverse?not?in?['0',?'1']:
????????reverse?=?input("输入无效！请选择（0升序；1降序）：")
????reverse?=?bool(int(reverse))
????students?=?pickle.load(open("students.pkl",?"rb"))
????way?=?input("请选择排序方式"
????????????????"（1按语文成绩排序；"
????????????????"2按数学成绩排序；"
????????????????"3按英语成绩排序；"
????????????????"0按总成绩排序）：")
????while?way?not?in?['0',?'1',?'2',?'3']:
????????way?=?input("输入无效！请选择排序方式"
????????????????????"（1按语文成绩排序；"
????????????????????"2按数学成绩排序；"
????????????????????"3按英语成绩排序；"
????????????????????"0按总成绩排序）：")
????if?way?==?'0':
????????def?condition(x):
????????????return?x['chinese']?+?x['math']?+?x['english']
????elif?way?==?'1':
????????def?condition(x):
????????????return?x['chinese']
????elif?way?==?'2':
????????def?condition(x):
????????????return?x['math']
????else:
????????def?condition(x):
????????????return?x['english']
????result?=?sorted(students, key=condition, reverse=reverse)
????result_str?=?''
????result_str?+=?'ID'.center(6)
????result_str?+=?'姓名'.center(12)
????result_str?+=?'语文'.center(10)
????result_str?+=?'数学'.center(10)
????result_str?+=?'英语'.center(10)
????result_str?+=?'总成绩'.center(11)
????for?i?in?result:
????????result_str?+=?"\n"
????????result_str?+=?i['id'].center(6)
????????result_str?+=?i['name'].center(12)
????????result_str?+=?str(i['chinese']).center(12)
????????result_str?+=?str(i['math']).center(12)
????????result_str?+=?str(i['english']).center(12)
????????result_str?+=?str(i['chinese']?+?i['math']?+
??????????????????????????i['english']).center(13)
????print(result_str)
def?total():
????try:
????????print("一共有",
??????????????str(len(pickle.load(open("students.pkl",?"rb")))),?"名学生！")
????except?FileNotFoundError:
????????print("没有学生信息！")
def?show():
????if?not?isfile("students.pkl"):
????????print("没有学生信息！@_@")
????????return
????result?=?pickle.load(open("students.pkl",?"rb"))
????result_str?=?''
????result_str?+=?'ID'.center(6)
????result_str?+=?'姓名'.center(12)
????result_str?+=?'语文'.center(10)
????result_str?+=?'数学'.center(10)
????result_str?+=?'英语'.center(10)
????result_str?+=?'总成绩'.center(11)
????for?i?in?result:
????????result_str?+=?"\n"
????????result_str?+=?i['id'].center(6)
????????result_str?+=?i['name'].center(12)
????????result_str?+=?str(i['chinese']).center(12)
????????result_str?+=?str(i['math']).center(12)
????????result_str?+=?str(i['english']).center(12)
????????result_str?+=?str(i['chinese']?+?i['math']?+
??????????????????????????i['english']).center(13)
????print(result_str)
def?is_float(number):
????try:
????????float(number)
????except?ValueError:
????????return?False
????else:
????????return?True
if?__name__?==?'__main__':
????try:
????????main()
????except?BaseException as e:
????????print("啊哦，出错了 O_O")
????????print(e)
????????input("请按 <Enter> 键退出程序！")
| 