2018-10-15作业

import yxy_login
from add_stu import add_stu
import check_stu
from delete_stu import del_stu
from revise_stu import revise_stu

作业二：写一个登陆注册

1.登陆
2.注册
3.退出
====================
2
输入账号： aaa
输入密码：123456
提示注册成功！

====================
1.登陆
2.注册
3.退出
====================
1
账号：aaa
密码：123
密码错误

账号：bbb
账号不存在

def main():
    user_name = yxy_login.login_menu()
    if not user_name:
        print('退出系统!')
    else:
        # 编辑主界面循环
        while True:
            print('+' * 60)
            print('+' + '欢迎来到非常非常简易版的学生管理系统^_^'.center(40, ' ') + '+')
            print('+' + '1.添加学生信息'.center(52, ' ') + '+')
            print('+' + '2.查询学生信息'.center(52, ' ') + '+')
            print('+' + '3.删除学生信息'.center(52, ' ') + '+')
            print('+' + '4.修改学生信息'.center(52, ' ') + '+')
            print('+'+'5.退出'.center(56, ' ')+'+')
            print('+' * 60)
            try:
                step1 = int(input('请输入您想进行的操作：'))
            except TypeError:
                print('请输入正确的数字！')
                continue

            if step1 == 5:
                break

            # 编辑添加学生的循环界面
            elif step1 == 1:
                while True:
                    add_stu(user_name)
                    print('1.继续添加')
                    print('2.返回上一步')
                    try:
                        step2 = int(input('请输入您想进行的操作： '))
                    except TypeError:
                        print('请输入正确的数字！')
                        continue

                    if step2 == 2:
                        break
                    elif step2 == 1:
                        continue
                    else:
                        print('请输入正确的数字！')
            # 编辑查询学生的界面
            elif step1 == 2:
                while True:
                    print('1.查询指定学生信息')
                    print('\n2.查询所有学生信息')
                    print('\n3.查询指定学生平均分')
                    print('\n4.返回上一步')
                    try:
                        step3 = int(input('请输入您想进行的操作： '))
                    except TypeError:
                        print('请输入正确的数字！')
                        continue
                    if step3 == 4:
                        break
                    elif step3 == 1:
                        check_stu.search_stu(user_name)
                    elif step3 == 2:
                        check_stu.check_all(user_name)
                    elif step3 == 3:
                        check_stu.avg_score(user_name)
                    else:
                        print('请输入正确的数字！')
            # 编辑删除学生的界面
            elif step1 == 3:
                del_stu(user_name)
            # 编辑修改学生的界面
            elif step1 == 4:
                revise_stu(user_name)

            else:
                print('请输入正确的数字！')


if __name__ == '__main__':

    main()