|
|
|
@ -3,7 +3,7 @@ from datetime import datetime |
|
|
|
from django.contrib.auth import get_user_model |
|
|
|
from django.contrib.auth import get_user_model |
|
|
|
|
|
|
|
|
|
|
|
from rest_framework import status, views, viewsets, generics |
|
|
|
from rest_framework import status, views, viewsets, generics |
|
|
|
from rest_framework.decorators import detail_route, list_route, action |
|
|
|
from rest_framework.decorators import detail_route, list_route, action, permission_classes, authentication_classes |
|
|
|
from rest_framework.response import Response |
|
|
|
from rest_framework.response import Response |
|
|
|
|
|
|
|
|
|
|
|
from . import ExtendedModelViewSet |
|
|
|
from . import ExtendedModelViewSet |
|
|
|
@ -453,7 +453,6 @@ class CommentViewSet(ExtendedModelViewSet): |
|
|
|
class ObjectCommentsViewSet(ExtendedModelViewSet): |
|
|
|
class ObjectCommentsViewSet(ExtendedModelViewSet): |
|
|
|
queryset = Comment.objects.all() |
|
|
|
queryset = Comment.objects.all() |
|
|
|
serializer_class = CommentCreateSerializer |
|
|
|
serializer_class = CommentCreateSerializer |
|
|
|
permission_classes = (IsAuthorObjectOrAdmin,) |
|
|
|
|
|
|
|
ordering_fields = ('update_at', ) |
|
|
|
ordering_fields = ('update_at', ) |
|
|
|
|
|
|
|
|
|
|
|
def get_queryset(self): |
|
|
|
def get_queryset(self): |
|
|
|
@ -497,6 +496,7 @@ class ObjectCommentsViewSet(ExtendedModelViewSet): |
|
|
|
except Exception as e: |
|
|
|
except Exception as e: |
|
|
|
print(e) |
|
|
|
print(e) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@permission_classes((IsAuthorObjectOrAdmin,)) |
|
|
|
def perform_destroy(self, instance): |
|
|
|
def perform_destroy(self, instance): |
|
|
|
obj_type = None |
|
|
|
obj_type = None |
|
|
|
obj_id = None |
|
|
|
obj_id = None |
|
|
|
@ -516,6 +516,7 @@ class ObjectCommentsViewSet(ExtendedModelViewSet): |
|
|
|
print(e) |
|
|
|
print(e) |
|
|
|
instance.delete() |
|
|
|
instance.delete() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@authentication_classes([]) |
|
|
|
def list(self, request, *args, **kwargs): |
|
|
|
def list(self, request, *args, **kwargs): |
|
|
|
queryset = self.filter_queryset(self.get_queryset()).filter(parent__isnull=True) |
|
|
|
queryset = self.filter_queryset(self.get_queryset()).filter(parent__isnull=True) |
|
|
|
serializer = self.get_serializer(queryset, many=True) |
|
|
|
serializer = self.get_serializer(queryset, many=True) |
|
|
|
|