[Update] 修改用户public_key生成密钥的view

pull/4082/head
Bai 2020-06-04 17:05:36 +08:00
parent c411b0a38e
commit afc7f3bb9c
1 changed files with 1 additions and 2 deletions

View File

@ -46,8 +46,7 @@ class UserPublicKeyGenerateView(PermissionsMixin, View):
def get(self, request, *args, **kwargs):
username = request.user.username
private, public = ssh_key_gen(username=username, hostname='jumpserver')
request.user.public_key = public
request.user.save()
request.user.set_public_key(public)
response = HttpResponse(private, content_type='text/plain')
filename = "{0}-jumpserver.pem".format(username)
response['Content-Disposition'] = 'attachment; filename={}'.format(filename)