python购物_Python-购物车系统

# coding=utf-8

import os, pickle

class color:

def echo_error(self, red):

print(f"\033[31;1m {red}\033[0m".strip())

def echo_ok(self, green):

print(f"\033[32;1m {green}\033[0m".strip())

def echo_warn(self, warning):

print(f"\033[33;1m {warning}\033[0m".strip())

color = color()

def file_open(file, mode, *args):

if mode == "wb":

data = args[0]

with open(file, mode) as f:

pickle.dump(data, f)

elif mode == "rb":

with open(file, mode) as f:

user_dict = pickle.load(f)

return user_dict

elif mode == "clear":

with open(file, mode) as f:

data = args[0]

pickle.dump(data, f)

def user(user, pwd, mode):

file = user + ".db"

if mode == "regist":

if not os.path.isfile(file):

user_dict = {"name": user, "pwd": pwd, "stat": 0}

file_open(file, "wb", user_dict)

color.echo_ok("注册成功！")

else:

color.echo_error("用户已注册")

elif mode == "login":

if os.path.isfile(file):

user_dict = file_open(file, "rb")

if user_dict["name"] == user and user_dict["pwd"] == pwd:

color.echo_ok("登录成功")

return user_dict

else:

color.echo_error("登录失败，用户名或密码错误")

else:

color.echo_error("不存在此用户，请先注册")

return None

else:

color.echo_error("错误")

def shopping(user_dict, list_goods):

if user_dict["shopping_car"]:

list_shopping = user_dict["shopping_car"]

else:

list_shopping = []

while True:

color.echo_ok("商品列表如下：\n"

"##########################\n"

"序号 商品名称 商品价格")

for i, k in enumerate(list_goods):

print(i, k["name"], k["price"])

chooice = input("请输入购买商品序号 [t 查看购物车 q 退出]")

if not chooice: continue

if chooice.isdigit() and 0 <= int(chooice) < len(list_goods):

num = input("请输入要购买的数量")

if num.isdigit():

num = int(num)

else:

color.echo_error("数量必须是正整数")

goods = {"name": list_goods[int(chooice)]["name"], "num": num,

"total_price": list_goods[int(chooice)]["price"] * num}

list_shopping.append(goods)

color.echo_ok("购物成功！")

elif chooice == "t":

color.echo_ok("#########购物清单##########")

print("商品名 数量 价格")

if len(list_shopping) == 0:

color.echo_warn("您没有购买任何东西..")

any = input("输入任意字符继续..")

if any: continue

continue

price = 0

for i in list_shopping:

print(i["name"], i["num"], i["total_price"])

price += i["total_price"]

chooice = input("总价格：%s，是否购买(yes/no)，[e 查看余额 c 清空购物车]:" % price)

if chooice == "c":

list_shopping = []

user_dict["shopping_car"] = list_shopping

color.echo_ok("购物车清空成功")

file_open(user_dict["name"] + ".db", "wb", user_dict, "clear")

elif chooice == "e":

color.echo_ok("您的余额为 %s" % user_dict["money"])

elif chooice == "yes":

if user_dict["money"] > price:

res_money = user_dict["money"] - price

color.echo_ok(f"购买成功，您的余额为 {res_money}")

list_shopping = []

user_dict["shopping_car"] = list_shopping

user_dict["money"] = res_money

file_open(user_dict["name"] + ".db", "wb", user_dict, "clear")

any = input("输入任意字符继续..")

if any: continue

else:

color.echo_error("您的余额不够买这些东西，挣点钱再来吧。")

elif chooice == "no":

color.echo_warn("您取消购买")

any = input("输入任意字符继续..")

if any: continue

elif chooice == "q":

color.echo_warn("退出程序")

break

def start(list_goods):

while True:

print('''

1. 注册

2. 登录并购物

3. 登出

''')

chooce = input(">>>").strip()

if chooce == "1":

username = input("请输入你的用户名：").strip()

pwd = input("请输入你的密码：").strip()

user(username, pwd, "regist")

elif chooce == "2":

username = input("请输入你的用户名：").strip()

pwd = input("请输入你的密码：").strip()

user_dict = user(username, pwd, "login")

if user_dict == None: continue

if user_dict["stat"] == 0:

money = int(input("请输入你的首冲金额："))

user_dict["stat"] += 1

user_dict["money"] = money

user_dict["shopping_car"] = []

color.echo_ok("恭喜，充值成功！，可以尽情购物了~")

file_open(user_dict["name"] + ".db", "wb", user_dict)

shopping(user_dict, list_goods)

else:

shopping(user_dict, list_goods)

elif chooce == "3":

break

else:

color.echo_error("非法操作，请重输入")

if __name__ == "__main__":

list_goods = [

{"name": "苹果", "price": 10},

{"name": "芒果", "price": 20},

{"name": "提子", "price": 30},

{"name": "香蕉", "price": 40},

]

start(list_goods)