U
    P^                     @   s   d 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Zddlm	Z	 ddl
mZ ddlmZmZ eeZdd	 ZG d
d deZdddZdddZdS )zIBuild wheels/sdists by installing build deps to a temporary environment.
    N)
check_call)	get_paths)mkdtemp   )Pep517HookCallerLoggerWrapperc              	   C   sH   t tj| d}t|}W 5 Q R X |d }|d |d |dfS )Nzpyproject.tomlzbuild-systemrequireszbuild-backendzbackend-path)openospathjointomlloadget)
source_dirfZpyproject_dataZbuildsys r   </tmp/pip-install-220asx0h/pip/pip/_vendor/pep517/envbuild.py_load_pyproject   s    r   c                   @   s6   e Zd ZdZdZdddZdd Zdd	 Zd
d ZdS )BuildEnvironmentzContext manager to install build deps in a simple temporary environment

    Based on code I wrote for pip, which is MIT licensed.
    NTc                 C   s
   || _ d S )N)_cleanup)selfcleanupr   r   r   __init__9   s    zBuildEnvironment.__init__c                 C   s   t dd| _td| j tjdd | _tjdd | _tj	dkrHdnd}t
|| j| jdd	}|d
 }| jr|tj | j tjd< n|tj tj tjd< |d |d kr|d }n|d tj |d  }| jr|tj | j tjd< n
|tjd< | S )Nzpep517-build-env-)prefixzTemporary build environment: %sPATH
PYTHONPATHntposix_prefix)baseplatbase)varsscriptspurelibplatlib)r   r   loginfor
   environr   	save_pathsave_pythonpathnamer   pathsepdefpath)r   Zinstall_schemeZinstall_dirsr"   lib_dirsr   r   r   	__enter__<   s0    


zBuildEnvironment.__enter__c                 C   sT   |sdS t d| tjddddd| jgt| }t|tt tj	tt tj
d dS )	zAInstall dependencies into this env by calling pip in a subprocessNzCalling pip to install %sz-mpipinstallz--ignore-installedz--prefix)stdoutstderr)r%   r&   sys
executabler   listr   r   loggingINFOERROR)r   reqscmdr   r   r   pip_install\   s$         

zBuildEnvironment.pip_installc                 C   s~   | j o| jd k	otj| j}|r.t| j | jd krHtjdd  n| jtjd< | j	d krntjdd  n| j	tjd< d S )Nr   r   )
r   r   r
   isdirshutilrmtreer(   r'   popr)   )r   exc_typeexc_valexc_tbZneeds_cleanupr   r   r   __exit__j   s    

zBuildEnvironment.__exit__)T)	__name__
__module____qualname____doc__r   r   r.   r;   rC   r   r   r   r   r      s   
 r   c           	   
   C   sn   |dkri }t | \}}}t| ||}t 8}|| ||}|| |||W  5 Q R  S Q R X dS )am  Build a wheel from a source directory using PEP 517 hooks.

    :param str source_dir: Source directory containing pyproject.toml
    :param str wheel_dir: Target directory to create wheel in
    :param dict config_settings: Options to pass to build backend

    This is a blocking function which will run pip in a subprocess to install
    build requirements.
    N)r   r   r   r;   get_requires_for_build_wheelbuild_wheel)	r   	wheel_dirconfig_settingsr   backendbackend_pathhooksenvr9   r   r   r   rI   ~   s    



rI   c           	   
   C   sn   |dkri }t | \}}}t| ||}t 8}|| ||}|| |||W  5 Q R  S Q R X dS )am  Build an sdist from a source directory using PEP 517 hooks.

    :param str source_dir: Source directory containing pyproject.toml
    :param str sdist_dir: Target directory to place sdist in
    :param dict config_settings: Options to pass to build backend

    This is a blocking function which will run pip in a subprocess to install
    build requirements.
    N)r   r   r   r;   get_requires_for_build_sdistbuild_sdist)	r   Z	sdist_dirrK   r   rL   rM   rN   rO   r9   r   r   r   rQ      s    



rQ   )N)N)rG   r
   r6   r   r=   
subprocessr   r3   	sysconfigr   tempfiler   wrappersr   r   	getLoggerrD   r%   r   objectr   rI   rQ   r   r   r   r   <module>   s   
a
