a
XC?h�. � @ s� g d �Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z ddl Z ddl
mZ ddl
mZ ejjZddlmZmZmZmZmZ G dd � d e�Ze� ZG d
d � d e�Z G d d
� d
e�Z dS ))�Queue� SimpleQueue�
JoinableQueue� N)�Empty�Full� )�
connection)�context)�debug�info�Finalize�register_after_fork�
is_exitingc @ s� e Zd Zd-dd�Zdd� Zdd� Zdd � Zd.d d �Zd/dd�Zd0dd�Z dd� Z
dd� Z dd� Z dd� Z
dd � Zd d � Zd d � Zd!d"� Zd#d$� Zed%d&� �Zed'd(� �Zed)d*� �Zed+d,� �ZdS )1r r c C s� |dkrddl m} || _tjdd�\| _| _|�� | _t �
� | _ t j
dkrTd | _n
|�� | _|�|�| _d| _| �� t j
dkr�t| tj� d S )Nr r )�
SEM_VALUE_MAXF�Zduplex�win32)Z synchronizer �_maxsizer �Pipe�_reader�_writer�Lock�_rlock�os�getpid�_opid�sys�platform�_wlockZBoundedSemaphore�_sem�
_ignore_epipe�_resetr
r � _after_fork��self�maxsize�ctx� r& �;/opt/alt/python39/lib64/python3.9/multiprocessing/queues.py�__init__% s
zQueue.__init__c C s. t �| � | j| j| j| j| j| j| j| j fS �N)
r �assert_spawningr r r r r r r r �r# r& r& r' � __getstate__9 s
�zQueue.__getstate__c C s0 |\| _ | _| _| _| _| _| _| _| �� d S r) ) r r r r r r r r r �r# �stater& r& r' � __setstate__> s �zQueue.__setstate__c C s t d� | jdd� d S )NzQueue._after_fork()T)�
after_fork)r
r r+ r& r& r' r! C s zQueue._after_forkFc C sj |r| j �� nt�t�� �| _ t�� | _d | _d | _ d| _
d| _ d | _ | j
j| _| jj| _| jj| _d S �NF)� _notempty�_at_fork_reinit� threading� Conditionr � collections�deque�_buffer�_thread� _jointhread�_joincancelled�_closed�_closer �
send_bytes� _send_bytesr �
recv_bytes� _recv_bytes�poll�_poll)r# r0 r& r& r' r G s
z Queue._resetTNc C sz | j rtd| �d���| j�||�s(t�| j�8 | jd u rB| �� | j� |� | j�
� W d � n1 sl0 Y d S �NzQueue z
is closed) r< �
ValueErrorr �acquirer r2 r9 �
_start_threadr8 �append�notify�r# �obj�block�timeoutr& r& r' �putV s
z Queue.putc C s� | j rtd| �d���|r\|d u r\| j� | �� }W d � n1 sF0 Y | j�� n||rlt�� | }| j�||�s~t �zL|r�|t�� }| �
|�s�t �n | �
� s�t �| �� }| j�� W | j�� n | j�� 0 t � |�S rD )
r< rE r rA r �release�time� monotonicrF r rC �_ForkingPickler�loads)r# rL rM �resZdeadliner& r& r' �getb s* &
z Queue.getc C s | j | jj�� S r) )r r �_semlockZ
_get_valuer+ r& r& r' �qsize| s z Queue.qsizec C s
| � � S r) �rC r+ r& r& r' �empty� s z Queue.emptyc C s | j j�� S r) )r rV �_is_zeror+ r& r& r' �full� s z
Queue.fullc C s
| � d�S r1 )rU r+ r& r& r' �
get_nowait� s zQueue.get_nowaitc C s | � |d�S r1 )rN �r# rK r& r& r' �
put_nowait� s zQueue.put_nowaitc C s d| _ | j}|r d | _|� d S )NT)r< r= )r# �closer& r& r' r_ � s
z Queue.closec C s t d� | jr| �� d S )NzQueue.join_thread())r
r: r+ r& r& r' � join_thread� s zQueue.join_threadc C s4 t d� d| _z| j�� W n ty. Y n0 d S )NzQueue.cancel_join_thread()T)r
r; r: Zcancel�AttributeErrorr+ r& r& r' �cancel_join_thread� s zQueue.cancel_join_threadc C s� t d� | j�� tjtj| j| j| j| j | j
j | j j | j
| j| jf dd�| _d| j_t d� | j�� t d� | js�t| jtjt�| j�gdd�| _t| tj| j| jgd d�| _d S )
NzQueue._start_thread()ZQueueFeederThread)�target�args�nameTzdoing self._thread.start()z ... done self._thread.start()�����)Z exitpriority�
)r
r8 �clearr4 ZThreadr �_feedr2 r? r r r_ r r �_on_queue_feeder_errorr r9 Zdaemon�startr; r �_finalize_join�weakref�refr: �_finalize_closer= r+ r& r& r' rG � s4
��
�
�zQueue._start_threadc C s4 t d� | � }|d ur(|�� t d� nt d� d S )Nzjoining queue threadz... queue thread joinedz ... queue thread already dead)r
�join)Ztwr�threadr& r& r' rl � s
zQueue._finalize_joinc C sB t d� |�" | �t� |�� W d � n1 s40 Y d S )Nz telling queue thread to quit)r
rH � _sentinelrI )�buffer�notemptyr& r& r' ro � s
zQueue._finalize_closec
C sr t d� |j} |j}
|j} | j} t}
tjdkr<|j}|j}nd }z�| � z| sT| � W |
� n|
� 0 zn| � }||
u r�t d� |� |� W W d S t� |�}|d u r�||� qh|� z||� W |� qh|� 0 qhW n t
y� Y n0 W q@ t �yj } zf|�r$t |dd�t
jk�r$W Y d }~d S t� �rDtd|� W Y d }~d S |�� |||� W Y d }~q@d }~0 0 q@d S )Nz$starting thread to feed data to piper z%feeder thread got sentinel -- exiting�errnor zerror in queue thread: %s)r
rF rO �wait�popleftrr r r rR �dumps�
IndexError� Exception�getattrru ZEPIPEr r )rs rt r> Z writelockZ reader_closeZ writer_closeZ ignore_epipe�onerrorZ queue_semZnacquireZnreleaseZnwaitZbpopleft�sentinelZwacquireZwreleaserK �er& r& r' ri � sP
z Queue._feedc C s ddl }|�� dS )z�
Private API hook called when feeding data in the background thread
raises an exception. For overriding by concurrent.futures.
r N)� traceback� print_exc)r~ rK r r& r& r' rj s z Queue._on_queue_feeder_error)r )F)TN)TN)�__name__�
__module__� __qualname__r( r, r/ r! r rN rU rW rY r[ r\ r^ r_ r` rb rG � staticmethodrl ro ri rj r& r& r& r' r # s0
!
>r c @ s@ e Zd Zddd�Zdd� Zdd� Zdd
d �Zd d
� Zdd� Zd S )r r c C s* t j| ||d� |�d�| _|�� | _d S )N)r% r )r r( Z Semaphore�_unfinished_tasksr5 �_condr" r& r& r' r( ( s zJoinableQueue.__init__c C s t �| �| j| jf S r) )r r, r� r� r+ r& r& r' r, - s zJoinableQueue.__getstate__c C s, t �| |d d� � |dd � \| _| _d S )N�����)r r/ r� r� r- r& r& r' r/ 0 s zJoinableQueue.__setstate__TNc C s� | j rtd| �d���| j�||�s(t�| j�h | j�B | jd u rJ| �� | j �
|� | j � � | j�
� W d � n1 s~0 Y W d � n1 s�0 Y d S rD )r< rE r rF r r2 r� r9 rG r8 rH r� rO rI rJ r& r& r' rN 4 s
zJoinableQueue.putc C sT | j �: | j�d�s td��| jj�� r2| j �� W d � n1 sF0 Y d S )NFz!task_done() called too many times)r� r� rF rE rV rZ Z
notify_allr+ r& r& r' � task_doneA s
zJoinableQueue.task_donec C s@ | j �& | jj�� s | j �� W d � n1 s20 Y d S r) )r� r� rV rZ rv r+ r& r& r' rp H s zJoinableQueue.join)r )TN) r� r� r� r( r, r/ rN r� rp r& r& r& r' r & s
r c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
e j �Z
dS )r c C sH t jdd�\| _| _|�� | _| jj| _tj dkr:d | _
n
|�� | _
d S )NFr r ) r r r r r r rB rC r r r )r# r% r& r& r' r( S s
zSimpleQueue.__init__c C s | j �� | j�� d S r) )r r_ r r+ r& r& r' r_ \ s
zSimpleQueue.closec C s
| � � S r) rX r+ r& r& r' rY ` s zSimpleQueue.emptyc C s t �| � | j| j| j| jfS r) )r r* r r r r r+ r& r& r' r, c s
zSimpleQueue.__getstate__c C s" |\| _ | _| _| _| j j| _d S r) )r r r r rB rC r- r& r& r' r/ g s zSimpleQueue.__setstate__c C s: | j � | j�� }W d � n1 s&0 Y t�|�S r) )r r r@ rR rS )r# rT r& r& r' rU k s (zSimpleQueue.getc C sX t �|�}| jd u r"| j�|� n2| j� | j�|� W d � n1 sJ0 Y d S r) )rR rx r r r> r] r& r& r' rN q s
zSimpleQueue.putN)r� r� r� r( r_ rY r, r/ rU rN � classmethod�types� GenericAlias�__class_getitem__r& r& r& r' r Q s
r ) �__all__r r r4 r6 rP r� rm ru Zqueuer r Z_multiprocessing� r r Z reductionZForkingPicklerrR �utilr
r r r
r �objectr rr r r r& r& r& r' �
s& z
+