# _*_coding:utf-8_*_
from rest_framework.views import APIView
from rest_framework.response import Response
from utils.base_response import BaseResponse
from utils.redis_pool import POOL
from django.utils.timezone import now
from utils.my_auth import LoginAuth
import redis
from .views import SHOPPINGCAR_KEY
from .models import CouponRecord
import json

CONN = redis.Redis(connection_pool=POOL)
SETTLEMENT_KEY = 'SETTLEMENT_%s_%s'
GLOBAL_COUPON_KEY = 'GLOBAL_COUPON_%s'


class SettlementView(APIView):
    authentication_classes = [LoginAuth, ]

    def post(self, request):
        res = BaseResponse()
        # 1 ,获取前端的数据以及user_id
        course_list = request.data.get('course_list', "")
        user_id = request.user.pk
        # 2,校验数据的合法性
        for course_id in course_list:
            # 2.1  判断 course_id 是否在购物车里
            shopping_car_key = SHOPPINGCAR_KEY % (user_id, course_id)
            if not CONN.exists(shopping_car_key):
                res.code = 1050
                res.error = '课程ID不合法'
                return Response(res.dict)

            # 3，构建数据结构
            # 3.1 获取用户的所有合法优惠券
            user_all_coupons = CouponRecord.objects.filter(
                account_id=user_id,
                status=0,
                coupon__valid_begin_date__lte=now(),
                coupon__valid_end_date__gte=now(),
            ).all()
            # 3.2  构建优惠券 dict
            course_coupon_dict = {}
            global_coupon_dict = {}
            for coupon_record in user_all_coupons:
                coupon = coupon_record.coupon
                if coupon.object_id == course_id:
                    course_coupon_dict[coupon.id] = {
                        'id': coupon.id,
                        'name': coupon.name,
                        'coupon_type': coupon.get_coupon_type_display()(),
                        'object_id': coupon.object_id,
                        'money_equivalent_value': coupon.money_equivalent_value,
                        'off_percent': coupon.off_percent,
                        'minunum_consume': coupon.minimum_consume
                    }
                elif coupon.object_id == '':
                    global_coupon_dict[coupon.id] = {
                        'id': coupon.id,
                        'name': coupon.name,
                        'coupon_type': coupon.get_coupon_type_display()(),
                        'money_equivalent_value': coupon.money_equivalent_value,
                        'off_percent': coupon.off_percent,
                        'minunum_consume': coupon.minimum_consume
                    }
            # 3.3 构建写入redis的数据结构
            course_info = CONN.hgetall(shopping_car_key)
            price_policy_dict = json.loads(course_info['price_policy_dict'])
            default_policy_id = course_info['default_price_policy_id']
            valid_period = price_policy_dict[default_policy_id]['valid_period_display']
            price = price_policy_dict[default_policy_id]['price']
            settlement_info = {
                'id': course_info['id'],
                'title': course_info['title'],
                'course_img': course_info['course_img'],
                'valid_period': valid_period,
                'price': price,
                'course_coupon_dict': json.dumps(course_coupon_dict, ensure_ascii=False)
            }
            # 4，写入redis中
            settlement_key = SETTLEMENT_KEY % (user_id, course_id)
            global_coupon_key = GLOBAL_COUPON_KEY % user_id
            CONN.hmset(settlement_key, settlement_info)
            if global_coupon_dict:
                CONN.hmset(global_coupon_key, global_coupon_dict)
            # 5，删除购物车中的数据
            CONN.delete(shopping_car_key)
        res.data = '加入结算中心成功'
        return Response(res.dict)

    def get(self, request):
        res = BaseResponse()
        # 1,获取user_id
        user_id = request.user.pk
        # 2,拼接所有的key
        # 3,去redis取数据
        settlement_key = SETTLEMENT_KEY % (user_id, '*')
        global_coupon_key = GLOBAL_COUPON_KEY % user_id
        all_keys = CONN.scan_iter(settlement_key)
        ret = []
        for key in all_keys:
            ret.append(CONN.hgetall(key))
        global_coupon_info = CONN.hgetall(global_coupon_key)
        res.data = {
            'settlement_info': ret,
            'global_coupon_dict': global_coupon_info
        }
        return Response(res.dict)

    def put(self, request):
        # course_id  course_coupon_id  global_coupon_id
        res = BaseResponse()
        # 1,获取前端传过来的数据
        course_id = request.data.get('course_id', '')
        course_coupon_id = request.data.get('course_coupon_id', "")
        global_coupon_id = request.data.get('global_coupon_id', "")
        user_id = request.user.pk
        # 2，校验数据合法性
        # 2.1 校验 course_id
        key = SETTLEMENT_KEY % (user_id, course_id)
        if course_id:
            if not CONN.exists(key):
                res.code = 1060
                res.error = '课程ID不合法'
                return Response(res.dict)
        # 2.2 校验 course_coupon_id
        if course_coupon_id:
            course_coupon_dict = json.loads(CONN.hget(key, 'course_coupon_dict'))
            if course_coupon_id not in course_coupon_dict:
                res.code = 1061
                res.error = '课程优惠券ID不合法'
                return Response(res.dict)
        # 2.3 校验global_coupon_id
        if global_coupon_id:
            global_coupon_key = GLOBAL_COUPON_KEY % user_id
            if not CONN.exists(global_coupon_key):
                res.code = 1062
                res.error = '全局优惠券ID不合法'
                return Response(res.dict)
        # 3，修改redis中数据
        CONN.hset(key, 'default_coupon_id', course_coupon_id)
        res.data = '更新成功'
        return Response(res.dict)
