14 lines
377 B
Python
14 lines
377 B
Python
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)
|