from rest_framework import viewsets

from . import models
from bookmarks.models import Bookmark

class BookmarkViewSet(viewsets.ModelViewSet):
    serializer_class = models.BookmarkSerializer

    def perform_create(self, serializer):
        serializer.save(user = self.request.user)

    def get_queryset(self):
        return Bookmark.objects.filter(user=self.request.user)