YfK@s*dZddlmZmZddlmZmZddlZddlZddl Z d)Z d*Z ddZ edd d d d gZ d e _e jd+krde j_de j_de j_de j_GdddeZGddddeZGdddeZGdddeZeedrQGdddeZeed rvGd!d"d"eZeed#rGd$d%d%eZeed&rGd'd(d(eZd(ekreZnNd"ekreZn6d%ekreZndekr eZneZdS),z|Selectors module. This module allows high-level and efficient I/O multiplexing, built upon the `select` module primitives. )ABCMetaabstractmethod) namedtupleMappingNc Cst|tr|}nLyt|j}Wn3tttfk rctdj|dYnX|dkrtdj||S)zReturn 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}NrzInvalid file descriptor: {}) isinstanceintfilenoAttributeError TypeError ValueErrorformat)fileobjfdr./opt/alt/python35/lib64/python3.5/selectors.py_fileobj_to_fds    r SelectorKeyrreventsdatazSelectorKey(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@sFeZdZdZddZddZddZdd Zd S) _SelectorMappingz)Mapping of file objects to selector keys.cCs ||_dS)N) _selector)selfZselectorrrr__init__>sz_SelectorMapping.__init__cCst|jjS)N)lenr _fd_to_key)rrrr__len__Asz_SelectorMapping.__len__c CsUy$|jj|}|jj|SWn*tk rPtdj|dYnXdS)Nz{!r} is not registered)r_fileobj_lookuprKeyErrorr )rrrrrr __getitem__Ds  z_SelectorMapping.__getitem__cCst|jjS)N)iterrr)rrrr__iter__Ksz_SelectorMapping.__iter__N)__name__ __module__ __qualname____doc__rrr!r#rrrrr;s    rc@seZdZdZedddZeddZdddZedd 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. NcCs tdS)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.registercCs tdS)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.modifycCs tdS)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 Csa|j}|dkr$tdy ||SWn*tk r\tdj|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_keys     zBaseSelector.get_keycCs tdS)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__)r$r%r&r'rr*r+r,r.r/r3r0r4r6rrrrr(Os    r( metaclassc@s|eZdZdZddZddZdddZd d Zdd d Zd dZ ddZ ddZ dS)_BaseSelectorImplzBase selector implementation.cCsi|_t||_dS)N)rr_map)rrrrrs z_BaseSelectorImpl.__init__c CsZyt|SWnEtk rUx-|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_lookupNcCs| s|ttB@r+tdj|t||j|||}|j|jkrvtdj||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 CsPy|jj|j|}Wn*tk rKtdj|dYnX|S)Nz{!r} is not registered)rpoprr r )rrr;rrrr+s  z_BaseSelectorImpl.unregisterc Csy|j|j|}Wn*tk rFtdj|dYnX||jkr{|j||j|||}n1||jkr|jd|}||j|j <|S)Nz{!r} is not registeredr) 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|SWntk r'dSYnXdS)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) r$r%r&r'rrr*r+r,r/r0rArrrrr8s      r8cseZdZdZfddZdfddZfddZejd krldd d Z n e j Z dd d Z S)SelectSelectorzSelect-based selector.cs)tjt|_t|_dS)N)superrset_readers_writers)r) __class__rrr%s  zSelectSelector.__init__NcsVtj|||}|t@r5|jj|j|t@rR|jj|j|S)N)rCr*r<rEaddrr=rF)rrrrr;)rGrrr**s   zSelectSelector.registercs<tj|}|jj|j|jj|j|S)N)rCr+rEdiscardrrF)rrr;)rGrrr+2szSelectSelector.unregisterwin32cCs2tj||||\}}}|||gfS)N)r.)rrw_r-xrrr_select9s!zSelectSelector._selectc Cs|dkrdn t|d}g}y+|j|j|jg|\}}}Wntk rj|SYnXt|}t|}xs||BD]g}d}||kr|tO}||kr|tO}|j|}|r|j |||j @fqW|S)Nr) maxrOrErFInterruptedErrorrDr<r=rAappendr) rr-readyrKrLrMrrr;rrrr.?s$!+        zSelectSelector.select) r$r%r&r'rr*r+sysplatformrOr.rr)rGrrB"s  rBpollcs^eZdZdZfddZdfddZfddZdd d ZS) PollSelectorzPoll-based selector.cs tjtj|_dS)N)rCrr.rV_poll)r)rGrrrZs zPollSelector.__init__Ncsftj|||}d}|t@r5|tjO}|t@rL|tjO}|jj|j||S)Nr) rCr*r<r.POLLINr=POLLOUTrXr)rrrrr; poll_events)rGrrr*^s    zPollSelector.registercs)tj|}|jj|j|S)N)rCr+rXr)rrr;)rGrrr+hszPollSelector.unregisterc Cs|dkrd}n(|dkr*d}ntj|d}g}y|jj|}Wntk rq|SYnXxy|D]q\}}d}|tj@r|tO}|tj@r|t O}|j |}|ry|j |||j @fqyW|S)Nrg@@) mathceilrXrVrQr.rYr=rZr<rArRr)rr-rS fd_event_listreventrr;rrrr.ms(        zPollSelector.select)r$r%r&r'rr*r+r.rr)rGrrWWs  rWepollcs|eZdZdZfddZddZdfddZfd d Zdd d Zfd dZ S) EpollSelectorzEpoll-based selector.cs tjtj|_dS)N)rCrr.r`_epoll)r)rGrrrs zEpollSelector.__init__cCs |jjS)N)rbr )rrrrr szEpollSelector.filenoNc stj|||}d}|t@r5|tjO}|t@rL|tjO}y|jj|j|Wn%t k rtj |YnX|S)Nr) rCr*r<r.EPOLLINr=EPOLLOUTrbr BaseExceptionr+)rrrrr;Z epoll_events)rGrrr*s     zEpollSelector.registerc sBtj|}y|jj|jWntk r=YnX|S)N)rCr+rbrOSError)rrr;)rGrrr+s  zEpollSelector.unregisterc Cs|dkrd}n,|dkr*d}ntj|dd}tt|jd}g}y|jj||}Wntk r|SYnXxy|D]q\}}d}|tj @r|t O}|tj @r|t O}|j |}|r|j|||j@fqW|S)Nrrg@@gMbP?)r\r]rPrrrbrVrQr.rcr=rdr<rArRr) rr-max_evrSr^rr_rr;rrrr.s*        zEpollSelector.selectcs|jjtjdS)N)rbr/rC)r)rGrrr/s zEpollSelector.close) r$r%r&r'rr r*r+r.r/rr)rGrras    radevpollcs|eZdZdZfddZddZdfddZfd d Zdd d Zfd dZ S)DevpollSelectorzSolaris /dev/poll selector.cs tjtj|_dS)N)rCrr.ri_devpoll)r)rGrrrs zDevpollSelector.__init__cCs |jjS)N)rkr )rrrrr szDevpollSelector.filenoNcsftj|||}d}|t@r5|tjO}|t@rL|tjO}|jj|j||S)Nr) rCr*r<r.rYr=rZrkr)rrrrr;r[)rGrrr*s    zDevpollSelector.registercs)tj|}|jj|j|S)N)rCr+rkr)rrr;)rGrrr+szDevpollSelector.unregisterc Cs|dkrd}n(|dkr*d}ntj|d}g}y|jj|}Wntk rq|SYnXxy|D]q\}}d}|tj@r|tO}|tj@r|t O}|j |}|ry|j |||j @fqyW|S)Nrg@@) r\r]rkrVrQr.rYr=rZr<rArRr)rr-rSr^rr_rr;rrrr.s(        zDevpollSelector.selectcs|jjtjdS)N)rkr/rC)r)rGrrr/s zDevpollSelector.close) r$r%r&r'rr r*r+r.r/rr)rGrrjs   rjkqueuecs|eZdZdZfddZddZdfddZfd d Zdd d Zfd dZ S)KqueueSelectorzKqueue-based selector.cs tjtj|_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 @rtj|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@rotj|jtjtj}y|j j |gddWnt k rnYnX|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|dkrdn t|d}t|j}g}y|jjd||}Wntk rj|SYnXx|D]}|j}|j}d}|tj kr|t O}|tj kr|t O}|j |} | rr|j| || j@fqrW|S)Nr)rPrrrnrqrQZidentfilterr.rpr<rrr=rArRr) rr-rhrSZkev_listrsrZflagrr;rrrr.<s&!       zKqueueSelector.selectcs|jjtjdS)N)rnr/rC)r)rGrrr/Rs zKqueueSelector.close) r$r%r&r'rr r*r+r.r/rr)rGrrm s  rmr)rr)r'abcrr collectionsrrr\r.rTr<r=rr version_inforrrrrr(r8rBhasattrrWrarjrmglobalsZDefaultSelectorrrrrsH         ~U31G8M