U
Qab7 @ s
d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl m
Z
mZ d dlZejdrd dlm m m mZ n eje j ZzeZW n e k
r dZY nX e Z d dl m Z d dlm!Z! ddd d
gZ"d-d d Z#ej$d.d
dZ%ej$dd Z&ej$dd Z'ej$dd Z(G dd de)Z*G dd dZ+ej$dd Z,dd Z-ej$d d Z.ej$d d Z/d!d" Z0d#d$ Z1d%d
Z2G d&d dZ3e4e d're j5gZ6ng Z6G d(d de3Z7e8ej9d)d* d+: D Z;G d,d d e Z r4 r5 tbr r r __exit__{ s
zExceptionSaver.__exit__c C s6 dt | krdS ttj| j\}}t||| j dS )z"restore and re-raise any exceptionrA N)varsr r. loadsrA r ZreraiserB )r> r4 r5 r r r resume s zExceptionSaver.resumeN)r7 r8 r9 r: r? rD rG r r r r r< r s r< c # sV t j t } V W 5 Q R X t j fddt jD }t| | dS )z
Context in which imported modules are saved.
Translates exceptions internal to the context into the equivalent exception
outside the context.
c 3 s$ | ] }| kr| d s|V qdS )z
encodings.N
startswith).0mod_namer! r r s
z save_modules..N)r modulescopyr< update_clear_modulesrG ) saved_excZ del_modulesr r! r save_modules s
rR c C s t | D ] }tj|= qd S r )listr rM )Z module_namesrK r r r rP s rP c c s$ t } z
| V W 5 t | X d S r )r$ __getstate__ __setstate__r! r r r save_pkg_resources_state s
rV c c s t j| d}t x t f t t N t < t|( t | t
d d V W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X d S )NZtempZ
setuptools) r( r joinrV rR hide_setuptoolsr" r r' r,
__import__) setup_dirZtemp_dirr r r
setup_context s
r[ c C s t d}t|| S )aH
>>> _needs_hiding('setuptools')
True
>>> _needs_hiding('pkg_resources')
True
>>> _needs_hiding('setuptools_plugin')
False
>>> _needs_hiding('setuptools.__init__')
True
>>> _needs_hiding('distutils')
True
>>> _needs_hiding('os')
False
>>> _needs_hiding('Cython')
True
z1(setuptools|pkg_resources|distutils|Cython)(\.|$))rer boolmatch)rK patternr r r
_needs_hiding s
r` c C s t ttj} t| dS )a%
Remove references to setuptools' modules from sys.modules to allow the
invocation to import the most appropriate setuptools. This technique is
necessary to avoid issues such as #315 where setuptools upgrading itself
would fail to find a function declared in the metadata.
N)filterr` r rM rP )rM r r r rX s rX c C s t jt j| }t| z| gt| tjdd< tjd| t
t j dd t
| trl| n | t }t| t|dd}t| | W 5 Q R X W n4 tk
r } z|jr|jd rʂ W 5 d}~X Y nX W 5 Q R X dS )z8Run a distutils setup script, sandboxed in its directoryNr c S s | S r )Zactivate)Zdistr r r zrun_setup..__main__)__file__r7 )r( r abspathdirnamer[ rS r r insertr __init__Z callbacksappend
isinstancestrencodegetfilesystemencodingr dictr
SystemExitargs)Z setup_scriptrq rZ Z dunder_filensvr r r r s"
c @ s e Zd ZdZdZdd Zdd Zdd Zd d
Zd d Z d
d Z
dD ] Z e e
e rDe
e e e < qDd$ddZerzedeZedeZdD ] Z e e
e ree e e < qdd ZdD ] Z e e
e ree e e < qdd ZdD ] Z e e
e ree e e < qd d Zd d Zd d! Zd"d# ZdS )%r zDWrap 'os' module and 'open()' builtin for virtualizing setup scriptsFc s fddt tD _d S )Nc s$ g | ] }| d st |r|qS )_)rI hasattr)rJ namer= r r
s
z,AbstractSandbox.__init__..)dir_os_attrsr= r r= r ri s
zAbstractSandbox.__init__c C s" | j D ]}tt|t|| qd S r )rz setattrr( getattr)r> sourcerv r r r _copy s
zAbstractSandbox._copyc C s( | | tr| jt_| jt_d| _d S r@ )r~ _filer file_openr _activer= r r r r? s
zAbstractSandbox.__enter__c C s$ d| _ trtt_tt_| t d S NF)r r r r r r r~ ry )r> exc_type exc_value tracebackr r r rD s
zAbstractSandbox.__exit__c
C s" | | W 5 Q R S Q R X dS )z Run 'func' under os sandboxingNr )r> funcr r r run s zAbstractSandbox.runc s t t fdd}|S )Nc s2 | j r | j ||f||\}}||f||S r )r _remap_pair)r> srcdstrq kwrv originalr r wrap& s z3AbstractSandbox._mk_dual_path_wrapper..wrapr| ry rv r r r r _mk_dual_path_wrapper# s
z%AbstractSandbox._mk_dual_path_wrapper)renamelinksymlinkNc s p t t fdd}|S )Nc s* | j r| j |f||}|f||S r )r _remap_inputr> r rq r r r r r 4 s z5AbstractSandbox._mk_single_path_wrapper..wrapr )rv r r r r r _mk_single_path_wrapper1 s z'AbstractSandbox._mk_single_path_wrapperr r )statlistdirr* r chmodchownmkdirremoveunlinkrmdirutimelchownchrootlstatZ startfilemkfifomknodpathconfaccessc s t t fdd}|S )Nc sB | j r2| j |f||}| |f||S |f||S r )r r
_remap_outputr r r r r I s z4AbstractSandbox._mk_single_with_return..wrapr r r r r _mk_single_with_returnF s
z&AbstractSandbox._mk_single_with_return)readlinktempnamc s t t fdd}|S )Nc s ||}| j r | |S |S r )r r )r> rq r Zretvalr r r r X s
z'AbstractSandbox._mk_query..wrapr r r r r _mk_queryU s
zAbstractSandbox._mk_query)r) tmpnamc C s |S )z=Called to remap or validate any path, whether input or outputr )r> r r r r _validate_pathd s z AbstractSandbox._validate_pathc O s
| |S zCalled for path inputsr r> operationr rq r r r r r h s z AbstractSandbox._remap_inputc C s
| |S )zCalled for path outputsr )r> r r r r r r l s z AbstractSandbox._remap_outputc O s0 | j |d |f||| j |d |f||fS )?Called for path pairs like rename, link, and symlink operationsz-fromz-to)r r> r r r rq r r r r r p s zAbstractSandbox._remap_pair)N)r7 r8 r9 r: r ri r~ r? rD r r rv ru ry r r r r r r r r r r r r r r r s<
devnullc @ s e Zd ZdZedddddddd d
d d d
dg
ZdgZefddZ dd Z
e rXd'ddZ d(ddZ dd Z
dd Zd d Zd d Zd!d" Zd)d$d%Zd&S )*r z.)
r( r r r _sandboxrW _prefix _exceptionsr ri )r> Zsandbox
exceptionsr r r ri s zDirectorySandbox.__init__c O s ddl m} ||||d S )Nr )r
)r1 r
)r> r rq r r
r r r
_violation s zDirectorySandbox._violationrc O s: |dkr(| |s(| jd||f|| t||f||S )Nr Zrtr ZrUUr )_okr r r> r r rq r r r r r s zDirectorySandbox._filec O s: |dkr(| |s(| jd||f|| t||f||S )Nr r )r r r r r r r r s zDirectorySandbox._openc C s | d d S )Nr )r r= r r r r s zDirectorySandbox.tmpnamc C sR | j }z>d| _ tjtj|}| |p@|| jkp@|| jW S || _ X d S r ) r r( r r r _exemptedr rI r )r> r Zactiver r r r r s
zDirectorySandbox._okc s< fdd| j D } fdd| jD }t||}t|S )Nc 3 s | ]} |V qd S r rH )rJ Z exceptionfilepathr r rL s z-DirectorySandbox._exempted..c 3 s | ]}t | V qd S r )r\ r^ )rJ r_ r r r rL s )r _exception_patterns itertoolschainany)r> r Z
start_matchesZpattern_matchesZ
candidatesr r r r s
zDirectorySandbox._exemptedc O s4 || j kr0| |s0| j|tj|f|| |S r ) write_opsr r r( r r r r r r r s z DirectorySandbox._remap_inputc O s2 | |r| |s*| j|||f|| ||fS )r )r r r r r r r s z DirectorySandbox._remap_pair c O s@ |t @ r*| |s*| jd|||f|| tj|||f||S )z Called for low-level os.open()zos.open) WRITE_FLAGSr r ry r )r> r flagsr rq r r r r r s zDirectorySandbox.openN)r )r )r )r7 r8 r9 r: ro fromkeysr r _EXCEPTIONSri r r r r r r r r r r r r r r ~ s:
c C s g | ]}t t|d qS )r r )rJ ar r r rw s rw z4O_WRONLY O_RDWR O_APPEND O_CREAT O_TRUNC O_TEMPORARYc @ s& e Zd ZdZed Zdd ZdS )r
zEA setup script attempted to modify the filesystem outside the sandboxa
SandboxViolation: {cmd}{args!r} {kwargs}
The package setup script has attempted to modify files on your system
that are not within the EasyInstall build area, and has been aborted.
This package cannot be safely installed by EasyInstall, and may not
support alternate installation locations even if you run its setup
script by hand. Please inform the package's author and the EasyInstall
maintainers to find out if a fix or workaround is available.
c C s | j \}}}| jjf t S r )rq tmplformatr )r> cmdrq kwargsr r r __str__ s zSandboxViolation.__str__N) r7 r8 r9 r: textwrapdedentlstripr r r r r r r
s )N)N)=r( r r& operator functoolsr r\
contextlibr. r Zsetuptools.externr Zsetuptools.extern.six.movesr r Zpkg_resources.py31compatr$ platformrI Z$org.python.modules.posix.PosixModulepythonrM posixZ PosixModulery rv r r NameErrorr r Zdistutils.errorsr r __all__r contextmanagerr r" r' r, r0 r- r< rR rP rV r[ r` rX r r ru r r r reduceor_splitr r
r r r r sx
w
V