server/ammicosrv/authentication/views.py
author rougeronj
Wed, 10 Jun 2015 10:57:27 +0200
changeset 149 8e117699857a
parent 143 server/src/authentication/views.py@ea633c8f9bfa
parent 135 server/src/authentication/views.py@27065f8a19d3
child 150 78e82bf8ff89
permissions -rw-r--r--
Merge with 925337e6983dc5e5a011681dafd37ce986e07a1a


import json

from django.contrib.auth import get_user_model
import requests
from rest_framework import serializers, status, permissions, parsers, renderers
from rest_framework.authtoken.models import Token
from rest_framework.response import Response
from rest_framework.views import APIView

from config import URL_JAMESPOT
from ammicosrv.ammico.views import populateVisit


#from ammico.views import populateVisit
class UserSerializer(serializers.ModelSerializer):
    class Meta:
        model = get_user_model()

class User(APIView):
    """
    get list user or add user
    """
    permission_classes = (permissions.AllowAny,)
    
    def get(self, request):
        user = get_user_model().objects.all()
        serializer = UserSerializer(user, many=True)
        return Response(serializer.data)
        
    def post(self, request):
        VALID_USER_FIELDS = [f.name for f in get_user_model()._meta.fields]
        serialized = UserSerializer(data=request.data)
        
        if serialized.is_valid():
            user_data = {field: data for (field, data) in request.DATA.items() if field in VALID_USER_FIELDS}
            
            params = {'o': 'user', 'f': 'get', 'mail': user_data['email']}
            r = requests.get(URL_JAMESPOT, params=params)
            infoUser = json.loads(r.content.decode('utf-8'))
            if ('idUser' in infoUser['VAL']):
                user_data.update({"idUser":infoUser['VAL']['idUser']})
            user = get_user_model().objects.create_user(
                **user_data
            )
            return Response(UserSerializer(instance=user).data, status=status.HTTP_201_CREATED)
        else:
            return Response(serialized._errors, status=status.HTTP_400_BAD_REQUEST)
        
class ObtainAuthToken(APIView):
    throttle_classes = ()
    permission_classes = ()
    parser_classes = (parsers.FormParser, parsers.MultiPartParser, parsers.JSONParser,)
    renderer_classes = (renderers.JSONRenderer,)

    def post(self, request):
        user = get_user_model().objects.get(email = request.data['email'], password = request.data['password'])
        if (user.idUser and 'idExpo' in request.GET):
            populateVisit(user, request.GET['idExpo']);
        token, _ = Token.objects.get_or_create(user=user)
        return Response({'token': token.key})