Django Serializer HiddenField隱藏字段實例

更新時間：2020年03月31日 08:33:53 作者：我就是全世界

這篇文章主要介紹了Django Serializer HiddenField隱藏字段實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

Django rest_framework serializer.HiddenField(default=serializer.CurrentUser())

在用Dajngo RestFramework時，有時候需要這么一個場景，前端不需要傳一個或多個字段，這些字是直接根據(jù)用戶登錄信息判斷自動賦值的，如果用mixin和viewset進行搭配寫接口，要么重寫create, update等方法，要么就是在serializer_class時就定義默認值，而第二種方法明顯簡單一些。

具體看代碼

# 假如我定義了一個serliazer，shop和status都是自動判斷后賦值的
# 用HiddenField定義字段，default方法來寫具體賦值函數(shù)
# default方法是一個類，需要重寫set_context的方法
from rest_framework.compat import (
 InvalidTimeError, MaxLengthValidator, MaxValueValidator,
 MinLengthValidator, MinValueValidator, unicode_repr, unicode_to_repr
)
class CurrentShopDefault(object):
 def set_context(self, serializer_field):
  self.shop = serializer_field.context['request'].user.shop

 def __call__(self):
  return self.shop

 def __repr__(self):
  return unicode_to_repr('%s()' % self.__class__.__name__)

class CurrentStatus(object):
 def set_context(self, serializer_field):
  user_info = serializer_field.context['request'].user
  if user_info.shop.shop_type == 'SHOP_TYPE_COMCIRCLE':
   self.status = 1
  else:
   self.status = 2

 def __call__(self):
  return self.status

 def __repr__(self):
  return unicode_to_repr('%s()' % self.__class__.__name__)

# 定義serializer
class GameSerializer(serializers.ModelSerializer):
 class Meta:
  model = Game
  exclude = ['created_time']

 shop = serializers.HiddenField(
  default=CurrentShopDefault()
 )
 status = serializers.HiddenField(
  default=CurrentStatus()
 )

最后在view文件的class里直接指定serializer_class就好了

class GameOpView(mixins.CreateModelMixin, mixins.UpdateModelMixin, mixins.RetrieveModelMixin, mixins.DestroyModelMixin, viewsets.GenericViewSet):
 """
  retrieve:
   獲取-游戲詳情
  create:
   創(chuàng)建-游戲
  update:
   更新-游戲
  delete:
   刪除-游戲
  partial_update:
   更新游戲狀態(tài)
  total_info:
   游戲數(shù)據(jù)總覽
 """
 permission_classes = [IsShop, ]
 pagination_class = StandardResultsSetPagination

 def get_queryset(self):
  return Game.objects.filter(
   Q(shop=self.request.user.shop) | Q(shop__com_circle=self.request.user.shop))

 def get_serializer_class(self):
  if self.action == 'partial_update':
   return GameStatusSerializer
  elif self.action == 'retrieve':
   return GameReadSerializer
  else:
   return GameSerializer

補充知識：drf 中serializers.HiddenField使用方法

class CurrentUser(object):
 def set_context(self, serializer_field):
  self.user_obj = serializer_field.context['request'].user

 def __call__(self):
  return self.user_obj

class GetObjectFlow(object):
 def set_context(self, serializer_field):
  self.object_flow = serializer_field.context['request'].data.get('object_flow') # 這里需要在views中傳進去呀，djangocontext上下文中才能取到

 def __call__(self):
  return self.object_flow


class AddHolidaySerializer(serializers.ModelSerializer, BaseModelSerializer):
 user = serializers.HiddenField(default=CurrentUser()) # 給用戶外鍵賦當前請求的user
 object_flow = serializers.HiddenField(default=GetObjectFlow()) # 后端自己處理一些特殊的外鍵
 status = serializers.HiddenField(default=0) # 默認新增時給status賦初值

 class Meta:
  model = Holiday
  exclude = ('deleted',)

以上這篇Django Serializer HiddenField隱藏字段實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: