o
    f.                     @   s   g d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl	m
Z
mZ ddlZddlmZ ddlmZ ejjZddlmZmZmZmZmZ G dd	 d	eZe ZG d
d deZG dd deZdS ))QueueSimpleQueueJoinableQueue    N)EmptyFull   )
connection)context)debuginfoFinalizeregister_after_fork
is_exitingc                   @   s   e Zd Zd-ddZdd Zdd Zdd	 Zd.ddZd/ddZd/ddZ	dd Z
dd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zed%d& Zed'd( Zed)d* Zed+d, ZdS )0r   r   c                C   s   |dkr
ddl m} || _tjdd\| _| _| | _t	
 | _tjdkr*d | _n| | _||| _d| _|   tjdkrIt| tj d S d S )Nr   r   )SEM_VALUE_MAXFduplexwin32)synchronizer   _maxsizer   Pipe_reader_writerLock_rlockosgetpid_opidsysplatform_wlockBoundedSemaphore_sem_ignore_epipe_resetr   r   _after_forkselfmaxsizectx r)   -/usr/lib/python3.10/multiprocessing/queues.py__init__%   s   




zQueue.__init__c                 C   s.   t |  | j| j| j| j| j| j| j| j	fS N)
r	   assert_spawningr"   r   r   r   r   r   r!   r   r&   r)   r)   r*   __getstate__9   s   
zQueue.__getstate__c              	   C   s0   |\| _ | _| _| _| _| _| _| _|   d S r,   )	r"   r   r   r   r   r   r!   r   r#   r&   stater)   r)   r*   __setstate__>   s   zQueue.__setstate__c                 C   s   t d | jdd d S )NzQueue._after_fork()T)
after_fork)r
   r#   r.   r)   r)   r*   r$   C   s   zQueue._after_forkFc                 C   sj   |r| j   ntt | _ t | _d | _d | _	d| _
d| _d | _| jj| _| jj| _| jj| _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&   r3   r)   r)   r*   r#   G   s   


zQueue._resetTNc                 C   s|   | j rtd| d| j||st| j | jd u r!|   | j	| | j
  W d    d S 1 s7w   Y  d S NzQueue z
 is closed)r?   
ValueErrorr!   acquirer   r5   r<   _start_threadr;   appendnotifyr&   objblocktimeoutr)   r)   r*   putV   s   
"z	Queue.putc                 C   s   | j rtd| d|r.|d u r.| j |  }W d    n1 s#w   Y  | j  n>|r6t | }| j||s?t	z&|rP|t  }| 
|sOt	n| 
 sVt	|  }| j  W | j  n| j  w t|S rG   )r?   rH   r   rD   r!   releasetime	monotonicrI   r   rF   _ForkingPicklerloads)r&   rO   rP   resdeadliner)   r)   r*   getb   s.   


z	Queue.getc                 C   s   | j | jj  S r,   )r   r!   _semlock
_get_valuer.   r)   r)   r*   qsize|   s   zQueue.qsizec                 C   
   |    S r,   rF   r.   r)   r)   r*   empty      
zQueue.emptyc                 C   s   | j j S r,   )r!   rZ   _is_zeror.   r)   r)   r*   full      z
Queue.fullc                 C   s
   |  dS r4   )rY   r.   r)   r)   r*   
