'''
此模块需要使用用户登陆模块代码
用户入口：
1、输入用户名和密码
2、商品信息存在文件里
3、已购商品和余额记录在文件里
注：用户购买商品后退出，再次登陆会自动读取上次用户的余额
商户入口:
1、添加商品
2、修改商品名称和价格
3、删除商品
备注：需要新建shop文件和购物车文件，其中shop文件保存商品信息：
'''
import os
import userLogin
import types
def read_info(_name):
if os.path.getsize("购物车"):
remaining = 0
with open("购物车","r",encoding="utf-8") as f:
for line in f:
if line.strip() == "":
continue
line = eval(line)
if line["name"] == _name:
remaining = line["shopping"][2]
if remaining > line["shopping"][2]:
remaining = line["shopping"][2]
if remaining == 0:
return None
else:
return remaining
else:
return None
def memu(num):
if os.path.getsize("shop"):
with open("shop","r",encoding="utf-8") as f:
print("------------商品------------")
for index,line in enumerate(f):
if line.strip() == "":
continue
index -= 1
line = eval(line)
info2 = '''
{_index}、{_goods},价格：{_price}
'''.format(_index = index + 1,
_goods = line["goods"],
_price = line["price"])
print(info2)
else:
print("当前无商品")
if num == 1:
exit()
def shopping_list(name):
sum = 0
flag = 0
with open("购物车", "r+", encoding="utf-8") as f:
for temp in f:
if temp.strip() == "":
continue
temp = eval(temp)
if temp["name"] == name:
sum += temp["shopping"][1]
info2 = '''
商品：{_goods}，价格:{_price}
'''.format(_goods=temp["shopping"][0],
_price=temp["shopping"][1])
if flag == 0:
print("---------购物清单---------")
print(info2)
flag = 1
print("总计金额： %d" % sum)
def user(_momey,name):
choose = input("输入你要购买商品的编号 >>>：")
if choose.isdigit():
choose = int(choose)
elif choose == 'q':
shopping_list(name)
exit()
else:
print("输入错误")
res = read_info(name)
if res != None:
_momey = res
f = open("shop", "r", encoding="utf-8")
content = f.readlines()
for i in range(len(content)):
if choose - 1 == i:
content[i] = eval(content[i])
surplus = _momey - content[i]["price"]
if surplus >= 0:
_momey = surplus
print(" 33[33;1m %s 33[0m 已加入购物车，你的当前余额为： 33[32;1m %d 33[0m "%(content[i]["goods"],_momey))
save(content[i],name,_momey)
return _momey
else:
print(" 33[31;1m 你的余额不足，当前余额为：%d 33[0m"%_momey)
return _momey
f.close()
def save(d,name,momey):
dict = {}
dict["name"] = name
dict["shopping"] = [d["goods"],d["price"],momey]
with open("购物车","a+",encoding="utf-8") as f:
f.write(str(dict) + '
')
def deposit(name):
while True:
res = read_info(name)
if res == None:
momey = input("输入你的存款金额 >>>：")
else:
momey = res
print(" 33[34;1m %s你好，你的当前余额是：%d 33[0m"%(name,momey))
return momey
if type(momey) != type(1):
if momey.isdigit():
momey = int(momey)
return momey
elif momey == 'q':
exit()
else:
print("存款输入错误")
continue
def add_shop():
with open("shop","a+",encoding="utf-8") as f:
dict = {}
goods = input("输入你要添加的商品名称 >>>")
price = input("输入商品的价格(正整数) >>>")
if price.isdigit():
price = int(price)
dict["goods"] = goods
dict["price"] = price
f.write(str(dict) +'
')
print(" 33[31;1m 添加商品成功 33[0m")
def modify_shop():
list = []
f = open("shop", "r+", encoding="utf-8")
content = f.readlines()
num = input("请输入你要修改的商品编号 >>>")
if num.isdigit():
num = int(num)
else:
print("输入错误")
for i in range(len(content)):
if num - 1 == i:
content[i] = eval(content[i])
choose = input("请输入你要修改商品的信息1、名称，2、价格 >>>")
if choose.isdigit():
choose = int(choose)
else:
print("输入错误")
if choose == 1:
goods = input("请输入修改后的商品名称 >>>")
content[i]["goods"] = goods
elif choose == 2:
price = input("请输入修改后的商品价格 >>>")
if price.isdigit():
price = int(price)
content[i]["price"] = price
else:
print("输入错误")
list.append(content[i])
f.truncate(0)
f.close()
with open("shop", "a+", encoding="utf-8") as f1:
for temp in list:
f1.write(str(temp))
print(" 33[31;1m 修改商品成功 33[0m")
def del_shop():
list = []
f = open("shop", "r+", encoding="utf-8")
content = f.readlines()
num = input("请输入你要删除的商品编号 >>>")
if num.isdigit():
num = int(num)
else:
print("输入错误")
for i in range(len(content)):
if num - 1 == i:
continue
list.append(content[i])
f.truncate(0)
f.close()
with open("shop", "a+", encoding="utf-8") as f1:
for temp in list:
f1.write(temp)
print(" 33[31;1m 删除商品成功 33[0m")
def main1(name):
res = input("请输入你的选择:1、用户，2、商户 >>>")
if res.isdigit():
res = int(res)
if res == 1:
momey = deposit(name)
while True:
memu(res)
momey = user(momey,name)
elif res == 2:
while True:
memu(res)
res1 = input("请输入你的选择:1、添加，2、修改，3、删除 >>>")
if res1.isdigit():
res1 = int(res1)
if res1 == 1:
add_shop()
elif res1 == 2:
modify_shop()
elif res1 == 3:
del_shop()
else:
print("输入错误")
elif res1 == 'q':
exit()
else:
print("输入错误")
elif res == 'q':
exit()
else:
print("输入错误")
if __name__ == "__main__":
while True:
userLogin.main()
'''
-------用户登陆模块代码------
'''
'''
先创建user、lock文件，user存储账号密码，用于登录时判断，lock文件为账号错误3次后，冻结
user内容格式为：
{'name':'zhangsan','passwd':'123'}
{'name':'lisi','passwd':'123'}
{'name':'wangwu','passwd':'123'}
'''
import os
import shopping
def lock(_name):
'''判断一个用户是否冻结'''
if os.path.getsize("lock"):
with open("lock","r") as f:
for line in f:
if line.strip() == "":
continue
line = eval(line)
if line["name"] == _name and line["num"] == 3:
return 0
else:
return -1
else:
return -1
def clear_lock():
'''当程序退出时，删除非冻结的账号信息'''
list = []
with open("lock","r+") as f:
for line in f:
if line.strip() == "":
continue
line = eval(line)
if line["num"] == 3:
list.append(line)
f.truncate(0)
with open("lock","w") as f1:
for i in list:
f1.write(str(i) + '
')
def login(_name,_passwd):
'''用户登陆'''
with open("user","r") as f:
flag = 0
for line in f:
if line.strip() == "":
continue
line = eval(line)
if line["name"] == _name:
if line["name"] == _name and line["passwd"] == _passwd:
return 0
else:
return 1
return -1
def write_lock(_name):
'''将输入错误的账号写入lock文件'''
#文件不为空
if os.path.getsize("lock"):
list = []
with open("lock","r+") as f:
flag = 0
for line in f:
if line.strip() == "":
continue
line = eval(line)
#判断账号是否存在
if line["name"] == _name:
line["num"] += 1
list.append(line)
else:
dict2 = {}
dict2["name"] = _name
dict2["num"] = 1
list.append(dict2)
with open("lock","w") as f1:
for i in list:
f1.write(str(i) + '
')
#空文件直接写入
else:
list1 = []
dict1 = {}
dict1["name"] = _name
dict1["num"] = 1
list1.append(str(dict1))
with open("lock","w") as f2:
for j in list1:
f2.write(str(j) + '
')
def main():
name = input("用户名:")
res = lock(name)
if res == 0:
print("%s已被冻结，请联系管理员"%name)
clear_lock()
exit()
passwd = input("密码:")
res1 = login(name,passwd)
if res1 == 0:
print("欢迎%s登陆CBD购物中心"%name)
while True:
shopping.main1(name)
elif res1 == 1:
print("密码错误")
write_lock(name)
else:
print("账号不存在")
res2 = lock(name)
if res2 == 0:
print("%s已被冻结，请联系管理员" %name)
clear_lock()
exit()
if __name__ == "__main__":
while True:
main()