U
e5d�H � @ s� d dl mZmZ d dlmZ d dlmZ d dlZd dlZd dl Z dZ
dZ dd� Z ed d
d d d
g�Z
de
_e jdkr�de
j_de
j_de
j_de
j_G dd� de�ZG dd� ded�ZG dd� de�ZG dd � d e�ZG d d � d e�Zeed �r�G d d!� d!e�Zeed"��rG d#d$� d$e�Zeed%��r6G d&d'� d'e�Z eed(��rRG d)d*� d*e�Z d*e � k�rde Z n:d$e � k�rveZ n(d'e � k�r�e Z nd!e � k�r�eZ neZ dS )+� )�ABCMeta�abstractmethod)�
namedtuple)�MappingN� � c
C sf t | t�r| }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 | jj�S 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 Zeddd��Zedd� �Zddd�Zeddd ��Zd
d � Zd d
� Z edd� �Z
dd� Z dd� Z dS )� BaseSelectorNc C s t �d S r ��NotImplementedError�r 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 krtd��z
|| 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� Zddd�Zdd � 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� fdd�Zd
� fdd� Z� fdd�ZejdkrDdd 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 �r�w�_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 �sys�platformrW r4 �
__classcell__r r rO r rJ " s
rJ c sZ e Zd ZdZdZdZ� fdd�Zd � fdd� Z� fdd�Zd
� 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|�d��d �Y nX d}||jkr�d}|t@ r^|| jO }|t@ rp|| jO }z| j� |j
|� W n t � � |� � Y nX d}||j
kr�d}|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 � @�@) �math�ceilr �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
ddd�Z� fdd�Z � 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| j�d�}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 )N�����r re g����MbP?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
� fdd�Z � 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� fdd�Zdd� 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 @ r�t�|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| j�d ||�}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 Zident�filterr4 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