change: change api

core/apps/finance/views/income_chat.py

@@ -3,7 +3,7 @@ from django.shortcuts import get_object_or_404
 from rest_framework import generics
 from rest_framework.response import Response
 
-from core.apps.finance.models import IncomeMessage, IncomeChat
+from core.apps.finance.models import IncomeMessage, IncomeChat, Income
 from core.apps.finance.serializers import income_chat as serializers
 from core.apps.accounts.permissions.permissions import HasRolePermission
 
@@ -35,9 +35,11 @@ class IncomeMessageListApiView(generics.GenericAPIView):
     serializer_class = serializers.IncomeMessageListSerializer
     queryset = IncomeMessage.objects.select_related('user')
     permission_classes = [HasRolePermission]
+    pagination_class = None
 
-    def get(self, request, chat_id):
-        chat = get_object_or_404(IncomeChat, id=chat_id)
+    def get(self, request, id):
+        income = get_object_or_404(Income, id=id)
+        chat, created = IncomeChat.objects.get_or_create(income=income)
         messages = self.queryset.filter(chat=chat)
         serializer = self.serializer_class(messages, many=True)
         return Response(