o ڭj@sUdZddlZddlZddlZddlmZddlmZddlmZddlmZGdddZ ej d r^zddl Z WneyHed Ynwe Zee ed <d ejd dfddZdSd ejd dfddZdZdS)aAllow bash-completion for argparse with argcomplete if installed. Needs argcomplete>=0.5.6 for python 3.2/3.3 (older versions fail to find the magic string, so _ARGCOMPLETE env. var is never set, and this does not need special code). Function try_argcomplete(parser) should be called directly before the call to ArgumentParser.parse_args(). The filescompleter is what you normally would use on the positional arguments specification, in order to get "dirname/" after "dirn" instead of the default "dirname ": optparser.add_argument(Config._file_or_dir, nargs='*').completer=filescompleter Other, application specific, completers should go in the file doing the add_argument calls as they need to be specified as .completer attributes as well. (If argcomplete is not installed, the function the attribute points to will not be used). SPEEDUP ======= The generic argcomplete script for bash-completion (/etc/bash_completion.d/python-argcomplete.sh) uses a python program to determine startup script generated by pip. You can speed up completion somewhat by changing this script to include # PYTHON_ARGCOMPLETE_OK so the python-argcomplete-check-easy-install-script does not need to be called to find the entry point of the code and see if that is marked with PYTHON_ARGCOMPLETE_OK. INSTALL/DEBUGGING ================= To include this support in another application that has setup.py generated scripts: - Add the line: # PYTHON_ARGCOMPLETE_OK near the top of the main python entry point. - Include in the file calling parse_args(): from _argcomplete import try_argcomplete, filescompleter Call try_argcomplete just before parse_args(), and optionally add filescompleter to the positional arguments' add_argument(). If things do not work right away: - Switch on argcomplete debugging with (also helpful when doing custom completers): export _ARC_DEBUG=1 - Run: python-argcomplete-check-easy-install-script $(which appname) echo $? will echo 0 if the magic line has been found, 1 if not. - Sometimes it helps to find early on errors using: _ARGCOMPLETE=1 _ARC_DEBUG=1 appname which should throw a KeyError: 'COMPLINE' (which is properly set by the global argcomplete script). N)glob)Any)List)Optionalc@s>eZdZdZd deddfddZded edeefd d Z dS) FastFilesCompleterzFast file completer class.T directoriesreturnNcCs ||_dSN)r)selfrr Q/var/www/html/CropPilot/venv/lib/python3.10/site-packages/_pytest/_argcomplete.py__init__Ls zFastFilesCompleter.__init__prefixkwargscKstj|ddvrttj|tj}nd}g}g}d|vr9d|vr9|r,|dtjkr5|t|d|d7}|t|t|D]}tj|rP|d7}| ||dqD|S)Nr*?z.*/) osseplenpathdirnameextendrsortedisdirappend)r rr prefix_dir completionglobbedxr r r __call__Os  zFastFilesCompleter.__call__)T) __name__ __module__ __qualname____doc__boolr strrrr"r r r r rIsr _ARGCOMPLETErfilescompleterparserrcCstj|dddS)NF)always_complete_options) argcomplete autocompleter+r r r try_argcompletelsr0cCsdSr r r/r r r r0qs)r&argparsersysrtypingrrrrenvirongetargcomplete.completersr- ImportErrorexitr*__annotations__ArgumentParserr0r r r r s&?