add pos device - fix closed quotas pagination

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

@@ -1,12 +1,16 @@
 from apps.pos_device.web.api.v1.serilaizers import serializers as pos_serializer
 from apps.authentication.api.v1.api import UserViewSet
+from apps.authorization.models import UserRelations
 from rest_framework.exceptions import APIException
 from apps.pos_device import models as pos_models
 from rest_framework.response import Response
+from rest_framework.decorators import action
 from common.tools import CustomOperations
 from rest_framework import viewsets
+from django.db import transaction
 from rest_framework import status
 
+
 class ProviderCompanyViewSet(viewsets.ModelViewSet):  # noqa
     queryset = pos_models.ProviderCompany.objects.all()
     serializer_class = pos_serializer.ProviderCompanySerializer
@@ -15,21 +19,69 @@ class ProviderCompanyViewSet(viewsets.ModelViewSet):  # noqa
         """ custom create of provider client """
 
         try:
-            client = CustomOperations().custom_create(
-                request=request,
-                view=UserViewSet(),
-                data=request.data
-            )
-        except Exception as e:
-            raise APIException(detail="data is invalid", code=403)
+            # creating user & relations
+            client = UserViewSet().create(request=request)
 
-        return Response(client, status=status.HTTP_201_CREATED)
+            if client.status_code == 201:
+
+                # create provider
+                serializer = self.serializer_class(data=request.data['provider'])
+                if serializer.is_valid():
+                    provider = serializer.save()
+                    provider.user_relation = UserRelations.objects.get(
+                        user_id=client.data['id']
+                    )
+                    provider.save()
+
+                    return Response(serializer.data, status=status.HTTP_201_CREATED)
+                return Response(serializer.errors, status=status.HTTP_403_FORBIDDEN)
+            else:
+                return client
+        except Exception as e:
+            raise e
 
 
 class DeviceViewSet(viewsets.ModelViewSet):
     queryset = pos_models.Device.objects.all()
     serializer_class = pos_serializer.DeviceSerializer
 
+    def create(self, request, *args, **kwargs):
+        """ Custom create of pos devices """
+
+        company = pos_models.ProviderCompany.objects.get(
+            user_relation__user=request.user
+        )
+
+        # create device
+        request.data.update({'company': company.id})
+        serializer = self.serializer_class(data=request.data)
+        if serializer.is_valid():
+            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_devices',
+        url_path='my_devices',
+        name='my_devices'
+    )
+    @transaction.atomic
+    def my_devices(self, request):
+        """ list of company devices """
+
+        devices = self.queryset.filter(
+            company__user_relation__user=request.user
+        )
+
+        # paginate devices
+        page = self.paginate_queryset(devices)
+        if page is not None:
+            serializer = self.get_serializer(page, many=True)
+            return self.get_paginated_response(serializer.data)
+
 
 class DeviceVersionViewSet(viewsets.ModelViewSet):
     queryset = pos_models.DeviceVersion.objects.all()