24 lines
723 B
Python
24 lines
723 B
Python
from .models import Bookmark
|
|
from django.urls import reverse_lazy
|
|
from django.views.generic.list import ListView
|
|
from django.views.generic import CreateView
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
|
|
from .models import Bookmark
|
|
|
|
class BookmarkListView(LoginRequiredMixin, ListView):
|
|
model = Bookmark
|
|
paginate_by = 20
|
|
|
|
def get_queryset(self):
|
|
return Bookmark.objects.filter(user_id=self.request.user.id)
|
|
|
|
class BookmarkCreate(LoginRequiredMixin, CreateView):
|
|
model = Bookmark
|
|
fields = ['title','url']
|
|
success_url = reverse_lazy('bookmarks:bookmark-list')
|
|
|
|
def form_valid(self, form):
|
|
form.instance.user = self.request.user
|
|
return super().form_valid(form)
|