heart

backend/community/views.py

@@ -284,6 +284,22 @@ class TopicViewSet(viewsets.ModelViewSet):
             return Response({'error': '请先登录'}, status=401)
         return super().create(request, *args, **kwargs)
 
+    @action(detail=True, methods=['post'])
+    def like(self, request, pk=None):
+        obj = self.get_object()
+        user = get_current_wechat_user(request)
+        if not user:
+             return Response({'error': '请先登录'}, status=401)
+        
+        if obj.likes.filter(id=user.id).exists():
+            obj.likes.remove(user)
+            liked = False
+        else:
+            obj.likes.add(user)
+            liked = True
+            
+        return Response({'liked': liked, 'count': obj.likes.count()})
+
     def retrieve(self, request, *args, **kwargs):
         instance = self.get_object()
         instance.view_count += 1
@@ -310,6 +326,22 @@ class ReplyViewSet(viewsets.ModelViewSet):
             return Response({'error': '请先登录'}, status=401)
         return super().create(request, *args, **kwargs)
 
+    @action(detail=True, methods=['post'])
+    def like(self, request, pk=None):
+        obj = self.get_object()
+        user = get_current_wechat_user(request)
+        if not user:
+             return Response({'error': '请先登录'}, status=401)
+        
+        if obj.likes.filter(id=user.id).exists():
+            obj.likes.remove(user)
+            liked = False
+        else:
+            obj.likes.add(user)
+            liked = True
+            
+        return Response({'liked': liked, 'count': obj.likes.count()})
+
 import requests
 
 class TopicMediaViewSet(viewsets.ViewSet):