from apps.authentication.api.v1.serializers.serializer import (
    UserSerializer,
    OrganizationSerializer,
    ProvinceSerializer,
    CitySerializer
)
from rest_framework import serializers
from apps.herd.models import Herd, Rancher


class HerdSerializer(serializers.ModelSerializer):
    """ Herd Serializer """
    class Meta:
        model = Herd
        fields = '__all__'

    def to_representation(self, instance):
        """ Customize serializer output """
        representation = super().to_representation(instance)
        if isinstance(instance, Herd):
            representation['owner'] = UserSerializer(instance.owner).data
            representation['cooperative'] = OrganizationSerializer(instance.cooperative).data
            representation['province'] = ProvinceSerializer(instance.province).data
            representation['city'] = CitySerializer(instance.city).data
            representation['contractor'] = OrganizationSerializer(instance.contractor).data

        return representation


class RancherSerializer(serializers.ModelSerializer):
    class Meta:
        model = Rancher
        fields = '__all__'

    def to_representation(self, instance):
        representation = super().to_representation(instance)

        representation['province'] = {
            'id': instance.province.id,
            'name': instance.province.name
        }

        representation['city'] = {
            'id': instance.city.id,
            'name': instance.city.name
        }

        return representation