'''
登陆预注册系统
版本：v1.0
作者：Euan
日期：2018/10/29
附带文件：userdata (用于保存用户账号与密码信息)
功能：实现用户账号注册与登陆：
1.目前无法实现重复注册；
2.用户名注册后无法重新选择登陆或注册；
3.没有加入登陆验证码。
'''
## 用户注册函数函数开始
def enroll():
# 读取userdata中的信息
f = open("loginSystem_userdata", mode="r+" ,encoding="utf-8")
while 1:
# 输入用户名和密码
username = input("请输入你的用户名:")
password = input("请输入密码:")
# 判断账号是否注册过
for data in f:
if data.strip().split(":")[0] == username:
print("\n**********该用户名已注册**********\n")
print("\n**********请重新注册**********\n")
break
# 将注册账号信息存储在userdata文件中
else:
print("\n**********恭喜, 注册成功**********\n")
f.write( username + ":" + password + "\n")
f.flush()
f.close()
return # 直接结束函数的调用
## 用户注册函数函数结束
## 用户登陆登陆函数开始
def entry():
username = input("请输入用户名:")
password = input("请输入密码:")
f = open("loginSystem_userdata", encoding="utf-8")
for line in f:
if line.strip() == username + ":" + password:
print("\n**********恭喜你.登录成功**********\n")
return True # True, 登录成功
else:
print("\n**********对不起, 用户名或密码错误**********\n")
return False # False 登录失败
## 用户登陆登陆函数结束
## 用户注册登陆函数调用开始
while 1:
print("注册请选择(1)\n登陆请选择(2)")
choice = input('请输入你的选择：')
if int(choice) == 1:
enroll()
break
elif int(choice) == 2:
entry()
break
else:
print("\n**********输入有误，请重新输入**********\n")
## 用户注册登陆函数调用结束
