mirror of https://github.com/jumpserver/jumpserver
22 lines
715 B
Python
22 lines
715 B
Python
|
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()
|