mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-06-07 22:14:52 +00:00
35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
# ~*~ coding: utf-8 ~*~
|
|
|
|
from django.http import HttpResponse, JsonResponse
|
|
from django.views import View
|
|
|
|
from authentication.serializers import SSHKeySerializer
|
|
from common.permissions import IsValidUser
|
|
from common.utils import get_logger, ssh_key_gen
|
|
from common.views.mixins import PermissionsMixin
|
|
from users.exceptions import CreateSSHKeyExceedLimit
|
|
|
|
__all__ = ['UserPublicKeyGenerateView']
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class UserPublicKeyGenerateView(PermissionsMixin, View):
|
|
permission_classes = [IsValidUser]
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
username = request.user.username
|
|
serializer = SSHKeySerializer(data=request.GET, context={'request': request})
|
|
if not serializer.is_valid():
|
|
return JsonResponse(serializer.errors, status=400)
|
|
if not request.user.can_create_ssh_key():
|
|
return HttpResponse(
|
|
CreateSSHKeyExceedLimit().default_detail, status=400
|
|
)
|
|
private, public = ssh_key_gen(username=username, hostname='jumpserver')
|
|
request.user.set_ssh_key(public, private, **serializer.validated_data)
|
|
response = HttpResponse(private, content_type='text/plain')
|
|
filename = "{0}-jumpserver.pem".format(username)
|
|
response['Content-Disposition'] = 'attachment; filename={}'.format(filename)
|
|
return response
|