B
YI~d R �
@ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZe �e�Z G dd� d�Z
d
dd�Z ddd�Z ed kr�d
Z
ye � d Z
W n4 ek
r� Z zed e ejd � W ddZ[X Y nX e�e
� dS )� Nc @ s� e Zd Zd dd�Zdd� Zdd� Zd d
� Zd d � Zej d
krHd dd�Z
n
d dd�Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � ZdS ) �
EnvBuilderFNc C s( || _ || _|| _|| _|| _|| _d S )N)�system_site_packages�clear�symlinks�upgrade�with_pip�prompt)�selfr r r r r r � r
�2/opt/alt/python37/lib64/python3.7/venv/__init__.py�__init__+ s zEnvBuilder.__init__c C sx t j�|�}| �|�}| j}d| _| �|� | �|� | jrF| �|� | j s`| �
|� | � |� |rtd| _| �|� d S )NFT) �os�path�abspath�ensure_directoriesr �create_configuration� setup_pythonr �
_setup_pipr �
setup_scripts�
post_setup)r �env_dir�contextZtrue_system_site_packagesr
r
r �create4 s
zEnvBuilder.createc C s` xZt �|�D ]L}t j�||�}t j�|�s6t j�|�rBt �|� q t j�|�r t� |� q W d S )N)
r
�listdirr �join�islink�isfile�remove�isdir�shutilZrmtree)r r �fnr
r
r �clear_directoryN s zEnvBuilder.clear_directoryc C s� dd� }t j�|�r$| jr$| �|� t�� }||_t j�|�d |_ | j
d k rT| j
n|j }d| |_
||� t j }t t
dt
j�}t j�t j�|��\}}||_||_||_t
jdkr�d} d}
t j�|d d
�} n(d } d }
t j�|d
dt
jd d� d
�} t j�||
� |_} || � || � t
jdk�rft jdk�rft
jdk�rft j�|d�}
t j�|
��sft �d
|
� t j�|| � |_}| |_t j�||�|_||� |S )Nc S s@ t j�| �st �| � n$t j�| �s0t j�| �r.create_if_needed� z(%s) �_base_executable�win32ZScriptsZInclude�Libz
site-packages�binZinclude�libz python%d.%d� l �posix�darwin�lib64) r
r r" r r! �types�SimpleNamespacer �split�env_namer �environ�getattr�sys�
executabler �
python_dirZ
python_exe�platformr � version_infoZinc_path�maxsize�name�symlink�bin_path�bin_name�env_exe)r r r& r r �envr8 �dirnameZexenameZbinnameZincpathZlibpathr Z link_path�binpathr
r
r r V sJ
z EnvBuilder.ensure_directoriesc C sz t j�|jd� |_}t|ddd��L}|�d|j � | jrBd}nd}|�d| � |�d t j
d d
� � W d Q R X d S ) Nz
pyvenv.cfg�wzutf-8)�encodingz
home = %s
�trueZfalsez"include-system-site-packages = %s
zversion = %d.%d.%d
� ) r
r r r Zcfg_path�open�writer9 r r7 r; )r r r �fZinclr
r
r r � s z EnvBuilder.create_configuration�ntc C s~ | j }|sjy6tj�|�s@|r4t�tj�|�|� n t�||� W n& tk
rh t�d||� d}Y nX |rzt �
||� d S )NzUnable to symlink %r to %rT) r r
r r r> �basename� Exception�logger�warningr �copyfile)r �src�dst�relative_symlinks_okZ
force_copyr
r
r �symlink_or_copy� s
zEnvBuilder.symlink_or_copyc C sZ t j�|�ot j�|� }| jr~|s~t j�|�s~y*|rLt �t j�|�|� n t �||� d S tk
r| t �
d||� Y nX t j� t j�|��\}}t j� t j�
t�dd|| �}t�d�s�t j�|��s"|�d�r�d| }|d d� }|dkr�d}n|d k�rd
}t j� t j�
|�|| �}n|}t j�|��sJ|�sFt �
d |� d S t�||� d S ) NzUnable to symlink %r to %r�scriptsrL TZ_d������pythonZ venvlauncherZpythonwZ
venvwlauncherzUnable to copy %r)r
r �lexistsr" r r r> rM rN rO rP �splitextr rC �__file__� sysconfig�is_python_buildr �endswithr rQ )r rR rS rT Zbad_srcrM ZextZsrcfnr
r
r rU � s:
c C s� |j }|j}| j}|j}tjdkr�||j|� tj�|�sFt� |d� xNdD ]F}tj�
||�}tj� |�sL||j|dd� tj�|�sLt� |d� qLW �n| j r�dd� t�
|�D �}t�d�r�dd� |D �}n d d
d d g}x:|D ]2}tj�
||�}tj�|�r�||tj�
||�� q�W t�d��r�x�t�|j�D ]v\} }
} d
| k�r0tj�| �} tj�
|jd| �} tj� | ��szt�| � tj�
| d
�}tj�
| d
�}
t�||
� P �q0W d S )NrL i� )rX Zpython3T)rT c S s, g | ]$}t j�t j�|�d �dkr|�qS )r' )z.exez.dll)r
r �normcaserZ )�.0rK r
r
r �
� s z+EnvBuilder.setup_python..c S s" g | ]}t j�|��d �r|�qS ))rX Z vcruntime)r
r r_ �
startswith)r` rK r
r
r ra s z
python.exez python_d.exez pythonw.exez
pythonw_d.exezinit.tclr* )r? rA rU r9 r
r= r8 r r �chmodr r" r r r\ r] rY �walkrM r r# r rQ )r r rD r ZcopierrC �suffix�suffixesrR �root�dirs�filesZtcldirrS r
r
r r � sF
zEnvBuilder.setup_pythonc C s$ |j ddddg}tj|tjd� d S )Nz-ImZ ensurepipz --upgradez
--default-pip)�stderr)rA �
subprocessZ check_outputZSTDOUT)r r �cmdr
r
r r s
zEnvBuilder._setup_pipc C s2 t j�t j�t��}t j�|d�}| �||� d S )NrV )r
r r rC r[ r �install_scripts)r r r r
r
r r " s zEnvBuilder.setup_scriptsc C s d S )Nr
)r r r
r
r r 1 s zEnvBuilder.post_setupc C sJ |� d|j�}|� d|j�}|� d|j�}|� d|j�}|� d|j�}|S )NZ __VENV_DIR__Z
__VENV_NAME__Z__VENV_PROMPT__Z__VENV_BIN_NAME__Z__VENV_PYTHON__)�replacer r4 r r@ rA )r �textr r
r
r �replace_variables; s z EnvBuilder.replace_variablesc C s� |j }t|�}�x�t�|�D �]�\}}}||kr`x,|d d � D ] }|dtjfkr>|�|� q>W q �xR|D �]H} tjdkr�| �d�r�| �d�r�qhtj� || �}
||d � �
tj �dd � } | s�|} ntjj |f| �� } tj� | �s�t�
| � tj� | | �}
t|
d��} | �� }W d Q R X |
�d��s~y$|�d�}| �||�}|�d�}W n6 tk
�r| } zd }t�d|
|� W d d }~X Y nX |d k rht|
d ��} | �|� W d Q R X t�|
|
� qhW q W d S )
N�commonrL rX )z.exez.pdbr- �rbzutf-8z+unable to copy script %r, may be binary: %s�wb)r? �lenr
rd r= r rb r^ r r r3 �sepr" r# rI �read�decoderp �encode� UnicodeErrorrO rP rJ r Zcopymode)r r r rD Zplenrg rh ri r% rK Zsrcfilere ZdstdirZdstfile�data�er
r
r rm M sF
zEnvBuilder.install_scripts)FFFFFN)F)F)�__name__�
__module__� __qualname__r r r! r r r
r= rU r r r r rp rm r
r
r
r r s
5
+<
r Fc C s t |||||d�}|�| � d S )N)r r r r r )r r )r r r r r r �builderr
r
r r � s r c C sb d}t jdk rd}ntt d�s"d}|s2td���n,dd l}|jtddd �}|jd
d d d
d� |jdddddd� tj dkr�d}nd}|�
� }|jd|dddd� |jd| dddd� |jdddd d d� |jd ddd d d� |jd!d"ddd#d$� |jd%d&d'� |� | �}|j �r"|j
�r"td(��t|j|j
|j|j |j|jd)�}x|jD ]}|�|� �qJW d S )*NT)rH rH F� base_prefixz.This script is only for use with Python >= 3.3r zFCreates virtual Python environments in one or more target directories.z|Once an environment has been created, you may wish to activate it, e.g. by sourcing an activate script in its bin directory.)�prog� description�epilogrh ZENV_DIR�+z)A directory to create the environment in.)�metavar�nargs�helpz--system-site-packages�
store_true� system_sitezDGive the virtual environment access to the system site-packages dir.)�default�action�destr� rL z
--symlinksr z[Try to use symlinks rather than copies, when symlinks are not the default for the platform.z--copies� store_falsez\Try to use copies rather than symlinks, even when symlinks are the default for the platform.z--clearr zcDelete the contents of the environment directory if it already exists, before environment creation.z --upgrader zlUpgrade the environment directory to use this version of Python, assuming Python has been upgraded in-place.z
--without-pipr z]Skips installing or upgrading pip in the virtual environment (pip is bootstrapped by default))r� r� r� r� z--promptz;Provides an alternative prompt prefix for this environment.)r� z1you cannot supply --upgrade and --clear together.)r r r r r r )r7 r; �hasattrr$ �argparse�ArgumentParserr| � add_argumentr
r= � add_mutually_exclusive_group�
parse_argsr r r r� r r r rh r ) �argsZ
compatibler� �parserZ use_symlinks�group�optionsr r% r
r
r �main� s\
r� �__main__r' z Error: %s)�file)FFFFN)N)Zloggingr
r rk r7 r\ r1 Z getLoggerr| rO r r r� ZrcrN r{ �printrj �exitr
r
r
r � s*
p
H$