编辑代码

import os
import re
filename = "students.txt"
account = {}
account['login'] = str(input("请创建您的账号:"))
account['password'] = str(input("请输入您的密码："))
print("注册成功,您的账号是:", account['login'].strip())
def main():
    ctrl = True  # 定义循环
    print("*****欢迎登陆学生信息管理系统*****")
    login = input("请输入您的账号:")
    password = input("请输入您的密码:")
    while ctrl:
        if login.strip() == account['login'].strip() and password.strip() == account['password'].strip():
            menu()  # 显示页面菜单
            option = input("请选择：")  # 选择菜单项
            open_str = int(re.sub("\D", "", option))  # 提取数字(在输入的值中用空格替换所有字母符号，保留数字【"\D"】，d:去掉数字)
            if open_str == 0:  # 退出选择界面
                print("您已经退出学生信息管理系统！")
                ctrl = False  # 循环结束
            elif open_str == 1:  # 录入学生成绩信息
                insert()
            elif open_str == 2:  # 查询学生成绩信息
                search()
            elif open_str == 3:  # 删除学生成绩信息
                delete()
            elif open_str == 4:  # 修改学生成绩信息
                modify()
            elif open_str == 5:  # 对学生成绩排序
                sort()
            elif open_str == 6:  # 统计学生总人数
                total()
            elif open_str == 7:  # 显示所有学生信息
                show()
        else:
            print("您输入的账号或密码错误!!!")
            break
def menu():
    print('='*30)
    print('欢迎使用学生信息管理系统1.0')
    print('1.录入学生信息')
    print('2.查找学生信息')
    print('3.删除学生信息')
    print('4.修改学生信息')
    print('5.学生信息排序')
    print('6.统计学生人数')
    print('7.显示学生信息')
    print('8.退出管理系统')
def save(student): # 建立保存学生信息函数
    try:
        students_txt = open(filename, 'a', encoding='UTF8')  # 追加模式打开学生信息文本文件
    except Exception as e:
        students_txt = open(filename, 'w', encoding='UTF8')  # 如果文件不存在，则写入内容
    for info in student:
        students_txt.write(str(info) + "\n")  # 将添加的学生信息写入到student_txt文件中
    students_txt.close()
def insert():
    stdentList = []  # 建立学生信息表为空列表
    mark = True
    while mark:
        id = input("请输入ID（如242560）：").strip() # 用户输入学生ID
        if not id:  # 如果id为空或者不合格，退出循环
            break
        name = input("请输入名字：").strip() # 输入名字
        if not name:  # 如果没有名字，循环结束
            break
        try:
            chinese = int(input("请输入语文成绩："))  # 输入语文成绩
            maths = int(input("请输入数学成绩："))  # 输入数学成绩
            english = int(input("请输入英语成绩"))  # 输入英语成绩
        except:
            print("输入无效，不是整数型。。。。。。重新输入！")  # 如果输入无效，去掉输入的信息
            continue
        # 将学生信息保存到字典
        stdent = {'id': id, 'name': name, 'chinese': chinese, 'maths': maths, 'english': english}
        stdentList.append(stdent) #将学生成绩以字典的形式添加到stdentList空列表中
        inputMark = input("是否继续添加？ （y/n）：")
        if inputMark == "y":
            mark = True
        else:
            mark = False
    save(stdentList)
    print("学生信息录入完毕！！！")
def search():
    mark = True
    student_query = []
    while mark:
        id = ""  # 定义ID为空
        name = ""  # 定义名字为空
        if os.path.exists(filename):  # 如果文件存在
            mode = int(input("按ID查输入1，按姓名查输入2："))
            if mode == 1:
                id = input("请输入要查找的ID：")
            elif mode == 2:
                name = input("请输入要查找的名字：")
            else:
                print("您输入的有误，请重新输入！")
                search()
            with open(filename, 'r', encoding='UTF8') as file:  # 读取保存文件里面的内容
                student = file.readlines()  # 按行读取
                for list in student:  # 实际上student是sava(student)函数的变量，与insert函数保存的stdentList列表一样
                    d = dict(eval(list))  # 将结果转成字典，eval作用：打印里面的结果；例：eval('3+2-9'),→-4
                    if id is not None:  # 如果输入的是ID
                        if d['id'] == id:  # 如果输入的ID与文件中的字典键值对应
                            student_query.append(d)  # 将查询到的结果保存在该列表内
                    elif name is not None:  # 如果输入的是名字
                        if d['name'] == name:
                            student_query.append(d)
                show_student(student_query)  # 显示查询结果
                student_query.clear()  # 清理查询结果
                inputMark = input("是否继续查询？ （y/n）：")  # 是否继续查询
                if inputMark == "y":
                    mark = True
                else:
                    mark = False
        else:
            print("暂未保存数据！\n")
            return
