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

_python_modules()
{
    COMPREPLY+=($(compgen -W \
        "$(${1:-python} ${BASH_SOURCE[0]%/*}/../helpers/python $cur \
            2>/dev/null)" -- "$cur"))
}

_python_warning_actions()
{
    COMPREPLY+=($(compgen -W "ignore default all module once error" \
        ${prefix:+-P "$prefix"} -- "$cur"))
}

_python()
{
    local cur prev words cword prefix
    _init_completion || return

    case $cur in
        -[QWX]?*)
            prefix=${cur:0:2}
            prev=$prefix
            cur=${cur:2}
            ;;
    esac

    case $prev in
        --help | --version | -!(-*)[?hVcX])
            return
            ;;
        -!(-*)m)
            _python_modules "$1"
            return
            ;;
        -!(-*)Q)
            COMPREPLY=($(compgen -W "old new warn warnall" -P "$prefix" \
                -- "$cur"))
            return
            ;;
        -!(-*)W)
            _python_warning_actions
            return
            ;;
        --jit)
            # TODO: quite a few others, parse from "--jit help" output?
            COMPREPLY=($(compgen -W "help off" -- "$cur"))
            return
            ;;
        !(?(*/)python*([0-9.])|?(*/)pypy*([0-9.])|-?))
            [[ $cword -lt 2 || ${words[cword - 2]} != -[QWX] ]] && _filedir
            ;;
    esac

    # if -c or -m is already given, complete all kind of files.
    if [[ ${words[*]::cword} == *\ -[cm]\ * ]]; then
        _filedir
    elif [[ $cur != -* ]]; then
        _filedir 'py?([cowz])'
    else
        COMPREPLY=($(compgen -W '$(_parse_help "$1" -h)' -- "$cur"))
    fi
} &&
    complete -F _python python python2 python2.7 python3 python3.{3..9} pypy pypy3 micropython

# ex: filetype=sh