1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
"""Application base class for providing a list of data as output.
"""
import abc
import logging
import pkg_resources
from .command import Command
LOG = logging.getLogger(__name__)
class Lister(Command):
"""Command base class for providing a list of data as output.
"""
__metaclass__ = abc.ABCMeta
def __init__(self, app, app_args):
super(Lister, self).__init__(app, app_args)
self.load_formatter_plugins()
def load_formatter_plugins(self):
self.formatters = {}
for ep in pkg_resources.iter_entry_points('cliff.formatter.list'):
try:
self.formatters[ep.name] = ep.load()()
except Exception as err:
LOG.error(err)
if self.app_args.debug:
raise
def get_parser(self, prog_name):
parser = super(Lister, self).get_parser(prog_name)
formatter_group = parser.add_argument_group(
title='Output Formatters',
description='List output formatter options',
)
formatter_choices = sorted(self.formatters.keys())
formatter_default = 'table'
if formatter_default not in formatter_choices:
formatter_default = formatter_choices[0]
formatter_group.add_argument(
'-f', '--format',
dest='formatter',
action='store',
choices=formatter_choices,
default=formatter_default,
help='the output format to use, defaults to %s' % formatter_default,
)
for name, formatter in sorted(self.formatters.items()):
formatter.add_argument_group(parser)
return parser
@abc.abstractmethod
def get_data(self, parsed_args):
"""Return a tuple containing the column names and an iterable
containing the data to be listed.
"""
def run(self, parsed_args):
column_names, data = self.get_data(parsed_args)
formatter = self.formatters[parsed_args.formatter]
formatter.emit_list(column_names, data, self.app.stdout, parsed_args)
return 0
|
