如何覆盖批量添加的 POST 方法 - Django Rest Framework [英] How to override POST method for bulk adding - Django Rest Framework

问题描述

我正在使用 Django Rest Framework 编写 REST API，我希望我的其中一个路由接受批量添加 POST 方法，以创建多个对象.其他方法(GET、PUT、PATCH、DELETE)将保持不变，一次只接受一个.

I'm writing a REST API using Django Rest Framework and I would like that one of my routes accepts bulk adding on POST method, to create multiple objects. Others methods (GET, PUT, PATCH, DELETE) will still remain the same, accepting only one at a time.

到目前为止，我所拥有的内容如下，目前可以一次发布一个.

What I have so far is below and it's currently working just fine for posting one at a time.

在我的 urls.py 中:

In my urls.py:

path('book', books.BookViewSet.as_view()),

books.py:

class BookViewSet(viewsets.ModelViewSet):
    serializer_class = BookSerializer
    queryset = Book.objects.all()
    permission_classes = (IsAuthenticated, )

serializer.py:

serializer.py:

class BookSerializer(serializers.ModelSerializer):

    def create(self, validated_data):
        # I assume this is the method to be overridden to get this

    class Meta:
        model = Book
        fields = ('id', 'name', 'author_id', 'page_number', 'active')

推荐答案

Serializer create 方法，不幸的是按对象创建数据对象.您可以覆盖 ModelViewSet 的 create 方法，验证后使用 bulk_create 方法.

Serializer create method, unfortunatelly creates data object by object.You can override create method of ModelViewSet and after validation use bulk_create method.

def create(self, request, *args, **kwargs):
    many = True if isinstance(request.data, list) else False
    serializer = BookSerializer(data=request.data, many=many)
    serializer.is_valid(raise_exception=True)
    author = request.user # you can change here
    book_list = [Book(**data, author=author) for data in serializer.validated_data]
    Book.objects.bulk_create(book_list)
    return Response({}, status=status.HTTP_201_CREATED)

这篇关于如何覆盖批量添加的 POST 方法 - Django Rest Framework的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！