mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-05-12 02:02:10 +00:00
24 lines
790 B
Python
24 lines
790 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]
|
|
filterset_fields = ('name', 'is_active')
|
|
search_fields = ('name',)
|
|
|
|
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()
|