关于REST风格API的设计2

from django.shortcuts import render

# Create your views here.
from django.shortcuts import render
from django_redis import get_redis_connection
from rest_framework import status
from rest_framework.generics import GenericAPIView
from rest_framework.response import Response
import pickle
import base64

from goods.models import SKU
from . import constants
from .serializers import CartSerializer, CartSKUSerializer, CartDeleteSerializer, CartSelectAllSerializer

# Create your views here.

class CartView(GenericAPIView):
"""购物车"""
serializer_class = CartSerializer

# 身份验证重写留白
def perform_authentication(self, request):
"""将执行具体请求方法前的身份认证关掉，由视图自己来进行身份认证"""
pass

def post(self, request):
# sku_id count selected
# 校验
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
# order_dict:validated_data
sku_id = serializer.validated_data['sku_id']
count = serializer.validated_data['count']
selected = serializer.validated_data['selected']

# 判断用户登录状态
'''
用户认证不成功,user方法会直接抛出异常,所以身份认证需要进行错误捕捉
'''
try:
user = request.user # 匿名用户 AnonymoseUser
except Exception:
user = None

#
if user and user.is_authenticated:
# 如果用户已登录，保存到redis
redis_conn = get_redis_connection('cart')
pl = redis_conn.pipeline()

# 用户购物车数据 redis hash哈希

'''
没有就直接添加新的键值对,否则就直接追加数据
'''
pl.hincrby('cart_%s' % user.id, sku_id, count)

# 用户购物车勾选数据 redis set
if selected:
pl.sadd('cart_selected_%s' % user.id, sku_id)

pl.execute()
return Response(serializer.data)

else:
# 如果用户未登录，保存到cookie reponse = Response() response.set_cookie
# 取出cookie中的购物车数据
cart_str = request.COOKIES.get('cart')

if cart_str: # 购物车不为空
# 解析
cart_str = cart_str.encode() # str -> bytes
cart_bytes = base64.b64decode(cart_str) # b64decode(byes类型）
cart_dict = pickle.loads(cart_bytes) # 字典类型
else:
cart_dict = {}

# cart_dict = {
# sku_id_1: {
# 'count': 10
# 'selected': True
# },
# sku_id_2: {
# 'count': 10
# 'selected': False
# },
# sku_id_3: {
# 'count': 10
# 'selected': True
# }
# }

if sku_id in cart_dict:
# 如果商品存在购物车中，累加
cart_dict[sku_id]['count'] += count
cart_dict[sku_id]['selected'] = selected
else:
# 如果商品不在购物车中，设置
cart_dict[sku_id] = {
'count': count,
'selected': selected
}

cart_cookie = base64.b64encode(pickle.dumps(cart_dict)).decode()

# 设置cookie
response = Response(serializer.data)
response.set_cookie('cart', cart_cookie, max_age=constants.CART_COOKIE_EXPIRES)

return response

def get(self, request):
"""查询购物车"""
"""
检测用户是否登录
"""
try:
user = request.user # user方法
except Exception:
user = None

# 查询
if user and user.is_authenticated:
# 如果用户已登录，从redis中查询 sku_id count selected
redis_conn = get_redis_connection('cart')
redis_cart = redis_conn.hgetall('cart_%s' % user.id)
# redis_cart = {
# 商品的sku_id bytes字节类型: 数量 bytes字节类型
# 商品的sku_id bytes字节类型: 数量 bytes字节类型
# ...
# }
redis_cart_selected = redis_conn.smembers('cart_selected_%s' % user.id)
# redis_cart_selected = set(勾选的商品sku_id bytes字节类型, ....)

# 遍历 redis_cart，形成cart_dict
cart_dict = {}
for sku_id, count in redis_cart.items():
cart_dict[int(sku_id)] = {
'count': int(count),
'selected': sku_id in redis_cart_selected
}
else:
# 如果用户未登录，从cookie中查询
cookie_cart = request.COOKIES.get('cart')

if cookie_cart:
# 表示cookie中有购物车数据
# 解析
cart_dict = pickle.loads(base64.b64decode(cookie_cart.encode()))
else:
# 表示cookie中没有购物车数据
cart_dict = {}

# cart_dict = {
# sku_id_1: {
# 'count': 10
# 'selected': True
# },
# sku_id_2: {
# 'count': 10
# 'selected': False
# },
# }

# 查询数据库
sku_id_list = cart_dict.keys()
sku_obj_list = SKU.objects.filter(id__in=sku_id_list)

