3 \]@sddlmZddlZddlZddlmZddlmZddlZeedZ Gdddej Z Gdd d e Z e ejejee r|e ndejd ZdS) )absolute_importN)queue)Eventwebsocket_handshakecs.eZdZdZgiffdd ZddZZS)Threadz This wrapper class provides gevent Greenlet interface that is compatible with the standard library's Thread class. cstt|j|f||dS)N)superr__init__)selftargetargskwargs) __class__N/tmp/pip-build-mqc4i71p/python-engineio/engineio/async_drivers/gevent_uwsgi.pyrszThread.__init__cCs|jS)N)run)r rrr_runsz Thread._run)__name__ __module__ __qualname____doc__rr __classcell__rr)r rr src@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)uWSGIWebSocketzx This wrapper class provides a uWSGI WebSocket interface that is compatible with eventlet's implementation. cCs||_d|_dS)N)app_sock)r rrrrrszuWSGIWebSocket.__init__cstj|_||_tjd|_ttdr4tj|_nTddlm }ddl m }ddl m ||_||_fdd}tj||j|j|_|j|dS)Nrequest_contextr)r)Queue)selectc s@x:|jy|gggdWqtk r6PYqXqWdS)z5Sets event when data becomes available to read on fd.rN)set ValueError)fdevent)rrrselect_greenlet_runner4s z7uWSGIWebSocket.__call__..select_greenlet_runner)uwsgiZ connection_fdrenvironr_req_ctxhasattrr gevent.eventrZ gevent.queuerZ gevent.selectr_event _send_queuegeventspawn_select_greenletr)r r#Zstart_responserrr!r)rr__call__!s"        zuWSGIWebSocket.__call__cCs*tj|jdkr&|jj|jjdS)z"Disconnects uWSGI from the client.N)r"Z disconnectr$r+killr'r)r rrrcloseCs  zuWSGIWebSocket.closecCs@t|tjrtj}ntj}|jdk r4|||jdn||dS)zUTransmits message either in binary or UTF-8 text mode, depending on its type.N)r) isinstancesix binary_typer"Zwebsocket_send_binaryZwebsocket_sendr$)r msgmethodrrr_sendKs   zuWSGIWebSocket._sendcCs8t|tjs|Stj|dd}|dkr4|jdS|S)z7Returns either bytes or str, depending on message type.r0zutf-8)r/r0r1byte2intdecode)r r2typerrr_decode_receivedWs   zuWSGIWebSocket._decode_receivedcCs0|jdk r|j|n|jj||jjdS)zQueues a message for sending. Real transmission is done in wait method. Sends directly if uWSGI version is new enough.N)r$r4r(putr'r)r r2rrrsendcs   zuWSGIWebSocket.sendc Csx|jdk r>ytj|jd}Wntk r2dSX|j|S|jjdd}|r|jjg}x8y|j|j j ddWq`t j j k rPYq`Xq`Wx|D]}|j|qWy tj}Wntk r|jjdSX|r|j|SqWdS)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.N)r)timeoutF)block)r$r"Zwebsocket_recvIOErrorr:r'waitclearappendr(getr)rEmptyr4Zwebsocket_recv_nbr+r-)r r2Z event_setZmsgsrrrrAms2       zuWSGIWebSocket.waitN) rrrrrr,r.r4r:r<rArrrrrs"   r)threadrZ queue_emptyr Z websocketsleep) __future__rr0r)rr&rr"r%Z_websocket_availableZGreenletrobjectrZ JoinableQueuerErGZ_asyncrrrrs     ~