File: //proc/276314/root/usr/lib/python3.9/test/libregrtest/__pycache__/save_env.cpython-39.pyc
a
kx�g(. � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl Z d dl m
Z
d dlmZ zd dlZd dlZW n ey� dZY n0 G dd� d�ZdS )� N)�support)�
print_warningc @ s e Zd ZdZdZd}dd�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� Z5dfdg� Z6dhdi� Z7djdk� Z8dldm� e9e:�D �Z;dndo� Z<dpdq� Z=drds� Z>dtdu� Z?dvdw� Z@dxdy� ZAdzd{� ZBd|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 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.9/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_policyz urllib.requests._url_tempfileszurllib.requests._openerc C s t tjj�S r )�list�urllib�requestZ_url_tempfiles�r
r r r �"get_urllib_requests__url_tempfilesK s 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_tempfilesM s z=saved_test_environment.restore_urllib_requests__url_tempfilesc C s t jjS r �r r Z_openerr r r r �get_urllib_requests__openerQ s z2saved_test_environment.get_urllib_requests__openerc C s |t j_d S r r )r
Zopenerr r r � restore_urllib_requests__openerS s z6saved_test_environment.restore_urllib_requests__openerc C s t �� S r )r Zmaybe_get_event_loop_policyr r r r �%get_asyncio_events__event_loop_policyV 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_policyX 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_argv[ 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_argv] s
z'saved_test_environment.restore_sys_argvc C s t �� S r )�os�getcwdr r r r �get_cwda s z saved_test_environment.get_cwdc C s t �|� d S r )r( �chdir)r
Z saved_cwdr r r � restore_cwdc s z"saved_test_environment.restore_cwdc C s t jS r �r! �stdoutr r r r �get_sys_stdoutf s z%saved_test_environment.get_sys_stdoutc C s
|t _d S r r- )r
Z saved_stdoutr r r �restore_sys_stdouth s z)saved_test_environment.restore_sys_stdoutc C s t jS r �r! �stderrr r r r �get_sys_stderrk s z%saved_test_environment.get_sys_stderrc C s
|t _d S r r1 )r
Z saved_stderrr r r �restore_sys_stderrm s z)saved_test_environment.restore_sys_stderrc C s t jS r �r! �stdinr r r r �
get_sys_stdinp s z$saved_test_environment.get_sys_stdinc C s
|t _d S r r5 )r
Z saved_stdinr r r �restore_sys_stdinr s z(saved_test_environment.restore_sys_stdinc C s t tj�tjttj�fS r )r r( �environ�dictr r r r �get_os_environu s z%saved_test_environment.get_os_environc C s( |d t _t j�� t j�|d � d S r$ )r( r9 �clear�update)r
Z
saved_environr r r �restore_os_environw s
z)saved_test_environment.restore_os_environc C s t tj�tjtjd d � fS r )r r! �pathr r r r � get_sys_path| s z#saved_test_environment.get_sys_pathc C s |d t _|d t jd d �< d S r$ )r! r? )r
Z
saved_pathr r r �restore_sys_path~ 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_hooks� s z)saved_test_environment.get_sys_path_hooksc C s |d t _|d t jd d �< d S r$ )r! rB )r
Z saved_hooksr r r �restore_sys_path_hooks� 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 rI )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$ )rN rO )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
rR r r r �get_asyncore_socket_map� s z.saved_test_environment.get_asyncore_socket_mapc C s0 t j�d�}|d ur,|jdd� |j�|� d S )NrR T)Z
ignore_all)r! rS rT Z close_allrU r= )r
Z saved_maprR r r r �restore_asyncore_socket_map� s z2saved_test_environment.restore_asyncore_socket_mapc C s t jt j�� fS r )�shutil�_ARCHIVE_FORMATSrV 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% )rY rZ r< r= �r
�savedr r r � restore_shutil_archive_formats� s
z5saved_test_environment.restore_shutil_archive_formatsc C s t jt j�� fS r )rY �_UNPACK_FORMATSrV 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 r\ )rY r` r< r= r] r r r � restore_shutil_unpack_formats� s
z4saved_test_environment.restore_shutil_unpack_formatsc C s t tj�tjtj�� fS r )r �loggingZ _handlersrV 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 rc 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! rh )r
Z
saved_optionsr r r �restore_sys_warnoptions� s
z.saved_test_environment.restore_sys_warnoptionsc C s
t j�� S r )� threading� _danglingrV r r r r �get_threading__dangling� s z.saved_test_environment.get_threading__danglingc C s t j�� t j�|� d S r )rk rl r< r= r] 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_cleanuprl rV 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 )ro rp rl r< r= r] 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_VARSr: 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$ )rt ru r< r= r] r r r � restore_sysconfig__CONFIG_VARS� s
z5saved_test_environment.restore_sysconfig__CONFIG_VARSc C s t tj�tjtj�� fS r )r rt �_INSTALL_SCHEMESrV 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$ )rt rx r< r= r] 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( r? �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}||vrH|d |vrHtj�|�r2t �|� ntj�|�rHt �|� d S )Nr{ )r ZTESTFNr( r? �isfiler r} Zrmtree)r
Z saved_valuer r r r �
restore_files� s z$saved_test_environment.restore_filesc C s g | ]}|� d �rtt|��qS )ZLC_)�
startswith�getattrr )r~ �lcr r r �
<listcomp>� s
�z!saved_test_environment.<listcomp>c
C sJ g }| j D ]:}z|�|t�|d �f� W q
ttfyB Y q
Y q
0 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
r^ r� Zsettingr r r �restore_locale s z%saved_test_environment.restore_localec C s t jS r �rN � 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 )r~ r� rT �restorer r r r� � z3saved_test_environment.__enter__.<locals>.<genexpr>)r: 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! r2 )
r
�exc_typeZexc_valZexc_tbr� r� rT r� ZcurrentZoriginalr r r �__exit__ s
�z saved_test_environment.__exit__N)r F)C�__name__�
__module__� __qualname__�__doc__r� r
r� r r r r r r r# r' r* r, r/ r0 r3 r4 r7 r8 r; r> r@ rA rC rD rF rH rK rM rP rQ rW rX r[ r_ ra rb rd re rf rg ri rj rm rn rq rr rv rw ry rz r� r� �dirr r� r� r� r� r� r� r� r� r r r r r s|
r )r rJ r rc r( rY r! rt rk Zurllib.requestr rN �testr Ztest.libregrtest.utilsr Z_multiprocessingZmultiprocessing.processro � ImportErrorr r r r r �<module> s"