143 lines
5.0 KiB
Python
143 lines
5.0 KiB
Python
from django.db.models import Sum
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from rest_framework import generics, views, filters
|
|
from rest_framework.response import Response
|
|
from django_filters.rest_framework.backends import DjangoFilterBackend
|
|
|
|
from core.apps.accounts.permissions.permissions import HasRolePermission
|
|
from core.apps.shared.paginations.custom import CustomPageNumberPagination
|
|
from core.apps.counterparty.models import Counterparty, CounterpartyFolder
|
|
from core.apps.counterparty.serializers import counterparty as serializers
|
|
from core.apps.counterparty.filters.counterparty import CounterpartyFilter
|
|
|
|
|
|
class CounterpartyListApiView(generics.ListAPIView):
|
|
serializer_class = serializers.CounterpartyListSerializer
|
|
queryset = Counterparty.objects.exclude(is_archived=True).exclude(folder__isnull=False)
|
|
pagination_class = [HasRolePermission]
|
|
required_permissions = []
|
|
pagination_class = CustomPageNumberPagination
|
|
filter_backends = [DjangoFilterBackend, filters.SearchFilter]
|
|
filterset_class = CounterpartyFilter
|
|
search_fields = [
|
|
'name', 'inn'
|
|
]
|
|
|
|
|
|
class CounterpartyCreateApiView(generics.GenericAPIView):
|
|
serializer_class = serializers.CounterpartyCreateSerializer
|
|
queryset = Counterparty.objects.all()
|
|
permission_classes = [HasRolePermission]
|
|
required_permissions = []
|
|
|
|
def post(self, request):
|
|
serializer = self.serializer_class(data=request.data)
|
|
if serializer.is_valid(raise_exception=True):
|
|
serializer.save()
|
|
return Response(
|
|
{'success': True, 'message': 'Conterparty Created'},
|
|
status=201
|
|
)
|
|
return Response(
|
|
{'success': False, 'message': serializer.errors},
|
|
status=400
|
|
)
|
|
|
|
|
|
class ArchiveCounterpartyApiView(views.APIView):
|
|
permission_classes = [HasRolePermission]
|
|
required_permissions = []
|
|
|
|
def get(self, request, id):
|
|
counterparty = get_object_or_404(Counterparty, id=id)
|
|
counterparty.is_archived = True
|
|
counterparty.save()
|
|
return Response(
|
|
{'success': True, 'message': 'counterparty archived'},
|
|
status=200
|
|
)
|
|
|
|
|
|
class ArchivedCounterpartyListApiView(generics.ListAPIView):
|
|
serializer_class = serializers.CounterpartyListSerializer
|
|
queryset = Counterparty.objects.exclude(is_archived=False)
|
|
pagination_class = [HasRolePermission]
|
|
required_permissions = []
|
|
pagination_class = CustomPageNumberPagination
|
|
|
|
|
|
class CounterpartyDeleteApiView(views.APIView):
|
|
permission_classes = [HasRolePermission]
|
|
required_permissions = []
|
|
|
|
def delete(self, request, id):
|
|
counterparty = get_object_or_404(Counterparty, id=id)
|
|
counterparty.delete()
|
|
return Response(
|
|
{'success': True, 'message': 'counterparty deleted'},
|
|
status=204
|
|
)
|
|
|
|
|
|
class CounterpartyUpdateApiView(generics.UpdateAPIView):
|
|
permission_classes = [HasRolePermission]
|
|
required_permissions = []
|
|
lookup_field = 'id'
|
|
serializer_class = serializers.CounterpartyUpdateSerializer
|
|
queryset = Counterparty.objects.all()
|
|
|
|
|
|
class FolderCounterpartyListApiView(generics.GenericAPIView):
|
|
serializer_class = serializers.CounterpartyListSerializer
|
|
queryset = Counterparty.objects.exclude(is_archived=True)
|
|
permission_classes = [HasRolePermission]
|
|
filter_backends = [filters.SearchFilter]
|
|
search_fields = [
|
|
'name', 'inn'
|
|
]
|
|
|
|
def get(self, reuqest, folder_id):
|
|
folder = get_object_or_404(CounterpartyFolder, id=folder_id)
|
|
queryset = self.queryset.filter(folder=folder).exclude(folder__isnull=True)
|
|
page = self.paginate_queryset(self.filter_queryset(queryset))
|
|
if page is not None:
|
|
serializer = self.serializer_class(page, many=True)
|
|
return self.get_paginated_response(serializer.data)
|
|
|
|
|
|
class CounterpartyStatisticsApiView(views.APIView):
|
|
permission_classes = [HasRolePermission]
|
|
|
|
def get(self, request):
|
|
counterparty_ids = request.query_params.getlist('counterparty')
|
|
if counterparty_ids:
|
|
queryset = Counterparty.objects.filter(id__in=counterparty_ids)
|
|
queryset = Counterparty.objects.all()
|
|
|
|
res = queryset.aggregate(
|
|
kredit_usd=Sum('kredit_usd'),
|
|
kredit_uzs=Sum('kredit_uzs'),
|
|
total_kredit=Sum('total_kredit'),
|
|
debit_usd=Sum('debit_usd'),
|
|
debit_uzs=Sum('debit_uzs'),
|
|
total_debut=Sum('total_debit'),
|
|
)
|
|
return Response(res)
|
|
|
|
|
|
class CounterpartiesApiView(generics.GenericAPIView):
|
|
serializer_class = serializers.CounterpartyListSerializer
|
|
queryset = Counterparty.objects.all()
|
|
permission_classes = [HasRolePermission]
|
|
filter_backends = [filters.SearchFilter]
|
|
search_fields = [
|
|
'name', 'inn'
|
|
]
|
|
|
|
def get(self, request):
|
|
page = self.paginate_queryset(self.filter_queryset(self.queryset))
|
|
if page is not None:
|
|
ser = self.serializer_class(page, many=True)
|
|
return self.get_paginated_response(ser.data)
|
|
|