# bash completion for python                               -*- shell-script -*-

proc _python_modules
{
    setglobal COMPREPLY = ''( $( compgen -W "$( ${1:-python} -c 'import pkgutil
for mod in pkgutil.iter_modules(): print(mod[1])' )" 2>/dev/null -- "$cur" ) )
}

proc _python
{
    local cur prev words cword
    _init_completion || return

    match $prev {
        with -'?'|-h|--help|-V|--version|-c|-X
            return 0
            
        with -m
            _python_modules $1
            return 0
            
        with -Q
            setglobal COMPREPLY = ''( $( compgen -W "old new warn warnall" -- "$cur" ) )
            return 0
            
        with -W
            setglobal COMPREPLY = ''( $( compgen -W "ignore default all module once error" \
                -- "$cur" ) )
            return 0
            
        with !(?(*/)python*([0-9.])|?(*/)pypy*([0-9.])|-?)
            [[ $cword -lt 2 || ${words[cword-2]} != -[QWX] ]] && _filedir
            
    }


    # if '-c' is already given, complete all kind of files.
    local i
    for (( i=0; i < ${#words[@]}-1; i++ )); do
        if [[ ${words[i]} == -c ]]; then
            _filedir
        fi
    done


    if [[ "$cur" != -* ]] {
        _filedir 'py?([co])'
    } else {
        setglobal COMPREPLY = ''( $( compgen -W '$( _parse_help "$1" -h )' -- "$cur" ) )
    }

    return 0
} &&
complete -F _python python python2 python3 pypy

# ex: ts=4 sw=4 et filetype=sh