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()
