之前有一篇博文写到关于购物车的业务逻辑,分别运用cookie和redis存储未登录和登录用户的购物车数据,虽然已经很好的完成了业务逻辑,但是会发现代码的冗余很严重,也不够具有python特色,今天就让我来用python特色,高逼格的来改写一下购物车逻辑把。

回看之前代码,我们会发现,会经常取出购物车数据,然后也会修改数据,我们能否对读和写的操作进行封装,第二,cookie中和redis中存储的我们对数据存储的格式不一样,能否考虑将两者存储为一种格式,第三,考虑数据通过网络传输,我们能否尽量存储少的数据。

之前我们在redis中是将商品数量和商品勾选状态分别存为hash和set,而在cookie中则是存成一个大字典,然后将大字典转化为字符串,在redis中,我们也可以采用相同的存储方式,而相对于字典中嵌套字典,我们的小字典中只有两组数据,我们可以用一个列表直接存储商品数量和勾选状态,记住对应的下标即可,这样我们就不用花空间去存储冗余的key值。确定好存储方式,我们需要想一下怎么去进行封装。在python是面向对象语言,同时具有多继承的特色,所以我们可以用拓展类的方式去封装读写操作,在对应的视图类中采取多继承的方式,引入读写的方法。

class CartMixin(object):
    """购物车扩展类"""
    str_to_dict_class = (str.encode, base64.b64decode, pickle.loads)
    dict_to_dict_class = (pickle.dumps, base64.b64encode, bytes.decode)

    def conversion_and_encryption(self, key):
     # 将数据转化为需要格式的方法 if isinstance(key, dict): for func in self.dict_to_dict_class: key = func(key) else: for func in self.str_to_dict_class: key = func(key) return key def get_cart_dict(self, request):
     # 获取购物车的方法 try: user = request.user except Exception: user = None if user and user.is_authenticated: return self.get_cart_from_redis(request) else: return self.get_cart_from_cookie(request) def write_cart_dict(self, cart_dict, request, response):
     # 写入购物车数据的方法 try: user = request.user except Exception: user = None if user and user.is_authenticated: self.write_cart_to_redis(cart_dict, user) return response else: response = self.write_cart_to_cookie(cart_dict, response) return response def get_cart_from_redis(self,request):
     # 从redis中获取购物车 redis_conn = get_redis_connection('cart') redis_cart = redis_conn.get('cart:%s'%request.user.id) if not redis_cart: return {} # redis_cart = pickle.loads(base64.b64decode(redis_cart)) cart_dict = self.conversion_and_encryption(redis_cart.decode()) return cart_dict def get_cart_from_cookie(self,request):
     # 从cookie中获取购物车的方法 cart_dict = request.COOKIES.get('cart') if cart_dict: # cart_dict = pickle.loads(base64.b64decode(cart_dict.encode())) cart_dict = self.conversion_and_encryption(cart_dict) else: cart_dict = {} return cart_dict def write_cart_to_redis(self, cart_dict, user):
     # 写入购物车数据到redis的方法 redis_conn = get_redis_connection('cart') # cart_dict = base64.b64encode(pickle.dumps(cart_dict)).decode() cart_dict = self.conversion_and_encryption(cart_dict) redis_conn.set('cart:%s'%user.id, cart_dict) def write_cart_to_cookie(self, cart_dict, response):
     # 写入购物车数据到cookie的方法 # cart_cookie = base64.b64encode(pickle.dumps(cart_dict)).decode() cart_cookie = self.conversion_and_encryption(cart_dict) response.set_cookie('cart', cart_cookie, max_age=constants.CART_COOKIE_EXPIRES) return response

为了保证数据的可靠性和安全性,我们会对数据进行一些处理在进行存储,考虑到有多个地方可能用到数据处理的操作,我们将数据处理的操作也封装成对应的方法,同时,我们也可能用不同的方式处理数据,所以将处理数据的方法放到一个列表中,用类属性方法进行存储,需要的更改的时候就只需要更改对应的类属性即可。这样我们就完成了读写操作的封装,在需要进行读写操作的地方调用对应的方法即可。这样能减少代码的冗余。

而在对应的业务逻辑中,我们应该只需要考虑业务方面的需要,无需考虑用户的登录状态,同时,购物车数据的读取和写入也是与我们的业务逻辑无关。关于登录状态的判断,已经封装到扩展类当中了,而对于读写,我们可以不必在视图的业务逻辑中进行操作,在drf框架中在请求进来之前会进行一个初始化操作,而在响应返回,传到前端之前,也会进行最后的处理,我们可以在视图类中,重写这两个方法,继承父类的操作,在加上我们需要进行的读写操作即可,这样就实现了读写操作和业务逻辑的分离。

 

class CartView(CartMixin, APIView):
    '''购物车'''
    permission_classes = [IsAuthenticated]

    def perform_authentication(self, request):
        """
        重写父类的用户验证方法,不在进入视图前就检查JWT
        """
        pass

    def initial(self, request, *args, **kwargs):
        super().initial(request, *args, **kwargs)
        self.cart_dict = self.get_cart_dict(request)

    def finalize_response(self, request, response, *args, **kwargs):
        response = super().finalize_response(request, response, *args, **kwargs)
        self.response = self.write_cart_dict(self.cart_dict, request, response)
        return self.response

 

至此,代码改写完成。

 

 

新人写博客锻炼自己,想一起交流的可以加我的qq595395786