3
�QgqM � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZe je�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 Y ddZ[X nX eje
� dS )� Nc @ s� e Zd Zd dd�Zdd� Zdd� Zd d
� Zd d � Zej d
krDdd� Z
d dd�Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � ZdS ) �
EnvBuilderFNc C s( || _ || _|| _|| _|| _|| _d S )N)�system_site_packages�clear�symlinks�upgrade�with_pip�prompt)�selfr r r r r r � r
�%/usr/lib64/python3.6/venv/__init__.py�__init__+ s zEnvBuilder.__init__c C sx t jj|�}| j|�}| j}d| _| j|� | j|� | jrF| j|� | j s`| j
|� | j |� |rtd| _| j|� 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 j|�D ]L}t jj||�}t jj|�s6t jj|�rBt j|� q t jj|�r tj |� 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 jj|�r$| jr$| j|� tj� }||_t jj|�d |_ | j
d k rT| j
n|j }d| |_
||� t j }t j
dkr�d|kr�t j d }nt j}t jjt jj|��\}}||_||_||_t j
dkr�d} d }
t jj|d
d �} n(d } d
}
t jj|ddt jd d� d �} t jj||
� |_} || � || � t jdk�r|t jdk�r|t j
dk�r|t jj|d�}
t jj|
��s|t jd|
� t jj|| � |_}| |_t jj||�|_||� |S )Nc S s@ t jj| �st j| � n$t jj| �s0t jj| �r<td| ��d S )Nz Unable to create directory %r)r
r �exists�makedirsr r �
ValueError)�dr
r
r �create_if_needed^ s z7EnvBuilder.ensure_directories.<locals>.create_if_needed� z(%s) �darwin�__PYVENV_LAUNCHER__Zwin32ZScriptsZInclude�Libz
site-packages�bin�include�libz python%d.%d� � �posix�lib64l )r
r r" r r! �types�SimpleNamespacer �split�env_namer �environ�sys�platform�
executabler �
python_dirZ
python_exer � version_infoZinc_path�maxsize�name�symlink�bin_path�bin_name�env_exe)r r r& r r �envr9 �dirnameZexenameZbinnameZincpathZlibpathr Z link_path�binpathr
r
r r V sN
z EnvBuilder.ensure_directoriesc C sz t jj|jd� |_}t|ddd��L}|jd|j � | jrBd}nd}|jd| � |jd 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�writer: r r7 r; )r r r �fZinclr
r
r r � s z EnvBuilder.create_configuration�ntc C s( |j d�rd}n|jd�o"|j d�}|S )N�.pyd�.dllT�pythonz.exe)rM rN )�endswith�
startswith)r rK �resultr
r
r �include_binary� s
zEnvBuilder.include_binaryc C s~ | j }|sjy6tjj|�s@|r4tjtjj|�|� n tj||� W n& tk
rh tjd||� d}Y nX |rzt j
||� 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 s� |j }|j}| j}||j|� |j}tjdkr�tjj|�sFtj |d� xNd
D ]F}tjj
||�}tjj |�sL||j|dd� tjj|�sLtj |d� qLW �nRd}| j � � fdd �tj
|�D �}x<|D ]4} tjj
|| �}
tjj
|| �} | |jkr�||
| � q�W tjj
||�}tjj|��rd� fd
d �tj
|�D �}x4|D ],} tjj
|| �}
tjj
|| �} ||
| � �q4W x�tj|j�D ]v\} }
}d |k�rrtjj| �}tjj
|jd |�}tjj |��s�tj|� tjj
| d �}
tjj
|d �} tj|
| � P �qrW d S )NrL i� rO �python3T)r[ ZDLLsc s g | ]}� |�r|�qS r
r
)�.0rK )r, r
r �
<listcomp>� s z+EnvBuilder.setup_python.<locals>.<listcomp>c s g | ]}� |�r|�qS r
r
)r^ rK )r, r
r r_ � s zinit.tclr* )rO r] )r? rA r\ r9 r: r
r= r r �chmodr r" rS r r �walkrT r r# r rX )r r rD r ZcopierrC �suffixZsubdir�filesrK rY rZ �root�dirsZtcldirr
)r, r r � sN
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 jjt jjt��}t jj|d�}| j||� d S )N�scripts)r
r r rC �__file__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 s zEnvBuilder.post_setupc s� |j |j|j|j|jd�}dd� }dd� }tj� |j}|jd�rF|� n|jd�rV|� ntj� � fdd �|j � D �}x |j � D ]\}}|j
||�}q|W |S )
N)Z __VENV_DIR__Z
__VENV_NAME__Z__VENV_PROMPT__Z__VENV_BIN_NAME__Z__VENV_PYTHON__c S s | j dd�} d| � d�S )N�'z'')�replace)�sr
r
r � quote_ps1$ s z/EnvBuilder.replace_variables.<locals>.quote_ps1c S s | S )Nr
)rn r
r
r � quote_bat. s z/EnvBuilder.replace_variables.<locals>.quote_batz.ps1z.batc s i | ]\}}� |�|�qS r
r
)r^ �keyrn )�quoter
r �
<dictcomp>< s z0EnvBuilder.replace_variables.<locals>.<dictcomp>) r r5 r r@ rA �shlexrr � script_pathrP �itemsrm ) r �textr Z replacementsro rp ru rq Zquotedr
)rr r �replace_variables s$
z EnvBuilder.replace_variablesc ! C s� |j }t|�}�x�tj|�D �]~\}}}||kr`x,|d d � D ] }|dtjfkr>|j|� q>W q �x8|D �].} tjj|| �}
||d � jtj �dd � } | s�|} ntjj|f| �� } tjj
| �s�tj | � tjj| | �}
t |
d��} | j
� }W d Q R X |
jd��sd|
|_y$|jd�}| j||�}|jd�}W n6 tk
�rb } zd }tjd|
|� W Y d d }~X nX |d k rht |
d��} | j|� W d Q R X tj|
|
� qhW q W d S )N�commonr. �rbz.exezutf-8z+unable to copy script %r, may be binary: %s�wb)r? �lenr
ra r= r r r r4 �sepr" r# rI �readrP ru �decoderx �encode� UnicodeErrorrV rW rJ r Zcopymode)r r r rD Zplenrd re rc r% rK Zsrcfilerb ZdstdirZdstfile�data�er
r
r rk A sB
zEnvBuilder.install_scripts)FFFFFN)F)�__name__�
__module__� __qualname__r r r! r r r
r= rS r\ r r r r rx rk r
r
r
r r s
8
3
0r Fc C s t |||||d�}|j| � d S )N)r r r r r )r r )r r r r r r �builderr
r
r r r s r c C sb d}t jd*k rd}ntt d�s"d}|s2td���n,dd l}|jtddd �}|jd
d d d
d� |jdddddd� tj dkr�d}nd}|j
� }|jd|dddd� |jd| dddd� |jdddd d d� |jd ddd d d� |jd!d"ddd#d$� |jd%d&d'� |j | �}|j �r"|j
�r"td(��t|j|j
|j|j |j|jd)�}x|jD ]}|j|� �qJW d S )+NTrH 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.)�progZ descriptionZepilogre 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--copiesZ 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 )rH rH )r7 r; �hasattrr$ �argparse�ArgumentParserr� � add_argumentr
r= Z add_mutually_exclusive_group�
parse_argsr r r r� r r r re r ) �argsZ
compatibler� �parserZ use_symlinks�groupZoptionsr� r% r
r
r �mainz s\
r� �__main__r' z Error: %s)�file)FFFFN)N)Zloggingr
r rg r7 r2 rt Z getLoggerr� rV r r r� ZrcrU r� �printrf �exitr
r
r
r �<module> s*
b
H$