U /eA@sHdZddlmZddlZeeZddlmZm Z m Z m Z m Z ddl mZmZddlmZddlmZmZdd lmZdd lmZdd lmZmZmZdd lmZdd lm Z ddl!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*ddl+m,Z,er,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6dZ7GdddZ8e"dS)z Implements a very low level facility for communicating with a Bokeh Server. Users will always want to use :class:`~bokeh.client.session.ClientSession` instead for standard usage. ) annotationsN) TYPE_CHECKINGAnyCallableDictList)HTTPClientError HTTPRequest)IOLoop)WebSocketErrorwebsocket_connect)ID)Protocol) MessageError ProtocolErrorValidationError)Receiver)format_url_query_arguments)fixup_windows_event_loop_policy)CONNECTED_AFTER_ACKCONNECTED_BEFORE_ACK DISCONNECTEDNOT_YET_CONNECTEDWAITING_FOR_REPLY ErrorReasonState) WebSocketClientConnectionWrapper)Document)DocumentChangedEvent)Message) ServerInfo) ClientSession)ClientConnectionc@seZdZUdZded<ded<ded<d[d d d d dddddZeddddZeddddZed dddZ eddddZ edddd Z ed dd!d"Z ddd#d$Z d\d dd&d'd(Zddd)d*Zddd+d,Zd-dd.d/d0Zd-d1d.d2d3Zd4dd5d6Zd1dd7d8d9Zddd:d;Zddddd?d@dAZdddBdCZdDddEdFZd1dDd7dGdHZdIdJddKdLdMZd4ddNdOZdddPdQZdddRdSdTZdUddVdWdXZdddYdZZ dS)]r$zR A low-level class used to connect to a Bokeh server. .. autoclasstoc:: r_stater _loopzCallable[[], bool] | None_until_predicateN@r#strz IOLoop | NonezDict[str, str] | NoneintNone)session websocket_urlio_loop argumentsmax_message_sizereturncCs^||_||_||_||_t|_t|j|_d|_t |_ |dk rF|nt |_ d|_ d|_dS)z6 Opens a websocket connection to the server. N)_url_session _arguments_max_message_sizer _protocolr _receiver_socketrr%r r&r' _server_info)selfr,r-r.r/r0r;;/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/client/connection.py__init__]s zClientConnection.__init__boolr1cCs t|jtS)zg Whether we've connected the Websocket and have exchanged initial handshake messages. ) isinstancer%rr:r;r;r< connectedrszClientConnection.connectedcCs|jS)z2 The Tornado ``IOLoop`` this connection is using. )r&rAr;r;r<r.zszClientConnection.io_loopcCs|jS)z1 The URL of the websocket this Connection is to. )r2rAr;r;r<urlszClientConnection.urlzErrorReason | NonecCst|jtsdS|jjS)zX The reason of the connection loss encoded as a ``DISCONNECTED.ErrorReason`` enum value N)r@r%r error_reasonrAr;r;r<rDs zClientConnection.error_reasonz int | NonecCst|jtsdS|jjS)zv If there was an error that caused a disconnect, this property holds the error code. None otherwise. r)r@r%r error_coderAr;r;r<rEs zClientConnection.error_codecCst|jtsdS|jjS)z If there was an error that caused a disconnect, this property holds the error detail. Empty string otherwise. )r@r%r error_detailrAr;r;r<rGs zClientConnection.error_detailcs ddfdd }|dS)Nr>r?cstjtptjtSN)r@r%rrr;rAr;r<connected_or_closedsz5ClientConnection.connect..connected_or_closed) _loop_until)r:rIr;rAr<connectszClientConnection.connectclosed)whyr1cCs|jdk r|jd|dS)z* Close the Websocket connection. Ni)r8close)r:rMr;r;r<rNs zClientConnection.closecCs |dS)a  Force a round-trip request/reply to the server, sometimes needed to avoid race conditions. Mostly useful for testing. Outside of test suites, this method hurts performance and should not be needed. Returns: None N)_send_request_server_inforAr;r;r<force_roundtrips z ClientConnection.force_roundtripcs>tjtrt_nddfdd }|dS)a6 Execute a blocking loop that runs and executes event callbacks until the connection is closed (e.g. by hitting Ctrl-C). While this method can be used to run Bokeh application code "outside" the Bokeh server, this practice is HIGHLY DISCOURAGED for any real use case. r>r?cs tjtSrH)r@r%rr;rAr;r<rLsz2ClientConnection.loop_until_closed..closedN)r@r%r_tell_session_about_disconnectrrJ)r:rLr;rAr<loop_until_closeds  z"ClientConnection.loop_until_closedr)documentr1cCsX|jd}||}|dkr(tdn,|jddkrJtd|jdn ||dS)z Pull a document from the server, overwriting the passed-in document Args: document : (Document) The document to overwrite with server content. Returns: None z PULL-DOC-REQNConnection to server was lostmsgtypeERRORzFailed to pull document: text)r6create_send_message_wait_for_reply RuntimeErrorheadercontentZpush_to_documentr:rSmsgreplyr;r;r<pull_docs   zClientConnection.pull_doc Message[Any]cCsT|jd|}||}|dkr*tdn&|jddkrLtd|jdn|SdS)z Push a document to the server, overwriting any existing server-side doc. Args: document : (Document) A Document to push to the server Returns: The server reply zPUSH-DOCNrTrUrVzFailed to push document: rW)r6rXrYrZr[r\r]r;r;r<push_docs   zClientConnection.push_docr"cCs|jdkr||_|jS)zq Ask for information about the server. Returns: A dictionary of server attributes. N)r9rOrAr;r;r<request_server_infos  z$ClientConnection.request_server_info)messager1c s||jdkrtd|n`z$||jIdH}td||Wn:tk rv}ztd||jddW5d}~XYnXdS)Nz-We're disconnected, so not sending message %rzSent %r [%d bytes]z#Error sending message to server: %rzreceived error while sendingrM)r8loginfosenddebugr rN)r:rdsenter;r;r< send_messages  zClientConnection.send_messagec st|j|j}t|}z,t|d|jjg|jdIdH}t||_ Wnrt k r}z(| t t j|j|jIdHWYdSd}~XYn.tk r}ztd|W5d}~XYnX|j dkr| t t jddIdHn|tIdHdS)NZbokeh)Z subprotocolsr0zFailed to connect to server: %rzSocket invalid.)rr2r4r r r3tokenr5rr8r_transition_to_disconnectedrr HTTP_ERRORcoderd ExceptionrfrgZ NETWORK_ERROR _transitionr)r:Z formatted_urlrequestsocketrkr;r;r<_connect_async#s zClientConnection._connect_asynccsr|IdH}|dkr2|ttjddIdHn<|jdkrTtd|j |n td|| IdHdS)NInternal server error. PATCH-DOCz"Got PATCH-DOC, applying to sessionz Ignoring %r) _pop_messagernrrrorUrfrir3Z _handle_patch_nextr:rdr;r;r<_handle_messages4s   z!ClientConnection._handle_messageszCallable[[], bool]) predicater1cCsF||_z|j|j|jWntk r@|dYnXdS)Nzuser interruption)r'r& add_callbackrzstartKeyboardInterruptrN)r:r}r;r;r<rJAs zClientConnection._loop_untilcsh|jdk r>|r>td|jjj|jjd|_|jdStd|jjj|j|IdHdS)Nz4Stopping client loop in state %s due to True from %szRunning state ) r'rfrir% __class____name__r&stoprunrAr;r;r<rzLs zClientConnection._nextzMessage[Any] | Nonec s|jdkrdSd}z|jIdH}Wn.tk rT}ztd|W5d}~XYnX|dkrltddSz2|j|IdH}|dk rtd||WSWqtt t fk r}z tj d|dd|j ddW5d}~XYqXqdS) NzError reading from socket %rzConnection closed by serverzReceived message %rz%rT)exc_infoz!error parsing message from serverre) r8Z read_messagerqrfrirgr7consumerrrerrorrN)r:fragmentrkrdr;r;r<ryWs$   zClientConnection._pop_messagecst|jd_gddddfdd }j||dd fd d }|dd fd d }|jS)Nmsgidraz List[None]r+)rd send_resultr1cs|IdH}||dSrH)rlappend)rdrresultrAr;r<handle_messagevszEClientConnection._send_message_wait_for_reply..handle_messager>r?cstdkpjkS)Nr)lenr%r;r:rwaiterr;r< have_send_result_or_disconnected{szWClientConnection._send_message_wait_for_reply..have_send_result_or_disconnectedcsjkpjdk SrH)r%r_r;)r:rr;r<have_reply_or_disconnectedszQClientConnection._send_message_wait_for_reply..have_reply_or_disconnected)rr[r%r&r~rJr_)r:rdrrrr;rr<rYqs  z-ClientConnection._send_message_wait_for_replyrr ) session_ideventr1cCsRddlm}t|dr*t|j|r*d|j_|jjd|gdd}|j |j |dS)Nr)ColumnDataChangedEventhintrxF)Z use_buffers) Zbokeh.document.eventsrhasattrr@rcolsr6rXr&r~rl)r:rrrr^r;r;r<_send_patch_documents  z%ClientConnection._send_patch_documentcCs,|jd}||}|dkr&td|jS)NzSERVER-INFO-REQzs4         $       g