from rest_framework import serializers
from company_gallery.models.company_gallery import CompanyGallery

class CompanyGallerySerializer(serializers.ModelSerializer):
    id = serializers.ReadOnlyField()
    image_url = serializers.SerializerMethodField('get_image_url')
    # image_url = serializers.SerializerMethodField()

    class Meta:
        model = CompanyGallery
        fields = ['id', 'company', 'image','image_url', 'caption']
        read_only_fields = ['id']

    def get_image_url(self, obj):
        request = self.context.get("request")
        if obj.image and hasattr(obj.image, 'url'):
            return request.build_absolute_uri(obj.image.url)
        return None