U
e5d~6 @ 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(dAd,d-Z)d.d
Z*da+eee)e j,e j-fd/d0Z.e/e. G d1d de'Z0G d2d dej1Z2ze 3d3Z4W n e5k
r d4Z4Y nX d5d Z6d6d7 Z7d8d9 Z8d:d; Z9dd? Z;dS )B N) _args_from_interpreter_flags )process sub_debugdebuginfo sub_warning
get_logger
log_to_stderr get_temp_dirregister_after_fork
is_exitingFinalizeForkAwareThreadLockForkAwareLocalclose_all_fds_exceptSUBDEBUG
SUBWARNING
multiprocessingz+[%(levelname)s/%(processName)s] %(message)sFc G s t rt jt| f| d S N)_loggerlogr msgargs r ,/usr/lib64/python3.8/multiprocessing/util.pyr , s c G s t rt jt| f| d S r )r r DEBUGr r r r r 0 s c G s t rt jt| f| d S r )r r INFOr r r r r 4 s c G s t rt jt| f| d S r )r r r r r r r r 8 s c C s| dd l } | z\tsj| tadt_ttdrFt t
t t
n$tj
t
di f tj t
di f W 5 | X tS )Nr
unregisterr )loggingZ _acquireLockZ _releaseLockr Z getLogger LOGGER_NAMEZ propagatehasattratexitr# _exit_functionregisterZ
_exithandlersremoveappend)r$ r r r r < s
c C sJ dd l }t }|t}| }|| || | rB|| dat S )Nr T)
r$ r Z FormatterDEFAULT_LOGGING_FORMATZ
StreamHandlerZ setFormatterZ
addHandlerZsetLevel_log_to_stderrr )levelr$ ZloggerZ formatterZhandlerr r r r
W s
c C s t jdkrdS tt dr dS dS )NZlinuxTZgetandroidapilevelF)sysplatformr& r r r r #_platform_supports_abstract_socketsl s
r1 c C s@ | sdS t | tr | d dkS t | tr4| d dkS tdd S )NFr z(address type of {address!r} unrecognized)
isinstancebytesstr TypeError)Zaddressr r r is_abstract_socket_namespacet s
r7 c C s& | | t }|d k r"d |jd< d S )Ntempdir)r current_process_config)rmtreer8 r9 r r r _remove_temp_dir s r< c C sf t jd} | d krbdd l}dd l}|jdd} td| td t |j
| fdd | t jd< | S )Nr8 r zpymp-)prefixzcreated temp directory %si)r exitpriority) r r9 r: getshutiltempfileZmkdtempr r r< r; )r8 r@ rA r r r r s
c C sf t t } | | D ]H\\}}}}z || W q tk
r^ } ztd| W 5 d }~X Y qX qd S )Nz after forker raised exception %s)list_afterfork_registryitemssort Exceptionr )rD indexZidentfuncobjer r r _run_after_forkers s rK c C s | t ttt| |f< d S r )rC next_afterfork_counterid)rI rH r r r r s c @ sB e Zd Zd
ddZdeeejfddZdd Z d d
Z
d d Z dS )r 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)r3 intr6 formattypeweakrefref_weakref
ValueError _callback_args_kwargsrL _finalizer_counter_keyosgetpid_pid_finalizer_registry)selfrI callbackr kwargsr> r r r __init__ s"
zFinalize.__init__c C s z || j = W n tk
r( |d Y nbX | j| krD|d d }n$|d| j| j| j | j| j| j}d | _ | _ | _ | _| _ |S d S )Nz finalizer no longer registeredz+finalizer ignored because different processz/finalizer calling %s with args %s and kwargs %s)rZ KeyErrorr] rV rW rX rT )r_ Zwrr^ r r\ resr r r __call__ s$ zFinalize.__call__c C sD z t | j= W n tk
r Y n X d | _ | _ | _ | _| _d S r )r^ rZ rc rT rV rW rX r_ r r r cancel s zFinalize.cancelc C s
| j tkS r )rZ r^ rf r r r still_active s zFinalize.still_activec C s z | }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, exitpriority=>) rT AttributeErrorr6 __class__ri getattrrV rW r5 rX rZ )r_ rI xr r r __repr__ s"
zFinalize.__repr__)r NN) ri
__module__ __qualname__rb r^ r r[ r\ re rg rh ro r r r r r s
c s t d kr d S d kr dd n fdd fddtt D }|jdd |D ]P}t |}|d k rPtd| z
| W qP tk
r d d l}| Y qPX qPd krt d S )
Nc S s | d d k S Nr r pr r r z!_run_finalizers..c s | d d k o| d kS rr r rs ) minpriorityr r ru rv c s g | ]} |r|qS r r ).0key)fr r
# s z#_run_finalizers..T)reversez
calling %sr ) r^ rB rE r? r rF traceback print_excclear)rw keysry finalizerr} r )rz rw r _run_finalizers s$
r c C s t p
t d kS r )_exitingr r r r r
8 s c C s t sda | d |d |d | d k rr| D ] }|jr0| d|j |j q0| D ]}| d|j | qX|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_childrenr9 rt r r r r( @ s
r( c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
r c C s | t| tj d S r )_resetr r rf r r r rb q s z ForkAwareThreadLock.__init__c C s" t | _| jj| _| jj| _d S r ) threadingZLock_lockacquirereleaserf r r r r u s
zForkAwareThreadLock._resetc C s
| j S r )r __enter__rf r r r r z s z ForkAwareThreadLock.__enter__c G s | j j| S r )r __exit__)r_ r r r r r } s z ForkAwareThreadLock.__exit__N)ri rp rq rb 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 r )__dict__r )rI r r r ru rv z)ForkAwareLocal.__init__..)r rf r r r rb s zForkAwareLocal.__init__c C s t | dfS )Nr )rQ rf r r r
__reduce__ s zForkAwareLocal.__reduce__N)ri rp rq rb r r r r r r s SC_OPEN_MAX c C sN t | dtg } | tt| d D ] }t| | d | |d q(d S )Nr )rB MAXFDrE rangelenr[
closerange)fdsir r r r s c C s t jd krd S zt j W n ttfk
r4 Y nX z@ttjtj} zt| ddt _W n t| Y nX W n ttfk
r Y nX d S )NF)closefd) r/ stdincloseOSErrorrU r[ opendevnullO_RDONLY)fdr r r _close_stdin s
r c C sT zt j W n ttfk
r& Y nX zt j W n ttfk
rN Y nX d S r )r/ stdoutflushrk rU stderrr r r r _flush_std_streams s r c C sx dd l }tttt|}t \}}z6||t | gd|d d dddddd||ddd W S t| t| X d S )Nr Tr F)
_posixsubprocesstuplesortedmaprO r[ piper Z fork_execfsencode)pathr Zpassfdsr Z errpipe_readZ
errpipe_writer r r spawnv_passfds s2
r c G s | D ]}t | qd S r )r[ r )r r r r r close_fds s r c C sZ ddl m} t ddlm} |j ddlm} |j t
| | d S )Nr )support)
forkserver)resource_tracker)
Ztestr r Z_cleanupr r Z _forkserverZ_stopr Z_resource_trackerr Z
gc_collectZ
reap_children)r r r r r r _cleanup_tests s
r )N)N)
s
V
,
*