diff --git a/judge/comments.py b/judge/comments.py index 405b2ce..2a2f515 100755 --- a/judge/comments.py +++ b/judge/comments.py @@ -11,6 +11,7 @@ from django.http import ( HttpResponseForbidden, HttpResponseNotFound, HttpResponseRedirect, + Http404, ) from django.urls import reverse_lazy from django.utils.decorators import method_decorator @@ -154,7 +155,10 @@ class CommentedDetailView(TemplateResponseMixin, SingleObjectMixin, View): pre_query = None if "comment-id" in request.GET: comment_id = int(request.GET["comment-id"]) - comment_obj = Comment.objects.get(pk=comment_id) + try: + comment_obj = Comment.objects.get(pk=comment_id) + except Comment.DoesNotExist: + raise Http404 pre_query = comment_obj while comment_obj is not None: pre_query = comment_obj @@ -174,7 +178,9 @@ class CommentedDetailView(TemplateResponseMixin, SingleObjectMixin, View): queryset = queryset.filter(parent=None, hidden=False) queryset_all = None comment_count = len(queryset) + context["comment_remove"] = -1 if (pre_query != None): + comment_remove = pre_query.id queryset_all = pre_query.get_descendants(include_self=True) queryset_all = ( queryset_all.select_related("author__user") @@ -182,6 +188,7 @@ class CommentedDetailView(TemplateResponseMixin, SingleObjectMixin, View): .defer("author__about") .annotate(revisions=Count("versions", distinct=True)) ) + context["comment_remove"] = comment_remove else: queryset = ( queryset.select_related("author__user") @@ -227,7 +234,7 @@ class CommentedDetailView(TemplateResponseMixin, SingleObjectMixin, View): context["comment_root_id"] = 0 context["comment_parrent_none"] = 1 if (pre_query != None): - context["offset"] = 0 + context["offset"] = 1 else: context["offset"] = 10 diff --git a/judge/views/comment.py b/judge/views/comment.py index 976abd2..abacb52 100755 --- a/judge/views/comment.py +++ b/judge/views/comment.py @@ -123,8 +123,11 @@ def get_comments(request, limit=10): if comment_id and not Comment.objects.filter(id=comment_id).exists(): raise Http404() offset = 0 - if "offset" in request.GET: + if "offset" in request.GET: offset = int(request.GET["offset"]) + comment_remove = -1 + if "comment_remove" in request.GET: + comment_remove = int(request.GET["comment_remove"]) comment_root_id = 0 if (comment_id): comment_obj = Comment.objects.get(pk=comment_id) @@ -134,6 +137,8 @@ def get_comments(request, limit=10): queryset = comment_obj.linked_object.comments if parrent_none: queryset = queryset.filter(parent=None, hidden=False) + if (comment_remove != -1): + queryset.get(pk=comment_remove).delete() else: queryset = queryset.filter(parent=comment_obj, hidden=False) comment_count = len(queryset) @@ -165,6 +170,7 @@ def get_comments(request, limit=10): "limit": limit, "comment_count": comment_count, "comment_parrent_none": parrent_none, + "comment_remove": comment_remove, } ) diff --git a/templates/comments/content-list.html b/templates/comments/content-list.html index 711e632..2ab123e 100755 --- a/templates/comments/content-list.html +++ b/templates/comments/content-list.html @@ -1,3 +1,5 @@ +{% set logged_in = request.user.is_authenticated %} +{% set profile = request.profile if logged_in else None %} {% for node in mptt_tree(comment_list) recursive %}