device identity - device assignment - fix active quotas list bug

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

@@ -1,4 +1,4 @@
-from apps.pos_device.web.api.v1.serilaizers import serializers as pos_serializer
+from apps.pos_device.web.api.v1.serilaizers import client as client_serializer
 from apps.pos_device import models as pos_models
 from rest_framework.response import Response
 from common.tools import CustomOperations
@@ -8,7 +8,7 @@ from rest_framework import status
 
 class POSClientViewSet(viewsets.ModelViewSet):
     queryset = pos_models.POSClient.objects.all()
-    serializer_class = pos_serializer.POSClientSerializer
+    serializer_class = client_serializer.POSClientSerializer
 
     def create(self, request, *args, **kwargs):
         """ Custom create of pos client """
@@ -61,9 +61,9 @@ class POSClientViewSet(viewsets.ModelViewSet):
 
 class POSClientAttributeViewSet(viewsets.ModelViewSet):
     queryset = pos_models.POSClientAttribute.objects.all()
-    serializer_class = pos_serializer.POSClientAttributeSerializer
+    serializer_class = client_serializer.POSClientAttributeSerializer
 
 
 class POSClientAttributeValueViewSet(viewsets.ModelViewSet):
     queryset = pos_models.POSClientAttributeValue.objects.all()
-    serializer_class = pos_serializer.POSClientAttributeValueSerializer
+    serializer_class = client_serializer.POSClientAttributeValueSerializer

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

@@ -1,4 +1,4 @@
-from apps.pos_device.web.api.v1.serilaizers import serializers as pos_serializer
+from apps.pos_device.web.api.v1.serilaizers import device as device_serializer
 from apps.authentication.api.v1.api import UserViewSet
 from apps.authorization.models import UserRelations
 from rest_framework.exceptions import APIException
@@ -13,7 +13,7 @@ from rest_framework import status
 
 class ProviderCompanyViewSet(viewsets.ModelViewSet):  # noqa
     queryset = pos_models.ProviderCompany.objects.all()
-    serializer_class = pos_serializer.ProviderCompanySerializer
+    serializer_class = device_serializer.ProviderCompanySerializer
 
     def create(self, request, *args, **kwargs):
         """ custom create of provider client """
@@ -43,7 +43,7 @@ class ProviderCompanyViewSet(viewsets.ModelViewSet):  # noqa
 
 class DeviceViewSet(viewsets.ModelViewSet):
     queryset = pos_models.Device.objects.all()
-    serializer_class = pos_serializer.DeviceSerializer
+    serializer_class = device_serializer.DeviceSerializer
 
     def create(self, request, *args, **kwargs):
         """ Custom create of pos devices """
@@ -85,9 +85,58 @@ class DeviceViewSet(viewsets.ModelViewSet):
 
 class DeviceVersionViewSet(viewsets.ModelViewSet):
     queryset = pos_models.DeviceVersion.objects.all()
-    serializer_class = pos_serializer.DeviceVersionSerializer
+    serializer_class = device_serializer.DeviceVersionSerializer
 
 
 class SessionViewSet(viewsets.ModelViewSet):  # noqa
     queryset = pos_models.Sessions.objects.all()
-    serializer_class = pos_serializer.SessionSerializer
+    serializer_class = device_serializer.SessionSerializer
+
+
+class DeviceAssignmentViewSet(viewsets.ModelViewSet):
+    queryset = pos_models.DeviceAssignment.objects.all()
+    serializer_class = device_serializer.DeviceAssignmentSerializer
+
+    @action(
+        methods=['post'],
+        detail=False,
+        url_path='assignment',
+        url_name='assignment',
+        name='assignment',
+    )
+    @transaction.atomic
+    def device_assignment(self, request):
+        """ assign pos device to client by company """
+
+        company = pos_models.ProviderCompany.objects.get(
+            user_relation__user=request.user
+        )
+        request.data.update({'company': company.id})
+
+        # create assignment
+        serializer = self.serializer_class(data=request.data)
+        if serializer.is_valid():
+            assignment = 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_name='my_assignment',
+        url_path='my_assignments',
+        name='my_assignment'
+    )
+    def my_assignment(self, request):
+        """ list of company device assignment to clients """
+
+        company = pos_models.ProviderCompany.objects.get(
+            user_relation__user=request.user
+        )
+
+        # get device assignment
+        assignments = self.queryset.filter(company=company)
+
+        serializer = self.serializer_class(assignments, many=True)
+        return Response