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

发布时间：2019-09-26 07:27:17编辑：auto阅读（1755）

    '''

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

    '''

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


关键字：

上一篇： python3 在不同操作系统安装第三方

下一篇： python3 三级菜单-基础版