Current File : //snap/core22/current/lib/python3.10/test/libregrtest/__pycache__/save_env.cpython-310.pyc
o

E�h�2�@snddlZddlZddlZddlZddlZddlmZddlmZddl	m
Z
Gdd�de�ZGdd�d�Z
dS)	�N)�support)�	os_helper)�
print_warningc@seZdZdS)�SkipTestEnvironmentN)�__name__�
__module__�__qualname__�r	r	�0/usr/lib/python3.10/test/libregrtest/save_env.pyrsrc@seZdZdZd�dd�dd�ZdZdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zd d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�Zd4d5�Zd6d7�Zd8d9�Zd:d;�Zd<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.
    rF)�pgocCs||_||_||_||_dS�N)�testname�verbose�quietr)�selfrrrrr	r	r
�__init__%s
zsaved_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.showwarning�shutil_archive_formats�shutil_unpack_formatsz!asyncio.events._event_loop_policyzurllib.requests._url_tempfileszurllib.requests._openercCs
tj|Sr
)�sys�modules�r�namer	r	r
�
get_moduleD�
z!saved_test_environment.get_modulecCs z|�|�WStyt�wr
)r�KeyErrorrrr	r	r
�try_get_moduleHs
�z%saved_test_environment.try_get_modulecCs|�d�}t|j�S�Nzurllib.request)r �list�_url_tempfiles�r�urllib_requestr	r	r
�"get_urllib_requests__url_tempfilesO�

