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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
#!/usr/bin/env python3
import argparse
from crazy_complete import argparse_mod
argp = argparse.ArgumentParser(prog='crazy-complete-test',
description='Test argument parser for shell completion')
argp.add_argument('--version', action='version')
# =============================================================================
# SUBCOMMANDS
# =============================================================================
subp = argp.add_subparsers(description='commands')
# =============================================================================
# Command 'complete'
# =============================================================================
cmdp = subp.add_parser('complete', help='Test complete commands')
cmdp.add_argument('--exec', help='Parse output').complete('exec', 'printf "%s\\t%s\\n" "Item 1" "Description 1" "Item 2" "Description 2"')
cmdp.add_argument('--exec-fast', help='Parse output (fast)').complete('exec_fast', 'printf "%s\\t%s\\n" "Item_1" "Description 1" "Item_2" "Description 2"')
cmdp.add_argument('--exec-internal', help='Execute internal code').complete('exec_internal', '_complete_internal')
cmdp.add_argument('--exec-large', help='Parse output (large date)').complete('exec', 'seq 1 99999')
cmdp.add_argument('--file', help='Complete a file').complete('file')
cmdp.add_argument('--directory', help='Complete a directory').complete('directory')
cmdp.add_argument('--file-tmp', help='Complete a file in /tmp').complete('file', {'directory': '/tmp'})
cmdp.add_argument('--directory-tmp', help='Complete a directory in /tmp').complete('directory', {'directory': '/tmp'})
cmdp.add_argument('--user', help='Complete a user').complete('user')
cmdp.add_argument('--group', help='Complete a group').complete('group')
cmdp.add_argument('--uid', help='Complete a uid').complete('uid')
cmdp.add_argument('--gid', help='Complete a gid').complete('gid')
cmdp.add_argument('--command', help='Complete a command').complete('command')
cmdp.add_argument('--process', help='Complete a process').complete('process')
cmdp.add_argument('--pid', help='Complete a pid').complete('pid')
cmdp.add_argument('--signal', help='Complete a signal').complete('signal')
cmdp.add_argument('--hostname', help='Complete a hostname').complete('hostname')
cmdp.add_argument('--variable', help='Complete a variable').complete('variable')
cmdp.add_argument('--environment', help='Complete a enviroment variable').complete('environment')
cmdp.add_argument('--service', help='Complete a service').complete('service')
cmdp.add_argument('--choices', help='Complete from choices', choices={'foo:bar': 'Description for foo:bar', 1:'one', 2:'two'})
cmdp.add_argument('--value-list', help='Complete a list').complete('value_list', {'values': ['foo', 'bar', 'baz']})
cmdp.add_argument('--value-list-2', help='Complete a list').complete('value_list', {'values': {'foo': 'Some foo', 'bar': 'Some bar'}})
cmdp.add_argument('--value-list-dup', help='Complete a list (with duplicates)').complete('value_list', {'values': ['foo', 'bar', 'baz'], 'duplicates': True})
cmdp.add_argument('--value-list-colon', help='Complete a list (colon)').complete('value_list', {'values': ['foo', 'bar', 'baz'], 'separator': ':'})
cmdp.add_argument('--range-1', help='Complete a range', type=int, choices=range(1,10))
cmdp.add_argument('--range-2', help='Complete a range', type=int, choices=range(1,10,2))
cmdp.add_argument('--history', help='Complete a history').complete('history', '[^"\' ]+@[^"\' ]+(/[^"\' ]+)?')
cmdp.add_argument('--commandline', help='Complete a command line').complete('commandline_string')
cmdp.add_argument('--mountpoint', help='Complete a mountpoint').complete('mountpoint')
cmdp.add_argument('--net', help='Complete a network interface').complete('net_interface')
cmdp.add_argument('--shell', help='Complete a login shell').complete('login_shell')
cmdp.add_argument('--locale', help='Complete a locale').complete('locale')
cmdp.add_argument('--charset', help='Complete a charset').complete('charset')
cmdp.add_argument('--timezone', help='Complete a timezone').complete('timezone')
cmdp.add_argument('--alsa-card', help='Complete an ALSA card').complete('alsa_card')
cmdp.add_argument('--alsa-device', help='Complete an ALSA device').complete('alsa_device')
cmdp.add_argument('--mime-image', help='Complete an image').complete('mime_file', 'image/')
cmdp.add_argument('--mime-video', help='Complete a video').complete('mime_file', 'video/')
cmdp.add_argument('--date', help='Complete a date').complete('date', '%Y-%m-%d')
cmdp.add_argument('--date-format', help='Complete a date format').complete('date_format')
cmdp.add_argument('--file-list', help='Complete a list of files').complete('file_list')
cmdp.add_argument('--directory-list', help='Complete a list of directories').complete('directory_list')
cmdp.add_argument('--filesystem-type', help='Complete a filesystem type').complete('filesystem_type')
cmdp.add_argument('--prefix', help='Complete with a prefix').complete('prefix', 'prefix:', ['file'])
cmdp.add_argument('--prefix-list', help='Complete with a prefix').complete('prefix', 'prefix:', ['file_list'])
cmdp.add_argument('--ip', help='Complete an ip address').complete('ip_address')
cmdp.add_argument('--ip-v4', help='Complete an ipv4 address').complete('ip_address', 'ipv4')
cmdp.add_argument('--ip-v6', help='Complete an ipv6 address').complete('ip_address', 'ipv6')
cmdp.add_argument('--integer', help='Complete an integer').complete('integer', {'suffixes': {'s': 'seconds', 'm': 'minutes', 'h': 'hours'}})
cmdp.add_argument('--float', help='Complete a float').complete('float', {'suffixes': {'s': 'seconds', 'm': 'minutes', 'h': 'hours'}})
# =============================================================================
# Command 'when'
# =============================================================================
cmdp = subp.add_parser('when', help='Test the "when"-feature')
cmdp.add_argument('--var', '-var', '-V', help='Conditional variable')
cmdp.add_argument('--optional', '-optional', '-O', help='Conditional variable', nargs='?')
cmdp.add_argument('--if-var', help='Only show option if --var is given').when('has_option --var -var -V')
cmdp.add_argument('--if-var-is-foo', help='Only show option if --var is foo or bar').when('option_is --var -var -V -- foo bar')
cmdp.add_argument('--if-optional', help='Only show option if --optional is given').when('has_option --optional -optional -O')
cmdp.add_argument('--if-optional-is-foo', help='Only show option if --optional is foo').when('option_is --optional -optional -O -- foo')
#cmdp.add_argument('--if-var-or-optional-is-foo').when('option_is --var --optional -- foo')
#cmdp.add_argument('--if-not-var', help='Only show option if --var is not given').when('not_has_option --var -v')
# =============================================================================
# Command 'argparse-actions'
# =============================================================================
cmdp = subp.add_parser('argparse-actions', help='argparse tool actions')
cmdp.add_argument('--store-true', help='A option flag', action='store_true')
cmdp.add_argument('--store-false', help='A option flag', action='store_false')
cmdp.add_argument('--store-const', help='A option flag', action='store_const', const='bar')
cmdp.add_argument('--append-const', help='A option flag', action='append_const', const='bar')
cmdp.add_argument('--append', help='A option flag', action='append')
cmdp.add_argument('--count', help='A option flag', action='count')
cmdp.add_argument('--extend', help='A option flag', action='extend')
# =============================================================================
# Command 'subcommand'
# =============================================================================
cmdp = subp.add_parser('subcommand', help='Test nested subcommands')
cmdp.add_argument('--subcommand-choices', help='Complete from choices', choices=(1,'two and a half',3))
subp1 = cmdp.add_subparsers(description='commands')
cmdp1 = subp1.add_parser('sub-subcommand', help='Nested subcommand')
cmdp1.add_argument('--sub-subcommand-choices', help='Complete from choices', choices=(1,'two and a half',3))
# =============================================================================
# Command 'test'
# =============================================================================
cmdp = subp.add_parser('test', help='For testing the completer').aliases(['alias1', 'alias2'])
cmdp.add_argument('-F', '-flag', '--flag', help='A option flag', action='store_true')
cmdp.add_argument('-O', '-optional', '--optional', help='Option with optional arg', nargs='?', choices=(1,2,3))
cmdp.add_argument('-A', '-arg', '--arg', help='Option with arg', choices=(1,2,3))
cmdp.add_argument('--special-chars-in-description', help='Here are some special chars: $"\'\\[]*`)')
cmdp.add_argument('--repeatable-flag', action='store_true').set_repeatable()
cmdp.add_argument('--repeatable-arg').set_repeatable()
cmdp.add_argument('-H', '-hidden', '--hidden', help=argparse.SUPPRESS, choices=[1,2,3])
cmdp.add_argument('--hidden-flag', help=argparse.SUPPRESS, action='store_true')
group = cmdp.add_mutually_exclusive_group()
group.add_argument('--exclusive-1', action='store_true')
group.add_argument('--exclusive-2', action='store_true')
cmdp.add_argument('positional_1', help='First positional', choices=('first1', 'first2', 'first3'))
cmdp.add_argument('positional_2', help='Second positional', choices=('second1', 'second2'))
cmdp.add_argument('positional_3', help='Repeated positional', choices=('repeated1', 'repeated2'), nargs='+')
if __name__ == '__main__':
argp.parse_args()
|
