ó mÜJ]c@`s6dZddlmZmZmZmZddlZejeƒZ ddl Z ddl m Z ddl mZmZddlmZmZmZddlmZd d lmZd d lmZmZmZd d lmZd d lmZddl m!Z!ddl"m#Z#dfZ$defd„ƒYZ%e&Z'dS(uA Provide a web socket handler for the Bokeh Server application. i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(turlparse(tgentlocks(tStreamClosedErrortWebSocketHandlertWebSocketClosedErrori(tProtocolHandleri(tProtocol(t MessageErrort ProtocolErrortValidationError(tMessage(tReceiver(tcheck_session_id_signature(tsettingsu WSHandlert WSHandlercB`sÚeZdZd„Zd„Zd„Zd„Zejd„ƒZ ejd„ƒZ d„Z ejd„ƒZ eje ed „ƒZd „Zejd „ƒZejd „ƒZejd „ƒZd„Zd„ZRS(uI Implements a custom Tornado WebSocketHandler for the Bokeh Server. cO`s`d|_d|_d|_|d|_d|_tjƒ|_t t |ƒj |||ŽdS(Nuapplication_contextiÿÿÿÿ( tNonetreceiverthandlert connectiontapplication_contextt latest_pongRtLockt write_locktsuperRt__init__(tselft tornado_apptargstkw((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyR?s     cC`sdS(N((RRtbokeh_websocket_path((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyt initializeLscC`s’ddlm}t|ƒ}|jjƒ}|jj}tjƒrXt tjƒƒ}n|||ƒ}|rqt St j d||||ƒt SdS(uš Implement a check_origin policy for Tornado to call. The supplied 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 i(tcheck_whitelistu³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 %rN(tutilR$Rtnetloctlowert applicationtwebsocket_originsRtallowed_ws_origintsettTruetlogterrortFalse(RtoriginR$t parsed_origint origin_hostt allowed_hoststallowed((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyt check_originOs   cC`sùtjdƒ|jddd ƒ}|d krG|jƒtdƒ‚n|jddd ƒ}|d kr|jƒtdƒ‚nt|d|jjd|jj ƒsÄtj d |ƒtd ƒ‚nd „}|j ||ƒ}|jj j ||ƒd S( uR Initialize a connection to a client. Returns: None uWebSocket connection openedubokeh-protocol-versiontdefaultu#No bokeh-protocol-version specifiedubokeh-session-iduNo bokeh-session-id specifiedtsignedt secret_keyu$Session id had invalid signature: %ruInvalid session IDcS`s/|jƒ}|dk r+tjd|ƒndS(Nu"Failed to fully open connection %r(t exceptionRR-tdebug(tfuturete((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyton_fully_opened‡s  N(R-tinfot get_argumentRtcloseR RR(t sign_sessionsR8R.t _async_opentio_loopt add_future(Rt proto_versiont session_idR=R;((s4lib/python2.7/site-packages/bokeh/server/views/ws.pytopenns"        cc`sy¢|jj||jƒV|jj|ƒ}t|ƒ}t|ƒ|_tjd|ƒt ƒ|_ tjd|ƒ|j j |||j|ƒ|_ tjdƒWn3tk r×}tjd|ƒ|jƒ|‚nX|j jjdƒ}|j|ƒVtjdƒ‚dS(u‰ Perform the specific steps needed to open a connection to a Bokeh session Specifically, 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 uReceiver created for %ruProtocolHandler created for %ruServerConnection createdu/Could not create new server session, reason: %suACKN(Rtcreate_session_if_neededtrequestt get_sessionR RRR-R:R RR(tnew_connectionRR>R R.R@tprotocoltcreatet send_messageRtReturnR(RRFREtsessionRLR<tmsg((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyRB’s"  !  cc`sòy|j|ƒV}Wn9tk rO}tjd||dtƒ|jdƒnXyS|r¢tdk rxtjj |ƒn|j |ƒV}|r¢|j |ƒVq¢nWn9tk rÞ}tjd||dtƒ|jdƒnXt j dƒ‚dS(u 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 u/Unhandled exception receiving a message: %r: %rtexc_infou server failed to parse a messageu.Handler or its work threw an exception: %r: %ru!server failed to handle a messageN(t_receivet ExceptionR-R.R,t_internal_errort_message_test_portRtreceivedtappendt_handlet _scheduleRRO(RtfragmenttmessageR<twork((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyt on_messageÁs  cC`ssyttj|dƒƒ|_WnMtk rHtjd|dtƒn'tk rntjd|dtƒnXdS(Nuutf-8u#received invalid unicode in pong %rRRu#received invalid integer in pong %r( tinttcodecstdecodeRtUnicodeDecodeErrorR-ttraceR,t ValueError(Rtdata((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyton_pongès   cc`sky1tdk r"tjj|ƒn|j|ƒVWn$ttfk rWtjdƒnXt j dƒ‚dS(u‰ Send a Bokeh Server protocol message to the connected client. Args: message (Message) : a message to send u/Failed sending message as connection was closedN( RVRtsentRXtsendR RR-twarningRRO(RR\((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyRNòs cc`sX|r:|jjƒVtt|ƒj||ƒVWdQXntt|ƒj||ƒVdS(uj Override parent write_message with a version that acquires a write lock before writing. N(RtacquireRRt write_message(RR\tbinarytlocked((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyRks#cC`sBtjd|j|jƒ|jdk r>|jj|jƒndS(u2 Clean up when the connection is closed. u/WebSocket connection closed: code=%s, reason=%rN(R-R>t close_codet close_reasonRRR(t client_lost(R((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyton_closescc`sky&|jj|ƒV}tj|ƒ‚Wn>tttfk rf}|jt|ƒƒtjdƒ‚nXdS(N( RtconsumeRROR R Rt_protocol_errortstrR(RR[R\R<((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyRSs cc`sqy,|jj||jƒV}tj|ƒ‚Wn>tttfk rl}|jt |ƒƒtjdƒ‚nXdS(N( RthandleRRROR R RRURtR(RR\R]R<((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyRY!s cc`sJt|tƒr |j|ƒVn|jdt|ƒƒtjdƒ‚dS(Nuexpected a Message not (t isinstanceRRNRUtreprRROR(RR]((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyRZ+scC`s$tjd|ƒ|jd|ƒdS(Nu3Bokeh Server internal error: %s, closing connectioni'(R-R.R@(RR\((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyRU4scC`s$tjd|ƒ|jd|ƒdS(Nu3Bokeh Server protocol error: %s, closing connectioni'(R-R.R@(RR\((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyRs8s(t__name__t __module__t__doc__RR#R5RGRt coroutineRBR^RfRNR/R,RkRqRSRYRZRURs(((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyR;s"   $/'     ((Rzt __future__RRRRtloggingt getLoggerRxR-R`tsix.moves.urllib.parseRttornadoRRttornado.websocketRRR tprotocol_handlerR RLR tprotocol.exceptionsR R Rtprotocol.messageRtprotocol.receiverRtbokeh.util.session_idRtbokeh.settingsRt__all__RRRV(((s4lib/python2.7/site-packages/bokeh/server/views/ws.pyt s$"   ÿ