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)