U
    @vge                     @   s  d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	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m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! d dl"m#Z#m$Z$m%Z% d dl&m'Z'm(Z(m)Z)m*Z* d dl+m,Z,m-Z- d dl.m/Z/m0Z0 er4d dlm1Z2 nd dlm2Z2 e,rd dl3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z< d dl=m>Z> e;e?e?e?f Z@ddddddddddddd d!gZAeBeCZDd"d# ZEd$d% ZFd&d ZGd'd ZHed(d)d*dd,dZId-d. ZJd/d0 ZKd1d ZLdd3dZMd4d5 ZNd6d7 ZOd8d ZPd9d: ZQd;d< ZRd=d ZSd>d ZTejUfd?d@ZVddBdZWdCd ZXdDd ZYdEdF ZZdGdH Z[dIdJ Z\dKdL Z]dMdN Z^dAe)dAd+d+dfdOdPZ_dQdR Z`dSdT ZadUdV ZbG dWdX dXecZdG dYdZ dZe2Zeejfd[d\ Zgd]d Zhd^d_ ZiG d`da daecZjddbd Zkdcdd Zldedf Zmdgdh ZnddjdkZodldm Zpdndo Zqdpdq Zrdrds Zsdtdu Ztdvdw Zudxdy Zvdzd! Zwd{d| ZxG d}d~ d~ecZydd Zzdd Z{dd Z|dd Z}dddZ~dd ZdS )    )absolute_importNdeque)pkg_resources)retry)PY2	text_type)input)parse)unquote)__version__)CommandError)get_major_minor_versionsite_packages	user_site)WINDOWS
expanduserstdlib_pkgsstr_to_display)MYPY_CHECK_RUNNINGcast)running_under_virtualenvvirtualenv_no_global)BytesIO)StringIO)	AnyAnyStr	ContainerIterableListOptionalTextTupleUnion)Distributionrmtreedisplay_path
backup_dirasksplitextformat_sizeis_installable_dirnormalize_pathrenamesget_progcaptured_stdout
ensure_dirget_installed_versionremove_auth_from_urlc                  C   s4   t jt jtdd} t j| } dt| t S )Nz..zpip {} from {} (python {}))	ospathjoindirname__file__abspathformatr   r   )pip_pkg_dir r;   </tmp/pip-unpacked-wheel-gw11q0wt/pip/_internal/utils/misc.pyget_pip_versionK   s      r=   c                 C   sD   t | dk r"| dt |  d 7 } nt | dkr:| dd } td| S )ax  
    Convert a tuple of ints representing a Python version to one of length
    three.

    :param py_version_info: a tuple of ints representing a Python version,
        or None to specify no version. The tuple can have any length.

    :return: a tuple of length three if `py_version_info` is non-None.
        Otherwise, return `py_version_info` unchanged (i.e. None).
       )r   NVersionInfo)lenr   )py_version_infor;   r;   r<   normalize_version_infoW   s
    rB   c              
   C   sP   zt |  W n< tk
rJ } z|jtjkr:|jtjkr: W 5 d}~X Y nX dS )z os.path.makedirs without EEXIST.N)r3   makedirsOSErrorerrnoEEXIST	ENOTEMPTY)r4   er;   r;   r<   r0   k   s
    c               
   C   sP   z0t jtjd } | dkr(dtj W S | W S W n tttfk
rJ   Y nX dS )Nr   )z__main__.pyz-cz	%s -m pippip)	r3   r4   basenamesysargv
executableAttributeError	TypeError
IndexError)progr;   r;   r<   r.   v   s    
i  i  )stop_max_delay
wait_fixedFc                 C   s   t j| |td d S )N)ignore_errorsonerror)shutilr%   rmtree_errorhandler)dirrT   r;   r;   r<   r%      s    c              	   C   sX   zt |jtj@  }W n ttfk
r2   Y dS X |rRt |tj | | dS  dS )zOn Windows, the files in .svn are read-only, so when rmtree() tries to
    remove them, an exception is thrown.  We catch that here, remove the
    read-only attribute, and hopefully continue without problems.N)r3   statst_modeS_IWRITEIOErrorrD   chmod)funcr4   exc_infohas_attr_readonlyr;   r;   r<   rW      s    rW   c                 C   sd   | dkrdS t | tr| S z| t d}W n0 tk
