File: //lib/python3.8/test/libregrtest/__pycache__/save_env.cpython-38.pyc
U
2�f$, � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl m
Z
zd dlZd dlZW n ek
r� dZY nX G dd� d�ZdS )� N)�support)�
print_warningc @ s� e Zd ZdZdZdudd�dd�ZdZdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� 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[� Z0d\d]� Z1d^d_� Z2d`da� Z3dbdc� Z4ddde� e5e6�D �Z7dfdg� Z8dhdi� Z9djdk� Z:dldm� Z;dndo� Z<dpdq� Z=drds� Z>dtS )v�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 attribute 'changed'
is initially False, but 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.
Fr )�pgoc C s || _ || _|| _|| _d S �N)�testname�verbose�quietr )�selfr r r r � r �//usr/lib/python3.8/test/libregrtest/save_env.py�__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_policyc C s t �� S r )r Zmaybe_get_event_loop_policy�r
r r r �%get_asyncio_events__event_loop_policyI s z<saved_test_environment.get_asyncio_events__event_loop_policyc C s t �|� d S r )�asyncioZset_event_loop_policy)r
Zpolicyr r r �)restore_asyncio_events__event_loop_policyK s z@saved_test_environment.restore_asyncio_events__event_loop_policyc C s t tj�tjtjd d � fS r )�id�sys�argvr r r r � get_sys_argvN s z#saved_test_environment.get_sys_argvc C s |d t _|d t jd d �< d S �N� � )r r )r
Z
saved_argvr r r �restore_sys_argvP s
z'saved_test_environment.restore_sys_argvc C s t �� S r )�os�getcwdr r r r �get_cwdT s z saved_test_environment.get_cwdc C s t �|� d S r )r �chdir)r
Z saved_cwdr r r � restore_cwdV s z"saved_test_environment.restore_cwdc C s t jS r �r �stdoutr r r r �get_sys_stdoutY s z%saved_test_environment.get_sys_stdoutc C s
|t _d S r r# )r
Z saved_stdoutr r r �restore_sys_stdout[ s z)saved_test_environment.restore_sys_stdoutc C s t jS r �r �stderrr r r r �get_sys_stderr^ s z%saved_test_environment.get_sys_stderrc C s
|t _d S r r' )r
Z saved_stderrr r r �restore_sys_stderr` s z)saved_test_environment.restore_sys_stderrc C s t jS r �r �stdinr r r r �
get_sys_stdinc s z$saved_test_environment.get_sys_stdinc C s
|t _d S r r+ )r
Z saved_stdinr r r �restore_sys_stdine s z(saved_test_environment.restore_sys_stdinc C s t tj�tjttj�fS r )r r �environ�dictr r r r �get_os_environh s z%saved_test_environment.get_os_environc C s( |d t _t j�� t j�|d � d S r )r r/ �clear�update)r
Z
saved_environr r r �restore_os_environj s
z)saved_test_environment.restore_os_environc C s t tj�tjtjd d � fS r )r r �pathr r r r � get_sys_patho s z#saved_test_environment.get_sys_pathc C s |d t _|d t jd d �< d S r )r r5 )r
Z
saved_pathr r r �restore_sys_pathq s
z'saved_test_environment.restore_sys_pathc C s t tj�tjtjd d � fS r )r r �
path_hooksr r r r �get_sys_path_hooksu s z)saved_test_environment.get_sys_path_hooksc C s |d t _|d t jd d �< d S r )r r8 )r
Z saved_hooksr r r �restore_sys_path_hooksw s
z-saved_test_environment.restore_sys_path_hooksc C s t �� S r )r �gettracer r r r �get_sys_gettrace{ s z'saved_test_environment.get_sys_gettracec C s t �|� d S r )r �settrace)r
Z trace_fxnr r r �restore_sys_gettrace} s z+saved_test_environment.restore_sys_gettracec C s t jS r ��builtinsr r r r r �get___import__� s z%saved_test_environment.get___import__c C s
|t _d S r r? )r
�import_r r r �restore___import__� s z)saved_test_environment.restore___import__c C s t tj�tjtjd d � fS r )r �warnings�filtersr r r r �get_warnings_filters� s z+saved_test_environment.get_warnings_filtersc C s |d t _|d t jd d �< d S r )rD rE )r
Z
saved_filtersr r r �restore_warnings_filters� s
z/saved_test_environment.restore_warnings_filtersc C s t j�d�}|r|j�� p i S )N�asyncore)r �modules�get�
socket_map�copy)r
rH r r r �get_asyncore_socket_map� s z.saved_test_environment.get_asyncore_socket_mapc C s0 t j�d�}|d k r,|jdd� |j�|� d S )NrH T)Z
ignore_all)r rI rJ Z close_allrK r3 )r
Z saved_maprH r r r �restore_asyncore_socket_map� s z2saved_test_environment.restore_asyncore_socket_mapc C s t jt j�� fS r )�shutil�_ARCHIVE_FORMATSrL r r r r �get_shutil_archive_formats� s z1saved_test_environment.get_shutil_archive_formatsc C s( |d t _t j�� t j�|d � d S �Nr r )rO rP r2 r3 �r
�savedr r r � restore_shutil_archive_formats� s
z5saved_test_environment.restore_shutil_archive_formatsc C s t jt j�� fS r )rO �_UNPACK_FORMATSrL r r r r �get_shutil_unpack_formats� s z0saved_test_environment.get_shutil_unpack_formatsc C s( |d t _t j�� t j�|d � d S rR )rO rV r2 r3 rS r r r � restore_shutil_unpack_formats� s
z4saved_test_environment.restore_shutil_unpack_formatsc C s t tj�tjtj�� fS r )r �loggingZ _handlersrL r r r r �get_logging__handlers� s z,saved_test_environment.get_logging__handlersc C s d S r r )r
Zsaved_handlersr r r �restore_logging__handlers� s z0saved_test_environment.restore_logging__handlersc C s t tj�tjtjd d � fS r )r rY Z _handlerListr r r r �get_logging__handlerList� s z/saved_test_environment.get_logging__handlerListc C s d S r r )r
Zsaved_handlerListr r r � restore_logging__handlerList� s z3saved_test_environment.restore_logging__handlerListc C s t tj�tjtjd d � fS r )r r � warnoptionsr r r r �get_sys_warnoptions� s z*saved_test_environment.get_sys_warnoptionsc C s |d t _|d t jd d �< d S r )r r^ )r
Z
saved_optionsr r r �restore_sys_warnoptions� s
z.saved_test_environment.restore_sys_warnoptionsc C s
t j�� S r )� threading� _danglingrL r r r r �get_threading__dangling� s z.saved_test_environment.get_threading__danglingc C s t j�� t j�|� d S r )ra rb r2 r3 rS r r r �restore_threading__dangling� s
z2saved_test_environment.restore_threading__danglingc C s t sd S t j�� t jj�� S r )�multiprocessing�processZ_cleanuprb rL r r r r �%get_multiprocessing_process__dangling� s
z<saved_test_environment.get_multiprocessing_process__danglingc C s& t sd S t jj�� t jj�|� d S r )re rf rb r2 r3 rS r r r �)restore_multiprocessing_process__dangling� s z@saved_test_environment.restore_multiprocessing_process__danglingc C s" t �d� tt j�t jtt j�fS )N�prefix)� sysconfigZget_config_varr � _CONFIG_VARSr0 r r r r �get_sysconfig__CONFIG_VARS� s
�z1saved_test_environment.get_sysconfig__CONFIG_VARSc C s( |d t _t j�� t j�|d � d S r )rj rk r2 r3 rS r r r � restore_sysconfig__CONFIG_VARS� s
z5saved_test_environment.restore_sysconfig__CONFIG_VARSc C s t tj�tjtj�� fS r )r rj �_INSTALL_SCHEMESrL r r r r � get_sysconfig__INSTALL_SCHEMES� s �z5saved_test_environment.get_sysconfig__INSTALL_SCHEMESc C s( |d t _t j�� t j�|d � d S r )rj rn r2 r3 rS r r r �"restore_sysconfig__INSTALL_SCHEMES� s
z9saved_test_environment.restore_sysconfig__INSTALL_SCHEMESc C s t dd� t�� D ��S )Nc s s& | ] }|t j�|�rd nd V qdS )�/� N)r r5 �isdir)�.0�fnr r r � <genexpr>� s �z3saved_test_environment.get_files.<locals>.<genexpr>)�sortedr �listdirr r r r � get_files� s �z saved_test_environment.get_filesc C sL t j}||krH|d |krHtj�|�r2t �|� ntj�|�rHt �|� d S )Nrq )r ZTESTFNr r5 �isfile�unlinkrs Zrmtree)r
Z saved_valueru r r r �
restore_files� s z$saved_test_environment.restore_filesc C s g | ]}|� d �rtt|��qS )ZLC_)�
startswith�getattrr )rt �lcr r r �
<listcomp>� s
�z!saved_test_environment.<listcomp>c
C sL g }| j D ]<}z|�|t�|d �f� W q
ttfk
rD Y q
Y q
X q
|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
rT r Zsettingr r r �restore_locale� s z%saved_test_environment.restore_localec C s t jS r �rD � showwarningr r r r �get_warnings_showwarning� s z/saved_test_environment.get_warnings_showwarningc C s
|t _d S r r� )r
�fxnr r r � restore_warnings_showwarning� s z3saved_test_environment.restore_warnings_showwarningc c sD | j D ]8}|�dd�}d| }d| }|t| |�t| |�fV qd S )N�.�_Zget_Zrestore_)� resources�replacer~ )r
�nameZ
method_suffixZget_nameZ restore_namer r r �
resource_info s
z$saved_test_environment.resource_infoc C s t dd� | �� D ��| _| S )Nc s s | ]\}}}||� fV qd S r r )rt r� rJ �restorer r r rv s z3saved_test_environment.__enter__.<locals>.<genexpr>)r0 r� � saved_valuesr r r r � __enter__ s �
z saved_test_environment.__enter__c
C s� | j }| ` t�� | jtjO _| �� D ]n\}}}|� }|�|�} || kr*d| _|| � | js*| js*t |� d| j
� �� t d| � d|� d�t j
dd� q*dS )NTz was modified by z
Before: z
After: � )�file�flushF)r� r Z
gc_collect�changedZenvironment_alteredr� �popr r r r �printr r( )
r
�exc_typeZexc_valZexc_tbr� r� rJ r� ZcurrentZoriginalr r r �__exit__
s"
�z saved_test_environment.__exit__N)r F)?�__name__�
__module__� __qualname__�__doc__r� r
r� r r r r r r" r% r& r) r* r- r. r1 r4 r6 r7 r9 r: r< r> rA rC rF rG rM rN rQ rU rW rX rZ r[ r\ r] r_ r` rc rd rg rh rl rm ro rp ry r| �dirr r� r� r� r� r� r� r� r� r r r r r st
r )r r@ r rY r rO r rj ra rD Ztestr Ztest.libregrtest.utilsr Z_multiprocessingZmultiprocessing.processre � ImportErrorr r r r r �<module> s