github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

[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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
apps/users/api.py
View File

@ -180,15 +180,24 @@ class UserConnectionTokenApi(APIView):
'asset': asset_id, 'asset': asset_id,
'system_user': system_user_id 'system_user': system_user_id
} }
cache.set(token, value, timeout=3600) cache.set(token, value, timeout=60)
return Response({"token": token}, status=201) return Response({"token": token}, status=201)
def get(self, request): def get(self, request):
token = request.query_params.get('token') token = request.query_params.get('token')
user_only = request.query_params.get('user-only', None)
value = cache.get(token, None) value = cache.get(token, None)
if value:
if not value:
return Response('', status=404)
if not user_only:
cache.delete(token) 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()