File: //usr/local/lib/python3.7/test/support/__pycache__/script_helper.cpython-37.opt-2.pyc
B
��g�) � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z m Z da
dd� ZG dd� de �dd��Zd d
� Zd d � Zd
d� Zdd� Zejejd�dd�Zdd� Zd!dd�Zd"dd�Zd#d d �Zd$d d �Z dS )%� N)�source_from_cache)�make_legacy_pyc�strip_python_stderrc C sV t d krRdtjkrda dS yt�tjdddg� W n tjk
rL da Y nX da t S )NZ
PYTHONHOMETz-Ez-czimport sys; sys.exit(0)F)�$__cached_interp_requires_environment�os�environ�
subprocessZ
check_call�sys�
executableZCalledProcessError� r r �6/usr/local/lib/python3.7/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 � }|�dd��� }|�dd��� }td| j|||f ��d S )Ni@ s (... truncated stdout ...)s (... truncated stderr ...)�ascii�replacezRProcess return code is %d
command line: %r
stdout:
---
%s
---
stderr:
---
%s
---)�out�err�len�decode�rstrip�AssertionError�rc)�self�cmd_line�maxlenr r r r r �fail; s z_PythonRunResult.failN)�__name__�
__module__� __qualname__r r r r r r 8 s r )r r r c O s& t � }|�dd �}d|kr&|�d�}n | o0| }tjddg}|rN|�d� n|s`|s`|�d� |�dd �r�i }tjdkr�tjd |d < n
tj�� }d
|kr�d |d
< |� |� |�
| � t j |t j
t j
t j
||d �}|�* z|�� \}} W d |�� t �� X W d Q R X |j}
t| �} t|
|| �|fS )
NZ__cwdZ
__isolatedz-XZ faulthandlerz-Iz-EZ
__cleanenv�win32Z
SYSTEMROOT�TERM� )�stdin�stdout�stderr�env�cwd)r
�popr r
�append�platformr r �copy�update�extendr �Popen�PIPEZ communicate�kill�_cleanup�
returncoder r ) �args�env_varsZ env_requiredr& �isolatedr r% �procr r r r r r �run_python_until_endX s<
r6 c O s0 t ||�\}}|jr| s"|js,| s,|�|� |S )N)r6 r r )Zexpected_successr2 r3 Zresr r r r �_assert_python� s
r7 c O s t d| �|�S )NT)T)r7 )r2 r3 r r r �assert_python_ok� s r8 c O s t d| �|�S )NF)F)r7 )r2 r3 r r r �assert_python_failure� s r9 )r# r$ c O sX t jg}t� s|�d� |�|� |�dttj��}d|d< t j
|ft j | |d�|��S )Nz-Er% Zvt100r )r" r# r$ ) r r
r
r( r, �
setdefault�dictr r r r- r. )r# r$ r2 �kwr r% r r r � spawn_python� s
r= c C s2 | j �� | j�� }| j�� | �� t�� |S )N)r" �closer# �read�waitr r0 )�p�datar r r � kill_python� s
rC Fc C sP |}|s|t jd 7 }t j�| |�}t|ddd�}|�|� |�� t�� |S )N�py�wzutf-8)�encoding) r �extsep�path�join�open�writer> � importlib�invalidate_caches)Z
script_dir�script_basename�sourceZ omit_suffixZscript_filename� script_nameZ script_filer r r � make_script� s
rQ c C s� |t j d }t j�| |�}t�|d�}|d kr~|�t j�}t|�dkrr|d dkrrt t
|��}t j� |�}|}n t j� |�}|� ||� |�
� |t j�||�fS )N�ziprE � ���� __pycache__)r rG rH rI �zipfile�ZipFile�split�sepr r r �basenamerK r> ) �zip_dir� zip_basenamerP Z name_in_zip� zip_filename�zip_name�zip_file�partsZ
legacy_pycr r r �make_zip_script� s ra r! c C s t �| � t| d|� d S )N�__init__)r �mkdirrQ )Zpkg_dirZ init_sourcer r r �make_pkg� s
rd � c s0 g }t | dd�}|�|� tj�|�} t | ||�}
|�|
� |rjtj|dd�}tj|
dd�}
|�||
f� � fdd�td|d �D �} tj� | d tj�|
��} |tj
d }
tj� | |
�}t � |d
�}x&| D ] }tj� || �}|�
||� q�W |�
|
| � |�� x|D ]}t�|� �q
W |tj� || �fS ) Nrb r! T)�doraisec s g | ]}t j�� g| ��qS r )r rY rI )�.0�i)�pkg_namer r �
<listcomp>� s z make_zip_pkg.<locals>.<listcomp>re ���rR rE )rQ r( r rH rZ �
py_compile�compiler, �rangerI rG rV rW rK r> �unlink)r[ r\ ri rN rO �depthZcompiledro Z init_nameZ
init_basenamerP Z pkg_namesZscript_name_in_zipr] r^ r_ �nameZinit_name_in_zipr )ri r � make_zip_pkg� s.
rr )F)N)r! )re F) � collectionsrL r r Zos.pathr rl rV �importlib.utilr Z test.supportr r r r
�
namedtupler r6 r7 r8 r9 r. ZSTDOUTr= rC rQ ra rd rr r r r r �<module> s. $ 4