3 \]i@sddlZy ddlZWnek r0ddlZYnXddlZddlZddlZddlZddlZddl m Z y ddl Z Wnek rdZ YnXy ddl Z Wnek rdZ YnXddl mZddl mZddl mZejdZgZejreZdd ZejejeZGd d d eZdS) N)urllib) exceptions)packet)payloadzengineio.clientcCs^x:tddD]*}|jr,|j|jddq|jddqWttrNt||Stj||SdS)zdSIGINT handler. Disconnect all active clients and then invoke the original signal handler. NT)abort)connected_clientsis_asyncio_basedstart_background_task disconnectcallableoriginal_signal_handlersignaldefault_int_handler)sigframeclientr:/tmp/pip-build-mqc4i71p/python-engineio/engineio/client.pysignal_handler s rc@seZdZdZdddgZd=d d Zd d Zd>d dZiddfddZddZ d?ddZ d@ddZ ddZ ddZ dAddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,ZdBd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<ZdS)CClientaAn Engine.IO client. This class implements a fully compliant Engine.IO web client with support for websocket and long-polling transports. :param logger: To enable logging set to ``True`` or pass a logger object to use. To disable logging set to ``False``. The default is ``False``. :param json: An alternative json module to use for encoding and decoding packets. Custom json modules must have ``dumps`` and ``loads`` functions that are compatible with the standard library versions. :param request_timeout: A timeout in seconds for requests. The default is 5 seconds. :param ssl_verify: ``True`` to verify SSL certificates, or ``False`` to skip SSL certificate verification, allowing connections to servers with self signed certificates. The default is ``True``. connectr messageFNTcCsi|_d|_d|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_||_|dk r||tj_t|ts||_nPt|_tjj r|jjtjkr|r|jjtjn|jjtj|jj tj!||_"dS)NT disconnected)#handlersbase_url transportscurrent_transportsidupgrades ping_interval ping_timeout pong_receivedhttpwsread_loop_taskwrite_loop_taskping_loop_taskping_loop_eventqueuestate ssl_verifyrPacketjson isinstanceboolloggerdefault_loggerloggingrootlevelNOTSETsetLevelINFOERROR addHandler StreamHandlerrequest_timeout)selfr1r.r<r,rrr__init__Js<  zClient.__init__cCsdS)NFr)r=rrrr rszClient.is_asyncio_basedcs8jkrtdfdd}|dkr,|S||dS)aRegister an event handler. :param event: The event name. Can be ``'connect'``, ``'message'`` or ``'disconnect'``. :param handler: The function that should be invoked to handle the event. When this parameter is not given, the method acts as a decorator for the handler function. Example usage:: # as a decorator: @eio.on('connect') def connect_handler(): print('Connection request') # as a method: def message_handler(msg): print('Received message: ', msg) eio.send('response') eio.on('message', message_handler) z Invalid eventcs|j<|S)N)r)handler)eventr=rr set_handlers zClient.on..set_handlerN) event_names ValueError)r=r@r?rAr)r@r=ronus  z Client.onz engine.iocs|jdkrtdddg|dk rRt|tjr4|g}fdd|D}|sRtd|pX|_|j|_t|d |jd |||S) aConnect to an Engine.IO server. :param url: The URL of the Engine.IO server. It can include custom query string parameters if required by the server. :param headers: A dictionary with custom headers to send with the connection request. :param transports: The list of allowed transports. Valid transports are ``'polling'`` and ``'websocket'``. If not given, the polling transport is connected first, then an upgrade to websocket is attempted. :param engineio_path: The endpoint where the Engine.IO server is installed. The default value is appropriate for most cases. Example usage:: eio = engineio.Client() eio.connect('http://localhost:5000') rz%Client is not in a disconnected statepolling websocketNcsg|]}|kr|qSrr).0 transport)valid_transportsrr sz"Client.connect..zNo valid transports providedZ _connect_r) r+rCr/six string_typesr create_queuer*getattr)r=urlheadersr engineio_pathr)rIrrs    zClient.connectcCs|jr|jjdS)zWait until the connection with the server ends. Client applications can use this function to block the main thread during the life of the connection. N)r&join)r=rrrwaitsz Client.waitcCs|jtjtj||ddS)aSend a message to a client. :param data: The data to send to the client. Data can be of type ``str``, ``bytes``, ``list`` or ``dict``. If a ``list`` or ``dict``, the data will be serialized as JSON. :param binary: ``True`` to send packet as binary, ``False`` to send as text. If not given, unicode (Python 2) and str (Python 3) are sent as text, and str (Python 2) and bytes (Python 3) are sent as binary. )databinaryN) _send_packetrr-MESSAGE)r=rTrUrrrsends z Client.sendc Cs|jdkr|jtjtj|jjdd|_|jddd|jdkrP|j j |s^|j j d|_yt j|Wntk rYnX|jdS) zDisconnect from the server. :param abort: If set to ``True``, do not wait for background tasks associated with the connection to end. connectedNZ disconnectingr F) run_asyncrFr)r+rVrr-CLOSEr*put_trigger_eventrr%closer&rRrremoverC_reset)r=rrrrr s     zClient.disconnectcCs|jS)zReturn the name of the transport currently in use. The possible values returned by this function are ``'polling'`` and ``'websocket'``. )r)r=rrrrHszClient.transportcOstj|||d}|j|S)aStart a background task. This is a utility function that applications can use to start a background task. :param target: the target function to execute. :param args: arguments to pass to the function. :param kwargs: keyword arguments to pass to the function. This function returns an object compatible with the `Thread` class in the Python standard library. The `start()` method on this object is already called by this function. )targetargskwargs) threadingThreadstart)r=rarbrcthrrrr szClient.start_background_taskrcCs tj|S)z'Sleep for the requested amount of time.)timesleep)r=secondsrrrrisz Client.sleepcOstj||}tj|_|S)zCreate a queue object.)r*QueueEmpty)r=rbrcqrrrrMs zClient.create_queuecOs tj||S)zCreate an event object.)rdEvent)r=rbrcrrr create_event szClient.create_eventcCsd|_d|_dS)Nr)r+r)r=rrrr`sz Client._resetc Cstdkr|jjddS|j||d|_|jjd|j|jd|j|j||jd}|dkrr|j t j d|j dks|j d krt j d j |j ytj|jd }Wn&tk rtjt j d dYnX|jd }|jtjkrt j d|jjdt|j|jd|_|jd|_|jdd|_|jdd|_d|_|jd|j7_d|_t j!||j"dddx"|jddD]}|j#|qWd|jkrd|j$kr|j%|||rdS|j&|j'|_(|j&|j)|_*|j&|j+|_,dS)z. cert_reqs)headerrZsslopt)rrz*WebSocket upgrade failed: connection errorzConnection errorZprobe)rTz7WebSocket upgrade failed: unexpected send exception: %sz7WebSocket upgrade failed: unexpected recv exception: %s)encoded_packetz(WebSocket upgrade failed: no PONG packetz WebSocket upgrade was successfulzUnexpected recv exception: zno OPEN packetz#WebSocket connection accepted with rr rtg@@rurYr)rZ)1rFr1warningrxrryrr$rRcookiesr,create_connectionr{ssl CERT_NONEr|IOErrorrrr-PINGrK text_typeencoderX ExceptionrrecvrPONGrTUPGRADErrr r!r"r+rrr]r%r rr(rr'_read_loop_websocketr&) r=rOrPrQZ websocket_urlupgraderr%rerrrrrrLs                    zClient._connect_websocketcCs|jttjkrtj|jnd}|jjd|t|jts<|jnd|jtj krb|j d|jddnL|jtj krvd|_ n8|jtj kr|jddn|jtjkrn|jjd|jd S) z(Handle incoming packets from the server.UNKNOWNzReceived packet %s data %szrT)rZ)rz%Received unexpected packet of type %sN)rlenr packet_namesr1ryr/rTbytesrWr]rr#r[r ZNOOPrw)r=rZ packet_namerrrrs     zClient._receive_packetcCsH|jdkrdS|jj||jjdtj|jt|j t s>|j nddS)z(Queue a packet to be sent to the server.rYNzSending packet %s data %sz) r+r*r\r1ryrrrr/rTr)r=rrrrrVs   zClient._send_packetcCsl|jdkrtj|_y|jj||||||jdStjjk rf}z|jjd|||WYdd}~XnXdS)N)rPrTrqverifyz+HTTP %s request to %s failed with error %s.) r$rvSessionrequestr,rRequestExceptionr1ry)r=methodrOrPbodyrqexcrrrrzs   zClient._send_requestc Os`|jdd}||jkr\|r0|j|j|f|Sy|j||S|jj|dYnXdS)zInvoke an event handler.rZFz handler errorN)poprr r1 exception)r=r@rbrcrZrrrr]s  zClient._trigger_eventcCsp|jd}tjj|}|dkr$d}n|dkr2d}ntd|jdkrL|d 7}d j||j||j|jrfd nd |d S)z&Generate the Engine.IO connection URL./rEr$rFr%zinvalid transporthttpswssszC{scheme}://{netloc}/{path}/?{query}{sep}transport={transport}&EIO=3&)schemenetlocpathqueryseprH)rr) striprparseurlparserCrr~rr)r=rOrQrH parsed_urlrrrrrxs    zClient._get_engineio_urlcCsdttjS)z.Generate the Engine.IO query string timestamp.z&t=)rrh)r=rrrr{szClient._get_url_timestampcCsd|_|jdkr|j|_n |jjxf|jdkr|jsb|jjd|jrT|jj|j j dPd|_|j t j t j|jj|jdq(W|jjddS)z[This background task sends a PING to the server at the requested interval. TNrYz-PONG response has not been received, abortingF)rqzExiting ping task)r#r)roclearr+r1ryr%shutdownr*r\rVrr-rrSr!)r=rrrrs       zClient._ping_loopc Csx|jdkr|jjd|j|jd|j|jt|j|jdd}|dkrh|jj d|j j dP|j dks||j d kr|jj d |j |j j dPyt j|jd }Wn.tk r|jj d |j j dPYnXx|jD]}|j|qWqW|jjd |jj|jjd|jr.|jj|jj|jdkr|jdddytj|Wntk rvYnX|j|jjddS)z-Read packets by polling the Engine.IO server.rYzSending polling GET request to rpr)rqNz*Connection refused by the server, abortingrri,z6Unexpected status code %s in server response, aborting)rsz'Unexpected packet from server, abortingz"Waiting for write loop task to endz!Waiting for ping loop task to endr F)rZzExiting read loop task)r+r1ryrrzr{maxr!r"rr*r\r}rrrrCrrr'rRr)setr(r]rr_r`)r=rrrrrrrsN            zClient._read_loop_pollingcCsTx|jdkrd}y|jj}Wnptjk rN|jjd|jjdPYnBt k r}z&|jj dt ||jjdPWYdd}~XnXt |t jr|jd}tj|d}|j|qW|jj d|jj|jj d|jr|jj|jj|jdkrD|jd d d ytj|Wntk r:YnX|j|jj d dS) z5Read packets from the Engine.IO WebSocket connection.rYNz)WebSocket connection was closed, abortingzUnexpected error "%s", abortingzutf-8)rz"Waiting for write loop task to endz!Waiting for ping loop task to endr F)rZzExiting read loop task)r+r%rrF"WebSocketConnectionClosedExceptionr1rr*r\rryrr/rKrrrr-rr'rRr)rr(r]rr_rCr`)r=rrrrrrrBs@             zClient._read_loop_websocketcCsx|jdkrt|j|jd}d}y|jj|dg}Wn&|jjk r`|jjdPYnX|dgkr||jj g}n^x\y|j |jjddWn|jjk rPYnX|ddkr~|dd}|jj Pq~W|sP|j d kr~t j |d }|jd |j|jd d i|jd}x|D]}|jj qW|dkrJ|jjdP|jdksb|jdkr|jjd|j|jPqyLxF|D]>}|jdd}|jr|jj|n |jj||jj qWWqtjk r|jjdPYqXqW|jjddS)zhThis background task sends packages to the server as they are pushed to the send queue. rYrN)rqzpacket queue is empty, abortingF)blockrrE)rPOSTz Content-Typezapplication/octet-stream)rrPrqz*Connection refused by the server, abortingrri,z6Unexpected status code %s in server response, aborting)Z always_bytesz)WebSocket connection was closed, abortingzExiting write loop taskr)r+rr!r"r*getrlr1rw task_donerrrrrzrrr<rr}r`rUr%Z send_binaryrXrFrry)r=rqrrrrrrrrrfsf                zClient._write_loop)FNrT)N)N)F)r)NNN)__name__ __module__ __qualname____doc__rBr>r rDrrSrXr rHr rirMror`rrrrVrzr]rxr{rrrrrrrrr4s@  $ ! #   9g   +$r)r3r* ImportErrorrkrrrdrhrKZ six.movesrrvrFrrrr getLoggerr2rPY2OSErrorr|rSIGINTr objectrrrrrs8