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
66
67
68
69
70
|
# 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
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
"""Output formatters using shell syntax."""
import argparse
import collections.abc
import typing as ty
from cliff import columns
from cliff.formatters import base
class ShellFormatter(base.SingleFormatter):
def add_argument_group(self, parser: argparse.ArgumentParser) -> None:
group = parser.add_argument_group(
title='shell formatter',
description='a format a UNIX shell can parse (variable="value")',
)
group.add_argument(
'--variable',
action='append',
default=[],
dest='variables',
metavar='VARIABLE',
help=argparse.SUPPRESS,
)
group.add_argument(
'--prefix',
action='store',
default='',
dest='prefix',
help='add a prefix to all variable names',
)
def emit_one(
self,
column_names: collections.abc.Sequence[str],
data: collections.abc.Sequence[ty.Any],
stdout: ty.TextIO,
parsed_args: argparse.Namespace,
) -> None:
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:
value = (
str(value.machine_readable())
if isinstance(value, columns.FormattableColumn)
else value
)
if isinstance(value, str):
value = value.replace('"', '\\"')
if isinstance(name, str):
# Colons and dashes may appear as a resource property but
# are invalid to use in a shell, replace them with an
# underscore.
name = name.replace(':', '_')
name = name.replace('-', '_')
stdout.write(f'{parsed_args.prefix}{name}="{value}"\n')
return
|
