1.4

# -*- coding： utf-8 -*-
"""
扩展需求：
1.用户下一次登录后 输入用户名密码 直接回到上次得状态 即上次消费得余额 物品 再次登录可继续购买
2.允许查询之前得消费记录
"""
import os

goods = [
    {"name": "电脑", "price": 1999},
    {"name": "鼠标", "price": 10},
    {"name": "游艇", "price": 20},
    {"name": "美女", "price": 998}
]


def show_goods():
    # 展示商品列表
    print('展示商品列表'.center(20,'-'))
    for index, i in enumerate(goods):
        print('%d. %s   %d' % (index, i['name'], i['price']))


def judge_is_buy():
    # 判断用户曾经是否购买过商品
    if os.path.exists('last_goods_new.txt'):
        last_history = []
        f = open('last_goods_new.txt', 'r', encoding='utf-8')
        for line in f:
            last_history = eval(line)
        print('这是您第33[1;31m%d33[0m次登录，您上次消费余额33[1;31m%s33[0m' % (last_history[0]+1, last_history[1]))
        for index, i in enumerate(last_history[2], 1):
            print('%d. %s  %s' % (index, i['name'], i['price']))
        return last_history[0], last_history[1]
    else:
        return False


def save_to_file(salary, buy_goods, buy_count):
    # 本次购买商品的记录存到文件中
    f = open('last_goods_new.txt', 'a', encoding='utf-8')
    f.write('%d,%s,%s
' % (buy_count, salary, buy_goods))
    f.close()


def show_buy_goods(salary, buy_goods, buy_count):
    # 展示本次购买的商品
    print('本次：您够买了33[1;31m%s33[0m件商品,余额为33[1;31m%s33[0m' % (len(buy_goods),salary))
    for index, i in enumerate(buy_goods, 1):
        print('%d. %s  %d' % (index, i['name'], i['price']))
    save_to_file(salary, buy_goods, buy_count)


def search_history(history_num):
    # 查询购买记录
    f = open('last_goods_new.txt', 'r', encoding='utf-8')
    for line in f:
        if history_num == eval(line)[0]:
            print(eval(line))
            print('您第%s次购买了：'%history_num)
            for line in eval(line)[2]:
                print('%s'%line['name'])
            break


def buy_goods_now(salary, buy_count):
    # 当前时刻 购买商品
    show_goods()
    buy_goods = []
    while True:
        choice = input('请选择购买商品的编号(输入q表示退出,输入x表示查询历史购买情况)：').strip()
        if choice.isdigit():
            choice = int(choice)
            if choice < len(goods):
                if salary >= goods[choice]['price']:
                    buy_goods.append(goods[choice])
                    salary -= goods[choice]['price']
                    print('商品33[1;35;46m%s33[0m已加入购物车' % goods[choice]['name'])
                    continue
                else:
                    print('您的余额不足，请重新选择商品')
                    continue
            else:
                print('编号不在商品范围内，请重新输入')
                continue
        elif choice == 'q':
            show_buy_goods(salary, buy_goods, buy_count)
            exit()
        elif choice == 'x':
            if os.path.exists('last_goods_new.txt'):
                history_num = input('请输入历史纪录编号：').strip()
                if history_num.isdigit():
                    history_num = int(history_num)
                    search_history(history_num)
                else:
                    print('请重新选择')

        else:
            print('请重新输入')
            continue


def login(username,password):
    # 登录认证
    _username = "alice"
    _password = "123"
    if username == _username and password == _password:
        print('登录成功'.center(20, '-'))
        data = judge_is_buy()
        if data:
            buy_count = data[0]+1
            salary = data[1]
            buy_goods_now(salary, buy_count)
        else:  # 用户第一次登录
            while True:
                salary = input('请输入工资：').strip()
                if salary.isdigit():
                    salary = int(salary)
                elif len(salary.split('.')) == 2 and salary.split('.')[0].isdigit() and salary.split('.')[1].isdigit():
                    salary = float(salary)
                else:
                    print('请重新输入工资：')
                    continue
                buy_goods_now(salary, 1)
    else:
        return False


def main():
    # 主函数
    while True:
        username = input('请输入用户名(输入q表示退出)：').strip()
        if username == 'q':
            break
        password = input('请输入密码(输入q表示退出)：').strip()
        if password == 'q':
            break
        if not login(username, password): continue


if __name__ == '__main__':
    main()