github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-18 00:08:31 +00:00
Code Issues Projects Releases Wiki Activity

perf: add TERMINAL_SSH_KEY_LIMIT_COUNT conf

Browse Source
This commit is contained in:
wangruidong
2024-08-13 16:06:01 +08:00
committed by Bryan
parent 3b1701b1aa
commit 32ae77c42d
12 changed files with 167 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
apps/users/models/user/_auth.py
View File

@@ -18,6 +18,7 @@ from common.utils import (
lazyproperty,
)
from users.signals import post_user_change_password
from users.exceptions import CreateSSHKeyExceedLimit
logger = get_logger(__file__)
@@ -133,6 +134,15 @@ class AuthMixin:
post_user_change_password.send(self.__class__, user=self)
super().set_password(raw_password) # noqa
def set_ssh_key(self, name, public_key, private_key):
if self.can_update_ssh_key():
from authentication.models import SSHKey
SSHKey.objects.create(name=name, public_key=public_key, private_key=private_key, user=self)
post_user_change_password.send(self.__class__, user=self)
def can_create_ssh_key(self):
return self.ssh_keys.count() < settings.TERMINAL_SSH_KEY_LIMIT_COUNT
def can_update_password(self):
return self.is_local