Gracias @astevanovic apuntando en la dirección correcta.
Encontré que reemplazar el método dehydrate
para procesar solo algunos campos especificados es un poco menos tedioso que anular el método full_hydrate
para omitir campos.
En la búsqueda de la reutilización, se me ocurrieron los siguientes fragmentos de código. Espero que sería útil para algunos:
class BeeModelResource(ModelResource):
def dehydrate(self, bundle):
bundle = super(BeeModelResource, self).dehydrate(bundle)
bundle = self.dehydrate_partial(bundle)
return bundle
def dehydrate_partial(self, bundle):
for field_name, resource_field in self.fields.items():
if not isinstance(resource_field, RelatedField):
continue
if resource_field.full: # already dehydrated
continue
if not field_name in self._meta.partial_fields:
continue
if isinstance(resource_field, ToOneField):
fk_object = getattr(bundle.obj, resource_field.attribute)
fk_bundle = Bundle(obj=fk_object, request=bundle.request)
fk_resource = resource_field.get_related_resource(fk_object)
bundle.data[field_name] = fk_resource.dehydrate_selected(
fk_bundle, self._meta.partial_fields[field_name]).data
elif isinstance(resource_field, ToManyField):
data = []
fk_objects = getattr(bundle.obj, resource_field.attribute)
for fk_object in fk_objects.all():
fk_bundle = Bundle(obj=fk_object, request=bundle.request)
fk_resource = resource_field.get_related_resource(fk_object)
fk_bundle = fk_resource.dehydrate_selected_fields(
fk_bundle, self._meta.partial_fields[field_name])
data.append(fk_bundle.data)
bundle.data[field_name] = data
return bundle
def dehydrate_selected_fields(self, bundle, selected_field_names):
# Dehydrate each field.
for field_name, field_object in self.fields.items():
# A touch leaky but it makes URI resolution work.
# (borrowed from tastypie.resources.full_dehydrate)
if field_name in selected_field_names and not self.is_special_fields(field_name):
if getattr(field_object, 'dehydrated_type', None) == 'related':
field_object.api_name = self._meta.api_name
field_object.resource_name = self._meta.resource_name
bundle.data[field_name] = field_object.dehydrate(bundle)
bundle.data['resource_uri'] = self.get_resource_uri(bundle.obj)
bundle.data['id'] = bundle.obj.pk
return bundle
@staticmethod
def is_special_fields(field_name):
return field_name in ['resource_uri']
Con @sigmus ejemplo, los recursos se necesitan 3 modificaciones:
- tanto de recursos utilizará
BeeModuleResource
como su superclase (o, añadir dehydrate_partial
a uno recursos y dehydrate_selected
a la otra.)
- desarmar
full=True
en cualquiera de los recursos
partial_fields
añadir en el recurso Meta
el recurso unset
`` `
class ClientResource(BeeModelResource): # make BeeModelResource a super class
projects = fields.ToManyField(
'api.resources.ProjectResource', 'project_set'
) # remove full=True
class Meta:
queryset = Client.objects.all()
resource_name = 'client'
partial_fields = {'projects': ['memo', 'title']} # add partial_fields
class ProjectResource(BeeModelResource): # make BeeModelResource a super class
client = fields.ForeignKey(ClientResource, 'client', full=True)
class Meta:
queryset = Project.objects.all()
resource_name = 'project'
Gracias por la punta de dos recursos, esto me ha ayudado mucho ... :) –
@MarkShust: Estoy frente a el mismo tipo de problema, puede por favor elaborar más en cualquiera de el método que funcionó para ti? –
@NikhilAgrawal si mal no recuerdo, acabo de hacer dos recursos, uno con full = True, uno con full = False, con el sufijo full = False con Simple, así que sé qué recurso es qué. luego haz referencia al recurso apropiado. –