get_nowait   r`   zQueue.get_nowaitc                 C   s   |  |dS r4   )rQ   r&   rN   r)   r)   r*   
put_nowait   rc   zQueue.put_nowaitc                 C   s$   d| _ | j}|rd | _|  d S d S )NT)r?   r@   )r&   closer)   r)   r*   rg      s   
zQueue.closec                 C   s2   t d | jsJ d| | jr|   d S d S )NzQueue.join_thread()zQueue {0!r} not closed)r
   r?   formatr=   r.   r)   r)   r*   join_thread   s
   zQueue.join_threadc                 C   s4   t d d| _z| j  W d S  ty   Y d S w )NzQueue.cancel_join_thread()T)r
   r>   r=   cancelAttributeErrorr.   r)   r)   r*   cancel_join_thread   s   zQueue.cancel_join_threadc                 C   s   t d | j  tjtj| j| j| j| j	| j
j| jj| j| j| jf	dd| _d| j_t d | j  t d | jsKt| jtjt| jgdd| _t| tj| j| jgd	d| _d S )
NzQueue._start_thread()QueueFeederThread)targetargsnameTzdoing self._thread.start()z... done self._thread.start())exitpriority
   )r
   r;   clearr7   Threadr   _feedr5   rB   r   r   rg   r   r"   _on_queue_feeder_errorr!   r<   daemonstartr>   r   _finalize_joinweakrefrefr=   _finalize_closer@   r.   r)   r)   r*   rJ      s4   


zQueue._start_threadc                 C   s6   t d |  }|d ur|  t d d S t d d S )Nzjoining queue threadz... queue thread joinedz... queue thread already dead)r
   join)twrthreadr)   r)   r*   rz      s   zQueue._finalize_joinc                 C   sD   t d | | t |  W d    d S 1 sw   Y  d S )Nztelling queue thread to quit)r
   rK   	_sentinelrL   )buffernotemptyr)   r)   r*   r}      s
   

"zQueue._finalize_closec	              
   C   sf  t d |j}	|j}
|j}| j}t}tjdkr|j}|j}nd }	 zT|	  z
| s+|  W |
  n|
  w z6	 | }||u rKt d |  |  W W d S t	|}|d u rY|| n|  z	|| W |  n|  w q6 t
ys   Y nw W n< ty } z0|rt|ddtjkrW Y d }~d S t rtd| W Y d }~d S |  ||| W Y d }~nd }~ww q!)Nz$starting thread to feed data to piper   r   z%feeder thread got sentinel -- exitingerrnor   zerror in queue thread: %s)r
   rI   rR   waitpopleftr   r   r   rU   dumps
IndexError	Exceptiongetattrr   EPIPEr   r   )r   r   rA   	writelockreader_closewriter_closeignore_epipeonerror	queue_semnacquirenreleasenwaitbpopleftsentinelwacquirewreleaserN   er)   r)   r*   rv      sb   




zQueue._feedc                 C   s   ddl }|  dS )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   rN   r   r)   r)   r*   rw     s   zQueue._on_queue_feeder_errorr   )FTN)__name__
__module____qualname__r+   r/   r2   r$   r#   rQ   rY   r\   r_   rb   rd   rf   rg   ri   rl   rJ   staticmethodrz   r}   rv   rw   r)   r)   r)   r*   r   #   s2    



!
	

>r   c                   @   s@   e Zd ZdddZdd Zdd Zdd
dZdd Zdd Zd	S )r   r   c                C   s*   t j| ||d |d| _| | _d S )N)r(   r   )r   r+   	Semaphore_unfinished_tasksr8   _condr%   r)   r)   r*   r+   (  s   zJoinableQueue.__init__c                 C   s   t | | j| jf S r,   )r   r/   r   r   r.   r)   r)   r*   r/   -  s   zJoinableQueue.__getstate__c                 C   s,   t | |d d  |dd  \| _| _d S )N)r   r2   r   r   r0   r)   r)   r*   r2   0  s   zJoinableQueue.__setstate__TNc              	   C   s   | j rtd| d| j||st| j= | j! | jd u r%|   | j	
| | j  | j  W d    n1 s?w   Y  W d    d S W d    d S 1 sWw   Y  d S rG   )r?   rH   r!   rI   r   r5   r   r<   rJ   r;   rK   r   rR   rL   rM   r)   r)   r*   rQ   4  s   

PzJoinableQueue.putc                 C   sf   | j & | jdstd| jj r!| j   W d    d S W d    d S 1 s,w   Y  d S )NFz!task_done() called too many times)r   r   rI   rH   rZ   ra   
notify_allr.   r)   r)   r*   	task_doneA  s   "zJoinableQueue.task_donec                 C   sR   | j  | jj s| j   W d    d S W d    d S 1 s"w   Y  d S r,   )r   r   rZ   ra   r   r.   r)   r)   r*   r~   H  s   "zJoinableQueue.joinr   r   )	r   r   r   r+   r/   r2   rQ   r   r~   r)   r)   r)   r*   r   &  s    

r   c                   @   sN   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	e
ejZdS )r   c                C   sJ   t jdd\| _| _| | _| jj| _tj	dkrd | _
d S | | _
d S )NFr   r   )r   r   r   r   r   r   rE   rF   r   r   r   )r&   r(   r)   r)   r*   r+   S  s   



zSimpleQueue.__init__c                 C   s   | j   | j  d S r,   )r   rg   r   r.   r)   r)   r*   rg   \  s   
zSimpleQueue.closec                 C   r]   r,   r^   r.   r)   r)   r*   r_   `  r`   zSimpleQueue.emptyc                 C   s   t |  | j| j| j| jfS r,   )r	   r-   r   r   r   r   r.   r)   r)   r*   r/   c  s   
zSimpleQueue.__getstate__c                 C   s"   |\| _ | _| _| _| j j| _d S r,   )r   r   r   r   rE   rF   r0   r)   r)   r*   r2   g  s   zSimpleQueue.__setstate__c                 C   s:   | j  | j }W d    n1 sw   Y  t|S r,   )r   r   rC   rU   rV   )r&   rW   r)   r)   r*   rY   k  s   
zSimpleQueue.getc                 C   s\   t |}| jd u r| j| d S | j | j| W d    d S 1 s'w   Y  d S r,   )rU   r   r   r   rA   re   r)   r)   r*   rQ   q  s   

"zSimpleQueue.putN)r   r   r   r+   rg   r_   r/   r2   rY   rQ   classmethodtypesGenericAlias__class_getitem__r)   r)   r)   r*   r   Q  s    	
r   )__all__r   r   r7   r9   rS   r   r{   r   queuer   r   _multiprocessing r   r	   	reductionForkingPicklerrU   utilr
   r   r   r   r   objectr   r   r   r   r)   r)   r)   r*   <module>   s(   	 z
+