From 45510b1adc401244297fb281b8f6ecd22f7c4b0e Mon Sep 17 00:00:00 2001 From: Oliver Date: Wed, 2 May 2018 20:57:04 +1000 Subject: [PATCH] Add category info to part serializer --- InvenTree/part/serializers.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/InvenTree/part/serializers.py b/InvenTree/part/serializers.py index 902c2b16a1..5b2a0f481a 100644 --- a/InvenTree/part/serializers.py +++ b/InvenTree/part/serializers.py @@ -8,16 +8,31 @@ class PartSerializer(serializers.ModelSerializer): Used when displaying all details of a single component. """ + def _category_name(self, part): + if part.category: + return part.category.name + return '' + + def _category_url(self, part): + if part.category: + return part.category.get_absolute_url() + return '' + + category_name = serializers.SerializerMethodField('_category_name') + category_url = serializers.SerializerMethodField('_category_url') + class Meta: model = Part fields = [ + 'pk', 'url', # Link to the part detail page 'name', 'IPN', 'URL', # Link to an external URL (optional) 'description', 'category', - 'category_path', + 'category_name', + 'category_url', 'total_stock', 'available_stock', 'units',