mirror of https://github.com/jumpserver/jumpserver
80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
from rest_framework.generics import CreateAPIView
|
|
|
|
from orgs.mixins.api import OrgBulkModelViewSet
|
|
from rbac.permissions import RBACPermission
|
|
|
|
from common.mixins import RecordViewLogMixin
|
|
from common.permissions import UserConfirmation
|
|
from authentication.const import ConfirmType
|
|
from assets.models import Account
|
|
from assets.filters import AccountFilterSet
|
|
from assets.tasks.account_connectivity import test_accounts_connectivity_manual
|
|
from assets import serializers
|
|
|
|
__all__ = ['AccountViewSet', 'AccountSecretsViewSet', 'AccountTaskCreateAPI']
|
|
|
|
|
|
class AccountViewSet(OrgBulkModelViewSet):
|
|
model = Account
|
|
search_fields = ('username', 'asset__address', 'name')
|
|
filterset_class = AccountFilterSet
|
|
serializer_classes = {
|
|
'default': serializers.AccountSerializer,
|
|
'verify': serializers.AssetTaskSerializer
|
|
}
|
|
rbac_perms = {
|
|
'verify': 'assets.test_account',
|
|
'partial_update': 'assets.change_assetaccountsecret',
|
|
}
|
|
|
|
@action(methods=['post'], detail=True, url_path='verify')
|
|
def verify_account(self, request, *args, **kwargs):
|
|
account = super().get_object()
|
|
task = test_accounts_connectivity_manual.delay([account.id])
|
|
return Response(data={'task': task.id})
|
|
|
|
|
|
class AccountSecretsViewSet(RecordViewLogMixin, AccountViewSet):
|
|
"""
|
|
因为可能要导出所有账号，所以单独建立了一个 viewset
|
|
"""
|
|
serializer_classes = {
|
|
'default': serializers.AccountSecretSerializer
|
|
}
|
|
http_method_names = ['get']
|
|
# Todo: 记得打开
|
|
# permission_classes = [RBACPermission, UserConfirmation.require(ConfirmType.MFA)]
|
|
rbac_perms = {
|
|
'list': 'assets.view_assetaccountsecret',
|
|
'retrieve': 'assets.view_assetaccountsecret',
|
|
}
|
|
|
|
|
|
class AccountTaskCreateAPI(CreateAPIView):
|
|
serializer_class = serializers.AccountTaskSerializer
|
|
search_fields = AccountViewSet.search_fields
|
|
filterset_class = AccountViewSet.filterset_class
|
|
|
|
def check_permissions(self, request):
|
|
return request.user.has_perm('assets.test_assetconnectivity')
|
|
|
|
def get_accounts(self):
|
|
queryset = Account.objects.all()
|
|
queryset = self.filter_queryset(queryset)
|
|
return queryset
|
|
|
|
def perform_create(self, serializer):
|
|
account_ids = self.get_accounts().values_list('id', flat=True)
|
|
task = test_accounts_connectivity_manual.delay(account_ids)
|
|
data = getattr(serializer, '_data', {})
|
|
data["task"] = task.id
|
|
setattr(serializer, '_data', data)
|
|
return task
|
|
|
|
def get_exception_handler(self):
|
|
def handler(e, context):
|
|
return Response({"error": str(e)}, status=400)
|
|
return handler
|