o
    JAf?%                     @   sH  d 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mZmZmZ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% ddl&m'Z' ddl(m)Z)m*Z* erddl+m,Z, e-e.Z/G dd dZ0de1fddZ2G dd dZ3G dd de3Z4dS )z;Build Environment used for isolation during sdist building
    N)OrderedDict)	get_paths)TracebackType)TYPE_CHECKINGIterableListOptionalSetTupleType)where)Requirement)Version)__file__)open_spinner)get_platlibget_prefixed_libsget_purelib)get_default_environmentget_environment)call_subprocess)TempDirectorytempdir_kinds)PackageFinderc                   @   s   e Zd ZdeddfddZdS )_PrefixpathreturnNc                 C   s@   || _ d| _ttjdkrdnd||ddd | _t|| _d S )NFntposix_prefix)baseplatbase)varsscripts)r   setupr   osnamebin_dirr   lib_dirs)selfr    r)   </usr/lib/python3.10/site-packages/pip/_internal/build_env.py__init__    s   z_Prefix.__init__)__name__
__module____qualname__strr+   r)   r)   r)   r*   r      s    r   r   c                  C   s.   t t j} |  st| S t| d S )zGet a file to pass to a Python executable, to run the currently-running pip.

    This is used to run a pip subprocess, for installing requirements into the build
    environment.
    z__pip-runner__.py)	pathlibPathpip_locationresolveparentis_dirr/   r$   fsdecode)sourcer)   r)   r*   _get_runnable_pip*   s   r8   c                   @   s   e Zd ZdZdddZdddZdeee  d	ee d
ee	 ddfddZ
dee deeeeef  ee f fddZdddee dededdf
ddZededddee dededdfddZdS )BuildEnvironmentzACreates and manages an isolated environment to install build depsr   Nc                    s  t tjdd t fdddD | _g | _g | _tt| j	 D ]}| j
|j | j|j q"dd t t fD }tj jd| _tj| jsTt| j ttj| jd	d
dd}|tdj|| jd W d    d S 1 s{w   Y  d S )NT)kindglobally_managedc                 3   s(    | ]}|t tj j|fV  qd S N)r   r$   r   join.0r%   temp_dirr)   r*   	<genexpr>@   s
    
z,BuildEnvironment.__init__.<locals>.<genexpr>)normaloverlayc                 S   s   h | ]}t j|qS r)   )r$   r   normcase)r?   siter)   r)   r*   	<setcomp>N   s    z,BuildEnvironment.__init__.<locals>.<setcomp>rF   zsitecustomize.pywzutf-8)encodinga  
                import os, site, sys

                # First, drop system-sites related paths.
                original_sys_path = sys.path[:]
                known_paths = set()
                for path in {system_sites!r}:
                    site.addsitedir(path, known_paths=known_paths)
                system_paths = set(
                    os.path.normcase(path)
                    for path in sys.path[len(original_sys_path):]
                )
                original_sys_path = [
                    path for path in original_sys_path
                    if os.path.normcase(path) not in system_paths
                ]
                sys.path = original_sys_path

                # Second, add lib directories.
                # ensuring .pth file are processed.
                for path in {lib_dirs!r}:
                    assert not path in sys.path
                    site.addsitedir(path)
                )system_sitesr'   )r   r   	BUILD_ENVr   	_prefixes	_bin_dirs	_lib_dirsreversedlistvaluesappendr&   extendr'   r   r   r$   r   r=   	_site_direxistsmkdiropenwritetextwrapdedentformat)r(   prefixrJ   fpr)   r@   r*   r+   =   s6   

"zBuildEnvironment.__init__c                 C   sn   dd dD | _ | jd d  }| j d }|r||tj | jg}tjtj	|dtj	|d d S )Nc                 S   s   i | ]
}|t j|d qS r<   )r$   environgetr>   r)   r)   r*   
<dictcomp>u   s    z.BuildEnvironment.__enter__.<locals>.<dictcomp>)PATHPYTHONNOUSERSITE
PYTHONPATHra   1)
	_save_envrM   rS   splitr$   pathseprT   r^   updater=   )r(   r   old_path
pythonpathr)   r)   r*   	__enter__t   s   


