如何以python风格高逼格的改成购物车逻辑
之前有一篇博文写到关于购物车的业务逻辑，分别运用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