Source code for api.viewsets

from rest_framework import viewsets
from rest_framework.permissions import IsAuthenticated

from api import permissions


[docs]class OwnerViewSet(viewsets.ModelViewSet):
    """
    A simple ViewSet for objects filtered by their 'owner' attribute.

    To use it, at minimum you'll need to provide the `serializer_class` attribute and
    the `model` attribute shortcut.
    """
    permission_classes = [IsAuthenticated, permissions.IsOwner]

[docs]    def get_queryset(self):
        return self.model.objects.filter(owner=self.request.user)

[docs]    def perform_create(self, serializer):
        obj = serializer.save(owner=self.request.user)
        self.post_save(obj)

[docs]    def post_save(self, obj):
        """A post_save hook for performing actions after the object has been pushed to the
        database.

        Leave it up to child classes to implement."""
        pass