Your IP : 216.73.216.213
B
�ÔW� � @ s� d Z ddlZddlZddlZddlZddlmZ dZdZej dkrLej
j Z nej
Z e jZG dd� de �ZG d d
� d
e�Zeed �d d
� �ZdS )z>Monkey-patching to add multiprocessing support for coverage.py� N)�contractz_coverage$patchedZ_COVERAGE_RCFILE)� � c @ s e Zd ZdZdd� ZdS )�ProcessWithCoveragez<A replacement for multiprocess.Process that starts coverage.c C sJ ddl m} tjt }|d|d�}|�� zt| �S |�� |�� X dS )z,Wrapper around _bootstrap to start coverage.r )�CoverageT)Z data_suffix� config_fileN) �coverager �os�environ�COVERAGE_RCFILE_ENV�start�original_bootstrap�stopZsave)�selfr �rcfileZcov� r �E/opt/alt/python37/lib64/python3.7/site-packages/coverage/multiproc.py�
_bootstrap s
z ProcessWithCoverage._bootstrapN)�__name__�
__module__� __qualname__�__doc__r r r r r r s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �StowawayzLAn object to pickle, so when it is unpickled, it can apply the monkey-patch.c C s
|| _ d S )N)r )r r r r r �__init__. s zStowaway.__init__c C s
d| j iS )Nr )r )r r r r � __getstate__1 s zStowaway.__getstate__c C s t |d � d S )Nr )�patch_multiprocessing)r �stater r r � __setstate__4 s zStowaway.__setstate__N)r r r r r r r r r r r r , s r )r c s� t tt�rdS tjdkr"tjt_ntt_�t j
t < yddlm } |j
� W n ttfk
r` Y nX � �fdd�}||_
tttd� dS )z�Monkey-patch the multiprocessing module.
This enables coverage measurement of processes started by multiprocessing.
This involves aggressive monkey-patching.
`rcfile` is the path to the rcfile being used.
N)r r r )�spawnc s � | �}t ��|d< |S )z@Get the original preparation data, and also insert our stowaway.Zstowaway)r )�name�d)� original_get_preparation_datar r r �"get_preparation_data_with_stowaway[ s zApatch_multiprocessing.<locals>.get_preparation_data_with_stowawayT)�hasattr�multiprocessing�PATCHED_MARKER�sys� version_infor r �OriginalProcess�Processr r
r r Zget_preparation_data� ImportError�AttributeError�setattr)r r r" r )r! r r r 8 s
r )r r$ Zmultiprocessing.processr r& Z
coverage.miscr r% r r' �processZ BaseProcessr( r) r r
r �objectr �strr r r r r �<module> s