python3 练习题(用函数完成登录注册以及购物车的功能)

'''
用函数完成登录注册以及购物车的功能
作业需求:
1，启动程序，用户可选择四个选项：登录，注册，购物，退出。
2，用户注册，用户名不能重复，注册成功之后，用户名密码记录到文件中。
3，用户登录，用户名密码从文件中读取，进行三次验证，验证不成功则退出整个程序。
4，用户登录成功之后才能选择购物功能进行购物，购物功能（就是将购物车封装到购物的函数中）。
5，退出则是退出整个程序。
'''
'''
分析:将用户信息保存到文件userinfo.txt(不考虑多个账号同时使用)
存储格式为{"alex":{"pwd":"123456", "status":"logged", "money": 8000, "shopping":{}},
            "wusir":{"pwd": "123456", "status":"logout","money": 6000, "shopping":{}},
            ......
            }
'''
#判断文件是否存在,不存在,则创建空文件
def mk_file():
    if not os.path.exists(filename):
        f = open(filename, mode="w", encoding="utf-8")
        f.close()

#获取文件中的内容,并将内容转换为dict类型
def get_file_content():
    usr_dic = {}
    with open(filename, mode="r", encoding="utf-8") as f:
        content = f.read().strip()
        if content != "":
            usr_dic = eval(content)
    return usr_dic

#修改文件内容
def set_file_content(usr_dic):
    with open("%s.bak" % filename, mode="w", encoding="utf-8") as f:
        f.write(str(usr_dic))
    os.remove(filename)
    os.rename("%s.bak" % filename, filename)

#用户注册
def register():
    while 1:
        usr = input("Register Username: ").strip()
        pwd = input("Register Password: ").strip()
        if usr != "" and pwd != "": # 判断用户名和密码是否合法
            if len(usr) < 3 or len(usr) > 30:
                print("\033[31;0m用户名的长度必须在3~30.\033[0m")
                continue

            if len(pwd) < 6 or len(pwd) > 20:
                print("\033[31;0m密码的长度必须在6~20.\033[0m")
                continue

            mk_file()   # 判断文件是否存在,不存在直接创建空文件
            usr_dic = get_file_content()    #读文件,判断用户是否存在,存在则提醒已存在,不存在则直接添加
            if usr in usr_dic:
                print("\033[31;0m注册失败, 用户名: %s已存在!\033[0m" % usr)
                return
            else:
                usr_dic[usr] = {"pwd": pwd}
                set_file_content(usr_dic)
                print("\033[32;0m恭喜您,用户名: %s注册成功!\033[0m" % usr)
                return True
        else:
            print("\033[31;0m用户名和密码不能为空!\033[0m")

#用户登录
def login():
    while 1:
        usr = input("Login Username: ").strip()
        pwd = input("Login Password: ").strip()
        if usr != "" and pwd != "": ## 判断用户名和密码是否为空
            mk_file()  #判断文件是否存在,不存在则创建空文件
            usr_dic = get_file_content() #读文件,判断用户名和密码是否正确
            if usr in usr_dic:
                if usr_dic[usr]["pwd"] == pwd:  #登录成功
                    return usr
            #用户密或密码错误
            return
        else:
            print("\033[31;0m用户名和密码不能为空.\033[0m")

#记录用户信息
def record_userinfo(usr, stat="logged", money=0, shopping={}):
    usr_dic = get_file_content()
    usr_dic[usr]["status"] = stat
    usr_dic[usr]["money"] = usr_dic[usr].get("money", 0) + money
    #更新用户已购买的商品信息
    if shopping != {}:
        sp_dic = usr_dic[usr].get("shopping", {})
        new_dic = shopping
        for k, v in sp_dic.items():
            if k in shopping:
                v[2] += shopping[k][2]
            new_dic[k] = v
        usr_dic[usr]["shopping"] = new_dic
    set_file_content(str(usr_dic))

