File: //usr/local/lib/python3.10/test/libregrtest/__pycache__/save_env.cpython-310.opt-1.pyc
o
�i�2 � @ sn d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
G dd� de �Z G dd� d�Z
dS ) � N)�support)� os_helper)�
print_warningc @ s e Zd ZdS )�SkipTestEnvironmentN)�__name__�
__module__� __qualname__� r r �6/usr/local/lib/python3.10/test/libregrtest/save_env.pyr s r c @ s e Zd ZdZd�dd�dd�ZdZdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#dDdE� Z$dFdG� Z%dHdI� Z&dJdK� Z'dLdM� Z(dNdO� Z)dPdQ� Z*dRdS� Z+dTdU� Z,dVdW� Z-dXdY� Z.dZd[� Z/d\d]� Z0d^d_� Z1d`da� Z2dbdc� Z3ddde� Z4dfdg� Z5dhdi� Z6djdk� Z7dldm� Z8dndo� Z9dpdq� e:e;�D �Z<drds� Z=dtdu� Z>dvdw� Z?dxdy� Z@dzd{� ZAd|d}� ZBd~d� ZCd�S )��saved_test_environmenta� Save bits of the test environment and restore them at block exit.
with saved_test_environment(testname, verbose, quiet):
#stuff
Unless quiet is True, a warning is printed to stderr if any of
the saved items was changed by the test. The support.environment_altered
attribute is set to True if a change is detected.
If verbose is more than 1, the before and after state of changed
items is also printed.
r F)�pgoc C s || _ || _|| _|| _d S �N)�testname�verbose�quietr )�selfr r r r r r r
�__init__% s
z saved_test_environment.__init__)zsys.argv�cwdz sys.stdinz
sys.stdoutz
sys.stderrz
os.environzsys.pathzsys.path_hooks�
__import__zwarnings.filterszasyncore.socket_mapzlogging._handlerszlogging._handlerListz sys.gettracezsys.warnoptionsz!multiprocessing.process._danglingzthreading._danglingzsysconfig._CONFIG_VARSzsysconfig._INSTALL_SCHEMES�files�localezwarnings.showwarningZshutil_archive_formatsZshutil_unpack_formatsz!asyncio.events._event_loop_policyz urllib.requests._url_tempfileszurllib.requests._openerc C s
t j| S r
)�sys�modules�r �namer r r
�
get_moduleD �
z!saved_test_environment.get_modulec C s z| � |�W S ty t�w r
)r �KeyErrorr r r r r
�try_get_moduleH s
�z%saved_test_environment.try_get_modulec C s | � d�}t|j�S �Nzurllib.request)r �listZ_url_tempfiles�r �urllib_requestr r r
�"get_urllib_requests__url_tempfilesO �
z9saved_test_environment.get_urllib_requests__url_tempfilesc C s |D ]}t �|� qd S r
)r �unlink)r Z tempfiles�filenamer r r
�&restore_urllib_requests__url_tempfilesR s �z=saved_test_environment.restore_urllib_requests__url_tempfilesc C � | � d�}|jS r )r �_openerr! r r r
�get_urllib_requests__openerV �
z2saved_test_environment.get_urllib_requests__openerc C � | � d�}||_d S r )r r) )r �openerr" r r r
� restore_urllib_requests__openerY r$ z6saved_test_environment.restore_urllib_requests__openerc C s | � d� t�� S �N�asyncio)r r Zmaybe_get_event_loop_policy�r r r r
�%get_asyncio_events__event_loop_policy] s
z<saved_test_environment.get_asyncio_events__event_loop_policyc C s | � d�}|�|� d S r/ )r Zset_event_loop_policy)r Zpolicyr0 r r r
�)restore_asyncio_events__event_loop_policy` s
z@saved_test_environment.restore_asyncio_events__event_loop_policyc C � t tj�tjtjd d � fS r
)�idr �argvr1 r r r
� get_sys_argvd � z#saved_test_environment.get_sys_argvc C � |d t _|d t jd d �< d S �N� � )r r6 )r Z
saved_argvr r r
�restore_sys_argvf �
z'saved_test_environment.restore_sys_argvc C � t �� S r
)�os�getcwdr1 r r r
�get_cwdj � z saved_test_environment.get_cwdc C � t �|� d S r
)r@ �chdir)r Z saved_cwdr r r
� restore_cwdl � z"saved_test_environment.restore_cwdc C � t jS r
�r �stdoutr1 r r r
�get_sys_stdouto � z%saved_test_environment.get_sys_stdoutc C �
|t _d S r
rI )r Z saved_stdoutr r r
�restore_sys_stdoutq �
z)saved_test_environment.restore_sys_stdoutc C rH r
�r �stderrr1 r r r
�get_sys_stderrt rL z%saved_test_environment.get_sys_stderrc C rM r
rP )r Z saved_stderrr r r
�restore_sys_stderrv rO z)saved_test_environment.restore_sys_stderrc C rH r
�r �stdinr1 r r r
�
get_sys_stdiny rL z$saved_test_environment.get_sys_stdinc C rM r
rT )r Z saved_stdinr r r
�restore_sys_stdin{ rO z(saved_test_environment.restore_sys_stdinc C s t tj�tjttj�fS r
)r5 r@ �environ�dictr1 r r r
�get_os_environ~ s z%saved_test_environment.get_os_environc C s( |d t _t j�� t j�|d � d S r: )r@ rX �clear�update)r Z
saved_environr r r
�restore_os_environ� s
z)saved_test_environment.restore_os_environc C r4 r
)r5 r �pathr1 r r r
� get_sys_path� r8 z#saved_test_environment.get_sys_pathc C r9 r: )r r^ )r Z
saved_pathr r r
�restore_sys_path� r> z'saved_test_environment.restore_sys_pathc C r4 r
)r5 r �
path_hooksr1 r r r
�get_sys_path_hooks� r8 z)saved_test_environment.get_sys_path_hooksc C r9 r: )r ra )r Z saved_hooksr r r
�restore_sys_path_hooks� r> z-saved_test_environment.restore_sys_path_hooksc C r? r
)r �gettracer1 r r r
�get_sys_gettrace� rC z'saved_test_environment.get_sys_gettracec C rD r
)r �settrace)r Z trace_fxnr r r
�restore_sys_gettrace� rG z+saved_test_environment.restore_sys_gettracec C rH r
��builtinsr r1 r r r
�get___import__� rL z%saved_test_environment.get___import__c C rM r
rh )r �import_r r r
�restore___import__� rO z)saved_test_environment.restore___import__c C �& | � d�}t|j�|j|jd d � fS �N�warnings)r r5 �filters�r ro r r r
�get_warnings_filters� s
z+saved_test_environment.get_warnings_filtersc C s* | � d�}|d |_|d |jd d �< d S )Nro r; r< )r rp )r Z
saved_filtersro r r r
�restore_warnings_filters� s
z/saved_test_environment.restore_warnings_filtersc C s t j�d�}|r
|j�� pi S )N�asyncore)r r �get�
socket_map�copy)r rt r r r
�get_asyncore_socket_map� s z.saved_test_environment.get_asyncore_socket_mapc C s4 t j�d�}|d ur|jdd� |j�|� d S d S )Nrt T)Z
ignore_all)r r ru Z close_allrv r\ )r Z saved_maprt r r r
�restore_asyncore_socket_map� s
�z2saved_test_environment.restore_asyncore_socket_mapc C � | � d�}|j|j�� fS �N�shutil)r �_ARCHIVE_FORMATSrw �r r| r r r
�get_shutil_archive_formats� s
z1saved_test_environment.get_shutil_archive_formatsc C �2 | � d�}|d |_|j�� |j�|d � d S �Nr| r r; )r r} r[ r\ �r �savedr| r r r
� restore_shutil_archive_formats� �
z5saved_test_environment.restore_shutil_archive_formatsc C rz r{ )r �_UNPACK_FORMATSrw r~ r r r
�get_shutil_unpack_formats� �
z0saved_test_environment.get_shutil_unpack_formatsc C r� r� )r r� r[ r\ r� r r r
� restore_shutil_unpack_formats� r� z4saved_test_environment.restore_shutil_unpack_formatsc C �" | � d�}t|j�|j|j�� fS �N�logging)r r5 � _handlersrw �r r� r r r
�get_logging__handlers� s
z,saved_test_environment.get_logging__handlersc C � d S r
r )r Zsaved_handlersr r r
�restore_logging__handlers� � z0saved_test_environment.restore_logging__handlersc C rm r� )r r5 � _handlerListr� r r r
�get_logging__handlerList� s
z/saved_test_environment.get_logging__handlerListc C r� r
r )r Zsaved_handlerListr r r
� restore_logging__handlerList� r� z3saved_test_environment.restore_logging__handlerListc C r4 r
)r5 r � warnoptionsr1 r r r
�get_sys_warnoptions� r8 z*saved_test_environment.get_sys_warnoptionsc C r9 r: )r r� )r Z
saved_optionsr r r
�restore_sys_warnoptions� r> z.saved_test_environment.restore_sys_warnoptionsc C s
t j�� S r
)� threading� _danglingrw r1 r r r
�get_threading__dangling� r z.saved_test_environment.get_threading__danglingc C s t j�� t j�|� d S r
)r� r� r[ r\ )r r� r r r
�restore_threading__dangling� r� z2saved_test_environment.restore_threading__danglingc C s | � d�}|�� |j�� S �Nzmultiprocessing.process)r �_cleanupr� rw )r �multiprocessing_processr r r
�%get_multiprocessing_process__dangling� s
z<saved_test_environment.get_multiprocessing_process__danglingc C s$ | � d�}|j�� |j�|� d S r� )r r� r[ r\ )r r� r� r r r
�)restore_multiprocessing_process__dangling� s
z@saved_test_environment.restore_multiprocessing_process__danglingc C s, | � d�}|�d� t|j�|jt|j�fS )N� sysconfig�prefix)r Zget_config_varr5 � _CONFIG_VARSrY �r r� r r r
�get_sysconfig__CONFIG_VARS� s
�z1saved_test_environment.get_sysconfig__CONFIG_VARSc C r� �Nr� r; r< )r r� r[ r\ �r r� r� r r r
� restore_sysconfig__CONFIG_VARS� r� z5saved_test_environment.restore_sysconfig__CONFIG_VARSc C r� )Nr� )r r5 �_INSTALL_SCHEMESrw r� r r r
� get_sysconfig__INSTALL_SCHEMES� s
�z5saved_test_environment.get_sysconfig__INSTALL_SCHEMESc C r� r� )r r� r[ r\ r� r r r
�"restore_sysconfig__INSTALL_SCHEMES� r� z9saved_test_environment.restore_sysconfig__INSTALL_SCHEMESc C s t dd� t�� D ��S )Nc s s( � | ]}|t j�|�r
d nd V qdS )�/� N)r@ r^ �isdir)�.0�fnr r r
� <genexpr> s � �z3saved_test_environment.get_files.<locals>.<genexpr>)�sortedr@ �listdirr1 r r r
� get_files s �z saved_test_environment.get_filesc C sZ t j}||vr'|d |vr)tj�|�rt �|� d S tj�|�r+t �|� d S d S d S d S )Nr� )r ZTESTFNr@ r^ �isfiler% r� �rmtree)r Z saved_valuer� r r r
�
restore_files s �z$saved_test_environment.restore_filesc C s g | ] }|� d �rtt|��qS )�LC_)�
startswith�getattrr )r� �lcr r r
�
<listcomp> s �z!saved_test_environment.<listcomp>c
C sD g }| j D ]}z
|�|t�|d �f� W q ttfy Y qw |S r
)�_lc�appendr � setlocale� TypeError�
ValueError)r Zpairingsr� r r r
�
get_locale s
�z!saved_test_environment.get_localec C s |D ]
\}}t �||� qd S r
)r r� )r r� r� Zsettingr r r
�restore_locale s �z%saved_test_environment.restore_localec C r( rn )r � showwarningrq r r r
�get_warnings_showwarning r+ z/saved_test_environment.get_warnings_showwarningc C r, rn )r r� )r �fxnro r r r
� restore_warnings_showwarning r$ z3saved_test_environment.restore_warnings_showwarningc c sF � | j D ] }|�dd�}d| }d| }|t| |�t| |�fV qd S )N�.�_Zget_Zrestore_)� resources�replacer� )r r Z
method_suffix�get_nameZ restore_namer r r
�
resource_info# s �
�z$saved_test_environment.resource_infoc C sP g | _ | �� D ] \}}}z|� }W n ty Y qw | j �||||f� q| S r
)� saved_valuesr� r r� )r r ru �restore�originalr r r
� __enter__* s
�z saved_test_environment.__enter__c
C s� | j }d | _ t�� |D ]3\}}}}|� } | |kr?dt_||� | js?| js?t|� d| j� �� td|� d| � d�t j
dd� q dS )NTz was modified by z
Before: z
After: � )�file�flushF) r� r �
gc_collectZenvironment_alteredr r r r �printr rQ )
r �exc_type�exc_val�exc_tbr� r ru r� r� �currentr r r
�__exit__5 s ��z saved_test_environment.__exit__N)r F)Dr r r �__doc__r r� r r r# r' r* r. r2 r3 r7 r= rB rF rK rN rR rS rV rW rZ r] r_ r` rb rc re rg rj rl rr rs rx ry r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �dirr r� r� r� r� r� r� r� r� r r r r
r s�
r )ri r r@ r r� �testr Z test.supportr Ztest.libregrtest.utilsr � Exceptionr r r r r r
�<module> s