perf: optimize adhoc asset selection experience

apps/ops/api/__init__.py

@@ -2,6 +2,7 @@
 #
 from .adhoc import *
 from .celery import *
+from .inventory import *
 from .job import *
 from .playbook import *
 from .variable import *

apps/ops/api/inventory.py

@@ -0,0 +1,42 @@
+import os
+import uuid
+
+from django.conf import settings
+from rest_framework.response import Response
+from rest_framework.views import APIView
+
+from assets.models import Asset
+from common.permissions import IsValidUser
+from ops.models.job import JMSPermedInventory
+
+__all__ = ['InventoryClassifiedHostsAPI']
+
+
+class InventoryClassifiedHostsAPI(APIView):
+    permission_classes = [IsValidUser]
+
+    def post(self, request, **kwargs):
+        asset_ids = request.data.get('assets', [])
+        node_ids = request.data.get('nodes', [])
+        runas_policy = request.data.get('runas_policy', 'privileged_first')
+        account_prefer = request.data.get('account_prefer', 'root,Administrator')
+        module = request.data.get('module', 'shell')
+        # 合并节点和资产
+        assets = list(Asset.objects.filter(id__in=asset_ids).all())
+
+        # 创建临时目录
+        tmp_dir = os.path.join(settings.PROJECT_DIR, 'inventory', str(uuid.uuid4()))
+        os.makedirs(tmp_dir, exist_ok=True)
+
+        # 创建库存对象并获取分类的主机
+        inventory = JMSPermedInventory(
+            assets=assets,
+            nodes=node_ids,
+            module=module,
+            account_policy=runas_policy,
+            account_prefer=account_prefer,
+            user=self.request.user
+        )
+        classified_hosts = inventory.get_classified_hosts(tmp_dir)
+
+        return Response(data=classified_hosts)