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
76
77
78
79
80
81
|
#!/usr/bin/env python3
# coding=utf-8
# PYTHON_ARGCOMPLETE_OK - This is required at the beginning of the file to enable argcomplete support
"""A simple example demonstrating integration with argcomplete"""
import argparse
optional_strs = ['Apple', 'Banana', 'Cranberry', 'Durian', 'Elderberry']
bash_parser = argparse.ArgumentParser(prog='base')
bash_parser.add_argument('option', choices=['load', 'export', 'reload'])
bash_parser.add_argument('-u', '--user', help='User name')
bash_parser.add_argument('-p', '--passwd', help='Password')
input_file = bash_parser.add_argument('-f', '--file', type=str, help='Input File')
if __name__ == '__main__':
from cmd2.argcomplete_bridge import bash_complete
# bash_complete flags this argument telling AutoCompleter to yield to bash to perform
# tab completion of a file path
bash_complete(input_file)
flag_opt = bash_parser.add_argument('-o', '--optional', help='Optional flag with choices')
setattr(flag_opt, 'arg_choices', optional_strs)
# Handle bash completion if it's installed
# This early check allows the script to bail out early to provide tab-completion results
# to the argcomplete library. Putting this at the end of the file would cause the full application
# to load fulfill every tab-completion request coming from bash. This can cause a notable delay
# on the bash prompt.
try:
# only move forward if we can import CompletionFinder and AutoCompleter
from cmd2.argcomplete_bridge import CompletionFinder
from cmd2.argparse_completer import AutoCompleter
import sys
if __name__ == '__main__':
completer = CompletionFinder()
# completer will return results to argcomplete and exit the script
completer(bash_parser, AutoCompleter(bash_parser))
except ImportError:
pass
# Intentionally below the bash completion code to reduce tab completion lag
import cmd2
class DummyApp(cmd2.Cmd):
"""
Dummy cmd2 app
"""
def __init__(self):
super().__init__()
if __name__ == '__main__':
args = bash_parser.parse_args()
# demonstrates some handling of the command line parameters
if args.user is None:
user = input('Username: ')
else:
user = args.user
if args.passwd is None:
import getpass
password = getpass.getpass()
else:
password = args.passwd
if args.file is not None:
print('Loading file: {}'.format(args.file))
# Clear the argumentns so cmd2 doesn't try to parse them
sys.argv = sys.argv[:1]
app = DummyApp()
app.cmdloop()
|
