##// END OF EJS Templates
Fix documentation of Kernel.stop_channels
Andrea Bedini -
Show More
@@ -1,502 +1,502 b''
1 1 //----------------------------------------------------------------------------
2 2 // Copyright (C) 2008-2011 The IPython Development Team
3 3 //
4 4 // Distributed under the terms of the BSD License. The full license is in
5 5 // the file COPYING, distributed as part of this software.
6 6 //----------------------------------------------------------------------------
7 7
8 8 //============================================================================
9 9 // Kernel
10 10 //============================================================================
11 11
12 12 /**
13 13 * @module IPython
14 14 * @namespace IPython
15 15 * @submodule Kernel
16 16 */
17 17
18 18 var IPython = (function (IPython) {
19 19
20 20 var utils = IPython.utils;
21 21
22 22 // Initialization and connection.
23 23 /**
24 24 * A Kernel Class to communicate with the Python kernel
25 25 * @Class Kernel
26 26 */
27 27 var Kernel = function (base_url) {
28 28 this.kernel_id = null;
29 29 this.shell_channel = null;
30 30 this.iopub_channel = null;
31 31 this.stdin_channel = null;
32 32 this.base_url = base_url;
33 33 this.running = false;
34 34 this.username = "username";
35 35 this.session_id = utils.uuid();
36 36 this._msg_callbacks = {};
37 37
38 38 if (typeof(WebSocket) !== 'undefined') {
39 39 this.WebSocket = WebSocket;
40 40 } else if (typeof(MozWebSocket) !== 'undefined') {
41 41 this.WebSocket = MozWebSocket;
42 42 } else {
43 43 alert('Your browser does not have WebSocket support, please try Chrome, Safari or Firefox β‰₯ 6. Firefox 4 and 5 are also supported by you have to enable WebSockets in about:config.');
44 44 };
45 45 this.bind_events();
46 46 };
47 47
48 48
49 49 Kernel.prototype._get_msg = function (msg_type, content) {
50 50 var msg = {
51 51 header : {
52 52 msg_id : utils.uuid(),
53 53 username : this.username,
54 54 session : this.session_id,
55 55 msg_type : msg_type
56 56 },
57 57 metadata : {},
58 58 content : content,
59 59 parent_header : {}
60 60 };
61 61 return msg;
62 62 };
63 63
64 64 Kernel.prototype.bind_events = function() {
65 65 var that = this;
66 66 $([IPython.events]).on('send_input_reply.Kernel', function(evt, data) {
67 67 that.send_input_reply(data);
68 68 });
69 69 }
70 70
71 71 /**
72 72 * Start the Python kernel
73 73 * @method start
74 74 */
75 75 Kernel.prototype.start = function (notebook_id) {
76 76 var that = this;
77 77 if (!this.running) {
78 78 var qs = $.param({notebook:notebook_id});
79 79 var url = this.base_url + '?' + qs;
80 80 $.post(url,
81 81 $.proxy(that._kernel_started,that),
82 82 'json'
83 83 );
84 84 };
85 85 };
86 86
87 87 /**
88 88 * Restart the python kernel.
89 89 *
90 90 * Emit a 'status_restarting.Kernel' event with
91 91 * the current object as parameter
92 92 *
93 93 * @method restart
94 94 */
95 95 Kernel.prototype.restart = function () {
96 96 $([IPython.events]).trigger('status_restarting.Kernel', {kernel: this});
97 97 var that = this;
98 98 if (this.running) {
99 99 this.stop_channels();
100 100 var url = this.kernel_url + "/restart";
101 101 $.post(url,
102 102 $.proxy(that._kernel_started, that),
103 103 'json'
104 104 );
105 105 };
106 106 };
107 107
108 108
109 109 Kernel.prototype._kernel_started = function (json) {
110 110 console.log("Kernel started: ", json.kernel_id);
111 111 this.running = true;
112 112 this.kernel_id = json.kernel_id;
113 113 var ws_url = json.ws_url;
114 114 if (ws_url.match(/wss?:\/\//) == null) {
115 115 // trailing 's' in https will become wss for secure web sockets
116 116 prot = location.protocol.replace('http', 'ws') + "//";
117 117 ws_url = prot + location.host + ws_url;
118 118 };
119 119 this.ws_url = ws_url;
120 120 this.kernel_url = this.base_url + "/" + this.kernel_id;
121 121 this.start_channels();
122 122 $([IPython.events]).trigger('status_started.Kernel', {kernel: this});
123 123 };
124 124
125 125
126 126 Kernel.prototype._websocket_closed = function(ws_url, early) {
127 127 this.stop_channels();
128 128 $([IPython.events]).trigger('websocket_closed.Kernel',
129 129 {ws_url: ws_url, kernel: this, early: early}
130 130 );
131 131 };
132 132
133 133 /**
134 134 * Start the `shell`and `iopub` channels.
135 135 * Will stop and restart them if they already exist.
136 136 *
137 137 * @method start_channels
138 138 */
139 139 Kernel.prototype.start_channels = function () {
140 140 var that = this;
141 141 this.stop_channels();
142 142 var ws_url = this.ws_url + this.kernel_url;
143 143 console.log("Starting WebSockets:", ws_url);
144 144 this.shell_channel = new this.WebSocket(ws_url + "/shell");
145 145 this.stdin_channel = new this.WebSocket(ws_url + "/stdin");
146 146 this.iopub_channel = new this.WebSocket(ws_url + "/iopub");
147 147 send_cookie = function(){
148 148 // send the session id so the Session object Python-side
149 149 // has the same identity
150 150 this.send(that.session_id + ':' + document.cookie);
151 151 };
152 152 var already_called_onclose = false; // only alert once
153 153 var ws_closed_early = function(evt){
154 154 if (already_called_onclose){
155 155 return;
156 156 }
157 157 already_called_onclose = true;
158 158 if ( ! evt.wasClean ){
159 159 that._websocket_closed(ws_url, true);
160 160 }
161 161 };
162 162 var ws_closed_late = function(evt){
163 163 if (already_called_onclose){
164 164 return;
165 165 }
166 166 already_called_onclose = true;
167 167 if ( ! evt.wasClean ){
168 168 that._websocket_closed(ws_url, false);
169 169 }
170 170 };
171 171 var channels = [this.shell_channel, this.iopub_channel, this.stdin_channel];
172 172 for (var i=0; i < channels.length; i++) {
173 173 channels[i].onopen = send_cookie;
174 174 channels[i].onclose = ws_closed_early;
175 175 }
176 176 // switch from early-close to late-close message after 1s
177 177 setTimeout(function() {
178 178 for (var i=0; i < channels.length; i++) {
179 179 if (channels[i] !== null) {
180 180 channels[i].onclose = ws_closed_late;
181 181 }
182 182 }
183 183 }, 1000);
184 184 this.shell_channel.onmessage = $.proxy(this._handle_shell_reply, this);
185 185 this.iopub_channel.onmessage = $.proxy(this._handle_iopub_reply, this);
186 186 this.stdin_channel.onmessage = $.proxy(this._handle_input_request, this);
187 187 };
188 188
189 189 /**
190 * Start the `shell`and `iopub` channels.
190 * Stop the `shell`and `iopub` channels.
191 191 * @method stop_channels
192 192 */
193 193 Kernel.prototype.stop_channels = function () {
194 194 var channels = [this.shell_channel, this.iopub_channel, this.stdin_channel];
195 195 for (var i=0; i < channels.length; i++) {
196 196 if ( channels[i] !== null ) {
197 197 channels[i].onclose = function (evt) {};
198 198 channels[i].close();
199 199 }
200 200 };
201 201 this.shell_channel = this.iopub_channel = this.stdin_channel = null;
202 202 };
203 203
204 204 // Main public methods.
205 205
206 206 /**
207 207 * Get info on object asynchronoulsy
208 208 *
209 209 * @async
210 210 * @param objname {string}
211 211 * @param callback {dict}
212 212 * @method object_info_request
213 213 *
214 214 * @example
215 215 *
216 216 * When calling this method pass a callbacks structure of the form:
217 217 *
218 218 * callbacks = {
219 219 * 'object_info_reply': object_info_reply_callback
220 220 * }
221 221 *
222 222 * The `object_info_reply_callback` will be passed the content object of the
223 223 *
224 224 * `object_into_reply` message documented in
225 225 * [IPython dev documentation](http://ipython.org/ipython-doc/dev/development/messaging.html#object-information)
226 226 */
227 227 Kernel.prototype.object_info_request = function (objname, callbacks) {
228 228 if(typeof(objname)!=null && objname!=null)
229 229 {
230 230 var content = {
231 231 oname : objname.toString(),
232 232 detail_level : 0,
233 233 };
234 234 var msg = this._get_msg("object_info_request", content);
235 235 this.shell_channel.send(JSON.stringify(msg));
236 236 this.set_callbacks_for_msg(msg.header.msg_id, callbacks);
237 237 return msg.header.msg_id;
238 238 }
239 239 return;
240 240 }
241 241
242 242 /**
243 243 * Execute given code into kernel, and pass result to callback.
244 244 *
245 245 * TODO: document input_request in callbacks
246 246 *
247 247 * @async
248 248 * @method execute
249 249 * @param {string} code
250 250 * @param [callbacks] {Object} With the optional following keys
251 251 * @param callbacks.'execute_reply' {function}
252 252 * @param callbacks.'output' {function}
253 253 * @param callbacks.'clear_output' {function}
254 254 * @param callbacks.'set_next_input' {function}
255 255 * @param {object} [options]
256 256 * @param [options.silent=false] {Boolean}
257 257 * @param [options.user_expressions=empty_dict] {Dict}
258 258 * @param [options.user_variables=empty_list] {List od Strings}
259 259 * @param [options.allow_stdin=false] {Boolean} true|false
260 260 *
261 261 * @example
262 262 *
263 263 * The options object should contain the options for the execute call. Its default
264 264 * values are:
265 265 *
266 266 * options = {
267 267 * silent : true,
268 268 * user_variables : [],
269 269 * user_expressions : {},
270 270 * allow_stdin : false
271 271 * }
272 272 *
273 273 * When calling this method pass a callbacks structure of the form:
274 274 *
275 275 * callbacks = {
276 276 * 'execute_reply': execute_reply_callback,
277 277 * 'output': output_callback,
278 278 * 'clear_output': clear_output_callback,
279 279 * 'set_next_input': set_next_input_callback
280 280 * }
281 281 *
282 282 * The `execute_reply_callback` will be passed the content and metadata
283 283 * objects of the `execute_reply` message documented
284 284 * [here](http://ipython.org/ipython-doc/dev/development/messaging.html#execute)
285 285 *
286 286 * The `output_callback` will be passed `msg_type` ('stream','display_data','pyout','pyerr')
287 287 * of the output and the content and metadata objects of the PUB/SUB channel that contains the
288 288 * output:
289 289 *
290 290 * http://ipython.org/ipython-doc/dev/development/messaging.html#messages-on-the-pub-sub-socket
291 291 *
292 292 * The `clear_output_callback` will be passed a content object that contains
293 293 * stdout, stderr and other fields that are booleans, as well as the metadata object.
294 294 *
295 295 * The `set_next_input_callback` will be passed the text that should become the next
296 296 * input cell.
297 297 */
298 298 Kernel.prototype.execute = function (code, callbacks, options) {
299 299
300 300 var content = {
301 301 code : code,
302 302 silent : true,
303 303 store_history : false,
304 304 user_variables : [],
305 305 user_expressions : {},
306 306 allow_stdin : false
307 307 };
308 308 callbacks = callbacks || {};
309 309 if (callbacks.input_request !== undefined) {
310 310 content.allow_stdin = true;
311 311 }
312 312 $.extend(true, content, options)
313 313 $([IPython.events]).trigger('execution_request.Kernel', {kernel: this, content:content});
314 314 var msg = this._get_msg("execute_request", content);
315 315 this.shell_channel.send(JSON.stringify(msg));
316 316 this.set_callbacks_for_msg(msg.header.msg_id, callbacks);
317 317 return msg.header.msg_id;
318 318 };
319 319
320 320 /**
321 321 * When calling this method pass a callbacks structure of the form:
322 322 *
323 323 * callbacks = {
324 324 * 'complete_reply': complete_reply_callback
325 325 * }
326 326 *
327 327 * The `complete_reply_callback` will be passed the content object of the
328 328 * `complete_reply` message documented
329 329 * [here](http://ipython.org/ipython-doc/dev/development/messaging.html#complete)
330 330 *
331 331 * @method complete
332 332 * @param line {integer}
333 333 * @param cursor_pos {integer}
334 334 * @param {dict} callbacks
335 335 * @param callbacks.complete_reply {function} `complete_reply_callback`
336 336 *
337 337 */
338 338 Kernel.prototype.complete = function (line, cursor_pos, callbacks) {
339 339 callbacks = callbacks || {};
340 340 var content = {
341 341 text : '',
342 342 line : line,
343 343 block : null,
344 344 cursor_pos : cursor_pos
345 345 };
346 346 var msg = this._get_msg("complete_request", content);
347 347 this.shell_channel.send(JSON.stringify(msg));
348 348 this.set_callbacks_for_msg(msg.header.msg_id, callbacks);
349 349 return msg.header.msg_id;
350 350 };
351 351
352 352
353 353 Kernel.prototype.interrupt = function () {
354 354 if (this.running) {
355 355 $([IPython.events]).trigger('status_interrupting.Kernel', {kernel: this});
356 356 $.post(this.kernel_url + "/interrupt");
357 357 };
358 358 };
359 359
360 360
361 361 Kernel.prototype.kill = function () {
362 362 if (this.running) {
363 363 this.running = false;
364 364 var settings = {
365 365 cache : false,
366 366 type : "DELETE"
367 367 };
368 368 $.ajax(this.kernel_url, settings);
369 369 };
370 370 };
371 371
372 372 Kernel.prototype.send_input_reply = function (input) {
373 373 var content = {
374 374 value : input,
375 375 };
376 376 $([IPython.events]).trigger('input_reply.Kernel', {kernel: this, content:content});
377 377 var msg = this._get_msg("input_reply", content);
378 378 this.stdin_channel.send(JSON.stringify(msg));
379 379 return msg.header.msg_id;
380 380 };
381 381
382 382
383 383 // Reply handlers
384 384
385 385 Kernel.prototype.get_callbacks_for_msg = function (msg_id) {
386 386 var callbacks = this._msg_callbacks[msg_id];
387 387 return callbacks;
388 388 };
389 389
390 390
391 391 Kernel.prototype.set_callbacks_for_msg = function (msg_id, callbacks) {
392 392 this._msg_callbacks[msg_id] = callbacks || {};
393 393 }
394 394
395 395
396 396 Kernel.prototype._handle_shell_reply = function (e) {
397 397 var reply = $.parseJSON(e.data);
398 398 $([IPython.events]).trigger('shell_reply.Kernel', {kernel: this, reply:reply});
399 399 var header = reply.header;
400 400 var content = reply.content;
401 401 var metadata = reply.metadata;
402 402 var msg_type = header.msg_type;
403 403 var callbacks = this.get_callbacks_for_msg(reply.parent_header.msg_id);
404 404 if (callbacks !== undefined) {
405 405 var cb = callbacks[msg_type];
406 406 if (cb !== undefined) {
407 407 cb(content, metadata);
408 408 }
409 409 };
410 410
411 411 if (content.payload !== undefined) {
412 412 var payload = content.payload || [];
413 413 this._handle_payload(callbacks, payload);
414 414 }
415 415 };
416 416
417 417
418 418 Kernel.prototype._handle_payload = function (callbacks, payload) {
419 419 var l = payload.length;
420 420 // Payloads are handled by triggering events because we don't want the Kernel
421 421 // to depend on the Notebook or Pager classes.
422 422 for (var i=0; i<l; i++) {
423 423 if (payload[i].source === 'page') {
424 424 var data = {'text':payload[i].text}
425 425 $([IPython.events]).trigger('open_with_text.Pager', data);
426 426 } else if (payload[i].source === 'set_next_input') {
427 427 if (callbacks.set_next_input !== undefined) {
428 428 callbacks.set_next_input(payload[i].text)
429 429 }
430 430 }
431 431 };
432 432 };
433 433
434 434
435 435 Kernel.prototype._handle_iopub_reply = function (e) {
436 436 var reply = $.parseJSON(e.data);
437 437 var content = reply.content;
438 438 var msg_type = reply.header.msg_type;
439 439 var metadata = reply.metadata;
440 440 var callbacks = this.get_callbacks_for_msg(reply.parent_header.msg_id);
441 441 if (msg_type !== 'status' && callbacks === undefined) {
442 442 // Message not from one of this notebook's cells and there are no
443 443 // callbacks to handle it.
444 444 return;
445 445 }
446 446 var output_types = ['stream','display_data','pyout','pyerr'];
447 447 if (output_types.indexOf(msg_type) >= 0) {
448 448 var cb = callbacks['output'];
449 449 if (cb !== undefined) {
450 450 cb(msg_type, content, metadata);
451 451 }
452 452 } else if (msg_type === 'status') {
453 453 if (content.execution_state === 'busy') {
454 454 $([IPython.events]).trigger('status_busy.Kernel', {kernel: this});
455 455 } else if (content.execution_state === 'idle') {
456 456 $([IPython.events]).trigger('status_idle.Kernel', {kernel: this});
457 457 } else if (content.execution_state === 'restarting') {
458 458 // autorestarting is distinct from restarting,
459 459 // in that it means the kernel died and the server is restarting it.
460 460 // status_restarting sets the notification widget,
461 461 // autorestart shows the more prominent dialog.
462 462 $([IPython.events]).trigger('status_autorestarting.Kernel', {kernel: this});
463 463 $([IPython.events]).trigger('status_restarting.Kernel', {kernel: this});
464 464 } else if (content.execution_state === 'dead') {
465 465 this.stop_channels();
466 466 $([IPython.events]).trigger('status_dead.Kernel', {kernel: this});
467 467 };
468 468 } else if (msg_type === 'clear_output') {
469 469 var cb = callbacks['clear_output'];
470 470 if (cb !== undefined) {
471 471 cb(content, metadata);
472 472 }
473 473 };
474 474 };
475 475
476 476
477 477 Kernel.prototype._handle_input_request = function (e) {
478 478 var request = $.parseJSON(e.data);
479 479 var header = request.header;
480 480 var content = request.content;
481 481 var metadata = request.metadata;
482 482 var msg_type = header.msg_type;
483 483 if (msg_type !== 'input_request') {
484 484 console.log("Invalid input request!", request);
485 485 return;
486 486 }
487 487 var callbacks = this.get_callbacks_for_msg(request.parent_header.msg_id);
488 488 if (callbacks !== undefined) {
489 489 var cb = callbacks[msg_type];
490 490 if (cb !== undefined) {
491 491 cb(content, metadata);
492 492 }
493 493 };
494 494 };
495 495
496 496
497 497 IPython.Kernel = Kernel;
498 498
499 499 return IPython;
500 500
501 501 }(IPython));
502 502
General Comments 0
You need to be logged in to leave comments. Login now