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.
41 lines
1.2 KiB
41 lines
1.2 KiB
# -*- coding: utf-8 -*-
|
|
#
|
|
from celery import shared_task
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from common.utils import get_object_or_none, get_logger
|
|
from orgs.utils import tmp_to_org, tmp_to_root_org
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
def task_activity_callback(self, pid, trigger, *args, **kwargs):
|
|
from accounts.models import AccountBackupAutomation
|
|
with tmp_to_root_org():
|
|
plan = get_object_or_none(AccountBackupAutomation, pk=pid)
|
|
if not plan:
|
|
return
|
|
if not plan.latest_execution:
|
|
return
|
|
resource_ids = plan.latest_execution.backup_accounts
|
|
org_id = plan.org_id
|
|
return resource_ids, org_id
|
|
|
|
|
|
@shared_task(
|
|
verbose_name=_('Execute account backup plan'),
|
|
activity_callback=task_activity_callback,
|
|
description=_(
|
|
"When performing scheduled or manual account backups, this task is used"
|
|
)
|
|
)
|
|
def execute_account_backup_task(pid, trigger, **kwargs):
|
|
from accounts.models import AccountBackupAutomation
|
|
with tmp_to_root_org():
|
|
plan = get_object_or_none(AccountBackupAutomation, pk=pid)
|
|
if not plan:
|
|
logger.error("No account backup route plan found: {}".format(pid))
|
|
return
|
|
with tmp_to_org(plan.org):
|
|
plan.execute(trigger)
|