Fix chat box and add Delete feature

chat_box/views.py

@@ -1,6 +1,6 @@
 from django.utils.translation import gettext as _
 from django.views.generic import ListView
-from django.http import HttpResponse
+from django.http import HttpResponse, JsonResponse
 from django.core.paginator import Paginator
 
 from judge.jinja2.gravatar import gravatar
@@ -28,7 +28,7 @@ class ChatView(ListView):
     template_name = 'chat/chat.html'
     title = _('Chat Box')
     paginate_by = 50
-    paginator = Paginator(Message.objects.all(), paginate_by)
+    paginator = Paginator(Message.objects.filter(hidden=False), paginate_by)
 
     def get(self, request, *args, **kwargs):
         page = request.GET.get('page')
@@ -46,3 +46,25 @@ class ChatView(ListView):
             msg.time = format_time(msg.time)
 
         return context
+
+def delete_message(request):
+    ret = {'delete': 'done'}
+    
+    if request.method == 'GET':
+        return JsonResponse(ret)
+
+    if request.user.is_staff:
+        author = request.POST.get('author')
+        time = request.POST.get('messtime')
+        all_mess = Message.objects.all()
+        
+        for mess in all_mess:
+            if mess.author.__str__() == author and format_time(mess.time) == time:
+                mess.hidden = True
+                mess.save()
+                new_elt = {'time': format_time(mess.time), 'content': mess.body}
+                ret = new_elt
+        
+        return JsonResponse(ret)
+    
+    return JsonResponse(ret)