63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
#!/usr/bin/env python
|
|
|
|
bashcode = r'''
|
|
# Run something, muting output or redirecting it to the debug stream
|
|
# depending on the value of _ARC_DEBUG.
|
|
__python_argcomplete_run() {
|
|
if [[ -z "$_ARC_DEBUG" ]]; then
|
|
"$@" 8>&1 9>&2 1>/dev/null 2>&1
|
|
else
|
|
"$@" 8>&1 9>&2 1>&9 2>&1
|
|
fi
|
|
}
|
|
|
|
_python_argcomplete() {
|
|
local IFS=$'\013'
|
|
local SUPPRESS_SPACE=0
|
|
if compopt +o nospace 2> /dev/null; then
|
|
SUPPRESS_SPACE=1
|
|
fi
|
|
COMPREPLY=( $(IFS="$IFS" \
|
|
COMP_LINE="$COMP_LINE" \
|
|
COMP_POINT="$COMP_POINT" \
|
|
COMP_TYPE="$COMP_TYPE" \
|
|
_ARGCOMPLETE_COMP_WORDBREAKS="$COMP_WORDBREAKS" \
|
|
_ARGCOMPLETE=1 \
|
|
_ARGCOMPLETE_SUPPRESS_SPACE=$SUPPRESS_SPACE \
|
|
__python_argcomplete_run "$1") )
|
|
if [[ $? != 0 ]]; then
|
|
unset COMPREPLY
|
|
elif [[ $SUPPRESS_SPACE == 1 ]] && [[ "$COMPREPLY" =~ [=/:]$ ]]; then
|
|
compopt -o nospace
|
|
fi
|
|
}
|
|
complete %(complete_opts)s -F _python_argcomplete "%(executable)s"
|
|
'''
|
|
|
|
tcshcode = '''\
|
|
complete "%(executable)s" 'p@*@`python-argcomplete-tcsh "%(executable)s"`@'
|
|
'''
|
|
|
|
def shellcode(executable, use_defaults=True, shell='bash', complete_arguments=None):
|
|
'''
|
|
Provide the shell code required to register a python executable for use with the argcomplete module.
|
|
|
|
:param str executable: Executable to be completed (when invoked exactly with this name
|
|
:param bool use_defaults: Whether to fallback to readline's default completion when no matches are generated.
|
|
:param str shell: Name of the shell to output code for (bash or tcsh)
|
|
:param complete_arguments: Arguments to call complete with
|
|
:type complete_arguments: list(str) or None
|
|
'''
|
|
|
|
if complete_arguments is None:
|
|
complete_options = '-o nospace -o default' if use_defaults else '-o nospace'
|
|
else:
|
|
complete_options = " ".join(complete_arguments)
|
|
|
|
if shell == 'bash':
|
|
code = bashcode
|
|
else:
|
|
code = tcshcode
|
|
|
|
return code % dict(complete_opts=complete_options, executable=executable)
|