Source code for robot.libdocpkg.consoleviewer

#  Copyright 2008-2015 Nokia Networks
#  Copyright 2016-     Robot Framework Foundation
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  See the License for the specific language governing permissions and
#  limitations under the License.

import textwrap

from robot.utils import MultiMatcher, console_encode, unicode
from robot.errors import DataError

[docs]class ConsoleViewer(object): def __init__(self, libdoc): self._libdoc = libdoc self._keywords = KeywordMatcher(libdoc)
[docs] @classmethod def handles(cls, command): return command.lower() in ['list', 'show', 'version']
[docs] @classmethod def validate_command(cls, command, args): if not cls.handles(command): raise DataError("Unknown command '%s'." % command) if command.lower() == 'version' and args: raise DataError("Command 'version' does not take arguments.")
[docs] def view(self, command, *args): self.validate_command(command, args) getattr(self, command.lower())(*args)
[docs] def list(self, *patterns): for kw in'*%s*' % p for p in patterns): self._console(
[docs] def show(self, *names): if MultiMatcher(names, match_if_no_patterns=True).match('intro'): self._show_intro(self._libdoc) if self._libdoc.inits: self._show_inits(self._libdoc) for kw in self._show_keyword(kw)
[docs] def version(self): self._console(self._libdoc.version or 'N/A')
def _console(self, msg): print(console_encode(msg)) def _show_intro(self, lib): self._header(, underline='=') self._data([('Version', lib.version), ('Scope', lib.scope if lib.type == 'LIBRARY' else None)]) self._doc(lib.doc) def _show_inits(self, lib): self._header('Importing', underline='-') for init in lib.inits: self._show_keyword(init, show_name=False) def _show_keyword(self, kw, show_name=True): if show_name: self._header(, underline='-') self._data([('Arguments', '[%s]' % unicode(kw.args))]) self._doc(kw.doc) def _header(self, name, underline): self._console('%s\n%s' % (name, underline * len(name))) def _data(self, items): ljust = max(len(name) for name, _ in items) + 3 for name, value in items: if value: text = '%s%s' % ((name+':').ljust(ljust), value) self._console(self._wrap(text, subsequent_indent=' '*ljust)) def _doc(self, doc): self._console('') for line in doc.splitlines(): self._console(self._wrap(line)) if doc: self._console('') def _wrap(self, text, width=78, **config): return '\n'.join(textwrap.wrap(text, width=width, **config))
[docs]class KeywordMatcher(object): def __init__(self, libdoc): self._keywords = libdoc.keywords
[docs] def search(self, patterns): matcher = MultiMatcher(patterns, match_if_no_patterns=True) for kw in self._keywords: if matcher.match( yield kw