jumpserver/apps/accounts/automations/backup_account/manager.py

# -*- coding: utf-8 -*-
#
import time

from django.utils import timezone
from django.utils.translation import gettext_lazy as _

from common.utils.timezone import local_now_display
from .handlers import AccountBackupHandler


class AccountBackupManager:
    def __init__(self, execution):
        self.execution = execution
        self.date_start = timezone.now()
        self.time_start = time.time()
        self.date_end = None
        self.time_end = None
        self.timedelta = 0

    def do_run(self):
        execution = self.execution
        account_backup_execution_being_executed = _('The account backup plan is being executed')
        print(f'\n\033[33m# {account_backup_execution_being_executed}\033[0m')
        handler = AccountBackupHandler(execution)
        handler.run()

    def pre_run(self):
        self.execution.date_start = self.date_start
        self.execution.save()

    def post_run(self):
        self.time_end = time.time()
        self.date_end = timezone.now()

        print('\n\n' + '-' * 80)
        plan_execution_end = _('Plan execution end')
        print('{} {}\n'.format(plan_execution_end, local_now_display()))
        self.timedelta = self.time_end - self.time_start
        time_cost = _('Time cost')
        print('{}: {}s'.format(time_cost, self.timedelta))
        self.execution.timedelta = self.timedelta
        self.execution.save()

    def run(self):
        self.pre_run()
        self.do_run()
        self.post_run()