identity.py

fromdjango.contrib.authimportmodelsasauth_models
fromdjango.utilsimportencoding

fromrest_frameworkimportviewsets, generics, renderers, parsers, serializers
fromrest_framework.decoratorsimportlist_route, detail_route
fromrest_framework.responseimportResponse
fromrest_framework_json_apiimportmixins, utils
from ..serializers.identityimportIdentitySerializer
from ..serializers.postimportPostSerializer


classIdentity(mixins.MultipleIDMixin, viewsets.ModelViewSet):
queryset=auth_models.User.objects.all()
serializer_class=IdentitySerializer

@list_route()
defempty_list(self, request):
"""
 This is a hack/workaround to return an empty result on a list
 endpoint because the delete operation in the test_empty_pluralization
 test doesn't prevent the /identities endpoint from still returning
 records when called in the same test. Suggestions welcome.
 """
self.queryset=self.queryset.filter(pk=None)
returnsuper(Identity, self).list(request)

# demonstrate sideloading data for use at app boot time
@list_route()
defposts(self, request):
self.resource_name=False

identities=self.queryset
posts= [{'id': 1, 'title': 'Test Blog Post'}]

data= {
encoding.force_text('identities'): IdentitySerializer(identities, many=True).data,
encoding.force_text('posts'): PostSerializer(posts, many=True).data,
 }
returnResponse(utils.format_keys(data, format_type='camelize'))

@detail_route()
defmanual_resource_name(self, request, *args, **kwargs):
self.resource_name='data'
returnsuper(Identity, self).retrieve(request, args, kwargs)

@detail_route()
defvalidation(self, request, *args, **kwargs):
raiseserializers.ValidationError('Oh nohs!')


classGenericIdentity(generics.GenericAPIView):
"""
 An endpoint that uses DRF's default format so we can test that.

 GET /identities/generic
 """
serializer_class=IdentitySerializer
allowed_methods= ['GET']
renderer_classes= (renderers.JSONRenderer, )
parser_classes= (parsers.JSONParser, )


defget_queryset(self):
returnauth_models.User.objects.all()

defget(self, request, pk=None):
"""
 GET request
 """
obj=self.get_object()
returnResponse(IdentitySerializer(obj).data)