python脚本实现用户注册登录功能

python脚本实现用户注册登录功能

本脚本通过模块调用实现用户注册登录功能,满足用户可以先注册后登陆未注册的账户无法登陆等条件.
具体代码实现如下

#!/root/bin/python
import passcreate
import getpass
adict = {}

def zhu_ce(zname,zpass):
    if zname in list(adict):
        print('用户已存在')
        return
    if not zpass:
        print('密码不能为空,已为您随机生成八位密码%s'% passabc.abc())
        zpass = passcreate.create()
    adict.update({zname:zpass})
    # adict[zname] = zpass
    print('注册成功,请登录')
    return zname,zpass

def deng_lu(dname,dpass):
    if dname not in list(adict):
        print('用户名或密码错误,请重新输入')
        return
    elif dpass not in list(adict.values()):
        print('用户名或密码错误,请重新输入')
        return
    else:
        print('登录成功!欢迎回来')
        quit()

if __name__ == '__main__':
     while 1:
        prompt = '''请选择
1.登录
2.注册
其他任意键退出'''
        choise = input(prompt).strip()
        if choise == '1':
            dname = input('请输入用户名:')
            dpass = getpass.getpass('请输入密码:')
            deng_lu(dname,dpass)
        elif choise == '2':
            zname = input('请输入用户名: ')
            zpass = getpass.getpass('请输入密码')
            zhu_ce(zname,zpass)
        else:
            break