[Update] 支持获取token的用户

pull/1068/head
ibuler 2018-03-12 19:39:27 +08:00
parent e4823a21e3
commit 8e4ab9f3a9
1 changed files with 14 additions and 5 deletions

View File

@ -180,15 +180,24 @@ class UserConnectionTokenApi(APIView):
'asset': asset_id,
'system_user': system_user_id
}
cache.set(token, value, timeout=3600)
cache.set(token, value, timeout=60)
return Response({"token": token}, status=201)
def get(self, request):
token = request.query_params.get('token')
user_only = request.query_params.get('user-only', None)
value = cache.get(token, None)
if value:
if not value:
return Response('', status=404)
if not user_only:
cache.delete(token)
return Response(value)
return Response(value)
else:
return Response({'user': value['user']})
def get_permissions(self):
if self.request.query_params.get('user-only', None):
self.permission_classes = (AllowAny,)
return super().get_permissions()