mirror of https://github.com/jumpserver/jumpserver
20 lines
673 B
Python
20 lines
673 B
Python
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',)
|
|
ordering = ('-date_last_used', '-date_created')
|
|
|
|
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()
|