o

E�hM-�@s�
dd
lZdd
l
Z
dd
lZddlmZ
ddlmZ
dZGdd�de�ZGdd	�d	e�Z	Gd
d�de�Z
Gdd
�d
e�ZGdd�de�Z
Gdd�dej�ZGdd�de
�Ze
jdkr�Gdd�dej�ZGdd�dej�ZGdd�dej�ZGdd�de
�ZGdd�de
�ZGdd �d e
�Ze�e�e�d!�Ze
jd"kr�eed#�
Zn#eed$�
ZnGd%d�dej�ZGd&d�de
�Zd#e�i
Zeed#�
Zd'd(�Ze��Zd)d*�Zd+d,�Zd-d.�Zd
S)/�N�
)
�process)
�	reduction�c
@�eZ
dZd
S)�ProcessErrorN��__name__�
__module__�__qualname__rrr�./usr/lib/python3.10/multiprocessing/context.pyr�
rc
@r)�BufferTooShortNrrrrrrr
rc
@r)�TimeoutErrorNrrrrrrr
rc
@r)�AuthenticationErrorNrrrrrrr
rc@s\
eZ
dZeZeZeZeZeej	�
Z	eej
�
Z
eej�
Zd
d�Zdd�Z
dCdd�
Zdd	�Zd
d�ZdDd
d�
ZdEdd�
ZdEdd�
Zdd�ZdFdd�
ZdGdd�
ZdGdd�
Zdd�Z		dHd d!�
Zd"d#�Zd$d%�Zdd&�
d'd(�Zdd&�
d)d*�Zd+d,�Zd-d.�ZdDd/d0�
Z d1d2�Z!d3d4�Z"d5d6�Z#dDd7d8�
Z$dId:d;�
Z%dId<d=�
Z&e'd>d?��
Z(e(j)d@d?��
Z(dAdB�Z*dS)J�BaseContextc
Cst�
�}
|
d
urtd�
�
|
S)z(Returns the number of CPUs in the systemNzcannot determine number of cpus)�os�	cpu_count�NotImplementedError)�self�numrrrr)s

zBaseContext.cpu_countc
Cs&d
dlm
}

|
|��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)�managersr�get_context�start)rr�
mrrr�Manager1s


zBaseContext.ManagerTcC�d
dlm
}
||
�
S)z1Returns two connection object connected by a piper)
�Pipe)�
connectionr )r�duplexr rrrr <�
zBaseContext.Pipec
C�d
dlm
}

|
|��d�
S)z#Returns a non-recursive lock objectr)
�Lockr)�synchronizer%r)rr%rrrr%A�
zBaseContext.Lockc
Cr$)zReturns a recursive lock objectr)
�RLockr)r&r(r)rr(rrrr(Fr'zBaseContext.RLockNcC�d
dlm
}
||
|��d�S)zReturns a condition objectr)
�	Conditionr)r&r*r)r�lockr*rrrr*K�
zBaseContext.ConditionrcCr))zReturns a semaphore objectr)
�	Semaphorer)r&r-r)r�valuer-rrrr-Pr,zBaseContext.SemaphorecCr))z"Returns a bounded semaphore objectr)
�BoundedSemaphorer)r&r/r)rr.r/rrrr/Ur,zBaseContext.BoundedSemaphorec
Cr$)zReturns an event objectr)
�Eventr)r&r0r)rr0rrrr0Zr'zBaseContext.EventcC� d
dlm
}
||
|||��d�S)zReturns a barrier objectr)
�Barrierr)r&r2r)r�parties�action�timeoutr2rrrr2_s
zBaseContext.Barrierr
cCr))�Returns a queue objectr)
�Queuer)�queuesr7r)r�maxsizer7rrrr7dr,zBaseContext.QueuecCr))r6r)
�
JoinableQueuer)r8r:r)rr9r:rrrr:ir,zBaseContext.JoinableQueuec
Cr$)r6r)
�SimpleQueuer)r8r;r)rr;rrrr;nr'zBaseContext.SimpleQueuercCs"d
dlm
}
||
||||��d�S)zReturns a process pool objectr)
�Pool)
�context)�poolr<r)r�	processes�initializer�initargs�maxtasksperchildr<rrrr<ss


�zBaseContext.PoolcGsd
dlm
}
||
g
|�
R�S)zReturns a shared objectr)
�RawValue)�sharedctypesrC)r�typecode_or_type�argsrCrrrrCzr,zBaseContext.RawValuecCsd
dlm
}
||
|�S)zReturns a shared arrayr)
�RawArray)rDrG)rrE�size_or_initializerrGrrrrGs

zBaseContext.RawArray)
r+c
Gs(d
dlm
}
||
g
|�
R||��d��
S)z$Returns a synchronized shared objectr)
�Value�r+r)rDrIr)rrEr+rFrIrrrrI�s

�zBaseContext.Valuec
Cr1)z#Returns a synchronized shared arrayr)
�ArrayrJ)rDrKr)rrErHr+rKrrrrK�s

�zBaseContext.Arrayc
Cs4tj
d
krttdd�rddlm}

|
�
dSdSdS)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�spawnrN)rrNrrrrN�s


�zBaseContext.freeze_supportc
Csd
dlm
}

|
�S)zZReturn package logger -- if it does not already exist then
        it is created.
        r)
�
get_logger)�utilrS)rrSrrrrS�s
zBaseContext.get_loggercCr)z8Turn on logging and add a handler which prints to stderrr)
�
log_to_stderr)rTrU)r�levelrUrrrrU�r#zBaseContext.log_to_stderrc
Csd
dlm
}

