3
_VjqM @ s d Z ddlZddlZddlZddlZddlZddlZddlZeje Z
G dd dZ dddZ ddd Z
e d
krd Zye
dZW n4 ek
r Z zed e ejd
W Y ddZ[X nX eje dS )z
Virtual environment (venv) package for Python. Based on PEP 405.
Copyright (C) 2011-2014 Vinay Sajip.
Licensed to the PSF under a contributor agreement.
Nc @ s e Zd ZdZd ddZdd Zdd Zd
d Zd d
Ze j
dkrHdd Z d ddZ dd Z
dd Zdd Zdd Zdd Zd d ZdS )!
EnvBuildera
This class exists to allow virtual environment creation to be
customized. The constructor parameters determine the builder's
behaviour when called upon to create a virtual environment.
By default, the builder makes the system (global) site-packages dir
*un*available to the created environment.
If invoked using the Python -m option, the default is to use copying
on Windows platforms but symlinks elsewhere. If instantiated some
other way, the default is to *not* use symlinks.
:param system_site_packages: If True, the system (global) site-packages
dir is available to created environments.
:param clear: If True, delete the contents of the environment directory if
it already exists, before environment creation.
:param symlinks: If True, attempt to symlink rather than copy files into
virtual environment.
:param upgrade: If True, upgrade an existing virtual environment.
:param with_pip: If True, ensure pip is installed in the virtual
environment
:param prompt: Alternative terminal prefix for the environment.
FNc C s( || _ || _|| _|| _|| _|| _d S )N)system_site_packagesclearsymlinksupgradewith_pipprompt)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| dS )z
Create a virtual environment in a directory.
:param env_dir: The target directory to create an environment in.
FTN) ospathabspathensure_directoriesr create_configuration setup_pythonr
_setup_pipr
setup_scripts
post_setup)r env_dircontextZtrue_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 joinislinkisfileremoveisdirshutilZrmtree)r r fnr
r
r clear_directoryN s zEnvBuilder.clear_directoryc C s dd }t jj|r$| jr$| j| tj }||_t jj|d |_ | j
dk rT| j
n|j }d| |_
|| t j }t j
dkrd|krt j d }nt j}t jjt jj|\}}||_||_||_t j
dkrd } d
}
t jj|d d } n(d
} d}
t jj|ddt jdd d } t jj||
|_} || || t jdkr|t jdkr|t j
dkr|t jj|d}
t jj|
s|t jd|
t jj|| |_}| |_t jj|||_|| |S )z
Create the directories for the environment.
Returns a context object which holds paths in the environment,
for use by subsequent logic.
c S s@ t jj| st j| n$t jj| s0t jj| r.create_if_needed Nz(%s) darwin__PYVENV_LAUNCHER__Zwin32ZScriptsZIncludeLibz
site-packagesbinincludelibz python%d.%d posixlib64l )r
r r" r r! typesSimpleNamespacer splitenv_namer environsysplatform
executabler
python_dirZ
python_exer version_infoZinc_pathmaxsizenamesymlinkbin_pathbin_nameenv_exe)r r r& r r envr9 dirnameZexenameZbinnameZincpathZlibpathr Z link_pathbinpathr
r
r r V sN
z EnvBuilder.ensure_directoriesc C sz t jj|jd |_}t|dddL}|jd|j | jrBd}nd}|jd| |jd t j
d
d W d
Q R X d
S ) aA
Create a configuration file indicating where the environment's Python
was copied from, and whether the system site-packages should be made
available in the environment.
:param context: The information for the environment creation request
being processed.
z
pyvenv.cfgwzutf-8)encodingz
home = %s
trueZfalsez"include-system-site-packages = %s
zversion = %d.%d.%d
N ) r
r r r Zcfg_pathopenwriter: r r7 r; )r r r fZinclr
r
r r s z EnvBuilder.create_configurationntc C s( |j drd}n|jdo"|j d}|S )N.pyd.dllTpythonz.exe)rM rN )endswith
startswith)r rK resultr
r
r include_binary s
zEnvBuilder.include_binaryc C s | j }|syRtjj|s\|rPtjj|tjj|ks:ttjtjj|| n tj|| W n& tk
r t j
d|| d}Y nX |rt j || dS )zQ
Try symlinking a file, and if that fails, fall back to copying.
zUnable to symlink %r to %rTN)
r r
r r rC AssertionErrorr> 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}tjdkrtjj|sFtj |d xNdD ]F}tjj
||}tjj |sL||j|dd tjj|sLtj |d qLW nRd}| j fdd tj
|D }x<|D ]4} tjj
|| }
tjj
|| } | |jkr||
| qW tjj
||}tjj|rd fd
d tj
|D }x4|D ],} tjj
|| }
tjj
|| } ||
| q4W xtj|jD ]v\} }
}d |krrtjj| }tjj
|jd |}tjj |stj| tjj
| d }
tjj
|d } tj|
| P qrW d
S )z
Set up a Python executable in the environment.
:param context: The information for the environment creation request
being processed.
rL i rO python3T)r\ ZDLLsc s g | ]} |r|qS r
r
).0rK )r, r
r
s z+EnvBuilder.setup_python..c s g | ]} |r|qS r
r
)r_ rK )r, r
r r` s zinit.tclr* N)rO r^ )r? rA r] r9 r: r
r= r r chmodr r" rS r r walkrU r r# r rY )r r rD r ZcopierrC suffixZsubdirfilesrK rZ r[ rootdirsZtcldirr
)r, r r sN
zEnvBuilder.setup_pythonc C s$ |j ddddg}tj|tjd dS )z1Installs or upgrades pip in a virtual environmentz-ImZ ensurepipz --upgradez
--default-pip)stderrN)rA
subprocessZ check_outputZSTDOUT)r r cmdr
r
r r s
zEnvBuilder._setup_pipc C s2 t jjt jjt}t jj|d}| j|| dS )a
Set up scripts into the created environment from a directory.
This method installs the default scripts into the environment
being created. You can prevent the default installation by overriding
this method if you really need to, or if you need to specify
a different location for the scripts to install. By default, the
'scripts' directory in the venv package is used as the source of
scripts to install.
scriptsN)r
r r rC __file__r install_scripts)r r r r
r
r r s zEnvBuilder.setup_scriptsc C s dS )a
Hook for post-setup modification of the venv. Subclasses may install
additional packages or scripts here, add activation shell scripts, etc.
:param context: The information for the environment creation request
being processed.
Nr
)r r r
r
r r s zEnvBuilder.post_setupc s |j |j|j|j|jd}dd }dd }tj |j}|jdrF| n|jdrV| ntj fdd |j D }x |j D ]\}}|j
||}q|W |S )
ai
Replace variable placeholders in script text with context-specific
variables.
Return the text passed in , but with variables replaced.
:param text: The text in which to replace placeholder variables.
:param context: The information for the environment creation request
being processed.
)Z __VENV_DIR__Z
__VENV_NAME__Z__VENV_PROMPT__Z__VENV_BIN_NAME__Z__VENV_PYTHON__c S s | j dd} d| dS )a
This should satisfy PowerShell quoting rules [1], unless the quoted
string is passed directly to Windows native commands [2].
[1]: https://learn.microsoft.com/en-us/powershell/module/microsoft.powershell.core/about/about_quoting_rules
[2]: https://learn.microsoft.com/en-us/powershell/module/microsoft.powershell.core/about/about_parsing#passing-arguments-that-contain-quote-characters
'z'')replace)sr
r
r quote_ps1$ s z/EnvBuilder.replace_variables..quote_ps1c S s | S )Nr
)ro r
r
r quote_bat. s z/EnvBuilder.replace_variables..quote_batz.ps1z.batc s i | ]\}} ||qS r
r
)r_ keyro )quoter
r
< s z0EnvBuilder.replace_variables..) r r5 r r@ rA shlexrs script_pathrP itemsrn ) r textr Z replacementsrp rq rv rr Zquotedr
)rs r replace_variables s$
z EnvBuilder.replace_variablesc ! C s |j }t|}xtj|D ]~\}}}||kr`x,|dd D ] }|dtjfkr>|j| q>W q x8|D ].} tjj|| }
||d jtj dd } | s|} ntjj|f| } tjj
| stj | tjj| | }
t |
d} | j
}W dQ R X |
jdsd|
|_y$|jd}| j||}|jd}W n6 tk
rb } zd}tjd|
| W Y dd}~X nX |dk rht |
d} | j| W dQ R X tj|
|
qhW q W dS ) as
Install scripts into the created environment from a directory.
:param context: The information for the environment creation request
being processed.
:param path: Absolute pathname of a directory containing script.
Scripts in the 'common' subdirectory of this directory,
and those in the directory named for the platform
being run on, are installed in the created environment.
Placeholder variables are replaced with environment-
specific values.
Ncommonr. rbz.exezutf-8z+unable to copy script %r, may be binary: %swb)r? lenr
rb r= r r r r4 sepr" r# rI readrP rv decodery encode UnicodeErrorrW rX rJ r Zcopymode)r r r rD Zplenre rf rd r% rK Zsrcfilerc ZdstdirZdstfiledataer
r
r rl A sB
zEnvBuilder.install_scripts)FFFFFN)F)__name__
__module__ __qualname____doc__r r r! r r r
r= rS r] r r r r ry rl r
r
r
r r s
8
3
0r Fc C s t |||||d}|j| dS )z,Create a virtual environment in a directory.)r r r r r N)r r )r r r r r r builderr
r
r r r s r c C sb d}t jd*k rd}ntt ds"d}|s2tdn,dd l}|jtddd }|jd
d d d
d |jdddddd tj dkrd}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 descriptionZepilogrf ZENV_DIR+z)A directory to create the environment in.)metavarnargshelpz--system-site-packages
store_true system_sitezDGive the virtual environment access to the system site-packages dir.)defaultactiondestr 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$ argparseArgumentParserr add_argumentr
r= Z add_mutually_exclusive_group
parse_argsr r r r r r r rf r ) argsZ
compatibler parserZ use_symlinksgroupZoptionsr r% r
r
r mainz s\
r __main__r' z Error: %s)file)FFFFN)N)r Zloggingr
r rh r7 r2 ru Z getLoggerr rW r r r ZrcrV r printrg exitr
r
r
r s,
b
H$