diff --git a/example/api/resources/identity.py b/example/api/resources/identity.py index 3da13a3a..aa1dec24 100644 --- a/example/api/resources/identity.py +++ b/example/api/resources/identity.py @@ -1,7 +1,6 @@ from django.contrib.auth import models as auth_models from django.utils import encoding from rest_framework import generics, parsers, renderers, serializers, viewsets -from rest_framework.decorators import detail_route, list_route from rest_framework.response import Response from rest_framework_json_api import mixins, utils @@ -9,6 +8,18 @@ from ..serializers.identity import IdentitySerializer from ..serializers.post import PostSerializer +try: + from rest_framework.decorators import action + + def detail_route(**kwargs): + return action(detail=True, **kwargs) + + def list_route(**kwargs): + return action(detail=False, **kwargs) + +except ImportError: + from rest_framework.decorators import detail_route, list_route + class Identity(mixins.MultipleIDMixin, viewsets.ModelViewSet): queryset = auth_models.User.objects.all().order_by('pk')