Moderation CheckUntested Code

pennlabs · Nov 5, 2024 · c890ef2 · c890ef2
1 parent aea59b7
commit c890ef2
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/backend/review/views.py b/backend/review/views.py
@@ -1011,6 +1011,15 @@ def create(self, request):
             return Response(
                 {"message": "Insufficient fields provided."}, status=status.HTTP_400_BAD_REQUEST
             )
+
+        # run moderation check 
+        text = request.data.get("text")
+        flagged, moderation_results = check_text_moderation(text)
+        if flagged:
+            return Response(
+                {"message": "Comment flagged for moderation.", "moderation_results": moderation_results},
+                status=status.HTTP_400_BAD_REQUEST
+            )
 
         # verify section is real
         try:
@@ -1052,7 +1061,17 @@ def update(self, request, pk=None):
             )
 
         if "text" in request.data:
-            comment.text = request.data.get("text")
+            # Run moderation check on the new text 
+            text = request.data.get("text")
+            flagged, moderation_results = check_text_moderation(text)
+            if flagged:
+                return Response(
+                    {"message": "Comment flagged for moderation.", "moderation_results": moderation_results},
+                    status=status.HTTP_400_BAD_REQUEST
+                )
+
+            # update and save comment
+            comment.text = text
             comment.save()
             return Response({"message": "Successfully edited."}, status=status.HTTP_201_CREATED)
         else: