def register(startname):
while True:
user_name = input("请输入注册用户名(q退出)：").strip()
if user_name == "q": break
if not user_name:
print("用户名不能为空")
continue
pwd = input("请输入密码：").strip()
if not pwd:
print("密码不能为空")
continue
pwd1 = input("请再次输入密码：").strip()
pwd2 = DB_handler.halib_file(pwd1)
if not pwd1:
print("密码不能为空")
continue
if pwd != pwd1:
print("两次密码不一致")
continue
# 将数据按照不同的用户传给对应的接口层
if startname == "admin":
obj = admin_interface.admin_resister(user_name, pwd2)
if obj:
print("管理员注册成功")
logger = get_logging("admin")
logger.info("%s管理员注册成功" % (user_name))
return
else:
print("用户名已存在，")
elif startname == "student":
obj = student_interface.student_resister(user_name, pwd2)
if obj:
print("学生注册成功,请重新启动")
logger = get_logging("student")
logger.info("%s学生注册成功" % (user_name))
exit(0)
else:
print("用户名已存在，")
else:
obj = teacher_interface.teacher_resister(user_name, pwd2)
if not obj:
print("老师注册成功")
logger = get_logging("teacher")
logger.info("%s老师注册成功" % user_name)
return user_name
else:
print(obj)
def login(startname):
global Logging, name
while True:
username = input("请输入登录用户名：").strip()
filenames = DB_handler.get_all_filename(startname.lower())
if username in filenames:
pwd = input("请输入登录密码：").strip()
obj1 = DB_handler.halib_file(pwd)
obj = DB_handler.load_obj_from_file(startname, username)
pwd2 = getattr(obj, "pwd")
if obj1 == pwd2:
Logging = True
print("登录成功！！！")
name = obj
return name
print("密码错误！！！")
continue
while True:
if startname == "student" or startname == "admin":
print("""
======================
1.注册
2.返回
======================
""")
choice = input("请选择：")
if choice == "1":
obj = register(startname)
if not obj:
return
elif choice == "2":
return
else:
print("输入错误！！！")
else:
print("请联系管理员帮你注册")
return