from django.db import transaction
class OrderCommitView(View):
    '''订单创建'''

    @transaction.atomic  ##  使用事务
    def post(self,request):
        '''订单创建'''
        ##  判断用户是否登录
        user = request.user
        if not user:
            return JsonResponse({'res':0,'errmsg':'用户未登录'})

        ##  1、接收数据
        pay_method = request.POST.get('pay_method')
        addr_id = request.POST.get('addr_id')
        skus_id = request.POST.get('skus_id')
        ##  将skus_id的格式转化成列表形式
        skus_id = skus_id.split(',')


        ##  2、校验数据
        if not all([pay_method,addr_id,skus_id]):
            return JsonResponse({'res':1,'errmsg':'数据不完整'})
        skus = []
        for sku_id in skus_id:
            ##  遍历skus_id，从数据库中查询sku
            try:
                sku = GoodsSKU.objects.get(id=sku_id)
                skus.append(sku)
            except GoodsSKU.DoesNoExist:
                return JsonResponse({'res':2,'errmsg':'商品不存在'})
        try:
            addr = Address.objects.get(id=addr_id)
        except Address.DoesNoExist:
            return JsonResponse({'res':3,'errmsg':'地址不存在'})
        ##  校验支付方式
        if str(pay_method) not in OrderInfo.PAY_METHODS.keys():
            return JsonResponse({'res':4,'errmsg':'错误的支付方式'})
        ##  设置事务的节点
        save_id = transaction.savepoint()

        ##  3、业务处理
        ##########  （1）先向订单数据库中添加一条数据,准备数据：order_id、user、addr、pay_method、total_count、total_price、transit_price
        ##  计算total_count和total_price，先初始化为0，后面查询各个商品的数量和价格之后再进行累加，以及重新保存
        try:
            total_count = 0
            total_price = 0
            ##  生成order_id：格式：当前的时间+sku_id
            order_id = datetime.now().strftime('%Y%m%d%H%M%S') + str(user.id)
            ##  添加数据，创建新的订单数据
            order = OrderInfo.objects.create(order_id=order_id,
                                     user=user,
                                     addr=addr,
                                     pay_method=pay_method,
                                     total_count=total_count,
                                     total_price=total_price,
                                     transit_price=10)
    
            ##########  （2）向商品订单中添加多条数据，准备数据：order、sku、count、price
            conn = get_redis_connection('default')
            cart_key = 'cart_%d'%user.id
            for sku in skus:
                count = conn.hget(cart_key,sku.id)
                price = sku.price
                OrderGoods.objects.create(order=order,
                                          sku=sku,
                                          count=count,
                                          price=price)
                if sku.stock < int(count):
                    ##  业务失败时，回滚事务
                    transaction.rollback(save_id)
                    return JsonResponse({'res':6,'errmsg':'库存不足'})
                ##  MySQL数据库的库存减少，销量增加
                sku.stock -= int(count)
                sku.sales += int(count)
                sku.save()
                ##  删除Redis数据库中添加到订单中的数据
                conn.hdel(cart_key,sku.id)
                ##  累加total_count 和total_price
                total_count += int(count)
                total_price += int(count)*int(price)
            ##  重新更新订单中的total_count和total_price
            order.total_count = total_count
            order.total_price = total_price
            order.save()
        
        except Exception:
            ##  业务失败时，回滚事务
            transaction.rollback(save_id)
            return JsonResponse({'res':7,'errmsg':'订单创建失败'})

        ##  4、返回应答
        return JsonResponse({'res':5,'message':'创建订单成功'})