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 )z@
Return process object representing the current process
)�_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�� dS )zQ
Method to be run in sub-process; can be overridden in sub-class
N)r* r, r. r r r r �rung s zBaseProcess.runc C s> | � � t� | �| �| _| jj| _| `| `| `t �
| � dS )z%
Start child process
N) r< r
r r �sentinel� _sentinelr* r, r. r r4 r r r r �startn s
zBaseProcess.startc C s | � � | j�� dS )zT
Terminate process; sends SIGTERM signal or uses TerminateProcess()
N)r< r � terminater r r r rA � s zBaseProcess.terminatec C s | � � | j�� dS )zT
Terminate process; sends SIGKILL signal or uses TerminateProcess()
N)r< r �killr r r r rB � s zBaseProcess.killc C s* | � � | j�|�}|dur&t�| � dS )z5
Wait until child process terminates
N)r< r �waitr r )r �timeout�resr r r r1 � s zBaseProcess.joinc C sJ | � � | tu rdS | jdu r"dS | j�� }|du r8dS t�| � dS dS )z1
Return whether process is alive
TNF)r< r r r r r )r �
returncoder r r �is_alive� s
zBaseProcess.is_alivec C sH | j dur>| j �� du r td��| j �� d| _ | `t�| � d| _dS )z�
Close the Process object.
This method releases resources held by the Process object. It is
an error to call this method if the child process is still running.
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 )z4
Return whether process is a daemon
r F)r" �getr r r r r � s zBaseProcess.daemonc C s || j d< dS )z1
Set whether process is a daemon
r N�r" )r Zdaemonicr r r r � s c C s
| j d S )N�authkeyrK r r r r rL � s zBaseProcess.authkeyc C s t |�| jd< dS )z2
Set authorization key of process
rL N)�AuthenticationStringr" )r rL r r r rL � s c C s" | � � | jdu r| jS | j�� S )zM
Return exit code of process or `None` if it has yet to stop
N)r< r r r r r r �exitcode� s
zBaseProcess.exitcodec C s* | � � | tu rt�� S | jo$| jjS dS )zU
Return identifier (PID) of process or `None` if it has yet to start
N)r< r r$ r% r �pidr r r r �ident� s zBaseProcess.identc C s2 | � � z| jW S ty, td�d�Y n0 dS )z{
Return a file descriptor (Unix) or handle (Windows) suitable for
waiting for process termination.
zprocess not startedN)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 �appendrO �_exitcode_to_namerJ r r1 )r rN 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_forkersrX r= Z_exit_function�
SystemExit�code�
isinstance�int�sys�stderr�writer � tracebackr' � print_excZ _shutdownZ_flush_std_streams)r Zparent_sentinelr[ r\ Z old_processrN �erk r r r �
_bootstrap" sT
�
�
zBaseProcess._bootstrap)N)N)r0 �
__module__� __qualname__�__doc__r r: r< r= r@ rA rB r1 rG rH �propertyr' �setterr rL rN rP rO r> rY rn r r r r r G sD �
r c @ s e Zd Zdd� ZdS )rM c C s, ddl m} |� d u r td��tt| �ffS )NrZ )�get_spawning_popenzJPickling an AuthenticationString object is disallowed for security reasons)r\ rt � TypeErrorrM �bytes)r rt r r r �
__reduce__X s
�z AuthenticationString.__reduce__N)r0 ro rp rw r r r r rM W s rM c @ s6 e Zd Zdd� Zdd� Zedd� �Zd
dd �ZeZdS ) ra c C s4 d| _ || _|| _d | _d | _d| _|| _i | _d S )Nr F)r! r2 �_pidr& r r) r? r" )r r' rO 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 rP v s z_ParentProcess.identNc C s ddl m} || jg|d� dS )z6
Wait until parent process terminates
r ry rz Nr{ )r rD rC r r r r1 z s z_ParentProcess.join)N) r0 ro rp r: rG rr rP r1 rO r r r r ra f s
ra 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)rL Z semprefix) r! r2 r& r r) rM 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 ro rp r: rH r r r r r| � s r| rZ � ZSIG�_r )$�__all__r$ rh �signalr_ rb Z _weakrefsetr �path�abspath�getcwdZ ORIGINAL_DIR�OSErrorr r r r
�objectr rv rM ra r| r
r r9 r r` r rW r �__dict__�itemsr' Zsignumr3 r r r r �
s<
!