stake holders sharing & quota distribution to CMP - list of stake holder sharings

apps/pos_device/web/api/v1/viewsets/device.py

@@ -1,7 +1,7 @@
 import random
 import string
 from datetime import timedelta
-
+from apps.product.web.api.v1.viewsets.quota_distribution_api import QuotaDistributionViewSet
 from apps.pos_device.web.api.v1.serilaizers import device as device_serializer
 from apps.authentication.exceptions import OrganizationBankAccountException
 from apps.authorization.api.v1.serializers import UserRelationSerializer
@@ -337,3 +337,102 @@ class StakeHoldersViewSet(viewsets.ModelViewSet, DynamicSearchMixin, SoftDeleteM
             serializer = self.get_serializer(page, many=True)
             return self.get_paginated_response(serializer.data)
 
+    @action(
+        methods=['get'],
+        detail=False,
+        url_path='list_by_organization',
+        url_name='list_by_organization',
+        name='list_by_organization',
+    )
+    def list_by_organization(self, request):
+        """ list of stakeholders by organization """
+
+        org = get_organization_by_user(request.user)
+
+        stakeholders = self.queryset.filter(
+            assignment__client__organization=org,
+            organization__type__key='CMP'
+        )
+
+        # paginate stakeholders
+        page = self.paginate_queryset(stakeholders)
+        if page is not None:
+            serializer = self.get_serializer(page, many=True)
+            return self.get_paginated_response(serializer.data)
+
+
+class StakeHolderShareAmountViewSet(viewsets.ModelViewSet, DynamicSearchMixin, SoftDeleteMixin):
+    queryset = pos_models.StakeHolderShareAmount.objects.select_related('quota_distribution', 'stakeholders')
+    serializer_class = device_serializer.StakeHolderShareAmountSerializer
+
+    @transaction.atomic
+    def create(self, request, *args, **kwargs):
+        """ create share amount for company stakeholders """
+
+        data = request.data.copy()
+
+        organization = get_organization_by_user(request.user)
+        data.update({'registering_organization': organization.id})
+
+        assigner_organization = get_organization_by_user(request.user)
+        data['distribution'].update({'assigner_organization': assigner_organization.id})
+
+        # create distribution
+        if 'distribution' in data.keys():
+            distribution = CustomOperations().custom_create(
+                request=request,
+                view=QuotaDistributionViewSet(),
+                data=data['distribution']
+            )
+            data.update({'quota_distribution': distribution['id']})
+
+        serializer = self.serializer_class(data=data)
+        if serializer.is_valid(raise_exception=True):
+            serializer.save()
+
+            return Response(serializer.data, status=status.HTTP_201_CREATED)
+        return Response(serializer.errors, status=status.HTTP_403_FORBIDDEN)
+
+    @action(
+        methods=['get'],
+        detail=False,
+        url_path='my_sharing_distributes',
+        url_name='my_sharing_distributes',
+        name='my_sharing_distributes',
+    )
+    def my_shared_distributes(self, request):
+        """ list of my shared stakeholders with detail """
+
+        organization = get_organization_by_user(request.user)
+
+        stakeholders_sharing = self.queryset.filter(
+            registering_organization=organization
+        ).order_by('-create_date')
+
+        # paginate stakeholders
+        page = self.paginate_queryset(stakeholders_sharing)
+        if page is not None:
+            serializer = self.get_serializer(page, many=True)
+            return self.get_paginated_response(serializer.data)
+
+    @action(
+        methods=['get'],
+        detail=True,
+        url_path='shared_by_distribution',
+        url_name='shared_by_distribution',
+        name='shared_by_distribution',
+    )
+    @transaction.atomic
+    def shared_by_distribution(self, request, pk=None):
+        """ list of shared stakeholder with distribution """
+
+        stakeholder_sharing = self.queryset.filter(
+            quota_distribution_id=pk
+        ).order_by('-create_date')
+
+        # paginate stakeholders
+        page = self.paginate_queryset(stakeholder_sharing)
+        if page is not None:
+            serializer = self.get_serializer(page, many=True)
+            return self.get_paginated_response(serializer.data)
+