# 1、作业：# 先让你登录   # {   # "nhy":{"passwd":123,'role':1,'money':10000,"carts":['mac']},   # "nhy2":{"passwd":123,'role':2,'money':10000,"carts":[]}   # }   # role是1的话，代表管理员，2代表普通用户   # 普通用户：      # 1、查看所有的商品 2、输入商品名称购买商品，购买完了之后要加到自己的购物车里面，钱要减去 3、可以查看自己购物车和余额，你已经买了xx东西，你还xxx钱 4、退出

   # 管理员：      #1、添加商品，商品名称，商品价格 2、充值，输入用户名，给他加钱 3、退出

# 需求分析# 1、存用户信息的这个文件、存商品信息的文件，都需要初始化# 2、登录# 3、登录成功之后判断角色展示不同的菜单

#普通用户：# 1、查看所有商品，打印字典 2、购买商品，字典的修改，修改完之后写文件 3、查看购物车和余额，打印字典 4、退出

#管理员：# 1、添加商品，字典的修改，修改完之后写文件 2、充值、输入用户名，字典的修改，修改完之后写文件 3、退出

# 公共的，就是读文件、写文件

USER_FILE = 'user.txt' # 用常量保存用户信息GOODS_FILE = 'goods.txt' # 用常量保存用户信息

def op_file(file_name,content = None): # content = None，如果content为真是写操作    f = open(file_name,'a+',encoding = 'utf8')    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): # 普通用户查看商品    products = op_file(GOODS_FILE)    for k in products:        print('商品名称是%s,价格是%s' % (k,products[k]))    return 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('商品购买成功！')                    return 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)    return 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()        passwd = input('请输入密码：').strip()        if not_null(username) and not_null(passwd):#账号和密码都不为空            if username in users:                cur_user = users[username]  # 代表通过账号获取到的小字典                if cur_user['passwd'] == passwd:                    if cur_user['role'] == 1:                        print('欢迎管理员登录')                        admin()                    else:                        print('欢迎%s登录' % username)                        buyer(username)            else:                print('用户不存在！')        else:            print('账号/密码不能为空！')

login()