initial commit

core/apps/contracts/serializers/__init__.py

@@ -0,0 +1,4 @@
+from .attached_files import *  # noqa
+from .contracts import *  # noqa
+from .file_contents import *  # noqa
+from .owners import *  # noqa

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

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

core/apps/contracts/serializers/attached_files/attached_files.py

@@ -0,0 +1,47 @@
+from rest_framework import serializers  # type: ignore
+
+from core.apps.contracts.models import ContractAttachedFileModel
+from core.apps.contracts.serializers.file_contents import (
+    RetrieveContractFileContentSerializer
+)
+
+
+class BaseContractAttachedFileSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ContractAttachedFileModel
+        fields = "__all__"
+        read_only_fields = (
+            "id",
+            "created_at",
+            "updated_at",
+        )
+
+
+class ListContractAttachedFileSerializer(BaseContractAttachedFileSerializer):
+    class Meta(BaseContractAttachedFileSerializer.Meta): 
+        fields = (
+            "id",
+            "name",
+            "created_at",
+            "updated_at",
+        )
+
+
+class RetrieveContractAttachedFileSerializer(BaseContractAttachedFileSerializer):
+    contents = RetrieveContractFileContentSerializer(many=True, read_only=True)
+    
+    class Meta(BaseContractAttachedFileSerializer.Meta):
+        fields = "__all__"
+
+
+class CreateContractAttachedFileSerializer(BaseContractAttachedFileSerializer):
+    class Meta(BaseContractAttachedFileSerializer.Meta): ...
+
+
+class UpdateContractAttachedFileSerializer(BaseContractAttachedFileSerializer):
+    class Meta(BaseContractAttachedFileSerializer.Meta): ...
+
+
+class DestroyContractAttachedFileSerializer(BaseContractAttachedFileSerializer):
+    class Meta(BaseContractAttachedFileSerializer.Meta): 
+        fields = ["id"]

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

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

core/apps/contracts/serializers/contracts/contracts.py

@@ -0,0 +1,42 @@
+from rest_framework import serializers
+
+from core.apps.contracts.models import ContractModel
+
+
+class BaseContractSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ContractModel
+        fields = "__all__"
+        read_only_fields = (
+            "id",
+            "created_at",
+            "updated_at"
+        )
+
+
+class ListContractSerializer(BaseContractSerializer):
+    class Meta(BaseContractSerializer.Meta): 
+        fields = (
+            "id",
+            "name",
+            "identifier",
+            "created_at",
+            "updated_at",
+        )
+
+
+class RetrieveContractSerializer(BaseContractSerializer):
+    class Meta(BaseContractSerializer.Meta): ...
+
+
+class CreateContractSerializer(BaseContractSerializer):
+    class Meta(BaseContractSerializer.Meta): ...
+
+
+class UpdateContractSerializer(BaseContractSerializer):
+    class Meta(BaseContractSerializer.Meta): ...
+
+
+class DestroyContractSerializer(BaseContractSerializer):
+    class Meta(BaseContractSerializer.Meta): 
+        fields = ["id"]

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

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

core/apps/contracts/serializers/file_contents/file_contents.py

@@ -0,0 +1,51 @@
+from rest_framework import serializers  # type: ignore
+from core.apps.contracts.models import ContractFileContentModel
+
+
+class BaseContractFileContentSerializer(serializers.ModelSerializer):
+    document = serializers.FileField(required=True)
+
+    class Meta:
+        model = ContractFileContentModel
+        fields = "__all__"
+        read_only_fields = (
+            "id",
+            "created_at",
+            "updated_at",
+        )
+
+
+class ListContractFileContentSerializer(BaseContractFileContentSerializer):
+    class Meta(BaseContractFileContentSerializer.Meta): ...
+
+
+class RetrieveContractFileContentSerializer(BaseContractFileContentSerializer):
+    class Meta(BaseContractFileContentSerializer.Meta): ...
+
+
+class CreateContractFileContentSerializer(BaseContractFileContentSerializer):
+    class Meta(BaseContractFileContentSerializer.Meta): ...
+
+
+class UpdateContractFileContentSerializer(BaseContractFileContentSerializer):
+    class Meta(BaseContractFileContentSerializer.Meta): ...
+
+
+class DestroyContractFileContentSerializer(BaseContractFileContentSerializer):
+    class Meta(BaseContractFileContentSerializer.Meta): 
+        fields = ["id"]
+
+
+class CreateContractFileContentFromOwnerSerializer(BaseContractFileContentSerializer):
+    class Meta(BaseContractFileContentSerializer.Meta):
+        fields = "__all__"
+        read_only_fields = (
+            *BaseContractFileContentSerializer.Meta.read_only_fields,
+            "file",
+            "contract_owner",
+        )
+    def create(self, validated_data):  # type: ignore
+        validated_data["contract_owner_id"] = self.context["owner_id"]
+        validated_data["file_id"] = self.context["file_id"]
+        return super().create(validated_data)  # type: ignore
+    

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"]
+