- 新增学生
- 显示学生
- 查找学生
- 删除学生
- 存档到文件
约定好数据的存储格式:
约定把数据保存在和py文件同级目录中,文件名为record.txt
文件内容按照行文本的方式来表示
首先这是一个文本文件,里面包含了很多行,每一行代表一个学生
学生的信息之间,使用\t来分割(水平制表符)
编写代码:
save存档
load加载
# 实现一个命令行版本的学生管理系统
import sys
import os.path
# 使用这个全局变量,来管理学生所有的信息
# 这个列表里面的每个元素都是一个“字典”,每个字典分别表示一个同学
students = []
def save():
"""
用于存档
:return:
"""
# 此处的路径不是以D盘开头的“绝对路径”,而是相对路径
# 此时这个写法的含义是让record.txt和当前的student_guanli.py在同一个目录里
with open('record.txt', 'w', encoding='utf8') as f:
for s in students:
f.write(f"{s['studentId']}\t{s['name']}\t{s['gender']}\t{s['className']}\n")
print(f'[存档成功] 共存储了{len(students)}条记录!')
def load():
"""
用于读档
:return:
"""
# 如果存档文件不存在,直接跳过读档流程
# 为了避免读方式打开文件的时候,文件不存在引起异常
if not os.path.exists('record.txt'):
return
# 读档的时候要保证先把旧的数据清理干净
global students
students = []
with open('record.txt', 'r', encoding='utf8') as f:
for line in f:
# 针对这一行数据,按照\t进行切分操作
# 切分之前,去除末尾的换行
line = line.strip() # 去掉字符串开头和末尾的空白符——空格,换行,回车...
tokens = line.split('\t')
if len(tokens) != 4:
print(f'当前行格式存在问题!Line={line}')
continue
student = {
'studentId': tokens[0],
'name': tokens[1],
'gender': tokens[2],
'className': tokens[3]
}
students.append(student)
print(f'[读档成功] 共读取了{len(students)}条记录!')
def menu():
print('1.新增学生')
print('2.显示学生')
print('3.查找学生')
print('4.删除学生')
print('0.退出程序')
choice = input('请输入您的选择:')
return int(choice)
def insert():
print('[新增学生] 开始!')
studentId = input('请输入学生的学号:')
name = input('请输入学生的姓名:')
gender = input('请输入学生的性别:')
if gender not in ('男', '女'):
print('性别不符合要求,新增失败!')
return
className = input('请输入学生的班级:')
# 使用一个字典,把上述信息聚合起来
student = {
'studentId': studentId,
'name': name,
'gender': gender,
'className': className
}
global students
students.append(student)
# 增加保存操作
save()
print('[新增学生] 完毕!')
def show():
# 遍历全局变量的这个列表,把每个学生的信息打印出来
print('[显示学生] 开始!')
for s in students:
print(f"[{s['studentId']}]\t{s['name']}\t{s['gender']}\t{s['className']}")
print(f'[显示学生] 完毕!共显示{len(students)}条数据!')
def find():
# 根据学生姓名,来进行查找
print('[查找学生] 开始!')
name = input('请输入要查找的学生姓名:')
count = 0
for s in students:
if name == s['name']:
print(f"[{s['studentId']}]\t{s['name']}\t{s['gender']}\t{s['className']}")
count += 1
print(f'[查找学生] 结束!共找到了{count}个匹配的同学!')
def delete():
print('[删除学生] 开始!')
studentId = input('请输入要删除学生的学号:')
# 看看学号对应的同学是哪个字典,然后把这个字典从列表上处即可
for s in students:
if studentId == s['studentId']:
print(f"删除{s['name']}同学的信息!")
students.remove(s)
save()
print('[删除学生] 结束!')
def main():
"""
入口函数
:return:
"""
# 通过控制台和用户进行交互
print('-------------------------------')
print(' 欢迎来到学生管理系统 ')
print('-------------------------------')
# 在程序启动的时候调用load
load()
while True:
# 通过menu函数打印菜单项
choice = menu()
if choice == 1:
# 新增学生
insert()
elif choice == 2:
# 显示学生
show()
elif choice == 3:
# 查找学生
find()
elif choice == 4:
delete()
elif choice == 0:
# 退出程序
print('goodbye!')
sys.exit(0)
else:
print('您的输入有误!请重新输入:')
# 需要进入下次循环
# continue
main()