File: //usr/lib/python3/dist-packages/twisted/internet/__pycache__/asyncioreactor.cpython-310.pyc
o
�b{+ � @ s� d Z ddlZddlZddlmZmZ ddlmZmZm Z ddl
m Z ddl m
Z
ddlmZ ddlmZmZmZ dd lmZ dd
lmZ e e�G d d � d e��Zdd
d�ZdS )z'
asyncio-based reactor implementation.
� N)�AbstractEventLoop�get_event_loop)�Dict�Optional�Type)� implementer)�FileDescriptor)�
IReactorFDSet)� _NO_FILEDESC�PosixReactorBase�_ContinuousPolling)�Logger)�callWithLoggerc s� e Zd ZdZdZe� Zd,dee f� fdd�
Z dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd-d d �Z� fd d �Z� fd d!�Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Z� Z S ).�AsyncioSelectorReactora�
Reactor running on top of L{asyncio.SelectorEventLoop}.
On POSIX platforms, the default event loop is
L{asyncio.SelectorEventLoop}.
On Windows, the default event loop on Python 3.7 and older
is C{asyncio.WindowsSelectorEventLoop}, but on Python 3.8 and newer
the default event loop is C{asyncio.WindowsProactorEventLoop} which
is incompatible with L{AsyncioSelectorReactor}.
Applications that use L{AsyncioSelectorReactor} on Windows
with Python 3.8+ must call
C{asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())}
before instantiating and running L{AsyncioSelectorReactor}.
FN� eventloopc sx |d u rt � }n|}tjdkr!ddlm} t||�r!td|� ���|| _i | _i | _ t
| �| _ d | _ d | _
t� �� d S )N�win32r )�ProactorEventLoopz)ProactorEventLoop is not supported, got: )r �sys�platform�asyncior �
isinstance� TypeError�_asyncioEventloop�_writers�_readersr �_continuousPolling� _scheduledAt� _timerHandle�super�__init__)�selfr �
_eventloopr �� __class__� �A/usr/lib/python3/dist-packages/twisted/internet/asyncioreactor.pyr 0 s
�
z AsyncioSelectorReactor.__init__c C s* z
| j j�|� W dS ty Y dS w )am
Compensate for a bug in asyncio where it will not unregister a FD that
it cannot handle in the epoll loop. It touches internal asyncio code.
A description of the bug by markrwilliams:
The C{add_writer} method of asyncio event loops isn't atomic because
all the Selector classes in the selector module internally record a
file object before passing it to the platform's selector
implementation. If the platform's selector decides the file object
isn't acceptable, the resulting exception doesn't cause the Selector to
un-track the file object.
The failing/hanging stdio test goes through the following sequence of
events (roughly):
* The first C{connection.write(intToByte(value))} call hits the asyncio
reactor's C{addWriter} method.
* C{addWriter} calls the asyncio loop's C{add_writer} method, which
happens to live on C{_BaseSelectorEventLoop}.
* The asyncio loop's C{add_writer} method checks if the file object has
been registered before via the selector's C{get_key} method.
* It hasn't, so the KeyError block runs and calls the selector's
register method
* Code examples that follow use EpollSelector, but the code flow holds
true for any other selector implementation. The selector's register
method first calls through to the next register method in the MRO
* That next method is always C{_BaseSelectorImpl.register} which
creates a C{SelectorKey} instance for the file object, stores it under
the file object's file descriptor, and then returns it.
* Control returns to the concrete selector implementation, which asks
the operating system to track the file descriptor using the right API.
* The operating system refuses! An exception is raised that, in this
case, the asyncio reactor handles by creating a C{_ContinuousPolling}
object to watch the file descriptor.
* The second C{connection.write(intToByte(value))} call hits the
asyncio reactor's C{addWriter} method, which hits the C{add_writer}
method. But the loop's selector's get_key method now returns a
C{SelectorKey}! Now the asyncio reactor's C{addWriter} method thinks
the asyncio loop will watch the file descriptor, even though it won't.
N)r � _selector�
unregister�
BaseException)r �fdr$ r$ r% �_unregisterFDInAsyncioK s
2 �z-AsyncioSelectorReactor._unregisterFDInAsyncioc
C s� |r|j n|j}|�� dkr| �|t|� d S z|� }W n ty6 } z|}| j�d � W Y d }~nd }~ww |rB| �|||� d S d S �N���)�doRead�doWrite�fileno�_disconnectSelectabler
� Exception�_log�failure)r �
selectable�read�method�why�er$ r$ r% � _readOrWrite� s
���z#AsyncioSelectorReactor._readOrWritec
C s� || j �� v s
|| jj v rd S |�� }z| j�|t|| j|d� || j |< W d S tyM } z| � |� |j
t
j krA| j� |� n� W Y d }~d S d }~ww )NT)
r �keysr r/ r �
add_readerr r9 �OSErrorr* �errno�EPERM� addReader)r �readerr) r8 r$ r$ r% r? � s �
���z AsyncioSelectorReactor.addReaderc C s� || j �� v s
|| jj v rd S |�� }z| j�|t|| j|d� || j |< W d S ty< | � |� | j�
|� Y d S t yJ | � |� Y d S t yV | � |� � w )NF)
r r: r r/ r �
add_writerr r9 �PermissionErrorr* � addWriter�BrokenPipeErrorr( �r �writerr) r$ r$ r% rC � s" �
�z AsyncioSelectorReactor.addWriterc C �t || j �� v s| j�|�sd S | j�|�r | j�|� d S |�� }|dkr,| j �|�}n| j �|� | j�|� d S r+ ) r r: r � isReading� removeReaderr/ �popr �
remove_reader)r r@ r) r$ r$ r% rI � s z#AsyncioSelectorReactor.removeReaderc C rG r+ ) r r: r � isWriting� removeWriterr/ rJ r �
remove_writerrE r$ r$ r% rM � s z#AsyncioSelectorReactor.removeWriterc C s" | � | j�� | j�� �| j�� S �N)�
_removeAllr r: r r � removeAll�r r$ r$ r% rQ � s ��z AsyncioSelectorReactor.removeAllc C � t | j�� �| j�� S rO )�listr r: r �
getReadersrR r$ r$ r% rU � � z!AsyncioSelectorReactor.getReadersc C rS rO )rT r r: r �
getWritersrR r$ r$ r% rW � rV z!AsyncioSelectorReactor.getWritersc C s$ | j �|d | j j� | j �� d S )Ng{�G�z�?)r �
call_later�stop� run_forever)r �timeoutr$ r$ r% �iterate� s z AsyncioSelectorReactor.iterateTc C s* | j |d� | j�� | jrd| _d S d S )N)�installSignalHandlersF)� startRunningr rZ � _justStopped)r r] r$ r$ r% �run� s
�zAsyncioSelectorReactor.runc s t � �� | �ddd� � d S )Nr c S s d S rO r$ r$ r$ r$ r% �<lambda> s z-AsyncioSelectorReactor.stop.<locals>.<lambda>)r rY � callLaterrR r" r$ r% rY s
zAsyncioSelectorReactor.stopc s t � �� | j�� d S rO )r �crashr rY rR r" r$ r% rc s
z AsyncioSelectorReactor.crashc C s d | _ | �� | �� d S rO )r �runUntilCurrent� _reschedulerR r$ r$ r% �_onTimer
s z AsyncioSelectorReactor._onTimerc C sR | � � }|d ur'| j�� | }|| _| jd ur | j�� | j�|| j�| _d S d S rO )r[ r �timer r �cancel�call_atrf )r r[ �abs_timer$ r$ r% re s
�z"AsyncioSelectorReactor._reschedulec C s t �| |� | �� d S rO )r �_moveCallLaterSoonerre )r �tpler$ r$ r% rk s z+AsyncioSelectorReactor._moveCallLaterSoonerc O sN t j| ||g|�R i |��}| j�� | �� }| jd u s!|| jk r%| �� |S rO )r rb r rg r[ r re )r �seconds�f�args�kwargs�dcrj r$ r$ r% rb s
z AsyncioSelectorReactor.callLaterc s" � ���fdd�}�j �|� d S )Nc s �j d�g� �R i ���S )Nr )rb r$ �ro rn rp r r$ r% ra ' s z7AsyncioSelectorReactor.callFromThread.<locals>.<lambda>)r �call_soon_threadsafe)r rn ro rp �gr$ rr r% �callFromThread& s z%AsyncioSelectorReactor.callFromThreadrO )T) �__name__�
__module__� __qualname__�__doc__� _asyncClosedr
r2 r r r r* r9 r? rC rI rM rQ rU rW r\ r` rY rc rf re rk rb ru �
__classcell__r$ r$ r" r% r s. 7
r c C s t | �}ddlm} ||� dS )z�
Install an asyncio-based reactor.
@param eventloop: The asyncio eventloop to wrap. If default, the global one
is selected.
r )�installReactorN)r �twisted.internet.mainr| )r �reactorr| r$ r$ r% �install+ s r rO )ry r= r r r r �typingr r r �zope.interfacer �twisted.internet.abstractr �twisted.internet.interfacesr �twisted.internet.posixbaser
r r �twisted.loggerr
�twisted.python.logr r r r$ r$ r$ r% �<module> s