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.
78 lines
2.3 KiB
78 lines
2.3 KiB
# ~*~ coding: utf-8 ~*~
|
|
|
|
from __future__ import absolute_import, unicode_literals
|
|
|
|
import json
|
|
import time
|
|
import uuid
|
|
|
|
from django.utils import timezone
|
|
|
|
from assets.models import Asset
|
|
from common.utils import get_logger
|
|
from .ansible.runner import AdHocRunner
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
def run_AdHoc(task_tuple, assets,
|
|
task_name='Ansible AdHoc runner',
|
|
task_id=None, pattern='all',
|
|
record=True, verbose=True):
|
|
"""
|
|
:param task_tuple: (('module_name', 'module_args'), ('module_name', 'module_args'))
|
|
:param assets: [asset1, asset2]
|
|
:param task_name:
|
|
:param task_id:
|
|
:param pattern:
|
|
:param record:
|
|
:param verbose:
|
|
:return:
|
|
"""
|
|
|
|
if not assets:
|
|
logger.warning('Empty assets, runner cancel')
|
|
return
|
|
if isinstance(assets[0], Asset):
|
|
assets = [asset._to_secret_json() for asset in assets]
|
|
if task_id is None:
|
|
task_id = str(uuid.uuid4())
|
|
|
|
runner = AdHocRunner(assets)
|
|
if record:
|
|
from .models import Task
|
|
if not Task.objects.filter(uuid=task_id):
|
|
record = Task(uuid=task_id,
|
|
name=task_name,
|
|
assets=','.join(str(asset['id']) for asset in assets),
|
|
module_args=task_tuple,
|
|
pattern=pattern)
|
|
record.save()
|
|
else:
|
|
record = Task.objects.get(uuid=task_id)
|
|
record.date_start = timezone.now()
|
|
record.date_finished = None
|
|
record.timedelta = None
|
|
record.is_finished = False
|
|
record.is_success = False
|
|
record.save()
|
|
ts_start = time.time()
|
|
if verbose:
|
|
logger.debug('Start runner {}'.format(task_name))
|
|
result = runner.run(task_tuple, pattern=pattern, task_name=task_name)
|
|
timedelta = round(time.time() - ts_start, 2)
|
|
summary = runner.clean_result()
|
|
if record:
|
|
record.date_finished = timezone.now()
|
|
record.is_finished = True
|
|
if verbose:
|
|
record.result = json.dumps(result, indent=4, sort_keys=True)
|
|
record.summary = json.dumps(summary)
|
|
record.timedelta = timedelta
|
|
if len(summary['failed']) == 0:
|
|
record.is_success = True
|
|
else:
|
|
record.is_success = False
|
|
record.save()
|
|
return summary, result
|