3
Lj) @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl
mZmZ dadd ZG dd de jdd#Zd d Zd
d Zdd Zdd ZejejdddZdd Zd$ddZ d%dd Z d&d d Z d'd!d"Z dS )( N)source_from_cache)make_legacy_pycstrip_python_stderrc C sV t d krRdtjkrda dS ytjtjdddg W n tjk
rL da Y nX da t S )NZ
PYTHONHOMETz-Ez-czimport sys; sys.exit(0)F)$__cached_interp_requires_environmentosenviron
subprocessZ
check_callsys
executableZCalledProcessError r r 2/usr/lib64/python3.6/test/support/script_helper.py interpreter_requires_environment s
r
c @ s e Zd Zdd ZdS )_PythonRunResultc C s d}| j | j }}t||kr0d|| d }t||krNd|| d }|jddj }|jddj }td| j|||f d S ) NP d s (... truncated stdout ...)s (... truncated stderr ...)asciireplacezRProcess return code is %d
command line: %r
stdout:
---
%s
---
stderr:
---
%s
---i@ )outerrlendecoderstripAssertionErrorrc)selfcmd_linemaxlenr r r r r fail> s z_PythonRunResult.failN)__name__
__module__ __qualname__r r r r r r ; s r r r r c
O s t }d|kr|jd}n | o$| }tjddg}|rB|jd n| rX| rX|jd |jdd ri }tjdkrtjd |d< n
tjj }d |krd
|d < |j | |j
| t j |t j
t j
t j
|d }|* z|j \}}W d |j t j X W d Q R X |j} t|}t| |||fS ) NZ
__isolatedz-XZ faulthandlerz-Iz-EZ
__cleanenvZwin32Z
SYSTEMROOTTERM )stdinstdoutstderrenv)r
popr r
appendplatformr r copyupdateextendr PopenPIPEZ communicatekill_cleanup
returncoder r )
argsenv_varsZ env_requiredisolatedr r& procr r r r r r run_python_until_end[ s:
r6 c O s4 t ||\}}|jr| s&|j r0| r0|j| |S )N)r6 r r )Zexpected_successr2 r3 resr r r r _assert_python s
r8 c O s t d| |S )NT)T)r8 )r2 r3 r r r assert_python_ok s r9 c O s t d| |S )NF)F)r8 )r2 r3 r r r assert_python_failure s r: )r$ r% c O sX t jg}t s|jd |j| |jdttj}d|d< t j
|ft j | |d|S )Nz-Er& Zvt100r! )r# r$ r% ) r r
r
r( r,
setdefaultdictr r r r- r. )r$ r% r2 kwr r& r r r spawn_python s
r> c C s2 | j j | jj }| jj | j tj |S )N)r# closer$ readwaitr r0 )pdatar r r kill_python s
rD Fc C sP |}|s|t jd 7 }t jj| |}t|ddd}|j| |j tj |S )Npywzutf-8)encoding) r extseppathjoinopenwriter? importlibinvalidate_caches)Z
script_dirscript_basenamesourceZ omit_suffixZscript_filename script_nameZ script_filer r r make_script s
rR c C s |t j d }t jj| |}tj|d}|d kr~|jt j}t|dkrr|d dkrrt t
|}t jj |}|}n t jj |}|j || |j
|t jj||fS )NziprF __pycache__)r rH rI rJ zipfileZipFilesplitsepr r r basenamerL r? ) zip_dir zip_basenamerQ Z name_in_zip zip_filenamezip_namezip_filepartsZ
legacy_pycr r r make_zip_script s rb r" c C s t j| t| d| d S )N__init__)r mkdirrR )Zpkg_dirZ init_sourcer r r make_pkg s
re c s0 g }t | dd}|j| tjj|} t | ||}
|j|
|rjtj|dd}tj|
dd}
|j||
f fddtd|d D } tjj | d
tjj|
} |tj
d }
tjj | |
}t j |d }x&| D ] }tjj || }|j
|| qW |j
|
| |j x|D ]}tj| q
W |tjj || fS ) Nrc r" T)doraisec s g | ]}t jj g| qS r )r rZ rJ ).0i)pkg_namer r
s z make_zip_pkg..rf rS rF )rR r( r rI r[
py_compilecompiler, rangerJ rH rW rX rL r? unlink)r\ r] rj rO rP ZdepthZcompiledrp Z init_nameZ
init_basenamerQ Z pkg_namesZscript_name_in_zipr^ r_ r` nameZinit_name_in_zipr )rj r make_zip_pkg s.
rr )r r r )F)N)r" )rf F) collectionsrM r r Zos.pathZtempfiler rm
contextlibZshutilrW importlib.utilr Z test.supportr r r r
namedtupler r6 r8 r9 r: r. ZSTDOUTr> rD rR rb re rr r r r r s4 $ 3