B ‹æ@\¢/ã@sÞdZddlmZmZmZmZddlZe e¡Z ddl Z ddl m Z ddl mZmZddlmZmZmZddlmZd d lmZd d lmZmZmZd d lmZd d lmZddl m!Z!ddl"m#Z#dZ$Gdd„deƒZ%dZ&dS)zA Provide a web socket handler for the Bokeh Server application. é)Úabsolute_importÚdivisionÚprint_functionÚunicode_literalsN)Úurlparse)ÚgenÚlocks)ÚStreamClosedErrorÚWebSocketHandlerÚWebSocketClosedErroré)ÚProtocolHandleré)ÚProtocol)Ú MessageErrorÚ ProtocolErrorÚValidationError)ÚMessage)ÚReceiver)Úcheck_session_id_signature)Úsettings)Ú WSHandlercsÀeZdZdZ‡fdd„Zdd„Zdd„Zdd „Zej d d „ƒZ ej d d „ƒZ dd„Z ej dd„ƒZ ej d"‡fdd„ ƒZdd„Zej dd„ƒZej dd„ƒZej dd„ƒZdd„Zd d!„Z‡ZS)#rzI Implements a custom Tornado WebSocketHandler for the Bokeh Server. csHd|_d|_d|_|d|_d|_t ¡|_tt |ƒj |f|ž|ŽdS)NÚapplication_contextéÿÿÿÿ) ÚreceiverÚhandlerÚ connectionrÚ latest_pongrZLockÚ write_lockÚsuperrÚ__init__)ÚselfZ tornado_appÚargsÚkw)Ú __class__©ú4lib/python3.7/site-packages/bokeh/server/views/ws.pyr ?s  zWSHandler.__init__cCsdS)Nr%)r!rZbokeh_websocket_pathr%r%r&Ú initializeLszWSHandler.initializecCsfddlm}t|ƒ}|j ¡}|jj}t ¡r:t t ¡ƒ}|||ƒ}|rLdSt   d||||¡dSdS)a™ Implement a check_origin policy for Tornado to call. The suplied origin will be compared to the Bokeh server whitelist. If the origin is not allow, an error will be logged and ``False`` will be returned. Args: origin (str) : The URL of the connection origin Returns: bool, True if the connection is allowed, False otherwise r )Úcheck_whitelistTz³Refusing websocket connection from Origin '%s'; use --allow-websocket-origin=%s or set BOKEH_ALLOW_WS_ORIGIN=%s to permit this; currently we allow origins %rFN) Úutilr(rZnetlocÚlowerÚ applicationZwebsocket_originsrZallowed_ws_originÚsetÚlogÚerror)r!Úoriginr(Z parsed_originZ origin_hostZ allowed_hostsZallowedr%r%r&Ú check_originOs     zWSHandler.check_origincCs¨t d¡|jddd}|dkr0| ¡tdƒ‚|jddd}|dkrV| ¡tdƒ‚t||jj|jjds€t  d |¡td ƒ‚d d „}|  ||¡}|jj   ||¡dS) zR Initialize a connection to a client. Returns: None zWebSocket connection openedzbokeh-protocol-versionN)Údefaultz#No bokeh-protocol-version specifiedzbokeh-session-idzNo bokeh-session-id specified)ÚsignedÚ secret_keyz$Session id had invalid signature: %rzInvalid session IDcSs | ¡}|dk rt d|¡dS)Nz"Failed to fully open connection %r)Z exceptionr-Údebug)ÚfutureÚer%r%r&Úon_fully_opened‡sz'WSHandler.open..on_fully_opened) r-ÚinfoZ get_argumentÚcloserrr+Z sign_sessionsr3r.Ú _async_openZio_loopZ add_future)r!Ú proto_versionÚ session_idr7r5r%r%r&Úopenns"    zWSHandler.openc csØyt|j ||j¡V|j |¡}t|ƒ}t|ƒ|_t d|¡t ƒ|_ t d|¡|j   |||j|¡|_ t d¡Wn:tk r®}zt d|¡| ¡|‚Wdd}~XYnX|j j d¡}| |¡Vt d¡‚dS)a‰ Perform the specific steps needed to open a connection to a Bokeh session Sepcifically, this method coordinates: * Getting a session for a session ID (creating a new one if needed) * Creating a protocol receiver and hander * Opening a new ServerConnection and sending it an ACK Args: session_id (str) : A session ID to for a session to connect to If no session exists with the given ID, a new session is made proto_version (str): The protocol version requested by the connecting client. Returns: None zReceiver created for %rzProtocolHandler created for %rzServerConnection createdz/Could not create new server session, reason: %sNZACK)rZcreate_session_if_neededZrequestZ get_sessionrrrr-r4r rr+Znew_connectionrr8rr.r9ÚprotocolZcreateÚ send_messagerÚReturn)r!r<r;Zsessionr>r6Úmsgr%r%r&r:’s"      zWSHandler._async_openc csÖy| |¡V}Wn>tk rN}z tjd||dd| d¡Wdd}~XYnXy8|r†tdk rjtj |¡| |¡V}|r†|  |¡VWn>tk rÆ}z tjd||dd| d¡Wdd}~XYnXt   d¡‚dS)a Process an individual wire protocol fragment. The websocket RFC specifies opcodes for distinguishing text frames from binary frames. Tornado passes us either a text or binary string depending on that opcode, we have to look at the type of the fragment to see what we got. Args: fragment (unicode or bytes) : wire fragment to process z/Unhandled exception receiving a message: %r: %rT)Úexc_infoz server failed to parse a messageNz.Handler or its work threw an exception: %r: %rz!server failed to handle a message) Ú_receiveÚ Exceptionr-r.Ú_internal_errorÚ_message_test_portZreceivedÚappendÚ_handleÚ _schedulerr@)r!ÚfragmentÚmessager6Úworkr%r%r&Ú on_messageÁs   zWSHandler.on_messagecCsbytt |d¡ƒ|_WnFtk r:tjd|ddYn$tk r\tjd|ddYnXdS)Nzutf-8z#received invalid unicode in pong %rT)rBz#received invalid integer in pong %r)ÚintÚcodecsÚdecoderÚUnicodeDecodeErrorr-ZtraceÚ ValueError)r!Údatar%r%r&Úon_pongès zWSHandler.on_pongc csVy$tdk rtj |¡| |¡VWn"ttfk rFt d¡YnXt  d¡‚dS)z‰ Send a Bokeh Server protocol message to the connected client. Args: message (Message) : a message to send Nz/Failed sending message as connection was closed) rFZsentrGÚsendr r r-Zwarningrr@)r!rKr%r%r&r?òs zWSHandler.send_messageFTc #sJ|r2|j ¡Vtt|ƒ ||¡VWdQRXntt|ƒ ||¡VdS)zj Override parent write_message with a version that acquires a write lock before writing. N)rÚacquirerrÚ write_message)r!rKZbinaryÚlocked)r$r%r&rWs zWSHandler.write_messagecCs.t d|j|j¡|jdk r*|j |j¡dS)z2 Clean up when the connection is closed. z/WebSocket connection closed: code=%s, reason=%rN)r-r8Z close_codeZ close_reasonrr+Z client_lost)r!r%r%r&Úon_closes zWSHandler.on_closec csby|j |¡V}t |¡‚Wn@tttfk r\}z| t|ƒ¡t d¡‚Wdd}~XYnXdS)N) rZconsumerr@rrrÚ_protocol_errorÚstr)r!rJrKr6r%r%r&rCs zWSHandler._receivec csfy |j ||j¡V}t |¡‚Wn@tttfk r`}z| t |ƒ¡t d¡‚Wdd}~XYnXdS)N) rZhandlerrr@rrrrEr[)r!rKrLr6r%r%r&rH!s zWSHandler._handleccs8t|tƒr| |¡Vn| dt|ƒ¡t d¡‚dS)Nzexpected a Message not )Ú isinstancerr?rEÚreprrr@)r!rLr%r%r&rI+s zWSHandler._schedulecCst d|¡| d|¡dS)Nz3Bokeh Server internal error: %s, closing connectioni')r-r.r9)r!rKr%r%r&rE4s zWSHandler._internal_errorcCst d|¡| d|¡dS)Nz3Bokeh Server protocol error: %s, closing connectioni')r-r.r9)r!rKr%r%r&rZ8s zWSHandler._protocol_error)FT)Ú__name__Ú __module__Ú __qualname__Ú__doc__r r'r0r=rÚ coroutiner:rMrTr?rWrYrCrHrIrErZÚ __classcell__r%r%)r$r&r;s" $/'     r)'raZ __future__rrrrZloggingZ getLoggerr^r-rOZsix.moves.urllib.parserZtornadorrZtornado.websocketr r r Zprotocol_handlerr r>rZprotocol.exceptionsrrrZprotocol.messagerZprotocol.receiverrZbokeh.util.session_idrZbokeh.settingsrÚ__all__rrFr%r%r%r&Ú s&