Modified bedCreate API View to support bulk creating beds

care/facility/api/serializers/bed.py

@@ -2,6 +2,7 @@
 from rest_framework.exceptions import ValidationError
 from rest_framework.serializers import (
     BooleanField,
+    IntegerField,
     ModelSerializer,
     SerializerMethodField,
     UUIDField,
@@ -31,6 +32,8 @@ class BedSerializer(ModelSerializer):
     location = UUIDField(write_only=True, required=True)
     facility = UUIDField(write_only=True, required=True)
 
+    number_of_beds = IntegerField(required=False, default=1, write_only=True)
+
     class Meta:
         model = Bed
         exclude = ("deleted", "external_id", "assets")
@@ -57,6 +60,11 @@ def validate(self, attrs):
             raise ValidationError(
                 {"location": "Field is Required", "facility": "Field is Required"}
             )
+
+        if attrs["number_of_beds"] > 100:
+            raise ValidationError(
+                {"number_of_beds": "Cannot create more than 100 beds at once."}
+            )
         return super().validate(attrs)
 
 

care/facility/api/viewsets/bed.py

@@ -3,6 +3,7 @@
 from django_filters import rest_framework as filters
 from drf_spectacular.utils import extend_schema, extend_schema_view
 from rest_framework import filters as drf_filters
+from rest_framework import status
 from rest_framework.exceptions import PermissionDenied
 from rest_framework.exceptions import ValidationError as DRFValidationError
 from rest_framework.fields import get_error_detail
@@ -14,6 +15,7 @@
     UpdateModelMixin,
 )
 from rest_framework.permissions import IsAuthenticated
+from rest_framework.response import Response
 from rest_framework.viewsets import GenericViewSet
 
 from care.facility.api.serializers.bed import (
@@ -57,6 +59,28 @@ class BedViewSet(
     search_fields = ["name"]
     filterset_class = BedFilter
 
+    def create(self, request, *args, **kwargs):
+        serializer = self.get_serializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        validated_data = serializer.validated_data
+        number_of_beds = validated_data.pop("number_of_beds", 1)
+        # Bulk creating n number of beds
+        if number_of_beds > 1 and number_of_beds <= 100:
+            objs = []
+            for i in range(1, number_of_beds + 1):
+                temp_data = dict(validated_data.copy())
+                temp_data["name"] = temp_data["name"] + f" - {i}"
+                objs.append(Bed(**temp_data))
+
+            res = Bed.objects.bulk_create(objs=objs, batch_size=number_of_beds)
+            return Response(res, status=status.HTTP_201_CREATED)
+
+        self.perform_create(serializer)
+        headers = self.get_success_headers(serializer.data)
+        return Response(
+            serializer.data, status=status.HTTP_201_CREATED, headers=headers
+        )
+
     def get_queryset(self):
         user = self.request.user
         queryset = self.queryset

care/facility/tests/test_bed_create.py

@@ -0,0 +1,61 @@
+from rest_framework import status
+from rest_framework.test import APIRequestFactory, APITestCase
+
+from care.facility.api.viewsets.bed import BedViewSet
+from care.facility.models import AssetLocation
+from care.facility.tests.mixins import TestClassMixin
+from care.utils.tests.test_base import TestBase
+
+
+class SingleBedTest(TestBase, TestClassMixin, APITestCase):
+    def setUp(self):
+        self.factory = APIRequestFactory()
+        state = self.create_state()
+        district = self.create_district(state=state)
+        self.user = self.create_user(district=district, username="test user")
+        self.facility = self.create_facility(district=district, user=self.user)
+        self.asset_location = AssetLocation.objects.create(
+            name="asset location", location_type=1, facility=self.facility
+        )
+
+    def test_create(self):
+        user = self.user
+        sample_data = {
+            "bed_type": "REGULAR",
+            "description": "Testing creation of beds.",
+            "facility": self.facility.external_id,
+            "location": self.asset_location.external_id,
+            "name": "Test Bed",
+            "number_of_beds": 1,
+        }
+        response = self.new_request(
+            ("/api/v1/bed/", sample_data, "json"), {"post": "create"}, BedViewSet, user
+        )
+        self.assertIs(response.status_code, status.HTTP_201_CREATED)
+
+
+class MultipleBedTest(TestBase, TestClassMixin, APITestCase):
+    def setUp(self):
+        self.factory = APIRequestFactory()
+        state = self.create_state()
+        district = self.create_district(state=state)
+        self.user = self.create_user(district=district, username="test user")
+        self.facility = self.create_facility(district=district, user=self.user)
+        self.asset_location = AssetLocation.objects.create(
+            name="asset location", location_type=1, facility=self.facility
+        )
+
+    def test_create(self):
+        user = self.user
+        sample_data = {
+            "bed_type": "REGULAR",
+            "description": "Testing creation of beds.",
+            "facility": self.facility.external_id,
+            "location": self.asset_location.external_id,
+            "name": "Test Bed",
+            "number_of_beds": 5,
+        }
+        response = self.new_request(
+            ("/api/v1/bed/", sample_data, "json"), {"post": "create"}, BedViewSet, user
+        )
+        self.assertIs(response.status_code, status.HTTP_201_CREATED)