3
\�K � @ s� d dl mZmZ d dlmZmZ d dlZd dlZd dlZd(Z d)Z
dd� Z eddd d
d g�Z d e _
ejd*kr�de j_
de j_
de j_
de j_
G dd� de�ZG dd� ded�ZG dd� de�ZG dd� de�Zeed �r�G d d � d e�Zeed ��rG d d!� d!e�Zeed"��r G d#d$� d$e�Zeed%��r:G d&d'� d'e�Zd'e � k�rLeZ n:d!e � k�r^eZ n(d$e � k�rpeZ nd e � k�r�eZ neZ dS )+� )�ABCMeta�abstractmethod)�
namedtuple�MappingN� c
C sf t | t�r| }n s z_SelectorMapping.__init__c C s t | jj�S )N)�lenr �
_fd_to_key)r r r r �__len__A s z_SelectorMapping.__len__c C sB y| j j|�}| j j| S tk
r< tdj|��d �Y nX d S )Nz{!r} is not registered)r �_fileobj_lookupr �KeyErrorr
)r r r r r r � __getitem__D s
z _SelectorMapping.__getitem__c C s t | jj�S )N)�iterr r )r r r r �__iter__K s z_SelectorMapping.__iter__N)�__name__�
__module__� __qualname__r r r! r# r r r r r ; s r c @ sj e Zd Zeddd��Zedd� �Zddd�Zeddd ��Zd
d � Zd d
� Z edd� �Z
dd� Z dd� Z dS )� BaseSelectorNc C s t �d S )N)�NotImplementedError)r r r r r r r �register^ s zBaseSelector.registerc C s t �d S )N)r( )r r r r r �
unregisteru s zBaseSelector.unregisterc C s | j |� | j|||�S )N)r* r) )r r r r r r r �modify� s
zBaseSelector.modifyc C s t �d S )N)r( )r �timeoutr r r �select� s zBaseSelector.selectc C s d S )Nr )r r r r �close� s zBaseSelector.closec C sJ | j � }|d krtd��y|| S tk
rD tdj|��d �Y nX d S )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 �d S )N)r( )r r r r r/ � s zBaseSelector.get_mapc C s | S )Nr )r r r r � __enter__� s zBaseSelector.__enter__c G s | j � d S )N)r. )r �argsr r r �__exit__� s zBaseSelector.__exit__)N)N)N)
r$ r% r& r r) r* r+ r- r. r2 r/ r3 r5 r r r r r' O s
r' )� metaclassc @ sP e Zd Zdd� Zdd� Zddd�Zdd � Zdd
d �Zd d
� Zdd� Z dd� Z
dS )�_BaseSelectorImplc 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"| jj� D ]}|j|kr$|jS q$W � Y nX d S )N)r r r �valuesr r )r r �keyr r r r � s
z!_BaseSelectorImpl._fileobj_lookupNc C sd | s|t tB @ r"tdj|���t|| j|�||�}|j| jkrTtdj||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 j| j|��}W n$ tk
r: tdj|��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 | j|� }W n$ tk
r8 tdj|��d �Y nX ||jkr^| j|� | j|||�}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 j� d | _d S )N)r �clearr8 )r r r r r. s
z_BaseSelectorImpl.closec C s | j S )N)r8 )r r r r r/ s z_BaseSelectorImpl.get_mapc C s$ y
| j | S tk
r d S X d S )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. r/ r@ r r r r r7 � s
r7 c s\ e Zd Z� fdd�Zd
� fdd� Z� fdd�ZejdkrDdd d
�Zne j Zdd d �Z � Z
S )�SelectSelectorc s t � j� t� | _t� | _d S )N)�superr �set�_readers�_writers)r )� __class__r r r % s
zSelectSelector.__init__Nc s@ t � j|||�}|t@ r&| jj|j� |t@ r<| jj|j� |S )N)rB r) r; rD �addr r< rE )r r r r r: )rF r r r) * s zSelectSelector.registerc s, t � j|�}| jj|j� | jj|j� |S )N)rB r* rD �discardr rE )r r r: )rF r r r* 2 s zSelectSelector.unregisterZwin32c C s$ t j ||||�\}}}||| g fS )N)r- )r �r�w�_r, �xr r r �_select9 s zSelectSelector._selectc
C s� |d kr d nt |d�}g }y | j| j| jg |�\}}}W n tk
rL |S X t|�}t|�}xV||B D ]J}d}||kr�|tO }||kr�|tO }| j|�}|rh|j |||j
@ f� qhW |S )Nr ) �maxrM rD rE �InterruptedErrorrC r; r< r@ �appendr ) r r, �readyrI rJ rK r r r: r r r r- ? s$
zSelectSelector.select)N)N)N) r$ r% r& r r) r* �sys�platformrM r- �
__classcell__r r )rF r rA " s
rA �pollc s@ e Zd Z� fdd�Zd
� fdd� Z� fdd�Zd dd �Z� ZS ) � PollSelectorc s t � j� tj� | _d S )N)rB r r- rU �_poll)r )rF r r r Z s
zPollSelector.__init__Nc sL t � j|||�}d}|t@ r&|tjO }|t@ r8|tjO }| jj|j|� |S )Nr ) rB r) r; r- �POLLINr< �POLLOUTrW r )r r r r r: � poll_events)rF r r r) ^ s
zPollSelector.registerc s t � j|�}| jj|j� |S )N)rB r* rW r )r r r: )rF r r r* h s zPollSelector.unregisterc
C s� |d krd }n |dkr d}nt j|d �}g }y| jj|�}W n tk
rR |S X x^|D ]V\}}d}|tj @ rz|tO }|tj @ r�|t O }| j
|�}|rZ|j |||j @ f� qZW |S )Nr g @�@)
�math�ceilrW rU rO r- rX r< rY r; r@ rP r )r r, rQ �
fd_event_listr �eventr r: r r r r- m s(
zPollSelector.select)N)N)r$ r% r& r r) r* r- rT r r )rF r rV W s
rV �epollc sT e Zd Z� fdd�Zdd� Zd� fdd� Z� fdd �Zdd
d �Z� fd d
�Z� Z S )�
EpollSelectorc s t � j� tj� | _d S )N)rB r r- r_ �_epoll)r )rF r r r � s
zEpollSelector.__init__c C s
| j j� S )N)ra r )r r r r r � s zEpollSelector.filenoNc st t � j|||�}d}|t@ r&|tjO }|t@ r8|tjO }y| jj|j|� W n" t k
rn t � j
|� � Y nX |S )Nr ) rB r) r; r- �EPOLLINr< �EPOLLOUTra r �
BaseExceptionr* )r r r r r: Z epoll_events)rF r r r) � s
zEpollSelector.registerc s8 t � j|�}y| jj|j� W n tk
r2 Y nX |S )N)rB r* ra r �OSError)r r r: )rF r r r* � s zEpollSelector.unregisterc
C s� |d krd}n |dkr d}nt j|d �d }tt| j�d�}g }y| jj||�}W n tk
rh |S X x^|D ]V\}}d}|tj @ r�|t
O }|tj @ r�|t O }| j
|�}|rp|j|||j@ f� qpW |S )Nr r g @�@g����MbP?�����)r[ r\ rN r r ra rU rO r- rb r< rc r; r@ rP r ) r r, �max_evrQ r] r r^ r r: r r r r- � s*
zEpollSelector.selectc s | j j� t� j� d S )N)ra r. rB )r )rF r r r. � s
zEpollSelector.close)N)N)
r$ r% r& r r r) r* r- r. rT r r )rF r r` � s
r` �devpollc sT e Zd Z� fdd�Zdd� Zd� fdd� Z� fdd �Zdd
d �Z� fd d
�Z� Z S )�DevpollSelectorc s t � j� tj� | _d S )N)rB r r- rh �_devpoll)r )rF r r r � s
zDevpollSelector.__init__c C s
| j j� S )N)rj r )r r r r r � s zDevpollSelector.filenoNc sL t � j|||�}d}|t@ r&|tjO }|t@ r8|tjO }| jj|j|� |S )Nr ) rB r) r; r- rX r< rY rj r )r r r r r: rZ )rF r r r) � s
zDevpollSelector.registerc s t � j|�}| jj|j� |S )N)rB r* rj r )r r r: )rF r r r* � s zDevpollSelector.unregisterc
C s� |d krd }n |dkr d}nt j|d �}g }y| jj|�}W n tk
rR |S X x^|D ]V\}}d}|tj @ rz|tO }|tj @ r�|t O }| j
|�}|rZ|j |||j @ f� qZW |S )Nr g @�@)
r[ r\ rj rU rO r- rX r< rY r; r@ rP r )r r, rQ r] r r^ r r: r r r r- � s(
zDevpollSelector.selectc s | j j� t� j� d S )N)rj r. rB )r )rF r r r. s
zDevpollSelector.close)N)N)
r$ r% r& r r r) r* r- r. rT r r )rF r ri � s
ri �kqueuec sT e Zd Z� fdd�Zdd� Zd� fdd� Z� fdd �Zdd
d �Z� fd d
�Z� Z S )�KqueueSelectorc s t � j� tj� | _d S )N)rB r r- rk �_kqueue)r )rF r r r s
zKqueueSelector.__init__c C s
| j j� S )N)rm r )r r r r r s zKqueueSelector.filenoNc s� t � j|||�}y`|t@ r@tj|jtjtj�}| jj |gdd� |t
@ rntj|jtj tj�}| jj |gdd� W n" t k
r� t � j
|� � Y nX |S )Nr )rB r) r; r- �keventr �KQ_FILTER_READZ KQ_EV_ADDrm �controlr< �KQ_FILTER_WRITErd r* )r r r r r: �kev)rF r r r) s zKqueueSelector.registerc s� t � j|�}|jt@ rVtj|jtjtj�}y| j j
|gdd� W n t k
rT Y nX |jt @ r�tj|jtj
tj�}y| j j
|gdd� W n t k
r� Y nX |S )Nr )rB r* r r; r- rn r ro Z KQ_EV_DELETErm rp re r< rq )r r r: rr )rF r r r* ' s
zKqueueSelector.unregisterc
C s� |d kr d nt |d�}t| j�}g }y| jjd ||�}W n tk
rL |S X xb|D ]Z}|j}|j}d}|tj krz|t
O }|tj kr�|t O }| j
|�} | rT|j| || j@ f� qTW |S )Nr )rN r r rm rp rO Zident�filterr- ro r; rq r< r@ rP r )
r r, rg rQ Zkev_listrr r �flagr r: r r r r- < s&
zKqueueSelector.selectc s | j j� t� j� d S )N)rm r. rB )r )rF r r r. R s
zKqueueSelector.close)N)N)
r$ r% r& r r r) r* r- r. rT r r )rF r rl
s
rl r � )r r ) �abcr r � collectionsr r r[ r- rR r; r< r r �__doc__� version_infor r r r r r' r7 rA �hasattrrV r` ri rl �globalsZDefaultSelectorr r r r � sF
~U3
1 G 8 M