]2N @ s d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z e j e Z Gd d d Z d d d d d d Z
d d d Z e d k r d
Z y e d Z Wn< e
k
r Z z e d e d e j WYd d Z [ Xn Xe j e d S)
Nc @ s e Z d Z d d d d d d d Z d d Z d d Z d d Z d
d Z e j d k rr d
d Z
d d d Z d d Z d d Z
d d Z d d Z d d Z d d Z d S)
EnvBuilderFc C s1 | | _ | | _ | | _ | | _ | | _ d S)N)system_site_packagesclearsymlinksupgradewith_pip)selfr r r r r r -/opt/alt/python35/lib64/python3.5/__init__.py__init__? s
zEnvBuilder.__init__c C s t j j | } | j | } | j } d | _ | j | | j | | j rc | j | | j s | j
| | j | | r d | _ | j | d S)NFT) ospathabspathensure_directoriesr create_configuration setup_pythonr
_setup_pipr
setup_scripts
post_setup)r env_dircontextZtrue_system_site_packagesr r r
createG s
zEnvBuilder.createc C s x t j | D]n } t j j | | } t j j | sO t j j | r_ t j | q t j j | r t j | q Wd S)N)
r listdirr
joinislinkisfileremoveisdirshutilrmtree)r r
fnr r r
clear_directorya s $zEnvBuilder.clear_directoryc C sE d d } t j j | r4 | j r4 | j | t j } | | _ t j j | d | _ d | j | _
| | t j } t j
d k r d | k r t j d } n t j } t j j t j j | \ } } | | _ | | _ | | _ t j
d k r.d } d } t j j | d
d }
n8 d } d
} t j j | d d t j d d d }
t j j | | | _ } | | | |
t j d k rt j d k rt j
d k rt j j | d } t j j | st j d | t j j | | | _ }
| | _ t j j |
| | _ | |
| S)Nc S sZ t j j | s" t j | n4 t j j | sF t j j | rV t d | d S)Nz Unable to create directory %r)r r
existsmakedirsr r
ValueError)dr r r
create_if_neededq s $z7EnvBuilder.ensure_directories..create_if_needed z(%s) darwin__PYVENV_LAUNCHER__win32ZScriptsZIncludeLibz
site-packagesbinincludelibz python%d.%d posixlib64l )r r
r" r r! typesSimpleNamespacer splitenv_namepromptenvironsysplatform
executabler
python_dirZ
python_exer version_infoZinc_pathmaxsizenamesymlinkbin_pathbin_nameenv_exe)r r r& r envr; dirnameZexenameZbinnameZincpathZlibpathr
Z link_pathbinpathr r r
r i sL
$
z EnvBuilder.ensure_directoriesc C s t j j | j d | _ } t | d d d b } | j d | j | j r] d } n d } | j d | | j d t j
d d
Wd QRXd S) Nz
pyvenv.cfgwencodingzutf-8z
home = %s
truefalsez"include-system-site-packages = %s
zversion = %d.%d.%d
) r r
r r Zcfg_pathopenwriter< r r9 r= )r r r
fZinclr r r
r s z EnvBuilder.create_configurationntc C s: | j d r d } n | j d o3 | j d } | S)N.pyd.dllTpythonz.exe)rP rQ )endswith
startswith)r rN resultr r r
include_binary s zEnvBuilder.include_binaryc C s | j } | s yK t j j | sZ | rJ t j t j j | | n t j | | Wn+ t k
r t j d | | d } Yn X| r t j
| | d S)NzUnable to symlink %r to %rT) r r r
r r@ basename Exceptionloggerwarningr copyfile)r srcdstrelative_symlinks_okZ
force_copyr r r
symlink_or_copy s
zEnvBuilder.symlink_or_copyc s | j } | j } | j } | | j | | j } t j d k r t j j | se t j | d xBd
D]e } t j j
| | } t j j | sl | | j | d d t j j | sl t j | d ql Wnd } | j f d d t j
| D } xT | D]L } t j j
| | }
t j j
| | } | | j k r| |
| qWt j j
| | } t j j | r f d
d t j
| D } xE | D]= } t j j
| | }
t j j
| | } | |
| qWx t j | j D] \ } }
} d | k rt j j | } t j j
| j d | } t j j | skt j | t j j
| d }
t j j
| d } t j |
| PqWd S)NrO i rR python3r^ TZDLLsc s" g | ] } | r | q Sr r ).0rN )r- r r
s z+EnvBuilder.setup_python..c s" g | ] } | r | q Sr r )ra rN )r- r r
rb s zinit.tclr+ )rR r` )rA rC r_ r; r< r r? r
r chmodr r" rV r r walkrW r r# r r[ )r r rF r
copierrE suffixZsubdirfilesrN r\ r] rootdirsZtcldirr )r- r
r sN
"
"
"
zEnvBuilder.setup_pythonc C s2 | j d d d d g } t j | d t j d S)Nz-ImZ ensurepipz --upgradez
--default-pipstderr)rC
subprocess check_outputSTDOUT)r r cmdr r r
r s zEnvBuilder._setup_pipc C sG t j j t j j t } t j j | d } | j | | d S)Nscripts)r r
r rE __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 C sm | j d | j } | j d | j } | j d | j } | j d | j } | j d | j } | S)NZ __VENV_DIR__Z
__VENV_NAME__Z__VENV_PROMPT__Z__VENV_BIN_NAME__Z__VENV_PYTHON__)replacer r6 r7 rB rC )r textr r r r
replace_variables# s z EnvBuilder.replace_variablesc ! C s | j } t | } xt j | D]\ } } } | | k r x: | d d D]( } | d t j f k rQ | j | qQ Wq% x| D]} t j j | | }
| | d j t j d d } | s | } n t j j | | } t j j
| s t j | t j j | | }
t |
d } | j
} Wd QRX|
j d sy4 | j d } | j | | } | j d } Wn> t k
r} z d } t j d |
| WYd d } ~ Xn X| d k r t |
d } | j | Wd QRXt j |
|
q Wq% Wd S)Ncommonr/ rbz.exezutf-8z+unable to copy script %r, may be binary: %swb)rA lenr rd r? r r
r r5 sepr" r# rL readrS decodert encode UnicodeErrorrY rZ rM r copymode)r r r
rF Zplenrh ri rg r% rN Zsrcfilerf ZdstdirZdstfiledataer r r
rq 5 s@
&
zEnvBuilder.install_scriptsN)__name__
__module__ __qualname__r r r! r r r r? rV r_ r r r r rt rq r r r r
r ' s 7 3
r Fc C s2 t d | d | d | d | } | j | d S)Nr r r r )r r )r r r r r builderr r r
r e s r c C s" d } t j d, k r d } n t t d s3 d } | sH t d nd d l } | j d t d d d
d } | j d d
d d d d d | j d d d d d d d d d t j d k r d } n d } | j
} | j d d | d d d d d d | j d d | d d d d d d | j d! d d d d d d" d d# | j d$ d d d d d d% d d& | j d' d d( d d d d d d) | j | } | j r| j
rt d* t d+ | j d" | j
d | j d% | j d( | j } x | j D] } | j | qWd S)-NTrK F base_prefixz.This script is only for use with Python >= 3.3r prog descriptionzFCreates virtual Python environments in one or more target directories.epilogz|Once an environment has been created, you may wish to activate it, e.g. by sourcing an activate script in its bin directory.ri metavarZENV_DIRnargs+helpz)A directory to create the environment in.z--system-site-packagesdefaultaction
store_truedest system_sitezDGive the virtual environment access to the system site-packages dir.rO 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)z1you cannot supply --upgrade and --clear together.r )rK rK )r9 r= hasattrr$ argparseArgumentParserr add_argumentr r? add_mutually_exclusive_group
parse_argsr r r r r r ri r ) argsZ
compatibler parserZ use_symlinksgroupoptionsr r% r r r
main| sV
r __main__r' z Error: %sfile)loggingr r rk r9 r3 getLoggerr rY r r r rcrX r printrj exitr r r r
s& ?D
*