from apps.pos_device.web.api.v1.serilaizers import serializers as pos_serializer
from apps.authentication.api.v1.api import UserViewSet
from rest_framework.exceptions import APIException
from apps.pos_device import models as pos_models
from rest_framework.response import Response
from common.tools import CustomOperations
from rest_framework import viewsets
from rest_framework import status

class ProviderCompanyViewSet(viewsets.ModelViewSet):  # noqa
    queryset = pos_models.ProviderCompany.objects.all()
    serializer_class = pos_serializer.ProviderCompanySerializer

    def create(self, request, *args, **kwargs):
        """ 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)

        return Response(client, status=status.HTTP_201_CREATED)


class DeviceViewSet(viewsets.ModelViewSet):
    queryset = pos_models.Device.objects.all()
    serializer_class = pos_serializer.DeviceSerializer


class DeviceVersionViewSet(viewsets.ModelViewSet):
    queryset = pos_models.DeviceVersion.objects.all()
    serializer_class = pos_serializer.DeviceVersionSerializer


class SessionViewSet(viewsets.ModelViewSet):  # noqa
    queryset = pos_models.Sessions.objects.all()
    serializer_class = pos_serializer.SessionSerializer