python -m 模块的命令行自动完成

Question

是否可以获得 python -m package.subpackage.module 的命令行自动完成功能？

这与 python ./package/subpackage/module.py 类似，但不相同，它会自动完成目录和文件路径。然而，对于 -m，python 将库的模块作为具有适当命名空间和导入路径的脚本运行。

我希望能够 python -m package.s[TAB] 并自动完成 subpackage。

此功能是否内置于某处，或者我该如何设置？

Answer 1

如评论区所说，需要扩展bash-completion工具。然后，您将创建一个脚本来处理您需要的情况（即：当最后一个参数是 -m）时。

下面的这个小示例显示了自定义完成脚本的开始。我们将其命名为 python_completion.sh.

_python_target() {
    local cur prev opts

    # Retrieving the current typed argument
    cur="${COMP_WORDS[COMP_CWORD]}"

    # Retrieving the previous typed argument ("-m" for example)
    prev="${COMP_WORDS[COMP_CWORD-1]}"

    # Preparing an array to store available list for completions
    # COMREPLY will be checked to suggest the list
    COMPREPLY=()

    # Here, we'll only handle the case of "-m"
    # Hence, the classic autocompletion is disabled
    # (ie COMREPLY stays an empty array)
    if [[ "$prev" != "-m" ]]
    then
        return 0
    fi

    # Retrieving paths and converts their separators into dots
    # (if packages doesn't exist, same thing, empty array)
    if [[ ! -e "./package" ]]
    then
       return 0
    fi

    # Otherwise, we retrieve first the paths starting with "./package"
    # and converts their separators into dots
    opts="$(find ./package -type d | sed -e 's+/+.+g' -e 's/^\.//' | head)"

    # We store the whole list by invoking "compgen" and filling
    # COMREPLY with its output content.
    COMPREPLY=($(compgen -W "$opts" -- "$cur"))

}

complete -F _python_target python

（警告。这个脚本有一个缺陷，它不能处理包含空格的文件名）。要测试它，运行它在当前环境中：

. ./python_completion.sh

并测试它：

python -m packag[TAB]

Here 是继续这种方式的教程。

python -m 模块的命令行自动完成

Command-line autocompletion for python -m module

python

bash

autocomplete

bash-completion