U
    fmfe                     @   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   )Zpip_pkg_dir r:   z/home/dh_an3skk/arjun-chandrasekhar-teaching.com/tomato/tomatoenvy/lib/python3.8/site-packages/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
    rA   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EEXISTZ	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)dirrR   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IOErrorrC   chmod)funcr4   exc_infoZhas_attr_readonlyr:   r:   r;   rU      s    rU   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   decoderI   getfilesystemencodingUnicodeDecodeErrorr   r   r9   ascii)r4   r&   r:   r:   r;   path_to_display   s    	
rd   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   rI   version_infor`   ra   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)rV   extn	extensionr:   r:   r;   r'      s    c                 C   s2   t jdd D ]}||kr|  S qt| |S )NZ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.ZPIP_NO_INPUTz7No input was expected ($PIP_NO_INPUT set); question: %sN)r3   rw   rx   	Exceptionrz   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   )rz   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   Zsetup_pyZ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   rh   )r4   Z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   basers   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   ry   rq   rB   rT   move
removedirsrC   )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_schemerv   TFN)r,   rI   prefixr   rl   pip._internal.locationsr   values)r4   r   r   Z
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   rl   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   rl   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)rI   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   Zinclude_editablesZeditables_onlyZ	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   ZsitessiteZ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 )Nrv   )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)getattrrI   setattrr   r   )Z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>Zreverse_mappingEnumr:   )dictzipranger?   itemstype)Z
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   re   @[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  )NNrp   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>)rsplitry   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****rv   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  Ztransform_netlocZpurlZnetloc_tupleZ
url_piecesZsurlr:   r:   r;   _transform_url  s    


    
r"  c                 C   s   t | S r   )r  r	  r:   r:   r;   _get_netloc  s    r$  c                 C   s
   t | fS r   )r  r#  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))
    r"  r$  )r  Z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   )r"  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   r+  r,  r:   r:   r;   r     s    zHiddenText.__init__c                 C   s   d t| S )Nz<HiddenText {!r}>)r9   rr   r   r:   r:   r;   __repr__(  s    zHiddenText.__repr__c                 C   s   | j S r   r,  r   r:   r:   r;   __str__,  s    zHiddenText.__str__c                 C   s    t | t |krdS | j|jkS )NF)r   r+  r   otherr:   r:   r;   __eq__1  s    zHiddenText.__eq__c                 C   s
   | |k S r   r:   r0  r:   r:   r;   __ne__<  s    zHiddenText.__ne__N)r   r   r   r   r-  r/  r2  r3  r:   r:   r:   r;   r*    s
   	r*  c                 C   s   t | ddS )Nr  r.  )r*  )r   r:   r:   r;   
hide_valueA  s    r4  c                 C   s   t | }t| |dS )Nr.  )r)  r*  )r  r,  r:   r:   r;   hide_urlF  s    r5  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{}.{}.exeNre   z-mrG   rp   z3To modify pip, please run the following command:
{} )r9   rI   ri   r   r3   r4   rH   rJ   rK   r   r5   )Zmodifying_pipZ	pip_namesZshould_show_use_python_msgZnew_commandr:   r:   r;   (protect_pip_from_modification_on_windowsL  s*      r7  c                   C   s   t jdk	ot j S )z!Is this console interactive?
    N)rI   stdinisattyr:   r:   r:   r;   is_console_interactivej  s    r:     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    rF  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)rG  r:   r:   r;   is_wheel_installed  s
    rI  )F)ro   )T)N)r  )r;  )
__future__r   
contextlibrD   r   r=  iologgingr3   r   rT   rW   rI   collectionsr   Z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  rG   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<   rA   r0   r.   r%   rU   rd   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  r"  r$  r%  r'  r2   r)  r*  r4  r5  r7  r:  rF  rI  r:   r:   r:   r;   <module>   s   ,     	

"

	

#
	
D$



#
