python基础小练习

需求：实现一个简单用户注册、登录系统，在控制台实现即可

---

欢迎使用XXX系统

1 注册
2 登录
3 退出

请选择：1

---

请输入用户名：zhang3
请输入密码：123
注册成功！

---

欢迎使用XXX系统

1 注册
2 登录
3 退出

请选择：2

---

请输入用户名：zhang3
请输入密码：123

欢迎zhang3

---

欢迎使用XXX系统

1 注册
2 登录
3 退出

请选择：2

---

---------------------异常系---------------------------------

---

欢迎使用XXX系统

1 注册
2 登录
3 退出

请选择：1

---

请输入用户名：zhang3
用户名已被占用，请重新输入

---

欢迎使用XXX系统

1 注册
2 登录
3 退出

请选择：2

---

请输入用户名：zhang3
请输入密码：123

用户名或密码错误！

---

欢迎使用XXX系统

1 注册
2 登录
3 退出

请选择：

---

相关代码如下：

user = {"admin": 123, "zhangsan": 123, "lisi": 1234, "yk": 123}


def main():
    flag = True
    while flag:
        print("*"*16)
        print("欢迎使用本注册系统")
        print("")
        print("1 注册")
        print("2 登录")
        print("3 退出")
        print("")
        a = int(input("请选择："))

        if a == 1:
            zhuce()
        elif a == 2:
            denglu()
        elif a == 3:
            print("期待您的下次光临，再见~")
            flag = False
        else:
            print("输入有误，请重新输入：")


def zhuce():
    lname = input("请输入用户名：")
    if lname in user.keys():
        print("用户名重复！")
    else:
        lpass = input("请输入密码：")
        user[lname] = lpass


def denglu():
    uname = input("请输入用户名：")
    if uname in user.keys():
        upass = input("请输入密码：")
        if upass == str(user[uname]):
            print("登录成功，欢迎您，%s" % uname)
        else:
            print("密码输入有误")
    else:
        print("用户名输入有误")
    print(user)


if __name__ == '__main__':
    main()