diff options
| -rw-r--r-- | cliff/command.py | 5 | ||||
| -rw-r--r-- | cliff/display.py | 17 | ||||
| -rw-r--r-- | cliff/formatters/base.py | 8 | ||||
| -rw-r--r-- | cliff/formatters/shell.py | 3 | ||||
| -rw-r--r-- | cliff/lister.py | 16 | ||||
| -rw-r--r-- | cliff/show.py | 7 | ||||
| -rw-r--r-- | cliff/tests/test_show.py | 36 |
7 files changed, 54 insertions, 38 deletions
diff --git a/cliff/command.py b/cliff/command.py index 116ae5c..7dfe749 100644 --- a/cliff/command.py +++ b/cliff/command.py @@ -3,15 +3,16 @@ import abc import argparse import inspect +import six + +@six.add_metaclass(abc.ABCMeta) class Command(object): """Base class for command plugins. :param app: Application instance invoking the command. :paramtype app: cliff.app.App """ - __metaclass__ = abc.ABCMeta - def __init__(self, app, app_args): self.app = app self.app_args = app_args diff --git a/cliff/display.py b/cliff/display.py index e04fd29..6efcf23 100644 --- a/cliff/display.py +++ b/cliff/display.py @@ -1,8 +1,19 @@ """Application base class for displaying data. """ import abc + +try: + from itertools import compress +except ImportError: + # for py26 compat + from itertools import izip + + def compress(data, selectors): + return (d for d, s in izip(data, selectors) if s) + import logging +import six import stevedore from .command import Command @@ -11,10 +22,10 @@ from .command import Command LOG = logging.getLogger(__name__) +@six.add_metaclass(abc.ABCMeta) class DisplayCommandBase(Command): """Command base class for displaying data about a single object. """ - __metaclass__ = abc.ABCMeta def __init__(self, app, app_args): super(DisplayCommandBase, self).__init__(app, app_args) @@ -80,3 +91,7 @@ class DisplayCommandBase(Command): column_names, data = self.take_action(parsed_args) self.produce_output(parsed_args, column_names, data) return 0 + + @staticmethod + def _compress_iterable(iterable, selectors): + return compress(iterable, selectors) diff --git a/cliff/formatters/base.py b/cliff/formatters/base.py index 43b8f17..c477a54 100644 --- a/cliff/formatters/base.py +++ b/cliff/formatters/base.py @@ -3,9 +3,11 @@ import abc +import six + +@six.add_metaclass(abc.ABCMeta) class Formatter(object): - __metaclass__ = abc.ABCMeta @abc.abstractmethod def add_argument_group(self, parser): @@ -15,10 +17,10 @@ class Formatter(object): """ +@six.add_metaclass(abc.ABCMeta) class ListFormatter(Formatter): """Base class for formatters that know how to deal with multiple objects. """ - __metaclass__ = abc.ABCMeta @abc.abstractmethod def emit_list(self, column_names, data, stdout, parsed_args): @@ -32,10 +34,10 @@ class ListFormatter(Formatter): """ +@six.add_metaclass(abc.ABCMeta) class SingleFormatter(Formatter): """Base class for formatters that work with single objects. """ - __metaclass__ = abc.ABCMeta @abc.abstractmethod def emit_one(self, column_names, data, stdout, parsed_args): diff --git a/cliff/formatters/shell.py b/cliff/formatters/shell.py index fd4f29e..e613c22 100644 --- a/cliff/formatters/shell.py +++ b/cliff/formatters/shell.py @@ -3,6 +3,7 @@ from .base import SingleFormatter +import argparse import six @@ -19,7 +20,7 @@ class ShellFormatter(SingleFormatter): default=[], dest='variables', metavar='VARIABLE', - help='specify the variable(s) to include, can be repeated', + help=argparse.SUPPRESS, ) group.add_argument( '--prefix', diff --git a/cliff/lister.py b/cliff/lister.py index 1b6c3b6..6cf79a2 100644 --- a/cliff/lister.py +++ b/cliff/lister.py @@ -1,28 +1,20 @@ """Application base class for providing a list of data as output. """ import abc - -try: - from itertools import compress -except ImportError: - # for py26 compat - from itertools import izip - - def compress(data, selectors): - return (d for d, s in izip(data, selectors) if s) - import logging +import six + from .display import DisplayCommandBase LOG = logging.getLogger(__name__) +@six.add_metaclass(abc.ABCMeta) class Lister(DisplayCommandBase): """Command base class for providing a list of data as output. """ - __metaclass__ = abc.ABCMeta @property def formatter_namespace(self): @@ -56,7 +48,7 @@ class Lister(DisplayCommandBase): # of data that the user has expressed interest in # seeing. We have to convert the compress() output to a # list so the table formatter can ask for its length. - data_gen = (list(compress(row, selector)) + data_gen = (list(self._compress_iterable(row, selector)) for row in data) self.formatter.emit_list(columns_to_include, data_gen, diff --git a/cliff/show.py b/cliff/show.py index 855b2d2..fb01068 100644 --- a/cliff/show.py +++ b/cliff/show.py @@ -1,19 +1,20 @@ """Application base class for displaying data about a single object. """ import abc -import itertools import logging +import six + from .display import DisplayCommandBase LOG = logging.getLogger(__name__) +@six.add_metaclass(abc.ABCMeta) class ShowOne(DisplayCommandBase): """Command base class for displaying data about a single object. """ - __metaclass__ = abc.ABCMeta @property def formatter_namespace(self): @@ -38,7 +39,7 @@ class ShowOne(DisplayCommandBase): # Set up argument to compress() selector = [(c in columns_to_include) for c in column_names] - data = list(itertools.compress(data, selector)) + data = list(self._compress_iterable(data, selector)) self.formatter.emit_one(columns_to_include, data, self.app.stdout, diff --git a/cliff/tests/test_show.py b/cliff/tests/test_show.py index 41df5e1..5db07ec 100644 --- a/cliff/tests/test_show.py +++ b/cliff/tests/test_show.py @@ -1,5 +1,7 @@ #!/usr/bin/env python +import weakref + from cliff.show import ShowOne import mock @@ -9,15 +11,16 @@ class FauxFormatter(object): def __init__(self): self.args = [] + self.obj = weakref.proxy(self) - def emit_list(self, columns, data, stdout, args): + def emit_one(self, columns, data, stdout, args): self.args.append((columns, data)) class ExerciseShowOne(ShowOne): - def load_formatter_plugins(self): - self.formatters = { + def _load_formatter_plugins(self): + return { 'test': FauxFormatter(), } return @@ -29,21 +32,22 @@ class ExerciseShowOne(ShowOne): ) -# def test_formatter_args(): -# app = mock.Mock() -# test_lister = ExerciseLister(app, []) +def test_formatter_args(): + app = mock.Mock() + test_show = ExerciseShowOne(app, []) + + parsed_args = mock.Mock() + parsed_args.columns = ('Col1', 'Col2') + parsed_args.formatter = 'test' -# parsed_args = mock.Mock() -# parsed_args.columns = ('Col1', 'Col2') -# parsed_args.formatter = 'test' + test_show.run(parsed_args) + f = test_show._formatter_plugins['test'] + assert len(f.args) == 1 + args = f.args[0] + assert args[0] == list(parsed_args.columns) + data = list(args[1]) + assert data == [('a', 'A'), ('b', 'B')] -# test_lister.run(parsed_args) -# f = test_lister.formatters['test'] -# assert len(f.args) == 1 -# args = f.args[0] -# assert args[0] == list(parsed_args.columns) -# data = list(args[1]) -# assert data == [['a', 'A'], ['b', 'B']] def test_dict2columns(): app = mock.Mock() |