initial commit

core/apps/contracts/serializers/owners/__init__.py

@@ -0,0 +1 @@
+from .owner import *  # noqa

core/apps/contracts/serializers/owners/owner.py

@@ -0,0 +1,174 @@
+from rest_framework import serializers
+
+from core.apps.contracts.models import (
+    ContractOwnerModel,
+    LegalEntityModel,
+    IndividualModel,
+)
+
+from django.db import transaction
+
+# from core.apps.contracts.serializers.attached_files import (
+#     BaseContractAttachedFileSerializer,
+# )
+# from core.apps.contracts.serializers.file_contents import (
+#     BaseContractFileContentSerializer
+# )
+
+
+#! TODO fix: BaseContractOwnerSerializer (.create/.update) fix
+class BaseIndividualSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = IndividualModel
+        fields = "__all__"
+        read_only_fields = (
+            "id",
+            "created_at",
+            "updated_at",
+        )
+        extra_kwargs = {
+            "iin_code": {
+                "required": False
+            },
+            "person_code": {
+                "required": False
+            }
+        }
+
+
+
+class ListIndividualSerializer(BaseIndividualSerializer):
+    class Meta(BaseIndividualSerializer.Meta):
+        fields = (
+            "id",
+            "full_name",
+            "phone",
+            "created_at",
+            "updated_at",
+        )
+
+
+class BaseLegalEntitySerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LegalEntityModel
+        fields = "__all__"
+        read_only_fields = (
+            "id",
+            "created_at",
+            "updated_at",
+        )
+        extra_kwargs = {
+            "bin_code": {
+                "required": False
+            },
+            "identifier": {
+                "required": False
+            }
+        }
+
+
+
+class ListLegalEntitySerializer(BaseLegalEntitySerializer):
+    class Meta(BaseLegalEntitySerializer.Meta):
+        fields = (
+            "id",
+            "name",
+            "phone",
+            "created_at",
+            "updated_at",
+        )
+
+
+class BaseContractOwnerSerializer(serializers.ModelSerializer):
+    legal_entity = BaseLegalEntitySerializer(required=False)
+    individual = BaseIndividualSerializer(required=False)
+
+    class Meta:
+        model = ContractOwnerModel
+        fields = "__all__"
+        read_only_fields = (
+            "id",
+            "created_at",
+            "updated_at",
+        )
+        extra_kwargs = {
+            "legal_entity": {
+                "required": False
+            },
+            "individual": {
+                "required": False
+            }
+        }
+
+    def create(self, validated_data):
+        legal_entity_data = validated_data.pop("legal_entity", None)
+        individual_data = validated_data.pop("individual", None)
+
+        with transaction.atomic():
+            if legal_entity_data is not None:
+                legal_entity_serializer = BaseLegalEntitySerializer(data=legal_entity_data)
+                legal_entity_serializer.is_valid(raise_exception=True)
+                validated_data["legal_entity"] = legal_entity_serializer.save()
+
+            if individual_data is not None:
+                individual_serializer = BaseIndividualSerializer(data=individual_data)
+                individual_serializer.is_valid(raise_exception=True)
+                validated_data["individual"] = individual_serializer.save()
+
+            contract_owner = ContractOwnerModel.objects.create(**validated_data)
+            return contract_owner
+    
+    def update(self, instance, validated_data):
+        legal_entity_data = validated_data.pop("legal_entity", None)
+        individual_data = validated_data.pop("individual", None)
+
+        with transaction.atomic():
+            if legal_entity_data is not None:
+                if instance.legal_entity:
+                    for attr, value in legal_entity_data.items():
+                        setattr(instance.legal_entity, attr, value)
+                    instance.legal_entity.save()
+                else:
+                    legal_entity = LegalEntityModel.objects.create(**legal_entity_data)
+                    instance.legal_entity = legal_entity
+
+            if individual_data is not None:
+                if instance.individual:
+                    for attr, value in individual_data.items():
+                        setattr(instance.individual, attr, value)
+                    instance.individual.save()
+                else:
+                    individual = IndividualModel.objects.create(**individual_data)
+                    instance.individual = individual
+
+            # Update ContractOwnerModel fields
+            for attr, value in validated_data.items():
+                setattr(instance, attr, value)
+            instance.save()
+
+        return instance
+        
+
+class ListContractOwnerSerializer(BaseContractOwnerSerializer):
+    legal_entity = ListLegalEntitySerializer(read_only=True)
+    individual = ListIndividualSerializer(read_only=True)
+
+    class Meta(BaseContractOwnerSerializer.Meta): ...
+
+
+class RetrieveContractOwnerSerializer(BaseContractOwnerSerializer):
+    class Meta(BaseContractOwnerSerializer.Meta): ...
+
+
+class CreateContractOwnerSerializer(BaseContractOwnerSerializer):
+    class Meta(BaseContractOwnerSerializer.Meta): ...
+
+
+class UpdateContractOwnerSerializer(BaseContractOwnerSerializer):
+    class Meta(BaseContractOwnerSerializer.Meta): ...
+
+
+class DestroyContractOwnerSerializer(BaseContractOwnerSerializer):
+    class Meta(BaseContractOwnerSerializer.Meta): 
+        fields = ["id"]
+