def messages_view(request): if request.POST: form = MessageForm(request.POST, user=request.user) if form.is_valid(): form.save() form.post() return redirect(reverse('dialogs:chat', kwargs={ 'thread_id': form.get_thread_id() })) else: form = MessageForm() thread_list = Thread.objects.filter( participants=request.user).order_by("-last_message") unread_messages = 0 for thread in thread_list: thread.unread_messages_count = thread.get_user_unread_messages_count( request.user) unread_messages += thread.unread_messages_count return render(request, 'private_messages.html', { "thread_list": thread_list, 'form': form, 'unread_messages': unread_messages, })
class DialogView(TemplateView): template_name = 'dialogs/dialog.html' @method_decorator(login_required) def dispatch(self, request, *args, **kwargs): self.opponent = None self.dialog = None self.form = None if 'user_id' in kwargs: self.opponent = get_object_or_404(get_user_model(), pk=kwargs['user_id']) self.dialog = Dialog.objects.get_or_create(request.user, self.opponent) if not self.dialog: raise Http404 self.form = MessageForm(request.POST or None) return super(DialogView, self).dispatch(request, *args, **kwargs) def get_dialogs(self): qs = Dialog.objects.for_user(self.request.user).select_related('user1', 'user2').filter( last_message__isnull=False ).order_by('-last_message__created') paginator = Paginator(qs, 20) page = self.request.GET.get('dialogs-page') try: items = paginator.page(page) except PageNotAnInteger: items = paginator.page(1) except EmptyPage: items = paginator.page(paginator.num_pages) return items def get_messages(self): if not self.dialog: return paginator = Paginator(self.dialog.messages.all(), 20) page = self.request.GET.get('messages-page') try: items = paginator.page(page) except PageNotAnInteger: items = paginator.page(1) except EmptyPage: items = paginator.page(paginator.num_pages) return items def get_context_data(self, **kwargs): context = super(DialogView, self).get_context_data(**kwargs) context['dialogs'] = self.get_dialogs() context['opponent'] = self.opponent context['dialog_messages'] = self.get_messages() context['form'] = self.form return context def post(self, request, *args, **kwargs): if self.form and self.form.is_valid(): message = self.form.save(commit=False) message.dialog = self.dialog message.sender = request.user message.save() # return redirect('messages', kwargs={'user_id': self.opponent.pk}) return redirect(request.get_full_path()) return self.get(request, *args, **kwargs)
class DialogView(TemplateView): template_name = 'dialog/dialog.html' @method_decorator(login_required) def dispatch(self, request, *args, **kwargs): self.opponent = None self.dialog = None self.form = None if 'user_id' in kwargs: self.opponent = get_object_or_404(get_user_model(), pk=kwargs['user_id']) self.dialog = Dialog.objects.get_or_create(request.user, self.opponent) if not self.dialog: raise Http404 self.form = MessageForm(request.POST or None) return super(DialogView, self).dispatch(request, *args, **kwargs) def get_dialogs(self): qs = Dialog.objects.for_user(self.request.user).select_related('user1', 'user2').filter( last_message__isnull=False ).order_by('-last_message__created') paginator = Paginator(qs, 20) page = self.request.GET.get('dialogs-page') try: items = paginator.page(page) except PageNotAnInteger: items = paginator.page(1) except EmptyPage: items = paginator.page(paginator.num_pages) return items def get_messages(self): if not self.dialog: return paginator = Paginator(self.dialog.messages.select_related('sender'), 20) page = self.request.GET.get('messages-page') try: items = paginator.page(page) except PageNotAnInteger: items = paginator.page(1) except EmptyPage: items = paginator.page(paginator.num_pages) return items def get_context_data(self, **kwargs): context = super(DialogView, self).get_context_data(**kwargs) context['dialogs'] = self.get_dialogs() context['opponent'] = self.opponent context['dialog_messages'] = self.get_messages() context['form'] = self.form return context def post(self, request, *args, **kwargs): if self.form and self.form.is_valid(): message = self.form.save(commit=False) message.sender = request.user message.dialog = self.dialog message.save() return redirect(request.path) return self.get(request, *args, **kwargs)
def post(self, request, chat_id): form = MessageForm(data=request.POST) if form.is_valid(): message = form.save(commit=False) message.chat_id = chat_id message.author = request.user message.save() return redirect(reverse('messages', kwargs={'chat_id': chat_id}))