U
e5dH @ s d dl mZmZ d dlmZ d dlmZ d dlZd dlZd dl Z dZ
dZ dd Z ed d
d d d
gZ
de
_e jdkrde
j_de
j_de
j_de
j_G dd deZG dd dedZG dd deZG dd d eZG d d d eZeed rG d d! d!eZeed"rG d#d$ d$eZeed%r6G d&d' d'eZ eed(rRG d)d* d*eZ d*e krde Z n:d$e krveZ n(d'e kre Z nd!e kreZ neZ dS )+ )ABCMetaabstractmethod)
namedtuple)MappingN c
C sf t | tr| }n td|d Y nX d S N{!r} is not registered)r _fileobj_lookupr KeyErrorr )r r r r r r __getitem__E s
z _SelectorMapping.__getitem__c C s t | jjS r )iterr r r r r r __iter__L s z_SelectorMapping.__iter__N)__name__
__module__ __qualname__r r! r& r( r r r r r < s r c @ sj e Zd ZedddZedd ZdddZeddd Zd
d Zd d
Z edd Z
dd Z dd Z dS ) BaseSelectorNc C s t d S r NotImplementedErrorr r r r r r r register_ s zBaseSelector.registerc C s t d S r r- )r r r r r
unregisterv s zBaseSelector.unregisterc C s | | | |||S r )r1 r0 r/ r r r modify s
zBaseSelector.modifyc C s t d S r r- )r timeoutr r r select s zBaseSelector.selectc C s d S r r r r r r close s zBaseSelector.closec C sL | }|d krtdz
|| W S tk
rF td|d Y nX d S )NzSelector is closedr# )get_map RuntimeErrorr% r )r r mappingr r r get_key s
zBaseSelector.get_keyc C s t d S r r- r r r r r6 s zBaseSelector.get_mapc C s | S r r r r r r __enter__ s zBaseSelector.__enter__c G s | d S r )r5 )r argsr r r __exit__ s zBaseSelector.__exit__)N)N)N)
r) r* r+ r r0 r1 r2 r4 r5 r9 r6 r: r< r r r r r, P s
r, ) metaclassc @ sP e Zd Zdd Zdd ZdddZdd Zdd
d Zd d
Zdd Z dd Z
dS )_BaseSelectorImplc C s i | _ t| | _d S r )r r _mapr r r r r s z_BaseSelectorImpl.__init__c C sN z
t |W S tk
rH | j D ] }|j|kr$|j Y S q$ Y nX d S r )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 rB r r r r0 s z_BaseSelectorImpl.registerc C s@ z| j | |}W n$ tk
r: td|d Y nX |S r" )r popr$ r% r rA r r r r1 s
z _BaseSelectorImpl.unregisterc C s z| j | | }W n$ tk
r8 td|d Y nX ||jkr^| | | |||}n"||jkr|j|d}|| j |j < |S )Nr# )r )
r r$ r% r r r1 r0 r _replacer rE r r r r2 s
z_BaseSelectorImpl.modifyc C s | j d | _d S r )r clearr? r r r r r5 s
z_BaseSelectorImpl.closec C s | j S r )r? r r r r r6 s z_BaseSelectorImpl.get_mapc C s( z | j | W S tk
r" Y d S X d S r )r r% )r r r r r _key_from_fd s z _BaseSelectorImpl._key_from_fd)N)N) r) r* r+ r r$ r0 r1 r2 r5 r6 rI r r r r r> s
r> c s\ e Zd Z fddZd
fdd Z fddZejdkrDdd d
Zne j Zdd d Z Z
S )SelectSelectorc s t t | _t | _d S r )superr set_readers_writersr __class__r r r % s
zSelectSelector.__init__Nc s@ t |||}|t@ r&| j|j |t@ r<| j|j |S r )rK r0 rC rM addr rD rN rE rO r r r0 * s zSelectSelector.registerc s, t |}| j|j | j|j |S r )rK r1 rM discardr rN rA rO r r r1 2 s zSelectSelector.unregisterZwin32c C s$ t ||||\}}}||| g fS r )r4 )r rw_r3 xr r r _select9 s zSelectSelector._selectc C s |d kr d nt |d}g }z | | j| jg |\}}}W n tk
rP | Y S X t|}t|}||B D ]J}d}||kr|tO }||kr|tO }| |}|rj| |||j
@ f qj|S Nr ) maxrW rM rN InterruptedErrorrL rC rD rI appendr ) r r3 readyrS rT rU r r rB r r r r4 ? s$
zSelectSelector.select)N)N)N) r) r* r+ r r0 r1 sysplatformrW r4
__classcell__r r rO r rJ " s
rJ c sZ e Zd ZdZdZdZ fddZd fdd Z fddZd
fdd Z dd
d Z
Z S )_PollLikeSelectorNc s t | | _d S r )rK r
_selector_clsr r rO r r r [ s
z_PollLikeSelector.__init__c sl t |||}d}|t@ r&|| jO }|t@ r8|| jO }z| j|j| W n t | Y nX |S rX ) rK r0 rC _EVENT_READrD _EVENT_WRITEr r r1 )r r r r rB Z
poller_eventsrO r r r0 _ s
z_PollLikeSelector.registerc s8 t |}z| j|j W n tk
r2 Y nX |S r )rK r1 r r OSErrorrA rO r r r1 m s z _PollLikeSelector.unregisterc s z| j | | }W n$ tk
r8 t|dd Y nX d}||jkrd}|t@ r^|| jO }|t@ rp|| jO }z| 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 rC rb rD rc r r2 r rK r1 r rG )r r r r rB ZchangedZselector_eventsrO r r r2 w s.
z_PollLikeSelector.modifyc C s |d krd }n |dkr d}nt |d }g }z| j|}W n tk
rV | Y S X |D ]V\}}d}|| j @ r||tO }|| j @ r|tO }| |}|r\|
|||j @ f q\|S )Nr @@) mathceilr pollrZ rb rD rc rC rI r[ r )r r3 r\
fd_event_listr eventr rB r r r r4 s(
z_PollLikeSelector.select)N)N)N) r) r* r+ ra rb rc r r0 r1 r2 r4 r_ r r rO r r` U s
r` rh c @ s e Zd ZejZejZejZ dS ) PollSelectorN)
r) r* r+ r4 rh ra POLLINrb POLLOUTrc r r r r rk s rk epollc s@ e Zd ZejZejZejZ dd Z
dddZ fddZ Z S )
EpollSelectorc C s
| j S r r r
r r r r r
s zEpollSelector.filenoNc C s |d krd}n |dkr d}nt |d d }tt| jd}g }z| j||}W n tk
rl | Y S X |D ]V\}}d}|tj @ r|t
O }|tj @ r|t O }|
|}|rr||||j@ f qr|S )Nr re gMbP?r )rf rg rY r r r rh rZ r4 EPOLLINrD EPOLLOUTrC rI r[ r ) r r3 max_evr\ ri r rj r rB r r r r4 s*
zEpollSelector.selectc s | j t d S r r r5 rK r rO r r r5 s
zEpollSelector.close)N)
r) r* r+ r4 rn ra rr rb rs rc r
r5 r_ r r rO r ro s
ro devpollc s6 e Zd ZejZejZejZ dd Z
fddZ Z S )DevpollSelectorc C s
| j S r rp r r r r r
s zDevpollSelector.filenoc s | j t d S r ru r rO r r r5 s
zDevpollSelector.close)
r) r* r+ r4 rv ra rl rb rm rc r
r5 r_ r r rO r rw s
rw kqueuec sT e Zd Z fddZdd Zd fdd Z fdd Zdd
d Z fd d
Z Z S )KqueueSelectorc s t t | _d S r )rK r r4 rx r r rO r r r s
zKqueueSelector.__init__c C s
| j S r rp r r r r r
s zKqueueSelector.filenoNc s t |||}z`|t@ r@t|jtjtj}| j |gdd |t
@ rnt|jtj tj}| j |gdd W n t | Y nX |S rX )
rK r0 rC r4 keventr KQ_FILTER_READZ KQ_EV_ADDr controlrD KQ_FILTER_WRITEr1 )r r r r rB kevrO r r r0 s zKqueueSelector.registerc s t |}|jt@ rVt|jtjtj}z| j
|gdd W n t k
rT Y nX |jt @ rt|jtj
tj}z| j
|gdd W n t k
r Y nX |S rX )rK r1 r rC r4 rz r r{ Z KQ_EV_DELETEr r| rd rD r} )r r rB r~ rO r r r1 s$
zKqueueSelector.unregisterc
C s |d kr d nt |d}t| j}g }z| jd ||}W n tk
rP | Y S X |D ]Z}|j}|j}d}|tj kr||t
O }|tj kr|t O }|
|} | rV|| || j@ f qV|S rX )rY r r r r| rZ Zidentfilterr4 r{ rC r} rD rI r[ r )
r r3 rt r\ Zkev_listr~ r flagr rB r r r r4 ) s&
zKqueueSelector.selectc s | j t d S r ru r rO r r r5 ? s
zKqueueSelector.close)N)N)
r) r* r+ r r
r0 r1 r4 r5 r_ r r rO r ry s
ry ) abcr r collectionsr Zcollections.abcr rf r4 r] rC rD r r __doc__ version_infor r r r r r, r> rJ r` hasattrrk ro rw ry globalsZDefaultSelectorr r r r sJ
~T3Z
. M