登录合并购物车

在用户登录时，将cookie中的购物车数据合并到redis中，并清除cookie中的购物车数据。

普通登录和QQ登录都要合并，所以将合并逻辑放到公共函数里实现。

在carts/utils.py中创建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中 :param request: 用户的请求对象 :param user: 当前登录的用户 :param response: 响应对象，用于清楚购物车cookie :return: """ cookie_cart = request.COOKIES.get('cart') if cookie_cart is not None: cookie_cart = pickle.loads(base64.b64decode(cookie_cart.encode())) redis_conn = get_redis_connection('cart') redis_cart = redis_conn.hgetall('cart_%s' % usert.id) redis_cart_selected = redis_conn.smembers('cart_selected_%s' % user.id) cart = {} for sku_id, count in redis_cart.items(): cart[int(sku_id)] = int(count) for sku_id, count_selected_dict in cookie_cart.items(): cart[sku_id] = count_selected_dict['count'] if count_selected_dict['selected']: redis_cart_selected.add(sku_id) if cart: pl = redis_conn.pipeline() pl.hmset('cart_%s' % user.id, cart) pl.sadd('cart_selected_%s' % user.id, *redis_cart_selected) pl.execute() response.delete_cookie('cart') return response 

修改登录视图

rest_framework_jwt提供的obtain_jwt_token视图，实际从rest_framework_jwt.views.ObtainJSONWebToken类视图而来，我们可以重写此类视图里的post方法来添加合并逻辑

from rest_framework_jwt.views import ObtainJSONWebToken

class UserAuthorizeView(ObtainJSONWebToken): """ 用户认证 """ def post(self, request, *args, **kwargs): # 调用父类的方法，获取drf jwt扩展默认的认证用户处理结果 response = super().post(request, *args, **kwargs) # 仿照drf jwt扩展对于用户登录的认证方式，判断用户是否认证登录成功 # 如果用户登录认证成功，则合并购物车 serializer = self.get_serializer(data=request.data) if serializer.is_valid(): user = serializer.validated_data.get('user') response = merge_cart_cookie_to_redis(request, user, response) return response 

修改路径users/urls.py

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

转载于:https://www.cnblogs.com/hzlnice/p/9393139.html