def delete():
    mark = True
    while mark:
        studentId = input("请输入要删除的学生ID：")  # 用户输入要删除的学生ID
        if studentId is not None:  # 如果输入的ID存在
            if os.path.exists(filename):  # 判断目标文件是否存在
                with open(filename, 'r', encoding='UTF8') as rfile:  # 如果文件存在，打开文件
                    student_old = rfile.readlines()  # 每行读取存在的文件
            else:
                student_old = []  # 如果目标文件不在，则为空
            ifdel = False  # 标记是否删除
            if student_old:  # 如果文件里面不为空
                with open(filename, 'w', encoding='UTF8') as wfile:  # 以写的方式打开文件
                    d = {}  # 定义空字典
                    for list in student_old:  # 每行读取查询到的信息
                        d = dict(eval(list))
                        if d['id'] != studentId:  # 如果ID不等于学生ID
                            wfile.write(str(d) + "\n")  # 将一条学生信息写入文件
                        else:
                            ifdel = True  # 标记已经删除
                    if ifdel:
                        print("ID为 %s 的学生信息已经被删除。。。" % studentId)
                    else:
                        print("没有找到ID为 %s 的学生信息。。。" % studentId)
            else:
                print("无学生信息...")
                break
            show()
            inputMark = input("是否继续删除？ （y/n） ：")
            if inputMark == 'y':
                mark = True
            else:
                mark = False
def modify():
    show()
    if os.path.exists(filename):
        with open(filename, 'r', encoding='UTF8') as rfile:
            student_old = rfile.readlines()
    else:
        return
    studentid = input("请输入要修改学生的ID：")
    with open(filename, 'w', encoding='UTF8') as wfile:
        for student in student_old:
            d = dict(eval(student))
            if d["id"] == studentid:
                print("找到这名学生，可以修改他的信息！")
                while True:
                    try:
                        d["name"] = input("请输入姓名：")
                        d["chinese"] = input("请输入语文成绩：")
                        d["maths"] = input("请输入数学成绩：")
                        d["english"] = input("请输入英语成绩")
                    except:
                        print("您输入的有误，请重新输入！")
                    else:
                        break
                student = str(d)
                wfile.write(student + "\n")
                print("修改成功！")
            else:
                wfile.write(student)
    mark = input("是否继续修改其他学生信息？ （y/n）：")
    if mark == "y":
        modify()
def sort():
    global ascORdescBool
    show()
    if os.path.exists(filename):
        with open(filename, 'r', encoding='UTF8') as file:
            student_old = file.readlines()
            student_new = []
        for list in student_old:
            d = dict(eval(list))
            student_new.append(d)
    else:
        return
    ascORdesc = input("请选择（0升序、1降序）：")
    if ascORdesc == "0":
        ascORdescBool = False
    elif ascORdesc == "1":
        ascORdescBool = True
    else:
        print("您输入有误，请重新输入！")
        sort()
    mode = input("请选择排序方式（1按英语成绩排序、2按数学成绩排序、3按语文成绩排序、0按总成绩排序）：")
    if mode == "1":
        student_new.sort(key=lambda x: x["englishi"], reverse=ascORdescBool)
    elif mode == "2":
        student_new.sort(key=lambda x: x["maths"], reverse=ascORdescBool)
    elif mode == "3":
        student_new.sort(key=lambda x: x["chinese"], reverse=ascORdescBool)
    elif mode == "0":
        student_new.sort(key=lambda x: x["maths"] + x["english"] + x["chinese"], reverse=ascORdescBool)
    else:
        print("您输入的有误，请重新输入！")
        sort()
    show_student(student_new)
def total():
    if os.path.exists(filename):
        with open(filename, 'r', encoding='UTF8') as rfile:
            student_old = rfile.readlines()
            if student_old:
                print("一共有 %d 名学生！" % len(student_old))
            else:
                print("还没有录入学生信息！")
    else:
        print("暂未保存数据信息。。。")
def show():
    student_new = []
    if os.path.exists(filename):
        with open(filename, 'r', encoding='UTF8') as rfile:
            student_old = rfile.readlines()
        for list in student_old:
            student_new.append(eval(list))
        if student_new:
            show_student(student_new)
    else:
        print("暂未保存数据信息。。。")
def show_student(studentList):
    if not studentList:
        print("(o@.@o) 无数据信息 (o@.@o) \n")
        return
    format_title = "{:^6}{:^12}\t{:^8}\t{:^10}\t{:^10}\t{:^10}"
    print(format_title.format("ID", "名字", "英语成绩", "数学成绩", "语文成绩", "总成绩"))
    format_data = "{:^6}{:^12}\t{:^12}\t{:^12}\t{:^12}\t{:^12}"
    for info in studentList:
        print(format_data.format(info.get("id"), info.get("name"), str(info.get("english")), str(info.get("maths")),
                                 str(info.get("chinese")),
                                 str(info.get("english") + info.get("maths") + info.get("chinese")).center(12)))

# if __name__ == "__main__":
#     main()