论.技术.Python.(旧)零学Python第026讲:某一功能代码块可以分开定义避免函数臃肿

小蜂队

尝试编写一个用户登录程序，代码是课件源代码（要求：用函数封装）

user_data = {}

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

    passwd = input('请输入密码：')
    user_data[name] = passwd
    print('注册成功，赶紧试试登录吧^_^')

def old_user():
    prompt = '请输入用户名：'
    while True:
        name = input(prompt)
        if name not in user_data:
            prompt = '您输入的用户名不存在，请重新输入：'
            continue
        else:
            break
    
    passwd = input('请输入密码：')
    pwd = user_data.get(name)
    if passwd == pwd:
        print('欢迎进入XXOO系统，请点右上角的X结束程序！')
    else:
        print('密码错误！')

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

    while True:
        chosen = False
        while not chosen:
            choice = input(prompt)
            if choice not in 'NnEeQq':
                print('您输入的指令代码错误，请重新输入：')
            else:
                chosen = True

        if choice == 'q' or choice == 'Q':
            break
        if choice == 'n' or choice == 'N':
            new_user()
        if choice == 'e' or choice == 'E':
            old_user()

showmenu()

小蜂队