Django REST Framework serializer에 사용자 특정 필드를 추가하십시오.

Question

현재 요청을 작성하는 사용자에 대한 정보가 들어있는 serializer에 필드를 추가하고 싶습니다.이 필드는 별도의 끝점을 만들지는 않습니다. 여기에 내가 길을했던된다

뷰셋 :

class ArticleViewSet(viewsets.ModelViewSet): 
    queryset = Article.objects.all() 
    serializer_class = ArticleSerializer 
    filter_class = ArticleFilterSet 

    def prefetch_likes(self, ids): 
     self.current_user_likes = dict([(like.article_id, like.pk) for like in Like.objects.filter(user=self.request.user, article_id__in=ids)]) 

    def get_object(self, queryset=None): 
     article = super(ArticleViewSet, self).get_object(queryset) 
     self.prefetch_likes([article.pk]) 
     return article 

    def paginate_queryset(self, queryset, page_size=None): 
     page = super(ArticleViewSet, self).paginate_queryset(queryset, page_size) 
     if page is None: 
      return None 

     ids = [article.pk for article in page.object_list] 
     self.prefetch_likes(ids) 

     return page 

시리얼 라이저 :

class ArticleSerializer(serializers.ModelSerializer): 
    class Meta: 
     model = Article 

    def to_native(self, obj): 
     ret = super(ArticleSerializer, self).to_native(obj) 

     if obj: 
      view = self.context['view'] 
      ret['has_liked'] = False 
      if hasattr(view, 'current_user_liked'): 
       ret['has_liked'] = obj.pk in view.current_user_liked 

     return ret 

좋아하는 기사의 프리 페치를 주입 할 수있는 더 좋은 곳이 있는가 또는 일반적으로이 작업을 수행하는 더 좋은 방법입니까?

Answer 1

나는 Like 모델 개체에 가능한 한 많은 것을 넣고 나머지는 사용자 지정 serializer 필드에 넣으려고합니다.

는 시리얼 필드에서 당신은 그들의 부모 시리얼에서을 상속 context 매개 변수를 통해 request에 액세스 할 수 있습니다.

그래서 당신은 같은 것을 할 수 있습니다

class LikedByUserField(Field): 
    def to_native(self, article): 
     request = self.context.get('request', None) 
     return Like.user_likes_article(request.user, article) 

user_likes_article 클래스 방법은 다음 프리 페치 (캐싱) 로직을 캡슐화 할 수있다.

도움이 되었기를 바랍니다.

Answer 2

당신이 SerializerMethodField

예와 함께 할 수 있습니다

class PostSerializer(serializers.ModelSerializer): 
    fav = serializers.SerializerMethodField('likedByUser') 

    def likedByUser(self, obj): 
     request = self.context.get('request', None) 
     if request is not None: 
      try: 
       liked=Favorite.objects.filter(user=request.user, post=obj.id).count() 
       return liked == 1 
      except Favorite.DoesNotExist: 
       return False 
     return "error" 

    class Meta: 
     model = Post 

는 다음과 같이보기에서 시리얼을 호출해야합니다 :

class PostView(APIVIEW): 
    def get(self,request): 
     serializers = PostSerializer(PostObjects,context={'request':request}) 

Answer 3

을 Django Documentation - SerializerMethodField에 따르면, 나는에 있었다 rapid2share의 코드를 약간 변경하십시오.