File: //proc/self/root/usr/local/lib/python3.7/test/libregrtest/__pycache__/save_env.cpython-37.opt-2.pyc
B
��g+ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z yd dl
Z
d dlZW n ek
r� dZY nX G dd� d�ZdS )� N)�support)�
print_warningc @ s� e Zd ZdZdpdd�dd�ZdZdd� Zd d
� 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/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUdV� Z-dWdX� Z.dYdZ� Z/d[d\� Z0d]d^� Z1d_d`� e2e3�D �Z4dadb� Z5dcdd� Z6dedf� Z7dgdh� Z8didj� Z9dkdl� Z:dmdn� Z;doS )q�saved_test_environmentFr )�pgoc C s || _ || _|| _|| _d S )N)�testname�verbose�quietr )�selfr r r r � r
�5/usr/local/lib/python3.7/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_formatsc C s t tj�tjtjd d � fS )N)�id�sys�argv)r r
r
r � get_sys_argvG 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_argvI s
z'saved_test_environment.restore_sys_argvc C s t �� S )N)�os�getcwd)r r
r
r �get_cwdM s z saved_test_environment.get_cwdc C s t �|� d S )N)r �chdir)r Z saved_cwdr
r
r � restore_cwdO s z"saved_test_environment.restore_cwdc C s t jS )N)r �stdout)r r
r
r �get_sys_stdoutR s z%saved_test_environment.get_sys_stdoutc C s
|t _d S )N)r r )r Z saved_stdoutr
r
r �restore_sys_stdoutT s z)saved_test_environment.restore_sys_stdoutc C s t jS )N)r �stderr)r r
r
r �get_sys_stderrW s z%saved_test_environment.get_sys_stderrc C s
|t _d S )N)r r )r Z saved_stderrr
r
r �restore_sys_stderrY s z)saved_test_environment.restore_sys_stderrc C s t jS )N)r �stdin)r r
r
r �
get_sys_stdin\ s z$saved_test_environment.get_sys_stdinc C s
|t _d S )N)r r# )r Z saved_stdinr
r
r �restore_sys_stdin^ s z(saved_test_environment.restore_sys_stdinc C s t tj�tjttj�fS )N)r r �environ�dict)r r
r
r �get_os_environa s z%saved_test_environment.get_os_environc C s( |d t _t j�� t j�|d � d S )Nr r )r r& �clear�update)r Z
saved_environr
r
r �restore_os_environc s
z)saved_test_environment.restore_os_environc C s t tj�tjtjd d � fS )N)r r �path)r r
r
r � get_sys_pathh s z#saved_test_environment.get_sys_pathc C s |d t _|d t jd d �< d S )Nr r )r r, )r Z
saved_pathr
r
r �restore_sys_pathj s
z'saved_test_environment.restore_sys_pathc C s t tj�tjtjd d � fS )N)r r �
path_hooks)r r
r
r �get_sys_path_hooksn s z)saved_test_environment.get_sys_path_hooksc C s |d t _|d t jd d �< d S )Nr r )r r/ )r Z saved_hooksr
r
r �restore_sys_path_hooksp s
z-saved_test_environment.restore_sys_path_hooksc C s t �� S )N)r �gettrace)r r
r
r �get_sys_gettracet s z'saved_test_environment.get_sys_gettracec C s t �|� d S )N)r �settrace)r Z trace_fxnr
r
r �restore_sys_gettracev s z+saved_test_environment.restore_sys_gettracec C s t jS )N)�builtinsr )r r
r
r �get___import__y s z%saved_test_environment.get___import__c C s
|t _d S )N)r6 r )r �import_r
r
r �restore___import__{ s z)saved_test_environment.restore___import__c C s t tj�tjtjd d � fS )N)r �warnings�filters)r r
r
r �get_warnings_filters~ s z+saved_test_environment.get_warnings_filtersc C s |d t _|d t jd d �< d S )Nr r )r: r; )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 r> 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 )Nr> T)Z
ignore_all)r r? r@ Z close_allrA r* )r Z saved_mapr> r
r
r �restore_asyncore_socket_map� s z2saved_test_environment.restore_asyncore_socket_mapc C s t jt j�� fS )N)�shutil�_ARCHIVE_FORMATSrB )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 )rE rF r) r* )r �savedr
r
r � restore_shutil_archive_formats� s
z5saved_test_environment.restore_shutil_archive_formatsc C s t jt j�� fS )N)rE �_UNPACK_FORMATSrB )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 )Nr r )rE rJ r) r* )r rH r
r
r � restore_shutil_unpack_formats� s
z4saved_test_environment.restore_shutil_unpack_formatsc C s t tj�tjtj�� fS )N)r �loggingZ _handlersrB )r r
r
r �get_logging__handlers� s z,saved_test_environment.get_logging__handlersc C s d S )Nr
)r Zsaved_handlersr
r
r �restore_logging__handlers� s z0saved_test_environment.restore_logging__handlersc C s t tj�tjtjd d � fS )N)r rM Z _handlerList)r r
r
r �get_logging__handlerList� s z/saved_test_environment.get_logging__handlerListc C s d S )Nr
)r Zsaved_handlerListr
r
r � restore_logging__handlerList� s z3saved_test_environment.restore_logging__handlerListc C s t tj�tjtjd d � fS )N)r r � warnoptions)r r
r
r �get_sys_warnoptions� s z*saved_test_environment.get_sys_warnoptionsc C s |d t _|d t jd d �< d S )Nr r )r rR )r Z
saved_optionsr
r
r �restore_sys_warnoptions� s
z.saved_test_environment.restore_sys_warnoptionsc C s
t j�� S )N)� threading� _danglingrB )r r
r
r �get_threading__dangling� s z.saved_test_environment.get_threading__danglingc C s t j�� t j�|� d S )N)rU rV r) r* )r rH r
r
r �restore_threading__dangling� s
z2saved_test_environment.restore_threading__danglingc C s t sd S t j�� t jj�� S )N)�multiprocessing�processZ_cleanuprV rB )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 )N)rY rZ rV r) r* )r rH 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 )Nr r )r^ r_ r) r* )r rH r
r
r � restore_sysconfig__CONFIG_VARS� s
z5saved_test_environment.restore_sysconfig__CONFIG_VARSc C s t tj�tjtj�� fS )N)r r^ �_INSTALL_SCHEMESrB )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 )Nr r )r^ rb r) r* )r rH 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 �listdir)r r
r
r � get_files� s z saved_test_environment.get_filesc C sL t j}||krH|d |krHtj�|�r2t �|� ntj�|�rHt �|� d S )Nre )r ZTESTFNr r, �isfile�unlinkrg Zrmtree)r Z saved_valueri r
r
r �
restore_files� s z$saved_test_environment.restore_filesc C s g | ]}|� d �rtt|��qS )�LC_)�
startswith�getattrr )rh �lcr
r
r �
<listcomp>� s z!saved_test_environment.<listcomp>c
C sN g }xD| j D ]:}y|�|t�|d �f� W q ttfk
rD w Y q X q W |S )N)�_lc�appendr � setlocale� TypeError�
ValueError)r Zpairingsrt r
r
r �
get_locale� s z!saved_test_environment.get_localec C s" x |D ]\}}t �||� qW d S )N)r rx )r rH rt Zsettingr
r
r �restore_locale� s z%saved_test_environment.restore_localec C s t jS )N)r: � showwarning)r r
r
r �get_warnings_showwarning� s z/saved_test_environment.get_warnings_showwarningc C s
|t _d S )N)r: r} )r �fxnr
r
r � restore_warnings_showwarning� s z3saved_test_environment.restore_warnings_showwarningc c sH xB| j D ]8}|�dd�}d| }d| }|t| |�t| |�fV qW d S )N�.�_Zget_Zrestore_)� resources�replacers )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 )Nr
)rh r� r@ �restorer
r
r rj s z3saved_test_environment.__enter__.<locals>.<genexpr>)r' r� � saved_values)r r
r
r � __enter__ s z saved_test_environment.__enter__c
C s� | j }| ` t�� | jtjO _xz| �� D ]n\}}}|� }|�|�} || kr,d| _|| � | js,| js,t |� d| j
� �� t d| � d|� d�t j
dd� q,W 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� r@ r� ZcurrentZoriginalr
r
r �__exit__ s
z saved_test_environment.__exit__N)r F)<�__name__�
__module__� __qualname__r� r r� r r r r r r r! r" r$ r% r( r+ r- r. r0 r1 r3 r5 r7 r9 r< r= rC rD rG rI rK rL rN rO rP rQ rS rT rW rX r[ r\ r` ra rc rd rm rp �dirr rv r{ r| r~ r� r� r� r� r
r
r
r r sn
r )r6 r rM r rE r r^ rU r: Ztestr Ztest.libregrtest.utilsr Z_multiprocessingZmultiprocessing.processrY � ImportErrorr r
r
r
r �<module> s