Esempio n. 1
0
    def __init__(self, config):
        self.config = config
        # CLI user interface
        self.ui = UI(config)
        # CLI commands
        self.commands = {}
        # CLI context
        self.context = Context()
        # Readline matches
        self.matches = []
        # Nagios objects
        self.objects = Objects(self, config)
        # Nagios command pipe
        self.command = nagios.Command(config)

        self.setup_prompt()
        if readline:
            self.setup_readline()
        self.setup_commands()
Esempio n. 2
0
class CLI(object):
    def __init__(self, config):
        self.config = config
        # CLI user interface
        self.ui = UI(config)
        # CLI commands
        self.commands = {}
        # CLI context
        self.context = Context()
        # Readline matches
        self.matches = []
        # Nagios objects
        self.objects = Objects(self, config)
        # Nagios command pipe
        self.command = nagios.Command(config)

        self.setup_prompt()
        if readline:
            self.setup_readline()
        self.setup_commands()

    # Setup

    def setup_prompt(self):
        prompt = self.config.get("ui.prompt_separator", ":").join(map(str, self.context.stack))
        self.prompt = (
            self.ui.color(
                "prompt",
                self.config.get("ui.prompt")
                % (
                    self.ui.color(
                        "prompt.object", self.config.get("ui.prompt_separator", ":").join(map(str, self.context.stack))
                    )
                ),
            )
            + " "
        )

    def setup_readline(self):
        history = self.config.get("cli.history")
        if history:
            # Load history
            history = os.path.expanduser(history)
            try:
                readline.read_history_file(history)
            except IOError:
                pass

            # Save history at exit
            import atexit

            atexit.register(readline.write_history_file, history)

        readline.set_completer(self.complete)
        readline.parse_and_bind("tab: complete")

        # We need hyphen in hostnames
        readline.set_completer_delims(readline.get_completer_delims().replace("-", ""))

    def setup_commands(self):
        for item in command.DEFAULT:
            self.command_add(item)

    # I/O

    def flush(self):
        sys.stdout.flush()

    def send(self, text, error=False):
        if error:
            sys.stderr.write(text)
        else:
            sys.stdout.write(text)

    def sendline(self, text, error=False):
        self.send(text + "\n", error=error)

    def error(self, text):
        self.sendline(self.ui.color("error", text), error=True)

    def warn(self, text):
        self.sendline(self.ui.color("warn", text), error=True)

    def ask_bool(self, question, min_length=None, default=None):
        if min_length:
            question = question.ljust(min_length, " ")

        answer = None
        prompt = question + " [yn]: "
        if default is not None:
            if default is True or default in "yY":
                prompt = question + " [Yn]: "
                default = True

            elif default is False or default in "nN":
                prompt = question + " [yN]: "
                default = False

            else:
                default = None

        while answer is None:
            answer = raw_input(prompt)
            if answer:
                if answer in "yY":
                    answer = True
                elif answer in "nN":
                    answer = False
                else:
                    self.error("Invalid response")
                    answer = None
            else:
                answer = default

        return answer

    def ask_str(self, question, min_length=None):
        if min_length:
            question = question.ljust(min_length, " ")

        answer = None
        prompt = question + ": "
        while answer is None:
            answer = raw_input(prompt)
            if not answer:
                answer = None

        return answer

    # Commands

    def command_add(self, obj):
        self.commands[obj.name] = obj
        self.commands[obj.name].cli = self

    def command_run(self, name, args):
        if name not in self.commands:
            matches = []
            for state in xrange(0, len(self.commands)):
                match = self.complete(name, state)
                if match is None:
                    break
                else:
                    matches.append(match)

            # One possible match, complete it
            if len(matches) == 1:
                return self.command_run(matches[0], args)

            return False

        if not self.commands[name].valid_in_context(self.context):
            print "Not valid in context"
            return False

        try:
            self.commands[name].run(*args)
        except TypeError, e:
            msg = e.message
            info = RE_RUN_GIVEN.search(e.message)
            if info:
                # Function arguments count (minus self)
                args = self.commands[name].run.func_code.co_argcount - 1
                try:
                    least = args - len(self.commands[name].run.func_defaults)
                except TypeError:
                    least = 0
                # Given arguments count (minus self)
                given = int(info.groupdict()["given"]) - 1
                if args == least:
                    error = "exactly"
                    count = args
                elif given < least:
                    error = "at least"
                    count = least
                elif given > args:
                    error = "at most"
                    count = args
                else:
                    print "huh?", args, least, given
                    return

                plural = ""
                if count != 1:
                    plural = "s"
                self.error("Syntax error: %s takes %s %d argument%s (%d given)" % (name, error, count, plural, given))
                return True
            else:
                print dir(e)
                print e.message
                print e.args
                raise

        return True