97 lines
2.9 KiB
Python
97 lines
2.9 KiB
Python
|
# The following comment should be removed at some point in the future.
|
||
|
|
# mypy: disallow-untyped-defs=False
|
||
|
|
|
||
|
|
from __future__ import absolute_import
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import textwrap
|
||
|
|
|
||
|
|
from pip._internal.cli.base_command import Command
|
||
|
|
from pip._internal.utils.misc import get_prog
|
||
|
|
|
||
|
|
BASE_COMPLETION = """
|
||
|
|
# pip %(shell)s completion start%(script)s# pip %(shell)s completion end
|
||
|
|
"""
|
||
|
|
|
||
|
|
COMPLETION_SCRIPTS = {
|
||
|
|
'bash': """
|
||
|
|
_pip_completion()
|
||
|
|
{
|
||
|
|
COMPREPLY=( $( COMP_WORDS="${COMP_WORDS[*]}" \\
|
||
|
|
COMP_CWORD=$COMP_CWORD \\
|
||
|
|
PIP_AUTO_COMPLETE=1 $1 2>/dev/null ) )
|
||
|
|
}
|
||
|
|
complete -o default -F _pip_completion %(prog)s
|
||
|
|
""",
|
||
|
|
'zsh': """
|
||
|
|
function _pip_completion {
|
||
|
|
local words cword
|
||
|
|
read -Ac words
|
||
|
|
read -cn cword
|
||
|
|
reply=( $( COMP_WORDS="$words[*]" \\
|
||
|
|
COMP_CWORD=$(( cword-1 )) \\
|
||
|
|
PIP_AUTO_COMPLETE=1 $words[1] 2>/dev/null ))
|
||
|
|
}
|
||
|
|
compctl -K _pip_completion %(prog)s
|
||
|
|
""",
|
||
|
|
'fish': """
|
||
|
|
function __fish_complete_pip
|
||
|
|
set -lx COMP_WORDS (commandline -o) ""
|
||
|
|
set -lx COMP_CWORD ( \\
|
||
|
|
math (contains -i -- (commandline -t) $COMP_WORDS)-1 \\
|
||
|
|
)
|
||
|
|
set -lx PIP_AUTO_COMPLETE 1
|
||
|
|
string split \\ -- (eval $COMP_WORDS[1])
|
||
|
|
end
|
||
|
|
complete -fa "(__fish_complete_pip)" -c %(prog)s
|
||
|
|
""",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
class CompletionCommand(Command):
|
||
|
|
"""A helper command to be used for command completion."""
|
||
|
|
|
||
|
|
ignore_require_venv = True
|
||
|
|
|
||
|
|
def __init__(self, *args, **kw):
|
||
|
|
super(CompletionCommand, self).__init__(*args, **kw)
|
||
|
|
|
||
|
|
cmd_opts = self.cmd_opts
|
||
|
|
|
||
|
|
cmd_opts.add_option(
|
||
|
|
'--bash', '-b',
|
||
|
|
action='store_const',
|
||
|
|
const='bash',
|
||
|
|
dest='shell',
|
||
|
|
help='Emit completion code for bash')
|
||
|
|
cmd_opts.add_option(
|
||
|
|
'--zsh', '-z',
|
||
|
|
action='store_const',
|
||
|
|
const='zsh',
|
||
|
|
dest='shell',
|
||
|
|
help='Emit completion code for zsh')
|
||
|
|
cmd_opts.add_option(
|
||
|
|
'--fish', '-f',
|
||
|
|
action='store_const',
|
||
|
|
const='fish',
|
||
|
|
dest='shell',
|
||
|
|
help='Emit completion code for fish')
|
||
|
|
|
||
|
|
self.parser.insert_option_group(0, cmd_opts)
|
||
|
|
|
||
|
|
def run(self, options, args):
|
||
|
|
"""Prints the completion code of the given shell"""
|
||
|
|
shells = COMPLETION_SCRIPTS.keys()
|
||
|
|
shell_options = ['--' + shell for shell in sorted(shells)]
|
||
|
|
if options.shell in shells:
|
||
|
|
script = textwrap.dedent(
|
||
|
|
COMPLETION_SCRIPTS.get(options.shell, '') % {
|
||
|
|
'prog': get_prog(),
|
||
|
|
}
|
||
|
|
)
|
||
|
|
print(BASE_COMPLETION % {'script': script, 'shell': options.shell})
|
||
|
|
else:
|
||
|
|
sys.stderr.write(
|
||
|
|
'ERROR: You must pass %s\n' % ' or '.join(shell_options)
|
||
|
|
)
|