blob: abf37711f8997a1ae6c5f0567be14d590eeaef4b (
plain)
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
|
"""Application base class for providing a list of data as output.
"""
import abc
import itertools
import logging
from .display import DisplayCommandBase
LOG = logging.getLogger(__name__)
class Lister(DisplayCommandBase):
"""Command base class for providing a list of data as output.
"""
__metaclass__ = abc.ABCMeta
@property
def formatter_namespace(self):
return 'cliff.formatter.list'
@property
def formatter_default(self):
return 'table'
@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)
if not parsed_args.columns:
columns_to_include = column_names
data_gen = data
else:
columns_to_include = [c for c in column_names
if c in parsed_args.columns
]
if not columns_to_include:
raise ValueError('No recognized column names in %s' %
str(parsed_args.columns))
# Set up argument to compress()
selector = [(c in columns_to_include)
for c in column_names]
# Generator expression to only return the parts of a row
# 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(itertools.compress(row, selector))
for row in data)
formatter = self.formatters[parsed_args.formatter]
formatter.emit_list(columns_to_include, data_gen, self.app.stdout, parsed_args)
return 0
|