python06

import sys
import getpass
 
 
# 保存所有用户
users = []
 
 
def main_menu():
    print("~ * ~" * 10)
    print("\t\t\t 1、用户注册")
    print("\t\t\t 2、用户登录")
    print("\t\t\t 3、退出系统")
    return input("请输入您的操作：")
 
 
def user_exists(username):
    for user in users:
        # {"username": "ljh",  "password": "1234"}
        if user.get("username") == username:
            return user
    return None
 
 
def reg():
    while True:
        username = input("请输入用户名称：")
        password = getpass.getpass("请输入用户密码：")
        confirm_pwd = getpass.getpass("请再次输入密码：")
 
        # 数据校验
        if username == None or username == "":
            print("对不起，用户名不能为空！！")
            continue
        if len(password) < 3:
            print("对不起，密码不能少于3位！！")
            continue
        if password != confirm_pwd:
            print("对不起，两次密码不一致，请重新输入！！")
            continue
 
        # 用户名称是否重复
        if user_exists(username) != None:
            print("对不起，该用户名称已经存在，请重新注册！！")
            continue
 
        # 保存用户
        user = {"username": username, "password": password}
        users.append(user)
        chioce = input("输入N/n则退出到上一个界面，否则继续注册")
        # if choice == "N" or chioce == "n":
        if choice.upper() == "N":
            break
 
 
def login():
    while True:
        username = input("请输入用户名称：")
        password = getpass.getpass("请输入用户密码：")
        user = user_exists(username)
        if user is not None:
            if user.get("password") == password:
                print("登录成功！！")
                print("欢迎到您的个人中心页面！！")
            else:
                print("对不起，密码错误")
        else:
            print("对不起，没有这个用户")
 
 
if __name__ == '__main__':
    while True:
        print(users)
        choice = main_menu()
        if choice == "1":
            # 注册
            reg()
        elif choice == "2":
            # 登录
            login()
        elif choice == "3":
            sys.exit()