zBuildEnvironment.__enter__exc_typeexc_valexc_tbc                 C   s:   | j  D ]\}}|d u rtj|d  q|tj|< qd S r<   )re   itemsr$   r^   pop)r(   rl   rm   rn   varname	old_valuer)   r)   r*   __exit__   s
   zBuildEnvironment.__exit__reqsc           	      C   s   t  }t  }|rct| drt| jnt }|D ]K}t|}|jdur+|jddis+q||j	}|s9|
| qt|jtrI|j	 d|j }n	|j	 d|j }|jj|jddsb|
||f q||fS )	zReturn 2 sets:
        - conflicting requirements: set of (installed, wanted) reqs tuples
        - missing requirements: set of reqs
        rN   Nextra z==z===T)prereleases)sethasattrr   rN   r   r   markerevaluateget_distributionr%   add
isinstanceversionr   	specifiercontains)	r(   rt   missingconflictingenvreq_strreqdistinstalled_req_strr)   r)   r*   check_requirements   s,   
z#BuildEnvironment.check_requirementsfinderr   requirementsprefix_as_stringr:   c                C   s<   | j | }|jr
J d|_|sd S | jt ||||d d S )NT)r:   )rL   r#   _install_requirementsr8   )r(   r   r   r   r:   r\   r)   r)   r*   install_requirements   s   


z%BuildEnvironment.install_requirementspip_runnabler\   c                C   st  t j| dddd|jdg}t tjkr|d dD ]}t|j	|}|
d|d	d
 dt|p2dhf q|j}|rW|
d|d g |dd  D ]	}	|
d|	g qLn|d |jD ]	}
|
d|
g q_|jD ]	}|
d|g ql|jr~|d |jr|d |d |
| dt i}td| }t|d| ||d W d    d S 1 sw   Y  d S )Ninstallz--ignore-installedz	--no-userz--prefixz--no-warn-script-locationz-v)	no_binaryonly_binaryz--_-,z:none:z-ir      z--extra-index-urlz
--no-indexz--find-linksz--trusted-hostz--prez--prefer-binary_PIP_STANDALONE_CERTzInstalling zpip subprocess to install )command_descspinnerextra_environ)sys
executabler   loggergetEffectiveLevelloggingDEBUGrR   getattrformat_controlrS   replacer=   sorted
index_urls
find_linkstrusted_hostsallow_all_prereleasesprefer_binaryr   r   r   )r   r   r   r\   r:   argsr   formatsr   extra_indexlinkhostr   r   r)   r)   r*   r      sX   










"z&BuildEnvironment._install_requirementsr   N)r,   r-   r.   __doc__r+   rk   r   r   BaseExceptionr   rs   r   r/   r
   r	   r   r   staticmethodr   r   r)   r)   r)   r*   r9   :   sV    

7


"
r9   c                
   @   sz   e Zd ZdZdddZdddZdeee  d	ee d
ee	 ddfddZ
dddZdddee dededdf
ddZdS )NoOpBuildEnvironmentz0A no-op drop-in replacement for BuildEnvironmentr   Nc                 C      d S r<   r)   r(   r)   r)   r*   r+   	     zNoOpBuildEnvironment.__init__c                 C   r   r<   r)   r   r)   r)   r*   rk     r   zNoOpBuildEnvironment.__enter__rl   rm   rn   c                 C   r   r<   r)   )r(   rl   rm   rn   r)   r)   r*   rs     s   zNoOpBuildEnvironment.__exit__c                 C   r   r<   r)   r   r)   r)   r*   cleanup  r   zNoOpBuildEnvironment.cleanupr   r   r   r   r:   c                C   s   t  r<   )NotImplementedError)r(   r   r   r   r:   r)   r)   r*   r     s   z)NoOpBuildEnvironment.install_requirementsr   )r,   r-   r.   r   r+   rk   r   r   r   r   rs   r   r   r/   r   r)   r)   r)   r*   r     s2    




r   )5r   r   r$   r0   r   rY   collectionsr   	sysconfigr   typesr   typingr   r   r   r   r	   r
   r   pip._vendor.certifir   "pip._vendor.packaging.requirementsr   Zpip._vendor.packaging.versionr   pipr   r2   pip._internal.cli.spinnersr   pip._internal.locationsr   r   r   pip._internal.metadatar   r   pip._internal.utils.subprocessr   pip._internal.utils.temp_dirr   r   "pip._internal.index.package_finderr   	getLoggerr,   r   r   r/   r8   r9   r   r)   r)   r)   r*   <module>   s6    $
 M