"""Базовое представление примечания."""
from django.contrib.auth.mixins import LoginRequiredMixin
from django.urls import reverse_lazy
from ya_note.notes.models import ModelNote


class ViewNoteBase(LoginRequiredMixin):
    """Базовый класс для остальных CBV."""

    model = ModelNote
    template_name: str = ""
    success_url = reverse_lazy('notes:success')

    def get_queryset(self):
        """Пользователь может работать только со своими заметками."""
        return self.model.objects.filter(author=self.request.user)