U W[G@sdZddlmZmZddlmZddlZddlZddlZddl m Z m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd d Zed kreZnejZzddlmZWnek reZYnXeZe eGdddejeZddZdgZ dS)z Select reactor )divisionabsolute_import)sleepN)EINTREBADF) implementer) IReactorFDSet) posixbase)log) platformTypecCsj|s2|s2|dkrd}n t|d}t|gggfS|dksB|dkrFd}t||||\}}}|||gfS)zWin32 select wrapper.Ng{Gz?gMbP?g?)minrselect)rwetimeoutr@/usr/lib/python3/dist-packages/twisted/internet/selectreactor.py win32selects  rZwin32)_ThreadedWin32EventsMixinc@sleZdZdZddZddZddZeZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZdS) SelectReactora- A select() based reactor - runs on all POSIX platforms and on Win32. @ivar _reads: A set containing L{FileDescriptor} instances which will be checked for read events. @ivar _writes: A set containing L{FileDescriptor} instances which will be checked for writability. cCs t|_t|_tj|dS)zV Initialize file descriptor tracking dictionaries and the base class. N)set_reads_writesr PosixReactorBase__init__selfrrrrBszSelectReactor.__init__c Cstdt|j}t|j}|j|j|j|f|j|ffD]v\}}|D]h}zt|g|g|gdWn>tk r}z td||||dW5d}~XYqRX| |qRqFdS)Nz1Malformed file descriptor found. Preening lists.rzbad descriptor %sF) r msglistrrclearr Exception_disconnectSelectableadd)rZreadersZwritersZselSetZselList selectablerrrr_preenDescriptorsKs       zSelectReactor._preenDescriptorsc CsJzt|j|jg|\}}}Wntk r:|YdStk r^t|YdStj t j t fk r}zf|j ddkr|js|jsWYFdSn:|j dt krWY*dS|j dtkr|WY dSW5d}~XYnX|j}tj}|d|jf|d|jffD]4\}} } |D]"} | | kr0q|| || | qqdS)z Run one iteration of the I/O monitor loop. This will run all selectables who had input or output readiness waiting for them. Nr)rdoReadZdoWrite)_selectrr ValueErrorr% TypeErrorr errr errorsocketIOErrorargsrr_doReadOrWriteZcallWithLogger) rrrrZignoredZseZ_drdwZ_logrunZ selectablesmethodZfdsetr$rrrdoSelect]sB       zSelectReactor.doSelectcCsLzt||}Wn td}tYnX|rH||||dkdS)Nr')getattrsysexc_infor r+r")rr$r1Zwhyrrrr0s zSelectReactor._doReadOrWritecCs|j|dS)zR Add a FileDescriptor for notification of data available to read. N)rr#rreaderrrr addReaderszSelectReactor.addReadercCs|j|dS)zS Add a FileDescriptor for notification of data available to write. N)rr#rwriterrrr addWriterszSelectReactor.addWritercCs|j|dS)zQ Remove a Selectable for notification of data available to read. N)rdiscardr7rrr removeReaderszSelectReactor.removeReadercCs|j|dS)zR Remove a Selectable for notification of data available to write. N)rr=r:rrr removeWriterszSelectReactor.removeWritercCs||j|jSN)Z _removeAllrrrrrr removeAllszSelectReactor.removeAllcCs t|jSr@)rrrrrr getReadersszSelectReactor.getReaderscCs t|jSr@)rrrrrr getWritersszSelectReactor.getWritersN)__name__ __module__ __qualname____doc__rr%r2Z doIterationr0r9r<r>r?rArBrCrrrrr6s  4 rcCst}ddlm}||dS)zIConfigure the twisted mainloop to be run using the select() reactor. r)installReactorN)rZtwisted.internet.mainrH)ZreactorrHrrrinstalls rI)N)!rGZ __future__rrtimerr5r r-errnorrZzope.interfacerZtwisted.internet.interfacesrZtwisted.internetr Ztwisted.pythonr Ztwisted.python.runtimer rr(Z"twisted.internet.win32eventreactorr ImportErrorobjectZ _extraBaserrrI__all__rrrrs.