# -*- 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)