r^   trRtd| }nt	| }Y nX |S )z
    Convert a bytes (or text) path to text (unicode in Python 2) for display
    and logging purposes.

    This function should never error out. Also, this function is mainly needed
    for Python 2 since in Python 3 str paths are already text.
    Nstrictzb{!r})

isinstancer   decoderK   getfilesystemencodingUnicodeDecodeErrorr   r   r9   ascii)r4   r&   r;   r;   r<   path_to_display   s    	
rg   c                 C   st   t jt j| } tjd dkrB| t d} | t	 d} | 
t  t jj rpd| tt  d  } | S )zTGives the display value for a given path, making it relative to cwd
    if possible.r      replace.N)r3   r4   normcaser8   rK   version_inforc   rd   encodegetdefaultencoding
startswithgetcwdsepr@   )r4   r;   r;   r<   r&      s    .bakc                 C   s6   d}|}t j| | r.|d7 }|t| }q| | S )z\Figure out the name of a directory to back up the given dir to
    (adding .bak, .bak2, etc)   )r3   r4   existsstr)rX   extn	extensionr;   r;   r<   r'      s    c                 C   s2   t jdd D ]}||kr|  S qt| |S )NPIP_EXISTS_ACTION )r3   environgetsplitr(   )messageoptionsactionr;   r;   r<   ask_path_exists   s    
r   c                 C   s   t jdrtd|  dS )z&Raise an error if no input is allowed.PIP_NO_INPUTz7No input was expected ($PIP_NO_INPUT set); question: %sN)r3   r{   r|   	Exceptionr~   r;   r;   r<   _check_no_input   s    r   c                 C   sF   t |  t| }|  }||kr<td|d|f  q |S q dS )z@Ask the message interactively, with the given possible responsesz<Your response (%r) was not one of the expected responses: %sz, N)r   r	   striplowerprintr5   )r~   r   responser;   r;   r<   r(      s    c                 C   s   t |  t| S )zAsk for input interactively.)r   r	   r   r;   r;   r<   	ask_input   s    r   c                 C   s   t |  t| S )z!Ask for a password interactively.)r   getpassr   r;   r;   r<   ask_password  s    r   c                 C   sL   | dkrd| d d  S | dkr,d| d  S | dkr@d| d  S d|  S d S )	Ni@B z%.1f MBg     @@i  i'  z%i kBz%.1f kBz%i bytesr;   )bytesr;   r;   r<   r*   
  s    c                 C   sP   t j| sdS t j| d}t j|r.dS t j| d}t j|rLdS dS )zBIs path is a directory containing setup.py or pyproject.toml?
    Fzsetup.pyTzpyproject.toml)r3   r4   isdirr5   isfile)r4   setup_pypyproject_tomlr;   r;   r<   r+     s    c                 c   s   |  |}|sq|V  q dS )z7Yield pieces of data from a file-like object until EOF.N)read)filesizechunkr;   r;   r<   read_chunks%  s    
r   Tc                 C   s2   t | } |rtj| } ntj| } tj| S )zN
    Convert a path to its canonical, case-normalized, absolute version.

    )r   r3   r4   realpathr8   rk   )r4   resolve_symlinksr;   r;   r<   r,   .  s
    c                 C   s@   t | \}}| dr8|dd | }|dd }||fS )z,Like os.path.splitext, but take off .tar tooz.tarN)	posixpathr)   r   endswith)r4   baserv   r;   r;   r<   r)   <  s
    c                 C   sz   t j|\}}|r.|r.t j|s.t | t| | t j| \}}|rv|rvzt | W n tk
