def handle_related_field(self, resource, field, field_name, request):
links = self.dict_class()
linked_ids = self.dict_class()
related_field = get_related_field(field)
model = self.model_from_obj(related_field)
resource_type = self.model_to_resource_type(model)
if field_name in resource:
links[field_name] = {
"type": resource_type,
}
if is_related_many(field):
link_data = [
encoding.force_text(pk) for pk in resource[field_name]]
elif resource[field_name]:
link_data = encoding.force_text(resource[field_name])
else:
link_data = None
linked_ids[field_name] = link_data
return {"linked_ids": linked_ids, "links": links}
def handle_nested_serializer(self, resource, field, field_name, request):
serializer_field = get_related_field(field)
if hasattr(serializer_field, "opts"):
model = serializer_field.opts.model
else:
model = serializer_field.Meta.model
resource_type = self.model_to_resource_type(model)
linked_ids = self.dict_class()
links = self.dict_class()
linked = self.dict_class()
linked[resource_type] = []
if is_related_many(field):
items = resource[field_name]
else:
items = [resource[field_name]]
obj_ids = []
resource.serializer = serializer_field
for item in items:
converted = self.convert_resource(item, resource, request)
linked_obj = converted["data"]
linked_ids = converted.pop("linked_ids", {})
if linked_ids:
linked_obj["links"] = linked_ids
obj_ids.append(converted["data"]["id"])
field_links = self.prepend_links_with_name(
converted.get("links", {}), resource_type)
field_links[field_name] = {
"type": resource_type,
}
if "href" in converted["data"]:
url_field_name = api_settings.URL_FIELD_NAME
url_field = serializer_field.fields[url_field_name]
field_links[field_name]["href"] = self.url_to_template(
url_field.view_name, request, field_name,
)
links.update(field_links)
linked[resource_type].append(linked_obj)
if is_related_many(field):
linked_ids[field_name] = obj_ids
else:
linked_ids[field_name] = obj_ids[0]
return {"linked_ids": linked_ids, "links": links, "linked": linked}
def convert_resource(self, resource, view):
serializer_data = view.get_serializer(instance=None)
fields = serializer_data.fields
links = {}
if "links" in resource:
links = resource["links"]
del resource["links"]
for field_name, field in six.iteritems(fields):
if field_name not in links:
continue
related_field = get_related_field(field)
if isinstance(related_field, relations.HyperlinkedRelatedField):
if is_related_many(field):
pks = links[field_name]
model = related_field.queryset.model
resource[field_name] = []
for pk in pks:
obj = model(pk=pk)
try:
url = related_field.to_representation(obj)
except AttributeError:
url = related_field.to_native(obj)
resource[field_name].append(url)
else:
pk = links[field_name]
model = related_field.queryset.model
obj = model(pk=pk)
try:
url = related_field.to_representation(obj)
except AttributeError:
url = related_field.to_native(obj)
resource[field_name] = url
else:
resource[field_name] = links[field_name]
return resource
def convert_resource(self, resource, view):
serializer_data = view.get_serializer(instance=None)
fields = serializer_data.fields
links = {}
if "links" in resource:
links = resource["links"]
del resource["links"]
for field_name, field in six.iteritems(fields):
if field_name not in links:
continue
related_field = get_related_field(field)
if isinstance(related_field, relations.HyperlinkedRelatedField):
if is_related_many(field):
pks = links[field_name]
model = related_field.queryset.model
resource[field_name] = []
for pk in pks:
obj = model(pk=pk)
try:
url = related_field.to_representation(obj)
except AttributeError:
url = related_field.to_native(obj)
resource[field_name].append(url)
else:
pk = links[field_name]
model = related_field.queryset.model
obj = model(pk=pk)
try:
url = related_field.to_representation(obj)
except AttributeError:
url = related_field.to_native(obj)
resource[field_name] = url
else:
resource[field_name] = links[field_name]
return resource
def convert_resource(self, resource, data, request):
fields = self.fields_from_resource(resource, data)
if not fields:
raise WrapperNotApplicable('Items must have a fields attribute.')
data = self.dict_class()
linked_ids = self.dict_class()
links = self.dict_class()
linked = self.dict_class()
meta = self.dict_class()
for field_name, field in six.iteritems(fields):
converted = None
if field_name in self.convert_by_name:
converter_name = self.convert_by_name[field_name]
converter = getattr(self, converter_name)
converted = converter(resource, field, field_name, request)
else:
related_field = get_related_field(field)
for field_type, converter_name in \
six.iteritems(self.convert_by_type):
if isinstance(related_field, field_type):
converter = getattr(self, converter_name)
converted = converter(
resource, field, field_name, request)
break
if converted:
data.update(converted.pop("data", {}))
linked_ids.update(converted.pop("linked_ids", {}))
links.update(converted.get("links", {}))
linked = self.update_nested(linked,
converted.get('linked', {}))
meta.update(converted.get("meta", {}))
else:
data[field_name] = resource[field_name]
return {
'data': data,
'linked_ids': linked_ids,
'links': links,
'linked': linked,
'meta': meta,
}
def convert_resource(self, resource, data, request):
fields = self.fields_from_resource(resource, data)
if not fields:
raise WrapperNotApplicable('Items must have a fields attribute.')
data = self.dict_class()
linked_ids = self.dict_class()
links = self.dict_class()
linked = self.dict_class()
meta = self.dict_class()
for field_name, field in six.iteritems(fields):
converted = None
if field_name in self.convert_by_name:
converter_name = self.convert_by_name[field_name]
converter = getattr(self, converter_name)
converted = converter(resource, field, field_name, request)
else:
related_field = get_related_field(field)
for field_type, converter_name in \
six.iteritems(self.convert_by_type):
if isinstance(related_field, field_type):
converter = getattr(self, converter_name)
converted = converter(resource, field, field_name,
request)
break
if converted:
data.update(converted.pop("data", {}))
linked_ids.update(converted.pop("linked_ids", {}))
links.update(converted.get("links", {}))
linked = self.update_nested(linked,
converted.get('linked', {}))
meta.update(converted.get("meta", {}))
else:
data[field_name] = resource[field_name]
return {
'data': data,
'linked_ids': linked_ids,
'links': links,
'linked': linked,
'meta': meta,
}
def handle_url_field(self, resource, field, field_name, request):
links = self.dict_class()
linked_ids = self.dict_class()
related_field = get_related_field(field)
model = self.model_from_obj(related_field)
resource_type = self.model_to_resource_type(model)
links[field_name] = {
"href": self.url_to_template(related_field.view_name, request, field_name),
"type": resource_type,
}
if field_name in resource:
linked_ids[field_name] = self.url_to_pk(resource[field_name], field)
return {"linked_ids": linked_ids, "links": links}
def handle_url_field(self, resource, field, field_name, request):
links = self.dict_class()
linked_ids = self.dict_class()
related_field = get_related_field(field)
model = self.model_from_obj(related_field)
resource_type = self.model_to_resource_type(model)
links[field_name] = {
"href": self.url_to_template(related_field.view_name,
request,
field_name),
"type": resource_type,
}
if field_name in resource:
linked_ids[field_name] = self.url_to_pk(
resource[field_name], field)
return {"linked_ids": linked_ids, "links": links}
