Search code examples
djangodjango-rest-frameworkdjango-serializer

Send full image url in DRF


I am trying to get the full image URL in the API in DRF i am trying to use build_absolute_uri but i keep receiving the error The 'image' attribute has no file associated with it.

the serializer.py:

class VesselInfoSerializer(serializers.ModelSerializer):

    image_url = serializers.SerializerMethodField()

    def get_image_url(self, Vessel):
        request = self.context.get('request')
        image_url = Vessel.image.url
        return request.build_absolute_uri(image_url)

    vessel_component_count = serializers.IntegerField(
        source='vessel_components.count',
        read_only=True
    )
    vessel_inventory_category_count = serializers.IntegerField(
        source='vessel_inventory_category.count',
        read_only=True
    )
    vessel_inventory_item_count = serializers.IntegerField(
        source='category_items.count',
        read_only=True
    )

    class Meta:
        model = Vessel
        fields = '__all__'

models.py:

class Vessel(models.Model):
    name = models.CharField(max_length=255)
    imo = models.CharField(max_length=255)
    image = models.ImageField(blank=True, upload_to='vessel_image')

    def __str__(self):
        return self.name

the view:

@api_view(['GET'])
def getVesselInfo(request):
    vessels = Vessel.objects.all()
    vSerializer = VesselInfoSerializer(
        vessels, many=True, context={"request": request})
    return Response(vSerializer.data)

Solution

  • You should probably check that the url is not None :

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