rt   Y nX dS )z7Like os.renames(), but handles renaming across devices.N)	r3   r4   r}   rt   rC   rV   move
removedirsrD   )oldnewheadtailr;   r;   r<   r-   F  s    
c                 C   sp   t | } t tj}t r(| t tjS ddlm} | |rh|d D ]}| t |rJ dS qJdS dS dS )a  
    Return True if this is a path pip is allowed to modify.

    If we're in a virtualenv, sys.prefix points to the virtualenv's
    prefix; only sys.prefix is considered local.

    If we're not in a virtualenv, in general we can modify anything.
    However, if the OS vendor has configured distutils to install
    somewhere other than sys.prefix (which could be a subdirectory of
    sys.prefix, e.g. /usr/local), we consider sys.prefix itself nonlocal
    and the domain of the OS vendor. (In other words, everything _other
    than_ sys.prefix is considered local.)

    Caution: this function assumes the head of path has been normalized
    with normalize_path.
    r   )distutils_schemerz   TFN)r,   rK   prefixr   ro   pip._internal.locationsr   values)r4   r   r   
local_pathr;   r;   r<   is_localX  s    

r   c                 C   s   t t| S )zh
    Return True if given Distribution object is installed somewhere pip
    is allowed to modify.

    )r   dist_locationdistr;   r;   r<   dist_is_local{  s    r   c                 C   s   t | ttS )zF
    Return True if given Distribution is installed in user site.
    )r   ro   r,   r   r   r;   r;   r<   dist_in_usersite  s    r   c                 C   s   t | ttS )z[
    Return True if given Distribution is installed in
    sysconfig.get_python_lib().
    )r   ro   r,   r   r   r;   r;   r<   dist_in_site_packages  s    r   c                 C   s6   t jD ]*}tj|| jd }tj|r dS qdS )zC
    Return True if given Distribution is an editable install.
    	.egg-linkTF)rK   r4   r3   r5   project_namer   )r   	path_itemegg_linkr;   r;   r<   dist_is_editable  s
    
r   c                    s   |rt |}nt j}| r tndd |r6dd  ndd  |rLdd ndd |r^tnd	d
  fdd|D S )a^  
    Return a list of installed Distribution objects.

    If ``local_only`` is True (default), only return installations
    local to the current virtualenv, if in a virtualenv.

    ``skip`` argument is an iterable of lower-case project names to
    ignore; defaults to stdlib_pkgs

    If ``include_editables`` is False, don't report editables.

    If ``editables_only`` is True , only report editables.

    If ``user_only`` is True , only report installations in the user
    site directory.

    If ``paths`` is set, only report the distributions present at the
    specified list of locations.
    c                 S   s   dS NTr;   dr;   r;   r<   
local_test  s    z/get_installed_distributions.<locals>.local_testc                 S   s   dS r   r;   r   r;   r;   r<   editable_test  s    z2get_installed_distributions.<locals>.editable_testc                 S   s
   t |  S Nr   r   r;   r;   r<   r     s    c                 S   s   t | S r   r   r   r;   r;   r<   editables_only_test  s    z8get_installed_distributions.<locals>.editables_only_testc                 S   s   dS r   r;   r   r;   r;   r<   r     s    c                 S   s   dS r   r;   r   r;   r;   r<   	user_test  s    z.get_installed_distributions.<locals>.user_testc                    s:   g | ]2}|r|j kr |r|r|r|qS r;   )key).0r   r   r   r   skipr   r;   r<   
<listcomp>  s    
z/get_installed_distributions.<locals>.<listcomp>)r   
WorkingSetworking_setr   r   )
local_onlyr   include_editableseditables_only	user_onlypathsr   r;   r   r<   get_installed_distributions  s     

