change api

core/apps/contracts/serializers/contract_side.py

@@ -25,6 +25,11 @@ class ContractSideCreateSerializer(serializers.Serializer):
         return data
     
 
+
+class ContractSideListCreateSerializer(serializers.Serializer):
+    contract_side = ContractSideCreateSerializer(many=True)
+
+
 class ContractSideListSerializer(serializers.ModelSerializer):
     contract_signature = serializers.SerializerMethodField(method_name='get_contract_signature')
 

core/apps/contracts/urls.py

@@ -14,7 +14,7 @@ urlpatterns = [
         ]
     )),
     path('contract_side/', include([
-            path('create/', contract_side_views.ConstartSideCreateApiView.as_view(), name='contract-side-create'),
+            path('create/', contract_side_views.ContractSideCreateApiView.as_view(), name='contract-side-create'),
         ]
     )),
     path('contract_signature/', include(

core/apps/contracts/views/contract_side.py

@@ -1,20 +1,41 @@
+from django.db import transaction
+
 from rest_framework import generics, status, parsers
 from rest_framework.response import Response
 
 from core.apps.contracts.serializers import contract_side as contract_side_serializer
-from core.apps.contracts.models.contract import ContractSide
+from core.apps.contracts.models.contract import ContractSide, Contract
 from core.apps.contracts.tasks.contract_side import create_contract_side
 from core.apps.shared.utils.response import error_message, success_message
+from core.apps.accounts.models import User
 
-class ConstartSideCreateApiView(generics.GenericAPIView):
-    serializer_class = contract_side_serializer.ContractSideCreateSerializer
+class ContractSideCreateApiView(generics.GenericAPIView):
+    serializer_class = contract_side_serializer.ContractSideListCreateSerializer
     queryset = ContractSide.objects.all()
 
+    @transaction.atomic
     def post(self, request):
         serializer = self.serializer_class(data=request.data)
         if serializer.is_valid():
-            # TODO: call celery task
-            create_contract_side.delay(serializer.validated_data)
-            return success_message("Contract side created", 201)
-        return error_message(serializer.error_messages, 400)
-    
+            try:
+                with transaction.atomic():
+                    for side_data in serializer.validated_data['contract_side']:
+                        user = User.objects.get(phone=side_data['phone'])
+                        contract = Contract.objects.get(id=side_data['contract_id'])
+                        ContractSide.objects.create(
+                            full_name=side_data['full_name'],
+                            indentification=side_data['indentification'],
+                            position=side_data.get('position', ''),
+                            has_indentification=side_data['has_indentification'],
+                            user_role=side_data['user_role'],
+                            # phone=side_data['phone'],
+                            contract=contract,
+                            user=user
+                        )
+                return success_message("Contract side created", 201)
+
+            except Exception as e:
+                transaction.set_rollback(True)
+                return error_message(str(e), 400)
+
+        return error_message(serializer.errors, 400)