Python 学習ノート (3) - Yang Shujuan Python チュートリアルの宿題

133 学生情報管理システム

ジョブコード:

import json
import os

filename = 'student.txt'


def main():
    while True:
        menu()
        choice = int(input('请选择'))
        if choice in [0, 1, 2, 3, 4, 5, 6, 7]:
            if choice == 0:
                answer = input('确认退出? y/n ')
                if answer == 'y' or answer == 'Y':
                    print('谢谢使用!')
                    break
                else:
                    continue
            elif choice == 1:
                insert()  # 录入学生信息
            elif choice == 2:
                search()  # 查找学生信息
            elif choice == 3:
                delete()  # 删除
            elif choice == 4:
                modify()  # 修改
            elif choice == 5:
                sort()  # 排序
            elif choice == 6:
                total()  # 统计
            elif choice == 7:
                show()  # 显示


def menu():
    print('{:=^30}'.format('学生信息管理系统'))
    print('{:-^32}'.format('功能菜单'))
    print('     1.录入学生信息'.ljust(15))
    print('     2.查找学生信息'.ljust(15))
    print('     3.删除学生信息'.ljust(15))
    print('     4.修改学生信息'.ljust(15))
    print('     5.排序'.ljust(15))
    print('     6.统计学生总人数'.ljust(15))
    print('     7.显示所有学生信息'.ljust(15))
    print('     0.退出系统'.ljust(15))
    print('{:-^35}'.format(''))


def save(lst):
    try:
        stu_txt = open(filename, 'a', encoding='utf-8')
    except:
        stu_txt = open(filename, 'w', encoding='utf-8')
    for item in lst:
        stu_txt.write(str(item) + '\n')
    stu_txt.close()


def insert():
    student_list = []
    while True:
        id = input('请输入学生ID(如1001): ')
        if not id:
            break
        name = input('请输入学生姓名: ')
        if not name:
            break

        try:
            english = input('请输入英语成绩: ')
            python = input('请输入Python成绩: ')
            java = input('请输入Java成绩: ')
        except:
            print('输入无效,不是整数类型,请重新输入。')
            continue
        # 将输入的成绩保存
        student = {'id': id, 'name': name, 'English': english, 'Python': python, 'Java': java}
        # 信息添加
        student_list.append(student)
        answer = input('是否继续添加? y/n \n')
        if answer == 'y' or answer == 'Y':
            continue
        else:
            break

    save(student_list)
    print('学生信息录入完毕!')


def show_student(lst):
    if len(lst) == 0:
        print('没有查询到学生信息,无数据显示')
        return
    # 定义标题显示格式
    format_title = '{:^6}\t{:^12}\t{:^8}\t{:^10}\t{:^8}\t{:^8}\t'  # 冒号后面表示填充内容
    print(format_title.format('ID', '姓名', '英语成绩', 'Python成绩', 'Java成绩', '总成绩'))
    # 定义内容的显示格式
    # format_data = '{:^6}\t{:^12}\t{:^8}\t{:^10}\t{:^10}\t{:^8}\t'
    # for item in lst:
    #     print(format_data.format(item.get('id'),
    #                              item.get('name'),
    #                              item.get('English'),
    #                              item.get('Python'),
    #                              item.get('Java'),
    #                              int(item.get('English') + item.get('Python') + item.get('Java'))
    #                              ))
    format_data = '{:^6}\t{:^12}\t{:^8}\t{:^10}\t{:^10}\t{:^8}\t'
    for item in lst:
        print(format_data.format(
            item.get('id'),
            item.get('name'),
            '{:^8}'.format(item.get('English')),
            '{:^10}'.format(item.get('Python')),
            '{:^10}'.format(item.get('Java')),
            '{:^8}'.format(int(item.get('English')) + int(item.get('Python')) + int(item.get('Java')))
        ))


def search():
    student_query = []
    while True:
        id = ''
        name = ''
        if os.path.exists(filename):
            mode = input('按ID查找请输入1,按姓名查找请输入2:')
            if mode == '1':
                id = input('请输入学生ID')
            elif mode == '2':
                name = input('请输入学生姓名')
            else:
                print('输入错误,重新输入')
                search()
            with open(filename, 'r', encoding='utf-8') as rfile:
                student = rfile.readlines()
                for item in student:
                    d = dict(eval(item))
                    if id != '':
                        if d['id'] == id:
                            student_query.append(d)
                    elif name != '':
                        if d['name'] == name:
                            student_query.append(d)
            # 显示查询结果
            show_student(student_query)
            # 清空列表
            student_query.clear()
            answer = input('是否继续查询? y/n \n')
            if answer == 'y' or answer == 'Y':
                continue
            else:
                break
        else:
            print('暂未保存目标学生信息')
            return


