mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.6 KiB
47 lines
1.6 KiB
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
|
|
from common.drf.filters import BaseFilterSet
|
|
from orgs.mixins.api import OrgBulkModelViewSet
|
|
from .common import ACLFiltersetMixin
|
|
from .. import models, serializers
|
|
|
|
__all__ = ['CommandFilterACLViewSet', 'CommandGroupViewSet']
|
|
|
|
|
|
class CommandGroupViewSet(OrgBulkModelViewSet):
|
|
model = models.CommandGroup
|
|
filterset_fields = ('name',)
|
|
search_fields = filterset_fields
|
|
serializer_class = serializers.CommandGroupSerializer
|
|
|
|
|
|
class CommandACLFilter(ACLFiltersetMixin, BaseFilterSet):
|
|
class Meta:
|
|
model = models.CommandFilterACL
|
|
fields = ['name', 'users', 'assets']
|
|
|
|
|
|
class CommandFilterACLViewSet(OrgBulkModelViewSet):
|
|
model = models.CommandFilterACL
|
|
filterset_class = CommandACLFilter
|
|
search_fields = ['name']
|
|
serializer_class = serializers.CommandFilterACLSerializer
|
|
rbac_perms = {
|
|
'command_review': 'tickets.add_superticket'
|
|
}
|
|
|
|
@action(['POST'], detail=False, url_path='command-review')
|
|
def command_review(self, request, *args, **kwargs):
|
|
serializer = serializers.CommandReviewSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
data = {
|
|
'run_command': serializer.validated_data['run_command'],
|
|
'session': serializer.session,
|
|
'cmd_filter_acl': serializer.cmd_filter_acl,
|
|
'org_id': serializer.org.id
|
|
}
|
|
ticket = serializer.cmd_filter_acl.create_command_review_ticket(**data)
|
|
info = ticket.get_extra_info_of_review(user=request.user)
|
|
return Response(data=info)
|