make parent_header available from the Shell object

IPython/kernel/zmq/ipkernel.py

@@ -1,14 +1,5 b''
 #!/usr/bin/env python
-"""A simple interactive kernel that talks to a frontend over 0MQ.
-
-Things to do:
-
-* Implement `set_parent` logic. Right before doing exec, the Kernel should
-  call set_parent on all the PUB objects with the message about to be executed.
-* Implement random port and security key logic.
-* Implement control messages.
-* Implement event loop and poll version.
-"""
+"""An interactive kernel that talks to frontends over 0MQ."""
 
 #-----------------------------------------------------------------------------
 # Imports

IPython/kernel/zmq/zmqshell.py

@@ -478,6 +478,7 b' class ZMQInteractiveShell(InteractiveShell):'
     display_pub_class = Type(ZMQDisplayPublisher)
     data_pub_class = Type(ZMQDataPublisher)
     kernel = Any()
+    parent_header = Any()
 
     # Override the traitlet in the parent class, because there's no point using
     # readline for the kernel. Can be removed when the readline code is moved
@@ -589,6 +590,7 b' class ZMQInteractiveShell(InteractiveShell):'
     
     def set_parent(self, parent):
         """Set the parent header for associating output with its triggering input"""
+        self.parent_header = parent
         self.displayhook.set_parent(parent)
         self.display_pub.set_parent(parent)
         self.data_pub.set_parent(parent)
@@ -601,6 +603,9 b' class ZMQInteractiveShell(InteractiveShell):'
         except AttributeError:
             pass
     
+    def get_parent(self):
+        return self.parent_header
+    
     #-------------------------------------------------------------------------
     # Things related to magics
     #-------------------------------------------------------------------------