146 lines
4.8 KiB
Python
146 lines
4.8 KiB
Python
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
|
|
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 = device_serializer.ProviderCompanySerializer
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
""" custom create of provider client """
|
|
|
|
try:
|
|
# creating user & relations
|
|
client = UserViewSet().create(request=request)
|
|
|
|
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 = device_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()
|
|
serializer_class = device_serializer.DeviceVersionSerializer
|
|
|
|
|
|
class SessionViewSet(viewsets.ModelViewSet): # noqa
|
|
queryset = pos_models.Sessions.objects.all()
|
|
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_assignments',
|
|
url_path='my_assignments',
|
|
name='my_assignments'
|
|
)
|
|
def my_assignment(self, request):
|
|
""" list of company device assignment to clients """
|
|
|
|
try:
|
|
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(serializer.data, status=status.HTTP_200_OK)
|
|
except Exception as e:
|
|
raise APIException('Non Object Error', code=403)
|