o

�
bV�@s�dZd
dl
Z
d
dlmZmZmZmZmZmZ	m
Z

d
dlmZ
d
dl
mZ
d
dlmZ
d
dlmZ
ee�
Gdd	�d	ejej��
Zd
d�Zd	dgZdS)z�
A poll() based implementation of the twisted main loop.

To install the event loop (and you should do this before any connections,
listeners or connectors are added)::

    from twisted.internet import pollreactor
    pollreactor.install()
�N)�POLLERR�POLLHUP�POLLIN�POLLNVAL�POLLOUT�error�poll)
�implementer)
�	posixbase)
�
IReactorFDSet)
�logc@s�eZ
dZd
ZeeBeBZeZ	e
Zdd�Zdd�Z
dd�Zdd	�Zd
d�Zdd
�Zdd�Zdd�Zdd�ZeZdd�Zdd�ZdS)�PollReactora^
    A reactor that uses poll(2).

    @ivar _poller: A L{select.poll} which will be used to check for I/O
        readiness.

    @ivar _selectables: A dictionary mapping integer file descriptors to
        instances of L{FileDescriptor} which have been registered with the
        reactor.  All L{FileDescriptor}s which are currently receiving read or
        write readiness notifications will be present as values in this
        dictionary.

    @ivar _reads: A dictionary mapping integer file descriptors to arbitrary
        values (this is essentially a set).  Keys in this dictionary will be
        registered with C{_poller} for read readiness notifications which will
        be dispatched to the corresponding L{FileDescriptor} instances in
        C{_selectables}.

    @ivar _writes: A dictionary mapping integer file descriptors to arbitrary
        values (this is essentially a set).  Keys in this dictionary will be
        registered with C{_poller} for write readiness notifications which will
        be dispatched to the corresponding L{FileDescriptor} instances in
        C{_selectables}.
    c

Cs*t�|_
i|_i|_i|_tj�|�

d
S)zo
        Initialize polling object, file descriptor tracking dictionaries, and
        the base class.
        N)r�_poller�_selectables�_reads�_writesr
�PosixReactorBase�__init__�
�self�r�>/usr/lib/python3/dist-packages/twisted/internet/pollreactor.pyrCs




zPollReactor.__init__cCs�z|j�
|
�

Wn	ty


Yn
wd
}|
|jvr|tB}|
|jvr&|tB}|d
kr3|j�|
|�
dS|
|jvr>|j|
=dSdS)z*Register/unregister an fd with the poller.r
N)	r�
unregister�KeyErrorrrrr�registerr)r�fd�maskrrr�_updateRegistrationNs


�









�zPollReactor._updateRegistrationcCsnz
|
��}||
Wnt
y&


|j��D]
\}}|
|ur 
nqYdSYn
w||vr5||=|�|�

dSdS)
N)�fileno�
BaseExceptionr�itemsr)r�
selectable�mdictr�fdesrrr�_dictRemove`s



���


�zPollReactor._dictRemovecC�8|
��}||j
v
r|
|j|<d
|j
|<|�|�

dSdS)z@Add a FileDescriptor for notification of data available to read.�
N)rrrr)r�readerrrrr�	addReaderu�






�zPollReactor.addReadercCr%)zAAdd a FileDescriptor for notification of data available to write.r&N)rrrr)r�writerrrrr�	addWriter}r)zPollReactor.addWritercC�|�|
|j
�S)
z?Remove a Selectable for notification of data available to read.)r$r)rr'rrr�removeReader��zPollReactor.removeReadercCr,)
z@Remove a Selectable for notification of data available to write.)r$r)rr*rrr�removeWriter�r.zPollReactor.removeWriterc

s,���f
d
d��j
D�
�f
dd��jD�
�S)zD
        Remove all selectables, and return a list of them.
        c
�g|]}
�j|
�qSr�
r��.0rrrr�
<listcomp>��z)PollReactor.removeAll.<locals>.<listcomp>c
r0rr1r2rrrr4�r5)�
_removeAllrrrrrr�	removeAll�s

�zPollReactor.removeAllc
Cs�|
d
u
r
t|
d�
}
z|j
�|
�
}Wnty.
}
z|jdtjkr)WYd
}~d
S�d
}~w
w|j}|D]\}}z|j|}Wn	t	yH


Yq4wt
�|||||�
q4d
S)zPoll the poller for new events.Ni�r
)�intrr�SelectError�args�errno�EINTR�_doReadOrWriterrr�callWithLogger)r�timeout�
l�
e�_drdwr�eventr!rrr�doPoll�s&




��



��zPollReactor.doPollc

��f
d
d��jD�
S)Nc
r0rr1r2rrrr4�r5z*PollReactor.getReaders.<locals>.<listcomp>)
rrrrr�
getReaders��
zPollReactor.getReadersc

rE)Nc
r0rr1r2rrrr4�r5z*PollReactor.getWriters.<locals>.<listcomp>)
rrrrr�
getWriters�rGzPollReactor.getWritersN)�__name__�
__module__�__qualname__�__doc__rrr�_POLL_DISCONNECTEDr�_POLL_INr�	_POLL_OUTrrr$r(r+r-r/r7rD�doIterationrFrHrrrrr
$s"

	r
cCst�}d
dl
m}

|
|�

dS)zInstall the poll() reactor.r
)
�installReactorN)r
�twisted.internet.mainrQ)�
prQrrr�install�s
rT)rLr;�selectrrrrrrr9r�zope.interfacer	�twisted.internetr
�twisted.internet.interfacesr�twisted.pythonrr�_PollLikeMixinr
rT�__all__rrrr�<module>
s$