Format: application/x-python-code
1 o
2
3 ;'h{% � @ s� U d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
4 m Z m Z d dl
5 mZmZ ddlmZmZmZ dd lmZ dd
6 lmZ dd lmZ dd lmZm Z dd
7 l m Z m Z m Z ddl!m"Z"m#Z# ddl$m%Z% ej&dkr}d dl
8 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�
9 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
10 t�� �t�� }W n tyE t� �t � }t��� t�|� t
11 � � �� Y nw |pJt � 4 I dH ��O |r�|�
12 � \�}�jdu r�tt�j��ttt�j��� t� }g } |r�||d
13 d tk r~n|�� \}
14 }|
15 �� ��|
16 � | �|
17 � |sstd d
18 �� | 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 }
19 W d � n1 s�w Y |
20 dk�rt%d|
21 ����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
22 ��: 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�
23 Call the given function with the given arguments in a worker process.
24
25 If the ``cancellable`` option is enabled and the task waiting for its completion is
26 cancelled, the worker process running it will be abruptly terminated using SIGKILL
27 (or ``terminateProcess()`` on Windows).
28
29 :param func: a callable
30 :param args: positional arguments for the callable
31 :param cancellable: ``True`` to allow cancellation of the operation while it's
32 running
33 :param limiter: capacity limiter to use to limit the total amount of processes
34 running (if omitted, the default limiter is used)
35 :return: an awaitable that yields the return value of the function.
36
37 � 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�� ��
38 � I d H W d � n1 sYw Y W n t yh Y nw t |t
39 � �rp� t|�d }~ww t�|�}|dkr�t |t�s�J �|�|S ) N�
40 �2 � )� RETURN� EXCEPTION�-Worker process returned unexpected response: T��shieldr1 )�sendZ
41 receive_until�split� RuntimeErrorZreceive_exactly�int�
42 BaseException�discard�killr �aclose�ProcessLookupError�
43 isinstancer r
44 �pickle�loads)r* �response�status�lengthZpickled_response�exc�retval��buffered�process�stdin�workers� �FC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/to_process.py�send_raw_command= s> ��
45 �� � ��
46 z"run_sync.<locals>.send_raw_commandN�run)�protocolr r Tr3 z-uz-m)rI �stdout� � � READY
47 r2 �__main__�__file__�initz*Error during worker process initialization)r* r+ r) r, )-r r? �dumps�HIGHEST_PROTOCOLr �getr � LookupError�setr r Z#setup_process_pool_exit_at_shutdown� current_default_process_limiter�pop�
48 returncoder r rI r r rP r
49 �WORKER_MAX_IDLE_TIME�popleftr; �remove�appendr r< �sys�
50 executable�__name__r �
51 subprocess�PIPEr Zreceiver
52 �getattr�modules�pathr r9 �addr )r# r! r" r% rM �requestZ idle_workersZ
53 idle_since�nowZkilled_processesZprocess_to_killZkilled_process�command�message�main_module_path�pickledrD rK rF rL �run_sync& s� � !
54
55 �
56
57 �
58
59 � � �
60 �
61 �
62 �
63 �
64 �
65 �����
66
67 ��
68 F��0�rr r c C s< zt �� W S ty tt�� pd�} t �| � | Y S w )z�
69 Return the capacity limiter that is used by default to limit the number of worker
70 processes.
71
72 :return: a capacity limiter object
73
74 � )r rY rZ r �os� cpu_countr[ )r" rK rK rL r\ � s
75
76 �r\ �Nonec
77
78 C s� t j} t j}ttj�t _ttjd�t _|j�d� d }}z
79 t� | j�^}}W n t
80 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
81 d= |r�tj �|�r�ztd|�} | r�| jr�t| �}
82 | j�|
83 � |
84 t j
85 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
86 | t| �f � |j�| � t|t�r�|�q) N�wrS TrN rV rT � __mp_main__r1 r0 s %s %d
87 )rc rI rP �openrt �devnull�buffer�writer? �load�EOFErrorr9 rj ri �isfiler �loaderr � exec_modulerW rX �lenr> �
88 SystemExit)
89 rI rP rE � exceptionrn r% rD r# rp �spec�mainrB rq rK rK rL �process_worker� sp �� ��
90
91
92
93 ������
94 �r� rT )
95 r# r$ r% r& r! r' r" r( r) r )r) r )r) rv )5�
96 __future__r rt r? rf rc � collectionsr �collections.abcr �importlib.utilr r �typingr r Z_core._eventloopr
97 r r Z_core._exceptionsr
98 Z_core._subprocessesr Z_core._synchronizationr Z _core._tasksr r �abcr r r Zlowlevelr r Zstreams.bufferedr � version_infor r Ztyping_extensionsr_ r r r �__annotations__r r rr r\ r� re rK rK rK rL �<module> sJ
99 ��
100
101 ;
102 