Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
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
13 changes: 13 additions & 0 deletions README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -216,3 +216,16 @@ To display a specific error inline use the following::
{{/each}}
{{input name="title" value=title}}

======
Mixins
======

The following mixin classes are available to use with Rest Framework
resources.

-------------------------------------------
rest_framework_ember.mixins.MultipleIDMixin
-------------------------------------------

Overrides ``get_queryset`` to filter by ``ids[]`` in URL query params.

5 changes: 4 additions & 1 deletion example/api/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,5 +58,8 @@ class EmberUserModelViewSet(viewsets.ModelViewSet):
parser_classes = (parsers.EmberJSONParser, )


class EmberDataMixinUserModelViewSet(mixins.EmberDataModelMixin, EmberUserModelViewSet):
class EmberDataMixinUserModelViewSet(mixins.MultipleIDMixin,
EmberUserModelViewSet):

queryset = auth_models.User.objects.all()

1 change: 1 addition & 0 deletions example/settings.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@

PASSWORD_HASHERS = ('django.contrib.auth.hashers.UnsaltedMD5PasswordHasher', )

MIDDLEWARE_CLASSES = ()

REST_FRAMEWORK = {
'PAGINATE_BY': 1,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,22 +7,20 @@
from django.conf import settings


class EmberDataModelMixinTests(TestBase):
class MultipleIDMixin(TestBase):
"""
Test usage with EmberDataModelMixin
Test usage with MultipleIDMixin

[<RegexURLPattern user-list ^user-viewsets/$>, <RegexURLPattern user-detail ^user-viewsets/(?P<pk>[^/]+)/$>]
[<RegexURLPattern user-list ^user-viewsets/$>,
<RegexURLPattern user-detail ^user-viewsets/(?P<pk>[^/]+)/$>]
"""
list_url = reverse_lazy('user-list')

def setUp(self):
super(EmberDataModelMixinTests, self).setUp()

def test_single_id_in_query_params(self):
"""
Ensure single ID in query params returns correct result
"""
url = '/user-mixin-viewset/?ids[]=%s' % self.miles.pk
url = '/user-mixin-viewset/?ids[]={0}'.format(self.miles.pk)
response = self.client.get(url)
self.assertEqual(response.status_code, 200)

Expand All @@ -48,7 +46,8 @@ def test_multiple_ids_in_query_params(self):
"""
Ensure multiple IDs in query params return correct result
"""
url = '/user-mixin-viewset/?ids[]=%s&ids[]=%s' % (self.miles.pk, self.john.pk)
url = '/user-mixin-viewset/?ids[]={0}&ids[]={1}'.format(
self.miles.pk, self.john.pk)
response = self.client.get(url)
self.assertEqual(response.status_code, 200)

Expand All @@ -67,5 +66,8 @@ def test_multiple_ids_in_query_params(self):
self.assertEquals(expected.get('user'), json_content.get('user'))
self.assertEquals(meta.get('count', 0), 2)
self.assertEquals(meta.get("next"), 2)
self.assertEqual('http://testserver/user-mixin-viewset/?ids%5B%5D=2&ids%5B%5D=1&page=2', meta.get("next_link"))
self.assertEqual(
'http://testserver/user-mixin-viewset/?ids%5B%5D=2&ids%5B%5D=1&page=2',
meta.get("next_link"))
self.assertEqual(meta.get("page"), 1)

7 changes: 5 additions & 2 deletions rest_framework_ember/mixins.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
"""
Class Mixins.
"""


class EmberDataModelMixin(object):
class MultipleIDMixin(object):
"""
Override get_queryset for multiple id support
"""
Expand All @@ -9,3 +11,4 @@ def get_queryset(self):
if ids:
self.queryset = self.queryset.filter(id__in=ids)
return self.queryset