# 遍历sku_obj_list 向sku对象中添加count和selected属性
for sku in sku_obj_list:
sku.count = cart_dict[sku.id]['count']
sku.selected = cart_dict[sku.id]['selected']

# 序列化返回
serializer = CartSKUSerializer(sku_obj_list, many=True)
return Response(serializer.data)

def put(self, request):
"""修改购物车"""
# sku_id, count, selected
# 校验
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)

sku_id = serializer.validated_data['sku_id']
count = serializer.validated_data['count']
selected = serializer.validated_data['selected']

# 判断用户的登录状态
try:
user = request.user
except Exception:
user = None

# 保存
if user and user.is_authenticated:
# 如果用户已登录，修改redis
redis_conn = get_redis_connection('cart')
pl = redis_conn.pipeline()

# 处理数量 hash
pl.hset('cart_%s' % user.id, sku_id, count)

# 处理勾选状态 set
if selected:
# 表示勾选
pl.sadd('cart_selected_%s' % user.id, sku_id)
else:
# 表示取消勾选, 删除
pl.srem('cart_selected_%s' % user.id, sku_id)

pl.execute()
return Response(serializer.data)

else:
# 未登录，修改cookie
cookie_cart = request.COOKIES.get('cart')

if cookie_cart:
# 表示cookie中有购物车数据
# 解析
cart_dict = pickle.loads(base64.b64decode(cookie_cart.encode()))
else:
# 表示cookie中没有购物车数据
cart_dict = {}
# cart_dict = {
# sku_id_1: {
# 'count': 10
# 'selected': True
# },
# sku_id_2: {
# 'count': 10
# 'selected': False
# },
# }

response = Response(serializer.data)

if sku_id in cart_dict:
cart_dict[sku_id] = {
'count': count,
'selected': selected
}

cart_cookie = base64.b64encode(pickle.dumps(cart_dict)).decode()

# 设置cookie
response.set_cookie('cart', cart_cookie, max_age=constants.CART_COOKIE_EXPIRES)

return response

def delete(self, request):
"""删除购物车数据"""
# 校验数据: sku_id
serializer = CartDeleteSerializer(data=request.data)
serializer.is_valid(raise_exception=True)
sku_id = serializer.validated_data['sku_id']

# 判断用户的登录状态
try:
user = request.user
except Exception:
user = None

# 删除
if user and user.is_authenticated:
# 已登录，删除redis
redis_conn = get_redis_connection('cart')
pl = redis_conn.pipeline()
# 删除hash
pl.hdel('cart_%s' % user.id, sku_id)
# 删除set
pl.srem('cart_selected_%s' % user.id, sku_id)
pl.execute()
return Response(status=status.HTTP_204_NO_CONTENT)
else:
# 未登录，删除cookie
cookie_cart = request.COOKIES.get('cart')

cart_cookie = base64.b64encode(pickle.dumps(cart_dict)).decode()

# 设置cookie
response.set_cookie('cart', cart_cookie, max_age=constants.CART_COOKIE_EXPIRES)

return response

class CartSelectAllView(GenericAPIView):
"""
购物车全选
"""
serializer_class = CartSelectAllSerializer

def perform_authentication(self, request):
pass

def put(self, request):
# selected
# 校验
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
selected = serializer.validated_data['selected']

# 判断用户的登录状态
try:
user = request.user
except Exception:
user = None

if user and user.is_authenticated:
# 已登录，redis
redis_conn = get_redis_connection('cart')
redis_cart = redis_conn.hgetall('cart_%s' % user.id)
# redis_cart = {
# 商品的sku_id bytes字节类型: 数量 bytes字节类型
# 商品的sku_id bytes字节类型: 数量 bytes字节类型
# ...
# }

sku_id_list = redis_cart.keys()
if selected:
# 全选，所有的sku_id都添加到redis set
redis_conn.sadd('cart_selected_%s' % user.id, *sku_id_list)
else:
# 取消全选，清空redis中的set数据
redis_conn.srem('cart_selected_%s' % user.id, *sku_id_list)

return Response({'message': 'OK'})
else:
# 未登录， cookie
cookie_cart = request.COOKIES.get('cart')

response = Response({'message': 'OK'})

if cart_dict:
for count_selected_dict in cart_dict.values():
count_selected_dict['selected'] = selected

cart_cookie = base64.b64encode(pickle.dumps(cart_dict)).decode()

# 设置cookie
response.set_cookie('cart', cart_cookie, max_age=constants.CART_COOKIE_EXPIRES)

return response

关于REST风格API的设计2

猜你喜欢