python实现班级档案管理系统

import openpyxl
from openpyxl import Workbook
import sys

s_information = [{'学号':'202031108041','姓名':'小明','性别':'男','年龄':'20','备注':'帅比'},
         {'学号':'202031108042','姓名':'小谷','性别':'女','年龄':'20','备注':'漂比'},
         {'学号':'202031108043','姓名':'小啊','性别':'男','年龄':'20','备注':'帅比'},
         {'学号':'202031108044','姓名':'小额','性别':'男','年龄':'20','备注':'帅比'},
         ]

#登录密码

def print_menu():
    # 打印菜单
    print('-'*50)
    print('[1]:学生基本信息录入')
    print('[2]:学生基本信息显示')
    print('[3]:学生基本信息保存')
    print('[4]:学生基本信息删除')
    print('[5]:学生基本信息修改')
    print('[6]:学生基本信息查询')
    print('[7]:退出')
#基本信息录入
def add_infomation():
    dic = {}
    while True:
        xuehao = input('请输入你的学号：')
        for i in s_information:
            if i['学号'] == xuehao:
                print('学号已存在,请重新输入')
        else:
            dic['学号'] = xuehao
            dic['姓名'] = input('请输入姓名：')
            dic['性别'] = input('请输入性别：')
            dic['年龄'] = input('请输入年龄：')
            dic['备注'] = input('请输入备注：')
            break
    s_information.append(dic)
#显示所有信息
def show_information():
    for i in s_information:
        print(i)
#删除学生信息,可以添加学生信息不在时的情况
def del_information():
    a = input('请输入你要删除的学生的姓名：')
    for i in s_information:
        if a == i['姓名']:
            s_information.remove(i)

#学生基本信息修改：
def change_information():
    a = input('请输入你要修改人的信息，如姓名、电话')
    print('1、学号')
    print('2、姓名')
    print('3、性别')
    print('4、年龄')
    print('5、备注')
    b = input('请输入你要修改的选项：')

    for i in s_information:
        if i['姓名'] == a :
            if b=="1":
                i['学号']=input('请输入修改后的：')
            elif b=='2':
                i['姓名']=input('请输入修改后的：')
            elif b=='3':
                i['性别'] = input('请输入修改后的：')
            elif b=='4':
                i['年龄'] = input('请输入修改后的：')
            else:
                i['备注']= input('请输入修改后的：')
            break
        elif i['学号'] == a :
            if b=="1":
                i['学号']=input('请输入修改后的：')
            elif b=='2':
                i['姓名']=input('请输入修改后的：')
            elif b=='3':
                i['性别'] = input('请输入修改后的：')
            elif b=='4':
                i['年龄'] = input('请输入修改后的：')
            else:
                i['备注']= input('请输入修改后的：')
            break
        else:
            print('请输入学号、或姓名进行查询')
#学生基本信息查询
def select_information():
    a = input('请输入你要查询人的姓名或学号')
    for i in s_information:
        if i['姓名']  == a:
            print(i)
            break
        elif i['学号'] == a :
            print(i)
            break
        else:
            print(''请输入学号或者姓名进行查询')
#学生信息保存
def write_tofile():
    a = input('请输入你要保存文件的名称：')
    wb=Workbook()
    sheet1 = wb.create_sheet('学生信息表',0)
    sheet1.cell(row=1,column=1).value='学号'
    sheet1.cell(row=1,column=2).value='姓名'
    sheet1.cell(row=1,column=3).value='性别'
    sheet1.cell(row=1,column=4).value='年龄'
    sheet1.cell(row=1,column=5).value='备注'
    for i in range(len(s_information)):
       
        
        
        sheet1.cell(row=i+2,column=1).value=s_information[i]['学号']
        sheet1.cell(row=i+2,column=2).value=s_information[i]['姓名']
        sheet1.cell(row=i+2,column=3).value=s_information[i]['性别']
        sheet1.cell(row=i+2,column=4).value=s_information[i]['年龄']
        sheet1.cell(row=i+2,column=5).value=s_information[i]['备注']
        
    wb.save('{}.xlsx'.format(a))

#离开程序    
def quit_information():
    sys.exit(0)       

#设置主函数
def main2():
      while True:#设置登录密码
            print('------登陆-------')
            print('账号是学号，密码是学号后五位')
            key_word = input("账号：")
            password = input("密码：")

            while True and password ==  key_word[-5:] and len(key_word) > 5:
                print('------班级管理系统------')
           
                print_menu()    # 打印菜单
                num = input('请输入您的选项:')
                if num == '1':
                    add_infomation()  # 添加图书
                elif num == '2':
                    show_information()  # 删除图书
                elif num == '3':
                    write_tofile() # 查询所有图书
                elif num == '4':
                    del_information()
                elif num == '5':
                    change_information()
                elif num == '6':
                    select_information()
                elif num == '7':
                    quit_information()

                else:
                    print('您选择的有误,请重新选择')
            else:
                print('你输入的密码有误，请重新输入')

if __name__ == '__main__':
    main2()