U
    pd~6                     @   s  d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ ddlm	Z	 ddddd	d
dddddddddgZ
d ZdZdZdZdZdZdZdadadd Zdd Zdd Zdd Zdd	 Zd@d d
Zd!d" Zd#d$ Ze Zd%d& Zd'd Ze Z e! Z"d(d) Z#d*d Z$i Z%e! Z&G d+d de'Z(dAd,d-Z)d.d Z*da+eee)e	j,e	j-fd/d0Z.e/e. G d1d de'Z0G d2d dej1Z2ze 3d3Z4W n e5k
r   d4Z4Y nX d5d Z6d6d7 Z7d8d9 Z8d:d; Z9d<d= Z:d>d? Z;dS )B    N)_args_from_interpreter_flags   )process	sub_debugdebuginfosub_warning
get_loggerlog_to_stderrget_temp_dirregister_after_fork
is_exitingFinalizeForkAwareThreadLockForkAwareLocalclose_all_fds_exceptSUBDEBUG
SUBWARNING   
         multiprocessingz+[%(levelname)s/%(processName)s] %(message)sFc                 G   s   t rt jt| f|  d S N)_loggerlogr   msgargs r   */usr/lib/python3.8/multiprocessing/util.pyr   ,   s    c                 G   s   t rt jt| f|  d S r   )r   r   DEBUGr   r   r   r    r   0   s    c                 G   s   t rt jt| f|  d S r   )r   r   INFOr   r   r   r    r   4   s    c                 G   s   t rt jt| f|  d S r   )r   r   r   r   r   r   r    r   8   s    c                  C   s|   ddl } |   z\tsj| tadt_ttdrFt	t
 tt
 n$tjt
di f tjt
di f W 5 |   X tS )z0
    Returns logger used by multiprocessing
    r   N
unregisterr   )loggingZ_acquireLockZ_releaseLockr   Z	getLoggerLOGGER_NAMEZ	propagatehasattratexitr#   _exit_functionregisterZ_exithandlersremoveappend)r$   r   r   r    r	   <   s    



c                 C   sJ   ddl }t }|t}| }|| || | rB||  dat	S )zB
    Turn on logging and add a handler which prints to stderr
    r   NT)
r$   r	   Z	FormatterDEFAULT_LOGGING_FORMATZStreamHandlerZsetFormatterZ
addHandlerZsetLevel_log_to_stderrr   )levelr$   ZloggerZ	formatterZhandlerr   r   r    r
   W   s    



c                   C   s    t jdkrdS tt drdS dS )NZlinuxTZgetandroidapilevelF)sysplatformr&   r   r   r   r    #_platform_supports_abstract_socketsl   s
    

r1   c                 C   s@   | sdS t | tr| d dkS t | tr4| d dkS tdd S )NFr    z(address type of {address!r} unrecognized)
isinstancebytesstr	TypeError)Zaddressr   r   r    is_abstract_socket_namespacet   s    

r7   c                 C   s&   | | t  }|d k	r"d |jd< d S )Ntempdir)r   current_process_config)rmtreer8   r9   r   r   r    _remove_temp_dir   s    r<   c                  C   sf   t  jd} | d krbdd l}dd l}|jdd} td|  td t	|j
| fdd | t  jd< | S )Nr8   r   zpymp-)prefixzcreated temp directory %si)r   exitpriority)r   r9   r:   getshutiltempfileZmkdtempr   r   r<   r;   )r8   r@   rA   r   r   r    r      s    
c                  C   sf   t t } |   | D ]H\\}}}}z|| W q tk
r^ } ztd| W 5 d }~X Y qX qd S )Nz after forker raised exception %s)list_afterfork_registryitemssort	Exceptionr   )rD   indexZidentfuncobjer   r   r    _run_after_forkers   s    rK   c                 C   s   | t ttt| |f< d S r   )rC   next_afterfork_counterid)rI   rH   r   r   r    r      s    c                   @   sF   e Zd ZdZdddZdeeejfddZ	dd	 Z
d
d Zdd ZdS )r   zA
    Class which supports object finalization using weakrefs
    r   Nc                 C   s   |d k	r&t |ts&td|t||d k	r>t|| | _n|d krNtd|| _	|| _
|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   formattypeweakrefref_weakref
ValueError	_callback_args_kwargsrL   _finalizer_counter_keyosgetpid_pid_finalizer_registry)selfrI   callbackr   kwargsr>   r   r   r    __init__   s"     

zFinalize.__init__c                 C   s   z|| j = W n tk
r(   |d Y nbX | j| krD|d d}n$|d| j| j| j | j| j| j}d | _ | _ | _ | _| _ |S dS )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)rZ   KeyErrorr]   rV   rW   rX   rT   )r_   Zwrr^   r   r\   resr   r   r    __call__   s$       zFinalize.__call__c                 C   sD   zt | j= W n tk
r    Y n X d | _ | _ | _ | _| _dS )z3
        Cancel finalization of the object
        N)r^   rZ   rc   rT   rV   rW   rX   r_   r   r   r    cancel   s     zFinalize.cancelc                 C   s
   | j tkS )zS
        Return whether this finalizer is still waiting to invoke callback
        )rZ   r^   rf   r   r   r    still_active   s    zFinalize.still_activec              	   C   s   z|   }W n ttfk
r(   d }Y nX |d kr>d| jj S d| jjt| jd| jf }| jrr|dt| j 7 }| j	r|dt| j	 7 }| j
d d k	r|dt| 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"    
zFinalize.__repr__)r   NN)ri   
__module____qualname____doc__rb   r^   r   r[   r\   re   rg   rh   ro   r   r   r   r    r      s   
 
c              	      s   t dkrdS dkrdd  nfdd  fddtt D }|jdd |D ]P}t |}|dk	rPtd	| z
|  W qP tk
r   d
dl}|  Y qPX qPdkrt   dS )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    <lambda>      z!_run_finalizers.<locals>.<lambda>c                    s   | d d k	o| d  kS rs   r   rt   )minpriorityr   r    rv     rw   c                    s   g | ]} |r|qS r   r   ).0key)fr   r    
<listcomp>#  s      z#_run_finalizers.<locals>.<listcomp>T)reversez
calling %sr   )	r^   rB   rE   r?   r   rF   	traceback	print_excclear)rx   keysrz   	finalizerr~   r   )r{   rx   r    _run_finalizers  s$    



r   c                   C   s   t p
t dkS )z6
    Returns true if the process is shutting down
    N)_exitingr   r   r   r    r   8  s    c                 C   s   t sda | d |d |d | d k	rr| D ] }|jr0| d|j |j  q0| D ]}| d|j |  qX|d |  d S )NTzprocess 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   ZdaemonnameZ_popenZ	terminatejoin)r   r   r   active_childrenr9   ru   r   r   r    r(   @  s    	



r(   c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
r   c                 C   s   |    t| tj  d S r   )_resetr   r   rf   r   r   r    rb   q  s    zForkAwareThreadLock.__init__c                 C   s"   t  | _| jj| _| jj| _d S r   )	threadingZLock_lockacquirereleaserf   r   r   r    r   u  s    

zForkAwareThreadLock._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 Zd Zdd Zdd ZdS )r   c                 C   s   t | dd  d S )Nc                 S   s
   | j  S r   )__dict__r   )rI   r   r   r    rv     rw   z)ForkAwareLocal.__init__.<locals>.<lambda>)r   rf   r   r   r    rb     s    zForkAwareLocal.__init__c                 C   s   t | dfS )Nr   )rQ   rf   r   r   r    
__reduce__  s    zForkAwareLocal.__reduce__N)ri   rp   rq   rb   r   r   r   r   r    r     s   SC_OPEN_MAX   c                 C   sb   t | dtg } |   | d tks,tdtt| d D ] }t| | d | |d   q<d S )Nzfd too larger   )rB   MAXFDrE   AssertionErrorrangelenr[   
closerange)fdsir   r   r    r     s
    c               	   C   s   t jd krd S zt j  W n ttfk
r4   Y nX z@ttjtj} zt| ddt _W n   t|   Y nX W n ttfk
r   Y nX d S )NF)closefd)	r/   stdincloseOSErrorrU   r[   opendevnullO_RDONLY)fdr   r   r    _close_stdin  s    

r   c                	   C   sT   zt j  W n ttfk
r&   Y nX zt j  W n ttfk
rN   Y nX d S r   )r/   stdoutflushrk   rU   stderrr   r   r   r    _flush_std_streams  s    r   c                 C   sx   dd l }tttt|}t \}}z6||t	| gd|d d dddddd||ddd W S t| t| X d S )Nr   Tr   F)
_posixsubprocesstuplesortedmaprO   r[   piper   Z	fork_execfsencode)pathr   Zpassfdsr   Zerrpipe_readZerrpipe_writer   r   r    spawnv_passfds  s2     
             
r   c                  G   s   | D ]}t | qdS )z/Close each file descriptor given as an argumentN)r[   r   )r   r   r   r   r    	close_fds  s    r   c                  C   sZ   ddl m}  t  ddlm} |j  ddlm} |j	  t
  |   |   dS )zKCleanup multiprocessing resources when multiprocessing tests
    completed.r   )support)
forkserver)resource_trackerN)Ztestr   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)<r[   	itertoolsr/   rR   r'   r   
subprocessr    r   __all__ZNOTSETr   r!   r"   r   r%   r,   r   r-   r   r   r   r   r	   r
   r1   r7   Zabstract_sockets_supportedr<   r   ZWeakValueDictionaryrC   countrM   rK   r   r^   rY   objectr   r   r   r   r   r9   r(   r)   r   Zlocalr   sysconfr   rF   r   r   r   r   r   r   r   r   r   r    <module>
   s              

		V
,
*



