î
e f) ã @ s[ d d l Z d d l Z d d l Z d d l m Z g Z Gd d „ d e ƒ Z Gd d „ d e ƒ Z Gd d „ d e ƒ Z Gd
d „ d e ƒ Z
Gd d
„ d
e ƒ Z Gd d „ d e j
ƒ Z Gd d „ d e ƒ Z e d d „ e e ƒ Dƒ ƒ e _ e j d k rÌGd d „ d e j
ƒ Z Gd d „ d e j
ƒ Z Gd d „ d e j
ƒ Z Gd d „ d e ƒ Z Gd d „ d e ƒ Z Gd d „ d e ƒ Z i e ƒ d! 6e ƒ d" 6e ƒ d# 6Z e e d! ƒ Z nO Gd$ d „ d e j
ƒ Z Gd% d „ d e ƒ Z i e ƒ d" 6Z e e d" ƒ Z d& d' „ Z e j ƒ Z d( d) „ Z d* d+ „ Z d, d- „ Z d S).é Né )Úprocessc @ s e Z d Z d S)Ú ProcessErrorN)Ú__name__Ú
__module__Ú __qualname__© r r ú „ Z& d S)?Ú BaseContextc C s/ t j ƒ } | d k r' t d ƒ ‚ n | Sd S)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 s3 d d l m } | d | j ƒ ƒ } | j ƒ | 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 ÚManager/ s
zBaseContext.ManagerTc C s d d l m } | | ƒ S)z1Returns two connection object connected by a piper )ÚPipe)Ú
connectionr )r Zduplexr r r r r : s zBaseContext.Pipec C s# d d l m } | d | j ƒ ƒ S)z#Returns a non-recursive lock objectr )ÚLockr )Ú synchronizer r )r r r r r r ? s zBaseContext.Lockc C s# d d l m } | d | j ƒ ƒ S)z Returns a recursive lock objectr )ÚRLockr )r r r )r r r r r r D s zBaseContext.RLockNc C s& d d l m } | | d | j ƒ ƒS)zReturns a condition objectr )Ú Conditionr )r r r )r Úlockr r r r r I s zBaseContext.Conditionr c C s& d d l m } | | d | j ƒ ƒS)zReturns a semaphore objectr )Ú Semaphorer )r r r )r Úvaluer r r r r N s zBaseContext.Semaphorec C s& d d l m } | | d | j ƒ ƒS)z"Returns a bounded semaphore objectr )ÚBoundedSemaphorer )r r! r )r r r! r r r r! S s z BaseContext.BoundedSemaphorec C s# d d l m } | d | j ƒ ƒ S)zReturns an event objectr )ÚEventr )r r" r )r r" r r r r" X s zBaseContext.Eventc C s, d d l m } | | | | d | j ƒ ƒS)zReturns a barrier objectr )ÚBarrierr )r r# r )r ZpartiesÚactionZtimeoutr# r r r r# ] s zBaseContext.Barrierr c C s& d d l m } | | d | j ƒ ƒS)zReturns a queue objectr )ÚQueuer )Úqueuesr% r )r Úmaxsizer% r r r r% b s zBaseContext.Queuec C s& d d l m } | | d | j ƒ ƒS)zReturns a queue objectr )Ú
JoinableQueuer )r& r( r )r r' r( r r r r( g s zBaseContext.JoinableQueuec C s# d d l m } | d | j ƒ ƒ S)zReturns a queue objectr )Ú SimpleQueuer )r& r) r )r r) r r r r) l s zBaseContext.SimpleQueuec C s/ d d l m } | | | | | d | j ƒ ƒS)z Returns a process pool objectr )ÚPoolÚcontext)Zpoolr* r )r Z processesZ initializerZinitargsZmaxtasksperchildr* r r r r* q s zBaseContext.Poolc G s d d l m } | | | Œ S)zReturns a shared objectr )ÚRawValue)Ú sharedctypesr, )r Útypecode_or_typeÚargsr, r r r r, x s zBaseContext.RawValuec C s d d l m } | | | ƒ S)zReturns a shared arrayr )ÚRawArray)r- r0 )r r. Úsize_or_initializerr0 r r r r0 } s zBaseContext.RawArrayr c G s/ d d l m } | | d | d | j ƒ | ŒS)z$Returns a synchronized shared objectr )ÚValuer r )r- r2 r )r r. r r/ r2 r r r r2 ‚ s zBaseContext.Valuec C s/ d d l m } | | | d | d | j ƒ ƒS)z#Returns a synchronized shared arrayr )ÚArrayr r )r- r3 r )r r. r1 r r3 r r r r3 ˆ s zBaseContext.Arrayc C s? t j d k r; t t d d ƒ r; d d l m } | ƒ n d S)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Úspawnr6 )r r6 r r r r6 Ž s !zBaseContext.freeze_supportc C s d d l 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 d d l m } | | ƒ S)z8Turn on logging and add a handler which prints to stderrr )Ú
log_to_stderr)r< r= )r Úlevelr= r r r r= s zBaseContext.log_to_stderrc C s d d l m } d S)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 d d l m } | | ƒ d S)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: rA )r Ú
executablerA r r r rA ª s zBaseContext.set_executablec C s d d l m } | | ƒ d S)zkSet list of module names to try to load in forkserver process.
This is really just a hint.
r )Úset_forkserver_preloadN)Ú
forkserverrC )r Z module_namesrC r r r rC ² s z"BaseContext.set_forkserver_preloadc C sQ | d k r | Sy t | } Wn" t k
rB t d | ƒ ‚ Yn X| j ƒ | 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 Ú
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)rG )r rI r r r Úset_start_methodÆ s z BaseContext.set_start_methodc C s d S)Nr )r r r r rH É s z BaseContext._check_available)'r r r r r
r r Ú staticmethodr Zcurrent_processZactive_childrenr r r r r r r r! r" r# r% r( r) r* r, r0 r2 r3 r6 r; r= r@ rA rC r rL rM rH r r r r r
sF
r
c @ s( e Z d Z d Z e d d „ ƒ Z d S)ÚProcessNc C s t j ƒ j j | ƒ S)N)Ú_default_contextr rO Ú_Popen)Ú process_objr r r rQ Ò s zProcess._Popen)r r r Ú
_start_methodrN rQ r r r r rO Ð s rO c sa e Z d Z e Z d d „ Z d ‡ f d d † Z d d d „ Z d d d
„ Z d d „ Z ‡ S)
ÚDefaultContextc C s | | _ d | _ d S)N)rP Ú_actual_context)r r+ r r r Ú__init__Ù s zDefaultContext.__init__Nc sE | d k r1 | j d k r* | j | _ n | j St ƒ j | ƒ Sd S)N)rU rP Úsuperr )r rI )Ú __class__r r r Ý s
zDefaultContext.get_contextFc C sZ | j d k r% | r% t d ƒ ‚ n | d k rD | rD d | _ d S| j | ƒ | _ d S)Nz context has already been set)rU Ú RuntimeErrorr )r rI Zforcer r r rM å s z DefaultContext.set_start_methodc C s2 | j d k r( | r d S| j | _ n | j j S)N)rU rP rJ )r rK r r r rL í s
z DefaultContext.get_start_methodc C sJ t j d k r d g Sd d l m } | j r< d d d g Sd d g Sd S)Nr4 r: r )Ú reductionÚforkrD )r7 r8 r? rZ ÚHAVE_SEND_HANDLE)r rZ r r r Úget_all_start_methodsô s
z$DefaultContext.get_all_start_methods) r r r rO rV r rM rL r] r r )rX r rT Ö s rT c c s% | ] } | d d k r | Vq d S)r Ú_Nr )Ú.0Úxr r r ú þ s ra r4 c @ s( e Z d Z d Z e d d „ ƒ Z d S)Ú ForkProcessr[ c C s d d l m } | | ƒ S)Nr )ÚPopen)Z
popen_forkrc )rR rc r r r rQ s zForkProcess._PopenN)r r r rS rN rQ r r r r rb s rb c @ s( e Z d Z d Z e d d „ ƒ Z d S)Ú SpawnProcessr: c C s d d l m } | | ƒ S)Nr )rc )Zpopen_spawn_posixrc )rR rc r r r rQ s zSpawnProcess._PopenN)r r r rS rN rQ r r r r rd
s rd c @ s( e Z d Z d Z e d d „ ƒ Z d S)ÚForkServerProcessrD c C s d d l m } | | ƒ S)Nr )rc )Zpopen_forkserverrc )rR rc r r r rQ s zForkServerProcess._PopenN)r r r rS rN rQ r r r r re s re c @ s e Z d Z d Z e Z d S)Ú ForkContextr[ N)r r r rJ rb rO r r r r rf s rf c @ s e Z d Z d Z e Z d S)Ú SpawnContextr: N)r r r rJ rd rO r r r r rg s rg c @ s( e Z d Z d Z e Z d d „ Z d S)ÚForkServerContextrD c C s, d d l m } | j s( t d ƒ ‚ n d S)Nr )rZ z%forkserver start method not available)r? rZ r\ rG )r rZ r r r rH & s z"ForkServerContext._check_availableN)r r r rJ re rO rH r r r r rh # s rh r[ r: rD c @ s( e Z d Z d Z e d d „ ƒ Z d S)rd r: c C s d d l m } | | ƒ S)Nr )rc )Zpopen_spawn_win32rc )rR rc r r r rQ 6 s zSpawnProcess._PopenN)r r r rS rN rQ r r r r rd 4 s c @ s e Z d Z d Z e Z d S)rg r: N)r r r rJ rd rO r r r r rg ; s c C s t | t _ d S)N)rE rP rU )rI r r r Ú_force_start_methodH s ri c C s t t d d ƒ S)NÚspawning_popen)r9 Ú_tlsr r r r Úget_spawning_popenQ s rl c C s
| t _ d S)N)rk rj )Úpopenr r r Úset_spawning_popenT s rn c C s/ t ƒ d k r+ t d t | ƒ j ƒ ‚ n d S)NzF%s objects should only be shared between processes through inheritance)rl rY Útyper )Úobjr r r Úassert_spawningW s rq )!r r7 Z threadingr? r Ú__all__Ú Exceptionr r
r r Úobjectr
Z BaseProcessrO rT ÚlistÚdirr8 rb rd re rf rg rh rE rP ri Zlocalrk rl rn rq r r r r Ú sD ³("