1 

     2 import json

     3 

     4 from django.contrib.auth import get_user_model

     5 import requests

     6 from rest_framework import serializers, status, permissions, parsers, renderers

     7 from rest_framework.authtoken.models import Token

     8 from rest_framework.response import Response

     9 from rest_framework.views import APIView

    10 

    11 from ammico.views import populateVisit

    12 from django.conf import settings

    13 

    14 

    15 class UserSerializer(serializers.ModelSerializer):

    16     class Meta:

    17         model = get_user_model()

    18 

    19 class User(APIView):

    20     """

    21     get list user or add user

    22     """

    23     permission_classes = (permissions.AllowAny,)

    24 

    25     def get(self, request):

    26         user = get_user_model().objects.all()

    27         serializer = UserSerializer(user, many=True)

    28         return Response(serializer.data)

    29 

    30     def post(self, request):

    31         VALID_USER_FIELDS = [f.name for f in get_user_model()._meta.fields]

    32         serialized = UserSerializer(data=request.data)

    33 

    34         if serialized.is_valid():

    35             user_data = {field: data for (field, data) in request.DATA.items() if field in VALID_USER_FIELDS}

    36 

    37             params = {'o': 'user', 'f': 'get', 'mail': user_data['email']}

    38             r = requests.get(settings.URL_JAMESPOT, params=params)

    39             infoUser = json.loads(r.content.decode('utf-8'))

    40             if ('idUser' in infoUser['VAL']):

    41                 user_data.update({"idUser":infoUser['VAL']['idUser']})

    42             user = get_user_model().objects.create_user(

    43                 **user_data

    44             )

    45             return Response(UserSerializer(instance=user).data, status=status.HTTP_201_CREATED)

    46         else:

    47             return Response(serialized._errors, status=status.HTTP_400_BAD_REQUEST)

    48 

    49 class ObtainAuthToken(APIView):

    50     throttle_classes = ()

    51     permission_classes = ()

    52     parser_classes = (parsers.FormParser, parsers.MultiPartParser, parsers.JSONParser,)

    53     renderer_classes = (renderers.JSONRenderer,)

    54 

    55     def post(self, request):

    56         user = get_user_model().objects.get(email = request.data['email'], password = request.data['password'])

    57         if (user.idUser):

    58             populateVisit(user)

    59         token, _ = Token.objects.get_or_create(user=user)

    60         return Response({'token': token.key})