comm.py
139 lines
| 4.4 KiB
| text/x-python
|
PythonLexer
|
r13195 | """Base class for a Comm""" | ||
|
r16623 | # Copyright (c) IPython Development Team. | ||
# Distributed under the terms of the Modified BSD License. | ||||
|
r13195 | |||
import uuid | ||||
from IPython.config import LoggingConfigurable | ||||
from IPython.core.getipython import get_ipython | ||||
|
r16623 | from IPython.utils.jsonutil import json_clean | ||
|
r13195 | from IPython.utils.traitlets import Instance, Unicode, Bytes, Bool, Dict, Any | ||
class Comm(LoggingConfigurable): | ||||
shell = Instance('IPython.core.interactiveshell.InteractiveShellABC') | ||||
def _shell_default(self): | ||||
return get_ipython() | ||||
|
r13199 | |||
|
r13195 | iopub_socket = Any() | ||
def _iopub_socket_default(self): | ||||
|
r13199 | return self.shell.kernel.iopub_socket | ||
|
r13195 | session = Instance('IPython.kernel.zmq.session.Session') | ||
def _session_default(self): | ||||
if self.shell is None: | ||||
return | ||||
|
r13199 | return self.shell.kernel.session | ||
|
r13195 | |||
|
r13204 | target_name = Unicode('comm') | ||
|
r13195 | |||
topic = Bytes() | ||||
def _topic_default(self): | ||||
return ('comm-%s' % self.comm_id).encode('ascii') | ||||
|
r13205 | _open_data = Dict(help="data dict, if any, to be included in comm_open") | ||
|
r13195 | _close_data = Dict(help="data dict, if any, to be included in comm_close") | ||
_msg_callback = Any() | ||||
_close_callback = Any() | ||||
_closed = Bool(False) | ||||
comm_id = Unicode() | ||||
def _comm_id_default(self): | ||||
return uuid.uuid4().hex | ||||
primary = Bool(True, help="Am I the primary or secondary Comm?") | ||||
|
r13228 | def __init__(self, target_name='', data=None, **kwargs): | ||
if target_name: | ||||
kwargs['target_name'] = target_name | ||||
|
r13195 | super(Comm, self).__init__(**kwargs) | ||
if self.primary: | ||||
|
r13205 | # I am primary, open my peer. | ||
|
r13197 | self.open(data) | ||
|
r13195 | |||
|
r13224 | def _publish_msg(self, msg_type, data=None, metadata=None, **keys): | ||
|
r13195 | """Helper for sending a comm message on IOPub""" | ||
data = {} if data is None else data | ||||
|
r13224 | metadata = {} if metadata is None else metadata | ||
|
r16623 | content = json_clean(dict(data=data, comm_id=self.comm_id, **keys)) | ||
|
r13195 | self.session.send(self.iopub_socket, msg_type, | ||
|
r16623 | content, | ||
metadata=json_clean(metadata), | ||||
|
r13223 | parent=self.shell.get_parent(), | ||
|
r13195 | ident=self.topic, | ||
) | ||||
def __del__(self): | ||||
"""trigger close on gc""" | ||||
self.close() | ||||
# publishing messages | ||||
|
r13224 | def open(self, data=None, metadata=None): | ||
|
r13195 | """Open the frontend-side version of this comm""" | ||
if data is None: | ||||
data = self._open_data | ||||
|
r17450 | self._closed = False | ||
|
r17613 | ip = get_ipython() | ||
if hasattr(ip, 'comm_manager'): | ||||
ip.comm_manager.register_comm(self) | ||||
|
r13224 | self._publish_msg('comm_open', data, metadata, target_name=self.target_name) | ||
|
r13195 | |||
|
r13224 | def close(self, data=None, metadata=None): | ||
|
r13195 | """Close the frontend-side version of this comm""" | ||
if self._closed: | ||||
# only close once | ||||
return | ||||
if data is None: | ||||
data = self._close_data | ||||
|
r13224 | self._publish_msg('comm_close', data, metadata) | ||
|
r17613 | ip = get_ipython() | ||
if hasattr(ip, 'comm_manager'): | ||||
ip.comm_manager.unregister_comm(self) | ||||
|
r13195 | self._closed = True | ||
|
r13224 | def send(self, data=None, metadata=None): | ||
|
r13197 | """Send a message to the frontend-side version of this comm""" | ||
|
r13224 | self._publish_msg('comm_msg', data, metadata) | ||
|
r13195 | |||
# registering callbacks | ||||
def on_close(self, callback): | ||||
"""Register a callback for comm_close | ||||
Will be called with the `data` of the close message. | ||||
Call `on_close(None)` to disable an existing callback. | ||||
""" | ||||
self._close_callback = callback | ||||
def on_msg(self, callback): | ||||
"""Register a callback for comm_msg | ||||
Will be called with the `data` of any comm_msg messages. | ||||
Call `on_msg(None)` to disable an existing callback. | ||||
""" | ||||
self._msg_callback = callback | ||||
# handling of incoming messages | ||||
|
r13197 | def handle_close(self, msg): | ||
|
r13195 | """Handle a comm_close message""" | ||
|
r13197 | self.log.debug("handle_close[%s](%s)", self.comm_id, msg) | ||
|
r13195 | if self._close_callback: | ||
|
r13197 | self._close_callback(msg) | ||
|
r13195 | |||
|
r13197 | def handle_msg(self, msg): | ||
|
r13195 | """Handle a comm_msg message""" | ||
|
r13197 | self.log.debug("handle_msg[%s](%s)", self.comm_id, msg) | ||
|
r13195 | if self._msg_callback: | ||
|
r15605 | self.shell.events.trigger('pre_execute') | ||
|
r13197 | self._msg_callback(msg) | ||
|
r15605 | self.shell.events.trigger('post_execute') | ||
|
r13195 | |||
__all__ = ['Comm'] | ||||