mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-12-18 14:16:17 +03:00
137 lines
6.7 KiB
Markdown
137 lines
6.7 KiB
Markdown
# Tutorial 6: ViewSets & Routers
|
|
|
|
REST framework includes an abstraction for dealing with `ViewSets`, that allows the developer to concentrate on modeling the state and interactions of the API, and leave the URL construction to be handled automatically, based on common conventions.
|
|
|
|
`ViewSet` classes are almost the same thing as `View` classes, except that they provide operations such as `retrieve`, or `update`, and not method handlers such as `get` or `put`.
|
|
|
|
A `ViewSet` class is only bound to a set of method handlers at the last moment, when it is instantiated into a set of views, typically by using a `Router` class which handles the complexities of defining the URL conf for you.
|
|
|
|
## Refactoring to use ViewSets
|
|
|
|
Let's take our current set of views, and refactor them into view sets.
|
|
|
|
First of all let's refactor our `UserList` and `UserDetail` classes into a single `UserViewSet` class. In the `snippets/views.py` file, we can remove the two view classes and replace them with a single ViewSet class:
|
|
|
|
from rest_framework import viewsets
|
|
|
|
|
|
class UserViewSet(viewsets.ReadOnlyModelViewSet):
|
|
"""
|
|
This viewset automatically provides `list` and `retrieve` actions.
|
|
"""
|
|
queryset = User.objects.all()
|
|
serializer_class = UserSerializer
|
|
|
|
Here we've used the `ReadOnlyModelViewSet` class to automatically provide the default 'read-only' operations. We're still setting the `queryset` and `serializer_class` attributes exactly as we did when we were using regular views, but we no longer need to provide the same information to two separate classes.
|
|
|
|
Next we're going to replace the `SnippetList`, `SnippetDetail` and `SnippetHighlight` view classes. We can remove the three views, and again replace them with a single class.
|
|
|
|
from rest_framework import permissions
|
|
from rest_framework import renderers
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
|
|
|
|
class SnippetViewSet(viewsets.ModelViewSet):
|
|
"""
|
|
This ViewSet automatically provides `list`, `create`, `retrieve`,
|
|
`update` and `destroy` actions.
|
|
|
|
Additionally we also provide an extra `highlight` action.
|
|
"""
|
|
queryset = Snippet.objects.all()
|
|
serializer_class = SnippetSerializer
|
|
permission_classes = [permissions.IsAuthenticatedOrReadOnly,
|
|
IsOwnerOrReadOnly]
|
|
|
|
@action(detail=True, renderer_classes=[renderers.StaticHTMLRenderer])
|
|
def highlight(self, request, *args, **kwargs):
|
|
snippet = self.get_object()
|
|
return Response(snippet.highlighted)
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.save(owner=self.request.user)
|
|
|
|
This time we've used the `ModelViewSet` class in order to get the complete set of default read and write operations.
|
|
|
|
Notice that we've also used the `@action` decorator to create a custom action, named `highlight`. This decorator can be used to add any custom endpoints that don't fit into the standard `create`/`update`/`delete` style.
|
|
|
|
Custom actions which use the `@action` decorator will respond to `GET` requests by default. We can use the `methods` argument if we wanted an action that responded to `POST` requests.
|
|
|
|
The URLs for custom actions by default depend on the method name itself. If you want to change the way url should be constructed, you can include `url_path` as a decorator keyword argument.
|
|
|
|
## Binding ViewSets to URLs explicitly
|
|
|
|
The handler methods only get bound to the actions when we define the URLConf.
|
|
To see what's going on under the hood let's first explicitly create a set of views from our ViewSets.
|
|
|
|
In the `snippets/urls.py` file we bind our `ViewSet` classes into a set of concrete views.
|
|
|
|
from rest_framework import renderers
|
|
|
|
from snippets.views import api_root, SnippetViewSet, UserViewSet
|
|
|
|
snippet_list = SnippetViewSet.as_view({
|
|
'get': 'list',
|
|
'post': 'create'
|
|
})
|
|
snippet_detail = SnippetViewSet.as_view({
|
|
'get': 'retrieve',
|
|
'put': 'update',
|
|
'patch': 'partial_update',
|
|
'delete': 'destroy'
|
|
})
|
|
snippet_highlight = SnippetViewSet.as_view({
|
|
'get': 'highlight'
|
|
}, renderer_classes=[renderers.StaticHTMLRenderer])
|
|
user_list = UserViewSet.as_view({
|
|
'get': 'list'
|
|
})
|
|
user_detail = UserViewSet.as_view({
|
|
'get': 'retrieve'
|
|
})
|
|
|
|
Notice how we're creating multiple views from each `ViewSet` class, by binding the HTTP methods to the required action for each view.
|
|
|
|
Now that we've bound our resources into concrete views, we can register the views with the URL conf as usual.
|
|
|
|
urlpatterns = format_suffix_patterns([
|
|
path('', api_root),
|
|
path('snippets/', snippet_list, name='snippet-list'),
|
|
path('snippets/<int:pk>/', snippet_detail, name='snippet-detail'),
|
|
path('snippets/<int:pk>/highlight/', snippet_highlight, name='snippet-highlight'),
|
|
path('users/', user_list, name='user-list'),
|
|
path('users/<int:pk>/', user_detail, name='user-detail')
|
|
])
|
|
|
|
## Using Routers
|
|
|
|
Because we're using `ViewSet` classes rather than `View` classes, we actually don't need to design the URL conf ourselves. The conventions for wiring up resources into views and urls can be handled automatically, using a `Router` class. All we need to do is register the appropriate view sets with a router, and let it do the rest.
|
|
|
|
Here's our re-wired `snippets/urls.py` file.
|
|
|
|
from django.urls import path, include
|
|
from rest_framework.routers import DefaultRouter
|
|
|
|
from snippets import views
|
|
|
|
# Create a router and register our ViewSets with it.
|
|
router = DefaultRouter()
|
|
router.register(r'snippets', views.SnippetViewSet, basename='snippet')
|
|
router.register(r'users', views.UserViewSet, basename='user')
|
|
|
|
# The API URLs are now determined automatically by the router.
|
|
urlpatterns = [
|
|
path('', include(router.urls)),
|
|
]
|
|
|
|
Registering the ViewSets with the router is similar to providing a urlpattern. We include two arguments - the URL prefix for the views, and the view set itself.
|
|
|
|
The `DefaultRouter` class we're using also automatically creates the API root view for us, so we can now delete the `api_root` function from our `views` module.
|
|
|
|
## Trade-offs between views vs ViewSets
|
|
|
|
Using ViewSets can be a really useful abstraction. It helps ensure that URL conventions will be consistent across your API, minimizes the amount of code you need to write, and allows you to concentrate on the interactions and representations your API provides rather than the specifics of the URL conf.
|
|
|
|
That doesn't mean it's always the right approach to take. There's a similar set of trade-offs to consider as when using class-based views instead of function-based views. Using ViewSets is less explicit than building your API views individually.
|