mirror of https://github.com/jumpserver/jumpserver
30 lines
1018 B
Python
30 lines
1018 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.drf.api import JMSModelViewSet
|
|
from ..models import TempToken
|
|
from ..serializers import TempTokenSerializer
|
|
|
|
|
|
class TempTokenViewSet(JMSModelViewSet):
|
|
serializer_class = TempTokenSerializer
|
|
permission_classes = [RBACPermission]
|
|
http_method_names = ['post', 'get', 'options', 'patch']
|
|
rbac_perms = {
|
|
'expire': 'authentication.change_temptoken',
|
|
}
|
|
|
|
def get_queryset(self):
|
|
username = self.request.user.username
|
|
return TempToken.objects.filter(username=username).order_by('-date_created')
|
|
|
|
@action(methods=['PATCH'], detail=True, url_path='expire')
|
|
def expire(self, *args, **kwargs):
|
|
instance = self.get_object()
|
|
instance.date_expired = timezone.now()
|
|
instance.save()
|
|
serializer = self.get_serializer(instance)
|
|
return Response(serializer.data)
|