from rest_framework import serializers
from apps.product import models as product_models
from apps.authorization.api.v1 import serializers as authorize_serializers


class ReferenceProductSerializer(serializers.ModelSerializer):
    """ Serializer of reference product """

    class Meta:
        model = product_models.ReferenceProduct
        fields = '__all__'


class ProductSerializer(serializers.ModelSerializer):
    """ Serializer of product """

    class Meta:
        model = product_models.Product
        fields = '__all__'

    def to_representation(self, instance):
        """ Custom output of product serializer """

        representation = super().to_representation(instance)
        if instance.reference:
            representation['reference'] = ReferenceProductSerializer(instance.reference).data

        return representation


class AttributeSerializer(serializers.ModelSerializer):
    """ serialize attributes of reference product """

    class Meta:
        model = product_models.Attribute
        fields = '__all__'

    def to_representation(self, instance):
        representation = super().to_representation(instance)
        if instance.reference_product:
            representation['reference_product'] = ReferenceProductSerializer(
                instance.reference_product
            ).data

        return representation


class AttributeValueSerializer(serializers.ModelSerializer):
    """ serialize attribute values for child products """

    class Meta:
        model = product_models.AttributeValue
        fields = '__all__'

    def to_representation(self, instance):
        """ Custom output """

        representation = super().to_representation(instance)
        if instance.product:
            representation['product'] = ProductSerializer(instance.product).data
        if instance.attribute:
            representation['attribute'] = AttributeSerializer(instance.attribute).data

        return representation


class BrokerSerializer(serializers.ModelSerializer):
    """ serialize product broker """

    class Meta:
        model = product_models.Broker
        fields = '__all__'

    def to_representation(self, instance):
        representation = super().to_representation(instance)
        if instance.reference_product:
            representation['reference_product'] = ReferenceProductSerializer(
                instance.reference_product
            ).data
        if instance.organization_relations:
            representation['organization_relations'] = authorize_serializers.UserRelationSerializer(
                instance.organization_relations
            ).data

        return representation


class SaleUnitSerializer(serializers.ModelSerializer):
    """ serialize unit of products for sale """

    class Meta:
        model = product_models.SaleUnit
        fields = '__all__'

    def to_representation(self, instance):
        representation = super().to_representation(instance)
        if instance.reference_product:
            representation['reference_product'] = ReferenceProductSerializer(
                instance.reference_product
            ).data

        return representation