CuratorAction

Complexity

Total Complexity

21

Size/Duplication

Total Lines	87
Duplicated Lines	0 %

9 Methods

Rating	Name	Duplication	Size	Complexity
A	do_command()	0	15	2
A	__init__()	0	7	1
A	command()	0	5	2
A	action()	0	3	1
A	exit_msg()	0	11	3
A	do_show()	0	11	3
A	show_dry_run()	0	14	4
A	api()	0	5	2
A	act_on()	0	6	3

# pylint: disable=no-member

from curator_invoke import CuratorInvoke
from esbase_action import ESBaseAction
from curator.api.utils import index_closed
import logging
import sys

logger = logging.getLogger(__name__)


class CuratorAction(ESBaseAction):

    def __init__(self, config=None):
        super(CuratorAction, self).__init__(config=config)
        self.success = True
        self._act_on = None
        self._command = None
        self._api = None
        self._action = None

    @property
    def action(self):
        return self._action

    @property
    def act_on(self):
        if not self._act_on:
            _act_on = 'indices' if self.action.startswith('indices') else 'snapshots'
            self._act_on = _act_on
        return self._act_on

    @property
    def command(self):
        if not self._command:
            self._command = self.action.split('.')[1]
        return self._command

    @property
    def api(self):
        if not self._api:
            self._api = CuratorInvoke(**self.config)
        return self._api

    def show_dry_run(self):
        """
        Log dry run output with the command which would have been executed.
        """
        command = self.command
        items = self.api.fetch(act_on=self.act_on, on_nofilters_showall=True)
        print "DRY RUN MODE. No changes will be made."
        print "DRY RUN MODE. Executing command {} {}.".format(command, self.act_on)
        for item in items:
            if self.act_on == 'snapshots':
                print "DRY RUN: {0}: {1}".format(command, item)
            else:
                print "DRY RUN: {0}: {1}{2}".format(command, item, ' (CLOSED)'
                                                    if index_closed(self.client, item) else '')

    def do_show(self):
        """
        Show indices or snapshots command.
        """
        if not self.config.dry_run:
            for item in self.api.fetch(act_on=self.act_on,
                                       on_nofilters_showall=True):
                print item
        else:
            self.show_dry_run()
        sys.exit(0)

    @staticmethod
    def exit_msg(success):
        """
        Display a message corresponding to whether the job completed
        successfully or not then exit.
        """
        if success:
            logger.info("Job completed successfully.")
        else:
            logger.warn("Job did not complete successfully.")
        sys.exit(0) if success else sys.exit(1)

    def do_command(self):
        """
        Do the command.
        """
        # Show and exit
        self.command == "show" and self.do_show()

        if self.config.dry_run:
            self.show_dry_run()
        else:
            logger.info("Job starting: %s %s", self.command, self.act_on)
            logger.debug("Params: %s", self.config)

            success = self.api.invoke(command=self.command, act_on=self.act_on)
            self.exit_msg(success)