diff --git a/apps/product/web/api/v1/serializers/quota_serializers.py b/apps/product/web/api/v1/serializers/quota_serializers.py index d3b6a03..017ab43 100644 --- a/apps/product/web/api/v1/serializers/quota_serializers.py +++ b/apps/product/web/api/v1/serializers/quota_serializers.py @@ -1,6 +1,7 @@ from apps.authentication.api.v1.serializers.serializer import OrganizationSerializer from apps.authorization.api.v1 import serializers as authorize_serializers from apps.product.web.api.v1.serializers import product_serializers +from apps.livestock.web.api.v1.serializers import LiveStockTypeSerializer from apps.product import models as product_models from rest_framework import serializers @@ -122,6 +123,15 @@ class QuotaLiveStockAllocationSerializer(serializers.ModelSerializer): } } + def to_representation(self, instance): + """ custom output for live stock type """ + + representation = super().to_representation(instance) + if isinstance(instance, product_models.QuotaLivestockAllocation): + representation['livestock_type'] = LiveStockTypeSerializer(instance.livestock_type).data + + return representation + def update(self, instance, validated_data): """ Custom Update """