U
if+ @ s d dl Z d dlZd dlZddlmZ ddlmZ dZG dd deZG dd d eZ G d
d d eZ
G d d
d
eZ G dd de Z
G dd dejZG dd de
ZejdkrRG dd dejZG dd dejZG dd dejZG dd d e
ZG d d d e
ZG d d d e
Ze e e d!Zejd"krDeed# Zn eed$ Zn8G d%d dejZG 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 }|dkrtdn|S dS )z(Returns the number of CPUs in the systemNz cannot determine number of cpus)os cpu_countNotImplementedError)selfZnumr r r r ) s
zBaseContext.cpu_countc C s& ddl m} || d}| |S )zReturns 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 ) SyncManagerctx)Zmanagersr get_contextstart)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} || dS )z#Returns a non-recursive lock objectr )Lockr ) synchronizer r )r r r r r r A s zBaseContext.Lockc C s ddl m} || dS )z Returns a recursive lock objectr )RLockr )r r r )r r r r r r F s zBaseContext.RLockNc C s ddl m} ||| dS )zReturns a condition objectr ) Conditionr )r r r )r lockr r r r r K s zBaseContext.Conditionr c C s ddl m} ||| dS )zReturns a semaphore objectr ) Semaphorer )r r" r )r valuer" r r r r" P s zBaseContext.Semaphorec C s ddl m} ||| dS )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} || dS )zReturns an event objectr )Eventr )r r% r )r r% r r r r% Z s zBaseContext.Eventc C s ddl m} ||||| dS )zReturns a barrier objectr )Barrierr )r r& r )r ZpartiesactionZtimeoutr& r r r r& _ s zBaseContext.Barrierr c C s ddl m} ||| dS )Returns a queue objectr )Queuer )queuesr) r )r maxsizer) r r r r) d s zBaseContext.Queuec C s ddl m} ||| dS )r( r )
JoinableQueuer )r* r, r )r r+ r, r r r r, i s zBaseContext.JoinableQueuec C s ddl m} || dS )r( r ) SimpleQueuer )r* r- r )r r- r r r r- n s zBaseContext.SimpleQueuer c C s" ddl m} |||||| dS )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} ||f| S )zReturns a shared objectr )RawValue) sharedctypesr0 )r typecode_or_typeargsr0 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} ||f||| dS )z$Returns a synchronized shared objectr )Valuer! r )r1 r6 r )r r2 r! r3 r6 r r r r6 s zBaseContext.Valuec C s ddl m} ||||| dS )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 ddr(ddlm} | dS )zCheck whether this is a fake forked process in a frozen executable.
If so then run code specified by commandline and exit.
win32frozenFr )freeze_supportN)sysplatformgetattrspawnr; )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 )zSets 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 sH |d kr | S z t | }W n" tk
r: td| d Y nX | |S )Nzcannot find context for %r)_concrete_contextsKeyError
ValueError_check_available)r methodr r r r r s zBaseContext.get_contextFc C s | j S N)_namer Z
allow_noner r r get_start_method s z BaseContext.get_start_methodc C s t dd 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 dS )z_Controls how objects will be reduced to a form that can be
shared with other processes.r )globalsgetr 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._Popenr r r
Z
_start_methodrY r^ r r r r r\ s r\ c sF e Zd ZeZdd Zd
fdd ZdddZdd 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 kr | j d kr| 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 k r|std|d kr,|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 kr|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? darwinforkrI )r< r= r HAVE_SEND_HANDLEappend)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? Nr r r
rP rs r\ r r r r rx ) s rx c @ s e Zd ZdZeZdd ZdS )ForkServerContextrI c C s t jstdd 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 )Nspawning_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 kr 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 sL ?,