python3 简单登录，注册测试代码

'''

写个登录，注册的简单代码。只用来简单测试，实现简单功能，没有过多的条件判断。仅仅用来练习。

'''

def login(username,password):

'''

db这个文件的路径是当前目录

打开保存账号密码的文件，将文件中的账号密码读出来，以字典格式保存。

如果输入的账号已经存在于字典中的键，结果返回FALSE。

如果账号存在字典且对应的键值等于输入的密码，则返回True。

'''

fo=open('db','r')

dict1={}

for i in fo:

i=i.strip('\n')

li=i.split('  ')

dict1.setdefault(li[0],li[1])

if username in dict1.keys() and dict1[username]==password:

return True

else:

return False

fo.close()

def register():

'''

打开保存账号密码的文件，将账号名保存为单独的一个列表。

如果输入的账号已经存在于这个列表中，则提示账号已经存在

如果输入的账号不存在这个列表，则将账号和密码写入到打开的这个文件保存。

'''

new_user=input('请输入需要申请的用户名：')

fr=open('db','r')

info_lists=fr.readlines()

name_lists=[]

for i in info_lists:

name=i.strip('\n').split(' ')[0]

name_lists.append(name)

if  new_user not in name_lists:

new_pwd=input('请输入密码：')

fo=open('db','a')

up='\n'+new_user+'  '+new_pwd

fo.write(up)

fo.close()

else:

print('用户名已存在')

fr.close()

# #

def main():

'''

程序入口，选择登录或是注册

调用登录函数或是注册函数，根据返回值提示注册成功或是失败。

'''

info=input('1,login;2,register:')

if info=='1':

user=input('用户名：')

pwd=input('密码：')

if login(user,pwd):

print ('chenggong')

else:

print('shibai')

elif info=='2':

register()

main()