diff --git a/apps/product/web/api/v1/serializers/quota_serializers.py b/apps/product/web/api/v1/serializers/quota_serializers.py index 017ab43..07e1c02 100644 --- a/apps/product/web/api/v1/serializers/quota_serializers.py +++ b/apps/product/web/api/v1/serializers/quota_serializers.py @@ -149,6 +149,17 @@ class QuotaLiveStockAgeLimitationSerializer(serializers.ModelSerializer): model = product_models.QuotaLiveStockAgeLimitation fields = '__all__' + def to_representation(self, instance): + """ custom output for livestock type """ + + representation = super().to_representation(instance) + if isinstance(instance, product_models.QuotaLiveStockAgeLimitation): + representation['livestock_type'] = LiveStockTypeSerializer( + instance.livestock_type + ).data + + return representation + def update(self, instance, validated_data): """ Custom Update """