函数版注册登陆小程序

user = dict()   #定义一个空的字典


def new_user():
    prompt = '请输入用户名:'
    while True:
        yhm = input(prompt)
        if yhm in user:
            prompt= '此用户名已经被使用,请重新输入:'
            continue
        else:
            break


    valmm = input('请输入密码:')
    user[yhm] = valmm
    print('注册成功,赶紧登陆试试吧^_^')


def log_user():
    prompt = '请输入用户名:'
    while True:
        yhm = input(prompt)
        if yhm not in user:
            prompt = '您输入的用户名不存在,请重新输入:'
            continue
        else:
            break
    prompt = '请输入密码:'
    while True:
        valmm = input(prompt)
        pwdin = user.get(yhm)
        if valmm == pwdin:  #valmm == user[yhm]
            print('欢迎进入XXOO系统,请点右上角的X结束程序!')
            break
        else:
            prompt = '密码错误!请重新输入:'


def reg_log():
    prompt ='''|---新建用户:N/n---|
|---登陆账户:E/e---|
|---退出程序:Q/q---|
|---请输入指令代码:'''


    while True:
        chg = False
        while not chg:
            chgin = input(prompt)
            if chgin not in 'NnEeQq':
                print('您输入的指令不合规范,请重新输入:',end = '')
            else:
                chg = True
        if chgin == 'Q' or chgin == 'q':
            break
        if chgin == 'N' or chgin == 'n':
            new_user()
        if chgin == 'E' or chgin == 'e':
            log_user()


reg_log()