dS)zVInstall support for sending connections and sockets
        between processes
        r)
r!N)�r!)rr!rrr�allow_connection_pickling�sz%BaseContext.allow_connection_picklingcC�d
dlm
}
||
�

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)rRrZ)r�
executablerZrrrrZ�s
zBaseContext.set_executablecCrY)zkSet list of module names to try to load in forkserver process.
        This is really just a hint.
        r)
�set_forkserver_preloadN)�
forkserverr\)r�module_namesr\rrrr\�s
z"BaseContext.set_forkserver_preloadcCsB|
dur|Szt|
}Wnt
y


td
|
�
d�w|��
|S)Nzcannot find context for %r)�_concrete_contexts�KeyError�
ValueError�_check_available)r�methodrrrrr�s





�
zBaseContext.get_contextFc
Cs|jS�
N)
�_name�r�
allow_nonerrr�get_start_method�s
zBaseContext.get_start_methodcCstd
�
�
)Nz+cannot set start method of concrete context)
ra�rrc�forcerrr�set_start_method�s
zBaseContext.set_start_methodc

Cst��
d
�
S)z_Controls how objects will be reduced to a form that can be
        shared with other processes.r)�globals�get�
rrrr�reducer�szBaseContext.reducercCs|
t�d
<dS)Nr)
rl)rrrrrro��c


C�dSrdrrnrrrrb�s
zBaseContext._check_available)
Trd)
r)NN)
r
)NNrN�
F)+r	r
rrrrr�staticmethodr�current_process�parent_process�active_childrenrrr r%r(r*r-r/r0r2r7r:r;r<rCrGrIrKrNrSrUrXrZr\rrhrk�propertyro�setterrbrrrrrsT
















�








rc@s(eZ
dZd
Zedd��
Zedd��
Zd
S)�ProcessNc

Cst�
�j�|�
Srd)�_default_contextrry�_Popen)
�process_objrrrr{�szProcess._PopencCst�
�j��Srd)rzrry�_after_forkrrrrr}�rpzProcess._after_fork�r	r
r�
_start_methodrsr{r}rrrrry�s




rycsFeZ
dZeZd
d�Zd
�f
dd�	Zddd�
Zdd	d
�
Zdd�Z�Z	S)�DefaultContextcCs|
|_d|_
dSrd)rz�_actual_context)rr=rrr�__init__�s


zDefaultContext.__init__Ncs,|
dur|jdur
|j
|_|jSt��|
�
Srd)r�rz�superr)rrc�
�	__class__rrr�s





zDefaultContext.get_contextFcCs<|jdu
r|st
d
�
�
|
dur|rd|_dS|�|
�
|_dS)Nzcontext has already been set)r��RuntimeErrorrrirrrrk�s





zDefaultContext.set_start_methodcCs"|jdur
|
r	dS|j
|_|jjSrd)r�rzrerfrrrrh�s






zDefaultContext.get_start_methodc
Cs>tj
d
krdg
Stj
dkrddgnddg}
tjr|
�d�

|
S)NrLrR�darwin�forkr])rOrPr�HAVE_SEND_HANDLE�append)r�methodsrrr�get_all_start_methods
s






z$DefaultContext.get_all_start_methodsrdrr)
r	r
rryr�rrkrhr��
__classcell__rrr�rr��s


r�rLc@�eZ
dZd
Zedd��
ZdS)�ForkProcessr�c
C�d
dlm
}

|
|�
S�Nr)
�Popen)�
popen_forkr��r|r�rrrr{
r#zForkProcess._PopenN�r	r
rrrsr{rrrrr�
�


r�c@�(eZ
dZd
Zedd��
Zedd��
ZdS)�SpawnProcessrRc
Cr�r�)�popen_spawn_posixr�r�rrrr{
r#�SpawnProcess._Popenc
Crqrdrrrrrr}"
��SpawnProcess._after_forkNr~rrrrr�
�




r�c@r�)�ForkServerProcessr]c
Cr�r�)�popen_forkserverr�r�rrrr{)
r#zForkServerProcess._PopenNr�rrrrr�'
r�r�c
@�eZ
dZd
ZeZdS)�ForkContextr�N)r	r
rrer�ryrrrrr�.
�

r�c
@r���SpawnContextrRN�r	r
rrer�ryrrrrr�2
r�r�c@seZ
dZd
ZeZdd�ZdS)�ForkServerContextr]c

Cstj
std
�
�
dS)Nz%forkserver start method not available)rr�rarnrrrrb9
s

�z"ForkServerContext._check_availableN)r	r
rrer�ryrbrrrrr�6
s


r�)r�rRr]r�rRr�c@r�)r�rRc
Cr�r�)�popen_spawn_win32r�r�rrrr{M
r#r�c
Crqrdrrrrrr}R
r�r�Nr~rrrrr�K
r�c
@r�r�r�rrrrr�W
r�c

Cst|t
_dSrd)r_rzr�)
rcrrr�_force_start_methodd
s
r�cCstt
d
d�S)N�spawning_popen)rQ�_tlsrrrr�get_spawning_popenm
s
r�c

Cs
|t_
dSrd)r�r�)
�popenrrr�set_spawning_popenp
s

r�c

Cs t�durt
d
t|�
j�
�
dS)NzF%s objects should only be shared between processes through inheritance)r�r��typer	)
�objrrr�assert_spawnings
s




���r�) rrO�	threadingrWrr�__all__�	Exceptionrrrr�objectr�BaseProcessryr�rPr�r�r�r�r�r�r_rzr��localr�r�r�r�rrrr�<module>
sL


?

,

�
