U /e@sddZddlmZddlZeeZddlmZm Z m Z ddl m Z ddl mZd ZGd d d ZdS) zR Encapsulate handling of all Bokeh Protocol messages a Bokeh server may receive. ) annotationsN)AnyCallableDict) ProtocolError) ServerSession)ProtocolHandlerc@s8eZdZUdZded<ddddZdd Zd d Zd S) r a A Bokeh server may be expected to receive any of the following protocol messages: * ``PATCH-DOC`` * ``PULL-DOC-REQ`` * ``PUSH-DOC`` * ``SERVER-INFO-REQ`` The job of ``ProtocolHandler`` is to direct incoming messages to the right specialized handler for each message type. When the server receives a new message on a connection it will call ``handler`` with the message and the connection that the message arrived on. Most messages are ultimately handled by the ``ServerSession`` class, but some simpler messages types such as ``SERVER-INFO-REQ`` may be handled directly by ``ProtocolHandler``. Any unexpected messages will result in a ``ProtocolError``. zDict[str, Callable[..., Any]] _handlersNone)returncCs:i|_tj|jd<tj|jd<tj|jd<|j|jd<dS)Nz PULL-DOC-REQzPUSH-DOCz PATCH-DOCzSERVER-INFO-REQ)r r pullpushpatch_server_info_req)selfrA/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/server/protocol_handler.py__init__As    zProtocolHandler.__init__c s|j|j}|dkr$|j|j}|dkr8td|z|||IdH}WnDtk r}z&tjd||dd||t|}W5d}~XYnX|S)aI Delegate a received message to the appropriate handler. Args: message (Message) : The message that was receive that needs to be handled connection (ServerConnection) : The connection that received this message Raises: ProtocolError Nz%s not expected on serverz/error handling message message: %r error: %rT)exc_info)r getZmsgtyper Exceptionlogerrorrepr)rmessage connectionhandlerZworkerrrhandleIs "zProtocolHandler.handlecs|jd|jdS)NzSERVER-INFO-REPLYmsgid)protocolcreateheader)rrrrrrrhsz ProtocolHandler._server_info_reqN)__name__ __module__ __qualname____doc____annotations__rr rrrrrr +s r )r( __future__rlogging getLoggerr%rtypingrrrZprotocol.exceptionsrsessionr __all__r rrrrs