B Uvg@s`dZddlZddlZddlZddlmZddlmZmZddl m Z ddZ dd Z d d Z dS) zBLogic that powers autocompletion installed by ``pip completion``. N)create_main_parser) commands_dict get_summaries)get_installed_distributionscsdtjkrdStjddd}ttjd}y||dWntk rZdYnXddtDg}yfd d|Dd }Wntk rd}YnXt}|rX|d krtd|d koԈ d  }|rNg} }x} | d }| drH| d dddkrH|d7}t|qWndd|jD}| |jdd|D} d rxJ|D]$} | jtjkr| j| j7qWnt|||} | rt| tdfddDtddS)z?Entry Point for completion of main and subcommand options. PIP_AUTO_COMPLETEN COMP_WORDS COMP_CWORDcSsg|] \}}|qSr ).0cmdsummaryr r A/tmp/pip-install-r_9ig3yj/pip/pip/_internal/cli/autocompletion.py sz autocomplete..csg|]}|kr|qSr r )r w) subcommandsr rrsrhelp)show uninstall-T) local_onlycSsg|]}|ddqS)=r)split)r xr r rrAscs g|]\}}|kr||fqSr r )r rv) prev_optsr rrBscs"g|]\}}|r||fqSr ) startswith)r kr)currentr rrDscss|]}|dfVqdS)rNr )r optr r r Mszautocomplete..z--rcSsg|] }|jqSr ) option_list)r ir r rrWscss|]}|D] }|Vq qdS)Nr )r itor r rr!Ys csg|]}|r|qSr )r)r r)rr rrds)osenvironrint IndexErrorrrsysexitrlowerrkeyappendprintrparseroption_list_allroptparse SUPPRESS_HELP _long_opts _short_optsnargsget_path_completion_typeauto_complete_paths option_groupsr#join)cwordscwordoptionssubcommand_namer2should_list_installed installedlcdist subcommandr opt_strcompletion_typeoption opt_labeloptsr )rrrr autocomplete sx                rKcCs|dks||ddsdSxr|D]j}|jtjkr6q$xVt|dD]D}||ddd|krF|jrtdd|jdDrF|jSqFWq$WdS) aLGet the type of path completion (``file``, ``dir``, ``path`` or None) :param cwords: same as the environmental variable ``COMP_WORDS`` :param cword: same as the environmental variable ``COMP_CWORD`` :param opts: The available options to check :return: path completion type (``file``, ``dir``, ``path`` or None) r"rN/rrcss|]}|dkVqdS))pathfiledirNr )r rr r rr!xsz+get_path_completion_type..)rrr4r5strrmetavarany)r=r>rJr r&r r rr9hs  r9c#stj|\}tj|}t|tjs.dStjfddt|D}xh|D]`}tj||}tjtj||}|dkrtj |r|VqXtj |rXtj|dVqXWdS)aoIf ``completion_type`` is ``file`` or ``path``, list all regular files and directories starting with ``current``; otherwise only list directories starting with ``current``. :param current: The word to be completed :param completion_type: path completion type(`file`, `path` or `dir`)i :return: A generator of regular files and/or directories Nc3s$|]}tj|r|VqdS)N)r(rMnormcaser)r r)filenamer rr!sz&auto_complete_paths..rOr ) r(rMrabspathaccessR_OKrSlistdirr<isfileisdir)rrG directory current_path file_listfr comp_filer )rTrr:}s     r:)__doc__r4r(r,pip._internal.cli.main_parserrpip._internal.commandsrrpip._internal.utils.miscrrKr9r:r r r rs  [