o
i 1 @ sB d Z ddlmZ ddlmZ ddlmZ ddlmZ zddlZW n e y/ ddl
ZdZ Y nw ddl m
Z dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ dgZ dgZ e e Z eee e ddZ eeB Z!ej"Z"ej#Z#e"e#B Z$ej%Z%e&edej'Z(G dd de(Z)e)Z*dd Z+dS )an
This module provides :class:`GeventSelector`, a high-level IO
multiplexing mechanism. This is aliased to :class:`DefaultSelector`.
This module provides the same API as the selectors defined in :mod:`selectors`.
On Python 2, this module is only available if the `selectors2
`_ backport is installed.
.. versionadded:: 20.6.0
)absolute_import)division)print_function) defaultdictN
selectors2)_get_hub_noargs)sleep) iteritems)
itervalues) copy_globals)Lazy)Event)_EV_READ) _EV_WRITEDefaultSelectorGeventSelector)__all__)names_to_ignoredunder_names_to_keep_BaseSelectorImplc sl e Zd ZdZd fdd Zeedsdd Zdd Ze d
d Z
dd d
Z dd Z dddZ
dd Z ZS )r a
A selector implementation using gevent primitives.
This is a type of :class:`selectors.BaseSelector`, so the documentation
for that class applies here.
.. caution::
As the base class indicates, it is critically important to
unregister file objects before closing them. (Or close the selector
they are registered with before closing them.) Failure to do so
may crash the process or have other unintended results.
Nc s> |d ur|| _ i | _i | _tt| _t | _tt |
d S N) hub_active_watchers_inactive_watchersr int_accumulated_eventsr
_readysuperr __init__)selfr __class__ X/var/www/apps/myagent/mysuperagent/venv/lib/python3.10/site-packages/gevent/selectors.pyr n s
zGeventSelector.__init__ _key_from_fdc C s" z| j | W S ty Y d S w r )
_fd_to_keyKeyError)r fdr" r" r# r$ z s
zGeventSelector._key_from_fdc C sJ |dkr | j | }|t@ r|tO }|t@ r|tO }|| j |< | j d S )Nr )r r
EVENT_READr EVENT_WRITEr set)r eventsr' cur_event_for_fdr" r" r#
__callback s
zGeventSelector.__callbackc C s t S r )get_hub)r r" r" r# r s zGeventSelector.hubc
C s t | |||}|tkrt}n |tkrt}nt}| jj}|j }|j
}z||j | | j |j < } W n t
yA }
ztd|
d}
~
ww || _|S )a
Register a file object for selection, monitoring it for I/O events.
*fileobj* is the file object to monitor. It may either be an integer file descriptor
or an object with a ``fileno()`` method. *events* is a bitwise mask of events to
monitor. *data* is an opaque object.
:return: A new `SelectorKey` instance.
:raises ValueError: In case of invalid
event mask or file descriptor
:raises KeyError: if the file object is already registered.
.. versionchanged:: 25.8.1
More reliably raises a ``ValueError`` if the file descriptor
is invalid.
zInvalid file descriptorN)r register _ALL_EVENTS _POLL_ALLr( r r r loopioMAXPRIr' r OSError
ValueErrorpriority) r fileobjr+ datakeyflagsr2 r3 r4 watcherer" r" r# r/ s"
zGeventSelector.registerc C sZ t | |}|j| jv r| j|j}n| j|j}| | | j|jd |S r ) r
unregisterr' r popr stopcloser )r r8 r: r< r" r" r# r> s zGeventSelector.unregisterc C s |dur
|dk r
d}t | jD ]
\}}|j| j|dd q| j| j | j | j r1t | j
| | j g }t | j D ])\}}| |}| j
|}| |rh||||j@ f || j|< qC| qC| j |S )z
Poll for I/O.
Note that, like the built-in selectors, this will block
indefinitely if no timeout is given and no files have been
registered.
Nr T) pass_events)r r start_GeventSelector__callbackr updateclearr is_setr waitr r$ r? r@ appendr+ rA )r timeoutr' r< resulteventr: r" r" r# select s*
zGeventSelector.selectc C s` | j | jfD ]}|d u r
qt|D ]
}| | qqd | _ | _d | _d | _t| d S r )r r r
r@ rA r r r )r dr< r" r" r# rA s
zGeventSelector.closer )__name__
__module__ __qualname____doc__r hasattrr r$ rD r r r/ r> rM rA
__classcell__r" r" r r# r E s (
'
>c s | j d }| j}| dd l}d|jvr |jt |jd< | dd}|j|us*J t}|j }t
|drO|j |j|fv rOddl m fdd}||j _ d |_
|r\| |d
d d t|_t
|d
rst
|jdruddl m} ||j_d S d S d S )N
aggressiver selectorsrM _selectrM c s |i |S r r" )r argskwargsrX r" r# rW 8 s z(_gevent_do_monkey_patch.._selectT
EpollSelectorKqueueSelectorDevpollSelector PollSelector
_selector_cls)poll) patch_kwargs
target_moduledefault_patch_itemssysmodulesrO get_originalrM
__selectors__SelectSelectorrS rW
gevent.select_gevent_monkey remove_itemr r^ r` r_ )
patch_requestrU
target_modrd orig_select_selectrV rh rW r` r" rX r# _gevent_do_monkey_patch s:
ro ),rR
__future__r r r collectionsr rV rg ImportErrorr
__target__
gevent.hubr r. geventr gevent._compatr r
gevent._utilr r gevent.eventr
ri r r __implements__ __extra__r globals __imports__r1 r( r) r0 SelectorKeygetattr BaseSelectorr r r ro r" r" r" r# sX O