specify socket identity from kernel.js

minrk · minrk · commit 24dcb6ba4845 · 2013-04-25T12:24:37.000-07:00
required for stdin routing
diff --git a/IPython/frontend/html/notebook/handlers.py b/IPython/frontend/html/notebook/handlers.py
@@ -506,6 +506,12 @@ def _inject_cookie_message(self, msg):
             # Cookie constructor doesn't accept unicode strings
             # under Python 2.x for some reason
             msg = msg.encode('utf8', 'replace')
+        try:
+            bsession, msg = msg.split(':', 1)
+            self.session.session = bsession.decode('ascii')
+        except Exception:
+            logging.error("No bsession!", exc_info=True)
+            pass
         try:
             self.request._cookies = Cookie.SimpleCookie(msg)
         except:
@@ -528,7 +534,7 @@ def max_msg_size(self):
     def create_stream(self):
         km = self.kernel_manager
         meth = getattr(km, 'connect_%s' % self.channel)
-        self.zmq_stream = meth(self.kernel_id)
+        self.zmq_stream = meth(self.kernel_id, identity=self.session.bsession)
     
     def initialize(self, *args, **kwargs):
         self.zmq_stream = None
diff --git a/IPython/frontend/html/notebook/static/js/kernel.js b/IPython/frontend/html/notebook/static/js/kernel.js
@@ -129,7 +129,7 @@ var IPython = (function (IPython) {
         this.shell_channel = new this.WebSocket(ws_url + "/shell");
         this.iopub_channel = new this.WebSocket(ws_url + "/iopub");
         send_cookie = function(){
-            this.send(document.cookie);
+            this.send(that.session_id + ':' + document.cookie);
         };
         var already_called_onclose = false; // only alert once
         var ws_closed_early = function(evt){
