haha

server/api/scripts.py

@@ -9,8 +9,8 @@
 """
 from rest_framework import status
 from rest_framework.decorators import api_view
-from rest_framework.response import Response
 
+from server.core.response import api_success, api_error
 from server.models import ChatScript
 from server.serializers import ChatScriptSerializer
 
@@ -25,12 +25,12 @@ def script_list(request):
             qs = qs.filter(position=position)
         if script_type:
             qs = qs.filter(script_type=script_type)
-        return Response(ChatScriptSerializer(qs, many=True).data)
+        return api_success(ChatScriptSerializer(qs, many=True).data)
 
     ser = ChatScriptSerializer(data=request.data)
     ser.is_valid(raise_exception=True)
     ser.save()
-    return Response(ser.data, status=status.HTTP_201_CREATED)
+    return api_success(ser.data, http_status=status.HTTP_201_CREATED)
 
 
 @api_view(["GET", "PUT", "DELETE"])
@@ -38,16 +38,16 @@ def script_detail(request, pk):
     try:
         obj = ChatScript.objects.get(pk=pk)
     except ChatScript.DoesNotExist:
-        return Response({"detail": "话术不存在"}, status=status.HTTP_404_NOT_FOUND)
+        return api_error(status.HTTP_404_NOT_FOUND, "话术不存在")
 
     if request.method == "GET":
-        return Response(ChatScriptSerializer(obj).data)
+        return api_success(ChatScriptSerializer(obj).data)
 
     if request.method == "PUT":
         ser = ChatScriptSerializer(obj, data=request.data, partial=True)
         ser.is_valid(raise_exception=True)
         ser.save()
-        return Response(ser.data)
+        return api_success(ser.data)
 
     obj.delete()
-    return Response({"message": "话术已删除"})
+    return api_success(msg="话术已删除")