41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from rest_framework import generics
|
|
|
|
from comment import models
|
|
from comment.serializers import CommentBaseSerializer
|
|
from utils.permissions import IsCommentModerator
|
|
|
|
|
|
class CommentLstView(generics.ListCreateAPIView):
|
|
"""Comment list create view."""
|
|
def get_queryset(self):
|
|
from product.models import Product
|
|
from establishment.models import Establishment
|
|
|
|
allowed = {
|
|
"product": Product.__name__.lower(),
|
|
"establishment": Establishment.__name__.lower()
|
|
}
|
|
|
|
qs = models.Comment.objects.with_base_related()
|
|
|
|
if "object" in self.kwargs:
|
|
qs = qs.by_object_id(self.kwargs["object"])
|
|
|
|
if "type" in self.kwargs and self.kwargs["type"] in allowed:
|
|
model = allowed[self.kwargs["type"]]
|
|
qs = qs.by_content_type(self.kwargs["type"], model)
|
|
|
|
return qs.order_by('-created')
|
|
|
|
serializer_class = CommentBaseSerializer
|
|
# permission_classes = [permissions.IsAuthenticatedOrReadOnly| IsCommentModerator|IsCountryAdmin]
|
|
|
|
|
|
class CommentRUDView(generics.RetrieveUpdateDestroyAPIView):
|
|
"""Comment RUD view."""
|
|
serializer_class = CommentBaseSerializer
|
|
queryset = models.Comment.objects.all()
|
|
permission_classes = [IsCommentModerator]
|
|
# permission_classes = [IsCountryAdmin | IsCommentModerator]
|
|
lookup_field = 'id'
|