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 �� }|du rtd��n|S dS )z(Returns the number of CPUs in the systemNz cannot determine number of cpus)�os� cpu_count�NotImplementedError)�selfZnumr r r r ) s
zBaseContext.cpu_countc C s& ddl m} || �� d�}|�� |S )z�Returns a manager associated with a running server process
The managers methods such as `Lock()`, `Condition()` and `Queue()`
can be used to create shared objects.
r )� SyncManager��ctx)Zmanagersr � get_context�start)r r �mr r r �Manager1 s zBaseContext.ManagerTc C s ddl m} ||�S )z1Returns two connection object connected by a piper )�Pipe)�
connectionr )r Zduplexr r r r r < s zBaseContext.Pipec C s ddl m} || �� d�S )z#Returns a non-recursive lock objectr )�Lockr )� synchronizer r )r r r r r r A s zBaseContext.Lockc C s ddl m} || �� d�S )z Returns a recursive lock objectr )�RLockr )r r r )r r r r r r F s zBaseContext.RLockNc C s ddl m} ||| �� d�S )zReturns a condition objectr )� Conditionr )r r r )r �lockr r r r r K s zBaseContext.Conditionr c C s ddl m} ||| �� d�S )zReturns a semaphore objectr )� Semaphorer )r r" r )r �valuer" r r r r" P s zBaseContext.Semaphorec C s ddl m} ||| �� d�S )z"Returns a bounded semaphore objectr )�BoundedSemaphorer )r r$ r )r r# r$ r r r r$ U s z BaseContext.BoundedSemaphorec C s ddl m} || �� d�S )zReturns an event objectr )�Eventr )r r% r )r r% r r r r% Z s zBaseContext.Eventc C s ddl m} ||||| �� d�S )zReturns a barrier objectr )�Barrierr )r r& r )r Zparties�actionZtimeoutr& r r r r& _ s zBaseContext.Barrierr c C s ddl m} ||| �� d�S )�Returns a queue objectr )�Queuer )�queuesr) r )r �maxsizer) r r r r) d s zBaseContext.Queuec C s ddl m} ||| �� d�S )r( r )�
JoinableQueuer )r* r, r )r r+ r, r r r r, i s zBaseContext.JoinableQueuec C s ddl m} || �� d�S )r( r )� SimpleQueuer )r* r- r )r r- r r r r- n s zBaseContext.SimpleQueuer c C s" ddl m} |||||| �� d�S )z Returns a process pool objectr )�Pool)�context)Zpoolr. r )r Z processesZ initializerZinitargsZmaxtasksperchildr. r r r r. s s
�zBaseContext.Poolc G s ddl m} ||g|�R � S )zReturns a shared objectr )�RawValue)� sharedctypesr0 )r �typecode_or_type�argsr0 r r r r0 z s zBaseContext.RawValuec C s ddl m} |||�S )zReturns a shared arrayr )�RawArray)r1 r4 )r r2 �size_or_initializerr4 r r r r4 s zBaseContext.RawArray)r! c G s( ddl m} ||g|�R || �� d��S )z$Returns a synchronized shared objectr )�Value�r! r )r1 r6 r )r r2 r! r3 r6 r r r r6 � s �zBaseContext.Valuec C s ddl m} ||||| �� d�S )z#Returns a synchronized shared arrayr )�Arrayr7 )r1 r8 r )r r2 r5 r! r8 r r r r8 � s �zBaseContext.Arrayc C s, t jdkr(tt dd�r(ddlm} |� dS )z�Check whether this is a fake forked process in a frozen executable.
If so then run code specified by commandline and exit.
�win32�frozenFr )�freeze_supportN)�sys�platform�getattr�spawnr; )r r; r r r r; � s zBaseContext.freeze_supportc C s ddl m} |� S )zZReturn package logger -- if it does not already exist then
it is created.
r )�
get_logger)�utilr@ )r r@ r r r r@ � s zBaseContext.get_loggerc C s ddl m} ||�S )z8Turn on logging and add a handler which prints to stderrr )�
log_to_stderr)rA rB )r �levelrB r r r rB � s zBaseContext.log_to_stderrc C s ddl m} dS )zVInstall support for sending connections and sockets
between processes
r )r N)� r )r r r r r �allow_connection_pickling� s z%BaseContext.allow_connection_picklingc C s ddl m} ||� dS )z�Sets the path to a python.exe or pythonw.exe binary used to run
child processes instead of sys.executable when using the 'spawn'
start method. Useful for people embedding Python.
r )�set_executableN)r? rF )r �
executablerF r r r rF � s zBaseContext.set_executablec C s ddl m} ||� dS )zkSet list of module names to try to load in forkserver process.
This is really just a hint.
r )�set_forkserver_preloadN)�
forkserverrH )r Z module_namesrH r r r rH � 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)rL �r rN Zforcer r r �set_start_method� s z BaseContext.set_start_methodc C s t � �d�S )z_Controls how objects will be reduced to a form that can be
shared with other processes.r )�globals�get�r r r r �reducer� s zBaseContext.reducerc C s |t � d< d S )Nr )rU )r r r r r rX � s c C s d S rO r rW r r r rM � 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. r0 r4 r6 r8 r; r@ rB rE rF rH r rR rT �propertyrX �setterrM r r r r r sR
�
r c @ s e Zd ZdZedd� �ZdS )�ProcessNc C s t �� j�| �S rO )�_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 rO )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 rO )rb r] �superr )r rN �� __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 rS r r r rT � s z DefaultContext.set_start_methodc C s" | j d u r|rd S | j| _ | j jS rO )rb r] rP rQ r r r rR � s
z DefaultContext.get_start_methodc C sB t jdkrdgS t jdkr"ddgnddg}tjr:|�d� |S d S )Nr9 r? �darwin�forkrI )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 rT rR rm �
__classcell__r r re r ra � s
ra r9 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 )�ForkServerProcessrI 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
rP ro r\ r r r r rv % s rv c @ s e Zd ZdZeZdS �� SpawnContextr? N�r r r
rP rs r\ r r r r rx ) s rx c @ s e Zd ZdZeZdd� ZdS )�ForkServerContextrI c C s t jstd��d S )Nz%forkserver start method not available)r rj rL rW r r r rM 0 s z"ForkServerContext._check_availableN)r r r
rP ru r\ rM r r r r rz - s rz )ri r? rI 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 rO )rJ r] rb )rN 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 rO )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 threadingrD r r �__all__� Exceptionr r r
r �objectr Z BaseProcessr\ ra r= ro rs ru rv rx rz rJ r] r{ Zlocalr} r~ r� r� r r r r � sJ ?, � �