Règle un problème d'affichage des votes dans les MPs

zestedesavoir · Aug 12, 2022 · d0d1427 · d0d1427
1 parent 5ea18c2
commit d0d1427
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/zds/mp/models.py b/zds/mp/models.py
@@ -444,11 +444,7 @@ def set_user_vote(self, user, vote):
     def get_votes(self):
         """Get the non-anonymous votes"""
         if not hasattr(self, "votes"):
-            self.votes = (
-                PrivatePostVote.objects.filter(private_post=self)
-                .select_related("user")
-                .all()
-            )
+            self.votes = PrivatePostVote.objects.filter(private_post=self).select_related("user").all()
 
         return self.votes
 

diff --git a/zds/mp/views.py b/zds/mp/views.py
@@ -24,7 +24,7 @@
 from zds.mp.utils import send_mp, send_message_mp
 from zds.utils.paginator import ZdSPagingListView
 from .forms import PrivateTopicForm, PrivatePostForm, PrivateTopicEditForm
-from .models import PrivateTopic, PrivateTopicRead, PrivatePost, mark_read, NotReachableError
+from .models import PrivateTopic, PrivateTopicRead, PrivatePost, mark_read, NotReachableError, PrivatePostVote
 
 
 class PrivateTopicList(ZdSPagingListView):
@@ -255,6 +255,10 @@ def get_context_data(self, **kwargs):
         else:
             context["user_can_modify"] = []
 
+        votes = PrivatePostVote.objects.filter(user_id=self.request.user.pk, private_post__in=context["posts"]).all()
+        context["user_like"] = [vote.private_post_id for vote in votes if vote.positive]
+        context["user_dislike"] = [vote.private_post_id for vote in votes if not vote.positive]
+
         return context
 
     def get_queryset(self):