from rest_framework import generics

from comment import models
from comment.serializers import CommentBaseSerializer
from utils.permissions import IsModerator, IsCountryAdmin


class CommentLstView(generics.ListCreateAPIView):
    """Comment list create view."""
    serializer_class = CommentBaseSerializer
    queryset = models.Comment.objects.all()
    permission_classes = [
        IsModerator |
        IsCountryAdmin
    ]


class CommentRUDView(generics.RetrieveUpdateDestroyAPIView):
    """Comment RUD view."""
    serializer_class = CommentBaseSerializer
    queryset = models.Comment.objects.all()
    permission_classes = [IsModerator]
    lookup_field = 'id'