from rest_framework import serializers
from rest_framework_gis.serializers import GeoFeatureModelSerializer
from .models import Poi, PoiCategory, PoiIcon, PoiImages

class PoiSerializer(GeoFeatureModelSerializer):
    icon = serializers.SerializerMethodField()
    images = serializers.SerializerMethodField()

    class Meta:
        model = Poi
        geo_field = 'geom'
        fields = '__all__'

    def get_icon(self, Poi):
        """u
        :param Poi: poi obeject
        :return: a field ie property called icon for model Poi
        """
        if Poi.category and Poi.category.fk_poi_icon is not None:
            if Poi.category.fk_poi_icon.icon.url:
                icon_url = Poi.category.fk_poi_icon.icon.url
                if icon_url:
                    return icon_url
                else:
                    return None
            else:
                return None
        else:
            return None

    def get_images(self, Poi):
        """
        :param Poi: poi obeject
        :return: a field ie property called icon for model Poi
        """
        if PoiImages.objects.filter(poi_id=Poi.id):
            serializer = PoiImageSerializer(PoiImages.objects.filter(poi_id=Poi.id).order_by('sort_order'), many=True)
            return serializer.data
        else:
            return None

    def to_representation(self, instance):
        representation = super().to_representation(instance)
        representation['properties']['id'] = representation['id']
        representation['properties']['src_icon'] = "poi"
        return representation


class PoiCategorySerializer(serializers.ModelSerializer):
    icon = serializers.SerializerMethodField()
    class Meta:
        model = PoiCategory
        fields = ('id', 'cat_name', 'cat_name_en', 'cat_name_de', 'icon', 'fk_poi_icon', 'enabled', 'parent',
                  'html_content', 'html_content_de', 'tree_order', 'description')

    def get_icon(self, PoiCategory):
        if PoiCategory.fk_poi_icon and PoiCategory.fk_poi_icon.icon is not None:
            if hasattr(PoiCategory.fk_poi_icon.icon, "url"):
                return PoiCategory.fk_poi_icon.icon.url
            else:
                return None
        else:
            return None

class PoiIconSerializer(serializers.ModelSerializer):

    class Meta:
        model = PoiIcon
        fields = ('id', 'name', 'icon')

class PoiImageSerializer(serializers.ModelSerializer):
    class Meta:
        model = PoiImages
        fields = ( 'id', 'poi', 'image', 'thumbnail', 'alt_text', 'sort_order', 'is_default')
        # read_only_fields = ('thumb', 'thumbnail')