| author | rougeronj |
| Wed, 10 Jun 2015 15:45:21 +0200 | |
| changeset 158 | 8b4c073a7c2c |
| parent 156 | bf4ae7d9a517 |
| child 167 | 389540e1073f |
| permissions | -rw-r--r-- |
| 51 | 1 |
|
2 |
import json |
|
3 |
||
| 156 | 4 |
from django.conf import settings |
| 51 | 5 |
from django.contrib.auth import get_user_model |
6 |
import requests |
|
|
61
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
7 |
from rest_framework import serializers, status, permissions, parsers, renderers |
| 51 | 8 |
from rest_framework.authtoken.models import Token |
9 |
from rest_framework.response import Response |
|
10 |
from rest_framework.views import APIView |
|
11 |
||
| 135 | 12 |
from ammicosrv.ammico.views import populateVisit |
| 156 | 13 |
from ammicosrv.authentication.serializers import AuthTokenSerializer, \ |
14 |
UserSerializer |
|
| 51 | 15 |
|
16 |
||
|
143
ea633c8f9bfa
add exposition table and update the views to include the new foreign key to the exposition table
rougeronj
parents:
117
diff
changeset
|
17 |
#from ammico.views import populateVisit |
| 51 | 18 |
class User(APIView): |
19 |
""" |
|
20 |
get list user or add user |
|
21 |
""" |
|
22 |
permission_classes = (permissions.AllowAny,) |
|
23 |
||
24 |
def get(self, request): |
|
25 |
user = get_user_model().objects.all() |
|
26 |
serializer = UserSerializer(user, many=True) |
|
27 |
return Response(serializer.data) |
|
28 |
||
29 |
def post(self, request): |
|
30 |
VALID_USER_FIELDS = [f.name for f in get_user_model()._meta.fields] |
|
31 |
serialized = UserSerializer(data=request.data) |
|
32 |
||
33 |
if serialized.is_valid(): |
|
34 |
user_data = {field: data for (field, data) in request.DATA.items() if field in VALID_USER_FIELDS} |
|
35 |
||
36 |
params = {'o': 'user', 'f': 'get', 'mail': user_data['email']} |
|
| 150 | 37 |
r = requests.get(settings.URL_JAMESPOT, params=params) |
| 51 | 38 |
infoUser = json.loads(r.content.decode('utf-8')) |
|
61
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
39 |
if ('idUser' in infoUser['VAL']): |
| 156 | 40 |
user_data.update({"id_user":infoUser['VAL']['idUser']}) |
| 51 | 41 |
user = get_user_model().objects.create_user( |
42 |
**user_data |
|
43 |
) |
|
44 |
return Response(UserSerializer(instance=user).data, status=status.HTTP_201_CREATED) |
|
45 |
else: |
|
46 |
return Response(serialized._errors, status=status.HTTP_400_BAD_REQUEST) |
|
|
61
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
47 |
|
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
48 |
class ObtainAuthToken(APIView): |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
49 |
throttle_classes = () |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
50 |
permission_classes = () |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
51 |
parser_classes = (parsers.FormParser, parsers.MultiPartParser, parsers.JSONParser,) |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
52 |
renderer_classes = (renderers.JSONRenderer,) |
| 51 | 53 |
|
|
61
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
54 |
def post(self, request): |
| 156 | 55 |
serializer = AuthTokenSerializer(data=request.data) |
56 |
if serializer.is_valid(): |
|
57 |
user = serializer.validated_data['user'] |
|
58 |
if (user.id_user and 'idExpo' in request.GET): |
|
59 |
populateVisit(user, request.GET['idExpo']) |
|
60 |
token, _ = Token.objects.get_or_create(user=user) |
|
61 |
return Response({'token': token.key}) |
|
62 |
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) |