File: //usr/local/lib/python3.10/test/support/__pycache__/script_helper.cpython-310.opt-1.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 sV t du r)dtjv r
da dS z t�tjdddg� W n
tjy& da Y t S w da t S )a
Returns True if our sys.executable interpreter requires environment
variables in order to be able to run at all.
This is designed to be used with @unittest.skipIf() to annotate tests
that need to use an assert_python*() function to launch an isolated
mode (-I) or no environment mode (-E) sub-interpreter process.
A normal build & test does not run into this situation but it can happen
when trying to run the standard library test suite from an interpreter that
doesn't have an obvious home with Python's current home finding logic.
Setting PYTHONHOME is one way to get most of the testsuite to run in that
situation. PYTHONPATH or PYTHONUSERSITE are other common environment
variables that might impact whether or not the interpreter can start.
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 ZdZdd� ZdS )�_PythonRunResultz2Helper for reporting Python subprocess run resultsc C s� d}| j | j}}t|�|krd|| d� }t|�|kr'd|| d� }|�dd��� }|�dd��� }td| j|||f ��)z4Provide helpful details about failed subcommand runsi@ s (... truncated stdout ...)Ns (... 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__�__doc__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_successr9 r: �resr r r r �_assert_python� s
r@ c O � t dg| �R i |��S )a|
Assert that running the interpreter with `args` and optional environment
variables `env_vars` succeeds (rc == 0) and return a (return code, stdout,
stderr) tuple.
If the __cleanenv keyword is set, env_vars is used as a fresh environment.
Python is started in isolated mode (command line option -I),
except if the __isolated keyword is set to False.
T�r@ �r9 r: r r r �assert_python_ok� s rD c O rA )z�
Assert that running the interpreter with `args` and optional environment
variables `env_vars` fails (rc != 0) and return a (return code, stdout,
stderr) tuple.
See assert_python_ok() for more options.
FrB rC r r r �assert_python_failure� s rE )r) r* c O sX t jg}t� s |�d� |�|� |�dttj��}d|d< t j
|ft j | |d�|��S )z�Run a Python subprocess with the given arguments.
kw is extra keyword args to pass to subprocess.Popen. Returns a Popen
object.
r r+ Zvt100r& )r( r) r* ) r r r r. r2 �
setdefault�dictr r r r3 r4 )r) r* r9 �kwr r+ r r r � spawn_python� s
��rI c C s2 | j �� | j�� }| j�� | �� t�� |S )z?Run the given Popen process until completion and return stdout.)r( �closer) �read�waitr r7 )�p�datar r r � kill_python� s
rO 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�ziprQ � ���� __pycache__)
r rS rT rU �zipfile�ZipFile�split�sepr r r �basenamerW ) �zip_dir� zip_basenamer\ Z name_in_zip� zip_filename�zip_name�zip_file�partsZ
legacy_pycr r r �make_zip_script� s �rm r' c C s t �| � t| d|� d S )N�__init__)r �mkdirr] )Zpkg_dirZ init_sourcer r r �make_pkg� s
rp � 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 ) Nrn r' T)�doraisec s g | ] }t j�� g| ��qS r )r re rU )�.0�i��pkg_namer r �
<listcomp> s z make_zip_pkg.<locals>.<listcomp>rq ���r^ rQ )r] r. r rT rf �
py_compile�compiler2 �rangerU rS rb rc rW �unlink)rg rh rv rZ r[ �depthZcompiledr| Z init_nameZ
init_basenamer\ Z pkg_namesZscript_name_in_zipri rj rk �nameZinit_name_in_zipr ru 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 rT rf �printr r r �runr8 r rD )Zscriptr� r~ r9 r< r r r �run_test_script s
��r� )Fr> )r' )rq F) � collectionsrX r r Zos.pathr ry rb �importlib.utilr �testr Ztest.support.import_helperr r r �
namedtupler r= r@ rD rE r4 �STDOUTrI rO r] rm rp r r� r r r r �<module> s8 $� 4
