U /e @sddZddlmZddlZeeZddlmZm Z m Z ddl m Z ddl mZdZGdd d ZdS) zn Provide a low-level wrapper for Tornado Websockets that adds locking and smooths some compatibility issues. ) annotationsN)Any AwaitableCallable)locks)WebSocketClientConnection) WebSocketClientConnectionWrapperc@s\eZdZdZdddddZdd d d dd d d ZdddddddZddddddZdS)rzE Used for compatibility across Tornado versions and to add write_lockrNone)socketreturncCs||_t|_dS)N)_socketrLock write_lock)selfr r:/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/client/websocket.py__init__2sz)WebSocketClientConnectionWrapper.__init__FTz str | bytesbool)messagebinarylockedr c sB|r0|jIdH|j||W5QRXn|j||dS)zh Write a message to the websocket after obtaining the appropriate Bokeh Document lock. N)racquirer write_message)rrrrrrrr:sz.WebSocketClientConnectionWrapper.write_messageNz int | Nonez str | None)codereasonr cCs|j||S)z Close the websocket. )r close)rrrrrrrEsz&WebSocketClientConnectionWrapper.closezCallable[..., Any] | NonezAwaitable[None | str | bytes])callbackr cCs |j|S)z@ Read a message from websocket and execute a callback. )r read_message)rrrrrrIsz-WebSocketClientConnectionWrapper.read_message)FT)NN)N)__name__ __module__ __qualname____doc__rrrrrrrrr/s  r)r! __future__rlogging getLoggerrlogtypingrrrZtornadorZtornado.websocketr__all__rrrrrs