a
    sh                     @  s   d Z ddlmZ ddlZddlZddlZddlmZ ddlm	Z	 ddl
mZ ddlmZ ddlmZmZ dd	lmZ d
dddZdddddddZddddddZdS )zALogic that powers autocompletion installed by ``pip completion``.    )annotationsN)Iterable)chain)Any)create_main_parser)commands_dictcreate_command)get_default_environmentNone)returnc                    sH  dt jvrdS t jdr&t jds*dS t jd  dd tt jd } z| d   W n tyt   d Y n0 t }tt}g }d}D ]}||v r|} qq|dur|dkrt	
d  d o|d	v }|r*t }  fd
d|jddD }|r*|D ]}	t|	 qt	
d  d o>|dk}
|
rjt dD ]}t| qPt	
d t|}|jjD ]2jtjkrz|fddjj D 7 }qzdd d| d  D fdd|D } fdd|D }t| |jj}|rt |}dd |D }|D ]>}|d }|d rR|d dd dkrR|d7 }t| qtfdd| D s:| D ]}| rt| qndd |jD }||j t |} dr|D ]$jtjkr|jj 7 }qn t| |}|rtt |}td! fdd|D  t	
d dS )z:Entry Point for completion of main and subcommand options.PIP_AUTO_COMPLETEN
COMP_WORDS
COMP_CWORD    help-)show	uninstallc                   s0   g | ](}|j r|j  d d vr|j qS )r   N)canonical_name
startswith).0dist)cwordslc \/var/www/html/assistant/venv/lib/python3.9/site-packages/pip/_internal/cli/autocompletion.py
<listcomp>8   s   z autocomplete.<locals>.<listcomp>T)
local_onlyinstallpathc                   s   g | ]}| j fqS r   )nargs)r   opt_str)optr   r   r   P   s   c                 S  s   g | ]}| d d qS )=r   )splitr   xr   r   r   r   U       c                   s    g | ]\}}| vr||fqS r   r   )r   r'   v)	prev_optsr   r   r   V   r(   c                   s"   g | ]\}}|  r||fqS r   r   )r   kr)   currentr   r   r   X   r(   c                 S  s   g | ]}|d fqS )r   r   )r   r    r   r   r   r   c   r(   r      z--r$   c                 3  s   | ]}| v V  qd S Nr   )r   name)r   r   r   	<genexpr>l   r(   zautocomplete.<locals>.<genexpr>c                 S  s   g | ]
}|j qS r   )option_list)r   ir   r   r   r   s   r(    c                   s   g | ]}|  r|qS r   r+   r&   r-   r   r   r      r(   )"osenvirongetr%   int
IndexErrorr   listr   sysexitr   r	   lowerZiter_installed_distributionsprintauto_complete_pathsr   parseroption_list_allr   optparseSUPPRESS_HELP
_long_opts_short_optsget_path_completion_typeanyZhandler_mapoption_groupsappendr3   r   from_iterablejoin)cwordrA   subcommandsoptionssubcommand_namewordshould_list_installedenv	installedr   Zshould_list_installablesr    
subcommandcompletion_typepathsoption	opt_labelZhandler_nameoptsflattened_optsr   )r.   r   r   r#   r*   r   autocomplete   s    








 
r\   z	list[str]r9   zIterable[Any]z
str | None)r   rM   rZ   r   c                 C  s   |dk s| |d   dsdS |D ]n}|jtjkr4q"t|dD ]L}| |d  dd |krB|jrtdd |jdD rB|j    S qBq"dS )	aL  Get 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/   r   N/r$   r   c                 s  s   | ]}|d v V  qdS ))r    filedirNr   r&   r   r   r   r2      s   z+get_path_completion_type.<locals>.<genexpr>)r   r   rC   rD   strr%   metavarrH   )r   rM   rZ   r#   or   r   r   rG      s    

rG   r`   zIterable[str])r.   rV   r   c                 #  s   t j| \} t j|}t |t js.dS t j   fddt |D }|D ]`}t j||}t jt j||}|dkrt j	|r|V  qVt j
|rVt j|dV  qVdS )at  If ``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``)
    :return: A generator of regular files and/or directories
    Nc                 3  s$   | ]}t j| r|V  qd S r0   )r6   r    normcaser   r&   filenamer   r   r2      s   z&auto_complete_paths.<locals>.<genexpr>r_   r   )r6   r    r%   abspathaccessR_OKrc   listdirrL   isfileisdir)r.   rV   	directorycurrent_path	file_listfr#   	comp_filer   rd   r   r@      s    	
r@   )__doc__
__future__r   rC   r6   r<   collections.abcr   	itertoolsr   typingr   pip._internal.cli.main_parserr   pip._internal.commandsr   r   Zpip._internal.metadatar	   r\   rG   r@   r   r   r   r   <module>   s   s