学生信息管理系统

def get_all_students():
    students = []
    f = open('学生.txt')
    for line in f:
        line = line.replace('\n','')
        student = line.split(' ')
        students.append(student)
        
    f.close()
    return students

def add_student():
    students = get_all_students()
    
    id = input('学号：')
    name = input('姓名：')
    gender = input('性别：')
    age = input('年龄：')
    clazz = input('班级：')
    major = input('专业：')
    department = input('系部：')
    telephone = input('电话：')
    
    students.append([id, name, gender, age, clazz, major, department, telephone])
    
    
    f = open('学生.txt','w')
    for i in range(len(students)):
        student = ' '.join(students[i]) + '\n'
        f.write(student)
    f.close()
    print('\n恭喜，学生记录添加成功！\n')
    
def display_all_students():
    students = get_all_students()
    for i in range(len(students)):
        for j in range(len(students[i])):
            print(students[i][j], end= ' ')
        print()
            

    
def query_student():
    while True:
        print('查询学生记录')
        print('======================')
        print('1.按学号查询学生记录')
        print('2.按姓名查询学生记录')
        print('3.查询全部学生记录')
        print('4.返回上级菜单')
        print('=======================')
        mc3 = int(input('输入菜单号：'))
        if mc3 == 1:
            students = []
            
            f = open('学生.txt')
            
            for line in f:
                line = line.replace('\n', '')
                student = tuple(line.split(' '))
                students.append(student)
            
            id = input('请输入待查学生的学号：')
            found = False
            for student in students:
                if (student[0]  == id):
                    found = True
                    for i in range(len(student)):
                        print(student[i], end = ' ')
                    break
            if not found:
                print('学号为[{}]的学生没找到！'.format(id))

        elif mc3 == 2:
            students = []

            f = open('学生.txt')
            
            for line in f:
                line = line.replace('\n', '')
                student = tuple(line.split(' '))
                students.append(student)
            
            name = input('请输入待查学生的姓名：')
            found = False
            for student in students:
                if (student[1] == name):
                    found = True
                    for i in range(len(student)):
                        print(student[i], end= ' ')
                    
            if not found:
                print('姓名为[{}]的学生没找到！'.format(name))

        elif mc3 == 3:
            display_all_students()
        else:
            break
        
def modify_student():
    students = []

    f = open('学生.txt')
    
    for line in f:
        line = line.replace('\n', '')
        student = tuple(line.split(' '))
        students.append(student)
    
    id = input('请输入待查学生的学号：')
    found = False
    for student in students:
        if (student[0] == id):
            found = True
            
            student = list(student)
            new_age = input('输入新的年龄：')
            new_telephone = input('输入新的电话：')
            
            student[3] = new_age
            student[7] = new_telephone
            
            student = tuple(student)
            print('恭喜：学生记录修改成功！')
            for i in range(len(student)):
                print(student[i], end=' ')
            break
        
    if not found:
        print('学号为[{}]的学生没找到！'.format(id))

def delete_student():
    students = []
        
    f = open('学生.txt')
    
    for line in f:
        line = line.replace('\n', '')
        student = tuple(line.split(' '))
        students.append(student)
    
    id = input('请输入待删学生的学号：')
    found = False
    for student in students:
        if (student[0] == id):
            found = True
            
            students.remove(student)
            print('恭喜：学生记录删除成功！')
            break
            
    if not found:
        print('学号为[{}]的学生没找到！'.format(id))
    else:
        for student in students:
            for i in range(len(student)):
                print(student[i], end= ' ')

def login():
    username = input('输入用户名：')
    password = input('输入密码：')

    if username == 'howard' and password == '123456':
        while True:
            print('学生信息管理')
            print('======================')
            print('1.增加学生记录')
            print('2.查询学生记录')
            print('3.修改学生记录')
            print('4.删除学生记录')
            print('5.返回上级菜单')
            print('=======================')
            mc2 = int(input('输入菜单号：'))
            if mc2 == 1:
                add_student()
            elif mc2 ==2:
                query_student()
            elif mc2 == 3:
                modify_student()
            elif mc2 == 4:
                delete_student()
            else:
                break
    else:
        print('\n用户名或密码错误，请重新登录！\n')
        
while True:
    print('用户登录')
    print('============')
    print('1.登录')
    print('2.退出')
    print('============')
    mc1 = int(input('输入菜单号：'))
    if mc1 == 1:
        login()
    elif mc1 == 2:
        print('\n蟹蟹使用本程序！\n')
        break