a
R�fPL � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlZddl Z ddl
Z
dZ dZ dd � Z
ed
g d ��Zd e_ e
jd
kr�dej_ dej_ dej_ dej_ G dd� de�ZG dd� ded�ZG dd� de�ZG dd� de�ZG dd � d e�Zee d �r�G d d � d e�Zee d ��rG d!d"� d"e�Zee d#��r6G d$d%� d%e�Z ee d&��rRG d'd(� d(e�Z d)d*� Z e d&��rje Z n4e d ��rzeZ n$e d#��r�e Z ne d ��r�eZ neZ dS )+z|Selectors module.
This module allows high-level and efficient I/O multiplexing, built upon the
`select` module primitives.
� )�ABCMeta�abstractmethod)�
namedtuple)�MappingN� � c
C sd t | t�r| }n:zt| �� �}W n( tttfyH td�| ��d�Y n0 |dk r`td�|���|S )z�Return a file descriptor from a file object.
Parameters:
fileobj -- file object or file descriptor
Returns:
corresponding file descriptor
Raises:
ValueError if the object is invalid
zInvalid file object: {!r}Nr zInvalid file descriptor: {})�
isinstance�int�fileno�AttributeError� TypeError�
ValueError�format)�fileobj�fd� r �./opt/alt/python39/lib64/python3.9/selectors.py�_fileobj_to_fd s
��r � SelectorKey)r r �events�dataz�SelectorKey(fileobj, fd, events, data)
Object used to associate a file object to its backing
file descriptor, selected event mask, and attached data.
)� � zFile object registered.zUnderlying file descriptor.z3Events that must be waited for on this file object.zzOptional opaque data associated to this file object.
For example, this could be used to store a per-client session ID.c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �_SelectorMappingz)Mapping of file objects to selector keys.c C s
|| _ d S �N)� _selector)�self�selectorr r r �__init__@ s z_SelectorMapping.__init__c C s t | jj�S r )�lenr �
_fd_to_key�r r r r �__len__C s z_SelectorMapping.__len__c C sB z| j �|�}| j j| W S ty< td�|��d �Y n0 d S �N�{!r} is not registered)r �_fileobj_lookupr �KeyErrorr )r r r r r r � __getitem__F s
z _SelectorMapping.__getitem__c C s t | jj�S r )�iterr r r! r r r �__iter__M s z_SelectorMapping.__iter__N)�__name__�
__module__� __qualname__�__doc__r r" r' r) r r r r r = s
r c @ sn e Zd ZdZeddd��Zedd� �Zddd�Zedd d
��Zd d � Z d
d� Z
edd� �Z dd� Z dd� Z
dS )� BaseSelectora- Selector abstract base class.
A selector supports registering file objects to be monitored for specific
I/O events.
A file object is a file descriptor or any object with a `fileno()` method.
An arbitrary object can be attached to the file object, which can be used
for example to store context information, a callback, etc.
A selector can use various implementations (select(), poll(), epoll()...)
depending on the platform. The default `Selector` class uses the most
efficient implementation on the current platform.
Nc C s t �dS )a3 Register a file object.
Parameters:
fileobj -- file object or file descriptor
events -- events to monitor (bitwise mask of EVENT_READ|EVENT_WRITE)
data -- attached data
Returns:
SelectorKey instance
Raises:
ValueError if events is invalid
KeyError if fileobj is already registered
OSError if fileobj is closed or otherwise is unacceptable to
the underlying system call (if a system call is made)
Note:
OSError may or may not be raised
N��NotImplementedError�r r r r r r r �register` s zBaseSelector.registerc C s t �dS )aj Unregister a file object.
Parameters:
fileobj -- file object or file descriptor
Returns:
SelectorKey instance
Raises:
KeyError if fileobj is not registered
Note:
If fileobj is registered but has since been closed this does
*not* raise OSError (even if the wrapped syscall does)
Nr/ )r r r r r �
unregisterw s zBaseSelector.unregisterc C s | � |� | �|||�S )ay Change a registered file object monitored events or attached data.
Parameters:
fileobj -- file object or file descriptor
events -- events to monitor (bitwise mask of EVENT_READ|EVENT_WRITE)
data -- attached data
Returns:
SelectorKey instance
Raises:
Anything that unregister() or register() raises
)r3 r2 r1 r r r �modify� s
zBaseSelector.modifyc C s t �dS )aq Perform the actual selection, until some monitored file objects are
ready or a timeout expires.
Parameters:
timeout -- if timeout > 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
Nr/ )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 | � � }|du rtd��z
|| W S tyD td�|��d�Y n0 dS )zzReturn the key associated to a registered file object.
Returns:
SelectorKey for this file object
NzSelector is closedr$ )�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.Nr/ r! r r r r8 � s zBaseSelector.get_mapc C s | S r r r! r r r � __enter__� s zBaseSelector.__enter__c G s | � � d S r )r7 )r �argsr r r �__exit__� s zBaseSelector.__exit__)N)N)N)r* r+ r, r- r r2 r3 r4 r6 r7 r; r8 r< r> r r r r r. Q s
r. )� metaclassc @ sT e Zd ZdZdd� Zdd� Zddd�Zd d
� Zdd d �Zd
d� Z dd� Z
dd� Z dS )�_BaseSelectorImplz Base selector implementation.c C s i | _ t| �| _d S r )r r �_mapr! r r r r � s z_BaseSelectorImpl.__init__c C sL z
t |�W S tyF | j�� D ] }|j|u r"|j Y S q"� Y n0 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| jv rRtd�||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 rD r r r r2 � s � z_BaseSelectorImpl.registerc C s> z| j �| �|��}W n" ty8 td�|��d �Y n0 |S r# )r �popr% r& r rC r r r r3 � s
z _BaseSelectorImpl.unregisterc C s� z| j | �|� }W n" ty6 td�|��d �Y n0 ||jkr\| �|� | �|||�}n"||jkr~|j|d�}|| j |j < |S )Nr$ )r )
r r% r& r r r3 r2 r �_replacer rG r r r r4 � s
z_BaseSelectorImpl.modifyc C s | j �� d | _d S r )r �clearrA r! r r r r7
s
z_BaseSelectorImpl.closec C s | j S r )rA r! r r r r8 s z_BaseSelectorImpl.get_mapc C s& z | j | W S ty Y dS 0 dS )z�Return 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% r2 r3 r4 r7 r8 rK r r r r r@ � s
r@ c s` e Zd ZdZ� fdd�Zd� fdd� Z� fdd�Zejd krHdd
d �Z ne
j
Z dd d
�Z
� Z S )�SelectSelectorzSelect-based selector.c 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 )rM r2 rE rO �addr rF rP rG rQ r r r2 + s zSelectSelector.registerc s, t � �|�}| j�|j� | j�|j� |S r )rM r3 rO �discardr rP rC rQ r r r3 3 s zSelectSelector.unregisterZwin32c C s$ t � ||||�\}}}||| g fS r )r6 )r �r�w�_r5 �xr r r �_select: s zSelectSelector._selectc C s� |d u r d nt |d�}g }z | �| j| jg |�\}}}W n tyN | Y S 0 t|�}t|�}||B D ]J}d}||v r�|tO }||v r�|tO }| �|�}|rh|� |||j
@ f� qh|S �Nr ) �maxrY rO rP �InterruptedErrorrN rE rF rK �appendr ) r r5 �readyrU rV rW r r rD r r r r6 @ s$
zSelectSelector.select)N)N)N) r* r+ r, r- r r2 r3 �sys�platformrY r6 �
__classcell__r r rQ r rL # s
rL c s^ e Zd ZdZdZdZdZ� fdd�Zd
� fdd� Z� fdd�Z d� fd d
� Z
dd d �Z � Z S )�_PollLikeSelectorz<Base class shared between poll, epoll and devpoll selectors.Nc s t � �� | �� | _d S r )rM r �
_selector_clsr r! rQ r r r \ s
z_PollLikeSelector.__init__c sl t � �|||�}d}|t@ r&|| jO }|t@ r8|| jO }z| j�|j|� W n t � �|� � Y n0 |S rZ ) rM r2 rE � _EVENT_READrF � _EVENT_WRITEr r r3 )r r r r rD Z
poller_eventsrQ r r r2 ` s
z_PollLikeSelector.registerc s6 t � �|�}z| j�|j� W n ty0 Y n0 |S r )rM r3 r r �OSErrorrC rQ r r r3 n s z _PollLikeSelector.unregisterc s� z| j | �|� }W n" ty6 t|�d��d �Y n0 d}||jkr�d}|t@ r\|| jO }|t@ rn|| jO }z| j� |j
|� W n t � � |� � Y n0 d}||j
kr�d}|r�|j||d�}|| j |j
< |S )Nz is not registeredFr T)r r )r r% r& r rE rd rF re r r4 r rM r3 r rI )r r r r rD ZchangedZselector_eventsrQ r r r4 x s.
z_PollLikeSelector.modifyc C s� |d u rd }n |dkr d}nt �|d �}g }z| j�|�}W n tyT | Y S 0 |D ]V\}}d}|| j @ rz|tO }|| j @ r�|tO }| � |�}|rZ|�
|||j @ f� qZ|S )Nr � @�@) �math�ceilr �pollr\ rd rF re rE rK r] r )r r5 r^ �
fd_event_listr �eventr rD r r r r6 � s(
z_PollLikeSelector.select)N)N)N)
r* r+ r, r- rc rd re r r2 r3 r4 r6 ra r r rQ r rb V s
rb rj c @ s" e Zd ZdZejZejZej Z
dS )� PollSelectorzPoll-based selector.N) r* r+ r, r- r6 rj rc �POLLINrd �POLLOUTre r r r r rm � s rm �epollc sD e Zd ZdZejZejZej Z
dd� Z d dd�Z� fdd�Z � Z
S )
�
EpollSelectorzEpoll-based selector.c C s
| j �� S r �r r
r! r r r r
� s zEpollSelector.filenoNc C s� |d u rd}n |dkr d}nt �|d �d }tt| j�d�}g }z| j�||�}W n tyj | Y S 0 |D ]V\}}d}|tj @ r�|t
O }|tj @ r�|t O }| �
|�}|rp|�|||j@ f� qp|S )N���r rg g����MbP?r )rh ri r[ r r r rj r\ r6 �EPOLLINrF �EPOLLOUTrE rK r] r ) r r5 �max_evr^ rk r rl r rD r r r r6 � s*
zEpollSelector.selectc s | j �� t� �� d S r �r r7 rM r! rQ r r r7 � s
zEpollSelector.close)N)r* r+ r, r- r6 rp rc rt rd ru re r
r7 ra r r rQ r rq � s
rq �devpollc s: e Zd ZdZejZejZej Z
dd� Z � fdd�Z � Z
S )�DevpollSelectorzSolaris /dev/poll selector.c C s
| j �� S r rr r! r r r r
� s zDevpollSelector.filenoc s | j �� t� �� d S r rw r! rQ r r r7 � s
zDevpollSelector.close)r* r+ r, r- r6 rx rc rn rd ro re r
r7 ra r r rQ r ry � s ry �kqueuec sX e Zd ZdZ� fdd�Zdd� Zd� fdd� Z� fd d
�Zdd d �Z� fd
d�Z � Z
S )�KqueueSelectorzKqueue-based selector.c s t � �� t�� | _d S r )rM r r6 rz r r! rQ r r r � s
zKqueueSelector.__init__c C s
| j �� S r rr 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 n0 |S rZ )
rM r2 rE r6 �keventr �KQ_FILTER_READZ KQ_EV_ADDr �controlrF �KQ_FILTER_WRITEr3 )r r r r rD �kevrQ r r r2 s � � zKqueueSelector.registerc s� t � �|�}|jt@ rTt�|jtjtj�}z| j �
|gdd� W n t yR Y n0 |jt @ r�t�|jtj
tj�}z| j �
|gdd� W n t y� Y n0 |S rZ )rM r3 r rE r6 r| r r} Z KQ_EV_DELETEr r~ rf rF r )r r rD r� rQ r r r3 s$
�
� zKqueueSelector.unregisterc
C s� |d u r d nt |d�}t t| j�d�}g }z| j�d ||�}W n tyT | Y S 0 |D ]Z}|j}|j}d}|tj kr�|t
O }|tj kr�|t O }| �
|�} | rZ|�| || j@ f� qZ|S )Nr r )r[ r r r r~ r\ Zident�filterr6 r} rE r rF rK r] r )
r r5 rv r^ Zkev_listr� r �flagr rD r r r r6 * s&
zKqueueSelector.selectc s | j �� t� �� d S r rw r! rQ r r r7 C s
zKqueueSelector.close)N)N) r* r+ r, r- r r
r2 r3 r6 r7 ra r r rQ r r{ � s
r{ c C sZ t t| d�}|du rdS z(|� }| dkr4|�d� n|�� W dS tyT Y dS 0 dS )zJCheck if we can use the selector depending upon the
operating system. NFrj r T)�getattrr6 rj r7 rf )�methodr Z selector_objr r r �_can_useH s r� ) r- �abcr r � collectionsr �collections.abcr rh r6 r_ rE rF r r � version_infor r r r r r. r@ rL rb �hasattrrm rq ry r{ r� ZDefaultSelectorr r r r �<module> sN
~T3Z
. M