encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-12-01 05:54:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use a resource instance instead of a class to get the queryset

Browse Source
This commit is contained in:
Mjumbe Wawatu Poe 2012-08-29 17:58:44 -04:00
parent baab2f059f
commit a632e580e2
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
djangorestframework/mixins.py
View File

@ -514,8 +514,8 @@ class ModelMixin(object):
"""
Return the queryset for this view.
"""
return getattr(self.resource, 'queryset',
self.resource.model.objects.all())
return getattr(self._resource, 'queryset',
self._resource.model.objects.all())
def get_ordering(self):
"""

2
djangorestframework/tests/mixins.py
View File

@ -29,6 +29,7 @@ class TestModelRead(TestModelsTestCase):
request = self.req.get('/groups')
mixin = ReadModelMixin()
mixin.resource = GroupResource
mixin._resource = GroupResource(mixin)
response = mixin.get(request, id=group.id)
self.assertEquals(group.name, response.name)
@ -40,6 +41,7 @@ class TestModelRead(TestModelsTestCase):
request = self.req.get('/groups')
mixin = ReadModelMixin()
mixin.resource = GroupResource
mixin._resource = GroupResource(mixin)
self.assertRaises(ErrorResponse, mixin.get, request, id=12345)