# 1、作业：
# 先让你登录
# {
# "nhy":{"passwd":123,'role':1,'money':10000,"carts":['mac']},
# "nhy2":{"passwd":123,'role':2,'money':10000,"carts":[]}
# }
# role是1的话，代表管理员，2代表普通用户
# 需求分析
# 1、存用户信息的这个文件、存商品信息的文件，都需要初始化
# 2、登录
# 3、登录成功之后判断角色展示不同的菜单
#普通用户：
# 1、查看所有商品，打印字典 2、购买商品，字典的修改，修改完之后写文件 3、查看购物车和余额，打印字典 4、退出
#管理员：
# 1、添加商品，字典的修改，修改完之后写文件 2、充值、输入用户名，字典的修改，修改完之后写文件 3、退出
# 公共的，就是读文件、写文件
# user.txt = {
'ssj': {'password': '123456', 'role': 1, 'money': 10000.0, 'carts': ['mac']},
'hua': {'password': '123456', 'role': 2, 'money': 100000.0, 'carts': []}
}
# goods.txt = {
'mac': 5000,
'iphone': 9000,
'car': 20000
}
USER_FILE = 'user.txt' # 用常量保存用户信息
GOODS_FILE = 'goods.txt' # 用常量保存商品信息
def op_file(file_name, content=None): # content=None，如果content为真是写操作
f = open(file_name, 'a+')
f.seek(0)
if content:
f.truncate()
f.write(str(content))
f.flush()
res = None
else:
res = eval(f.read())
f.close()
return res
def not_null(var): # 输入的内容不为空的函数
if len(var) > 0:
return True
return False
def is_price(s): # 合法的价格函数
s = str(s)
if s.count('.') == 1:
new_s = s.split('.')
left_num = new_s[0]
right_num = new_s[-1]
if right_num.isdigit():
if left_num.isdigit():
return True
elif s.isdigit() and int(s) > 0:
return True
return False
def add_products(): # 管理员添加商品
products = op_file(GOODS_FILE)
while True:
product_name = input('请输入你要添加的商品名称：').strip()
money = input('请输入你的商品价格：').strip()
if not_null(product_name) and not_null(money):
if product_name not in products:
if is_price(money): # 价格合法的话
products[product_name] = float(money)
op_file(GOODS_FILE, products)
print('商品添加成功！')
return admin() # 添加成功，都会返回上级菜单
else:
print('价格输入不合法！')
else:
print('商品已存在！')
else:
print('商品名称/价格不能为空！')
def recharge(): # 管理员充值
users = op_file(USER_FILE)
while True:
username = input('请输入你要充值的账号：').strip()
money = input('请输入你要充值的钱：').strip()
if not_null(username) and not_null(money):
if username in users:
if is_price(money):
src_money = users[username]['money']
users[username]['money'] = src_money + float(money)
op_file(USER_FILE, users)
print('充值成功！')
return admin() # 充值成功，都会返回上级菜单
else:
print('钱输入不合法！')
else:
print('用户不存在！')
else:
print('账号/钱不能为空！')
def admin(): # 管理员菜单函数
choice = input('请输入你的选择：1、充值 2、添加商品 3、退出').strip()
menu = {
'1': recharge,
'2': add_products,
'3': exit
}
if choice in menu:
menu[choice]()
else:
print('请输入正确的选择！')
admin() # 输入不正确，再调用一下自己重新输入
def show_products(user): # 普通用户查看商品，要传入一个user，哪个账号查看商品
products = op_file(GOODS_FILE)
for k in products:
print('商品名称是%s,价格是%s' % (k, products[k]))
buyer(user)
def buy(user): # 普通用户购买商品，要传入user
# 1、把原来的用户信息读出来
# 2、买完之后，再写进去
users = op_file(USER_FILE)
products = op_file(GOODS_FILE)
money = users[user]['money']
carts = users[user]['carts']
for k in products:
print('商品名称是%s,价格是%s' % (k, products[k]))
while True:
product_name = input('请输入你要购买的商品名称：').strip()
if not_null(product_name):
if product_name in products:
product_price = products[product_name]
if money >= product_price:
carts.append(product_name)
money -= product_price
users[user]['carts'] = carts
users[user]['money'] = money
op_file(USER_FILE, users)
print('商品购买成功！')
buyer(user)
else:
print('对不起，你的余额不足，你的余额是 %s' % money)
else:
print('你输入的商品名称不存在！')
else:
print('商品名称不能为空！')
def user_info(user): # 把哪个用户的信息打印出来
users = op_file(USER_FILE)
money = users[user]['money']
carts = users[user]['carts']
print('你的余额是%s' % money)
print('你的购物车里有', carts)
buyer(user)
def buyer(user): # 普通用户菜单函数
choice = input('请输入你的选择：1、查看商品信息 2、购买商品 3、查看购物车 4、退出').strip()
menu = {
'1': show_products,
'2': buy,
'3': user_info,
'4': exit
}
if choice in menu:
menu[choice](user)
else:
print('请输入正确的选择！')
buyer(user)
def login(): # 登录函数读文件
users = op_file(USER_FILE)
for i in range(3):
username = input('请输入账号：').strip()
password = input('请输入密码：').strip()
if not_null(username) and not_null(password): # 账号和密码都不为空
if username in users:
cur_user = users[username] # 代表通过账号获取到的小字典
if cur_user['password'] == password:
if cur_user['role'] == 1:
print('欢迎管理员登录')
admin()
else:
print('欢迎%s登录' % username)
buyer(username)
else:
print('用户不存在！')
else:
print('账号/密码不能为空！')
login()