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
71
72
73
74
75
|
# Copyright 2020 The Pigweed Authors
#
# 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
#
# https://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.
"""The Pigweed command line interface."""
import logging
import os
import sys
from typing import NoReturn
import pw_cli.log
from pw_cli import arguments, plugins, pw_command_plugins
_LOG = logging.getLogger(__name__)
def main() -> NoReturn:
"""Entry point for the pw command."""
args = arguments.parse_args()
pw_cli.log.install(level=args.loglevel, debug_log=args.debug_log)
# Print the banner unless --no-banner or --tab-complete-command is provided.
# Note: args.tab_complete_command may be the empty string '' so check for
# None instead.
if not args.no_banner and args.tab_complete_command is None:
arguments.print_banner()
_LOG.debug('Executing the pw command from %s', args.directory)
os.chdir(args.directory)
pw_command_plugins.register()
if args.tab_complete_option is not None:
arguments.print_completions_for_option(
arguments.arg_parser(),
text=args.tab_complete_option,
tab_completion_format=args.tab_complete_format,
)
sys.exit(0)
if args.tab_complete_command is not None:
for name, plugin in sorted(pw_command_plugins.plugin_registry.items()):
if name.startswith(args.tab_complete_command):
if args.tab_complete_format == 'zsh':
print(':'.join([name, plugin.help()]))
else:
print(name)
sys.exit(0)
if args.help or args.command is None:
print(pw_command_plugins.format_help(), file=sys.stderr)
sys.exit(0)
try:
sys.exit(pw_command_plugins.run(args.command, args.plugin_args))
except (plugins.Error, KeyError) as err:
_LOG.critical('Cannot run command %s.', args.command)
_LOG.critical('%s', err)
sys.exit(2)
if __name__ == '__main__':
main()
|