r   c                 C   sx   g }t  r*|t t sBtrB|t ntr8|t |t |D ],}tj|| jd }tj	|rF|  S qFdS )a  
    Return the path for the .egg-link file if it exists, otherwise, None.

    There's 3 scenarios:
    1) not in a virtualenv
       try to find in site.USER_SITE, then site_packages
    2) in a no-global virtualenv
       try to find in site_packages
    3) in a yes-global virtualenv
       try to find in site_packages, then site.USER_SITE
       (don't look in global location)

    For #1 and #3, there could be odd cases, where there's an egg-link in 2
    locations.

    This method will just return the first one found.
    r   N)
r   appendr   r   r   r3   r4   r5   r   r   )r   sitessiteegglinkr;   r;   r<   egg_link_path  s    




r   c                 C   s   t | }|rt|S t| jS )aO  
    Get the site-packages location of this distribution. Generally
    this is dist.location, except in the case of develop-installed
    packages, where dist.location is the source code location, and we
    want to know where the egg-link file is.

    The returned location is normalized (in particular, with symlinks removed).
    )r   r,   location)r   r   r;   r;   r<   r   
  s    
r   c                 G   s   t j| f|  d S r   )loggerinfo)msgargsr;   r;   r<   write_output  s    r   c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	FakeFilezQWrap a list of lines in an object with readline() to make
    ConfigParser happy.c                 C   s   dd |D | _ d S )Nc                 s   s   | ]
}|V  qd S r   r;   )r   lr;   r;   r<   	<genexpr>#  s     z$FakeFile.__init__.<locals>.<genexpr>_gen)selflinesr;   r;   r<   __init__"  s    zFakeFile.__init__c                 C   sP   z4zt | jW W S  tk
r0   | j   Y W S X W n tk
rJ   Y dS X d S )Nrz   )nextr   	NameErrorStopIterationr   r;   r;   r<   readline%  s    zFakeFile.readlinec                 C   s   | j S r   r   r   r;   r;   r<   __iter__.  s    zFakeFile.__iter__N)__name__
__module____qualname____doc__r   r   r   r;   r;   r;   r<   r     s   	r   c                   @   s$   e Zd Zedd Zedd ZdS )StreamWrapperc                 C   s   || _ |  S r   )orig_stream)clsr   r;   r;   r<   from_stream4  s    zStreamWrapper.from_streamc                 C   s   | j jS r   )r   encodingr   r;   r;   r<   r   :  s    zStreamWrapper.encodingN)r   r   r   classmethodr   propertyr   r;   r;   r;   r<   r   2  s   
r   c              
   c   s@   t t| }tt| t| zt t| V  W 5 tt| | X dS )zReturn a context manager used by captured_stdout/stdin/stderr
    that temporarily replaces the sys stream *stream_name* with a StringIO.

    Taken from Lib/support/__init__.py in the CPython repo.
    N)getattrrK   setattrr   r   )stream_nameorig_stdoutr;   r;   r<   captured_output?  s
    
r   c                   C   s   t dS )zCapture the output of sys.stdout:

       with captured_stdout() as stdout:
           print('hello')
       self.assertEqual(stdout.getvalue(), 'hello
')

    Taken from Lib/support/__init__.py in the CPython repo.
    stdoutr   r;   r;   r;   r<   r/   N  s    	c                   C   s   t dS )z 
    See captured_stdout().
    stderrr   r;   r;   r;   r<   captured_stderrZ  s    r   c                   @   s    e Zd ZdZdd Zdd ZdS )cached_propertyzA property that is only computed once per instance and then replaces
       itself with an ordinary attribute. Deleting the attribute resets the
       property.

       Source: https://github.com/bottlepy/bottle/blob/0.11.5/bottle.py#L175
    c                 C   s   t |d| _|| _d S )Nr   )r   r   r^   )r   r^   r;   r;   r<   r   i  s    zcached_property.__init__c                 C   s(   |d kr| S |  | }|j| j j< |S r   )r^   __dict__r   )r   objr   valuer;   r;   r<   __get__m  s    zcached_property.__get__N)r   r   r   r   r   r   r;   r;   r;   r<   r   a  s   r   c                 C   s4   t j| }|dkrt  }||}|r0|jS dS )zCGet the installed version of dist_name avoiding pkg_resources cacheN)r   Requirementr
   r   findversion)	dist_namer   reqr   r;   r;   r<   r1   u  s
    
