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

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

具體看代碼

# 假如我定義了一個(gè)serliazer，shop和status都是自動(dòng)判斷后賦值的# 用HiddenField定義字段，default方法來(lái)寫(xiě)具體賦值函數(shù)# default方法是一個(gè)類(lèi)，需要重寫(xiě)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__)# 定義serializerclass 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

補(bǔ)充知識(shí)：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_objclass GetObjectFlow(object): def set_context(self, serializer_field): self.object_flow = serializer_field.context[’request’].data.get(’object_flow’) # 這里需要在views中傳進(jìn)去呀，djangocontext上下文中才能取到 def __call__(self): return self.object_flowclass AddHolidaySerializer(serializers.ModelSerializer, BaseModelSerializer): user = serializers.HiddenField(default=CurrentUser()) # 給用戶(hù)外鍵賦當(dāng)前請(qǐng)求的user object_flow = serializers.HiddenField(default=GetObjectFlow()) # 后端自己處理一些特殊的外鍵 status = serializers.HiddenField(default=0) # 默認(rèn)新增時(shí)給status賦初值 class Meta: model = Holiday exclude = (’deleted’,)

以上這篇Django Serializer HiddenField隱藏字段實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持好吧啦網(wǎng)。