from django.utils.translation import gettext as _ from django.views.generic import ListView from django.http import HttpResponse, JsonResponse from django.core.paginator import Paginator from django.shortcuts import render from django.forms.models import model_to_dict from django.utils import timezone from judge.jinja2.gravatar import gravatar from .models import Message, Profile import json def format_messages(messages): msg_list = [{ 'time': msg.time, 'author': msg.author, 'body': msg.body, 'image': gravatar(msg.author, 32), 'id': msg.id, 'css_class': msg.author.css_class, } for msg in messages] return json.dumps(msg_list, default=str) class ChatView(ListView): model = Message context_object_name = 'message' template_name = 'chat/chat.html' title = _('Chat Box') paginate_by = 50 paginator = Paginator(Message.objects.filter(hidden=False), paginate_by) def get_queryset(self): return Message.objects.filter(hidden=False) def get(self, request, *args, **kwargs): page = request.GET.get('page') if (page == None): # return render(request, 'chat/chat.html', {'message': format_messages(Message.objects.all())}) return super().get(request, *args, **kwargs) cur_page = self.paginator.get_page(page) return HttpResponse(format_messages(cur_page.object_list)) def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['title'] = self.title last_five_minutes = timezone.now()-timezone.timedelta(minutes=5) context['online_users'] = Profile.objects \ .filter(last_access__gte = last_five_minutes) return context def delete_message(request): ret = {'delete': 'done'} if request.method == 'GET': return JsonResponse(ret) if request.user.is_staff: messid = int(request.POST.get('messid')) all_mess = Message.objects.all() for mess in all_mess: if mess.id == messid: mess.hidden = True mess.save() new_elt = {'time': mess.time, 'content': mess.body} ret = new_elt break return JsonResponse(ret) return JsonResponse(ret)