feat: support configuring multiple SSH keys for users

This commit is contained in:
wangruidong
2024-07-29 19:37:50 +08:00
committed by Bryan
parent 7a38c9136e
commit 2a5c41dfaf
16 changed files with 162 additions and 87 deletions

View File

@@ -11,6 +11,7 @@ from .login_confirm import *
from .mfa import *
from .password import *
from .session import *
from .ssh_key import *
from .sso import *
from .temp_token import *
from .token import *

View File

@@ -0,0 +1,21 @@
from django.utils import timezone
from rest_framework.response import Response
from rest_framework.decorators import action
from rbac.permissions import RBACPermission
from common.api import JMSModelViewSet
from common.permissions import IsValidUser
from ..serializers import SSHKeySerializer
from users.notifications import ResetPublicKeySuccessMsg
class SSHkeyViewSet(JMSModelViewSet):
serializer_class = SSHKeySerializer
permission_classes = [IsValidUser]
def get_queryset(self):
return self.request.user.ssh_keys.all()
def perform_update(self, serializer):
super().perform_update(serializer)
ResetPublicKeySuccessMsg(self.request.user, self.request).publish_async()