File: //usr/local/lib/python3.9/site-packages/billiard/__pycache__/context.cpython-39.pyc
a
X>hb3 � @ s� d dl Z d dlZd dlZd dlZddlmZ g ZdZddlm Z m
Z
m Z m Z m
Z
mZmZ G dd� d�ZG dd � d ej�ZG d
d � d e�Zed d
� ee�D ��e_ejdk�r2G dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� de�ZG dd� de�ZG dd� de�Z e� e� e � d�Z ee d �Z n8G d d� dej�ZG d d� de�Zd e� iZ ee d �Z d d!� Z e� � Z!d"d#� Z"d$d%� Z#d&d'� Z$dS )(� N� )�processzIforce_execv is not supported as the billiard C extension is not installed)� ProcessError�BufferTooShort� TimeoutError�AuthenticationError�TimeLimitExceeded�SoftTimeLimitExceeded�WorkerLostErrorc @ sX e Zd ZeZeZeZeZeZeZe Z e
e j �Z e
e j
�Z
eed�rLdd� Zndd� Zdd� ZdEd d
�Zd d � Zd
d� ZdFdd�ZdGdd�ZdHdd�Zdd� ZdIdd�ZdJd d �ZdKd d �Zd d!� Z dLd#d$�Z d%d&� Z d'd(� Z d)d*� Z d+d,� Z!d-d.� Z"d/d0� Z#dMd1d2�Z$d3d4� Z%d5d6� Z&d7d8� Z'dNd9d:�Z(dOd;d<�Z)dPd=d>�Z*d?d@� Z+dAdB� Z,dCdD� Z-dS )Q� BaseContext� cpu_countc C s" t �� }|du rtd��n|S dS )z(Returns the number of CPUs in the systemN� cannot determine number of cpus)�osr �NotImplementedError)�self�num� r �:/usr/local/lib/python3.9/site-packages/billiard/context.pyr 2 s
zBaseContext.cpu_countc
C s� t jdkr:zttjd �}W q� ttfy6 d}Y q�0 n�dt jv sNt jdkr�d}t jdkrdd| }z:t�|�� }t|�� �}W d � n1 s�0 Y W q� ty� d}Y q�0 n,zt� d�}W n tt
t fy� d}Y n0 |d kr�|S t d
��d S ) N�win32ZNUMBER_OF_PROCESSORSr Zbsd�darwinz/sbin/sysctl -n hw.ncpuz/usr�SC_NPROCESSORS_ONLNr r
)
�sys�platform�intr �environ�
ValueError�KeyError�popen�read�sysconf�OSError�AttributeErrorr )r r Zcomm�pr r r r : s*
.
c 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 �ManagerT s zBaseContext.ManagerTFc C s ddl m} ||||�S )z1Returns two connection object connected by a piper )�Pipe)�
connectionr* )r ZduplexZ rnonblockZ wnonblockr* 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, d 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. i s zBaseContext.RLockNc C s ddl m} ||| �� d�S )zReturns a condition objectr )� Conditionr$ )r- r/ r& )r �lockr/ r r r r/ n s zBaseContext.Conditionr c C s ddl m} ||| �� d�S )zReturns a semaphore objectr )� Semaphorer$ )r- r1 r& )r �valuer1 r r r r1 s s zBaseContext.Semaphorec C s ddl m} ||| �� d�S )z"Returns a bounded semaphore objectr )�BoundedSemaphorer$ )r- r3 r& )r r2 r3 r r r r3 x s z BaseContext.BoundedSemaphorec C s ddl m} || �� d�S )zReturns an event objectr )�Eventr$ )r- r4 r& )r r4 r r r r4 } s zBaseContext.Eventc C s ddl m} ||||| �� d�S )zReturns a barrier objectr )�Barrierr$ )r- r5 r& )r �parties�action�timeoutr5 r r r r5 � s zBaseContext.Barrierr c C s ddl m} ||| �� d�S )�Returns a queue objectr )�Queuer$ )�queuesr: r& )r �maxsizer: r r r r: � s zBaseContext.Queuec C s ddl m} ||| �� d�S )r9 r )�
JoinableQueuer$ )r; r= r& )r r< r= r r r r= � s zBaseContext.JoinableQueuec C s ddl m} || �� d�S )r9 r )� SimpleQueuer$ )r; r> r& )r r> r r r r> � s zBaseContext.SimpleQueuer c C s< ddl m} |||||||||| |
| | |
||||| �� d�S )z Returns a process pool objectr )�Pool)�context)�poolr? r& )r Z processesZ initializerZinitargsZmaxtasksperchildr8 Z soft_timeoutZlost_worker_timeoutZ max_restartsZmax_restart_freqZ
on_process_upZon_process_downZon_timeout_setZon_timeout_cancel�threadsZ semaphoreZputlocksZ
allow_restartr? r r r r? � s
�zBaseContext.Poolc G s ddl m} ||g|�R � S )zReturns a shared objectr )�RawValue)� sharedctypesrC )r �typecode_or_type�argsrC r r r rC � s zBaseContext.RawValuec C s ddl m} |||�S )zReturns a shared arrayr )�RawArray)rD rG )r rE �size_or_initializerrG r r r rG � s zBaseContext.RawArrayc O s4 ddl m} |�dd�}||g|�R || �� d��S )z$Returns a synchronized shared objectr )�Valuer0 T�r0 r% )rD rI �getr& )r rE rF �kwargsrI r0 r r r rI � s
�zBaseContext.Valuec O s, ddl m} |�dd�}||||| �� d�S )z#Returns a synchronized shared arrayr )�Arrayr0 TrJ )rD rM rK r& )r rE rH rF rL rM r0 r r r rM � 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.
r �frozenFr )�freeze_supportN)r r �getattr�spawnrO )r rO r r r rO � 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)�utilrR )r rR r r r rR � s zBaseContext.get_loggerc C s ddl m} ||�S )z8Turn on logging and add a handler which prints to stderrr )�
log_to_stderr)rS rT )r �levelrT r r r rT � 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)rQ rX )r �
executablerX r r r rX � 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)�
forkserverrZ )r Z module_namesrZ r r r rZ � s z"BaseContext.set_forkserver_preloadc C sD |d u r | S z t | }W n ty6 td| ��Y n0 |�� |S )Nzcannot find context for %r)�_concrete_contextsr r �_check_available)r �methodr% r r r r& � s zBaseContext.get_contextc 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)r �r r^ r r r �set_start_method� s z BaseContext.set_start_methodc C s | � � p
ddkS )N�fork)rc �r r r r �forking_is_enabled� s z BaseContext.forking_is_enabledc C s6 |s2ddl m} |r$| jddd� nt�tt�� d S )Nr )�
supports_execrQ T)�force)Z_extri re �warnings�warn�RuntimeWarning�
W_NO_EXECV)r r2 ri r r r �forking_enable� s
zBaseContext.forking_enablec C s d S r_ r rg r r r r] s z BaseContext._check_available)TFF)N)r )r )NN)r )r )NNr NNNNNr NNNNTNFF)N)N)F)N).�__name__�
__module__� __qualname__r r r r r r r
� staticmethodr �current_process�active_children�hasattrr r r) r* r, r. r/ r1 r3 r4 r5 r: r= r> r? rC rG rI rM rO rR rT rW rX rZ r& rc re rh ro r] r r r r r $ s^
�
r c @ s e Zd ZdZedd� �ZdS )�ProcessNc C s t �� j�| �S r_ )�_default_contextr& rw �_Popen)� process_objr r r ry s zProcess._Popen�rp rq rr Z
_start_methodrs ry r r r r rw
s rw c sF e Zd ZeZdd� Zd
� fdd� Zddd�Zdd d
�Zd d � Z� Z S )�DefaultContextc C s || _ d | _d S r_ )rx �_actual_context)r r@ r r r �__init__ s zDefaultContext.__init__Nc s4 |d u r | j d u r| j| _ | j S tt| ��|�S d S r_ )r} rx �superr| r& rd �� __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)r} � RuntimeErrorr& )r r^ rj r r r re $ s z DefaultContext.set_start_methodc C s" | j d u r|rd S | j| _ | j jS r_ )r} rx r` ra r r r rc , s
z DefaultContext.get_start_methodc C s6 t jdkrdgS ddlm} |jr*g d�S ddgS d S )Nr rQ r �� reduction�rf rQ r[ rf )r r rV r� �HAVE_SEND_HANDLE�r r� r r r �get_all_start_methods3 s
z$DefaultContext.get_all_start_methods)N)F)F)
rp rq rr rw r~ r& re rc r� �
__classcell__r r r� r r| s
r| c c s | ]}|d dkr|V qdS )r �_Nr )�.0�xr r r � <genexpr>= � r� r c @ s e Zd ZdZedd� �ZdS )� ForkProcessrf c C s ddl m} || �S �Nr )�Popen)Z
popen_forkr� �rz r� r r r ry H s zForkProcess._PopenNr{ r r r r r� E s r� c @ s e Zd ZdZedd� �ZdS )� SpawnProcessrQ c C s ddl m} || �S r� )Zpopen_spawn_posixr� r� r r r ry P s �SpawnProcess._PopenNr{ r r r r r� M s r� c @ s e Zd ZdZedd� �ZdS )�ForkServerProcessr[ c C s ddl m} || �S r� )Zpopen_forkserverr� r� r r r ry X s zForkServerProcess._PopenNr{ r r r r r� U s r� c @ s e Zd ZdZeZdS )� ForkContextrf N)rp rq rr r` r� rw r r r r r� ] s r� c @ s e Zd ZdZeZdS �� SpawnContextrQ N�rp rq rr r` r� rw r r r r r� a s r� c @ s e Zd ZdZeZdd� ZdS )�ForkServerContextr[ c C s ddl m} |jstd��d S )Nr r� z%forkserver start method not available)rV r� r� r r� r r r r] i s z"ForkServerContext._check_availableN)rp rq rr r` r� rw r] r r r r r� e s r� r� rf c @ s e Zd ZdZedd� �ZdS )r� rQ c C s ddl m} || �S r� )Zpopen_spawn_win32r� r� r r r ry z s r� Nr{ r r r r r� w s c @ s e Zd ZdZeZdS r� r� r r r r r� s rQ c C s t | t_d S r_ )r\ rx r} )r^ r r r �_force_start_method� s r� c C s t tdd �S )N�spawning_popen)rP �_tlsr r r r �get_spawning_popen� s r� c C s
| t _d S r_ )r� r� )r r r r �set_spawning_popen� 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� r� �typerp )�objr r r �assert_spawning� s
��r� )%r r � threadingrk rV r �__all__rn �
exceptionsr r r r r r r
r Z BaseProcessrw r| �list�dirr r� r� r� r� r� r� r\ rx r� �localr� r� r� r� r r r r �<module> sD
$ j(
��