mirror of
https://github.com/encode/django-rest-framework.git
synced 2025-07-16 11:12:21 +03:00
Add BulkCreateModelMixin to allow bulk creation of models.
This commit is contained in:
parent
38a74b42da
commit
717bab5a90
|
@ -190,6 +190,17 @@ class CreateAPIView(mixins.CreateModelMixin,
|
|||
"""
|
||||
Concrete view for creating a model instance.
|
||||
"""
|
||||
|
||||
def post(self, request, *args, **kwargs):
|
||||
return self.create(request, *args, **kwargs)
|
||||
|
||||
|
||||
class BulkCreateAPIView(mixins.BulkCreateModelMixin,
|
||||
GenericAPIView):
|
||||
"""
|
||||
Concrete view for creating a model instance.
|
||||
"""
|
||||
|
||||
def post(self, request, *args, **kwargs):
|
||||
return self.create(request, *args, **kwargs)
|
||||
|
||||
|
@ -199,6 +210,7 @@ class ListAPIView(mixins.ListModelMixin,
|
|||
"""
|
||||
Concrete view for listing a queryset.
|
||||
"""
|
||||
|
||||
def get(self, request, *args, **kwargs):
|
||||
return self.list(request, *args, **kwargs)
|
||||
|
||||
|
@ -208,6 +220,7 @@ class RetrieveAPIView(mixins.RetrieveModelMixin,
|
|||
"""
|
||||
Concrete view for retrieving a model instance.
|
||||
"""
|
||||
|
||||
def get(self, request, *args, **kwargs):
|
||||
return self.retrieve(request, *args, **kwargs)
|
||||
|
||||
|
@ -217,6 +230,7 @@ class DestroyAPIView(mixins.DestroyModelMixin,
|
|||
"""
|
||||
Concrete view for deleting a model instance.
|
||||
"""
|
||||
|
||||
def delete(self, request, *args, **kwargs):
|
||||
return self.destroy(request, *args, **kwargs)
|
||||
|
||||
|
@ -226,6 +240,7 @@ class UpdateAPIView(mixins.UpdateModelMixin,
|
|||
"""
|
||||
Concrete view for updating a model instance.
|
||||
"""
|
||||
|
||||
def put(self, request, *args, **kwargs):
|
||||
return self.update(request, *args, **kwargs)
|
||||
|
||||
|
@ -239,6 +254,7 @@ class ListCreateAPIView(mixins.ListModelMixin,
|
|||
"""
|
||||
Concrete view for listing a queryset or creating a model instance.
|
||||
"""
|
||||
|
||||
def get(self, request, *args, **kwargs):
|
||||
return self.list(request, *args, **kwargs)
|
||||
|
||||
|
@ -252,6 +268,7 @@ class RetrieveUpdateAPIView(mixins.RetrieveModelMixin,
|
|||
"""
|
||||
Concrete view for retrieving, updating a model instance.
|
||||
"""
|
||||
|
||||
def get(self, request, *args, **kwargs):
|
||||
return self.retrieve(request, *args, **kwargs)
|
||||
|
||||
|
@ -268,6 +285,7 @@ class RetrieveDestroyAPIView(mixins.RetrieveModelMixin,
|
|||
"""
|
||||
Concrete view for retrieving or deleting a model instance.
|
||||
"""
|
||||
|
||||
def get(self, request, *args, **kwargs):
|
||||
return self.retrieve(request, *args, **kwargs)
|
||||
|
||||
|
@ -282,6 +300,7 @@ class RetrieveUpdateDestroyAPIView(mixins.RetrieveModelMixin,
|
|||
"""
|
||||
Concrete view for retrieving, updating or deleting a model instance.
|
||||
"""
|
||||
|
||||
def get(self, request, *args, **kwargs):
|
||||
return self.retrieve(request, *args, **kwargs)
|
||||
|
||||
|
|
|
@ -15,6 +15,7 @@ class CreateModelMixin:
|
|||
"""
|
||||
Create a model instance.
|
||||
"""
|
||||
|
||||
def create(self, request, *args, **kwargs):
|
||||
serializer = self.get_serializer(data=request.data)
|
||||
serializer.is_valid(raise_exception=True)
|
||||
|
@ -32,10 +33,33 @@ class CreateModelMixin:
|
|||
return {}
|
||||
|
||||
|
||||
class BulkCreateModelMixin:
|
||||
"""
|
||||
Bulk create model instances.
|
||||
"""
|
||||
|
||||
def create(self, request, *args, **kwargs):
|
||||
serializer = self.get_serializer(data=request.data, many=True)
|
||||
serializer.is_valid(raise_exception=True)
|
||||
self.perform_bulk_create(serializer)
|
||||
headers = self.get_success_headers(serializer.data)
|
||||
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
|
||||
|
||||
def perform_bulk_create(self, serializer):
|
||||
serializer.save()
|
||||
|
||||
def get_success_headers(self, data):
|
||||
try:
|
||||
return {'Location': str(data[api_settings.URL_FIELD_NAME])}
|
||||
except (TypeError, KeyError):
|
||||
return {}
|
||||
|
||||
|
||||
class ListModelMixin:
|
||||
"""
|
||||
List a queryset.
|
||||
"""
|
||||
|
||||
def list(self, request, *args, **kwargs):
|
||||
queryset = self.filter_queryset(self.get_queryset())
|
||||
|
||||
|
@ -52,6 +76,7 @@ class RetrieveModelMixin:
|
|||
"""
|
||||
Retrieve a model instance.
|
||||
"""
|
||||
|
||||
def retrieve(self, request, *args, **kwargs):
|
||||
instance = self.get_object()
|
||||
serializer = self.get_serializer(instance)
|
||||
|
@ -62,6 +87,7 @@ class UpdateModelMixin:
|
|||
"""
|
||||
Update a model instance.
|
||||
"""
|
||||
|
||||
def update(self, request, *args, **kwargs):
|
||||
partial = kwargs.pop('partial', False)
|
||||
instance = self.get_object()
|
||||
|
@ -91,6 +117,7 @@ class DestroyModelMixin:
|
|||
"""
|
||||
Destroy a model instance.
|
||||
"""
|
||||
|
||||
def destroy(self, request, *args, **kwargs):
|
||||
instance = self.get_object()
|
||||
self.perform_destroy(instance)
|
||||
|
|
Loading…
Reference in New Issue
Block a user