《云计算全栈》-python篇：模拟用户登陆信息系统

1 案例1：模拟用户登陆信息系统
1.1 问题

编写login.py脚本，实现以下目标：

支持新用户注册，新用户名和密码注册到字典中
支持老用户登陆，用户名和密码正确提示登陆成功
主程序通过循环询问进行何种操作，根据用户的选择，执行注册或是登陆操作
123
1.2 方案
创建空字典存储用户名、密码，用三个函数分别实现用户注册、用户登录以及判断调用函数这三个方法，完成模拟用户登录：
1.调用show_menu()函数后，利用while循环交互端输出提示，请用户input0/1/2任意数值，如果输入的值不是0/1/2，打印选择无效请重新输入并重新开始循环，如果输入的值是2，停止整个循环，如果输入的值是0/1/2通过字典键值对关联关系，调用相对应函数
2.如果输入的值是0，字典cmds中0键对应的值是register，register ()调用注册函数，函数利用if方法判断输入用户名是否存在，如果用户名在字典中，输出用户名已存在，否则用户输入密码，并将用户名与密码以键值对形式放入字典中
3.如上，如果输入的值是1，调用登录函数login()，利用if方法判断输入的用户名的对应的密码是否和字典中存储用户对应密码相同，如果不同显示登录失败，否则登录成功，此函数中导入getpass模块使用方法，作用是输入密码不可见。

1.3 步骤
实现此案例需要按照如下步骤进行。
步骤一：编写脚本
    [root@localhost day05]# vim login.py
    #!/usr/bin/env python3
    import getpass
    userdb = {}
    def register():
        username = input('username: ')
        if username in userdb:
            print('\033[31;1m%s already exists.\033[0m' % username)
        else:
            password = input('password: ')
            userdb[username] = password
    def login():
        username = input('username: ')
        password = getpass.getpass('password: ')
        # if username not in userdb or userdb['username'] != password:
        if userdb.get(username) != password:
            print('\033[31;1mLogin incorrect\033[0m')
        else:
            print('\033[32;1mLogin successful\033[0m')
    def show_menu():
        prompt = """(0) register
    (1) login
    (2) quit
    Please input your choice(0/1/2): """
        cmds = {'0': register, '1': login}
        while True:
            choice = input(prompt).strip()[0]
            if choice not in '012':
                print('Invalid choice. Try again.')
                continue
            if choice == '2':
                break
            cmds[choice]()
    if __name__ == '__main__':
        show_menu()
1234567891011121314151617181920212223242526272829303132333435
步骤二：测试脚本执行
[root@localhost day05]# python3 login.py 
(0) register
(1) login
(2) quit
Please input your choice(0/1/2): 0
username: a
password: 123
(0) register
(1) login
(2) quit
Please input your choice(0/1/2): 1
username: a
password: 
Login successful
(0) register
(1) login
(2) quit
Please input your choice(0/1/2): 1
username: b
password: 
Login incorrect
(0) register
(1) login
(2) quit
Please input your choice(0/1/2): 2
12345678910111213141516171819202122232425