day11 - 作业（学生系统）

#_*_encoding=utf-8_*_
def interface():
    """
    欢迎界面
    :return: 无
    """
    print('='.center(40, '='))
    print("欢迎ZH:")
    print(' '*15 + "1. 添加学生")
    print(' '*15 + "2. 查看学生")
    print(' '*15 + "3. 修改学生信息")
    print(' '*15 + "4. 删除学生")
    print(' '*15 + "5. 返回")
    print('='.center(40, '='))


def add_stu():
    """
    添加学生信息
    :return: 无
    """
    stu_file = open("D:\\学生.txt", 'a+')
    while True:
        print('='.center(40, '='))
        stu_name = input("请输入学生姓名：")
        stu_age = int(input("请输入学生的年龄："))
        stu_tel = input("请输入学生的电话：")
        stu_file.write('姓名:' + stu_name + ' 年龄:%d' % stu_age + ' 电话:' + stu_tel + '\n')
        print("添加成功!")
        print('='.center(40, '='))
        while True:
            print("1.继续\n2.返回")
            n = input("请输入选择(1-2):")
            if n == '1':
                break
            if n == '2':
                stu_file.close()
                return
            print("输入错误！")


def view_stu():
    """
    查看学生信息
    :return: 无
    """
    stu_file = open("D:\\学生.txt")
    while True:
        print('='.center(40, '='))
        stu_name = input("请输入查看学生的姓名：")
        flag = 0
        for stu_str in stu_file.readlines():
            name_str = ''
            for ch in stu_str[3:]:
                if ch == ' ':
                    break
                name_str += ch
            if name_str == stu_name:
                print(stu_str[:-2])
                flag = 1
                break
        if flag == 0:
            print("很遗憾，未找到该学生信息！")
        print('='.center(40, '='))
        while True:
            print("1.继续\n2.返回")
            n = input("请输入选择(1-2):")
            if n == '1':
                stu_file.seek(0)
                break
            if n == '2':
                stu_file.close()
                return
            print("输入错误！")


def modify_stu():
    """
    修改学生信息
    :return: 无
    """
    while True:
        print('='.center(40, '='))
        stu_name = input("请输入修改学生的姓名：")
        flag = 0
        file_data = ""
        with open("D:\\学生.txt", 'r') as stu_file:
            for stu_str in stu_file.readlines():
                name_str = ''
                for ch in stu_str[3:]:
                    if ch == ' ':
                        break
                    name_str += ch
                if name_str == stu_name:
                    stu_age = int(input("请输入修改学生的年龄："))
                    stu_tel = input("请输入修改学生的电话：")
                    stu_str = '姓名:' + stu_name + ' 年龄:%d' % stu_age + ' 电话:' + stu_tel + '\n'
                    flag = 1
                file_data += stu_str
        with open("D:\\学生.txt", 'w') as stu_file:
            stu_file.write(file_data)
        if flag == 0:
            print("很遗憾，未找到该学生信息！")
        if flag == 1:
            print("修改成功！")
        print('='.center(40, '='))
        while True:
            print("1.继续\n2.返回")
            n = input("请输入选择(1-2):")
            if n == '1':
                break
            if n == '2':
                return
            print("输入错误！")


def remove_stu():
    """
    删除学生信息
    :return: 无
    """
    while True:
        print('='.center(40, '='))
        stu_name = input("请输入删除学生的姓名：")
        flag = 0
        file_data = ""
        with open("D:\\学生.txt", 'r') as stu_file:
            for stu_str in stu_file.readlines():
                name_str = ''
                for ch in stu_str[3:]:
                    if ch == ' ':
                        break
                    name_str += ch
                if name_str == stu_name:
                    flag = 1
                    continue
                file_data += stu_str
        with open("D:\\学生.txt", 'w') as stu_file:
            stu_file.write(file_data)
        if flag == 0:
            print("很遗憾，未找到该学生信息！")
        if flag == 1:
            print("删除成功！")
        print('='.center(40, '='))
        while True:
            print("1.继续\n2.返回")
            n = input("请输入选择(1-2):")
            if n == '1':
                break
            if n == '2':
                return
            print("输入错误！")


def stu_sys():
    """
    学生系统
    :return: 无
    """
    while True:
        interface()
        n = input("请输入选择(1-5):")
        while True:
            if n == '1':
                add_stu()
            if n == '2':
                view_stu()
            if n == '3':
                modify_stu()
            if n == '4':
                remove_stu()
            if n == '5':
                return
            if not (1 <= int(n) <= 5):
                print("输入错误！")
                n = input("请输入选择(1-5):")
                continue
            interface()
            n = input("请输入选择(1-5):")