Add category info to part serializer

This commit is contained in:
Oliver 2018-05-02 20:57:04 +10:00
parent 89a1fc59bb
commit 45510b1adc

View File

@ -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',