iBddlZddlmZddlmZddlmZddlZeedZGddejZ Gdd Z e ej ej eere ndej d ZdS) N)queue)Event) selectorswebsocket_handshakec.eZdZdZgiffd ZdZxZS)Threadz This wrapper class provides gevent Greenlet interface that is compatible with the standard library's Thread class. cBtj|g|Ri|dSN)super__init__)selftargetargskwargs __class__s N/usr/local/lib/python3.11/dist-packages/engineio/async_drivers/gevent_uwsgi.pyr zThread.__init__s01$111&11111c*|Sr )runr s r_runz Thread._runsxxzzr)__name__ __module__ __qualname____doc__r r __classcell__)rs@rrr s]%'r222222rrc<eZdZdZdZdZdZdZdZdZ dZ d S) uWSGIWebSocketzx This wrapper class provides a uWSGI WebSocket interface that is compatible with eventlet's implementation. c0||_d|_g|_dSr )app_sockreceived_messages)r handlerservers rr zuWSGIWebSocket.__init__s !#rctj|_||_tjd|_t tdrtj|_nSt|_ tj |_ d}tj||j|j |_||tjdS)Nrequest_contextctj}||tj ||)#t j$r||YdSwxYw)z5Sets event when data becomes available to read on fd.TN) rDefaultSelectorregister EVENT_READselectsetgevent GreenletExit unregister)fdeventsels rselect_greenlet_runnerz7uWSGIWebSocket.__call__..select_greenlet_runner/s/11 R!5666'$  $*'''NN2&&&&&&'s*A$BB)uwsgi connection_fdr!environr_req_ctxhasattrr&r_eventrQueue _send_queuer-spawn_select_greenletr disconnect)r r7start_responser3s r__call__zuWSGIWebSocket.__call__s(**   !### 5+ , , !133DMM ''DK${}}D  ' ' '%+L&  %%D !  rrc||j4|j|jdSdS)z"Disconnects uWSGI from the client.N)r8r>killr:r,rs rclosezuWSGIWebSocket.closeBs? =  ! & & ( ( ( KOO      ! rct|tr tj}n tj}|j|||jdS||dS)zUTransmits message either in binary or UTF-8 text mode, depending on its type.Nr&) isinstancebytesr5websocket_send_binarywebsocket_sendr8)r msgmethods r_sendzuWSGIWebSocket._sendIs^ c5 ! ! *0FF)F = $ F3 6 6 6 6 6 6 F3KKKKKrct|ts|St|dd}|dkr|dS|S)z7Returns either bytes or str, depending on message type.r0zutf-8)rGrHorddecode)r rKtypes r_decode_receivedzuWSGIWebSocket._decode_receivedUsL#u%% J3qs8}} 2::::g&& & rc|j||dS|j||jdS)zQueues a message for sending. Real transmission is done in wait method. Sends directly if uWSGI version is new enough.N)r8rMr<putr:r,)r rKs rsendzuWSGIWebSocket.sendasN = $ JJsOOOOO    % % % KOO     rc |jV tj|j}n%#t$r|YdSwxYw||S|jr|jdS|j d}|r|j g} | |j dn#tjj$rYnwxYwK|D]>} ||#t$r|YdSwxYw tj}n%#t$r|YdSwxYw|r.|j ||nnl|jr|jdS) aTWaits and returns received messages. If running in compatibility mode for older uWSGI versions, it also sends messages that have been queued by send(). A return value of None means that connection was closed. This must be called repeatedly. For uWSGI < 2.1.x it must be called from the main greenlet.TNrFr)timeoutF)block)r8r5websocket_recvOSErrorrDrTr"popr:waitclearappendr<getr-rEmptyrMwebsocket_recv_nb)r rK event_setmsgss rr_zuWSGIWebSocket.waitks:, 9}( .t}MMMCC   JJLLL44 ,,S111)9155a888!K,,Q,77 (K%%'''D"" KK(8(<(<5(<(I(IJJJJ%|1"""!E"" $((( JJsOOOO&((( JJLLL#'444( $#577"$$$ #tt$.55 11#668888 )9155a888Y, 9sE%AA:.C))D?D DEEEE=<E=N) rrrrr rArDrMrTrWr_rrrrs$$$ !!!F      3939393939rr)threadr queue_emptyr1 websocketsleep)r-r gevent.eventrrr5r9_websocket_availableGreenletrr JoinableQueuercrk_asyncrgrrrqs  wu&;<<     V_   I9I9I9I9I9I9I9I9Z  ; #7AT \   r