def delete():
    while True:
        student_id = input('请输入要删除的学生的ID:')
        if student_id != '':
            if os.path.exists(filename):
                with open(filename, 'r', encoding='utf-8') as file:
                    student_old = file.readlines()
            else:
                student_old = []
            flag = False  # 标记是否删除
            if student_old:
                with open(filename, 'w', encoding='utf-8') as wfile:
                    d = {}
                    for item in student_old:
                        d = dict(eval(item))  # 将字符串转成字典
                        if d['id'] != student_id:
                            wfile.write(str(d) + '\n')
                        else:
                            flag = True  # 表示已经删除
                    if flag:
                        print(f'id为{student_id}的学生信息已被删除')
                    else:
                        print(f'没有找到ID为{student_id}的学生信息')
            else:
                print('学生信息列表为空,无学生信息')
                break
            show()  # 删除完成后重新显示所有学生的信息
            answer = input('是否继续删除? y/n \n')
            if answer == 'y' or answer == 'Y':
                continue
            else:
                break


def modify():
    show()
    if os.path.exists(filename):
        with open(filename, 'r', encoding='utf-8') as rfile:
            student_old = rfile.readlines()
    else:
        return
    student_id = input('请输入要修改的学员ID:')
    flag = False
    with open(filename, 'w', encoding='utf-8') as wfile:
        for item in student_old:
            d = dict(eval(item))
            if d['id'] == student_id:
                flag = True  # 查找成功的标志位
                print('找到学生信息,可以修改其内容')
                while True:
                    try:
                        d['name'] = input('请输入姓名')
                        d['English'] = int(input('请输入英语成绩'))
                        d['Python'] = int(input('请输入Python成绩'))
                        d['Java'] = int(input('请输入Java成绩'))
                    except:
                        print('输入有误,重新输入')
                    else:
                        break
                wfile.write(str(d) + '\n')
                print('修改成功')
            else:
                wfile.write(str(d) + '\n')
        if not flag:
            print(f'没有找到ID为{student_id}的学生信息')
        answer = input('是否继续修改? y/n \n')
        if answer == 'y' or answer == 'Y':
            modify()


def sort():
    show()
    student_new = []
    if os.path.exists(filename):
        with open(filename, 'r', encoding='utf-8') as rfile:
            students_lst = rfile.readlines()
            for item in students_lst:
                d = dict(eval(item))
                student_new.append(d)
    else:
        print('暂未保存数据信息……')
        return

    asc_or_desc = input('输入排序要求,0为升序,1为降序')
    if asc_or_desc == '0':
        asc_or_desc_flag = False
    elif asc_or_desc == '1':
        asc_or_desc_flag = True
    else:
        print('输入有误,重新输入')
        sort()
    mode = int(input('请选择排序方式(1.按英语 2.按Python 3.按Java 4.按总成绩'))
    if mode == 1:
        student_new.sort(key=lambda x: int(x['English']), reverse=asc_or_desc_flag)
    elif mode == 2:
        student_new.sort(key=lambda x: int(x['Python']), reverse=asc_or_desc_flag)
    elif mode == 3:
        student_new.sort(key=lambda x: int(x['Java']), reverse=asc_or_desc_flag)
    elif mode == 4:
        student_new.sort(key=lambda x: int(int(x['English']) + int(x['Python']) + int(x['Java'])),
                         reverse=asc_or_desc_flag)
    else:
        print('输入有误,重新输入')
        sort()
    show_student(student_new)


def total():
    if os.path.exists(filename):
        with open(filename, 'r', encoding='utf-8') as rfile:
            students = rfile.readlines()
            if students:
                print(f'一共有{len(students)}名学生')
            else:
                print('暂未录入学生信息')
    else:
        print('暂未保存数据信息……')


def show():
    student_lst = []
    if os.path.exists(filename):
        with open(filename, 'r', encoding='utf-8') as rfile:
            students = rfile.readlines()
            for item in students:
                student_lst.append(eval(item))
            if student_lst:
                show_student(student_lst)
    else:
        print('暂未保存数据信息……')


if __name__ == '__main__':
    main()

おすすめ

転載: blog.csdn.net/Suxiang1997/article/details/130582687