File: //usr/local/lib/python3.10/dist-packages/anyio/__pycache__/to_process.cpython-310.pyc
o
���g{% � @ s� U d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z m Z d dl
mZmZ ddlmZmZmZ dd lmZ dd
lmZ dd lmZ dd lmZm Z dd
l m Z m Z m Z ddl!m"Z"m#Z# ddl$m%Z% ej&dkr}d dl
m'Z'm(Z( nd dl)m'Z'm(Z( dZ*ed�Z+e'd�Z,e"d�Z-de.d< e"d�Z/de.d< e"d�Z0de.d< ddd �d/d&d'�Z1d0d)d*�Z2d1d,d-�Z3e4d.kr�e3� dS dS )2� )� annotationsN)�deque)�Callable)�module_from_spec�spec_from_file_location)�TypeVar�cast� )� current_time�get_async_backend�get_cancelled_exc_class)�BrokenWorkerProcess)� open_process)�CapacityLimiter)� CancelScope�
fail_after)�ByteReceiveStream�ByteSendStream�Process)�RunVar�checkpoint_if_cancelled)�BufferedByteReceiveStream)� � )� TypeVarTuple�Unpacki, �T_Retval�PosArgsT�_process_pool_workerszRunVar[set[Process]]�_process_pool_idle_workersz$RunVar[deque[tuple[Process, float]]]�_default_process_limiterzRunVar[CapacityLimiter]F)� cancellable�limiter�func�&Callable[[Unpack[PosArgsT]], T_Retval]�args�Unpack[PosArgsT]r! �boolr" �CapacityLimiter | None�returnc � sZ �d� ���fdd� }t � I dH tjd| |ftjd �}z
t�� �t�� }W n tyE t� �t � }t��� t�|� t
� � �� Y nw |pJt � 4 I dH ��O |r�|�
� \�}�jdu r�tt�j��ttt�j��� t� }g } |r�||d
d tk r~n|�� \}
}|
�� ��|
� | �|
� |sstd d
�� | D ] } | � � I dH q�W d � n1 s�w Y n����� |sSt j ddt g} t | t!j"t!j"d�I dH �zTtt�j��ttt�j��� t#d�� � �$d�I dH }
W d � n1 s�w Y |
dk�rt%d|
����t&t j'd dd�}tjdt j(|ftjd �}||�I dH W n! t%t)� f�y0 � t*�yE } z ��� t%d�|�d}~ww ��+�� t| d
��: z)tt,||�I dH �W ��v �rj|��t� f� W d � W d �I dH S ��v �r�|��t� f� w w 1 �s�w Y W d �I dH dS 1 I dH �s�w Y dS )a�
Call the given function with the given arguments in a worker process.
If the ``cancellable`` option is enabled and the task waiting for its completion is
cancelled, the worker process running it will be abruptly terminated using SIGKILL
(or ``terminateProcess()`` on Windows).
:param func: a callable
:param args: positional arguments for the callable
:param cancellable: ``True`` to allow cancellation of the operation while it's
running
:param limiter: capacity limiter to use to limit the total amount of processes
running (if omitted, the default limiter is used)
:return: an awaitable that yields the return value of the function.
� pickled_cmd�bytesr) �objectc � s �z/�� | �I d H � �dd�I d H }|�d�\}}|dvr%td|����� �t|��I d H }W nG tyw } z;���� z"��� t dd�� ��
� I d H W d � n1 sYw Y W n t yh Y nw t |t
� �rp� t|�d }~ww t�|�}|dkr�t |t�s�J �|�|S ) N�
�2 � )� RETURN� EXCEPTION�-Worker process returned unexpected response: T��shieldr1 )�send�
receive_until�split� RuntimeError�receive_exactly�int�
BaseException�discard�killr �aclose�ProcessLookupError�
isinstancer r
�pickle�loads)r* �response�status�length�pickled_response�exc�retval��buffered�process�stdin�workers� �;/usr/local/lib/python3.10/dist-packages/anyio/to_process.py�send_raw_command= s> ��
�� � ��
z"run_sync.<locals>.send_raw_commandN�run)�protocolr r Tr3 z-uz-m)rL �stdout� � � READY
r2 �__main__�__file__�initz*Error during worker process initialization)r* r+ r) r, )-r rA �dumps�HIGHEST_PROTOCOLr �getr � LookupError�setr r �#setup_process_pool_exit_at_shutdown� current_default_process_limiter�pop�
returncoder r rL r r rS r
�WORKER_MAX_IDLE_TIME�popleftr= �remove�appendr r> �sys�
executable�__name__r �
subprocess�PIPEr �receiver
�getattr�modules�pathr r; �addr )r# r! r" r% rP �request� idle_workers�
idle_since�now�killed_processes�process_to_kill�killed_process�command�message�main_module_path�pickledrG rN rI rO �run_sync& s� � !
�
�
� � �
�
�
�
�
�
�����
��
F��0�r| r c C s< zt �� W S ty tt�� pd�} t �| � | Y S w )z�
Return the capacity limiter that is used by default to limit the number of worker
processes.
:return: a capacity limiter object
� )r r\ r] r �os� cpu_countr^ )r" rN rN rO r` � s
�r` �Nonec
C s� t j} t j}ttj�t _ttjd�t _|j�d� d }}z
t� | j�^}}W n t
y2 Y d S t yE } z|}W Y d }~nmd }~ww |dkri|\}}z||� }W nY t yh } z|}W Y d }~nJd }~ww |dkr�|\t _ }t j
d= |r�tj �|�r�ztd|�} | r�| jr�t| �}
| j�|
� |
t j
d< W n t y� } z|}W Y d }~nd }~ww z|d ur�d} t�|tj�} n d } t�|tj�} W n t y� } z|}d} t�|tj�} W Y d }~nd }~ww |j�d
| t| �f � |j�| � t|t�r�|�q) N�wrV TrQ rY rW � __mp_main__r1 r0 s %s %d
)rg rL rS �openr~ �devnull�buffer�writerA �load�EOFErrorr; ro rn �isfiler �loaderr � exec_modulerZ r[ �lenr@ �
SystemExit)
rL rS rH � exceptionrx r% rG r# rz �spec�mainrD r{ rN rN rO �process_worker� sp �� ��
������
�r� rW )
r# r$ r% r& r! r' r" r( r) r )r) r )r) r� )5�
__future__r r~ rA rj rg � collectionsr �collections.abcr �importlib.utilr r �typingr r �_core._eventloopr
r r �_core._exceptionsr
�_core._subprocessesr �_core._synchronizationr � _core._tasksr r �abcr r r �lowlevelr r �streams.bufferedr � version_infor r �typing_extensionsrc r r r �__annotations__r r r| r` r� ri rN rN rN rO �<module> sJ
��
;
