53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
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, Income
|
|
from core.apps.finance.serializers import income_chat as serializers
|
|
from core.apps.accounts.permissions.permissions import HasRolePermission
|
|
|
|
|
|
class IncomeMessageCreateApiView(generics.GenericAPIView):
|
|
serializer_class = serializers.IncomeMessageCreateSerializer
|
|
queryset = IncomeMessage.objects.all()
|
|
permission_classes = [HasRolePermission]
|
|
|
|
def post(self, request):
|
|
serializer = self.serializer_class(data=request.data, context={'user': request.user})
|
|
if serializer.is_valid(raise_exception=True):
|
|
serializer.save()
|
|
return Response(
|
|
{
|
|
'success': True,
|
|
'message': 'created'
|
|
}, status=201
|
|
)
|
|
return Response(
|
|
{
|
|
'success': False,
|
|
'message': serializer.errors,
|
|
}, status=400
|
|
)
|
|
|
|
|
|
class IncomeMessageListApiView(generics.GenericAPIView):
|
|
serializer_class = serializers.IncomeMessageListSerializer
|
|
queryset = IncomeMessage.objects.select_related('user')
|
|
permission_classes = [HasRolePermission]
|
|
pagination_class = None
|
|
|
|
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(
|
|
{
|
|
'chat': {
|
|
'id': chat.id,
|
|
'income': chat.income.id
|
|
},
|
|
'messages': serializer.data
|
|
}, status=200
|
|
) |