z9saved_test_environment.get_urllib_requests__url_tempfilescCs|D]}t�|�qdSr
)r�unlink)r�	tempfiles�filenamer	r	r
�&restore_urllib_requests__url_tempfilesRs�z=saved_test_environment.restore_urllib_requests__url_tempfilescC�|�d�}|jSr!)r �_openerr$r	r	r
�get_urllib_requests__openerV�
z2saved_test_environment.get_urllib_requests__openercC�|�d�}||_dSr!)rr-)r�openerr%r	r	r
�restore_urllib_requests__openerYr'z6saved_test_environment.restore_urllib_requests__openercCs|�d�t��S�N�asyncio)r r�maybe_get_event_loop_policy�rr	r	r
�%get_asyncio_events__event_loop_policy]s
z<saved_test_environment.get_asyncio_events__event_loop_policycCs|�d�}|�|�dSr3)r�set_event_loop_policy)r�policyr4r	r	r
�)restore_asyncio_events__event_loop_policy`s
z@saved_test_environment.restore_asyncio_events__event_loop_policycC�ttj�tjtjdd�fSr
)�idr�argvr6r	r	r
�get_sys_argvd�z#saved_test_environment.get_sys_argvcC� |dt_|dtjdd�<dS�N��)rr=)r�
saved_argvr	r	r
�restore_sys_argvf�
z'saved_test_environment.restore_sys_argvcC�t��Sr
)�os�getcwdr6r	r	r
�get_cwdj�zsaved_test_environment.get_cwdcC�t�|�dSr
)rH�chdir)r�	saved_cwdr	r	r
�restore_cwdl�z"saved_test_environment.restore_cwdcC�tjSr
�r�stdoutr6r	r	r
�get_sys_stdouto�z%saved_test_environment.get_sys_stdoutcC�
|t_dSr
rR)r�saved_stdoutr	r	r
�restore_sys_stdoutq�
z)saved_test_environment.restore_sys_stdoutcCrQr
�r�stderrr6r	r	r
�get_sys_stderrtrUz%saved_test_environment.get_sys_stderrcCrVr
rZ)r�saved_stderrr	r	r
�restore_sys_stderrvrYz)saved_test_environment.restore_sys_stderrcCrQr
�r�stdinr6r	r	r
�
get_sys_stdinyrUz$saved_test_environment.get_sys_stdincCrVr
r_)r�saved_stdinr	r	r
�restore_sys_stdin{rYz(saved_test_environment.restore_sys_stdincCsttj�tjttj�fSr
)r<rH�environ�dictr6r	r	r
�get_os_environ~sz%saved_test_environment.get_os_environcCs(|dt_tj��tj�|d�dSrA)rHrd�clear�update)r�
saved_environr	r	r
�restore_os_environ�s

z)saved_test_environment.restore_os_environcCr;r
)r<r�pathr6r	r	r
�get_sys_path�r?z#saved_test_environment.get_sys_pathcCr@rA)rrk)r�
saved_pathr	r	r
�restore_sys_path�rFz'saved_test_environment.restore_sys_pathcCr;r
)r<r�
path_hooksr6r	r	r
�get_sys_path_hooks�r?z)saved_test_environment.get_sys_path_hookscCr@rA)rro)r�saved_hooksr	r	r
�restore_sys_path_hooks�rFz-saved_test_environment.restore_sys_path_hookscCrGr
)r�gettracer6r	r	r
�get_sys_gettrace�rKz'saved_test_environment.get_sys_gettracecCrLr
)r�settrace)r�	trace_fxnr	r	r
�restore_sys_gettrace�rPz+saved_test_environment.restore_sys_gettracecCrQr
��builtinsrr6r	r	r
�get___import__�rUz%saved_test_environment.get___import__cCrVr
rx)r�import_r	r	r
�restore___import__�rYz)saved_test_environment.restore___import__cC�&|�d�}t|j�|j|jdd�fS�N�warnings)r r<�filters�rrr	r	r
�get_warnings_filters�s
z+saved_test_environment.get_warnings_filterscCs*|�d�}|d|_|d|jdd�<dS)NrrBrC)rr�)r�
saved_filtersrr	r	r
�restore_warnings_filters�s

z/saved_test_environment.restore_warnings_filterscCstj�d�}|r
|j��piS)N�asyncore)rr�get�
socket_map�copy)rr�r	r	r
�get_asyncore_socket_map�sz.saved_test_environment.get_asyncore_socket_mapcCs4tj�d�}|dur|jdd�|j�|�dSdS)Nr�T)�
ignore_all)rrr��	close_allr�rh)r�	saved_mapr�r	r	r
�restore_asyncore_socket_map�s
�z2saved_test_environment.restore_asyncore_socket_mapcC�|�d�}|j|j��fS�N�shutil)r �_ARCHIVE_FORMATSr��rr�r	r	r
�get_shutil_archive_formats�s
z1saved_test_environment.get_shutil_archive_formatscC�2|�d�}|d|_|j��|j�|d�dS�Nr�rrB)rr�rgrh�r�savedr�r	r	r
�restore_shutil_archive_formats��


z5saved_test_environment.restore_shutil_archive_formatscCr�r�)r �_UNPACK_FORMATSr�r�r	r	r
�get_shutil_unpack_formats��
z0saved_test_environment.get_shutil_unpack_formatscCr�r�)rr�rgrhr�r	r	r
�restore_shutil_unpack_formats�r�z4saved_test_environment.restore_shutil_unpack_formatscC�"|�d�}t|j�|j|j��fS�N�logging)r r<�	_handlersr��rr�r	r	r
�get_logging__handlers�s
z,saved_test_environment.get_logging__handlerscC�dSr
r	)r�saved_handlersr	r	r
�restore_logging__handlers��z0saved_test_environment.restore_logging__handlerscCr}r�)r r<�_handlerListr�r	r	r
�get_logging__handlerList�s
z/saved_test_environment.get_logging__handlerListcCr�r
r	)r�saved_handlerListr	r	r
�restore_logging__handlerList�r�z3saved_test_environment.restore_logging__handlerListcCr;r
)r<r�warnoptionsr6r	r	r
�get_sys_warnoptions�r?z*saved_test_environment.get_sys_warnoptionscCr@rA)rr�)r�
saved_optionsr	r	r
�restore_sys_warnoptions�rFz.saved_test_environment.restore_sys_warnoptionscCs
tj��Sr
)�	threading�	_danglingr�r6r	r	r
�get_threading__dangling�rz.saved_test_environment.get_threading__danglingcCstj��tj�|�dSr
)r�r�rgrh)rr�r	r	r
�restore_threading__dangling�r�z2saved_test_environment.restore_threading__danglingcCs|�d�}|��|j��S�Nzmultiprocessing.process)r �_cleanupr�r�)r�multiprocessing_processr	r	r
�%get_multiprocessing_process__dangling�s

z<saved_test_environment.get_multiprocessing_process__danglingcCs$|�d�}|j��|j�|�dSr�)rr�rgrh)rr�r�r	r	r
�)restore_multiprocessing_process__dangling�s

z@saved_test_environment.restore_multiprocessing_process__danglingcCs,|�d�}|�d�t|j�|jt|j�fS)N�	sysconfig�prefix)r �get_config_varr<�_CONFIG_VARSre�rr�r	r	r
�get_sysconfig__CONFIG_VARS�s


�z1saved_test_environment.get_sysconfig__CONFIG_VARScCr��Nr�rBrC)rr�rgrh�rr�r�r	r	r
�restore_sysconfig__CONFIG_VARS�r�z5saved_test_environment.restore_sysconfig__CONFIG_VARScCr�)Nr�)r r<�_INSTALL_SCHEMESr�r�r	r	r
�get_sysconfig__INSTALL_SCHEMES�s
�z5saved_test_environment.get_sysconfig__INSTALL_SCHEMEScCr�r�)rr�rgrhr�r	r	r
�"restore_sysconfig__INSTALL_SCHEMES�r�z9saved_test_environment.restore_sysconfig__INSTALL_SCHEMEScCstdd�t��D��S)Ncss(�|]}|tj�|�r
dndVqdS)�/�N)rHrk�isdir)�.0�fnr	r	r
�	<genexpr>s� �z3saved_test_environment.get_files.<locals>.<genexpr>)�sortedrH�listdirr6r	r	r
�	get_filess�z saved_test_environment.get_filescCsZtj}||vr'|d|vr)tj�|�rt�|�dStj�|�r+t�|�dSdSdSdS)Nr�)r�TESTFNrHrk�isfiler(r��rmtree)r�saved_valuer�r	r	r
�
restore_filess�z$saved_test_environment.restore_filescCs g|]}|�d�rtt|��qS)�LC_)�
startswith�getattrr)r��lcr	r	r
�
<listcomp>s�z!saved_test_environment.<listcomp>c
CsDg}|jD]}z
|�|t�|d�f�WqttfyYqw|Sr
)�_lc�appendr�	setlocale�	TypeError�
ValueError)r�pairingsr�r	r	r
�
get_locales
�z!saved_test_environment.get_localecCs|D]
\}}t�||�qdSr
)rr�)rr�r��settingr	r	r
�restore_locales�z%saved_test_environment.restore_localecCr,r~)r �showwarningr�r	r	r
�get_warnings_showwarningr/z/saved_test_environment.get_warnings_showwarningcCr0r~)rr�)r�fxnrr	r	r
�restore_warnings_showwarningr'z3saved_test_environment.restore_warnings_showwarningccsF�|jD]}|�dd�}d|}d|}|t||�t||�fVqdS)N�.�_�get_�restore_)�	resources�replacer�)rr�
method_suffix�get_name�restore_namer	r	r
�
resource_info#s�
�z$saved_test_environment.resource_infoc	CsPg|_|��D]\}}}z|�}Wn	tyYqw|j�||||f�q|Sr
)�saved_valuesr�rr�)rrr��restore�originalr	r	r
�	__enter__*s
�z saved_test_environment.__enter__c
Cs�|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_collect�environment_alteredrrrr�printrr[)
r�exc_type�exc_val�exc_tbr�rr�r�r��currentr	r	r
�__exit__5s��zsaved_test_environment.__exit__N)rF)Drrr�__doc__rr�rr r&r+r.r2r7r:r>rErJrOrTrXr\r^rarcrfrjrlrnrprrrtrwrzr|r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r��dirrr�r�r�r�r�r�r�r�r	r	r	r
rs�
r)ryrrHrr��testr�test.supportr�test.libregrtest.utilsr�	Exceptionrrr	r	r	r
�<module>s