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 sz | � � | jdu sJ d��| jt�� ks0J d��tj�d�rDJ d��t� | � | �| _| jj
| _ | ` | `
| `t�| � dS )z%
Start child process
Nz cannot start a process twicez:can only start a process object created by current processr z3daemonic processes are not allowed to have children)r< r r& r$ r% r r" �getr
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 rB � s zBaseProcess.terminatec C s | � � | j�� dS )zT
Terminate process; sends SIGKILL signal or uses TerminateProcess()
N)r< r �killr r r r rC � s zBaseProcess.killc C sR | � � | jt�� ks J d��| jdus0J d��| j�|�}|durNt�| � dS )z5
Wait until child process terminates
z can only join a child processNz can only join a started process)r< r& r$ r% r �waitr r )r �timeout�resr r r r1 � s zBaseProcess.joinc C s` | � � | tu rdS | jt�� ks*J d��| jdu r8dS | j�� }|du rNdS t�| � dS dS )z1
Return whether process is alive
Tz can only test a child processNF) r< r r& 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 rI � s
zBaseProcess.closec C s | j S r )r2 r r r r r' � s zBaseProcess.namec C s t |t�sJ d��|| _d S )Nzname must be a string)�
isinstancer r2 )r r' r r r r' � s c C s | j �dd�S )z4
Return whether process is a daemon
r F)r" r> r r r r r � s zBaseProcess.daemonc C s | j du sJ d��|| jd< dS )z1
Set whether process is a daemon
Nzprocess has already startedr )r r" )r Zdaemonicr r r r � s c C s
| j d S )N�authkey)r" r r r r rK � s zBaseProcess.authkeyc C s t |�| jd< dS )z2
Set authorization key of process
rK N)�AuthenticationStringr" )r rK r r r rK � 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 �appendrN �_exitcode_to_namer> r r1 )r rM 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)'� rZ 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_forkersrW r= Z_exit_function�
SystemExit�coderJ �int�sys�stderr�writer � tracebackr' � print_excZ _shutdownZ_flush_std_streams)r Zparent_sentinelrZ r[ Z old_processrM �eri r r r �
_bootstrap" sT
�
�
zBaseProcess._bootstrap)N)N)r0 �
__module__� __qualname__�__doc__r r: r< r= rA rB rC r1 rH rI �propertyr' �setterr rK rM rO rN r? rX rl r r r r r G sD �
r c @ s e Zd Zdd� ZdS )rL c C s, ddl m} |� d u r td��tt| �ffS )NrY )�get_spawning_popenzJPickling an AuthenticationString object is disallowed for security reasons)r[ rr � TypeErrorrL �bytes)r rr r r r �
__reduce__X s
�z AuthenticationString.__reduce__N)r0 rm rn ru r r r r rL W s rL c @ s6 e Zd Zdd� Zdd� Zedd� �Zd
dd �ZeZdS ) r` c C s4 d| _ || _|| _d | _d | _d| _|| _i | _d S )Nr F)r! r2 �_pidr& r r) r@ r" )r r' rN r? r r r r: h s z_ParentProcess.__init__c C s ddl m} || jgdd� S )Nr �rD �rE �Zmultiprocessing.connectionrD r@ )r rD r r r rH r s z_ParentProcess.is_alivec C s | j S r )rv r r r r rO v s z_ParentProcess.identNc C s ddl m} || jg|d� dS )z6
Wait until parent process terminates
r rw rx Nry )r rE rD r r r r1 z s z_ParentProcess.join)N) r0 rm rn r: rH rp rO r1 rN r r r r r` f s
r` 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)rK Z semprefix) r! r2 r& r r) rL r$ �urandomr" r r r r r: � s �z_MainProcess.__init__c C s d S r r r r r r rI � s z_MainProcess.closeN)r0 rm rn r: rI r r r r rz � s rz rY � ZSIG�_r )$�__all__r$ rf �signalr^ ra Z _weakrefsetr �path�abspath�getcwdZ ORIGINAL_DIR�OSErrorr r r r
�objectr rt rL r` rz r
r r9 r r_ r rV r �__dict__�itemsr' Zsignumr3 r r r r �
s<
!