From 1de7ad09ec4918a895e6c7987a9a86562d3822f2 Mon Sep 17 00:00:00 2001 From: Nathan Chapman Date: Fri, 22 Jul 2022 14:45:25 -0600 Subject: [PATCH] Add exclude so the current user doesn't see their name --- src/core/views.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/core/views.py b/src/core/views.py index c0aafae..bdf5828 100644 --- a/src/core/views.py +++ b/src/core/views.py @@ -13,6 +13,7 @@ from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin from django.contrib.messages.views import SuccessMessageMixin from django.contrib.contenttypes.models import ContentType +from accounts.models import User from .models import Topic, Post, Comment from .forms import PostForm, PostCreateForm, CommentCreateForm @@ -127,6 +128,15 @@ class PostDetailView(LoginRequiredMixin, DetailView): model = Post pk_url_kwarg = 'post_pk' + def get_queryset(self): + queryset = Post.objects.all().prefetch_related( + models.Prefetch( + 'recipients', + queryset=User.objects.exclude(pk=self.request.user.pk) + ) + ) + return queryset + def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) object_content_type = ContentType.objects.get_for_model(self.object).pk