#购物功能
def shopping(usr):
    shopping_lst = [("电脑", 2999), ("鼠标", 30), ("键盘", 80), ("音响", 888), ("耳机", 60)]
    shopping_carts = {}

    print("\033[32;0m欢迎来到淘宝电脑城\033[0m".center(50, "*"))
    while 1:
        money = input("请输入您要充值的金额(元): ").strip()
        if money.isdigit():
            money = int(money)
            usr_dic = get_file_content().get(usr, {}) #获取账号余额
            money_total = usr_dic.get("money", 0) + money
            print("\033[32;0m恭喜您,成功充值%d元! 您的账号余额为%d元.\033[0m" % (money, money_total))
            record_userinfo(usr, money=money) #记录用户充值金额
            money = money_total
            break
        else:
            print("输入有误,请重新输入!")

    while 1:
        #显示商品
        print("\033[32;0m商品列表\033[0m".center(50, "*"))
        print("编号\t商品名称\t商品价格")
        for i, v in enumerate(shopping_lst, 1):
            print("%d\t\t%s\t\t%s" % (i, v[0], v[1]))
        print("n\t\t商品结算")
        print("q\t\t退出")

        #用户选择商品
        choice = input("请输入您要购买的商品编号: ").strip()
        if choice.isdigit():
            choice = int(choice)
            if choice >= 1 and choice <= len(shopping_lst):
                sp_name = shopping_lst[choice-1][0]
                sp_price = shopping_lst[choice-1][1]
                if choice in shopping_carts: #判断商品是否已在购物车
                    shopping_carts[choice][2] += 1
                else:
                    shopping_carts[choice] = [sp_name, sp_price, 1]
                print("\033[32;0m恭喜您,成功将商品: %s添加购物车!\033[0m" % sp_name)
            else:
                print("\033[31;0m商品不存在\033[0m")
        elif choice.upper() == "N":
            while 1:
                if shopping_carts == {}: #判断购物车是否为空
                    print("\033[33;0m购物车空空如也!请先选择您要购买的商品,再来结算吧.\033[0m")
                    break

                print("\033[32;0m购物车商品信息\033[0m".center(50, "*")) #显示购物车商品
                print("序号\t商品名称\t商品单价\t商品数量")
                consume_total = 0
                for i in shopping_carts:
                    goods = shopping_carts[i]
                    print("%s\t\t%s\t\t%s\t\t\t%s" % (i, goods[0], goods[1], goods[2]))
                    consume_total += goods[1] * goods[2]

                #钱够,直接购买商品
                if money - consume_total >= 0:
                    money -= consume_total
                    print("\033[32;0m恭喜您,商品结算成功,本次消费共: %d元, 账号余额为%d元.\033[0m" % (consume_total, money))
                    record_userinfo(usr, shopping=shopping_carts, money=-consume_total) #更新用户余额和已购买商品信息
                    return
                #余额不足
                else:
                    print("\033[31;0m商品总金额: %d元, 您还差 %d元.\033[0m" % (consume_total, consume_total-money))
                    del_choice = input("请选择您要删除的商品序号: ").strip() #移除购物车商品
                    if del_choice.isdigit():
                        del_choice = int(del_choice)
                        if del_choice in shopping_carts: #判断用户输入的商品序列号是否在购物车中,若在,直接移除
                            shopping_carts[del_choice][2] -= 1
                            if shopping_carts[del_choice][2] == 0:
                                del shopping_carts[del_choice]
                        else:
                            print("\033[31;0m输入有误,请重新输入!\033[0m")
                    else:
                        print("\033[31;0m输入有误,请重新输入!\033[0m")
        elif choice.upper() == "Q":
            break
        else:
            print("\033[31;0m商品不存在!\033[0m")

import os
choice_lst = ["登录", "注册", "购物", "退出"]
print("\033[32;0m欢迎来到淘宝购物城\033[0m".center(50, "*"))
filename = "userinfo.txt"
usr = ""
Flag = 1

while Flag:
    #显示4个选项
    for i in choice_lst:
        print(i)
    choice = input("请输入你要做的操作: ").strip()
    # 判断用户输入的选项是否正确
    if choice in choice_lst:
        if choice == "登录":
            #可以进行三次验证
            for i in range(2, -1, -1):
                usr = login()
                if usr:
                    print("\033[32;0m恭喜您,账号: %s登录成功!\033[0m" % usr)
                    #登录成功后,记录登录状态
                    record_userinfo(usr)
                    break
                else:
                    if i == 0:  #进行三次验证，验证不成功则退出整个程序
                        print("\033[31;0m用户名或密码错误,您的账号已被锁住.\033[0m")
                        Flag = 0
                        break
                    else:
                        print("\033[31;0m用户名或密码错误,您还有%d次机会.\033[0m" % i)
        elif choice == "注册":
            register()
        elif choice == "购物":
            usr_dic = get_file_content().get(usr, {})
            if usr_dic == {}:
                print("\033[31;0m请先登录再购物!\033[0m")
            else:
                if usr_dic.get("status", 0) != "logged":
                    print("\033[31;0m请先登录再购物!\033[0m")
                else:
                    shopping(usr)
        elif choice == "退出":
            print("\033[32;0m欢迎下次光临\033[0m".center(50, "*"))
            #修改用户的登录状态
            if usr != "":
                record_userinfo(usr, stat="logout")
            break
    else:
        print("\033[31;0m输入错误, 请重新输入!\033[0m")