Issue #12786: Implement hook for Comm messages

Author: sonthonaxrk

ipykernel/ipkernel.py

@@ -68,6 +68,8 @@ def _user_ns_changed(self, change):
     _sys_raw_input = Any()
     _sys_eval_input = Any()
 
+    comm_msg_types = [ 'comm_open', 'comm_msg', 'comm_close' ]
+
     def __init__(self, **kwargs):
         super(IPythonKernel, self).__init__(**kwargs)
 
@@ -95,8 +97,7 @@ def __init__(self, **kwargs):
         self.comm_manager = CommManager(parent=self, kernel=self)
 
         self.shell.configurables.append(self.comm_manager)
-        comm_msg_types = [ 'comm_open', 'comm_msg', 'comm_close' ]
-        for msg_type in comm_msg_types:
+        for msg_type in self.comm_msg_types:
             self.shell_handlers[msg_type] = getattr(self.comm_manager, msg_type)
 
         if _use_appnope() and self._darwin_app_nap:
@@ -566,6 +567,16 @@ def do_clear(self):
         self.shell.reset(False)
         return dict(status='ok')
 
+    def should_dispatch_immediately(self, msg):
+        try:
+            msg_type = msg['header']['msg_type']
+            if msg_type in self.comm_msg_types:
+                return True
+        except ValueError:
+            pass
+
+        return False
+
 
 # This exists only for backwards compatibility - use IPythonKernel instead
 

ipykernel/kernelbase.py

@@ -300,7 +300,7 @@ def should_handle(self, stream, msg, idents):
             return False
         return True
 
-    async def dispatch_shell(self, msg):
+    async def dispatch_shell(self, msg, idents=None):
         """dispatch shell requests"""
 
         # flush control queue before handling shell requests