61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
import uuid
|
|
|
|
from django.conf import settings
|
|
from django.core.cache import cache
|
|
from rest_framework.response import Response
|
|
from rest_framework.generics import CreateAPIView
|
|
|
|
from common.utils import get_logger
|
|
from common.permissions import IsSuperUserOrAppUser
|
|
from orgs.mixins.api import RootOrgViewMixin
|
|
|
|
from ..serializers import ConnectionTokenSerializer
|
|
|
|
logger = get_logger(__name__)
|
|
__all__ = ['UserConnectionTokenApi']
|
|
|
|
|
|
class UserConnectionTokenApi(RootOrgViewMixin, CreateAPIView):
|
|
permission_classes = (IsSuperUserOrAppUser,)
|
|
serializer_class = ConnectionTokenSerializer
|
|
|
|
def perform_create(self, serializer):
|
|
user = serializer.validated_data['user']
|
|
asset = serializer.validated_data['asset']
|
|
system_user = serializer.validated_data['system_user']
|
|
token = str(uuid.uuid4())
|
|
value = {
|
|
'user': str(user.id),
|
|
'username': user.username,
|
|
'asset': str(asset.id),
|
|
'hostname': asset.hostname,
|
|
'system_user': str(system_user.id),
|
|
'system_user_name': system_user.name
|
|
}
|
|
cache.set(token, value, timeout=20)
|
|
return token
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
if not settings.CONNECTION_TOKEN_ENABLED:
|
|
data = {'error': 'Connection token disabled'}
|
|
return Response(data, status=400)
|
|
|
|
if not request.user.is_superuser:
|
|
data = {'error': 'Only super user can create token'}
|
|
return Response(data, status=403)
|
|
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
token = self.perform_create(serializer)
|
|
return Response({"token": token}, status=201)
|
|
|
|
def get(self, request):
|
|
token = request.query_params.get('token')
|
|
value = cache.get(token, None)
|
|
|
|
if not value:
|
|
return Response('', status=404)
|
|
return Response(value)
|