kop narsalar qoshildi

core/apps/shared/utils/response_mixin.py

@@ -0,0 +1,64 @@
+from rest_framework import status
+from rest_framework.response import Response
+
+
+class ResponseMixin:
+    """
+        Mixin to customize the response format
+
+    Example Usage:
+
+    class MyAPIView(APIView, APIViewResponseMixin):
+        def get(self, request):
+            try:
+                # Your logic here
+                data = {"key": "value"}
+                return self.success_response(data=data, message="Data retrieved successfully")
+            except Exception as e:
+                return self.error_response(message=str(e))
+
+    """
+
+    SUCCESS = "success"  # 200
+    FAILURE = "failure"  # 400
+    ERROR = "error"  # 500
+
+    @classmethod
+    def success_response(cls, data=None, message=None, status_code=status.HTTP_200_OK):
+        """
+        Returns Success Response
+        """
+        response_data = {"status_code": status_code, "status": cls.SUCCESS}
+        if message is not None:
+            response_data["message"] = message
+        if data is not None:
+            response_data["data"] = data
+        return Response(response_data, status=status_code)
+
+    @classmethod
+    def failure_response(
+        cls, data=None, message=None, status_code=status.HTTP_400_BAD_REQUEST
+    ):
+        """
+        Returns Failure Response
+        """
+        response_data = {"status_code": status_code, "status": cls.FAILURE}
+        if message is not None:
+            response_data["message"] = message
+        if data is not None:
+            response_data["data"] = data
+        return Response(response_data, status=status_code)
+
+    @classmethod
+    def error_response(
+        cls, data=None, message=None, status_code=status.HTTP_500_INTERNAL_SERVER_ERROR
+    ):
+        """
+        Returns Error Response
+        """
+        response_data = {"status_code": status_code, "status": cls.ERROR}
+        if message is not None:
+            response_data["message"] = message
+        if data is not None:
+            response_data["data"] = data
+        return Response(response_data, status=status_code)