Dovodi_Backend/apps/hadis/serializers/reference_v2.py

from rest_framework import serializers
from rest_framework.pagination import PageNumberPagination
from ..models import BookReference, BookAuthor, BookEdition, BookVolume
from .category import get_localized_text
from apps.hadis.serializers import HadisListSerializer, HadisShortSerializer
from ..serializers.category import LocalizedField
from ..serializers.reference import BookAttributeSerializer, BookReferenceImageSerializer
from rest_framework.pagination import PageNumberPagination
from ..models import BookReference, BookAuthor, BookEdition, BookVolume
from .category import get_localized_text
from apps.hadis.serializers import HadisListSerializer
from ..serializers.category import LocalizedField

class BookAuthorV2Serializer(serializers.ModelSerializer):
    name = LocalizedField()
    class Meta:
        model = BookAuthor
        fields = ['id', 'name', 'slug']

class BookEditionV2Serializer(serializers.ModelSerializer):
    publisher = LocalizedField()
    notes = LocalizedField()
    tags = LocalizedField()
    publication_place = serializers.SerializerMethodField()

    class Meta:
        model = BookEdition
        fields = [
            'id', 'publisher', 'publication_place', 'edition_number', 
            'year_of_publication', 'number_of_volumes', 'notes', 'source_url', 'tags'
        ]

    def get_publication_place(self, obj):
        request = self.context.get("request")
        city = get_localized_text(obj.city_of_publication, request) if obj.city_of_publication else ""
        country = get_localized_text(obj.country_of_publication, request) if obj.country_of_publication else ""
        return {
            "city": city,
            "country": country
        }

class BookVolumeV2Serializer(serializers.ModelSerializer):
    class Meta:
        model = BookVolume
        fields = ['id', 'title']

class BookReferenceV2DetailSerializer(serializers.ModelSerializer):
    title = LocalizedField()
    authors = BookAuthorV2Serializer(many=True, read_only=True)
    
    information = serializers.SerializerMethodField()
    excerpts = serializers.SerializerMethodField()
    volumes = serializers.SerializerMethodField()

    class Meta:
        model = BookReference
        fields = ['id', 'title', 'authors', 'information', 'excerpts', 'volumes']

    def get_information(self, obj):
        request = self.context.get("request")
        summary = get_localized_text(obj.description, request) if obj.description else ""
        editions = BookEditionV2Serializer(obj.editions.all(), many=True, context=self.context).data
        
        return {
            "summary": summary,
            "editions": editions
        }

    def get_excerpts(self, obj):
        from apps.hadis.models import Hadis
        hadis_ids = obj.hadis_references.values_list('hadis', flat=True)
        hadis_qs = Hadis.objects.filter(id__in=hadis_ids, status=True).order_by('id')

        paginator = PageNumberPagination()
        paginator.page_size = 20
        request = self.context.get("request")
        
        paginated_hadis = paginator.paginate_queryset(hadis_qs, request)

        if paginated_hadis is not None:
            serializer = HadisListSerializer(paginated_hadis, many=True, context=self.context)
            return {
                'count': paginator.page.paginator.count,
                'next': paginator.get_next_link(),
                'previous': paginator.get_previous_link(),
                'results': serializer.data
            }

        return {
            'count': hadis_qs.count(),
            'next': None,
            'previous': None,
            'results': HadisListSerializer(hadis_qs, many=True, context=self.context).data
        }

    def get_volumes(self, obj):
        editions = obj.editions.all()
        if editions.exists():
            editions_data = []
            for ed in editions:
                editions_data.append({
                    "edition_id": ed.id,
                    "edition_statement": ed.edition_number or f"Edition {ed.id}",
                    "volumes": BookVolumeV2Serializer(ed.volumes.all(), many=True).data
                })
            return {
                "has_editions": True,
                "editions_volumes": editions_data,
                "all_volumes": []
            }
        else:
            return {
                "has_editions": False,
                "editions_volumes": [],
                "all_volumes": BookVolumeV2Serializer(obj.volumes.all(), many=True).data
            }

class BookReferenceV2SyncSerializer(serializers.ModelSerializer):
    title = LocalizedField()
    authors = BookAuthorV2Serializer(many=True, read_only=True)
    attributes = BookAttributeSerializer(many=True, read_only=True)
    images = BookReferenceImageSerializer(many=True, read_only=True)
    
    information = serializers.SerializerMethodField()
    hadises = serializers.SerializerMethodField()
    volumes = serializers.SerializerMethodField()

    class Meta:
        model = BookReference
        fields = [
            'id', 'title', 'slug', 'share_link', 'rate', 'authors', 
            'attributes', 'images', 'information', 'hadises', 'volumes'
        ]

    def get_information(self, obj):
        request = self.context.get("request")
        summary = get_localized_text(obj.description, request) if obj.description else ""
        editions = BookEditionV2Serializer(obj.editions.all(), many=True, context=self.context).data
        return {
            "summary": summary,
            "editions": editions,
            "language": get_localized_text(obj.language, request) if obj.language else "",
            "number_page": obj.number_page
        }

    def get_hadises(self, obj):
        references = obj.hadis_references.all()
        hadis_list = [ref.hadis for ref in references if ref.hadis]
        return HadisShortSerializer(hadis_list, many=True, context=self.context).data

    def get_volumes(self, obj):
        editions = obj.editions.all()
        if editions.exists():
            editions_data = []
            for ed in editions:
                editions_data.append({
                    "edition_id": ed.id,
                    "edition_statement": ed.edition_number or f"Edition {ed.id}",
                    "volumes": BookVolumeV2Serializer(ed.volumes.all(), many=True).data
                })
            return {
                "has_editions": True,
                "editions_volumes": editions_data,
                "all_volumes": []
            }
        else:
            return {
                "has_editions": False,
                "editions_volumes": [],
                "all_volumes": BookVolumeV2Serializer(obj.volumes.all(), many=True).data
            }