a
XC?h�+ � @ s� d dl Z d dlZd dlZddlmZ ddlmZ dZG dd� de�ZG dd � d e�Z G d
d � d e�Z
G d d
� d
e�Z G dd� de �Z
G dd� dej�ZG dd� de
�Zejdk�rRG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd � d e
�ZG d d � d e
�ZG d d � d e
�Ze� e� e� d!�Zejd"k�rDeed# �Zn eed$ �Zn8G d%d� dej�ZG d&d � d e
�Zd#e� iZeed# �Zd'd(� Ze�� Z d)d*� Z d+d,� Z d-d.� Z dS )/� N� )�process)� reduction� c @ s e Zd ZdS )� ProcessErrorN��__name__�
__module__� __qualname__r r r �d?� �Z(e(j)d@d?� �Z(dAdB� Z*d S )O� BaseContextc C s" t �� }|d u rtd��n|S d S )Nz cannot determine number of cpus)�os� cpu_count�NotImplementedError)�selfZnumr r r r ) s
zBaseContext.cpu_countc C s& ddl m} || �� d�}|�� |S )Nr )� SyncManager��ctx)Zmanagersr � get_context�start)r r �mr r r �Manager1 s zBaseContext.ManagerTc C s ddl m} ||�S )Nr )�Pipe)�
connectionr )r Zduplexr r r r r < s zBaseContext.Pipec C s ddl m} || �� d�S )Nr )�Lockr )� synchronizer r )r r r r r r A s zBaseContext.Lockc C s ddl m} || �� d�S )Nr )�RLockr )r r r )r r r r r r F s zBaseContext.RLockNc C s ddl m} ||| �� d�S )Nr )� Conditionr )r r r )r �lockr r r r r K s zBaseContext.Conditionr c C s ddl m} ||| �� d�S )Nr )� Semaphorer )r r" r )r �valuer" r r r r" P s zBaseContext.Semaphorec C s ddl m} ||| �� d�S )Nr )�BoundedSemaphorer )r r$ r )r r# r$ r r r r$ U s z BaseContext.BoundedSemaphorec C s ddl m} || �� d�S )Nr )�Eventr )r r% r )r r% r r r r% Z s zBaseContext.Eventc C s ddl m} ||||| �� d�S )Nr )�Barrierr )r r& r )r Zparties�actionZtimeoutr& r r r r& _ s zBaseContext.Barrierr c C s ddl m} ||| �� d�S )Nr )�Queuer )�queuesr( r )r �maxsizer( r r r r( d s zBaseContext.Queuec C s ddl m} ||| �� d�S )Nr )�
JoinableQueuer )r) r+ r )r r* r+ r r r r+ i s zBaseContext.JoinableQueuec C s ddl m} || �� d�S )Nr )� SimpleQueuer )r) r, r )r r, r r r r, n s zBaseContext.SimpleQueuer c C s" ddl m} |||||| �� d�S )Nr )�Pool)�context)Zpoolr- r )r Z processesZ initializerZinitargsZmaxtasksperchildr- r r r r- s s
�zBaseContext.Poolc G s ddl m} ||g|�R � S )Nr )�RawValue)� sharedctypesr/ )r �typecode_or_type�argsr/ r r r r/ z s zBaseContext.RawValuec C s ddl m} |||�S )Nr )�RawArray)r0 r3 )r r1 �size_or_initializerr3 r r r r3 s zBaseContext.RawArray)r! c G s( ddl m} ||g|�R || �� d��S )Nr )�Value�r! r )r0 r5 r )r r1 r! r2 r5 r r r r5 � s �zBaseContext.Valuec C s ddl m} ||||| �� d�S )Nr )�Arrayr6 )r0 r7 r )r r1 r4 r! r7 r r r r7 � s �zBaseContext.Arrayc C s, t jdkr(tt dd�r(ddlm} |� d S )N�win32�frozenFr )�freeze_support)�sys�platform�getattr�spawnr: )r r: r r r r: � s zBaseContext.freeze_supportc C s ddl m} |� S )Nr )�
get_logger)�utilr? )r r? r r r r? � s zBaseContext.get_loggerc C s ddl m} ||�S )Nr )�
log_to_stderr)r@ rA )r �levelrA r r r rA � s zBaseContext.log_to_stderrc C s ddl m} d S )Nr )r )� r )r r r r r �allow_connection_pickling� s z%BaseContext.allow_connection_picklingc C s ddl m} ||� d S )Nr )�set_executable)r> rE )r �
executablerE r r r rE � s zBaseContext.set_executablec C s ddl m} ||� d S )Nr )�set_forkserver_preload)�
forkserverrG )r Z module_namesrG r r r rG � s z"BaseContext.set_forkserver_preloadc C sF |d u r | S z t | }W n ty8 td| �d �Y n0 |�� |S )Nzcannot find context for %r)�_concrete_contexts�KeyError�
ValueError�_check_available)r �methodr r r r r � s zBaseContext.get_contextFc C s | j S �N)�_name�r Z
allow_noner r r �get_start_method� s z BaseContext.get_start_methodc C s t d��d S )Nz+cannot set start method of concrete context)rK �r rM Zforcer r r �set_start_method� s z BaseContext.set_start_methodc C s t � �d�S �Nr )�globals�get�r r r r �reducer� s zBaseContext.reducerc C s |t � d< d S rT )rU )r r r r r rX � s c C s d S rN r rW r r r rL � s z BaseContext._check_available)T)N)r )r )NN)r )r )NNr N)N)N)F)F)+r r r
r r r
r � staticmethodr Zcurrent_processZparent_processZactive_childrenr r r r r r r" r$ r% r& r( r+ r, r- r/ r3 r5 r7 r: r? rA rD rE rG r rQ rS �propertyrX �setterrL r r r r r sR
�
r c @ s e Zd ZdZedd� �ZdS )�ProcessNc C s t �� j�| �S rN )�_default_contextr r\ �_Popen)� process_objr r r r^ � s zProcess._Popen�r r r
Z
_start_methodrY r^ r r r r r\ � s r\ c sF e Zd ZeZdd� Zd
� fdd� Zddd�Zdd d
�Zd d � Z� Z S )�DefaultContextc C s || _ d | _d S rN )r] �_actual_context)r r. r r r �__init__� s zDefaultContext.__init__Nc s0 |d u r | j d u r| j| _ | j S t� �|�S d S rN )rb r] �superr )r rM �� __class__r r r � s
zDefaultContext.get_contextFc C s< | j d ur|std��|d u r,|r,d | _ d S | �|�| _ d S )Nz context has already been set)rb � RuntimeErrorr rR r r r rS � s z DefaultContext.set_start_methodc C s" | j d u r|rd S | j| _ | j jS rN )rb r] rO rP r r r rQ � s
z DefaultContext.get_start_methodc C sB t jdkrdgS t jdkr"ddgnddg}tjr:|�d� |S d S )Nr8 r> �darwin�forkrH )r; r< r �HAVE_SEND_HANDLE�append)r �methodsr r r �get_all_start_methods s
z$DefaultContext.get_all_start_methods)N)F)F)
r r r
r\ rc r rS rQ rm �
__classcell__r r re r ra � s
ra r8 c @ s e Zd ZdZedd� �ZdS )� ForkProcessri c C s ddl m} || �S �Nr )�Popen)Z
popen_forkrq �r_ rq r r r r^ s zForkProcess._PopenNr` r r r r ro s ro c @ s e Zd ZdZedd� �ZdS )� SpawnProcessr> c C s ddl m} || �S rp )Zpopen_spawn_posixrq rr r r r r^ s �SpawnProcess._PopenNr` r r r r rs s rs c @ s e Zd ZdZedd� �ZdS )�ForkServerProcessrH c C s ddl m} || �S rp )Zpopen_forkserverrq rr r r r r^ s zForkServerProcess._PopenNr` r r r r ru s ru c @ s e Zd ZdZeZdS )� ForkContextri N)r r r
rO ro r\ r r r r rv % s rv c @ s e Zd ZdZeZdS �� SpawnContextr> N�r r r
rO rs r\ r r r r rx ) s rx c @ s e Zd ZdZeZdd� ZdS )�ForkServerContextrH c C s t jstd��d S )Nz%forkserver start method not available)r rj rK rW r r r rL 0 s z"ForkServerContext._check_availableN)r r r
rO ru r\ rL r r r r rz - s rz )ri r> rH rh r> ri c @ s e Zd ZdZedd� �ZdS )rs r> c C s ddl m} || �S rp )Zpopen_spawn_win32rq rr r r r r^ D s rt Nr` r r r r rs B s c @ s e Zd ZdZeZdS rw ry r r r r rx I s c C s t | t_d S rN )rI r] rb )rM r r r �_force_start_methodV s r{ c C s t tdd �S )N�spawning_popen)r= �_tlsr r r r �get_spawning_popen_ s r~ c C s
| t _d S rN )r} r| )�popenr r r �set_spawning_popenb s r� c C s t � d u r tdt| �j ��d S )NzF%s objects should only be shared between processes through inheritance)r~ rg �typer )�objr r r �assert_spawninge s
��r� ) r r; Z threadingrC r r �__all__� Exceptionr r r
r �objectr Z BaseProcessr\ ra r< ro rs ru rv rx rz rI r] r{ Zlocalr} r~ r� r� r r r r � sJ ?, � �