Qh d Z ddlZddlZddlZddlZddlZddlZddlmZm Z dZ
ej j Z
e
j Z G d de
Z G d d Zd ed
dfd ZdS ) z>Monkey-patching to add multiprocessing support for coverage.py N)AnyDictz_coverage$patchedc e Zd ZdZd ZdS )ProcessWithCoveragezE A?GN)__name__
__module__ __qualname____doc__r# r" r r s) FF: : : : :r) r c ^ e Zd ZdZdeddfdZdeeef fdZdeeef ddfdZdS ) StowawayzLAn object to pickle, so when it is unpickled, it can apply the monkey-patch.rcfilereturnNc || _ d S )Nr, )r r, s r" __init__zStowaway.__init__7 s
r) c d| j iS Nr, r/ )r s r" __getstate__zStowaway.__getstate__: s $+&&r) statec 0 t |d d S r2 )patch_multiprocessing)r r4 s r" __setstate__zStowaway.__setstate__= s eHo.....r) ) r$ r% r& r' strr0 r r3 r7 r( r) r" r+ r+ 5 s VV s t 'd38n ' ' ' '/$sCx. /T / / / / / /r) r+ r, r- c t t t rdS t j t
_ t j t j d< ddlm
} |j dt dt t t f f fd }||_ n# t t f$ r Y nw xY wt# t t d dS ) zMonkey-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.
NCOVERAGE_RCFILEr )spawnnamer- c B | }t |d< |S )z@Get the original preparation data, and also insert our stowaway.stowaway)r+ )r< d original_get_preparation_datar, s r" "get_preparation_data_with_stowawayzApatch_multiprocessing..get_preparation_data_with_stowaway` s) --d33A $V , ,AjMHr) T)hasattrmultiprocessingPATCHED_MARKERr r# OriginalProcessospathabspathenvironr; get_preparation_datar8 r r ImportErrorAttributeErrorsetattr)r, r; rA r@ s ` @r" r6 r6 A s // !4!?O %'GOOF$;$;BJ ! H))))))(-(B% S T#s(^ &H""
(
O^T22222s "
B B21B2)r' rC multiprocessing.processrF os.pathr r typingr r rD process BaseProcessrE r# r r r+ r8 r6 r( r) r" rS s E D
% ")5$/ : : : : :/ : : :: / / / / / / / /'3# '3$ '3 '3 '3 '3 '3 '3r)