c                 C   s   t | dd dS )zConsume an iterable at C speed.r   )maxlenNr   )iteratorr;   r;   r<   consume  s    r  c                  O   s@   t t| tt| f|}dd | D }||d< tdd|S )Nc                 S   s   i | ]\}}||qS r;   r;   )r   r   r   r;   r;   r<   
<dictcomp>  s      zenum.<locals>.<dictcomp>reverse_mappingEnumr;   )dictzipranger@   itemstype)
sequentialnamedenumsreverser;   r;   r<   enum  s    r  c                 C   s*   |dkr| S d| krd | } d | |S )z.
    Build a netloc from a host-port pair
    N:[{}]z{}:{})r9   )hostportr;   r;   r<   build_netloc  s
    
r  httpsc                 C   s4   |  ddkr(d| kr(d| kr(d| } d|| S )z)
    Build a full URL from a netloc.
    r  rh   @[r  z{}://{})countr9   )netlocschemer;   r;   r<   build_url_from_netloc  s    
r  c                 C   s   t | }t|}|j|jfS )z2
    Return the host-port pair from a netloc.
    )r  urllib_parseurlparsehostnamer  )r  urlparsedr;   r;   r<   parse_netloc  s    
r"  c                 C   sX   d| kr| dfS |  dd\}} d|kr6|dd}n|df}tdd |D }| |fS )zp
    Parse out and remove the auth information from a netloc.

    Returns: (netloc, (username, password)).
    r  )NNrs   r  Nc                 s   s"   | ]}|d krd nt |V  qd S r   )urllib_unquote)r   xr;   r;   r<   r     s    z)split_auth_from_netloc.<locals>.<genexpr>)rsplitr}   tuple)r  auth	user_passr;   r;   r<   split_auth_from_netloc  s    r)  c                 C   sL   t | \} \}}|dkr| S |dkr.d}d}nt|}d}dj||| dS )z
    Replace the sensitive data in a netloc with "****", if it exists.

    For example:
        - "user:pass@example.com" returns "user:****@example.com"
        - "accesstoken@example.com" returns "****@example.com"
    N****rz   z:****z{user}{password}@{netloc})userpasswordr  )r)  r  quoter9   )r  r+  r,  r;   r;   r<   redact_netloc  s    	
r.  c                 C   s@   t | }||j}|j|d |j|j|jf}t |}||fS )aR  Transform and replace netloc in a url.

    transform_netloc is a function taking the netloc and returning a
    tuple. The first element of this tuple is the new netloc. The
    entire tuple is returned.

    Returns a tuple containing the transformed url as item 0 and the
    original tuple returned by transform_netloc as item 1.
    r   )r  urlsplitr  r  r4   queryfragment
urlunsplit)r   transform_netlocpurlnetloc_tuple
url_piecessurlr;   r;   r<   _transform_url  s    


    
r8  c                 C   s   t | S r   )r)  r  r;   r;   r<   _get_netloc  s    r:  c                 C   s
   t | fS r   )r.  r9  r;   r;   r<   _redact_netloc  s    r;  c                 C   s   t | t\}\}}|||fS )z
    Parse a url into separate netloc, auth, and url with no auth.

    Returns: (url_without_auth, netloc, (username, password))
    r8  r:  )r   url_without_authr  r'  r;   r;   r<   split_auth_netloc_from_url  s    r>  c                 C   s   t | td S )z7Return a copy of url with 'username:password@' removed.r   r<  r   r;   r;   r<   r2     s    c                 C   s   t | td S )z.Replace the password in a given url with ****.r   )r8  r;  r?  r;   r;   r<   redact_auth_from_url  s    r@  c                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )
