def student_id():
    """
    學(xué)號(hào)生成器
    :return:
    """
    i = 1
    while True:
        x = i
        yield 'py1902' + str(x).rjust(3, '0')
        i = int(i) + 1


ids = student_id()


all_stu = [
    {'stu_id': next(ids), 'name': '張1', 'age': 18, 'tel': '12333332222'},
    {'stu_id': next(ids), 'name': '張2', 'age': 19, 'tel': '12333334434'},
    {'stu_id': next(ids), 'name': '張3', 'age': 21, 'tel': '12332223468'}
]


def add_stu(name: str, age: int, tel: str, stu_id=next(ids)):
    """
    添加學(xué)生
    :param name:
    :param age:
    :param tel:
    :param stu_id:
    :return:
    """
    dict1 = {}
    dict1['stu_id'] = stu_id
    dict1['name'] = name
    dict1['age'] = age
    dict1['tel'] = tel
    all_stu.append(dict1)
    if dict1 in all_stu:
        print('添加成功!')
        return add_stu


def retrieval(x):
    """
    查詢學(xué)生信息
    :return:
    """
    if x == '1':
        print(all_stu)
    elif x == '2':
        name = input('請(qǐng)輸入學(xué)生姓名：')
        for dict1 in all_stu:
            if dict1['name'] == name:
                print(dict1)
            else:
                print('查無此人')
                break
    elif x == '3':
        stu_id = input('請(qǐng)輸入學(xué)生學(xué)號(hào)：')
        for dict1 in all_stu:
            if dict1['stu_id'] == stu_id:
                print(dict1)
            else:
                print('查無此人')
                break
    judge('2')


def revise1(x):
    if x == '1':
        stu_id = input('請(qǐng)輸入要修改的學(xué)生的學(xué)號(hào)')
        for dict1 in all_stu:
            if dict1['stu_id'] == stu_id:
                print('學(xué)生信息', dict1)
                dict1['name'] = input('請(qǐng)輸入新的姓名')
                print('修改成功！', dict1)
            else:
                print('查無此人')
                judge('3')
    elif x == '2':
        stu_id = input('請(qǐng)輸入要修改的學(xué)生的學(xué)號(hào)')
        for dict1 in all_stu:
            if dict1['stu_id'] == stu_id:
                print('學(xué)生信息', dict1)
                dict1['age'] = int(input('請(qǐng)輸入新的年齡'))
                print('修改成功羔巢！', dict1)
            else:
                print('查無此人')
                judge('3')
    elif x == '3':
        stu_id = input('請(qǐng)輸入要修改的學(xué)生的學(xué)號(hào)')
        for dict1 in all_stu:
            if dict1['stu_id'] == stu_id:
                print('學(xué)生信息', dict1)
                dict1['tel'] = input('請(qǐng)輸入新的電話')
                print('修改成功交惯！', dict1)
            else:
                print('查無此人')
                judge('3')
    judge('3')


def del_stu(name):
    for dict1 in all_stu:
        if dict1['name'] == name:
            all_stu.remove(dict1)
            print('刪除成功！')
        else:
            print('查無此人')
    home()


# 功能判斷
def judge(x: str):
    if x == '1':
        name = input('請(qǐng)輸入學(xué)生姓名：')
        age = int(input('請(qǐng)輸入學(xué)生年齡：'))
        tel = input('請(qǐng)輸入學(xué)生電話：')
        add_stu(name, age, tel)
        print('1.繼續(xù)添加')
        print('2.返回')
        x = input('請(qǐng)選擇1~2：')
        if x == '1':
            judge(x)
        else:
            home()
    elif x == '2':
        print('---------')
        print('1.查看所有學(xué)生')
        print('2.按姓名查找')
        print('3.按學(xué)號(hào)查找')
        print('4.返回')
        print('---------')
        r = input('請(qǐng)選擇1~5：')
        if r == '4':
            home()
        else:
            retrieval(r)
    elif x == '3':
        print('---------')
        print('1.修改學(xué)生姓名')
        print('2.修改學(xué)生年齡')
        print('3.修改學(xué)生電話')
        print('4.返回')
        print('---------')
        r = input('請(qǐng)選擇1~5：')
        if r == '4':
            home()
        else:
            revise1(r)
    elif x == '4':
        name = input('請(qǐng)輸入要?jiǎng)h除的學(xué)生的姓名：')
        del_stu(name)


def home():
    print('====學(xué)生管理系統(tǒng)====')
    print('')
    print('1.添加學(xué)生')
    print('2.查看學(xué)生')
    print('3.修改學(xué)生')
    print('4.刪除學(xué)生')
    print('5.返回')
    print('-------------------')
    x = input('請(qǐng)選擇1~5：')
    judge(x)


home()