Python每日一练10、模拟用户登录系统

                                需求
# 支持新用户注册(添加)，新用户名和密码注册到字典中
# 支持老用户登陆(查询)，用户名和密码正确提示登陆成功
# 主程序通过循环询问，进行何种操作，根据用户的选择，执行注册或是登陆操作

userdata = {'zhangsan':'123'} #定义一个字典，存储用户名和密码
def register(): #注册
    username = input('请输入用户名：')
    if len(username) != 0  and  username not in userdata.keys(): #userdb.keys()
        password = input('请输入密码：')
        userdata[username] = password
    else:
        print('username 不可用!')

def login(): #老用户登录
    username = input('请输入用户名：')
    password = input('输入密码：')
    if (username,password) in  userdata.items():
        print('登录成功！')
    else:
        print('密码错误！，已退出')

def show_menu(): #交互界面
    while True:
        xuanze = input('1-register 2-login 3-exit:')
        if xuanze not in ['1','2','3']:
            print('请输入正确选项！')
            continue
        elif xuanze == '3':
            print('Bye~')
            break
        elif xuanze == '1':
            register()
        else:
            login()


if __name__ == '__main__':
    show_menu()

运行结果：

1-register 2-login 3-exit:1
请输入用户名：uzi
请输入密码：123
1-register 2-login 3-exit: