a
XC?h�. � @ s2 g d �Z ddlZddlZddlZddlZddlZddlmZ zej� e�
� �Z W n e yf dZ Y n0 dd� Z
dd� Zdd � Zd
d � ZG d d
� d
e�ZG dd� de�ZG dd� de�ZG dd� de�Zdae� ae�d�ae� a [i Z e ej � � �D ]0\Z!Z"e!dd� dkr�de!vr�de!� �e e" < q�e� Z#dS ))� BaseProcess�current_process�active_children�parent_process� N)�WeakSetc C s t S �N)�_current_process� r r �^ � z'BaseProcess.__init__..)�next�_process_counterr � _identity�_config�copy�os�getpid� _parent_pid�name� _parent_namer �_closed�_target�tuple�_args�dict�_kwargs�type�__name__�join�_namer � _dangling�add)r �group�targetr' �args�kwargsr �countr r r
�__init__P s
�zBaseProcess.__init__c C s | j rtd��d S )Nzprocess object is closed)r) �
ValueErrorr r r r
�
_check_closedc s zBaseProcess._check_closedc C s | j r| j | ji | j�� d S r )r* r, r. r r r r
�rung s zBaseProcess.runc C s> | � � t� | �| �| _| jj| _| `| `| `t �
| � d S r ) r< r r r �sentinel� _sentinelr* r, r. r
r4 r r r r
�startn s
zBaseProcess.startc C s | � � | j�� d S r )r< r � terminater r r r
rA � s zBaseProcess.terminatec C s | � � | j�� d S r )r< r �killr r r r
rB � s zBaseProcess.killc C s* | � � | j�|�}|d ur&t�| � d S r )r< r �waitr
r )r �timeout�resr r r
r1 � s zBaseProcess.joinc C sJ | � � | tu rdS | jd u r"dS | j�� }|d u r8dS t�| � dS d S )NTF)r< r r r r
r )r �
returncoder r r
�is_alive� s
zBaseProcess.is_alivec C sH | j d ur>| j �� d u r td��| j �� d | _ | `t�| � d| _d S )Nz^Cannot close a process while it is still running. You should first call join() or terminate().T)r r r; �closer? r
r r) r r r r
rH � s
zBaseProcess.closec C s | j S r �r2 r r r r
r' � s zBaseProcess.namec C s
|| _ d S r rI )r r' r r r
r' � s c C s | j �dd�S )Nr F)r" �getr r r r
r � s zBaseProcess.daemonc C s || j d< d S )Nr �r" )r Zdaemonicr r r
r � s c C s
| j d S �N�authkeyrK r r r r
rM � s zBaseProcess.authkeyc C s t |�| jd< d S rL )�AuthenticationStringr" )r rM r r r
rM � s c C s" | � � | jd u r| jS | j�� S r )r< r r r r r r
�exitcode� s
zBaseProcess.exitcodec C s* | � � | tu rt�� S | jo$| jjS d S r )r< r r$ r% r �pidr r r r
�ident� s zBaseProcess.identc C s2 | � � z| jW S ty, td�d �Y n0 d S )Nzprocess not started)r< r? �AttributeErrorr; r r r r
r> � s
zBaseProcess.sentinelc C s� d }| t u rd}nL| jr d}n@| jt�� kr2d}n,| jd u rBd}n | j�� }|d urZd}nd}t| �jd| j g}| jd ur�|�
d| jj � |�
d| j � |�
|� |d ur�t �
||�}|�
d | � | jr�|�
d
� d d �|� S )
NZstarted�closed�unknown�initialZstoppedzname=%rzpid=%sz parent=%sz exitcode=%sr z<%s>� )r r) r& r$ r% r r r/ r0 r2 �appendrP �_exitcode_to_namerJ r r1 )r rO Zstatus�infor r r
�__repr__ s0
zBaseProcess.__repr__c
C s� ddl m}m} �z`z�| jd ur,|�| j� t�d�at� a |�
� t }| a t | j
| j|�atjrnt�� �� z|j�� |�� W ~n~0 |�d� z| �� d}W |�� n
|�� 0 W n� t�y } zJ|jd u r�d}n.t |jt �r�|j}nt j � t!|j�d � d}W Y d }~n:d }~0 d}dd l"}t j � d| j# � |�$� Y n0 W t�%� |�d| � |�&� n t�%� |�d| � |�&� 0 |S )N� )�util�contextz child process calling self.run()r �
z Process %s:
z process exiting with exitcode %d)'� r\ r] Z
_start_methodZ_force_start_method� itertoolsr9 r �setr
Z _close_stdinr �_ParentProcessr( r& r � threadingZ_HAVE_THREAD_NATIVE_IDZ main_threadZ_set_native_idZ_finalizer_registry�clearZ_run_after_forkersrY r= Z_exit_function�
SystemExit�code�
isinstance�int�sys�stderr�writer � tracebackr' � print_excZ _shutdownZ_flush_std_streams)r Zparent_sentinelr\ r] Z old_processrO �erl r r r
�
_bootstrap" sT
�
�
zBaseProcess._bootstrap)N)N)r0 �
__module__� __qualname__r r: r< r= r@ rA rB r1 rG rH �propertyr' �setterr rM rO rQ rP r> rZ ro r r r r
r G sB �
r c @ s e Zd Zdd� ZdS )rN c C s, ddl m} |� d u r td��tt| �ffS )Nr[ )�get_spawning_popenzJPickling an AuthenticationString object is disallowed for security reasons)r] rt � TypeErrorrN �bytes)r rt r r r
�
__reduce__X s
�z AuthenticationString.__reduce__N)r0 rp rq rw r r r r
rN W s rN c @ s6 e Zd Zdd� Zdd� Zedd� �Zd
dd �ZeZdS ) rb c C s4 d| _ || _|| _d | _d | _d| _|| _i | _d S )Nr F)r! r2 �_pidr& r r) r? r" )r r' rP r> r r r
r: h s z_ParentProcess.__init__c C s ddl m} || jgdd� S �Nr )rC )rD �Zmultiprocessing.connectionrC r? )r rC r r r
rG r s z_ParentProcess.is_alivec C s | j S r )rx r r r r
rQ v s z_ParentProcess.identNc C s ddl m} || jg|d� d S ry rz )r rD rC r r r
r1 z s z_ParentProcess.join)N) r0 rp rq r: rG rr rQ r1 rP r r r r
rb f s
rb c @ s e Zd Zdd� Zdd� ZdS )� _MainProcessc C s8 d| _ d| _d | _d | _d| _tt�d��dd�| _d S )Nr Z MainProcessF� z/mp)rM Z semprefix) r! r2 r& r r) rN r$ �urandomr" r r r r
r: � s �z_MainProcess.__init__c C s d S r r r r r r
rH � s z_MainProcess.closeN)r0 rp rq r: rH r r r r
r{ � s r{ r[ � ZSIG�_r )$�__all__r$ ri �signalr` rc Z _weakrefsetr �path�abspath�getcwdZ ORIGINAL_DIR�OSErrorr r r r �objectr rv rN rb r{ r r r9 r ra r
rX r �__dict__�itemsr' Zsignumr3 r r r r
�
s<
!