-
-
Notifications
You must be signed in to change notification settings - Fork 129
/
register-python-argcomplete
executable file
·62 lines (43 loc) · 1.7 KB
/
register-python-argcomplete
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
#!/usr/bin/env python
# PYTHON_ARGCOMPLETE_OK
# Copyright 2012-2019, Andrey Kislyuk and argcomplete contributors.
# Licensed under the Apache License. See https://github.com/kislyuk/argcomplete for more info.
'''
Register a Python executable for use with the argcomplete module.
To perform the registration, source the output of this script in your bash shell
(quote the output to avoid interpolation).
Example:
$ eval "$(register-python-argcomplete my-favorite-script.py)"
For Tcsh
$ eval `register-python-argcomplete --shell tcsh my-favorite-script.py`
For Fish
$ register-python-argcomplete --shell fish my-favourite-script.py > ~/.config/fish/my-favourite-script.py.fish
'''
import sys
import argparse
import argcomplete
parser = argparse.ArgumentParser(
description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)
parser.add_argument(
'--no-defaults',
dest='use_defaults', action='store_false', default=True,
help='When no matches are generated, do not fallback to readline\'s default completion')
parser.add_argument(
'--complete-arguments',
nargs=argparse.REMAINDER,
help='arguments to call complete with; use of this option discards default options')
parser.add_argument(
'-s', '--shell',
choices=('bash', 'tcsh', 'fish'), default='bash',
help='output code for the specified shell')
parser.add_argument(
'executable',
nargs='+',
help='executable to completed (when invoked by exactly this name)')
argcomplete.autocomplete(parser)
if len(sys.argv) == 1:
parser.print_help()
sys.exit(1)
args = parser.parse_args()
sys.stdout.write(argcomplete.shellcode(
args.executable, args.use_defaults, args.shell, args.complete_arguments))