因为需要下一次运行程序的时候可以使用上一次注册的账号,所以已经注册过的账号需要持久化。
1、确定持久化对象:所有已经注册过的账号和对应密码
2、创建保存数据的文件
3、确定文件初始内容:一个用户占一行,每一行内容是’用户名,密码’
4、需要数据的时候从文件中获取数据
5、数据发生改变把最新的数据写入文件中
# 注册
def register():
print('-------- 正在注册--------')
username = input('请输入用户名(不能包含逗号):')
password = input('请输入密码(不能包含逗号):')
if ',' in username or ',' in password:
return '注册失败,账号或密码不合法'
else:
f2 = open('用户信息.txt', 'r', encoding='utf-8')
result = f2.readlines()
for i in result:
if username == i.strip().split(',')[0]:
return '注册失败!该账号已经存在!!!'
else:
f3 = open('用户信息.txt', 'a', encoding='utf-8')
f3.write('\n' + username + ',' + password)
return '注册成功!'
# 登录
def login():
print('-------- 正在登录--------')
username = input('请输入用户名:')
password = input('请输入密码:')
f2 = open('用户信息.txt', 'r', encoding='utf-8')
result = f2.readlines()
for i in result:
index = i.strip().split(',')
if username == index[0] and password == index[1]:
return '登录成功!'
elif username == index[0] and password != index[1]:
return '登录失败!密码错误'
return '登录失败!该账号没有注册!'
def choice():
# 选择2---注册
num = int(input('请输入(1~2):'))
if num == 2:
print(register())
# 选择1登录
elif num == 1:
print(login())
else:
print('输入有误!请重新输入!!')
choice()
def welcome():
# 打印首页
f1 = open('首页.txt', 'r', encoding='utf-8')
print(f1.read())
choice()
if __name__ == '__main__':
welcome()