Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Drop get_iterable()
  • Loading branch information
lovelydinosaur committed Dec 10, 2014
commit 9d3810f313123ce48dea48f3c3f0f2cae5816bde
9 changes: 3 additions & 6 deletions rest_framework/relations.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def use_pk_only_optimization(self):
return False

def get_attribute(self, instance):
if self.use_pk_only_optimization():
if self.use_pk_only_optimization() and self.source_attrs:
try:
# Optimized case, return a mock object only containing the pk attribute.
instance = get_attribute(instance, self.source_attrs[:-1])
Expand All @@ -99,10 +99,6 @@ def get_attribute(self, instance):
# Standard case, return the object instance.
return get_attribute(instance, self.source_attrs)

def get_iterable(self, instance, source_attrs):
relationship = get_attribute(instance, source_attrs)
return relationship.all() if (hasattr(relationship, 'all')) else relationship

@property
def choices(self):
return dict([
Expand Down Expand Up @@ -349,7 +345,8 @@ def to_internal_value(self, data):
]

def get_attribute(self, instance):
return self.child_relation.get_iterable(instance, self.source_attrs)
relationship = get_attribute(instance, self.source_attrs)
return relationship.all() if (hasattr(relationship, 'all')) else relationship

def to_representation(self, iterable):
return [
Expand Down