|
1 | 1 | from datetime import datetime
|
2 |
| -from rest_framework_json_api import serializers, relations |
| 2 | +from django.db.models.query import QuerySet |
| 3 | +from rest_framework.utils.serializer_helpers import BindingDict |
| 4 | +from rest_framework_json_api import serializers, relations, utils |
3 | 5 | from example import models
|
4 | 6 |
|
5 | 7 |
|
@@ -44,13 +46,13 @@ def __init__(self, *args, **kwargs):
|
44 | 46 | source='comment_set', many=True, read_only=True)
|
45 | 47 | # many related from serializer
|
46 | 48 | suggested = relations.SerializerMethodResourceRelatedField(
|
47 |
| - source='get_suggested', model=Entry, many=True, read_only=True) |
| 49 | + source='get_suggested', model=models.Entry, many=True, read_only=True) |
48 | 50 | # single related from serializer
|
49 | 51 | featured = relations.SerializerMethodResourceRelatedField(
|
50 |
| - source='get_featured', model=Entry, read_only=True) |
| 52 | + source='get_featured', model=models.Entry, read_only=True) |
51 | 53 |
|
52 | 54 | def get_suggested(self, obj):
|
53 |
| - return models.Entry.objects.exclude(pk=obj.pk).first() |
| 55 | + return models.Entry.objects.exclude(pk=obj.pk) |
54 | 56 |
|
55 | 57 | def get_featured(self, obj):
|
56 | 58 | return models.Entry.objects.exclude(pk=obj.pk).first()
|
@@ -108,19 +110,48 @@ class Meta:
|
108 | 110 |
|
109 | 111 | class ProjectSerializer(serializers.ModelSerializer):
|
110 | 112 |
|
| 113 | + polymorphic_serializers = [ |
| 114 | + {'model': models.ArtProject, 'serializer': ArtProjectSerializer}, |
| 115 | + {'model': models.ResearchProject, 'serializer': ResearchProjectSerializer}, |
| 116 | + ] |
| 117 | + |
111 | 118 | class Meta:
|
112 | 119 | model = models.Project
|
113 | 120 | exclude = ('polymorphic_ctype',)
|
114 | 121 |
|
| 122 | + def _get_actual_serializer_from_instance(self, instance): |
| 123 | + for info in self.polymorphic_serializers: |
| 124 | + if isinstance(instance, info.get('model')): |
| 125 | + actual_serializer = info.get('serializer') |
| 126 | + return actual_serializer(instance, context=self.context) |
| 127 | + |
| 128 | + @property |
| 129 | + def fields(self): |
| 130 | + _fields = BindingDict(self) |
| 131 | + for key, value in self.get_fields().items(): |
| 132 | + _fields[key] = value |
| 133 | + return _fields |
| 134 | + |
| 135 | + def get_fields(self): |
| 136 | + if self.instance is not None: |
| 137 | + if not isinstance(self.instance, QuerySet): |
| 138 | + return self._get_actual_serializer_from_instance(self.instance).get_fields() |
| 139 | + else: |
| 140 | + raise Exception("Cannot get fields from a polymorphic serializer given a queryset") |
| 141 | + return super(ProjectSerializer, self).get_fields() |
| 142 | + |
115 | 143 | def to_representation(self, instance):
|
116 | 144 | # Handle polymorphism
|
117 |
| - if isinstance(instance, models.ArtProject): |
118 |
| - return ArtProjectSerializer( |
119 |
| - instance, context=self.context).to_representation(instance) |
120 |
| - elif isinstance(instance, models.ResearchProject): |
121 |
| - return ResearchProjectSerializer( |
122 |
| - instance, context=self.context).to_representation(instance) |
123 |
| - return super(ProjectSerializer, self).to_representation(instance) |
| 145 | + return self._get_actual_serializer_from_instance(instance).to_representation(instance) |
| 146 | + |
| 147 | + def to_internal_value(self, data): |
| 148 | + data_type = data.get('type') |
| 149 | + for info in self.polymorphic_serializers: |
| 150 | + actual_serializer = info['serializer'] |
| 151 | + if data_type == utils.get_resource_type_from_serializer(actual_serializer): |
| 152 | + self.__class__ = actual_serializer |
| 153 | + return actual_serializer(data, context=self.context).to_internal_value(data) |
| 154 | + raise Exception("Could not deserialize") |
124 | 155 |
|
125 | 156 |
|
126 | 157 | class CompanySerializer(serializers.ModelSerializer):
|
|
0 commit comments