B
�/g^�* c @ s� d dl Z d dlZd dlZddlmZ ddlmZ g ZG dd� de�ZG dd� de�Z G d d
� d
e�Z
G d d � d e�Z G d
d� de �Z
G dd� dej�ZG dd� de
�Zdd� ee�D �e_ejdk�rLG dd� dej�ZG dd� dej�ZG dd� dej�ZG d d � d e
�ZG d d � d e
�ZG d d!� d!e
�Ze� e� e� d"�Zeed# �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)� reductionc @ � e Zd ZdS )� ProcessErrorN��__name__�
__module__� __qualname__� r
r
�//usr/lib64/python3.7/multiprocessing/context.pyr � r c @ r )�BufferTooShortNr r
r
r
r r
r r
c @ r )� TimeoutErrorNr r
r
r
r r r r c @ r )�AuthenticationErrorNr r
r
r
r r r r c @ sN e Zd ZeZeZeZeZeej �Z eej
�Z
dd� Z dd� Z dCdd�Z
dd � Zd
d � ZdDd
d�ZdEdd�ZdFdd�Zdd� ZdGdd�ZdHdd�ZdIdd �Zd d � ZdJd d!�Zd"d#� Zd$d%� Zdd&�d'd(�Zdd&�d)d*�Z d+d,� Z d-d.� Z dKd/d0�Z d1d2� Z d3d4� Z!d5d6� Z"dLd7d8�Z#dMd:d;�Z$dNd<d=�Z%e&d>d?� �Z'e'j(d@d?� �Z'dAdB� Z)d S )O� BaseContextc C s" t �� }|dkrtd��n|S dS )z(Returns the number of CPUs in the systemNz cannot determine number of cpus)�os� cpu_countZNotImplementedError)�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_contextZstart)r r Zmr
r
r �Manager0 s zBaseContext.ManagerTc C � ddl m} ||�S )z1Returns two connection object connected by a piper )�Pipe)�
connectionr )r Zduplexr r
r
r r ; � zBaseContext.Pipec C � ddl m} || �� d�S )z#Returns a non-recursive lock objectr )�Lockr )� synchronizer r )r r r
r
r r @ r zBaseContext.Lockc C r )z Returns a recursive lock objectr )�RLockr )r r r )r r r
r
r r E r zBaseContext.RLockNc C � ddl m} ||| �� d�S )zReturns a condition objectr )� Conditionr )r r" r )r �lockr" r
r
r r" J r zBaseContext.Conditionr c C r! )zReturns a semaphore objectr )� Semaphorer )r r$ r )r �valuer$ r
r
r r$ O r zBaseContext.Semaphorec C r! )z"Returns a bounded semaphore objectr )�BoundedSemaphorer )r r& r )r r% r& r
r
r r& T r z BaseContext.BoundedSemaphorec C r )zReturns an event objectr )�Eventr )r r' r )r r' r
r
r r' Y r zBaseContext.Eventc C � ddl m} ||||| �� d�S )zReturns a barrier objectr )�Barrierr )r r) r )r ZpartiesZactionZtimeoutr) r
r
r r) ^ r zBaseContext.Barrierr c C r! )�Returns a queue objectr )�Queuer )�queuesr+ r )r �maxsizer+ r
r
r r+ c r zBaseContext.Queuec C r! )r* r )�
JoinableQueuer )r, r. r )r r- r. r
r
r r. h r zBaseContext.JoinableQueuec C r )r* r )� SimpleQueuer )r, r/ r )r r/ r
r
r r/ m r zBaseContext.SimpleQueuer
c C s" ddl m} |||||| �� d�S )z Returns a process pool objectr )�Pool)�context)Zpoolr0 r )r Z processesZ initializerZinitargsZmaxtasksperchildr0 r
r
r r0 r s
zBaseContext.Poolc G s ddl m} ||f|�� S )zReturns a shared objectr )�RawValue)� sharedctypesr2 )r �typecode_or_type�argsr2 r
r
r r2 y r zBaseContext.RawValuec C s ddl m} |||�S )zReturns a shared arrayr )�RawArray)r3 r6 )r r4 �size_or_initializerr6 r
r
r r6 ~ r zBaseContext.RawArray)r# c G s& ddl m} ||f|�|| �� d��S )z$Returns a synchronized shared objectr )�Value�r# r )r3 r8 r )r r4 r# r5 r8 r
r
r r8 � s zBaseContext.Valuec C r( )z#Returns a synchronized shared arrayr )�Arrayr9 )r3 r: r )r r4 r7 r# r: r
r
r r: � 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.
�win32ZfrozenFr )�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)�utilrA )r rA r
r
r rA � � zBaseContext.get_loggerc C r )z8Turn on logging and add a handler which prints to stderrr )�
log_to_stderr)rB rD )r ZlevelrD r
r
r rD � r 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 � 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@ rH )r Z
executablerH r
r
r rH � s zBaseContext.set_executablec C rG )zkSet list of module names to try to load in forkserver process.
This is really just a hint.
r )�set_forkserver_preloadN)�
forkserverrI )r Z module_namesrI r
r
r rI � rC z"BaseContext.set_forkserver_preloadc C sH |d kr | S y t | }W n" tk
r: td| �d �Y nX |�� |S )Nzcannot find context for %r)�_concrete_contextsZKeyError�
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� � 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� rS 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 )�globalsZget�r r
r
r �reducer� s zBaseContext.reducerc C s |t � d< d S )Nr )rV )r r r
r
r rX � � c C s d S rO r
rW r
r
r rM � rS z BaseContext._check_available)TrO �r rZ )NN�r r[ )NNr
NrO rO �Fr\ )*r r r r r
r r � staticmethodr Zcurrent_processZactive_childrenr r r r r r" r$ r& r' r) r+ r. r/ r0 r2 r6 r8 r: r< rA rD rF rH rI r rR rU ZpropertyrX ZsetterrM r
r
r
r r sJ
r c @ s e Zd ZdZedd� �ZdS )�ProcessNc C s t �� j�| �S rO )�_default_contextr r^ �_Popen)� process_objr
r
r r` � rY zProcess._Popen�r r r Z
_start_methodr] r` r
r
r
r r^ � � 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 r1 r
r
r �__init__� � zDefaultContext.__init__Nc s0 |d kr | j d kr| j| _ | j S t� �|�S d S rO )re r_ Zsuperr )r rN �Z __class__r
r r � s
zDefaultContext.get_contextFc C s< | j d k r|std��|d kr,|r,d | _ d S | �|�| _ d S )Nz context has already been set)re � RuntimeErrorr rT r
r
r rU � s z DefaultContext.set_start_methodc C s" | j d kr|rd S | j| _ | j jS rO )re r_ rP rQ r
r
r rR � s
z DefaultContext.get_start_methodc C s, t jdkrdgS tjr dddgS ddgS d S )Nr; r@ �forkrJ )r= r> r �HAVE_SEND_HANDLErW r
r
r �get_all_start_methods� s
z$DefaultContext.get_all_start_methodsrO r\ r\ )
r r r r^ rf r rU rR rl Z
__classcell__r
r
rh r rd � s
rd c C s g | ]}|d dkr|�qS )r Z_r
)Z.0Zxr
r
r �
<listcomp> s rm r; c @ � e Zd ZdZedd� �ZdS )� ForkProcessrj c C � ddl m} || �S �Nr )�Popen)Z
popen_forkrr �ra rr r
r
r r` r zForkProcess._PopenNrb r
r
r
r ro rc ro c @ rn )� SpawnProcessr@ c C rp rq )Zpopen_spawn_posixrr rs r
r
r r` r �SpawnProcess._PopenNrb r
r
r
r rt rc rt c @ rn )�ForkServerProcessrJ c C rp rq )Zpopen_forkserverrr rs r
r
r r` r zForkServerProcess._PopenNrb r
r
r
r rv rc rv c @ � e Zd ZdZeZdS )� ForkContextrj N)r r r rP ro r^ r
r
r
r rx % rc rx c @ rw �� SpawnContextr@ N�r r r rP rt r^ r
r
r
r rz ) rc rz c @ s e Zd ZdZeZdd� ZdS )�ForkServerContextrJ c C s t jstd��d S )Nz%forkserver start method not available)r rk rL rW r
r
r rM 0 rg z"ForkServerContext._check_availableN)r r r rP rv r^ rM r
r
r
r r| - s r| )rj r@ rJ rj c @ rn )rt r@ c C rp rq )Zpopen_spawn_win32rr rs r
r
r r` ? r ru Nrb r
r
r
r rt = rc c @ rw ry r{ r
r
r
r rz D rc r@ c C s t | t_d S rO )rK r_ re )rN r
r
r �_force_start_methodQ rS r} c C s t tdd �S )N�spawning_popen)r? �_tlsr
r
r
r �get_spawning_popenZ rS r� c C s
| t _d S rO )r r~ )Zpopenr
r
r �set_spawning_popen] rS r� c C s t � d kr tdt| �j ��d S )NzF%s objects should only be shared between processes through inheritance)r� ri Ztyper )Zobjr
r
r �assert_spawning` s
r� )!r r= Z threadingrE r r Z__all__Z Exceptionr r
r r Zobjectr Z BaseProcessr^ rd Zdirr> ro rt rv rx rz r| rK r_ r} Zlocalr r� r� r� r
r
r
r Z<module> sD >'