from __future__ import absolute_import

​

import sys

from pip.basecommand import Command

​

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 ) )

}

complete -o default -F _pip_completion pip

""", '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] ) )

}

compctl -K _pip_completion pip

"""}

​

​

class CompletionCommand(Command):

    """A helper command to be used for command completion."""

    name = 'completion'

    summary = 'A helper command used for command completion'

​

    def __init__(self, *args, **kw):

        super(CompletionCommand, self).__init__(*args, **kw)

​

        cmd_opts = self.cmd_opts

​

        cmd_opts.add_option(

            '--bash', '-b',