summaryrefslogtreecommitdiff
path: root/cliff/formatters/shell.py
blob: a4905500cc6b8589f5f4c54aae4fa4ba107988f9 (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
"""Output formatters using shell syntax.
"""

from .base import SingleFormatter


class ShellFormatter(SingleFormatter):

    def add_argument_group(self, parser):
        group = parser.add_argument_group(
            title='shell formatter',
            description='Print values in a format a UNIX shell can parse (variable="value")',
            )
        group.add_argument(
            '--variable',
            action='append',
            default=[],
            dest='variables',
            metavar='VARIABLE',
            help='specify the variable(s) to include, can be repeated',
            )

    def emit_one(self, column_names, data, stdout, parsed_args):
        variable_names = [c.lower().replace(' ', '_')
                          for c in column_names
                          ]
        desired_columns = parsed_args.variables
        for name, value in zip(variable_names, data):
            if name in desired_columns or not desired_columns:
                stdout.write('%s="%s"\n' % (name, value))
        return