123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- from django.contrib.contenttypes.models import ContentType
- from rest_framework import serializers
- from extras.models import CF_TYPE_SELECT, CustomField, CustomFieldChoice, CustomFieldValue
- class CustomFieldsSerializer(serializers.BaseSerializer):
- def to_representation(self, obj):
- return obj
- class CustomFieldModelSerializer(serializers.ModelSerializer):
- """
- Extends ModelSerializer to render any CustomFields and their values associated with an object.
- """
- custom_fields = CustomFieldsSerializer()
- def __init__(self, *args, **kwargs):
- super(CustomFieldModelSerializer, self).__init__(*args, **kwargs)
-
- content_type = ContentType.objects.get_for_model(self.Meta.model)
- custom_fields = {f.name: None for f in CustomField.objects.filter(obj_type=content_type)}
-
- for cfv in self.instance.custom_field_values.all():
- if cfv.field.type == CF_TYPE_SELECT:
- custom_fields[cfv.field.name] = CustomFieldChoiceSerializer(cfv.value).data
- else:
- custom_fields[cfv.field.name] = cfv.value
- self.instance.custom_fields = custom_fields
- class CustomFieldChoiceSerializer(serializers.ModelSerializer):
- """
- Imitate utilities.api.ChoiceFieldSerializer
- """
- value = serializers.IntegerField(source='pk')
- label = serializers.CharField(source='value')
- class Meta:
- model = CustomFieldChoice
- fields = ['value', 'label']
|