관리-도구
편집 파일: selectors.cpython-36.pyc
3 �Pf�R � @ sz d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ y ej Z W n e e fk rh ejZ Y nX d(Z d)Z dZe� ZdaG dd� de�Zdd� Zejd*kr�d d � Znd d � Zedddddg�ZG dd� de�ZG dd� de�Zeed��rG dd� de�Zeed��r G dd � d e�Z eed ��r:G d d � d e�Z eed ��rVG d!d"� d"e�Z eed��sfd#Zd$d%� Z d&d'� Z dS )+� N)� namedtuple�Mapping� Tc s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )� SelectorErrorc s t t| �j� || _d S )N)�superr �__init__�errno)�self�errcode)� __class__� � /usr/lib/python3.6/selectors.pyr s zSelectorError.__init__c C s dj | j�S )Nz<SelectorError errno={0}>)�formatr )r r r r �__repr__" s zSelectorError.__repr__c C s | j � S )N)r )r r r r �__str__% s zSelectorError.__str__)�__name__� __module__� __qualname__r r r � __classcell__r r )r r r s r c C sd t | t�r| }n:yt| j� �}W n( tttfk rH tdj| ���Y nX |dk r`tdj|���|S )zl Return a file descriptor from a file object. If given an integer will simply return that integer back. zInvalid file object: {0!r}r z Invalid file descriptor: {0})� isinstance�int�fileno�AttributeError� TypeError� ValueErrorr )�fileobj�fdr r r �_fileobj_to_fd) s r � � c O sV y | ||�S t ttjfk rP } z"d}t|d�r8|j}t|��W Y dd}~X nX dS )z� This is the short-circuit version of the below logic because in Python 3.5+ all system calls automatically restart and recalculate their timeouts. Nr )�OSError�IOError�select�error�hasattrr r )�func�_�args�kwargs�er r r r �_syscall_wrapper; s r* c O sR |j dd�}|dkr d}d}n t|�}|dk r4d}n t� | }t|�}|rZd|krZtd��t}x�|tk�rLy| ||�}W q` tttj fk �rH } z�d}t |d�r�|j }nt |d�r�|j d }|t j kp�t t d �o�|t jk} | �r&|dk �r$t� } | |k�rtt jd ��|�r$d|k�r$|| |d< w`|�r6t|��n� W Y dd}~X q`X q`W |S ) z� Wrapper function for syscalls that could fail due to EINTR. All functions should be retried if there is time left in the timeout in accordance with PEP 475. �timeoutNFg z4Timeout must be in args or kwargs to be recalculatedr r' r �WSAEINTR)r )�get�float� monotonic�listr �_SYSCALL_SENTINELr r! r" r# r$ r r' ZEINTRr, Z ETIMEDOUTr ) r% Zrecalc_timeoutr' r( r+ Zexpires�resultr) r Z is_interruptZ current_timer r r r* G sJ � SelectorKeyr r �events�datac @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �_SelectorMappingz* Mapping of file objects to selector keys c C s || _ d S )N)� _selector)r Zselectorr r r r � s z_SelectorMapping.__init__c C s t | jj�S )N)�lenr7 � _fd_to_key)r r r r �__len__� s z_SelectorMapping.__len__c C s@ y| j j|�}| j j| S tk r: tdj|���Y nX d S )Nz{0!r} is not registered.)r7 �_fileobj_lookupr9 �KeyErrorr )r r r r r r � __getitem__� s z _SelectorMapping.__getitem__c C s t | jj�S )N)�iterr7 r9 )r r r r �__iter__� s z_SelectorMapping.__iter__N)r r r �__doc__r r: r= r? r r r r r6 � s r6 c @ sv e Zd ZdZdd� Zdd� Zddd�Zd d � Zd d d �Zd d d�Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS ) � BaseSelectora/ Abstract Selector 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 info, a callback, etc. A selector can use various implementations (select(), poll(), epoll(), and kqueue()) depending on the platform. The 'DefaultSelector' class uses the most efficient implementation for the current platform. c C s i | _ t| �| _d S )N)r9 r6 �_map)r r r r r � s zBaseSelector.__init__c C sH yt |�S tk rB x"| jj� D ]}|j|kr$|jS q$W � Y nX dS )aa 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. 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 r9 �valuesr r )r r �keyr r r r; � s z BaseSelector._fileobj_lookupNc C sd | s|t tB @ r"tdj|���t|| j|�||�}|j| jkrTtdj||j���|| j|j< |S )z8 Register a file object for a set of events to monitor. zInvalid events: {0!r}z${0!r} (FD {1}) is already registered) � EVENT_READ� EVENT_WRITEr r r3 r; r r9 r< )r r r4 r5 rD r r r �register� s zBaseSelector.registerc C s� y| j j| j|��}W n� tk r8 tdj|���Y np tjk r� } zR|jtjkrZ� n<x:| j j � D ] }|j |krf| j j|j � P qfW tdj|���W Y dd}~X nX |S )z0 Unregister a file object from being monitored. z{0!r} is not registeredN) r9 �popr; r< r �socketr# r ZEBADFrC r r )r r rD r) r r r � unregister� s zBaseSelector.unregisterc C s� y| j | j|� }W n" tk r6 tdj|���Y nX ||jkr\| j|� | j|||�}n"||jkr~|j|d�}|| j |j < |S )z< Change a registered file object monitored events and data. z{0!r} is not registered)r5 ) r9 r; r< r r4 rJ rG r5 �_replacer )r r r4 r5 rD r r r �modify� s zBaseSelector.modifyc C s t � �dS )zj Perform the actual selection until some monitored file objects are ready or the timeout expires. N)�NotImplementedError)r r+ r r r r" � s zBaseSelector.selectc C s | j j� d| _dS )zd Close the selector. This must be called to ensure that all underlying resources are freed. N)r9 �clearrB )r r r r �close� s zBaseSelector.closec C sH | j � }|dkrtd��y|| S tk rB tdj|���Y nX dS )z: Return the key associated with a registered file object. NzSelector is closedz{0!r} is not registered)�get_map� RuntimeErrorr< r )r r �mappingr r r �get_key s zBaseSelector.get_keyc C s | j S )z3 Return a mapping of file objects to selector keys )rB )r r r r rP s zBaseSelector.get_mapc C s$ y | j | S tk r dS X dS )z_ Return the key associated to a given file descriptor Return None if it is not found. N)r9 r< )r r r r r � _key_from_fd s zBaseSelector._key_from_fdc C s | S )Nr )r r r r � __enter__ s zBaseSelector.__enter__c G s | j � d S )N)rO )r r' r r r �__exit__ s zBaseSelector.__exit__)N)N)N)r r r r@ r r; rG rJ rL r" rO rS rP rT rU rV r r r r rA � s rA r" c sN e Zd ZdZ� fdd�Zd � fdd� Z� fdd�Zdd d �Zdd d �Z� Z S )�SelectSelectorz Select-based selector. c s" t t| �j� t� | _t� | _d S )N)r rW r �set�_readers�_writers)r )r r r r ! s zSelectSelector.__init__Nc sD t t| �j|||�}|t@ r*| jj|j� |t@ r@| jj|j� |S )N) r rW rG rE rY �addr rF rZ )r r r4 r5 rD )r r r rG & s zSelectSelector.registerc s0 t t| �j|�}| jj|j� | jj|j� |S )N)r rW rJ rY �discardr rZ )r r rD )r r r rJ . s zSelectSelector.unregisterc C s t j ||g |�S )z? Wrapper for select.select because timeout is a positional arg )r" )r �r�wr+ r r r �_select4 s zSelectSelector._selectc C s� t | j� r t | j� r g S |d kr(d nt|d�}g }t| jd| j| j|�\}}}t|�}t|�}xV||B D ]J}d}||kr�|tO }||kr�|tO }| j |�}|rl|j |||j @ f� qlW |S )Ng Tr ) r8 rY rZ �maxr* r_ rX rE rF rT �appendr4 ) r r+ �readyr] r^ r& r r4 rD r r r r" 8 s$ zSelectSelector.select)N)N)N) r r r r@ r rG rJ r_ r" r r r )r r rW s rW �pollc sN e Zd ZdZ� fdd�Zd � fdd� Z� fdd�Zdd d �Zdd d �Z� Z S )� PollSelectorz Poll-based selector c s t t| �j� tj� | _d S )N)r rd r r" rc �_poll)r )r r r r S s zPollSelector.__init__Nc sP t t| �j|||�}d}|t@ r*|tjO }|t@ r<|tjO }| jj|j |� |S )Nr ) r rd rG rE r" �POLLINrF �POLLOUTre r )r r r4 r5 rD � event_mask)r r r rG W s zPollSelector.registerc s" t t| �j|�}| jj|j� |S )N)r rd rJ re r )r r rD )r r r rJ a s zPollSelector.unregisterc C s4 |dk r$|dkrd}nt j|d �}| jj|�}|S )zj Wrapper function for select.poll.poll() so that _syscall_wrapper can work with only seconds. Nr g @�@)�math�ceilre rc )r r+ r2 r r r � _wrap_pollf s zPollSelector._wrap_pollc C sx g }t | jd|d�}x^|D ]V\}}d}|tj @ r:|tO }|tj @ rN|tO }| j|�}|r|j|||j @ f� qW |S )NT)r+ r ) r* rk r" rf rF rg rE rT ra r4 )r r+ rb � fd_eventsr rh r4 rD r r r r" t s zPollSelector.select)N)N)N) r r r r@ r rG rJ rk r" r r r )r r rd Q s rd �epollc sX e Zd ZdZ� fdd�Zdd� Zd� fdd� Z� fd d �Zdd d �Z� fd d�Z � Z S )� EpollSelectorz Epoll-based selector c s t t| �j� tj� | _d S )N)r rn r r" rm �_epoll)r )r r r r � s zEpollSelector.__init__c C s | j j� S )N)ro r )r r r r r � s zEpollSelector.filenoNc sT t t| �j|||�}d}|t@ r*|tjO }|t@ r<|tjO }t| j jd|j |� |S )Nr F) r rn rG rE r" �EPOLLINrF �EPOLLOUTr* ro r )r r r4 r5 rD Z events_mask)r r r rG � s zEpollSelector.registerc s@ t t| �j|�}yt| jjd|j� W n tk r: Y nX |S )NF)r rn rJ r* ro r r )r r rD )r r r rJ � s zEpollSelector.unregisterc C s� |d k r2|dkrd}nt j|d �d }t|�}nd }tt| j�d�}g }t| jjd||d�}x^|D ]V\}}d}|t j @ r�|t O }|t j @ r�|t O }| j|�}|rd|j|||j@ f� qdW |S ) Nr g g @�@g����MbP?g �?r T)r+ Z maxeventsg �)ri rj r. r` r8 r9 r* ro rc r" rp rF rq rE rT ra r4 ) r r+ � max_eventsrb rl r rh r4 rD r r r r" � s* zEpollSelector.selectc s | j j� tt| �j� d S )N)ro rO r rn )r )r r r rO � s zEpollSelector.close)N)N) r r r r@ r r rG rJ r" rO r r r )r r rn � s !rn �kqueuec sX e Zd ZdZ� fdd�Zdd� Zd� fdd� Z� fd d �Zdd d �Z� fd d�Z � Z S )�KqueueSelectorz Kqueue / Kevent-based selector c s t t| �j� tj� | _d S )N)r rt r r" rs �_kqueue)r )r r r r � s zKqueueSelector.__init__c C s | j j� S )N)ru r )r r r r r � s zKqueueSelector.filenoNc s| t t| �j|||�}|t@ rFtj|jtjtj�}t | j j d|gdd� |t @ rxtj|jtj tj�}t | j j d|gdd� |S )NFr )r rt rG rE r" �keventr �KQ_FILTER_READZ KQ_EV_ADDr* ru �controlrF �KQ_FILTER_WRITE)r r r4 r5 rD rv )r r r rG � s zKqueueSelector.registerc s� t t| �j|�}|jt@ r^tj|jtjtj �}yt | j j d|gdd� W n t k r\ Y nX |jt@ r�tj|jtjtj �}yt | j j d|gdd� W n t k r� Y nX |S )NFr )r rt rJ r4 rE r" rv r rw Z KQ_EV_DELETEr* ru rx r rF ry )r r rD rv )r r r rJ � s$ zKqueueSelector.unregisterc C s� |d k rt |d�}t| j�d }i }t| jjdd ||�}x�|D ]�}|j}|j}d}|tj krd|t O }|tj krv|t O }| j |�} | r>| j|kr�| || j@ f|| j< q>|| j d } | || B | j@ f|| j< q>W t|j� �S )Nr � Tr )r` r8 r9 r* ru rx Zident�filterr" rw rE ry rF rT r r4 r0 rC ) r r+ rr Z ready_fdsZ kevent_listrv r rh r4 rD Z old_eventsr r r r" � s* zKqueueSelector.selectc s | j j� tt| �j� d S )N)ru rO r rt )r )r r r rO s zKqueueSelector.close)N)N) r r r r@ r r rG rJ r" rO r r r )r r rt � s rt Fc C sZ y0| dkr t j� }|jd� ntt | �� j� dS ttfk rT } zdS d}~X nX dS )a Checks that select structs can be allocated by the underlying operating system, not just advertised by the select module. We don't check select() because we'll be hopeful that most platforms that don't have it available will not advertise it. (ie: GAE) rc r TFN)r" rc �getattrrO r r )�struct�pr) r r r � _can_allocate s r c C sP t dkrJtd�rta n4td�r$ta n&td�r2ta nttd�rBta ntd��t � S )z� This function serves as a first call for DefaultSelector to detect if the select module is being monkey-patched incorrectly by eventlet, greenlet, and preserve proper behavior. Nrs rm rc r" z!Platform does not have a selector) �_DEFAULT_SELECTORr rt rn rd r$ r" rW r r r r r �DefaultSelector5 s r� r rz )r r )!r ri r" rI �sysZtime� collectionsr r r/ r � ImportErrorrE rF Z HAS_SELECT�objectr1 r� � Exceptionr r � version_infor* r3 r6 rA r$ rW rd rn rt r r� r r r r �<module> sH 8 1 4 B R