mirror of https://github.com/jumpserver/jumpserver
73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
from celery import shared_task
|
|
from django.utils.translation import gettext_lazy as _, gettext_noop
|
|
|
|
from accounts.const import AutomationTypes
|
|
from accounts.tasks.common import quickstart_automation_by_snapshot
|
|
from common.utils import get_logger, get_object_or_none
|
|
from orgs.utils import tmp_to_org, tmp_to_root_org
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
def task_activity_callback(self, pid, trigger, tp, *args, **kwargs):
|
|
model = AutomationTypes.get_type_model(tp)
|
|
with tmp_to_root_org():
|
|
instance = get_object_or_none(model, pk=pid)
|
|
if not instance:
|
|
return
|
|
if not instance.latest_execution:
|
|
return
|
|
resource_ids = instance.latest_execution.get_all_asset_ids()
|
|
return resource_ids, instance.org_id
|
|
|
|
|
|
@shared_task(
|
|
queue='ansible', verbose_name=_('Account execute automation'),
|
|
activity_callback=task_activity_callback
|
|
)
|
|
def execute_account_automation_task(pid, trigger, tp):
|
|
model = AutomationTypes.get_type_model(tp)
|
|
with tmp_to_root_org():
|
|
instance = get_object_or_none(model, pk=pid)
|
|
if not instance:
|
|
logger.error("No automation task found: {}".format(pid))
|
|
return
|
|
with tmp_to_org(instance.org):
|
|
instance.execute(trigger)
|
|
|
|
|
|
def record_task_activity_callback(self, record_id, *args, **kwargs):
|
|
from accounts.models import ChangeSecretRecord
|
|
with tmp_to_root_org():
|
|
record = get_object_or_none(ChangeSecretRecord, id=record_id)
|
|
if not record:
|
|
return
|
|
resource_ids = [record.id]
|
|
org_id = record.execution.org_id
|
|
return resource_ids, org_id
|
|
|
|
|
|
@shared_task(
|
|
queue='ansible', verbose_name=_('Execute automation record'),
|
|
activity_callback=record_task_activity_callback
|
|
)
|
|
def execute_automation_record_task(record_id, tp):
|
|
from accounts.models import ChangeSecretRecord
|
|
with tmp_to_root_org():
|
|
instance = get_object_or_none(ChangeSecretRecord, pk=record_id)
|
|
if not instance:
|
|
logger.error("No automation record found: {}".format(record_id))
|
|
return
|
|
|
|
task_name = gettext_noop('Execute automation record')
|
|
task_snapshot = {
|
|
'secret': instance.new_secret,
|
|
'secret_type': instance.execution.snapshot.get('secret_type'),
|
|
'accounts': [str(instance.account_id)],
|
|
'assets': [str(instance.asset_id)],
|
|
'params': {},
|
|
'record_id': record_id,
|
|
}
|
|
with tmp_to_org(instance.execution.org_id):
|
|
quickstart_automation_by_snapshot(task_name, tp, task_snapshot)
|