3 \K@sdZddlmZmZddlmZmZddlZddlZddl Z d)Z d*Z ddZ edd d d d gZ d e _e jd+krde j_de j_de j_de j_GdddeZGdddedZGdddeZGdddeZeedrGdddeZeed rGd!d"d"eZeed#r"Gd$d%d%eZeed&r>Gd'd(d(eZd(ekrPeZn:d"ekrbeZn(d%ekrteZndekreZneZdS),z|Selectors module. This module allows high-level and efficient I/O multiplexing, built upon the `select` module primitives. )ABCMetaabstractmethod) namedtupleMappingNc Csft|tr|}nsz_SelectorMapping.__init__cCs t|jjS)N)lenr _fd_to_key)rrrr__len__Asz_SelectorMapping.__len__c CsBy|jj|}|jj|Stk r<tdj|dYnXdS)Nz{!r} is not registered)r_fileobj_lookuprKeyErrorr )rrrrrr __getitem__Ds   z_SelectorMapping.__getitem__cCs t|jjS)N)iterrr)rrrr__iter__Ksz_SelectorMapping.__iter__N)__name__ __module__ __qualname____doc__rrr!r#rrrrr;s rc@sneZdZdZedddZeddZdddZedd d Zd d Z d dZ eddZ ddZ ddZ 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. NcCstdS)a3Register 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)rrrrrrrregister^szBaseSelector.registercCstdS)ajUnregister 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) N)r))rrrrr unregisteruszBaseSelector.unregistercCs|j||j|||S)ayChange 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 )r+r*)rrrrrrrmodifys zBaseSelector.modifycCstdS)aqPerform 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 N)r))rtimeoutrrrselectszBaseSelector.selectcCsdS)zmClose the selector. This must be called to make sure that any underlying resource is freed. Nr)rrrrcloseszBaseSelector.closec CsJ|j}|dkrtdy||Stk rDtdj|dYnXdS)zzReturn the key associated to a registered file object. Returns: SelectorKey for this file object NzSelector is closedz{!r} is not registered)get_map RuntimeErrorr r )rrmappingrrrget_keyszBaseSelector.get_keycCstdS)z2Return a mapping of file objects to selector keys.N)r))rrrrr0szBaseSelector.get_mapcCs|S)Nr)rrrr __enter__szBaseSelector.__enter__cGs |jdS)N)r/)rargsrrr__exit__szBaseSelector.__exit__)N)N)N)r$r%r&r'rr*r+r,r.r/r3r0r4r6rrrrr(Os      r() metaclassc@sTeZdZdZddZddZdddZd d Zdd d Zd dZ ddZ ddZ dS)_BaseSelectorImplzBase selector implementation.cCsi|_t||_dS)N)rr_map)rrrrrsz_BaseSelectorImpl.__init__c CsHyt|Stk rBx"|jjD]}|j|kr$|jSq$WYnXdS)alReturn 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)rr rvaluesrr)rrkeyrrrrs   z!_BaseSelectorImpl._fileobj_lookupNcCsd| s|ttB@r"tdj|t||j|||}|j|jkrTtdj||j||j|j<|S)NzInvalid events: {!r}z"{!r} (FD {}) is already registered) EVENT_READ EVENT_WRITEr r rrrrr )rrrrr;rrrr*s   z_BaseSelectorImpl.registerc Cs@y|jj|j|}Wn$tk r:tdj|dYnX|S)Nz{!r} is not registered)rpoprr r )rrr;rrrr+s z_BaseSelectorImpl.unregisterc Csy|j|j|}Wn$tk r8tdj|dYnX||jkr^|j||j|||}n"||jkr|j|d}||j|j <|S)Nz{!r} is not registered)r) rrr r rr+r*r_replacer)rrrrr;rrrr,s     z_BaseSelectorImpl.modifycCs|jjd|_dS)N)rclearr9)rrrrr/ s z_BaseSelectorImpl.closecCs|jS)N)r9)rrrrr0sz_BaseSelectorImpl.get_mapc Cs$y |j|Stk rdSXdS)zReturn the key associated to a given file descriptor. Parameters: fd -- file descriptor Returns: corresponding key, or None if not found N)rr )rrrrr _key_from_fds  z_BaseSelectorImpl._key_from_fd)N)N) r$r%r&r'rrr*r+r,r/r0rArrrrr8s  r8cs`eZdZdZfddZdfdd ZfddZejd krHdd d Z ne j Z dd d Z Z S)SelectSelectorzSelect-based selector.cstjt|_t|_dS)N)superrset_readers_writers)r) __class__rrr%s zSelectSelector.__init__Ncs@tj|||}|t@r&|jj|j|t@r<|jj|j|S)N)rCr*r<rEaddrr=rF)rrrrr;)rGrrr**s zSelectSelector.registercs,tj|}|jj|j|jj|j|S)N)rCr+rEdiscardrrF)rrr;)rGrrr+2s zSelectSelector.unregisterZwin32cCs$tj||||\}}}|||gfS)N)r.)rrw_r-xrrr_select9szSelectSelector._selectc Cs|dkr dnt|d}g}y|j|j|jg|\}}}Wntk rL|SXt|}t|}xV||BD]J}d}||kr|tO}||kr|tO}|j|}|rh|j |||j @fqhW|S)Nr) maxrNrErFInterruptedErrorrDr<r=rAappendr) rr-readyrJrKrLrrr;rrrr.?s$ zSelectSelector.select)N)N)N) r$r%r&r'rr*r+sysplatformrNr. __classcell__rr)rGrrB"s    rBpollcsDeZdZdZfddZd fdd ZfddZd d d ZZS) PollSelectorzPoll-based selector.cstjtj|_dS)N)rCrr.rV_poll)r)rGrrrZs zPollSelector.__init__NcsLtj|||}d}|t@r&|tjO}|t@r8|tjO}|jj|j||S)Nr) rCr*r<r.POLLINr=POLLOUTrXr)rrrrr; poll_events)rGrrr*^s  zPollSelector.registercstj|}|jj|j|S)N)rCr+rXr)rrr;)rGrrr+hs zPollSelector.unregisterc Cs|dkrd}n|dkrd}ntj|d}g}y|jj|}Wntk rR|SXx^|D]V\}}d}|tj@rz|tO}|tj@r|t O}|j |}|rZ|j |||j @fqZW|S)Nrg@@) mathceilrXrVrPr.rYr=rZr<rArQr)rr-rR fd_event_listreventrr;rrrr.ms(   zPollSelector.select)N)N) r$r%r&r'rr*r+r.rUrr)rGrrWWs   rWepollcsXeZdZdZfddZddZdfdd Zfd d Zdd d Zfd dZ Z S) EpollSelectorzEpoll-based selector.cstjtj|_dS)N)rCrr.r`_epoll)r)rGrrrs zEpollSelector.__init__cCs |jjS)N)rbr )rrrrr szEpollSelector.filenoNc sttj|||}d}|t@r&|tjO}|t@r8|tjO}y|jj|j|Wn"t k rntj |YnX|S)Nr) rCr*r<r.EPOLLINr=EPOLLOUTrbr BaseExceptionr+)rrrrr;Z epoll_events)rGrrr*s   zEpollSelector.registerc s8tj|}y|jj|jWntk r2YnX|S)N)rCr+rbrOSError)rrr;)rGrrr+s  zEpollSelector.unregisterc Cs|dkrd}n |dkrd}ntj|dd}tt|jd}g}y|jj||}Wntk rh|SXx^|D]V\}}d}|tj @r|t O}|tj @r|t O}|j |}|rp|j|||j@fqpW|S)Nrrg@@gMbP?)r\r]rOrrrbrVrPr.rcr=rdr<rArQr) rr-max_evrRr^rr_rr;rrrr.s*   zEpollSelector.selectcs|jjtjdS)N)rbr/rC)r)rGrrr/s zEpollSelector.close)N)N) r$r%r&r'rr r*r+r.r/rUrr)rGrras  radevpollcsXeZdZdZfddZddZdfdd Zfd d Zdd d Zfd dZ Z S)DevpollSelectorzSolaris /dev/poll selector.cstjtj|_dS)N)rCrr.ri_devpoll)r)rGrrrs zDevpollSelector.__init__cCs |jjS)N)rkr )rrrrr szDevpollSelector.filenoNcsLtj|||}d}|t@r&|tjO}|t@r8|tjO}|jj|j||S)Nr) rCr*r<r.rYr=rZrkr)rrrrr;r[)rGrrr*s  zDevpollSelector.registercstj|}|jj|j|S)N)rCr+rkr)rrr;)rGrrr+s zDevpollSelector.unregisterc Cs|dkrd}n|dkrd}ntj|d}g}y|jj|}Wntk rR|SXx^|D]V\}}d}|tj@rz|tO}|tj@r|t O}|j |}|rZ|j |||j @fqZW|S)Nrg@@) r\r]rkrVrPr.rYr=rZr<rArQr)rr-rRr^rr_rr;rrrr.s(   zDevpollSelector.selectcs|jjtjdS)N)rkr/rC)r)rGrrr/s zDevpollSelector.close)N)N) r$r%r&r'rr r*r+r.r/rUrr)rGrrjs   rjkqueuecsXeZdZdZfddZddZdfdd Zfd d Zdd d Zfd dZ Z S)KqueueSelectorzKqueue-based selector.cstjtj|_dS)N)rCrr.rl_kqueue)r)rGrrrs zKqueueSelector.__init__cCs |jjS)N)rnr )rrrrr szKqueueSelector.filenoNc stj|||}y`|t@r@tj|jtjtj}|jj |gdd|t @rntj|jtj tj}|jj |gddWn"t k rtj |YnX|S)Nr)rCr*r<r.keventrKQ_FILTER_READZ KQ_EV_ADDrncontrolr=KQ_FILTER_WRITErer+)rrrrr;kev)rGrrr*s   zKqueueSelector.registercstj|}|jt@rVtj|jtjtj}y|j j |gddWnt k rTYnX|jt @rtj|jtj tj}y|j j |gddWnt k rYnX|S)Nr)rCr+rr<r.rorrpZ KQ_EV_DELETErnrqrfr=rr)rrr;rs)rGrrr+'s      zKqueueSelector.unregisterc Cs|dkr dnt|d}t|j}g}y|jjd||}Wntk rL|SXxb|D]Z}|j}|j}d}|tj krz|t O}|tj kr|t O}|j |} | rT|j| || j@fqTW|S)Nr)rOrrrnrqrPZidentfilterr.rpr<rrr=rArQr) rr-rhrRZkev_listrsrflagrr;rrrr.<s&     zKqueueSelector.selectcs|jjtjdS)N)rnr/rC)r)rGrrr/Rs zKqueueSelector.close)N)N) r$r%r&r'rr r*r+r.r/rUrr)rGrrm s   rmr)rr)r'abcrr collectionsrrr\r.rSr<r=rr version_inforrrrrr(r8rBhasattrrWrarjrmglobalsZDefaultSelectorrrrrsH ~U3 1 G 8 M