二十九、登陆时购物车合并（商品部分）

1、重写登录功能的业务逻辑

• 在users/下添加merge_cart_cookie_to_redis合并购物车的方法

import pickle
import base64
from django_redis import get_redis_connection


def merge_cart_cookie_to_redis(request, user, response):
    """
    合并请求用户的购物车数据，将未登录保存在cookie里的保存到redis中
    遇到cookie与redis中出现相同的商品时以cookie数据为主，覆盖redis中的数据
    :param request: 用户的请求对象
    :param user: 当前登录的用户
    :param response: 响应对象，用于清楚购物车cookie
    :return:
    """

    # 1、获取cookie
    cart_cookie = request.COOKIES.get('cart_cookie', None)
    # 2、判断cookie是否存在
    if cart_cookie is None:
        return response
    # 3、存在则解密 cart = {10: {‘count’:2, selected: True}。cart = {}
    cart = pickle.loads(base64.b64decode(cart_cookie))
    # 4、判断字典是否为空
    if not cart:
        return response
    # 5、不为空拆分数据
    # 6、哈希 对应字典 {sku_id: count,sku_id2:count2}
    cart_dict = {}
    # 7、集合 对应列表 选中[sku_id1, sku_id2]。未选中[sku_id3]
    sku_id = []  # 选中
    sku_id_none = []  # 未选中
    for skuid, data in cart.items():
        # 哈希
        cart_dict[skuid] = data['count']
        if data['selected']:
            sku_id.append(skuid)
        else:
            sku_id_none.append(skuid)
    # 8、写入redis
    conn = get_redis_connection('carts')
    conn.hmset('cart_%s' % user.id, cart_dict)
    if sku_id:
        conn.sadd('cart_selected_%s' % user.id, *sku_id)
    if sku_id_none:
        conn.srem('cart_selected_%s' % user.id, *sku_id_none)
    # 9、删除cookie
    response.delete_cookie('cart_cookie')

    return response

• 在views下重写用户登录的视图函数

class UserLoginView(ObtainJSONWebToken):

    def post(self, request, *args, **kwargs):
        response=super().post(request, *args, **kwargs)

        serializer = self.get_serializer(data=request.data)

        if serializer.is_valid():
            user = serializer.object.get('user') or request.user
            # 合并购物车
            response=merge_cart_cookie_to_redis(request,user,response)

        return response

2、修改登陆子路由

urlpatterns = [
    ...
    # url(r'^authorizations/$', obtain_jwt_token), 
    url(r'^authorizations/$', views.UserAuthorizeView.as_view()),
    ...
]