U `{j@sUdZddlmZddlZddlmZddlZddlZddlmZGdddZ ej drz ddl Z Wnek red YnXe Zd ed <d d dddZnd d dddZdZdS)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). ) annotationsN)glob)Anyc@s4eZdZdZddddddZdd d d d d ZdS)FastFilesCompleterzFast file completer class.TboolNone) directoriesreturncCs ||_dSN)r)selfrr S/var/www/html/bwave/python/venv/lib/python3.8/site-packages/_pytest/_argcomplete.py__init__MszFastFilesCompleter.__init__strrz list[str])prefixkwargsr cKstj|ddkr*ttj|tj}nd}g}g}d|krrd|krr|rX|dtjkrj|t|d|d7}|t|t|D]*}tj|r|d7}| ||dq|S)Nr*?z.*/) osseplenpathdirnameextendrsortedisdirappend)r rrZ prefix_dir completionZglobbedxr r r __call__Ps  zFastFilesCompleter.__call__N)T)__name__ __module__ __qualname____doc__rr"r r r r rJsrZ _ARGCOMPLETErzFastFilesCompleter | Nonefilescompleterzargparse.ArgumentParserr)parserr cCstj|dddS)NF)Zalways_complete_options) argcomplete autocompleter(r r r try_argcompletemsr,cCsdSr r r+r r r r,rs)r& __future__rargparserrsystypingrrenvirongetZargcomplete.completersr) ImportErrorexitr'__annotations__r,r r r r s @