B
HxLK @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ ddlm Z ddddd d
d d d
ddddddgZ
d Z dZ dZ
dZdZdZdZdadadd Zd d Zd d Zd d Zd d Zd>d d
Zd!d" Zd#d$ Ze Z d%d& Z d'd Z e Z e! Z"d(d) Z#d*d Z$i Z%e! Z&G d+d de'Z(d?d,d-Z)d.d
Z*da+eee)e j,e j-fd/d0Z.e/e. G d1d de'Z0G d2d dej1Z2ye 3d3Z4W n e5k
r d4Z4Y nX d5d Z6d6d7 Z7d8d9 Z8d:d; Z9d r? r r r r s
c C sj t t } | xP| D ]H\\}}}}y || W q tk
r` } ztd| W d d }~X Y qX qW d S )Nz after forker raised exception %s)list_afterfork_registryitemssort Exceptionr )rB indexZidentfuncobjer r r _run_after_forkers s rI c C s | t ttt| |f< d S )N)rA next_afterfork_counterid)rG rF r r r r s c @ sF e Zd ZdZdddZdeeejfddZ dd Z
d
d Z d d
Z dS )r zA
Class which supports object finalization using weakrefs
r Nc C s |d k r&t |ts&td|t||d k r>t|| | _n|d krNtd|| _ || _
|p`i | _ |t t
f| _t | _| t| j< d S )Nz3Exitpriority ({0!r}) must be None or int, not {1!s}z+Without object, exitpriority cannot be None)r1 intr4 formattypeweakrefref_weakref
ValueError _callback_args_kwargsrJ _finalizer_counter_keyosgetpid_pid_finalizer_registry)selfrG callbackr kwargsr< r r r __init__ s
zFinalize.__init__c C s y || j = W n tk
r( |d Y nbX | j| krD|d d}n$|d| j| j| j | j| j| j}d | _ | _ | _ | _| _ |S dS )zQ
Run the callback unless it has already been called or cancelled
z finalizer no longer registeredz+finalizer ignored because different processNz/finalizer calling %s with args %s and kwargs %s)rX KeyErrorr[ rT rU rV rR )r] Zwrr\ r rZ Zresr r r __call__ s zFinalize.__call__c C sD y t | j= W n tk
r Y n X d | _ | _ | _ | _| _dS )z3
Cancel finalization of the object
N)r\ rX ra rR rT rU rV )r] r r r cancel s
zFinalize.cancelc C s
| j tkS )zS
Return whether this finalizer is still waiting to invoke callback
)rX r\ )r] r r r still_active s zFinalize.still_activec C s y | }W n ttfk
r( d }Y nX |d kr>d| jj S d| jjt| jd| jf }| jrr|dt| j 7 }| j r|dt| j 7 }| j
d d k r|dt| j
d 7 }|d S ) Nz<%s object, dead>z<%s object, callback=%s__name__z, args=z , kwargs=r z, exitprority=>) rR AttributeErrorr4 __class__re getattrrT rU r3 rV rX )r] rG xr r r __repr__ s
zFinalize.__repr__)r NN)
re
__module__ __qualname____doc__r` r\ r rY rZ rb rc rd rk r r r r r s
c s t dkr dS dkr dd n fdd fddtt D }|jdd xX|D ]P}t |}|dk rRtd | y
| W qR tk
r d
dl}| Y qRX qRW dkrt dS ) z
Run all finalizers whose exit priority is not None and at least minpriority
Finalizers with highest priority are called first; finalizers with
the same priority will be called in reverse order of creation.
Nc S s | d d k S )Nr r )pr r r z!_run_finalizers..c s | d d k o| d kS )Nr r )ro ) minpriorityr r rp rq c s g | ]} |r|qS r r ).0key)fr r
# s z#_run_finalizers..T)reversez
calling %sr ) r\ r@ rC r= r rD traceback print_excclear)rr keysrt finalizerrx r )ru rr r _run_finalizers s$
r} c C s t p
t dkS )z6
Returns true if the process is shutting down
N)_exitingr r r r r
8 s c C s t sda | d |d |d | d k rzx*| D ] }|jr2| d|j |j q2W x"| D ]}| d|j | q^W |d | d S )NTzprocess shutting downz2running all "atexit" finalizers with priority >= 0r z!calling terminate() for daemon %sz calling join() for process %sz)running the remaining "atexit" finalizers)r~ ZdaemonnameZ_popenZ terminatejoin)r r r} active_childrenr7 ro r r r r& @ s
r& c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
r c C s | t| tj d S )N)_resetr r )r] r r r r` q s z ForkAwareThreadLock.__init__c C s" t | _| jj| _| jj| _d S )N) threadingZLock_lockacquirerelease)r] r r r r u s
zForkAwareThreadLock._resetc C s
| j S )N)r __enter__)r] r r r r z s z ForkAwareThreadLock.__enter__c G s | j j| S )N)r __exit__)r] r r r r r } s z ForkAwareThreadLock.__exit__N)re rl rm r` r r r r r r r r p s c @ s e Zd Zdd Zdd ZdS )r c C s t | dd d S )Nc S s
| j S )N)__dict__rz )rG r r r rp rq z)ForkAwareLocal.__init__..)r )r] r r r r` s zForkAwareLocal.__init__c C s t | dfS )Nr )rO )r] r r r
__reduce__ s zForkAwareLocal.__reduce__N)re rl rm r` r r r r r r s SC_OPEN_MAX c C sR t | dtg } | x4tt| d D ] }t| | d | |d q*W d S )Nr )r@ MAXFDrC rangelenrY
closerange)Zfdsir r r r s c C s t jd krd S yt j W n ttfk
r4 Y nX y@ttjtj} yt| ddt _W n t| Y nX W n ttfk
r Y nX d S )NF)closefd) r- stdincloseOSErrorrS rY opendevnullO_RDONLY)fdr r r _close_stdin s
r c C sT yt j W n ttfk
r& Y nX yt j W n ttfk
rN Y nX d S )N)r- stdoutflushrg rS stderrr r r r _flush_std_streams s r c C st dd l }tttt|}t \}}z2||t| gd|d d dddddd||ddd S t | t | X d S )Nr Tr F)
_posixsubprocesstuplesortedmaprM rY pipeZ fork_execfsencoder )pathr Zpassfdsr Z errpipe_readZ
errpipe_writer r r spawnv_passfds s
r c C sD ddl m} t ddlm} |j t | |
dS )zKCleanup multiprocessing resources when multiprocessing tests
completed.r )support)
forkserverN) Ztestr r Z_cleanupr r Z _forkserverZ_stopr} Z
gc_collectZ
reap_children)r r r r r _cleanup_tests s
r )N)N);rY itertoolsr- rP r% r
subprocessr r __all__ZNOTSETr r r r r# r* r r+ r r r r r r
r/ r5 Zabstract_sockets_supportedr: r ZWeakValueDictionaryrA countrK rI r r\ rW objectr r} r
r~ r r7 r& r' r Zlocalr sysconfr rD r r r r r r r r r
sp
V
,(