| author | rougeronj |
| Thu, 23 Apr 2015 18:17:58 +0200 | |
| changeset 87 | 7b43de480a11 |
| parent 72 | ed2ee692ff6f |
| child 108 | 4a152f5f4a09 |
| permissions | -rw-r--r-- |
| 51 | 1 |
|
2 |
import json |
|
3 |
||
4 |
from django.contrib.auth import get_user_model |
|
5 |
import requests |
|
|
61
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
6 |
from rest_framework import serializers, status, permissions, parsers, renderers |
| 51 | 7 |
from rest_framework.authtoken.models import Token |
|
61
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
8 |
from rest_framework.authtoken.serializers import AuthTokenSerializer |
| 51 | 9 |
from rest_framework.response import Response |
10 |
from rest_framework.views import APIView |
|
11 |
||
12 |
from config import URL_JAMESPOT |
|
|
72
ed2ee692ff6f
Update PopulateVisit function to get only the new visits from Jamespot.
rougeronj
parents:
61
diff
changeset
|
13 |
from ammico.views import populateVisit |
| 51 | 14 |
|
15 |
||
16 |
class UserSerializer(serializers.ModelSerializer): |
|
17 |
class Meta: |
|
18 |
model = get_user_model() |
|
19 |
||
20 |
class User(APIView): |
|
21 |
""" |
|
22 |
get list user or add user |
|
23 |
""" |
|
24 |
permission_classes = (permissions.AllowAny,) |
|
25 |
||
26 |
def get(self, request): |
|
27 |
print ("here ?") |
|
28 |
user = get_user_model().objects.all() |
|
29 |
serializer = UserSerializer(user, many=True) |
|
30 |
return Response(serializer.data) |
|
31 |
||
32 |
def post(self, request): |
|
33 |
VALID_USER_FIELDS = [f.name for f in get_user_model()._meta.fields] |
|
34 |
DEFAULTS = { |
|
35 |
"groups":"", |
|
36 |
"user_permissions":"" |
|
37 |
} |
|
38 |
request.data.update(DEFAULTS) |
|
39 |
serialized = UserSerializer(data=request.data) |
|
40 |
||
41 |
if serialized.is_valid(): |
|
42 |
user_data = {field: data for (field, data) in request.DATA.items() if field in VALID_USER_FIELDS} |
|
43 |
||
44 |
params = {'o': 'user', 'f': 'get', 'mail': user_data['email']} |
|
45 |
r = requests.get(URL_JAMESPOT, params=params) |
|
46 |
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
|
47 |
if ('idUser' in infoUser['VAL']): |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
48 |
user_data.update({"idUser":infoUser['VAL']['idUser']}) |
| 51 | 49 |
user = get_user_model().objects.create_user( |
50 |
**user_data |
|
51 |
) |
|
52 |
return Response(UserSerializer(instance=user).data, status=status.HTTP_201_CREATED) |
|
53 |
else: |
|
54 |
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
|
55 |
|
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
56 |
class ObtainAuthToken(APIView): |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
57 |
throttle_classes = () |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
58 |
permission_classes = () |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
59 |
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
|
60 |
renderer_classes = (renderers.JSONRenderer,) |
| 51 | 61 |
|
|
61
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
62 |
def post(self, request): |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
63 |
serializer = AuthTokenSerializer(data=request.data) |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
64 |
serializer.is_valid(raise_exception=True) |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
65 |
user = serializer.validated_data['user'] |
|
87
7b43de480a11
get the books and slides from Jamespot in case the user have a idUser (is a registered user in jamespot plateform)
rougeronj
parents:
72
diff
changeset
|
66 |
if (user.idUser): |
|
7b43de480a11
get the books and slides from Jamespot in case the user have a idUser (is a registered user in jamespot plateform)
rougeronj
parents:
72
diff
changeset
|
67 |
populateVisit(user) |
|
61
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
68 |
token, created = Token.objects.get_or_create(user=user) |
|
5abd0a9eafbc
custom ObtainToken function to add PopulateVisite function when a user connects
rougeronj
parents:
51
diff
changeset
|
69 |
return Response({'token': token.key}) |