U P^@s`dZddlZddlZddlZddlmZddlmZmZddl m Z ddZ dd Z d d Z dS) zBLogic that powers autocompletion installed by ``pip completion``. N)create_main_parser) commands_dictcreate_command)get_installed_distributionscsdtjkrdStjddd}ttjd}z||dWntk rZdYnXttg}zfdd|Dd }Wntk rd}YnXt}|r:|d krt d|d kö d  }|r>g} }t d dD].}|j |r|j |ddkr||j q|r>|D]}t|q$t dt|} | jjD]8} | jtjkrN| j| jD]} || | jfqlqNdd|d|dDfdd|D}fdd|D}t||| jj} | rt| }dd|D}|D]>} | d }| dr,| d dddkr,|d7}t|qndd|jD}||jdd|D} d r|D]$} | jtjkrt| j| j7qtnt|||} | rt| tdfddDt ddS)z?Entry Point for completion of main and subcommand options. PIP_AUTO_COMPLETEN COMP_WORDS COMP_CWORDcsg|]}|kr|qSr ).0w) subcommandsr A/tmp/pip-install-220asx0h/pip/pip/_internal/cli/autocompletion.py !sz autocomplete..rhelp)show uninstall-T) local_onlycSsg|]}|ddqS)=r)splitr xr r rrDscs g|]\}}|kr||fqSr r )r rv) prev_optsr rrEscs"g|]\}}|r||fqSr  startswith)r krcurrentr rrGs css|]}|dfVqdS)rNr )r optr r r Pszautocomplete..z--rcSsg|] }|jqSr ) option_list)r ir r rrZscss|]}|D] }|Vq qdSNr )r itor r rr"\s csg|]}|r|qSr rrrr rrgs )osenvironrint IndexErrorlistrrsysexitrlowerrkeyappendprintrparseroption_list_allroptparse SUPPRESS_HELP _long_opts _short_optsnargsget_path_completion_typeauto_complete_paths option_groupsr$join)cwordscwordoptionssubcommand_namer5should_list_installed installedlcdist subcommandr!opt_strcompletion_typeoption opt_labeloptsr )r rrr autocompletes              rNcCs|dks||ddsdS|D]n}|jtjkr4q"t|dD]L}||ddd|krB|jrtdd|jdDrB|jSqBq"dS) 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 rr r rr"zsz+get_path_completion_type..)rrr7r8strrmetavarany)r@rArMr!r(r r rr<ks  r<c#stj|\}tj|}t|tjs.dStjfddt|D}|D]`}tj||}tjtj||}|dkrtj |r|VqVtj |rVtj|dVqVdS)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|VqdSr&)r*rPnormcaserrfilenamer rr"sz&auto_complete_paths..rRr ) r*rPrabspathaccessR_OKrVlistdirr?isfileisdir)r rJ directory current_path file_listfr! comp_filer rWrr=s    r=)__doc__r7r*r/pip._internal.cli.main_parserrpip._internal.commandsrrpip._internal.utils.miscrrNr<r=r r r rs  [