如何在Django Rest Framework中使用perform_create自动设置字段?

时间:2016-02-09 03:32:36

标签: python django django-rest-framework

我正在尝试为我的应用程序创建REST Api。

应用程序目录中的

urls.py 如下所示:

urlpatterns = [
    url(
        r'^professors/(?P<pk>[0-9]+)/reviews/$',
        views.ProfessorReviewList.as_view(),
        name = 'user-review-list',
    )
]

serializers.py

class ProfessorSerializer(serializers.HyperlinkedModelSerializer):
    reviews = serializers.HyperlinkedIdentityField(view_name='professor-review-list')
    class Meta:
        model = Professor
        fields = (
            'url', 'name', 'name_code', 'university',
            'department', 'total_rating_points',
            'number_of_reviews', 'rating', 'reviews',
        )

class ReviewSerializer(serializers.HyperlinkedModelSerializer):
    author = serializers.ReadOnlyField(source='author.username')

    class Meta:
        model = Review
        fields = (
            'url', 'author', 'professor',
            'created', 'updated', 'rating', 'text'
        )

最后,我的 views.py 如下所示:

class ProfessorReviewList(generics.ListCreateAPIView):
    queryset = Review.objects.all()
    serializer_class = ReviewSerializer
    def get_queryset(self):
        queryset = super(ProfessorReviewList, self).get_queryset()
        return queryset.filter(professor__pk=self.kwargs.get('pk'))

    permission_classes = (permissions.IsAuthenticatedOrReadOnly,
                          IsAuthorOrReadOnly,)

    def perform_create(self, serializer):
        # The request user is set as author automatically.
        serializer.save(author=self.request.user)

我要做的是在访问网址时获取一位特定教授的所有评论列表。如果他们愿意,我还想允许用户在此终点添加新评论。一切都很好,但我想要一个系统,用户在创建评论时不必选择教授。

假设我去 / professorors / 1 / reviews / ,获得了教授pk = 1的所有评论的列表。现在,如果用户添加了新评论,教授将自动设置为教授,pk = 1。用户无法选择。在 / professorors / 2 / reviews / ,审查的教授字段自动设置为pk = 2的教授,依此类推。

我该怎么做?谢谢你的帮助。

1 个答案:

答案 0 :(得分:6)

您应该像对作者一样添加教授:

serializer.save(
    author=self.request.user,
    professor_id=self.kwargs.get('pk'),
)

请注意,为了避免从DB中拉出教授,我明确地设置了FK。