File: //usr/local/lib/python3.10/test/support/__pycache__/script_helper.cpython-310.opt-2.pyc
o
�i�, � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z d dl m
Z
dadd� ZG dd� de �dd ��Zd
d � Zd d
� Zdd� Zdd� Zejejd�dd�Zdd� Zd$dd�Zd%dd�Zd&d d �Z d'd d!�Z d"d#� Z dS )(� N)�source_from_cache)�support)�make_legacy_pycc C sX t d u r*dtjv rda dS z t�tjdddg� W n
tjy' da Y t S w da t S )NZ
PYTHONHOMET�-Ez-czimport sys; sys.exit(0)F)�$__cached_interp_requires_environment�os�environ�
subprocess�
check_call�sys�
executable�CalledProcessError� r r �7/usr/local/lib/python3.10/test/support/script_helper.py� interpreter_requires_environment s
��r c @ s e Zd Z dd� ZdS )�_PythonRunResultc C s� d}| j | j}}t|�|krd|| d � }t|�|kr(d|| d � }|�dd��� }|�dd��� }td| j|||f ��)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 : s r )r r r c O sB t � }|�dd �}d|v r|�d�}n| o| }tjddg}|r'|�d� n |s0|s0|�d� |�dd �rEi }tjdkrDtjd |d < ntj�� }d
|vrRd |d
< |� |� |�
| � t j |t j
t j
t j
||d �}|�" z|�� \}} W |�� t �� n |�� t �� w W d � n1 s�w Y |j}
t|
|| �|fS )
NZ__cwdZ
__isolated�-X� faulthandlerz-Ir Z
__cleanenv�win32Z
SYSTEMROOT�TERM� )�stdin�stdout�stderr�env�cwd)r �popr r �append�platformr r �copy�update�extendr �Popen�PIPE� communicate�kill�_cleanup�
returncoder ) �args�env_varsZ env_requiredr+ �isolatedr r* �procr r r r r r �run_python_until_endZ sD
�
�
� �r< c O s4 t |i |��\}}|jr| s|js| s|�|� |S �N)r< r r )Zexpected_successr8 r9 �resr r r r �_assert_python� s
r? c O � t dg| �R i |��S )NT�r? �r8 r9 r r r �assert_python_ok� s
rC c O r@ )NFrA rB r r r �assert_python_failure� s rD )r( r) c O sZ t jg}t� s
|�d� |�|� |�dttj��}d|d< t j
|ft j | |d�|��S )Nr r* Zvt100r% )r' r( r) ) r r r r- r1 �
setdefault�dictr r r r2 r3 )r( r) r8 �kwr r* r r r � spawn_python� s
��rH c C s4 | j �� | j�� }| j�� | �� t�� |S r= )r' �closer( �read�waitr r6 )�p�datar r r � kill_python� s
rN Fc C sh |}|s |t jd 7 }t j�| |�}t|ddd��
}|�|� W d � n1 s)w Y t�� |S )N�py�wzutf-8)�encoding)r �extsep�path�join�open�write� importlib�invalidate_caches)Z
script_dir�script_basename�sourceZ omit_suffixZscript_filename� script_nameZ script_filer r r � make_script� s �r\ c C s� |t j d }t j�| |�}t�|d��9}|d u r@|�t j�}t|�dkr:|d dkr:t t
|��}t j� |�}|}nt j� |�}|� ||� W d � n1 sPw Y |t j�||�fS )N�ziprP � ���� __pycache__)
r rR rS rT �zipfile�ZipFile�split�sepr r r �basenamerV ) �zip_dir� zip_basenamer[ Z name_in_zip� zip_filename�zip_name�zip_file�partsZ
legacy_pycr r r �make_zip_script� s �rl r& c C s t �| � t| d|� d S )N�__init__)r �mkdirr\ )Zpkg_dirZ init_sourcer r r �make_pkg� s
ro � c s> g }t | dd�}|�|� tj�|�} t | ||�}
|�|
� |r5tj|dd�}tj|
dd�}
|�||
f� � fdd�td|d �D �} tj� | d tj�|
��} |tj
d }
tj� | |
�}t � |d
�� }| D ]}tj� || �}|�
||� qg|�
|
| � W d � n1 s�w Y |D ]}t�|� q�|tj� || �fS ) Nrm r& T)�doraisec s g | ] }t j�� g| ��qS r )r rd rT )�.0�i��pkg_namer r �
<listcomp> s z make_zip_pkg.<locals>.<listcomp>rp ���r] rP )r\ r- r rS re �
py_compile�compiler1 �rangerT rR ra rb rV �unlink)rf rg ru rY rZ �depthZcompiledr{ Z init_nameZ
init_basenamer[ Z pkg_namesZscript_name_in_ziprh ri rj �nameZinit_name_in_zipr rt r � make_zip_pkg� s.
� r~ c C s� t jrDdd� }dtj�| �� �}t� t||�dd� tjdddd | d
g}t� |�}t||� d |j
� ��dd� |j
rBt |� d ���d S t d | d
� d S )
Nc S s d| � d�S )Nz===== z ======r )�textr r r �title s z run_test_script.<locals>.titlezscript T)�flushr r" r# z-uz-vz completed: exit code z failed)
r �verboser rS re �printr r r �runr7 r rC )Zscriptr� r} r8 r; r r r �run_test_script s
��r� )Fr= )r& )rp F) � collectionsrW r r Zos.pathr rx ra �importlib.utilr �testr Ztest.support.import_helperr r r �
namedtupler r< r? rC rD r3 �STDOUTrH rN r\ rl ro r~ r� r r r r �<module> s8 $� 4
