B
!;) @ s d Z ddlmZmZ ddlmZ ddlmZ ddlZddl Z ddl
Z
dZ dZ dd Z
ed
d d d
dgZde_ e
jdkrdej_ dej_ dej_ dej_ G dd deZG dd dedZG dd deZG d d d eZG d d d eZee d rG d!d" d"eZee d#r G d$d% d%eZee d&r 0, this specifies the maximum wait time, in
seconds
if timeout <= 0, the select() call won't block, and will
report the currently ready file objects
if timeout is None, select() will block until a monitored
file object becomes ready
Returns:
list of (key, events) for ready file objects
`events` is a bitwise mask of EVENT_READ|EVENT_WRITE
N)r* )r timeoutr r r select s zBaseSelector.selectc C s dS )zmClose the selector.
This must be called to make sure that any underlying resource is freed.
Nr )r r r r close s zBaseSelector.closec C sJ | }|dkrtdy|| S tk
rD td|dY nX dS )zzReturn the key associated to a registered file object.
Returns:
SelectorKey for this file object
NzSelector is closedz{!r} is not registered)get_map RuntimeErrorr! r )r r mappingr r r get_key s zBaseSelector.get_keyc C s t dS )z2Return a mapping of file objects to selector keys.N)r* )r r r r r1 s zBaseSelector.get_mapc C s | S )Nr )r r r r __enter__ s zBaseSelector.__enter__c G s | d S )N)r0 )r argsr r r __exit__ s zBaseSelector.__exit__)N)N)N)r% r& r' r( r r+ r, r- r/ r0 r4 r1 r5 r7 r r r r r) P s
r) ) metaclassc @ sT e Zd ZdZdd Zdd ZdddZd d
Zdd d Zd
d Z dd Z
dd Z dS )_BaseSelectorImplz Base selector implementation.c C s i | _ t| | _d S )N)r r _map)r r r r r s z_BaseSelectorImpl.__init__c C sH yt |S tk
rB x"| j D ]}|j|kr$|jS q$W Y nX dS )al Return a file descriptor from a file object.
This wraps _fileobj_to_fd() to do an exhaustive search in case
the object is invalid but we still have it in our map. This
is used by unregister() so we can unregister an object that
was previously registered even if it is closed. It is also
used by _SelectorMapping.
N)r r
r valuesr r )r r keyr r r r s
z!_BaseSelectorImpl._fileobj_lookupNc C sb |r|t tB @ r td|t|| |||}|j| jkrRtd||j|| j|j< |S )NzInvalid events: {!r}z"{!r} (FD {}) is already registered)
EVENT_READ EVENT_WRITEr
r r r r r r! )r r r r r< r r r r+ s z_BaseSelectorImpl.registerc C s@ y| j | |}W n$ tk
r: td|d Y nX |S )Nz{!r} is not registered)r popr r! r )r r r< r r r r, s
z _BaseSelectorImpl.unregisterc C s y| j | | }W n$ tk
r8 td|d Y nX ||jkr^| | | |||}n"||jkr|j|d}|| j |j < |S )Nz{!r} is not registered)r )
r r r! r r r, r+ r _replacer )r r r r r< r r r r- s
z_BaseSelectorImpl.modifyc C s | j d | _d S )N)r clearr: )r r r r r0 s
z_BaseSelectorImpl.closec C s | j S )N)r: )r r r r r1 s z_BaseSelectorImpl.get_mapc C s$ y
| j | S tk
r dS X dS )zReturn the key associated to a given file descriptor.
Parameters:
fd -- file descriptor
Returns:
corresponding key, or None if not found
N)r r! )r r r r r _key_from_fd s
z _BaseSelectorImpl._key_from_fd)N)N) r% r& r' r( r r r+ r, r- r0 r1 rB r r r r r9 s
r9 c s` e Zd ZdZ fddZd fdd Z fddZejd krHdd
d Z ne
j
Z dd d
Z
Z S )SelectSelectorzSelect-based selector.c s t t | _t | _d S )N)superr set_readers_writers)r ) __class__r r r % s
zSelectSelector.__init__Nc s@ t |||}|t@ r&| j|j |t@ r<| j|j |S )N)rD r+ r= rF addr r> rG )r r r r r< )rH r r r+ * s zSelectSelector.registerc s, t |}| j|j | j|j |S )N)rD r, rF discardr rG )r r r< )rH r r r, 2 s zSelectSelector.unregisterZwin32c C s$ t ||||\}}}||| g fS )N)r/ )r rw_r. xr r r _select9 s zSelectSelector._selectc C s |d kr d nt |d}g }y | | j| jg |\}}}W n tk
rL |S X t|}t|}xV||B D ]J}d}||kr|tO }||kr|tO }| |}|rh| |||j
@ f qhW |S )Nr ) maxrO rF rG InterruptedErrorrE r= r> rB appendr ) r r. readyrK rL rM r r r< r r r r/ ? s$
zSelectSelector.select)N)N)N) r% r& r' r( r r+ r, sysplatformrO r/
__classcell__r r )rH r rC " s
rC c s^ e Zd ZdZdZdZdZ fddZd
fdd Z fddZ d fd d
Z
dd d Z Z S )_PollLikeSelectorz _EVENT_WRITEr r r, )r r r r r< Z
poller_events)rH r r r+ _ s
z_PollLikeSelector.registerc s8 t |}y| j|j W n tk
r2 Y nX |S )N)rD r, r r OSError)r r r< )rH r r r, m s z _PollLikeSelector.unregisterc s y| j | | }W n$ tk
r8 t|dd Y nX d}||jkrd}|t@ r^|| jO }|t@ rp|| jO }y| j |j
| W n t | Y nX d}||j
krd}|r|j||d}|| j |j
< |S )Nz is not registeredFr T)r r )r r r! r r= rY r> rZ r r- r rD r, r r@ )r r r r r< ZchangedZselector_events)rH r r r- w s.
z_PollLikeSelector.modifyc C s |d krd }n |dkr d}nt |d }g }y| j|}W n tk
rR |S X x^|D ]V\}}d}|| j @ rz|tO }|| j @ r|tO }| |}|rZ|
|||j @ f qZW |S )Nr g @@) mathceilr pollrQ rY r> rZ r= rB rR r )r r. rS
fd_event_listr eventr r< r r r r/ s(
z_PollLikeSelector.select)N)N)N)
r% r& r' r( rX rY rZ r r+ r, r- r/ rV r r )rH r rW U s
rW r^ c @ s" e Zd ZdZejZejZej Z
dS ) PollSelectorzPoll-based selector.N) r% r& r' r( r/ r^ rX POLLINrY POLLOUTrZ r r r r ra s ra epollc sD e Zd ZdZejZejZej Z
dd Z d ddZ fddZ Z
S )
EpollSelectorzEpoll-based selector.c C s
| j S )N)r r
)r r r r r
s zEpollSelector.filenoNc C s |d krd}n |dkr d}nt |d d }tt| jd}g }y| j||}W n tk
rh |S X x^|D ]V\}}d}|tj @ r|t
O }|tj @ r|t O }|
|}|rp||||j@ f qpW |S )Nr g @@gMbP?r )r\ r] rP r r r r^ rQ r/ EPOLLINr> EPOLLOUTr= rB rR r ) r r. max_evrS r_ r r` r r< r r r r/ s*
zEpollSelector.selectc s | j t d S )N)r r0 rD )r )rH r r r0 s
zEpollSelector.close)N)r% r& r' r( r/ rd rX rg rY rh rZ r
r0 rV r r )rH r re s
re devpollc s: e Zd ZdZejZejZej Z
dd Z fddZ Z
S )DevpollSelectorzSolaris /dev/poll selector.c C s
| j S )N)r r
)r r r r r
s zDevpollSelector.filenoc s | j t d S )N)r r0 rD )r )rH r r r0 s
zDevpollSelector.close)r% r& r' r( r/ rj rX rb rY rc rZ r
r0 rV r r )rH r rk s rk kqueuec sX e Zd ZdZ fddZdd Zd fdd Z fd d
Zdd d Z fd
dZ Z
S )KqueueSelectorzKqueue-based selector.c s t t | _d S )N)rD r r/ rl r )r )rH r r r s
zKqueueSelector.__init__c C s
| j S )N)r r
)r r r r r
s zKqueueSelector.filenoNc s t |||}y`|t@ r@t|jtjtj}| j |gdd |t
@ rnt|jtj tj}| j |gdd W n t | Y nX |S )Nr )
rD r+ r= r/ keventr KQ_FILTER_READZ KQ_EV_ADDr controlr> KQ_FILTER_WRITEr, )r r r r r< kev)rH r r r+ s zKqueueSelector.registerc s t |}|jt@ rVt|jtjtj}y| j
|gdd W n t k
rT Y nX |jt @ rt|jtj
tj}y| j
|gdd W n t k
r Y nX |S )Nr )rD r, r r= r/ rn r ro Z KQ_EV_DELETEr rp r[ r> rq )r r r< rr )rH r r r, s
zKqueueSelector.unregisterc
C s |d kr d nt |d}t| j}g }y| jd ||}W n tk
rL |S X xb|D ]Z}|j}|j}d}|tj krz|t
O }|tj kr|t O }|
|} | rT|| || j@ f qTW |S )Nr )rP r r r rp rQ Zidentfilterr/ ro r= rq r> rB rR r )
r r. ri rS Zkev_listrr r flagr r< r r r r/ ) s&
zKqueueSelector.selectc s | j t d S )N)r r0 rD )r )rH r r r0 ? s
zKqueueSelector.close)N)N) r% r& r' r( r r
r+ r, r/ r0 rV r r )rH r rm s
rm ) r( abcr r collectionsr Zcollections.abcr r\ r/ rT r= r> r r version_infor r r r r r) r9 rC rW hasattrra re rk rm globalsZDefaultSelectorr r r r sL
~T3Z . M