gealogiuya

apps/geology/serializers/__init__.py

@@ -0,0 +1,4 @@
+from .category import *  # noqa
+from .gallery import *  # noqa
+from .geology import *  # noqa
+from .photo import *  # noqa

apps/geology/serializers/category.py

@@ -0,0 +1,16 @@
+from typing import Type
+
+from rest_framework import serializers
+
+from apps.geology.models import Category, Geology
+
+
+class CategorySerializer(serializers.ModelSerializer):
+    count = serializers.SerializerMethodField()
+
+    class Meta:
+        model: Type[Category] = Category
+        fields: str = ("id", "count", "name", "created_at", "updated_at")
+
+    def get_count(self, obj):
+        return Geology.objects.filter(category=obj).count()

apps/geology/serializers/gallery.py

@@ -0,0 +1,17 @@
+from rest_framework import serializers
+
+from apps.geology.models import Gallery, GalleryCategory
+
+
+class GallerySerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Gallery
+        fields = "__all__"
+
+
+class GalleryCategorySerializer(serializers.ModelSerializer):
+    gallery = GallerySerializer(many=True, read_only=True)
+
+    class Meta:
+        model = GalleryCategory
+        fields = ("id", "name", "count", "gallery", "created_at", "updated_at")

apps/geology/serializers/geology.py

@@ -0,0 +1,161 @@
+import logging
+from urllib.parse import urlparse
+
+from drf_spectacular.utils import extend_schema_field
+from rest_framework import exceptions, status
+from rest_framework import serializers
+from rest_framework.response import Response
+
+from apps.geology.models import Geology, Photo
+
+
+class GeologySerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Geology
+        fields = (
+            "id",
+            "name",
+            "title",
+            "body1",
+            "body2",
+            "body3",
+            "banner_title",
+            "banner",
+            "address",
+            "image",
+            "location",
+            "composition",
+            "isp",
+            "analysis",
+            "category",
+            "created_at",
+            "updated_at",
+        )
+
+    def create(self, validated_data):
+        try:
+            request = self.context.get("request")
+            photo_data1 = []
+            photo_data2 = []
+
+            for key in request.data:
+                if key.startswith("description1[") and key.endswith("]"):
+                    index = key[len("description1[") : -1]
+                    file_key = f"image1[{index}]"
+                    photo_desc = request.data.get(key)
+                    photo_file = request.FILES.get(file_key)
+
+                    if not photo_file:
+                        return Response(
+                            {
+                                "error": f"File is required for photo item {index} in photos1"
+                            },
+                            status=status.HTTP_400_BAD_REQUEST,
+                        )
+                    if not photo_desc:
+                        photo_desc = photo_file.name
+                    photo_data1.append({"image": photo_file, "description": photo_desc})
+
+                if key.startswith("description2[") and key.endswith("]"):
+                    index = key[len("description2[") : -1]
+                    file_key = f"image2[{index}]"
+                    photo_desc = request.data.get(key)
+                    photo_file = request.FILES.get(file_key)
+
+                    if not photo_file:
+                        return Response(
+                            {
+                                "error": f"File is required for photo item {index} in photos2"
+                            },
+                            status=status.HTTP_400_BAD_REQUEST,
+                        )
+                    if not photo_desc:
+                        photo_desc = photo_file.name
+                    photo_data2.append({"image": photo_file, "description": photo_desc})
+
+            geology = Geology.objects.create(
+                name=validated_data.get("name"),
+                title=validated_data.get("title"),
+                body1=validated_data.get("body1"),
+                body2=validated_data.get("body2"),
+                body3=validated_data.get("body3"),
+                banner_title=validated_data.get("banner_title"),
+                banner=validated_data.get("banner"),
+                address=validated_data.get("address"),
+                location=validated_data.get("location"),
+                composition=validated_data.get("composition"),
+                isp=validated_data.get("isp"),
+                analysis=validated_data.get("analysis"),
+                category=validated_data.get("category"),
+            )
+
+            for photo in photo_data1:
+                photo_instance = Photo.objects.create(
+                    image=photo["image"], description=photo["description"]
+                )
+                geology.photos1.add(photo_instance)
+
+            for photo in photo_data2:
+                photo_instance = Photo.objects.create(
+                    image=photo["image"], description=photo["description"]
+                )
+                geology.photos2.add(photo_instance)
+            return geology
+        except Exception as e:
+            logging.error(f"Error in create method: {str(e)}")
+            raise exceptions.ValidationError({"detail": str(e)})
+
+
+class GeologyDetailSerializer(serializers.ModelSerializer):
+    category = serializers.SerializerMethodField()
+    photos1 = serializers.SerializerMethodField()
+    photos2 = serializers.SerializerMethodField()
+
+    class Meta:
+        model = Geology
+        fields = (
+            "id",
+            "category",
+            "banner_title",
+            "banner",
+            "name",
+            "title",
+            "body1",
+            "address",
+            "image",
+            "location",
+            "composition",
+            "isp",
+            "analysis",
+            "body2",
+            "photos1",
+            "body3",
+            "photos2",
+            "created_at",
+            "updated_at",
+        )
+
+    def to_representation(self, instance):
+        representation = super().to_representation(instance)
+        if representation.get("banner"):
+            parsed_url = urlparse(representation["banner"])
+            representation["banner"] = parsed_url.path
+        return representation
+
+    @extend_schema_field(serializers.DictField())
+    def get_category(self, obj):
+        from apps.geology.serializers.category import CategorySerializer
+
+        return CategorySerializer(obj.category).data
+
+    @extend_schema_field(serializers.ListField(child=serializers.DictField()))
+    def get_photos1(self, obj):
+        from apps.geology.serializers.photo import PhotoSerializer
+
+        return PhotoSerializer(obj.photos1.all(), many=True).data
+
+    @extend_schema_field(serializers.ListField(child=serializers.DictField()))
+    def get_photos2(self, obj):
+        from apps.geology.serializers.photo import PhotoSerializer
+
+        return PhotoSerializer(obj.photos2.all(), many=True).data

apps/geology/serializers/photo.py

@@ -0,0 +1,11 @@
+from typing import Type
+
+from rest_framework import serializers
+
+from apps.geology.models import Photo
+
+
+class PhotoSerializer(serializers.ModelSerializer):
+    class Meta:
+        model: Type[Photo] = Photo
+        fields: str = "__all__"