HiddenTextc                 C   s   || _ || _d S r   )secretredacted)r   rB  rC  r;   r;   r<   r     s    zHiddenText.__init__c                 C   s   d t| S )Nz<HiddenText {!r}>)r9   ru   r   r;   r;   r<   __repr__(  s    zHiddenText.__repr__c                 C   s   | j S r   rC  r   r;   r;   r<   __str__,  s    zHiddenText.__str__c                 C   s    t | t |krdS | j|jkS )NF)r  rB  r   otherr;   r;   r<   __eq__1  s    zHiddenText.__eq__c                 C   s
   | |k S r   r;   rG  r;   r;   r<   __ne__<  s    zHiddenText.__ne__N)r   r   r   r   rD  rF  rI  rJ  r;   r;   r;   r<   rA    s
   	rA  c                 C   s   t | ddS )Nr*  rE  )rA  )r   r;   r;   r<   
hide_valueA  s    rK  c                 C   s   t | }t| |dS )NrE  )r@  rA  )r   rC  r;   r;   r<   hide_urlF  s    rL  c                 C   sz   dd tjd dj tjdd  g}| oBtoBtjtjd |k}|rvtjddgtjd	d  }t	d
 d
|dS )zProtection of pip.exe from modification on Windows

    On Windows, any operation modifying pip should be run as:
        python -m pip ...
    zpip.exez	pip{}.exer   zpip{}.{}.exeNrh   z-mrI   rs   z3To modify pip, please run the following command:
{} )r9   rK   rl   r   r3   r4   rJ   rL   rM   r   r5   )modifying_pip	pip_namesshould_show_use_python_msgnew_commandr;   r;   r<   (protect_pip_from_modification_on_windowsL  s*      rR  c                   C   s   t jdk	ot j S )z!Is this console interactive?
    N)rK   stdinisattyr;   r;   r;   r<   is_console_interactivej  s    rU     c              	   C   sR   t  }d}t| d.}t||dD ]}|t|7 }|| q$W 5 Q R X ||fS )z:Return (hash, length) for path using hashlib.sha256()
    r   rb)r   )hashlibsha256openr   r@   update)r4   	blocksizehlengthfblockr;   r;   r<   	hash_fileq  s    ra  c                  C   s(   zddl } W n tk
r"   Y dS X dS )z8
    Return whether the wheel package is installed.
    r   NFT)wheelImportError)rb  r;   r;   r<   is_wheel_installed  s
    rd  )F)rr   )T)N)r  )rV  )
__future__r   
contextlibrE   r   rX  iologgingr3   r   rV   rY   rK   collectionsr   pip._vendorr   Zpip._vendor.retryingr   Zpip._vendor.sixr   r   Zpip._vendor.six.movesr	   Zpip._vendor.six.moves.urllibr
   r  Z"pip._vendor.six.moves.urllib.parser   r#  rI   r   pip._internal.exceptionsr   r   r   r   r   pip._internal.utils.compatr   r   r   r   pip._internal.utils.typingr   r   pip._internal.utils.virtualenvr   r   r   r   typingr   r   r   r   r   r    r!   r"   r#   Zpip._vendor.pkg_resourcesr$   intr?   __all__	getLoggerr   r   r=   rB   r0   r.   r%   rW   rg   r&   r'   r   r   r(   r   r   r*   r+   DEFAULT_BUFFER_SIZEr   r,   r)   r-   r   r   r   r   r   r   r   r   r   objectr   r   contextmanagerr   r/   r   r   r1   r  r  r  r  r"  r)  r.  r8  r:  r;  r>  r2   r@  rA  rK  rL  rR  rU  ra  rd  r;   r;   r;   r<   <module>   s   ,     	

"

	

#
	
D$



#
