class CategoriaAlojamientoConAlojamientoSerializer(serializers.ModelSerializer):

    children = serializers.SerializerMethodField()
    def get_children(self, obj):
        alojamientos = Alojamiento.objects.filter(categoria=obj, ciudad=self.ciudad_actual).distinct()
        serializer = AlojamientoSerializer(alojamientos, many=True, read_only=True)
        return serializer.data

    def __init__(self, deciudad=None,  *args,**kwargs):
        super(CategoriaAlojamientoConAlojamientoSerializer, self).__init__(*args, **kwargs)
        self.ciudad_actual = ciudad

    class Meta:
        model = CategoriaAlojamiento
        fields = '__all__'


class CiudadConAlojamientoPorCategoriaSerializer(serializers.ModelSerializer):
    children = serializers.SerializerMethodField()

    def get_children(self, obj):
        categorias = CategoriaAlojamiento.objects.filter(alojamientos__ciudad=obj).distinct()
        serializer = CategoriaAlojamientoConAlojamientoSerializer(categorias,  deciudad=obj, many=True, read_only=True)
        return serializer.data

    class Meta:
        model = Ciudad
        fields = '__all__'