github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-21 19:38:07 +00:00
Code Issues Projects Releases Wiki Activity

feat: 批量不是发布机 (#11150)

Browse Source 
Co-authored-by: feng <1304903146@qq.com>
This commit is contained in:
fit2bot
2023-08-01 17:42:16 +08:00
committed by GitHub
parent 44397caad4
commit 8ed823d587
3 changed files with 24 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
apps/terminal/api/applet/host.py
View File

@@ -1,3 +1,4 @@
from rest_framework import status
from rest_framework import viewsets
from rest_framework.decorators import action
from rest_framework.response import Response
@@ -7,8 +8,7 @@ from common.permissions import IsServiceAccount
from orgs.utils import tmp_to_builtin_org
from terminal.models import AppletHost, AppletHostDeployment
from terminal.serializers import (
AppletHostSerializer, AppletHostDeploymentSerializer,
AppletHostStartupSerializer, AppletHostDeployAppletSerializer
AppletHostSerializer, AppletHostDeploymentSerializer, AppletHostStartupSerializer
)
from terminal.tasks import run_applet_host_deployment, run_applet_host_deployment_install_applet
@@ -63,12 +63,20 @@ class AppletHostDeploymentViewSet(viewsets.ModelViewSet):
instance.save_task(task.id)
return Response({'task': str(task.id)}, status=201)
@action(methods=['post'], detail=False, serializer_class=AppletHostDeployAppletSerializer)
@action(methods=['post'], detail=False)
def applets(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
applet_id = serializer.validated_data.pop('applet_id', '')
instance = serializer.save()
task = run_applet_host_deployment_install_applet.delay(instance.id, applet_id)
instance.save_task(task.id)
return Response({'task': str(task.id)}, status=201)
hosts = request.data.get('hosts', [])
applet_id = request.data.get('applet_id', '')
model = self.get_queryset().model
hosts_qs = AppletHost.objects.filter(id__in=hosts)
if not hosts_qs.exists():
return Response(status=status.HTTP_404_NOT_FOUND)
objs = [model(host=host) for host in hosts_qs]
applet_host_deployments = model.objects.bulk_create(objs)
applet_host_deployment_ids = [str(obj.id) for obj in applet_host_deployments]
task = run_applet_host_deployment_install_applet.delay(applet_host_deployment_ids, applet_id)
task_id = str(task.id)
model.objects.filter(id__in=applet_host_deployment_ids).update(task=task_id)
return Response({'task': task_id}, status=201)