函数版注册登陆小程序

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