Your IP : 13.59.247.75
3
\ * � @ s� d ddgZ ddlZddlZddlZddlZddlZddlZddlZddlm Z m
Z
ddl Z ddl m
Z
ddl mZ ejjZdd lmZmZmZmZmZ G d
d � d e�Ze� ZG d d� de�ZG d d� de�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�Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd d � Zd d � Zd d!� Zed"d#� �Zed$d%� �Zed&d'� �Zd S )+r r c C s� |dkrddl m} || _tjdd�\| _| _|j� | _t j
� | _ t j
dkrTd | _n
|j� | _|j|�| _d| _| j� t j
dkr�t| tj� d S )Nr r )�
SEM_VALUE_MAXF)�duplex�win32)Z synchronizer �_maxsizer �Pipe�_reader�_writer�Lock�_rlock�os�getpid�_opid�sys�platform�_wlockZBoundedSemaphore�_sem�
_ignore_epipe� _after_forkr
r )�self�maxsize�ctx� r$ � /usr/lib64/python3.6/queues.py�__init__$ s
zQueue.__init__c C s. t j| � | 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 |\| _ | _| _| _| _| _| _| _| j� d S )N) r r r r r r r r r )r! �stater$ r$ r% � __setstate__> s $zQueue.__setstate__c C sb t d� tjtj� �| _tj� | _d | _d | _ d| _
d| _ d | _ | j
j| _| jj| _| jj| _d S )NzQueue._after_fork()F)r
� threading� Conditionr � _notempty� collections�deque�_buffer�_thread� _jointhread�_joincancelled�_closed�_closer �
send_bytes� _send_bytesr �
recv_bytes� _recv_bytes�poll�_poll)r! r$ r$ r% r C s
zQueue._after_forkTNc C sP | j j||�st�| j�. | jd kr,| j� | jj|� | jj� W d Q R X d S )N) r �acquirer r- r1 �
_start_threadr0 �append�notify)r! �obj�block�timeoutr$ r$ r% �putP s
z Queue.putc
C s� |r2|d kr2| j � | j� }W d Q R X | jj� nr|rBtj� | }| j j||�sTt�zB|rv|tj� }| j|�s�t�n | j� s�t�| j� }| jj� W d | j j� X t j
|�S )N) r r9 r �release�timeZ monotonicr< r r; �_ForkingPickler�loads)r! rA rB �resZdeadliner$ r$ r% �get[ s&
z Queue.getc C s | j | jjj� S )N)r r �_semlockZ
_get_value)r! r$ r$ r% �qsizes s z Queue.qsizec C s
| j � S )N)r; )r! r$ r$ r% �emptyw s z Queue.emptyc C s | j jj� S )N)r rJ �_is_zero)r! r$ r$ r% �fullz s z
Queue.fullc C s
| j d�S )NF)rI )r! r$ r$ r% �
get_nowait} s zQueue.get_nowaitc C s | j |d�S )NF)rC )r! r@ r$ r$ r% �
put_nowait� s zQueue.put_nowaitc C s2 d| _ z| jj� W d | j}|r,d | _|� X d S )NT)r4 r �closer5 )r! rQ r$ r$ r% rQ � s z Queue.closec C s t d� | jr| j� d S )NzQueue.join_thread())r
r2 )r! r$ r$ r% � join_thread� s zQueue.join_threadc C s6 t d� d| _y| jj� W n tk
r0 Y nX d S )NzQueue.cancel_join_thread()T)r
r3 r2 Zcancel�AttributeError)r! r$ r$ r% �cancel_join_thread� s zQueue.cancel_join_threadc C s� t d� | jj� tjtj| j| j| j| j | j
j | j fdd�| _
d| j
_t d� | j
j� t d� | js�t| j
tjtj| j
�gd
d�| _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
r0 �clearr+ ZThreadr �_feedr- r7 r r rQ r r1 Zdaemon�startr3 r �_finalize_join�weakref�refr2 �_finalize_closer5 )r! r$ r$ r% r= � s(
zQueue._start_threadc C s4 t d� | � }|d k r(|j� t d� nt d� d S )Nzjoining queue threadz... queue thread joinedz ... queue thread already dead)r
�join)ZtwrZthreadr$ r$ r% r^ � s
zQueue._finalize_joinc C s. t d� |� | jt� |j� W d Q R X d S )Nz telling queue thread to quit)r
r> � _sentinelr? )�buffer�notemptyr$ r$ r% ra � s
zQueue._finalize_closec C sP t d� |j}|j}|j}| j} t}
tjdkr<|j} |j} nd } �xy�|� z| sX|� W d |� X ybx\| � }
|
|
kr�t d� |� d S tj |
�}
| d kr�||
� qh| � z ||
� W d | � X qhW W n t
k
r� Y nX W qD t k
�rF } zJ|�rt |dd�t
jk�rd S t� �r&td|� d S dd l}|j� W Y d d }~X qDX qDW d S )Nz$starting thread to feed data to piper z%feeder thread got sentinel -- exiting�errnor zerror in queue thread: %s)r
r< rD �wait�popleftrc r r rF �dumps�
IndexError� Exception�getattrrf ZEPIPEr r � traceback� print_exc)rd re r6 Z writelockrQ Z ignore_epipeZnacquireZnreleaseZnwaitZbpopleft�sentinelZwacquireZwreleaser@ �erm r$ r$ r% r\ � sR
z Queue._feed)r )TN)TN)�__name__�
__module__� __qualname__r& r( r* r rC rI rK rL rN rO rP rQ rR rT r= � staticmethodr^ ra r\ r$ r$ r$ r% r " s$
c @ s@ e Zd Zddd�Zdd� Zdd� Zdd
d �Zd d
� Zdd� Zd S )r r c C s* t j| ||d� |jd�| _|j� | _d S )N)r# r )r r& Z Semaphore�_unfinished_tasksr, �_cond)r! r" r# r$ r$ r% r& s zJoinableQueue.__init__c C s t j| �| j| jf S )N)r r( rv ru )r! r$ r$ r% r( s zJoinableQueue.__getstate__c C s, t j| |d d� � |dd � \| _| _d S )N� ���rx )r r* rv ru )r! r) r$ r$ r% r* s zJoinableQueue.__setstate__TNc C sl | j j||�st�| j�J | j�8 | jd kr4| j� | jj|� | j j
� | jj � W d Q R X W d Q R X d S )N) r r< r r- rv r1 r= r0 r> ru rD r? )r! r@ rA rB r$ r$ r% rC s
zJoinableQueue.putc C s@ | j �0 | jjd�s td��| jjj� r2| j j� W d Q R X d S )NFz!task_done() called too many times)rv ru r< �
ValueErrorrJ rM Z
notify_all)r! r$ r$ r% � task_done' s
zJoinableQueue.task_donec C s, | j � | jjj� s | j j� W d Q R X d S )N)rv ru rJ rM rg )r! r$ r$ r% rb . s zJoinableQueue.join)r )TN) rq rr rs r& r( r* rC rz rb r$ r$ r$ r% r
s
c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )r c C sH t jdd�\| _| _|j� | _| jj| _tj dkr:d | _
n
|j� | _
d S )NF)r r ) r r r r r r r: r; r r r )r! r# r$ r$ r% r& 9 s
zSimpleQueue.__init__c C s
| j � S )N)r; )r! r$ r$ r% rL B s zSimpleQueue.emptyc C s t j| � | j| j| j| jfS )N)r r' r r r r )r! r$ r$ r% r( E s
zSimpleQueue.__getstate__c C s" |\| _ | _| _| _| j j| _d S )N)r r r r r: r; )r! r) r$ r$ r% r* I s zSimpleQueue.__setstate__c C s& | j � | jj� }W d Q R X tj|�S )N)r r r8 rF rG )r! rH r$ r$ r% rI M s zSimpleQueue.getc C sD t j|�}| jd kr"| jj|� n | j� | jj|� W d Q R X d S )N)rF ri r r r6 )r! r@ r$ r$ r% rC S s
zSimpleQueue.putN) rq rr rs r& rL r( r* rI rC r$ r$ r$ r% r 7 s ) �__all__r r r+ r. rE r_ rf Zqueuer r Z_multiprocessing� r r Z reductionZForkingPicklerrF �utilr
r r r
r �objectr rc r r r$ r$ r$ r% �<module>
s$
b
*