##// END OF EJS Templates
Rename history retrieval methods, and improve docstrings.
Thomas Kluyver -
Show More
@@ -1,667 +1,709 b''
1 1 """ History related magics and functionality """
2 2 #-----------------------------------------------------------------------------
3 3 # Copyright (C) 2010 The IPython Development Team.
4 4 #
5 5 # Distributed under the terms of the BSD License.
6 6 #
7 7 # The full license is in the file COPYING.txt, distributed with this software.
8 8 #-----------------------------------------------------------------------------
9 9
10 10 #-----------------------------------------------------------------------------
11 11 # Imports
12 12 #-----------------------------------------------------------------------------
13 13 from __future__ import print_function
14 14
15 15 # Stdlib imports
16 16 import datetime
17 17 import json
18 18 import os
19 19 import re
20 20 import sqlite3
21 21
22 22 from collections import defaultdict
23 23
24 24 # Our own packages
25 25 from IPython.config.configurable import Configurable
26 26 import IPython.utils.io
27 27
28 28 from IPython.testing import decorators as testdec
29 29 from IPython.utils.io import ask_yes_no
30 30 from IPython.utils.traitlets import Bool, Dict, Instance, Int, List, Unicode
31 31 from IPython.utils.warn import warn
32 32
33 33 #-----------------------------------------------------------------------------
34 34 # Classes and functions
35 35 #-----------------------------------------------------------------------------
36 36
37 37 class HistoryManager(Configurable):
38 38 """A class to organize all history-related functionality in one place.
39 39 """
40 40 # Public interface
41 41
42 42 # An instance of the IPython shell we are attached to
43 43 shell = Instance('IPython.core.interactiveshell.InteractiveShellABC')
44 44 # Lists to hold processed and raw history. These start with a blank entry
45 45 # so that we can index them starting from 1
46 46 input_hist_parsed = List([""])
47 47 input_hist_raw = List([""])
48 48 # A list of directories visited during session
49 49 dir_hist = List()
50 50 # A dict of output history, keyed with ints from the shell's
51 51 # execution count. If there are several outputs from one command,
52 52 # only the last one is stored.
53 53 output_hist = Dict()
54 54 # Contains all outputs, in lists of reprs.
55 55 output_hist_reprs = Instance(defaultdict)
56 56
57 57 # String holding the path to the history file
58 58 hist_file = Unicode()
59 59 # The SQLite database
60 60 db = Instance(sqlite3.Connection)
61 61 # The number of the current session in the history database
62 62 session_number = Int()
63 63 # Should we log output to the database? (default no)
64 64 db_log_output = Bool(False, config=True)
65 65 # Write to database every x commands (higher values save disk access & power)
66 66 # Values of 1 or less effectively disable caching.
67 67 db_cache_size = Int(0, config=True)
68 68 # The input and output caches
69 69 db_input_cache = List()
70 70 db_output_cache = List()
71 71
72 72 # Private interface
73 73 # Variables used to store the three last inputs from the user. On each new
74 74 # history update, we populate the user's namespace with these, shifted as
75 75 # necessary.
76 76 _i00, _i, _ii, _iii = '','','',''
77 77
78 78 # A set with all forms of the exit command, so that we don't store them in
79 79 # the history (it's annoying to rewind the first entry and land on an exit
80 80 # call).
81 81 _exit_commands = None
82 82
83 83 def __init__(self, shell, config=None):
84 84 """Create a new history manager associated with a shell instance.
85 85 """
86 86 # We need a pointer back to the shell for various tasks.
87 87 super(HistoryManager, self).__init__(shell=shell, config=config)
88 88
89 89 # list of visited directories
90 90 try:
91 91 self.dir_hist = [os.getcwd()]
92 92 except OSError:
93 93 self.dir_hist = []
94 94
95 95 # Now the history file
96 96 if shell.profile:
97 97 histfname = 'history-%s' % shell.profile
98 98 else:
99 99 histfname = 'history'
100 100 self.hist_file = os.path.join(shell.ipython_dir, histfname + '.sqlite')
101 101 self.init_db()
102 102 self.new_session()
103 103
104 104 self._i00, self._i, self._ii, self._iii = '','','',''
105 105 self.output_hist_reprs = defaultdict(list)
106 106
107 107 self._exit_commands = set(['Quit', 'quit', 'Exit', 'exit', '%Quit',
108 108 '%quit', '%Exit', '%exit'])
109 109
110 110 def init_db(self):
111 111 """Connect to the database, and create tables if necessary."""
112 112 self.db = sqlite3.connect(self.hist_file)
113 113 self.db.execute("""CREATE TABLE IF NOT EXISTS sessions (session integer
114 114 primary key autoincrement, start timestamp,
115 115 end timestamp, num_cmds integer, remark text)""")
116 116 self.db.execute("""CREATE TABLE IF NOT EXISTS history
117 117 (session integer, line integer, source text, source_raw text,
118 118 PRIMARY KEY (session, line))""")
119 119 # Output history is optional, but ensure the table's there so it can be
120 120 # enabled later.
121 121 self.db.execute("""CREATE TABLE IF NOT EXISTS output_history
122 122 (session integer, line integer, output text,
123 123 PRIMARY KEY (session, line))""")
124 124 self.db.commit()
125 125
126 126 def new_session(self):
127 127 """Get a new session number."""
128 128 with self.db:
129 129 cur = self.db.execute("""INSERT INTO sessions VALUES (NULL, ?, NULL,
130 130 NULL, "") """, (datetime.datetime.now(),))
131 131 self.session_number = cur.lastrowid
132 132
133 133 def end_session(self):
134 134 """Close the database session, filling in the end time and line count."""
135 135 self.writeout_cache()
136 136 with self.db:
137 137 self.db.execute("""UPDATE sessions SET end=?, num_cmds=? WHERE
138 138 session==?""", (datetime.datetime.now(),
139 139 len(self.input_hist_parsed)-1, self.session_number))
140 140 self.session_number = 0
141 141
142 142 def name_session(self, name):
143 143 """Give the current session a name in the history database."""
144 144 with self.db:
145 145 self.db.execute("UPDATE sessions SET remark=? WHERE session==?",
146 146 (name, self.session_number))
147 147
148 148 def reset(self, new_session=True):
149 149 """Clear the session history, releasing all object references, and
150 150 optionally open a new session."""
151 151 if self.session_number:
152 152 self.end_session()
153 153 self.input_hist_parsed[:] = [""]
154 154 self.input_hist_raw[:] = [""]
155 155 self.output_hist.clear()
156 156 # The directory history can't be completely empty
157 157 self.dir_hist[:] = [os.getcwd()]
158 158
159 159 if new_session:
160 160 self.new_session()
161 161
162 162 ## -------------------------------
163 163 ## Methods for retrieving history:
164 164 ## -------------------------------
165 def _get_hist_sql(self, sql, params, raw=True, output=False):
165 def _run_sql(self, sql, params, raw=True, output=False):
166 166 """Prepares and runs an SQL query for the history database.
167 167
168 168 Parameters
169 169 ----------
170 170 sql : str
171 171 Any filtering expressions to go after SELECT ... FROM ...
172 172 params : tuple
173 173 Parameters passed to the SQL query (to replace "?")
174 raw : bool
175 If True, get raw input.
176 output : bool
177 If True, include output where available.
174 raw, output : bool
175 See :meth:`get_range`
178 176
179 177 Returns
180 178 -------
181 An iterator over 3-tuples: (session, line_number, command), or if output
182 is True, (session, line_number, (command, output)).
179 Tuples as :meth:`get_range`
183 180 """
184 181 toget = 'source_raw' if raw else 'source'
185 182 sqlfrom = "history"
186 183 if output:
187 184 sqlfrom = "history LEFT JOIN output_history USING (session, line)"
188 185 toget = "history.%s, output_history.output" % toget
189 186 cur = self.db.execute("SELECT session, line, %s FROM %s " %\
190 187 (toget, sqlfrom) + sql, params)
191 188 if output: # Regroup into 3-tuples, and parse JSON
192 189 loads = lambda out: json.loads(out) if out else None
193 190 return ((ses, lin, (inp, loads(out))) \
194 191 for ses, lin, inp, out in cur)
195 192 return cur
196 193
197 194
198 def get_hist_tail(self, n=10, raw=True, output=False, include_latest=False):
195 def get_tail(self, n=10, raw=True, output=False, include_latest=False):
199 196 """Get the last n lines from the history database.
200 197
201 If include_latest is False (default), n+1 lines are fetched, and
202 the latest one is discarded. This is intended to be used where
203 the function is called by a user command, which it should not
204 return."""
198 Parameters
199 ----------
200 n : int
201 The number of lines to get
202 raw, output : bool
203 See :meth:`get_range`
204 include_latest : bool
205 If False (default), n+1 lines are fetched, and the latest one
206 is discarded. This is intended to be used where the function
207 is called by a user command, which it should not return.
208
209 Returns
210 -------
211 Tuples as :meth:`get_range`
212 """
205 213 self.writeout_cache()
206 214 if not include_latest:
207 215 n += 1
208 cur = self._get_hist_sql("ORDER BY session DESC, line DESC LIMIT ?",
216 cur = self._run_sql("ORDER BY session DESC, line DESC LIMIT ?",
209 217 (n,), raw=raw, output=output)
210 218 if not include_latest:
211 219 return reversed(list(cur)[1:])
212 220 return reversed(list(cur))
213 221
214 def get_hist_search(self, pattern="*", raw=True, search_raw=True,
222 def search(self, pattern="*", raw=True, search_raw=True,
215 223 output=False):
216 224 """Search the database using unix glob-style matching (wildcards
217 225 * and ?).
218 226
227 Parameters
228 ----------
229 pattern : str
230 The wildcarded pattern to match when searching
231 search_raw : bool
232 If True, search the raw input, otherwise, the parsed input
233 raw, output : bool
234 See :meth:`get_range`
235
219 236 Returns
220 237 -------
221 An iterator over tuples: (session, line_number, command)
238 Tuples as :meth:`get_range`
222 239 """
223 240 tosearch = "source_raw" if search_raw else "source"
224 241 if output:
225 242 tosearch = "history." + tosearch
226 243 self.writeout_cache()
227 return self._get_hist_sql("WHERE %s GLOB ?" % tosearch, (pattern,),
244 return self._run_sql("WHERE %s GLOB ?" % tosearch, (pattern,),
228 245 raw=raw, output=output)
229 246
230 def _get_hist_session(self, start=1, stop=None, raw=True, output=False):
247 def _get_range_session(self, start=1, stop=None, raw=True, output=False):
231 248 """Get input and output history from the current session. Called by
232 get_history, and takes similar parameters."""
249 get_range, and takes similar parameters."""
233 250 input_hist = self.input_hist_raw if raw else self.input_hist_parsed
234 251
235 252 n = len(input_hist)
236 253 if start < 0:
237 254 start += n
238 255 if not stop:
239 256 stop = n
240 257 elif stop < 0:
241 258 stop += n
242 259
243 260 for i in range(start, stop):
244 261 if output:
245 262 line = (input_hist[i], self.output_hist_reprs.get(i))
246 263 else:
247 264 line = input_hist[i]
248 265 yield (0, i, line)
249 266
250 def get_history(self, session=0, start=1, stop=None, raw=True,output=False):
267 def get_range(self, session=0, start=1, stop=None, raw=True,output=False):
251 268 """Retrieve input by session.
252 269
253 270 Parameters
254 271 ----------
255 272 session : int
256 273 Session number to retrieve. The current session is 0, and negative
257 274 numbers count back from current session, so -1 is previous session.
258 275 start : int
259 276 First line to retrieve.
260 277 stop : int
261 278 End of line range (excluded from output itself). If None, retrieve
262 279 to the end of the session.
263 280 raw : bool
264 281 If True, return untranslated input
265 282 output : bool
266 283 If True, attempt to include output. This will be 'real' Python
267 284 objects for the current session, or text reprs from previous
268 285 sessions if db_log_output was enabled at the time. Where no output
269 286 is found, None is used.
270 287
271 288 Returns
272 289 -------
273 290 An iterator over the desired lines. Each line is a 3-tuple, either
274 291 (session, line, input) if output is False, or
275 292 (session, line, (input, output)) if output is True.
276 293 """
277 294 if session == 0 or session==self.session_number: # Current session
278 return self._get_hist_session(start, stop, raw, output)
295 return self._get_range_session(start, stop, raw, output)
279 296 if session < 0:
280 297 session += self.session_number
281 298
282 299 if stop:
283 300 lineclause = "line >= ? AND line < ?"
284 301 params = (session, start, stop)
285 302 else:
286 303 lineclause = "line>=?"
287 304 params = (session, start)
288 305
289 return self._get_hist_sql("WHERE session==? AND %s""" % lineclause,
306 return self._run_sql("WHERE session==? AND %s""" % lineclause,
290 307 params, raw=raw, output=output)
291 308
292 def get_hist_from_rangestr(self, rangestr, raw=True, output=False):
309 def get_range_by_str(self, rangestr, raw=True, output=False):
293 310 """Get lines of history from a string of ranges, as used by magic
294 commands %hist, %save, %macro, etc."""
311 commands %hist, %save, %macro, etc.
312
313 Parameters
314 ----------
315 rangestr : str
316 A string specifying ranges, e.g. "5 ~2/1-4". See
317 :func:`magic_history` for full details.
318 raw, output : bool
319 As :meth:`get_range`
320
321 Returns
322 -------
323 Tuples as :meth:`get_range`
324 """
295 325 for sess, s, e in extract_hist_ranges(rangestr):
296 for line in self.get_history(sess, s, e, raw=raw, output=output):
326 for line in self.get_range(sess, s, e, raw=raw, output=output):
297 327 yield line
298 328
299 329 ## ----------------------------
300 330 ## Methods for storing history:
301 331 ## ----------------------------
302 332 def store_inputs(self, line_num, source, source_raw=None):
303 333 """Store source and raw input in history and create input cache
304 334 variables _i*.
305 335
306 336 Parameters
307 337 ----------
308 338 line_num : int
309 339 The prompt number of this input.
310 340
311 341 source : str
312 342 Python input.
313 343
314 344 source_raw : str, optional
315 345 If given, this is the raw input without any IPython transformations
316 346 applied to it. If not given, ``source`` is used.
317 347 """
318 348 if source_raw is None:
319 349 source_raw = source
320 350 source = source.rstrip('\n')
321 351 source_raw = source_raw.rstrip('\n')
322 352
323 353 # do not store exit/quit commands
324 354 if source_raw.strip() in self._exit_commands:
325 355 return
326 356
327 357 self.input_hist_parsed.append(source)
328 358 self.input_hist_raw.append(source_raw)
329 359
330 360 self.db_input_cache.append((self.session_number, line_num,
331 361 source, source_raw))
332 362 # Trigger to flush cache and write to DB.
333 363 if len(self.db_input_cache) >= self.db_cache_size:
334 364 self.writeout_cache()
335 365
336 366 # update the auto _i variables
337 367 self._iii = self._ii
338 368 self._ii = self._i
339 369 self._i = self._i00
340 370 self._i00 = source_raw
341 371
342 372 # hackish access to user namespace to create _i1,_i2... dynamically
343 373 new_i = '_i%s' % line_num
344 374 to_main = {'_i': self._i,
345 375 '_ii': self._ii,
346 376 '_iii': self._iii,
347 377 new_i : self._i00 }
348 378 self.shell.user_ns.update(to_main)
349 379
350 380 def store_output(self, line_num):
351 381 """If database output logging is enabled, this saves all the
352 382 outputs from the indicated prompt number to the database. It's
353 called by run_cell after code has been executed."""
383 called by run_cell after code has been executed.
384
385 Parameters
386 ----------
387 line_num : int
388 The line number from which to save outputs
389 """
354 390 if (not self.db_log_output) or not self.output_hist_reprs[line_num]:
355 391 return
356 392 output = json.dumps(self.output_hist_reprs[line_num])
357 393 db_row = (self.session_number, line_num, output)
358 394 if self.db_cache_size > 1:
359 395 self.db_output_cache.append(db_row)
360 396 else:
361 397 with self.db:
362 398 self.db.execute("INSERT INTO output_history VALUES (?,?,?)", db_row)
363 399
364 400 def writeout_cache(self):
365 401 #print(self.db_input_cache)
366 402 with self.db:
367 403 self.db.executemany("INSERT INTO history VALUES (?, ?, ?, ?)",
368 404 self.db_input_cache)
369 405 self.db.executemany("INSERT INTO output_history VALUES (?, ?, ?)",
370 406 self.db_output_cache)
371 407 self.db_input_cache = []
372 408 self.db_output_cache = []
373 409
374 410
375 411 # To match, e.g. ~5/8-~2/3
376 412 range_re = re.compile(r"""
377 413 ((?P<startsess>~?\d+)/)?
378 414 (?P<start>\d+) # Only the start line num is compulsory
379 415 ((?P<sep>[\-:])
380 416 ((?P<endsess>~?\d+)/)?
381 417 (?P<end>\d+))?
382 418 """, re.VERBOSE)
383 419
384 420 def extract_hist_ranges(ranges_str):
385 421 """Turn a string of history ranges into 3-tuples of (session, start, stop).
386 422
387 423 Examples
388 424 --------
389 425 list(extract_input_ranges("~8/5-~7/4 2"))
390 426 [(-8, 5, None), (-7, 1, 4), (0, 2, 3)]
391 427 """
392 428 for range_str in ranges_str.split():
393 429 rmatch = range_re.match(range_str)
394 430 if not rmatch:
395 431 continue
396 432 start = int(rmatch.group("start"))
397 433 end = rmatch.group("end")
398 434 end = int(end) if end else start+1 # If no end specified, get (a, a+1)
399 435 if rmatch.group("sep") == "-": # 1-3 == 1:4 --> [1, 2, 3]
400 436 end += 1
401 437 startsess = rmatch.group("startsess") or "0"
402 438 endsess = rmatch.group("endsess") or startsess
403 439 startsess = int(startsess.replace("~","-"))
404 440 endsess = int(endsess.replace("~","-"))
405 441 assert endsess >= startsess
406 442
407 443 if endsess == startsess:
408 444 yield (startsess, start, end)
409 445 continue
410 446 # Multiple sessions in one range:
411 447 yield (startsess, start, None)
412 448 for sess in range(startsess+1, endsess):
413 449 yield (sess, 1, None)
414 450 yield (endsess, 1, end)
415 451
416 452 def _format_lineno(session, line):
417 453 """Helper function to format line numbers properly."""
418 454 if session == 0:
419 455 return str(line)
420 456 return "%s#%s" % (session, line)
421 457
422 458 @testdec.skip_doctest
423 459 def magic_history(self, parameter_s = ''):
424 460 """Print input history (_i<n> variables), with most recent last.
425 461
426 462 %history -> print at most 40 inputs (some may be multi-line)\\
427 463 %history n -> print at most n inputs\\
428 464 %history n1 n2 -> print inputs between n1 and n2 (n2 not included)\\
429 465
430 466 By default, input history is printed without line numbers so it can be
431 directly pasted into an editor.
432
433 With -n, each input's number <n> is shown, and is accessible as the
434 automatically generated variable _i<n> as well as In[<n>]. Multi-line
435 statements are printed starting at a new line for easy copy/paste.
467 directly pasted into an editor. Use -n to show them.
468
469 Ranges of history can be indicated using the syntax:
470 4 : Line 4, current session
471 4-6 : Lines 4-6, current session
472 243/1-5: Lines 1-5, session 243
473 ~2/7 : Line 7, session 2 before current
474 ~8/1-~6/5 : From the first line of 8 sessions ago, to the fifth line
475 of 6 sessions ago.
476 Multiple ranges can be entered, separated by spaces
477
478 The same syntax is used by %macro, %save, %edit, %rerun
436 479
437 480 Options:
438 481
439 482 -n: print line numbers for each input.
440 483 This feature is only available if numbered prompts are in use.
441 484
442 485 -o: also print outputs for each input.
443 486
444 487 -p: print classic '>>>' python prompts before each input. This is useful
445 488 for making documentation, and in conjunction with -o, for producing
446 489 doctest-ready output.
447 490
448 491 -r: (default) print the 'raw' history, i.e. the actual commands you typed.
449 492
450 493 -t: print the 'translated' history, as IPython understands it. IPython
451 494 filters your input and converts it all into valid Python source before
452 495 executing it (things like magics or aliases are turned into function
453 496 calls, for example). With this option, you'll see the native history
454 497 instead of the user-entered version: '%cd /' will be seen as
455 498 'get_ipython().magic("%cd /")' instead of '%cd /'.
456 499
457 500 -g: treat the arg as a pattern to grep for in (full) history.
458 501 This includes the saved history (almost all commands ever written).
459 502 Use '%hist -g' to show full saved history (may be very long).
460 503
461 504 -l: get the last n lines from all sessions. Specify n as a single arg, or
462 505 the default is the last 10 lines.
463 506
464 507 -f FILENAME: instead of printing the output to the screen, redirect it to
465 508 the given file. The file is always overwritten, though IPython asks for
466 509 confirmation first if it already exists.
467 510
468 511 Examples
469 512 --------
470 513 ::
471 514
472 515 In [6]: %hist -n 4 6
473 516 4:a = 12
474 517 5:print a**2
475 518
476 519 """
477 520
478 521 if not self.shell.displayhook.do_full_cache:
479 522 print('This feature is only available if numbered prompts are in use.')
480 523 return
481 524 opts,args = self.parse_options(parameter_s,'noprtglf:',mode='string')
482 525
483 526 # For brevity
484 527 history_manager = self.shell.history_manager
485 528
486 529 def _format_lineno(session, line):
487 530 """Helper function to format line numbers properly."""
488 531 if session in (0, history_manager.session_number):
489 532 return str(line)
490 533 return "%s/%s" % (session, line)
491 534
492 535 # Check if output to specific file was requested.
493 536 try:
494 537 outfname = opts['f']
495 538 except KeyError:
496 539 outfile = IPython.utils.io.Term.cout # default
497 540 # We don't want to close stdout at the end!
498 541 close_at_end = False
499 542 else:
500 543 if os.path.exists(outfname):
501 544 if not ask_yes_no("File %r exists. Overwrite?" % outfname):
502 545 print('Aborting.')
503 546 return
504 547
505 548 outfile = open(outfname,'w')
506 549 close_at_end = True
507 550
508 551 print_nums = 'n' in opts
509 552 get_output = 'o' in opts
510 553 pyprompts = 'p' in opts
511 554 # Raw history is the default
512 555 raw = not('t' in opts)
513 556
514 557 default_length = 40
515 558 pattern = None
516 559
517 560 if 'g' in opts: # Glob search
518 561 pattern = "*" + args + "*" if args else "*"
519 hist = history_manager.get_hist_search(pattern, raw=raw,
520 output=get_output)
562 hist = history_manager.search(pattern, raw=raw, output=get_output)
521 563 elif 'l' in opts: # Get 'tail'
522 564 try:
523 565 n = int(args)
524 566 except ValueError, IndexError:
525 567 n = 10
526 hist = history_manager.get_hist_tail(n, raw=raw, output=get_output)
568 hist = history_manager.get_tail(n, raw=raw, output=get_output)
527 569 else:
528 570 if args: # Get history by ranges
529 hist = history_manager.get_hist_from_rangestr(args, raw, get_output)
571 hist = history_manager.get_range_by_str(args, raw, get_output)
530 572 else: # Just get history for the current session
531 hist = history_manager.get_history(raw=raw, output=get_output)
573 hist = history_manager.get_range(raw=raw, output=get_output)
532 574
533 575 # We could be displaying the entire history, so let's not try to pull it
534 576 # into a list in memory. Anything that needs more space will just misalign.
535 577 width = 4
536 578
537 579 for session, lineno, inline in hist:
538 580 # Print user history with tabs expanded to 4 spaces. The GUI clients
539 581 # use hard tabs for easier usability in auto-indented code, but we want
540 582 # to produce PEP-8 compliant history for safe pasting into an editor.
541 583 if get_output:
542 584 inline, output = inline
543 585 inline = inline.expandtabs(4).rstrip()
544 586
545 587 multiline = "\n" in inline
546 588 line_sep = '\n' if multiline else ' '
547 589 if print_nums:
548 590 print('%s:%s' % (_format_lineno(session, lineno).rjust(width),
549 591 line_sep), file=outfile, end='')
550 592 if pyprompts:
551 593 print(">>> ", end="", file=outfile)
552 594 if multiline:
553 595 inline = "\n... ".join(inline.splitlines()) + "\n..."
554 596 print(inline, file=outfile)
555 597 if get_output and output:
556 598 print("\n".join(output), file=outfile)
557 599
558 600 if close_at_end:
559 601 outfile.close()
560 602
561 603
562 604 def magic_rep(self, arg):
563 605 r""" Repeat a command, or get command to input line for editing
564 606
565 607 - %rep (no arguments):
566 608
567 609 Place a string version of last computation result (stored in the special '_'
568 610 variable) to the next input prompt. Allows you to create elaborate command
569 611 lines without using copy-paste::
570 612
571 613 In[1]: l = ["hei", "vaan"]
572 614 In[2]: "".join(l)
573 615 Out[2]: heivaan
574 616 In[3]: %rep
575 617 In[4]: heivaan_ <== cursor blinking
576 618
577 619 %rep 45
578 620
579 621 Place history line 45 on the next input prompt. Use %hist to find
580 622 out the number.
581 623
582 %rep 1-4 6-7 3
624 %rep 1-4
583 625
584 626 Combine the specified lines into one cell, and place it on the next
585 input prompt. History slice syntax is the same as in %macro and %save.
627 input prompt. See %history for the slice syntax.
586 628
587 629 %rep foo+bar
588 630
589 631 If foo+bar can be evaluated in the user namespace, the result is
590 632 placed at the next input prompt. Otherwise, the history is searched
591 633 for lines which contain that substring, and the most recent one is
592 634 placed at the next input prompt.
593 635 """
594 636 if not arg: # Last output
595 637 self.set_next_input(str(self.shell.user_ns["_"]))
596 638 return
597 639 # Get history range
598 histlines = self.history_manager.get_hist_from_rangestr(arg)
640 histlines = self.history_manager.get_range_by_str(arg)
599 641 cmd = "\n".join(x[2] for x in histlines)
600 642 if cmd:
601 643 self.set_next_input(cmd.rstrip())
602 644 return
603 645
604 646 try: # Variable in user namespace
605 647 cmd = str(eval(arg, self.shell.user_ns))
606 648 except Exception: # Search for term in history
607 histlines = self.history_manager.get_hist_search("*"+arg+"*")
649 histlines = self.history_manager.search("*"+arg+"*")
608 650 for h in reversed([x[2] for x in histlines]):
609 651 if 'rep' in h:
610 652 continue
611 653 self.set_next_input(h.rstrip())
612 654 return
613 655 else:
614 656 self.set_next_input(cmd.rstrip())
615 657 print("Couldn't evaluate or find in history:", arg)
616 658
617 659 def magic_rerun(self, parameter_s=''):
618 660 """Re-run previous input
619 661
620 662 By default, you can specify ranges of input history to be repeated
621 (as with %hist). With no arguments, it will repeat the last line.
663 (as with %history). With no arguments, it will repeat the last line.
622 664
623 665 Options:
624 666
625 667 -l <n> : Repeat the last n lines of input, not including the
626 668 current command.
627 669
628 670 -g foo : Repeat the most recent line which contains foo
629 671 """
630 672 opts, args = self.parse_options(parameter_s, 'l:g:', mode='string')
631 673 if "l" in opts: # Last n lines
632 674 n = int(opts['l'])
633 hist = self.history_manager.get_hist_tail(n)
675 hist = self.history_manager.get_tail(n)
634 676 elif "g" in opts: # Search
635 677 p = "*"+opts['g']+"*"
636 hist = list(self.history_manager.get_hist_search(p))
678 hist = list(self.history_manager.search(p))
637 679 for l in reversed(hist):
638 680 if "rerun" not in l[2]:
639 681 hist = [l] # The last match which isn't a %rerun
640 682 break
641 683 else:
642 684 hist = [] # No matches except %rerun
643 685 elif args: # Specify history ranges
644 hist = self.history_manager.get_hist_from_rangestr(args)
686 hist = self.history_manager.get_range_by_str(args)
645 687 else: # Last line
646 hist = self.history_manager.get_hist_tail(1)
688 hist = self.history_manager.get_tail(1)
647 689 hist = [x[2] for x in hist]
648 690 if not hist:
649 691 print("No lines in history match specification")
650 692 return
651 693 histlines = "\n".join(hist)
652 694 print("=== Executing: ===")
653 695 print(histlines)
654 696 print("=== Output: ===")
655 697 self.run_cell("\n".join(hist), store_history=False)
656 698
657 699
658 700 def init_ipython(ip):
659 701 ip.define_magic("rep", magic_rep)
660 702 ip.define_magic("recall", magic_rep)
661 703 ip.define_magic("rerun", magic_rerun)
662 704 ip.define_magic("hist",magic_history) # Alternative name
663 705 ip.define_magic("history",magic_history)
664 706
665 707 # XXX - ipy_completers are in quarantine, need to be updated to new apis
666 708 #import ipy_completers
667 709 #ipy_completers.quick_completer('%hist' ,'-g -t -r -n')
@@ -1,2527 +1,2527 b''
1 1 # -*- coding: utf-8 -*-
2 2 """Main IPython class."""
3 3
4 4 #-----------------------------------------------------------------------------
5 5 # Copyright (C) 2001 Janko Hauser <jhauser@zscout.de>
6 6 # Copyright (C) 2001-2007 Fernando Perez. <fperez@colorado.edu>
7 7 # Copyright (C) 2008-2011 The IPython Development Team
8 8 #
9 9 # Distributed under the terms of the BSD License. The full license is in
10 10 # the file COPYING, distributed as part of this software.
11 11 #-----------------------------------------------------------------------------
12 12
13 13 #-----------------------------------------------------------------------------
14 14 # Imports
15 15 #-----------------------------------------------------------------------------
16 16
17 17 from __future__ import with_statement
18 18 from __future__ import absolute_import
19 19
20 20 import __builtin__
21 21 import __future__
22 22 import abc
23 23 import atexit
24 24 import codeop
25 25 import os
26 26 import re
27 27 import sys
28 28 import tempfile
29 29 import types
30 30 from contextlib import nested
31 31
32 32 from IPython.config.configurable import Configurable
33 33 from IPython.core import debugger, oinspect
34 34 from IPython.core import history as ipcorehist
35 35 from IPython.core import page
36 36 from IPython.core import prefilter
37 37 from IPython.core import shadowns
38 38 from IPython.core import ultratb
39 39 from IPython.core.alias import AliasManager
40 40 from IPython.core.builtin_trap import BuiltinTrap
41 41 from IPython.core.compilerop import CachingCompiler
42 42 from IPython.core.display_trap import DisplayTrap
43 43 from IPython.core.displayhook import DisplayHook
44 44 from IPython.core.displaypub import DisplayPublisher
45 45 from IPython.core.error import TryNext, UsageError
46 46 from IPython.core.extensions import ExtensionManager
47 47 from IPython.core.fakemodule import FakeModule, init_fakemod_dict
48 48 from IPython.core.formatters import DisplayFormatter
49 49 from IPython.core.history import HistoryManager
50 50 from IPython.core.inputsplitter import IPythonInputSplitter
51 51 from IPython.core.logger import Logger
52 52 from IPython.core.magic import Magic
53 53 from IPython.core.payload import PayloadManager
54 54 from IPython.core.plugin import PluginManager
55 55 from IPython.core.prefilter import PrefilterManager, ESC_MAGIC
56 56 from IPython.external.Itpl import ItplNS
57 57 from IPython.utils import PyColorize
58 58 from IPython.utils import io
59 59 from IPython.utils.doctestreload import doctest_reload
60 60 from IPython.utils.io import ask_yes_no, rprint
61 61 from IPython.utils.ipstruct import Struct
62 62 from IPython.utils.path import get_home_dir, get_ipython_dir, HomeDirError
63 63 from IPython.utils.pickleshare import PickleShareDB
64 64 from IPython.utils.process import system, getoutput
65 65 from IPython.utils.strdispatch import StrDispatch
66 66 from IPython.utils.syspathcontext import prepended_to_syspath
67 67 from IPython.utils.text import num_ini_spaces, format_screen, LSString, SList
68 68 from IPython.utils.traitlets import (Int, Str, CBool, CaselessStrEnum, Enum,
69 69 List, Unicode, Instance, Type)
70 70 from IPython.utils.warn import warn, error, fatal
71 71 import IPython.core.hooks
72 72
73 73 #-----------------------------------------------------------------------------
74 74 # Globals
75 75 #-----------------------------------------------------------------------------
76 76
77 77 # compiled regexps for autoindent management
78 78 dedent_re = re.compile(r'^\s+raise|^\s+return|^\s+pass')
79 79
80 80 #-----------------------------------------------------------------------------
81 81 # Utilities
82 82 #-----------------------------------------------------------------------------
83 83
84 84 # store the builtin raw_input globally, and use this always, in case user code
85 85 # overwrites it (like wx.py.PyShell does)
86 86 raw_input_original = raw_input
87 87
88 88 def softspace(file, newvalue):
89 89 """Copied from code.py, to remove the dependency"""
90 90
91 91 oldvalue = 0
92 92 try:
93 93 oldvalue = file.softspace
94 94 except AttributeError:
95 95 pass
96 96 try:
97 97 file.softspace = newvalue
98 98 except (AttributeError, TypeError):
99 99 # "attribute-less object" or "read-only attributes"
100 100 pass
101 101 return oldvalue
102 102
103 103
104 104 def no_op(*a, **kw): pass
105 105
106 106 class SpaceInInput(Exception): pass
107 107
108 108 class Bunch: pass
109 109
110 110
111 111 def get_default_colors():
112 112 if sys.platform=='darwin':
113 113 return "LightBG"
114 114 elif os.name=='nt':
115 115 return 'Linux'
116 116 else:
117 117 return 'Linux'
118 118
119 119
120 120 class SeparateStr(Str):
121 121 """A Str subclass to validate separate_in, separate_out, etc.
122 122
123 123 This is a Str based trait that converts '0'->'' and '\\n'->'\n'.
124 124 """
125 125
126 126 def validate(self, obj, value):
127 127 if value == '0': value = ''
128 128 value = value.replace('\\n','\n')
129 129 return super(SeparateStr, self).validate(obj, value)
130 130
131 131 class MultipleInstanceError(Exception):
132 132 pass
133 133
134 134
135 135 #-----------------------------------------------------------------------------
136 136 # Main IPython class
137 137 #-----------------------------------------------------------------------------
138 138
139 139 class InteractiveShell(Configurable, Magic):
140 140 """An enhanced, interactive shell for Python."""
141 141
142 142 _instance = None
143 143 autocall = Enum((0,1,2), default_value=1, config=True)
144 144 # TODO: remove all autoindent logic and put into frontends.
145 145 # We can't do this yet because even runlines uses the autoindent.
146 146 autoindent = CBool(True, config=True)
147 147 automagic = CBool(True, config=True)
148 148 cache_size = Int(1000, config=True)
149 149 color_info = CBool(True, config=True)
150 150 colors = CaselessStrEnum(('NoColor','LightBG','Linux'),
151 151 default_value=get_default_colors(), config=True)
152 152 debug = CBool(False, config=True)
153 153 deep_reload = CBool(False, config=True)
154 154 display_formatter = Instance(DisplayFormatter)
155 155 displayhook_class = Type(DisplayHook)
156 156 display_pub_class = Type(DisplayPublisher)
157 157
158 158 exit_now = CBool(False)
159 159 # Monotonically increasing execution counter
160 160 execution_count = Int(1)
161 161 filename = Str("<ipython console>")
162 162 ipython_dir= Unicode('', config=True) # Set to get_ipython_dir() in __init__
163 163
164 164 # Input splitter, to split entire cells of input into either individual
165 165 # interactive statements or whole blocks.
166 166 input_splitter = Instance('IPython.core.inputsplitter.IPythonInputSplitter',
167 167 (), {})
168 168 logstart = CBool(False, config=True)
169 169 logfile = Str('', config=True)
170 170 logappend = Str('', config=True)
171 171 object_info_string_level = Enum((0,1,2), default_value=0,
172 172 config=True)
173 173 pdb = CBool(False, config=True)
174 174
175 175 profile = Str('', config=True)
176 176 prompt_in1 = Str('In [\\#]: ', config=True)
177 177 prompt_in2 = Str(' .\\D.: ', config=True)
178 178 prompt_out = Str('Out[\\#]: ', config=True)
179 179 prompts_pad_left = CBool(True, config=True)
180 180 quiet = CBool(False, config=True)
181 181
182 182 history_length = Int(10000, config=True)
183 183
184 184 # The readline stuff will eventually be moved to the terminal subclass
185 185 # but for now, we can't do that as readline is welded in everywhere.
186 186 readline_use = CBool(True, config=True)
187 187 readline_merge_completions = CBool(True, config=True)
188 188 readline_omit__names = Enum((0,1,2), default_value=2, config=True)
189 189 readline_remove_delims = Str('-/~', config=True)
190 190 readline_parse_and_bind = List([
191 191 'tab: complete',
192 192 '"\C-l": clear-screen',
193 193 'set show-all-if-ambiguous on',
194 194 '"\C-o": tab-insert',
195 195 '"\M-i": " "',
196 196 '"\M-o": "\d\d\d\d"',
197 197 '"\M-I": "\d\d\d\d"',
198 198 '"\C-r": reverse-search-history',
199 199 '"\C-s": forward-search-history',
200 200 '"\C-p": history-search-backward',
201 201 '"\C-n": history-search-forward',
202 202 '"\e[A": history-search-backward',
203 203 '"\e[B": history-search-forward',
204 204 '"\C-k": kill-line',
205 205 '"\C-u": unix-line-discard',
206 206 ], allow_none=False, config=True)
207 207
208 208 # TODO: this part of prompt management should be moved to the frontends.
209 209 # Use custom TraitTypes that convert '0'->'' and '\\n'->'\n'
210 210 separate_in = SeparateStr('\n', config=True)
211 211 separate_out = SeparateStr('', config=True)
212 212 separate_out2 = SeparateStr('', config=True)
213 213 wildcards_case_sensitive = CBool(True, config=True)
214 214 xmode = CaselessStrEnum(('Context','Plain', 'Verbose'),
215 215 default_value='Context', config=True)
216 216
217 217 # Subcomponents of InteractiveShell
218 218 alias_manager = Instance('IPython.core.alias.AliasManager')
219 219 prefilter_manager = Instance('IPython.core.prefilter.PrefilterManager')
220 220 builtin_trap = Instance('IPython.core.builtin_trap.BuiltinTrap')
221 221 display_trap = Instance('IPython.core.display_trap.DisplayTrap')
222 222 extension_manager = Instance('IPython.core.extensions.ExtensionManager')
223 223 plugin_manager = Instance('IPython.core.plugin.PluginManager')
224 224 payload_manager = Instance('IPython.core.payload.PayloadManager')
225 225 history_manager = Instance('IPython.core.history.HistoryManager')
226 226
227 227 # Private interface
228 228 _post_execute = set()
229 229
230 230 def __init__(self, config=None, ipython_dir=None,
231 231 user_ns=None, user_global_ns=None,
232 232 custom_exceptions=((), None)):
233 233
234 234 # This is where traits with a config_key argument are updated
235 235 # from the values on config.
236 236 super(InteractiveShell, self).__init__(config=config)
237 237
238 238 # These are relatively independent and stateless
239 239 self.init_ipython_dir(ipython_dir)
240 240 self.init_instance_attrs()
241 241 self.init_environment()
242 242
243 243 # Create namespaces (user_ns, user_global_ns, etc.)
244 244 self.init_create_namespaces(user_ns, user_global_ns)
245 245 # This has to be done after init_create_namespaces because it uses
246 246 # something in self.user_ns, but before init_sys_modules, which
247 247 # is the first thing to modify sys.
248 248 # TODO: When we override sys.stdout and sys.stderr before this class
249 249 # is created, we are saving the overridden ones here. Not sure if this
250 250 # is what we want to do.
251 251 self.save_sys_module_state()
252 252 self.init_sys_modules()
253 253
254 254 # While we're trying to have each part of the code directly access what
255 255 # it needs without keeping redundant references to objects, we have too
256 256 # much legacy code that expects ip.db to exist.
257 257 self.db = PickleShareDB(os.path.join(self.ipython_dir, 'db'))
258 258
259 259 self.init_history()
260 260 self.init_encoding()
261 261 self.init_prefilter()
262 262
263 263 Magic.__init__(self, self)
264 264
265 265 self.init_syntax_highlighting()
266 266 self.init_hooks()
267 267 self.init_pushd_popd_magic()
268 268 # self.init_traceback_handlers use to be here, but we moved it below
269 269 # because it and init_io have to come after init_readline.
270 270 self.init_user_ns()
271 271 self.init_logger()
272 272 self.init_alias()
273 273 self.init_builtins()
274 274
275 275 # pre_config_initialization
276 276
277 277 # The next section should contain everything that was in ipmaker.
278 278 self.init_logstart()
279 279
280 280 # The following was in post_config_initialization
281 281 self.init_inspector()
282 282 # init_readline() must come before init_io(), because init_io uses
283 283 # readline related things.
284 284 self.init_readline()
285 285 # init_completer must come after init_readline, because it needs to
286 286 # know whether readline is present or not system-wide to configure the
287 287 # completers, since the completion machinery can now operate
288 288 # independently of readline (e.g. over the network)
289 289 self.init_completer()
290 290 # TODO: init_io() needs to happen before init_traceback handlers
291 291 # because the traceback handlers hardcode the stdout/stderr streams.
292 292 # This logic in in debugger.Pdb and should eventually be changed.
293 293 self.init_io()
294 294 self.init_traceback_handlers(custom_exceptions)
295 295 self.init_prompts()
296 296 self.init_display_formatter()
297 297 self.init_display_pub()
298 298 self.init_displayhook()
299 299 self.init_reload_doctest()
300 300 self.init_magics()
301 301 self.init_pdb()
302 302 self.init_extension_manager()
303 303 self.init_plugin_manager()
304 304 self.init_payload()
305 305 self.hooks.late_startup_hook()
306 306 atexit.register(self.atexit_operations)
307 307
308 308 @classmethod
309 309 def instance(cls, *args, **kwargs):
310 310 """Returns a global InteractiveShell instance."""
311 311 if cls._instance is None:
312 312 inst = cls(*args, **kwargs)
313 313 # Now make sure that the instance will also be returned by
314 314 # the subclasses instance attribute.
315 315 for subclass in cls.mro():
316 316 if issubclass(cls, subclass) and \
317 317 issubclass(subclass, InteractiveShell):
318 318 subclass._instance = inst
319 319 else:
320 320 break
321 321 if isinstance(cls._instance, cls):
322 322 return cls._instance
323 323 else:
324 324 raise MultipleInstanceError(
325 325 'Multiple incompatible subclass instances of '
326 326 'InteractiveShell are being created.'
327 327 )
328 328
329 329 @classmethod
330 330 def initialized(cls):
331 331 return hasattr(cls, "_instance")
332 332
333 333 def get_ipython(self):
334 334 """Return the currently running IPython instance."""
335 335 return self
336 336
337 337 #-------------------------------------------------------------------------
338 338 # Trait changed handlers
339 339 #-------------------------------------------------------------------------
340 340
341 341 def _ipython_dir_changed(self, name, new):
342 342 if not os.path.isdir(new):
343 343 os.makedirs(new, mode = 0777)
344 344
345 345 def set_autoindent(self,value=None):
346 346 """Set the autoindent flag, checking for readline support.
347 347
348 348 If called with no arguments, it acts as a toggle."""
349 349
350 350 if not self.has_readline:
351 351 if os.name == 'posix':
352 352 warn("The auto-indent feature requires the readline library")
353 353 self.autoindent = 0
354 354 return
355 355 if value is None:
356 356 self.autoindent = not self.autoindent
357 357 else:
358 358 self.autoindent = value
359 359
360 360 #-------------------------------------------------------------------------
361 361 # init_* methods called by __init__
362 362 #-------------------------------------------------------------------------
363 363
364 364 def init_ipython_dir(self, ipython_dir):
365 365 if ipython_dir is not None:
366 366 self.ipython_dir = ipython_dir
367 367 self.config.Global.ipython_dir = self.ipython_dir
368 368 return
369 369
370 370 if hasattr(self.config.Global, 'ipython_dir'):
371 371 self.ipython_dir = self.config.Global.ipython_dir
372 372 else:
373 373 self.ipython_dir = get_ipython_dir()
374 374
375 375 # All children can just read this
376 376 self.config.Global.ipython_dir = self.ipython_dir
377 377
378 378 def init_instance_attrs(self):
379 379 self.more = False
380 380
381 381 # command compiler
382 382 self.compile = CachingCompiler()
383 383
384 384 # User input buffers
385 385 # NOTE: these variables are slated for full removal, once we are 100%
386 386 # sure that the new execution logic is solid. We will delte runlines,
387 387 # push_line and these buffers, as all input will be managed by the
388 388 # frontends via an inputsplitter instance.
389 389 self.buffer = []
390 390 self.buffer_raw = []
391 391
392 392 # Make an empty namespace, which extension writers can rely on both
393 393 # existing and NEVER being used by ipython itself. This gives them a
394 394 # convenient location for storing additional information and state
395 395 # their extensions may require, without fear of collisions with other
396 396 # ipython names that may develop later.
397 397 self.meta = Struct()
398 398
399 399 # Object variable to store code object waiting execution. This is
400 400 # used mainly by the multithreaded shells, but it can come in handy in
401 401 # other situations. No need to use a Queue here, since it's a single
402 402 # item which gets cleared once run.
403 403 self.code_to_run = None
404 404
405 405 # Temporary files used for various purposes. Deleted at exit.
406 406 self.tempfiles = []
407 407
408 408 # Keep track of readline usage (later set by init_readline)
409 409 self.has_readline = False
410 410
411 411 # keep track of where we started running (mainly for crash post-mortem)
412 412 # This is not being used anywhere currently.
413 413 self.starting_dir = os.getcwd()
414 414
415 415 # Indentation management
416 416 self.indent_current_nsp = 0
417 417
418 418 def init_environment(self):
419 419 """Any changes we need to make to the user's environment."""
420 420 pass
421 421
422 422 def init_encoding(self):
423 423 # Get system encoding at startup time. Certain terminals (like Emacs
424 424 # under Win32 have it set to None, and we need to have a known valid
425 425 # encoding to use in the raw_input() method
426 426 try:
427 427 self.stdin_encoding = sys.stdin.encoding or 'ascii'
428 428 except AttributeError:
429 429 self.stdin_encoding = 'ascii'
430 430
431 431 def init_syntax_highlighting(self):
432 432 # Python source parser/formatter for syntax highlighting
433 433 pyformat = PyColorize.Parser().format
434 434 self.pycolorize = lambda src: pyformat(src,'str',self.colors)
435 435
436 436 def init_pushd_popd_magic(self):
437 437 # for pushd/popd management
438 438 try:
439 439 self.home_dir = get_home_dir()
440 440 except HomeDirError, msg:
441 441 fatal(msg)
442 442
443 443 self.dir_stack = []
444 444
445 445 def init_logger(self):
446 446 self.logger = Logger(self.home_dir, logfname='ipython_log.py',
447 447 logmode='rotate')
448 448
449 449 def init_logstart(self):
450 450 """Initialize logging in case it was requested at the command line.
451 451 """
452 452 if self.logappend:
453 453 self.magic_logstart(self.logappend + ' append')
454 454 elif self.logfile:
455 455 self.magic_logstart(self.logfile)
456 456 elif self.logstart:
457 457 self.magic_logstart()
458 458
459 459 def init_builtins(self):
460 460 self.builtin_trap = BuiltinTrap(shell=self)
461 461
462 462 def init_inspector(self):
463 463 # Object inspector
464 464 self.inspector = oinspect.Inspector(oinspect.InspectColors,
465 465 PyColorize.ANSICodeColors,
466 466 'NoColor',
467 467 self.object_info_string_level)
468 468
469 469 def init_io(self):
470 470 # This will just use sys.stdout and sys.stderr. If you want to
471 471 # override sys.stdout and sys.stderr themselves, you need to do that
472 472 # *before* instantiating this class, because Term holds onto
473 473 # references to the underlying streams.
474 474 if sys.platform == 'win32' and self.has_readline:
475 475 Term = io.IOTerm(cout=self.readline._outputfile,
476 476 cerr=self.readline._outputfile)
477 477 else:
478 478 Term = io.IOTerm()
479 479 io.Term = Term
480 480
481 481 def init_prompts(self):
482 482 # TODO: This is a pass for now because the prompts are managed inside
483 483 # the DisplayHook. Once there is a separate prompt manager, this
484 484 # will initialize that object and all prompt related information.
485 485 pass
486 486
487 487 def init_display_formatter(self):
488 488 self.display_formatter = DisplayFormatter(config=self.config)
489 489
490 490 def init_display_pub(self):
491 491 self.display_pub = self.display_pub_class(config=self.config)
492 492
493 493 def init_displayhook(self):
494 494 # Initialize displayhook, set in/out prompts and printing system
495 495 self.displayhook = self.displayhook_class(
496 496 config=self.config,
497 497 shell=self,
498 498 cache_size=self.cache_size,
499 499 input_sep = self.separate_in,
500 500 output_sep = self.separate_out,
501 501 output_sep2 = self.separate_out2,
502 502 ps1 = self.prompt_in1,
503 503 ps2 = self.prompt_in2,
504 504 ps_out = self.prompt_out,
505 505 pad_left = self.prompts_pad_left
506 506 )
507 507 # This is a context manager that installs/revmoes the displayhook at
508 508 # the appropriate time.
509 509 self.display_trap = DisplayTrap(hook=self.displayhook)
510 510
511 511 def init_reload_doctest(self):
512 512 # Do a proper resetting of doctest, including the necessary displayhook
513 513 # monkeypatching
514 514 try:
515 515 doctest_reload()
516 516 except ImportError:
517 517 warn("doctest module does not exist.")
518 518
519 519 #-------------------------------------------------------------------------
520 520 # Things related to injections into the sys module
521 521 #-------------------------------------------------------------------------
522 522
523 523 def save_sys_module_state(self):
524 524 """Save the state of hooks in the sys module.
525 525
526 526 This has to be called after self.user_ns is created.
527 527 """
528 528 self._orig_sys_module_state = {}
529 529 self._orig_sys_module_state['stdin'] = sys.stdin
530 530 self._orig_sys_module_state['stdout'] = sys.stdout
531 531 self._orig_sys_module_state['stderr'] = sys.stderr
532 532 self._orig_sys_module_state['excepthook'] = sys.excepthook
533 533 try:
534 534 self._orig_sys_modules_main_name = self.user_ns['__name__']
535 535 except KeyError:
536 536 pass
537 537
538 538 def restore_sys_module_state(self):
539 539 """Restore the state of the sys module."""
540 540 try:
541 541 for k, v in self._orig_sys_module_state.iteritems():
542 542 setattr(sys, k, v)
543 543 except AttributeError:
544 544 pass
545 545 # Reset what what done in self.init_sys_modules
546 546 try:
547 547 sys.modules[self.user_ns['__name__']] = self._orig_sys_modules_main_name
548 548 except (AttributeError, KeyError):
549 549 pass
550 550
551 551 #-------------------------------------------------------------------------
552 552 # Things related to hooks
553 553 #-------------------------------------------------------------------------
554 554
555 555 def init_hooks(self):
556 556 # hooks holds pointers used for user-side customizations
557 557 self.hooks = Struct()
558 558
559 559 self.strdispatchers = {}
560 560
561 561 # Set all default hooks, defined in the IPython.hooks module.
562 562 hooks = IPython.core.hooks
563 563 for hook_name in hooks.__all__:
564 564 # default hooks have priority 100, i.e. low; user hooks should have
565 565 # 0-100 priority
566 566 self.set_hook(hook_name,getattr(hooks,hook_name), 100)
567 567
568 568 def set_hook(self,name,hook, priority = 50, str_key = None, re_key = None):
569 569 """set_hook(name,hook) -> sets an internal IPython hook.
570 570
571 571 IPython exposes some of its internal API as user-modifiable hooks. By
572 572 adding your function to one of these hooks, you can modify IPython's
573 573 behavior to call at runtime your own routines."""
574 574
575 575 # At some point in the future, this should validate the hook before it
576 576 # accepts it. Probably at least check that the hook takes the number
577 577 # of args it's supposed to.
578 578
579 579 f = types.MethodType(hook,self)
580 580
581 581 # check if the hook is for strdispatcher first
582 582 if str_key is not None:
583 583 sdp = self.strdispatchers.get(name, StrDispatch())
584 584 sdp.add_s(str_key, f, priority )
585 585 self.strdispatchers[name] = sdp
586 586 return
587 587 if re_key is not None:
588 588 sdp = self.strdispatchers.get(name, StrDispatch())
589 589 sdp.add_re(re.compile(re_key), f, priority )
590 590 self.strdispatchers[name] = sdp
591 591 return
592 592
593 593 dp = getattr(self.hooks, name, None)
594 594 if name not in IPython.core.hooks.__all__:
595 595 print "Warning! Hook '%s' is not one of %s" % \
596 596 (name, IPython.core.hooks.__all__ )
597 597 if not dp:
598 598 dp = IPython.core.hooks.CommandChainDispatcher()
599 599
600 600 try:
601 601 dp.add(f,priority)
602 602 except AttributeError:
603 603 # it was not commandchain, plain old func - replace
604 604 dp = f
605 605
606 606 setattr(self.hooks,name, dp)
607 607
608 608 def register_post_execute(self, func):
609 609 """Register a function for calling after code execution.
610 610 """
611 611 if not callable(func):
612 612 raise ValueError('argument %s must be callable' % func)
613 613 self._post_execute.add(func)
614 614
615 615 #-------------------------------------------------------------------------
616 616 # Things related to the "main" module
617 617 #-------------------------------------------------------------------------
618 618
619 619 def new_main_mod(self,ns=None):
620 620 """Return a new 'main' module object for user code execution.
621 621 """
622 622 main_mod = self._user_main_module
623 623 init_fakemod_dict(main_mod,ns)
624 624 return main_mod
625 625
626 626 def cache_main_mod(self,ns,fname):
627 627 """Cache a main module's namespace.
628 628
629 629 When scripts are executed via %run, we must keep a reference to the
630 630 namespace of their __main__ module (a FakeModule instance) around so
631 631 that Python doesn't clear it, rendering objects defined therein
632 632 useless.
633 633
634 634 This method keeps said reference in a private dict, keyed by the
635 635 absolute path of the module object (which corresponds to the script
636 636 path). This way, for multiple executions of the same script we only
637 637 keep one copy of the namespace (the last one), thus preventing memory
638 638 leaks from old references while allowing the objects from the last
639 639 execution to be accessible.
640 640
641 641 Note: we can not allow the actual FakeModule instances to be deleted,
642 642 because of how Python tears down modules (it hard-sets all their
643 643 references to None without regard for reference counts). This method
644 644 must therefore make a *copy* of the given namespace, to allow the
645 645 original module's __dict__ to be cleared and reused.
646 646
647 647
648 648 Parameters
649 649 ----------
650 650 ns : a namespace (a dict, typically)
651 651
652 652 fname : str
653 653 Filename associated with the namespace.
654 654
655 655 Examples
656 656 --------
657 657
658 658 In [10]: import IPython
659 659
660 660 In [11]: _ip.cache_main_mod(IPython.__dict__,IPython.__file__)
661 661
662 662 In [12]: IPython.__file__ in _ip._main_ns_cache
663 663 Out[12]: True
664 664 """
665 665 self._main_ns_cache[os.path.abspath(fname)] = ns.copy()
666 666
667 667 def clear_main_mod_cache(self):
668 668 """Clear the cache of main modules.
669 669
670 670 Mainly for use by utilities like %reset.
671 671
672 672 Examples
673 673 --------
674 674
675 675 In [15]: import IPython
676 676
677 677 In [16]: _ip.cache_main_mod(IPython.__dict__,IPython.__file__)
678 678
679 679 In [17]: len(_ip._main_ns_cache) > 0
680 680 Out[17]: True
681 681
682 682 In [18]: _ip.clear_main_mod_cache()
683 683
684 684 In [19]: len(_ip._main_ns_cache) == 0
685 685 Out[19]: True
686 686 """
687 687 self._main_ns_cache.clear()
688 688
689 689 #-------------------------------------------------------------------------
690 690 # Things related to debugging
691 691 #-------------------------------------------------------------------------
692 692
693 693 def init_pdb(self):
694 694 # Set calling of pdb on exceptions
695 695 # self.call_pdb is a property
696 696 self.call_pdb = self.pdb
697 697
698 698 def _get_call_pdb(self):
699 699 return self._call_pdb
700 700
701 701 def _set_call_pdb(self,val):
702 702
703 703 if val not in (0,1,False,True):
704 704 raise ValueError,'new call_pdb value must be boolean'
705 705
706 706 # store value in instance
707 707 self._call_pdb = val
708 708
709 709 # notify the actual exception handlers
710 710 self.InteractiveTB.call_pdb = val
711 711
712 712 call_pdb = property(_get_call_pdb,_set_call_pdb,None,
713 713 'Control auto-activation of pdb at exceptions')
714 714
715 715 def debugger(self,force=False):
716 716 """Call the pydb/pdb debugger.
717 717
718 718 Keywords:
719 719
720 720 - force(False): by default, this routine checks the instance call_pdb
721 721 flag and does not actually invoke the debugger if the flag is false.
722 722 The 'force' option forces the debugger to activate even if the flag
723 723 is false.
724 724 """
725 725
726 726 if not (force or self.call_pdb):
727 727 return
728 728
729 729 if not hasattr(sys,'last_traceback'):
730 730 error('No traceback has been produced, nothing to debug.')
731 731 return
732 732
733 733 # use pydb if available
734 734 if debugger.has_pydb:
735 735 from pydb import pm
736 736 else:
737 737 # fallback to our internal debugger
738 738 pm = lambda : self.InteractiveTB.debugger(force=True)
739 739 self.history_saving_wrapper(pm)()
740 740
741 741 #-------------------------------------------------------------------------
742 742 # Things related to IPython's various namespaces
743 743 #-------------------------------------------------------------------------
744 744
745 745 def init_create_namespaces(self, user_ns=None, user_global_ns=None):
746 746 # Create the namespace where the user will operate. user_ns is
747 747 # normally the only one used, and it is passed to the exec calls as
748 748 # the locals argument. But we do carry a user_global_ns namespace
749 749 # given as the exec 'globals' argument, This is useful in embedding
750 750 # situations where the ipython shell opens in a context where the
751 751 # distinction between locals and globals is meaningful. For
752 752 # non-embedded contexts, it is just the same object as the user_ns dict.
753 753
754 754 # FIXME. For some strange reason, __builtins__ is showing up at user
755 755 # level as a dict instead of a module. This is a manual fix, but I
756 756 # should really track down where the problem is coming from. Alex
757 757 # Schmolck reported this problem first.
758 758
759 759 # A useful post by Alex Martelli on this topic:
760 760 # Re: inconsistent value from __builtins__
761 761 # Von: Alex Martelli <aleaxit@yahoo.com>
762 762 # Datum: Freitag 01 Oktober 2004 04:45:34 nachmittags/abends
763 763 # Gruppen: comp.lang.python
764 764
765 765 # Michael Hohn <hohn@hooknose.lbl.gov> wrote:
766 766 # > >>> print type(builtin_check.get_global_binding('__builtins__'))
767 767 # > <type 'dict'>
768 768 # > >>> print type(__builtins__)
769 769 # > <type 'module'>
770 770 # > Is this difference in return value intentional?
771 771
772 772 # Well, it's documented that '__builtins__' can be either a dictionary
773 773 # or a module, and it's been that way for a long time. Whether it's
774 774 # intentional (or sensible), I don't know. In any case, the idea is
775 775 # that if you need to access the built-in namespace directly, you
776 776 # should start with "import __builtin__" (note, no 's') which will
777 777 # definitely give you a module. Yeah, it's somewhat confusing:-(.
778 778
779 779 # These routines return properly built dicts as needed by the rest of
780 780 # the code, and can also be used by extension writers to generate
781 781 # properly initialized namespaces.
782 782 user_ns, user_global_ns = self.make_user_namespaces(user_ns,
783 783 user_global_ns)
784 784
785 785 # Assign namespaces
786 786 # This is the namespace where all normal user variables live
787 787 self.user_ns = user_ns
788 788 self.user_global_ns = user_global_ns
789 789
790 790 # An auxiliary namespace that checks what parts of the user_ns were
791 791 # loaded at startup, so we can list later only variables defined in
792 792 # actual interactive use. Since it is always a subset of user_ns, it
793 793 # doesn't need to be separately tracked in the ns_table.
794 794 self.user_ns_hidden = {}
795 795
796 796 # A namespace to keep track of internal data structures to prevent
797 797 # them from cluttering user-visible stuff. Will be updated later
798 798 self.internal_ns = {}
799 799
800 800 # Now that FakeModule produces a real module, we've run into a nasty
801 801 # problem: after script execution (via %run), the module where the user
802 802 # code ran is deleted. Now that this object is a true module (needed
803 803 # so docetst and other tools work correctly), the Python module
804 804 # teardown mechanism runs over it, and sets to None every variable
805 805 # present in that module. Top-level references to objects from the
806 806 # script survive, because the user_ns is updated with them. However,
807 807 # calling functions defined in the script that use other things from
808 808 # the script will fail, because the function's closure had references
809 809 # to the original objects, which are now all None. So we must protect
810 810 # these modules from deletion by keeping a cache.
811 811 #
812 812 # To avoid keeping stale modules around (we only need the one from the
813 813 # last run), we use a dict keyed with the full path to the script, so
814 814 # only the last version of the module is held in the cache. Note,
815 815 # however, that we must cache the module *namespace contents* (their
816 816 # __dict__). Because if we try to cache the actual modules, old ones
817 817 # (uncached) could be destroyed while still holding references (such as
818 818 # those held by GUI objects that tend to be long-lived)>
819 819 #
820 820 # The %reset command will flush this cache. See the cache_main_mod()
821 821 # and clear_main_mod_cache() methods for details on use.
822 822
823 823 # This is the cache used for 'main' namespaces
824 824 self._main_ns_cache = {}
825 825 # And this is the single instance of FakeModule whose __dict__ we keep
826 826 # copying and clearing for reuse on each %run
827 827 self._user_main_module = FakeModule()
828 828
829 829 # A table holding all the namespaces IPython deals with, so that
830 830 # introspection facilities can search easily.
831 831 self.ns_table = {'user':user_ns,
832 832 'user_global':user_global_ns,
833 833 'internal':self.internal_ns,
834 834 'builtin':__builtin__.__dict__
835 835 }
836 836
837 837 # Similarly, track all namespaces where references can be held and that
838 838 # we can safely clear (so it can NOT include builtin). This one can be
839 839 # a simple list. Note that the main execution namespaces, user_ns and
840 840 # user_global_ns, can NOT be listed here, as clearing them blindly
841 841 # causes errors in object __del__ methods. Instead, the reset() method
842 842 # clears them manually and carefully.
843 843 self.ns_refs_table = [ self.user_ns_hidden,
844 844 self.internal_ns, self._main_ns_cache ]
845 845
846 846 def make_user_namespaces(self, user_ns=None, user_global_ns=None):
847 847 """Return a valid local and global user interactive namespaces.
848 848
849 849 This builds a dict with the minimal information needed to operate as a
850 850 valid IPython user namespace, which you can pass to the various
851 851 embedding classes in ipython. The default implementation returns the
852 852 same dict for both the locals and the globals to allow functions to
853 853 refer to variables in the namespace. Customized implementations can
854 854 return different dicts. The locals dictionary can actually be anything
855 855 following the basic mapping protocol of a dict, but the globals dict
856 856 must be a true dict, not even a subclass. It is recommended that any
857 857 custom object for the locals namespace synchronize with the globals
858 858 dict somehow.
859 859
860 860 Raises TypeError if the provided globals namespace is not a true dict.
861 861
862 862 Parameters
863 863 ----------
864 864 user_ns : dict-like, optional
865 865 The current user namespace. The items in this namespace should
866 866 be included in the output. If None, an appropriate blank
867 867 namespace should be created.
868 868 user_global_ns : dict, optional
869 869 The current user global namespace. The items in this namespace
870 870 should be included in the output. If None, an appropriate
871 871 blank namespace should be created.
872 872
873 873 Returns
874 874 -------
875 875 A pair of dictionary-like object to be used as the local namespace
876 876 of the interpreter and a dict to be used as the global namespace.
877 877 """
878 878
879 879
880 880 # We must ensure that __builtin__ (without the final 's') is always
881 881 # available and pointing to the __builtin__ *module*. For more details:
882 882 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
883 883
884 884 if user_ns is None:
885 885 # Set __name__ to __main__ to better match the behavior of the
886 886 # normal interpreter.
887 887 user_ns = {'__name__' :'__main__',
888 888 '__builtin__' : __builtin__,
889 889 '__builtins__' : __builtin__,
890 890 }
891 891 else:
892 892 user_ns.setdefault('__name__','__main__')
893 893 user_ns.setdefault('__builtin__',__builtin__)
894 894 user_ns.setdefault('__builtins__',__builtin__)
895 895
896 896 if user_global_ns is None:
897 897 user_global_ns = user_ns
898 898 if type(user_global_ns) is not dict:
899 899 raise TypeError("user_global_ns must be a true dict; got %r"
900 900 % type(user_global_ns))
901 901
902 902 return user_ns, user_global_ns
903 903
904 904 def init_sys_modules(self):
905 905 # We need to insert into sys.modules something that looks like a
906 906 # module but which accesses the IPython namespace, for shelve and
907 907 # pickle to work interactively. Normally they rely on getting
908 908 # everything out of __main__, but for embedding purposes each IPython
909 909 # instance has its own private namespace, so we can't go shoving
910 910 # everything into __main__.
911 911
912 912 # note, however, that we should only do this for non-embedded
913 913 # ipythons, which really mimic the __main__.__dict__ with their own
914 914 # namespace. Embedded instances, on the other hand, should not do
915 915 # this because they need to manage the user local/global namespaces
916 916 # only, but they live within a 'normal' __main__ (meaning, they
917 917 # shouldn't overtake the execution environment of the script they're
918 918 # embedded in).
919 919
920 920 # This is overridden in the InteractiveShellEmbed subclass to a no-op.
921 921
922 922 try:
923 923 main_name = self.user_ns['__name__']
924 924 except KeyError:
925 925 raise KeyError('user_ns dictionary MUST have a "__name__" key')
926 926 else:
927 927 sys.modules[main_name] = FakeModule(self.user_ns)
928 928
929 929 def init_user_ns(self):
930 930 """Initialize all user-visible namespaces to their minimum defaults.
931 931
932 932 Certain history lists are also initialized here, as they effectively
933 933 act as user namespaces.
934 934
935 935 Notes
936 936 -----
937 937 All data structures here are only filled in, they are NOT reset by this
938 938 method. If they were not empty before, data will simply be added to
939 939 therm.
940 940 """
941 941 # This function works in two parts: first we put a few things in
942 942 # user_ns, and we sync that contents into user_ns_hidden so that these
943 943 # initial variables aren't shown by %who. After the sync, we add the
944 944 # rest of what we *do* want the user to see with %who even on a new
945 945 # session (probably nothing, so theye really only see their own stuff)
946 946
947 947 # The user dict must *always* have a __builtin__ reference to the
948 948 # Python standard __builtin__ namespace, which must be imported.
949 949 # This is so that certain operations in prompt evaluation can be
950 950 # reliably executed with builtins. Note that we can NOT use
951 951 # __builtins__ (note the 's'), because that can either be a dict or a
952 952 # module, and can even mutate at runtime, depending on the context
953 953 # (Python makes no guarantees on it). In contrast, __builtin__ is
954 954 # always a module object, though it must be explicitly imported.
955 955
956 956 # For more details:
957 957 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
958 958 ns = dict(__builtin__ = __builtin__)
959 959
960 960 # Put 'help' in the user namespace
961 961 try:
962 962 from site import _Helper
963 963 ns['help'] = _Helper()
964 964 except ImportError:
965 965 warn('help() not available - check site.py')
966 966
967 967 # make global variables for user access to the histories
968 968 ns['_ih'] = self.history_manager.input_hist_parsed
969 969 ns['_oh'] = self.history_manager.output_hist
970 970 ns['_dh'] = self.history_manager.dir_hist
971 971
972 972 ns['_sh'] = shadowns
973 973
974 974 # user aliases to input and output histories. These shouldn't show up
975 975 # in %who, as they can have very large reprs.
976 976 ns['In'] = self.history_manager.input_hist_parsed
977 977 ns['Out'] = self.history_manager.output_hist
978 978
979 979 # Store myself as the public api!!!
980 980 ns['get_ipython'] = self.get_ipython
981 981
982 982 # Sync what we've added so far to user_ns_hidden so these aren't seen
983 983 # by %who
984 984 self.user_ns_hidden.update(ns)
985 985
986 986 # Anything put into ns now would show up in %who. Think twice before
987 987 # putting anything here, as we really want %who to show the user their
988 988 # stuff, not our variables.
989 989
990 990 # Finally, update the real user's namespace
991 991 self.user_ns.update(ns)
992 992
993 993 def reset(self, new_session=True):
994 994 """Clear all internal namespaces.
995 995
996 996 Note that this is much more aggressive than %reset, since it clears
997 997 fully all namespaces, as well as all input/output lists.
998 998
999 999 If new_session is True, a new history session will be opened.
1000 1000 """
1001 1001 # Clear histories
1002 1002 self.history_manager.reset(new_session)
1003 1003
1004 1004 # Reset counter used to index all histories
1005 1005 self.execution_count = 0
1006 1006
1007 1007 # Restore the user namespaces to minimal usability
1008 1008 for ns in self.ns_refs_table:
1009 1009 ns.clear()
1010 1010
1011 1011 # The main execution namespaces must be cleared very carefully,
1012 1012 # skipping the deletion of the builtin-related keys, because doing so
1013 1013 # would cause errors in many object's __del__ methods.
1014 1014 for ns in [self.user_ns, self.user_global_ns]:
1015 1015 drop_keys = set(ns.keys())
1016 1016 drop_keys.discard('__builtin__')
1017 1017 drop_keys.discard('__builtins__')
1018 1018 for k in drop_keys:
1019 1019 del ns[k]
1020 1020
1021 1021 # Restore the user namespaces to minimal usability
1022 1022 self.init_user_ns()
1023 1023
1024 1024 # Restore the default and user aliases
1025 1025 self.alias_manager.clear_aliases()
1026 1026 self.alias_manager.init_aliases()
1027 1027
1028 1028 def reset_selective(self, regex=None):
1029 1029 """Clear selective variables from internal namespaces based on a
1030 1030 specified regular expression.
1031 1031
1032 1032 Parameters
1033 1033 ----------
1034 1034 regex : string or compiled pattern, optional
1035 1035 A regular expression pattern that will be used in searching
1036 1036 variable names in the users namespaces.
1037 1037 """
1038 1038 if regex is not None:
1039 1039 try:
1040 1040 m = re.compile(regex)
1041 1041 except TypeError:
1042 1042 raise TypeError('regex must be a string or compiled pattern')
1043 1043 # Search for keys in each namespace that match the given regex
1044 1044 # If a match is found, delete the key/value pair.
1045 1045 for ns in self.ns_refs_table:
1046 1046 for var in ns:
1047 1047 if m.search(var):
1048 1048 del ns[var]
1049 1049
1050 1050 def push(self, variables, interactive=True):
1051 1051 """Inject a group of variables into the IPython user namespace.
1052 1052
1053 1053 Parameters
1054 1054 ----------
1055 1055 variables : dict, str or list/tuple of str
1056 1056 The variables to inject into the user's namespace. If a dict, a
1057 1057 simple update is done. If a str, the string is assumed to have
1058 1058 variable names separated by spaces. A list/tuple of str can also
1059 1059 be used to give the variable names. If just the variable names are
1060 1060 give (list/tuple/str) then the variable values looked up in the
1061 1061 callers frame.
1062 1062 interactive : bool
1063 1063 If True (default), the variables will be listed with the ``who``
1064 1064 magic.
1065 1065 """
1066 1066 vdict = None
1067 1067
1068 1068 # We need a dict of name/value pairs to do namespace updates.
1069 1069 if isinstance(variables, dict):
1070 1070 vdict = variables
1071 1071 elif isinstance(variables, (basestring, list, tuple)):
1072 1072 if isinstance(variables, basestring):
1073 1073 vlist = variables.split()
1074 1074 else:
1075 1075 vlist = variables
1076 1076 vdict = {}
1077 1077 cf = sys._getframe(1)
1078 1078 for name in vlist:
1079 1079 try:
1080 1080 vdict[name] = eval(name, cf.f_globals, cf.f_locals)
1081 1081 except:
1082 1082 print ('Could not get variable %s from %s' %
1083 1083 (name,cf.f_code.co_name))
1084 1084 else:
1085 1085 raise ValueError('variables must be a dict/str/list/tuple')
1086 1086
1087 1087 # Propagate variables to user namespace
1088 1088 self.user_ns.update(vdict)
1089 1089
1090 1090 # And configure interactive visibility
1091 1091 config_ns = self.user_ns_hidden
1092 1092 if interactive:
1093 1093 for name, val in vdict.iteritems():
1094 1094 config_ns.pop(name, None)
1095 1095 else:
1096 1096 for name,val in vdict.iteritems():
1097 1097 config_ns[name] = val
1098 1098
1099 1099 #-------------------------------------------------------------------------
1100 1100 # Things related to object introspection
1101 1101 #-------------------------------------------------------------------------
1102 1102
1103 1103 def _ofind(self, oname, namespaces=None):
1104 1104 """Find an object in the available namespaces.
1105 1105
1106 1106 self._ofind(oname) -> dict with keys: found,obj,ospace,ismagic
1107 1107
1108 1108 Has special code to detect magic functions.
1109 1109 """
1110 1110 #oname = oname.strip()
1111 1111 #print '1- oname: <%r>' % oname # dbg
1112 1112 try:
1113 1113 oname = oname.strip().encode('ascii')
1114 1114 #print '2- oname: <%r>' % oname # dbg
1115 1115 except UnicodeEncodeError:
1116 1116 print 'Python identifiers can only contain ascii characters.'
1117 1117 return dict(found=False)
1118 1118
1119 1119 alias_ns = None
1120 1120 if namespaces is None:
1121 1121 # Namespaces to search in:
1122 1122 # Put them in a list. The order is important so that we
1123 1123 # find things in the same order that Python finds them.
1124 1124 namespaces = [ ('Interactive', self.user_ns),
1125 1125 ('IPython internal', self.internal_ns),
1126 1126 ('Python builtin', __builtin__.__dict__),
1127 1127 ('Alias', self.alias_manager.alias_table),
1128 1128 ]
1129 1129 alias_ns = self.alias_manager.alias_table
1130 1130
1131 1131 # initialize results to 'null'
1132 1132 found = False; obj = None; ospace = None; ds = None;
1133 1133 ismagic = False; isalias = False; parent = None
1134 1134
1135 1135 # We need to special-case 'print', which as of python2.6 registers as a
1136 1136 # function but should only be treated as one if print_function was
1137 1137 # loaded with a future import. In this case, just bail.
1138 1138 if (oname == 'print' and not (self.compile.compiler_flags &
1139 1139 __future__.CO_FUTURE_PRINT_FUNCTION)):
1140 1140 return {'found':found, 'obj':obj, 'namespace':ospace,
1141 1141 'ismagic':ismagic, 'isalias':isalias, 'parent':parent}
1142 1142
1143 1143 # Look for the given name by splitting it in parts. If the head is
1144 1144 # found, then we look for all the remaining parts as members, and only
1145 1145 # declare success if we can find them all.
1146 1146 oname_parts = oname.split('.')
1147 1147 oname_head, oname_rest = oname_parts[0],oname_parts[1:]
1148 1148 for nsname,ns in namespaces:
1149 1149 try:
1150 1150 obj = ns[oname_head]
1151 1151 except KeyError:
1152 1152 continue
1153 1153 else:
1154 1154 #print 'oname_rest:', oname_rest # dbg
1155 1155 for part in oname_rest:
1156 1156 try:
1157 1157 parent = obj
1158 1158 obj = getattr(obj,part)
1159 1159 except:
1160 1160 # Blanket except b/c some badly implemented objects
1161 1161 # allow __getattr__ to raise exceptions other than
1162 1162 # AttributeError, which then crashes IPython.
1163 1163 break
1164 1164 else:
1165 1165 # If we finish the for loop (no break), we got all members
1166 1166 found = True
1167 1167 ospace = nsname
1168 1168 if ns == alias_ns:
1169 1169 isalias = True
1170 1170 break # namespace loop
1171 1171
1172 1172 # Try to see if it's magic
1173 1173 if not found:
1174 1174 if oname.startswith(ESC_MAGIC):
1175 1175 oname = oname[1:]
1176 1176 obj = getattr(self,'magic_'+oname,None)
1177 1177 if obj is not None:
1178 1178 found = True
1179 1179 ospace = 'IPython internal'
1180 1180 ismagic = True
1181 1181
1182 1182 # Last try: special-case some literals like '', [], {}, etc:
1183 1183 if not found and oname_head in ["''",'""','[]','{}','()']:
1184 1184 obj = eval(oname_head)
1185 1185 found = True
1186 1186 ospace = 'Interactive'
1187 1187
1188 1188 return {'found':found, 'obj':obj, 'namespace':ospace,
1189 1189 'ismagic':ismagic, 'isalias':isalias, 'parent':parent}
1190 1190
1191 1191 def _ofind_property(self, oname, info):
1192 1192 """Second part of object finding, to look for property details."""
1193 1193 if info.found:
1194 1194 # Get the docstring of the class property if it exists.
1195 1195 path = oname.split('.')
1196 1196 root = '.'.join(path[:-1])
1197 1197 if info.parent is not None:
1198 1198 try:
1199 1199 target = getattr(info.parent, '__class__')
1200 1200 # The object belongs to a class instance.
1201 1201 try:
1202 1202 target = getattr(target, path[-1])
1203 1203 # The class defines the object.
1204 1204 if isinstance(target, property):
1205 1205 oname = root + '.__class__.' + path[-1]
1206 1206 info = Struct(self._ofind(oname))
1207 1207 except AttributeError: pass
1208 1208 except AttributeError: pass
1209 1209
1210 1210 # We return either the new info or the unmodified input if the object
1211 1211 # hadn't been found
1212 1212 return info
1213 1213
1214 1214 def _object_find(self, oname, namespaces=None):
1215 1215 """Find an object and return a struct with info about it."""
1216 1216 inf = Struct(self._ofind(oname, namespaces))
1217 1217 return Struct(self._ofind_property(oname, inf))
1218 1218
1219 1219 def _inspect(self, meth, oname, namespaces=None, **kw):
1220 1220 """Generic interface to the inspector system.
1221 1221
1222 1222 This function is meant to be called by pdef, pdoc & friends."""
1223 1223 info = self._object_find(oname)
1224 1224 if info.found:
1225 1225 pmethod = getattr(self.inspector, meth)
1226 1226 formatter = format_screen if info.ismagic else None
1227 1227 if meth == 'pdoc':
1228 1228 pmethod(info.obj, oname, formatter)
1229 1229 elif meth == 'pinfo':
1230 1230 pmethod(info.obj, oname, formatter, info, **kw)
1231 1231 else:
1232 1232 pmethod(info.obj, oname)
1233 1233 else:
1234 1234 print 'Object `%s` not found.' % oname
1235 1235 return 'not found' # so callers can take other action
1236 1236
1237 1237 def object_inspect(self, oname):
1238 1238 info = self._object_find(oname)
1239 1239 if info.found:
1240 1240 return self.inspector.info(info.obj, oname, info=info)
1241 1241 else:
1242 1242 return oinspect.object_info(name=oname, found=False)
1243 1243
1244 1244 #-------------------------------------------------------------------------
1245 1245 # Things related to history management
1246 1246 #-------------------------------------------------------------------------
1247 1247
1248 1248 def init_history(self):
1249 1249 """Sets up the command history, and starts regular autosaves."""
1250 1250 self.history_manager = HistoryManager(shell=self, config=self.config)
1251 1251
1252 1252 def history_saving_wrapper(self, func):
1253 1253 """ Wrap func for readline history saving
1254 1254
1255 1255 Convert func into callable that saves & restores
1256 1256 history around the call """
1257 1257
1258 1258 if self.has_readline:
1259 1259 from IPython.utils import rlineimpl as readline
1260 1260 else:
1261 1261 return func
1262 1262
1263 1263 def wrapper():
1264 1264 self.save_history()
1265 1265 try:
1266 1266 func()
1267 1267 finally:
1268 1268 self.reload_history()
1269 1269 return wrapper
1270 1270
1271 1271
1272 1272 #-------------------------------------------------------------------------
1273 1273 # Things related to exception handling and tracebacks (not debugging)
1274 1274 #-------------------------------------------------------------------------
1275 1275
1276 1276 def init_traceback_handlers(self, custom_exceptions):
1277 1277 # Syntax error handler.
1278 1278 self.SyntaxTB = ultratb.SyntaxTB(color_scheme='NoColor')
1279 1279
1280 1280 # The interactive one is initialized with an offset, meaning we always
1281 1281 # want to remove the topmost item in the traceback, which is our own
1282 1282 # internal code. Valid modes: ['Plain','Context','Verbose']
1283 1283 self.InteractiveTB = ultratb.AutoFormattedTB(mode = 'Plain',
1284 1284 color_scheme='NoColor',
1285 1285 tb_offset = 1,
1286 1286 check_cache=self.compile.check_cache)
1287 1287
1288 1288 # The instance will store a pointer to the system-wide exception hook,
1289 1289 # so that runtime code (such as magics) can access it. This is because
1290 1290 # during the read-eval loop, it may get temporarily overwritten.
1291 1291 self.sys_excepthook = sys.excepthook
1292 1292
1293 1293 # and add any custom exception handlers the user may have specified
1294 1294 self.set_custom_exc(*custom_exceptions)
1295 1295
1296 1296 # Set the exception mode
1297 1297 self.InteractiveTB.set_mode(mode=self.xmode)
1298 1298
1299 1299 def set_custom_exc(self, exc_tuple, handler):
1300 1300 """set_custom_exc(exc_tuple,handler)
1301 1301
1302 1302 Set a custom exception handler, which will be called if any of the
1303 1303 exceptions in exc_tuple occur in the mainloop (specifically, in the
1304 1304 run_code() method.
1305 1305
1306 1306 Inputs:
1307 1307
1308 1308 - exc_tuple: a *tuple* of valid exceptions to call the defined
1309 1309 handler for. It is very important that you use a tuple, and NOT A
1310 1310 LIST here, because of the way Python's except statement works. If
1311 1311 you only want to trap a single exception, use a singleton tuple:
1312 1312
1313 1313 exc_tuple == (MyCustomException,)
1314 1314
1315 1315 - handler: this must be defined as a function with the following
1316 1316 basic interface::
1317 1317
1318 1318 def my_handler(self, etype, value, tb, tb_offset=None)
1319 1319 ...
1320 1320 # The return value must be
1321 1321 return structured_traceback
1322 1322
1323 1323 This will be made into an instance method (via types.MethodType)
1324 1324 of IPython itself, and it will be called if any of the exceptions
1325 1325 listed in the exc_tuple are caught. If the handler is None, an
1326 1326 internal basic one is used, which just prints basic info.
1327 1327
1328 1328 WARNING: by putting in your own exception handler into IPython's main
1329 1329 execution loop, you run a very good chance of nasty crashes. This
1330 1330 facility should only be used if you really know what you are doing."""
1331 1331
1332 1332 assert type(exc_tuple)==type(()) , \
1333 1333 "The custom exceptions must be given AS A TUPLE."
1334 1334
1335 1335 def dummy_handler(self,etype,value,tb):
1336 1336 print '*** Simple custom exception handler ***'
1337 1337 print 'Exception type :',etype
1338 1338 print 'Exception value:',value
1339 1339 print 'Traceback :',tb
1340 1340 print 'Source code :','\n'.join(self.buffer)
1341 1341
1342 1342 if handler is None: handler = dummy_handler
1343 1343
1344 1344 self.CustomTB = types.MethodType(handler,self)
1345 1345 self.custom_exceptions = exc_tuple
1346 1346
1347 1347 def excepthook(self, etype, value, tb):
1348 1348 """One more defense for GUI apps that call sys.excepthook.
1349 1349
1350 1350 GUI frameworks like wxPython trap exceptions and call
1351 1351 sys.excepthook themselves. I guess this is a feature that
1352 1352 enables them to keep running after exceptions that would
1353 1353 otherwise kill their mainloop. This is a bother for IPython
1354 1354 which excepts to catch all of the program exceptions with a try:
1355 1355 except: statement.
1356 1356
1357 1357 Normally, IPython sets sys.excepthook to a CrashHandler instance, so if
1358 1358 any app directly invokes sys.excepthook, it will look to the user like
1359 1359 IPython crashed. In order to work around this, we can disable the
1360 1360 CrashHandler and replace it with this excepthook instead, which prints a
1361 1361 regular traceback using our InteractiveTB. In this fashion, apps which
1362 1362 call sys.excepthook will generate a regular-looking exception from
1363 1363 IPython, and the CrashHandler will only be triggered by real IPython
1364 1364 crashes.
1365 1365
1366 1366 This hook should be used sparingly, only in places which are not likely
1367 1367 to be true IPython errors.
1368 1368 """
1369 1369 self.showtraceback((etype,value,tb),tb_offset=0)
1370 1370
1371 1371 def showtraceback(self,exc_tuple = None,filename=None,tb_offset=None,
1372 1372 exception_only=False):
1373 1373 """Display the exception that just occurred.
1374 1374
1375 1375 If nothing is known about the exception, this is the method which
1376 1376 should be used throughout the code for presenting user tracebacks,
1377 1377 rather than directly invoking the InteractiveTB object.
1378 1378
1379 1379 A specific showsyntaxerror() also exists, but this method can take
1380 1380 care of calling it if needed, so unless you are explicitly catching a
1381 1381 SyntaxError exception, don't try to analyze the stack manually and
1382 1382 simply call this method."""
1383 1383
1384 1384 try:
1385 1385 if exc_tuple is None:
1386 1386 etype, value, tb = sys.exc_info()
1387 1387 else:
1388 1388 etype, value, tb = exc_tuple
1389 1389
1390 1390 if etype is None:
1391 1391 if hasattr(sys, 'last_type'):
1392 1392 etype, value, tb = sys.last_type, sys.last_value, \
1393 1393 sys.last_traceback
1394 1394 else:
1395 1395 self.write_err('No traceback available to show.\n')
1396 1396 return
1397 1397
1398 1398 if etype is SyntaxError:
1399 1399 # Though this won't be called by syntax errors in the input
1400 1400 # line, there may be SyntaxError cases whith imported code.
1401 1401 self.showsyntaxerror(filename)
1402 1402 elif etype is UsageError:
1403 1403 print "UsageError:", value
1404 1404 else:
1405 1405 # WARNING: these variables are somewhat deprecated and not
1406 1406 # necessarily safe to use in a threaded environment, but tools
1407 1407 # like pdb depend on their existence, so let's set them. If we
1408 1408 # find problems in the field, we'll need to revisit their use.
1409 1409 sys.last_type = etype
1410 1410 sys.last_value = value
1411 1411 sys.last_traceback = tb
1412 1412
1413 1413 if etype in self.custom_exceptions:
1414 1414 # FIXME: Old custom traceback objects may just return a
1415 1415 # string, in that case we just put it into a list
1416 1416 stb = self.CustomTB(etype, value, tb, tb_offset)
1417 1417 if isinstance(ctb, basestring):
1418 1418 stb = [stb]
1419 1419 else:
1420 1420 if exception_only:
1421 1421 stb = ['An exception has occurred, use %tb to see '
1422 1422 'the full traceback.\n']
1423 1423 stb.extend(self.InteractiveTB.get_exception_only(etype,
1424 1424 value))
1425 1425 else:
1426 1426 stb = self.InteractiveTB.structured_traceback(etype,
1427 1427 value, tb, tb_offset=tb_offset)
1428 1428 # FIXME: the pdb calling should be done by us, not by
1429 1429 # the code computing the traceback.
1430 1430 if self.InteractiveTB.call_pdb:
1431 1431 # pdb mucks up readline, fix it back
1432 1432 self.set_readline_completer()
1433 1433
1434 1434 # Actually show the traceback
1435 1435 self._showtraceback(etype, value, stb)
1436 1436
1437 1437 except KeyboardInterrupt:
1438 1438 self.write_err("\nKeyboardInterrupt\n")
1439 1439
1440 1440 def _showtraceback(self, etype, evalue, stb):
1441 1441 """Actually show a traceback.
1442 1442
1443 1443 Subclasses may override this method to put the traceback on a different
1444 1444 place, like a side channel.
1445 1445 """
1446 1446 print >> io.Term.cout, self.InteractiveTB.stb2text(stb)
1447 1447
1448 1448 def showsyntaxerror(self, filename=None):
1449 1449 """Display the syntax error that just occurred.
1450 1450
1451 1451 This doesn't display a stack trace because there isn't one.
1452 1452
1453 1453 If a filename is given, it is stuffed in the exception instead
1454 1454 of what was there before (because Python's parser always uses
1455 1455 "<string>" when reading from a string).
1456 1456 """
1457 1457 etype, value, last_traceback = sys.exc_info()
1458 1458
1459 1459 # See note about these variables in showtraceback() above
1460 1460 sys.last_type = etype
1461 1461 sys.last_value = value
1462 1462 sys.last_traceback = last_traceback
1463 1463
1464 1464 if filename and etype is SyntaxError:
1465 1465 # Work hard to stuff the correct filename in the exception
1466 1466 try:
1467 1467 msg, (dummy_filename, lineno, offset, line) = value
1468 1468 except:
1469 1469 # Not the format we expect; leave it alone
1470 1470 pass
1471 1471 else:
1472 1472 # Stuff in the right filename
1473 1473 try:
1474 1474 # Assume SyntaxError is a class exception
1475 1475 value = SyntaxError(msg, (filename, lineno, offset, line))
1476 1476 except:
1477 1477 # If that failed, assume SyntaxError is a string
1478 1478 value = msg, (filename, lineno, offset, line)
1479 1479 stb = self.SyntaxTB.structured_traceback(etype, value, [])
1480 1480 self._showtraceback(etype, value, stb)
1481 1481
1482 1482 #-------------------------------------------------------------------------
1483 1483 # Things related to readline
1484 1484 #-------------------------------------------------------------------------
1485 1485
1486 1486 def init_readline(self):
1487 1487 """Command history completion/saving/reloading."""
1488 1488
1489 1489 if self.readline_use:
1490 1490 import IPython.utils.rlineimpl as readline
1491 1491
1492 1492 self.rl_next_input = None
1493 1493 self.rl_do_indent = False
1494 1494
1495 1495 if not self.readline_use or not readline.have_readline:
1496 1496 self.has_readline = False
1497 1497 self.readline = None
1498 1498 # Set a number of methods that depend on readline to be no-op
1499 1499 self.set_readline_completer = no_op
1500 1500 self.set_custom_completer = no_op
1501 1501 self.set_completer_frame = no_op
1502 1502 warn('Readline services not available or not loaded.')
1503 1503 else:
1504 1504 self.has_readline = True
1505 1505 self.readline = readline
1506 1506 sys.modules['readline'] = readline
1507 1507
1508 1508 # Platform-specific configuration
1509 1509 if os.name == 'nt':
1510 1510 # FIXME - check with Frederick to see if we can harmonize
1511 1511 # naming conventions with pyreadline to avoid this
1512 1512 # platform-dependent check
1513 1513 self.readline_startup_hook = readline.set_pre_input_hook
1514 1514 else:
1515 1515 self.readline_startup_hook = readline.set_startup_hook
1516 1516
1517 1517 # Load user's initrc file (readline config)
1518 1518 # Or if libedit is used, load editrc.
1519 1519 inputrc_name = os.environ.get('INPUTRC')
1520 1520 if inputrc_name is None:
1521 1521 home_dir = get_home_dir()
1522 1522 if home_dir is not None:
1523 1523 inputrc_name = '.inputrc'
1524 1524 if readline.uses_libedit:
1525 1525 inputrc_name = '.editrc'
1526 1526 inputrc_name = os.path.join(home_dir, inputrc_name)
1527 1527 if os.path.isfile(inputrc_name):
1528 1528 try:
1529 1529 readline.read_init_file(inputrc_name)
1530 1530 except:
1531 1531 warn('Problems reading readline initialization file <%s>'
1532 1532 % inputrc_name)
1533 1533
1534 1534 # Configure readline according to user's prefs
1535 1535 # This is only done if GNU readline is being used. If libedit
1536 1536 # is being used (as on Leopard) the readline config is
1537 1537 # not run as the syntax for libedit is different.
1538 1538 if not readline.uses_libedit:
1539 1539 for rlcommand in self.readline_parse_and_bind:
1540 1540 #print "loading rl:",rlcommand # dbg
1541 1541 readline.parse_and_bind(rlcommand)
1542 1542
1543 1543 # Remove some chars from the delimiters list. If we encounter
1544 1544 # unicode chars, discard them.
1545 1545 delims = readline.get_completer_delims().encode("ascii", "ignore")
1546 1546 delims = delims.translate(None, self.readline_remove_delims)
1547 1547 delims = delims.replace(ESC_MAGIC, '')
1548 1548 readline.set_completer_delims(delims)
1549 1549 # otherwise we end up with a monster history after a while:
1550 1550 readline.set_history_length(self.history_length)
1551 1551
1552 1552 # Load the last 1000 lines from history
1553 for _, _, cell in self.history_manager.get_hist_tail(1000,
1553 for _, _, cell in self.history_manager.get_tail(1000,
1554 1554 include_latest=True):
1555 1555 if cell.strip(): # Ignore blank lines
1556 1556 for line in cell.splitlines():
1557 1557 readline.add_history(line)
1558 1558
1559 1559 # Configure auto-indent for all platforms
1560 1560 self.set_autoindent(self.autoindent)
1561 1561
1562 1562 def set_next_input(self, s):
1563 1563 """ Sets the 'default' input string for the next command line.
1564 1564
1565 1565 Requires readline.
1566 1566
1567 1567 Example:
1568 1568
1569 1569 [D:\ipython]|1> _ip.set_next_input("Hello Word")
1570 1570 [D:\ipython]|2> Hello Word_ # cursor is here
1571 1571 """
1572 1572
1573 1573 self.rl_next_input = s
1574 1574
1575 1575 # Maybe move this to the terminal subclass?
1576 1576 def pre_readline(self):
1577 1577 """readline hook to be used at the start of each line.
1578 1578
1579 1579 Currently it handles auto-indent only."""
1580 1580
1581 1581 if self.rl_do_indent:
1582 1582 self.readline.insert_text(self._indent_current_str())
1583 1583 if self.rl_next_input is not None:
1584 1584 self.readline.insert_text(self.rl_next_input)
1585 1585 self.rl_next_input = None
1586 1586
1587 1587 def _indent_current_str(self):
1588 1588 """return the current level of indentation as a string"""
1589 1589 return self.input_splitter.indent_spaces * ' '
1590 1590
1591 1591 #-------------------------------------------------------------------------
1592 1592 # Things related to text completion
1593 1593 #-------------------------------------------------------------------------
1594 1594
1595 1595 def init_completer(self):
1596 1596 """Initialize the completion machinery.
1597 1597
1598 1598 This creates completion machinery that can be used by client code,
1599 1599 either interactively in-process (typically triggered by the readline
1600 1600 library), programatically (such as in test suites) or out-of-prcess
1601 1601 (typically over the network by remote frontends).
1602 1602 """
1603 1603 from IPython.core.completer import IPCompleter
1604 1604 from IPython.core.completerlib import (module_completer,
1605 1605 magic_run_completer, cd_completer)
1606 1606
1607 1607 self.Completer = IPCompleter(self,
1608 1608 self.user_ns,
1609 1609 self.user_global_ns,
1610 1610 self.readline_omit__names,
1611 1611 self.alias_manager.alias_table,
1612 1612 self.has_readline)
1613 1613
1614 1614 # Add custom completers to the basic ones built into IPCompleter
1615 1615 sdisp = self.strdispatchers.get('complete_command', StrDispatch())
1616 1616 self.strdispatchers['complete_command'] = sdisp
1617 1617 self.Completer.custom_completers = sdisp
1618 1618
1619 1619 self.set_hook('complete_command', module_completer, str_key = 'import')
1620 1620 self.set_hook('complete_command', module_completer, str_key = 'from')
1621 1621 self.set_hook('complete_command', magic_run_completer, str_key = '%run')
1622 1622 self.set_hook('complete_command', cd_completer, str_key = '%cd')
1623 1623
1624 1624 # Only configure readline if we truly are using readline. IPython can
1625 1625 # do tab-completion over the network, in GUIs, etc, where readline
1626 1626 # itself may be absent
1627 1627 if self.has_readline:
1628 1628 self.set_readline_completer()
1629 1629
1630 1630 def complete(self, text, line=None, cursor_pos=None):
1631 1631 """Return the completed text and a list of completions.
1632 1632
1633 1633 Parameters
1634 1634 ----------
1635 1635
1636 1636 text : string
1637 1637 A string of text to be completed on. It can be given as empty and
1638 1638 instead a line/position pair are given. In this case, the
1639 1639 completer itself will split the line like readline does.
1640 1640
1641 1641 line : string, optional
1642 1642 The complete line that text is part of.
1643 1643
1644 1644 cursor_pos : int, optional
1645 1645 The position of the cursor on the input line.
1646 1646
1647 1647 Returns
1648 1648 -------
1649 1649 text : string
1650 1650 The actual text that was completed.
1651 1651
1652 1652 matches : list
1653 1653 A sorted list with all possible completions.
1654 1654
1655 1655 The optional arguments allow the completion to take more context into
1656 1656 account, and are part of the low-level completion API.
1657 1657
1658 1658 This is a wrapper around the completion mechanism, similar to what
1659 1659 readline does at the command line when the TAB key is hit. By
1660 1660 exposing it as a method, it can be used by other non-readline
1661 1661 environments (such as GUIs) for text completion.
1662 1662
1663 1663 Simple usage example:
1664 1664
1665 1665 In [1]: x = 'hello'
1666 1666
1667 1667 In [2]: _ip.complete('x.l')
1668 1668 Out[2]: ('x.l', ['x.ljust', 'x.lower', 'x.lstrip'])
1669 1669 """
1670 1670
1671 1671 # Inject names into __builtin__ so we can complete on the added names.
1672 1672 with self.builtin_trap:
1673 1673 return self.Completer.complete(text, line, cursor_pos)
1674 1674
1675 1675 def set_custom_completer(self, completer, pos=0):
1676 1676 """Adds a new custom completer function.
1677 1677
1678 1678 The position argument (defaults to 0) is the index in the completers
1679 1679 list where you want the completer to be inserted."""
1680 1680
1681 1681 newcomp = types.MethodType(completer,self.Completer)
1682 1682 self.Completer.matchers.insert(pos,newcomp)
1683 1683
1684 1684 def set_readline_completer(self):
1685 1685 """Reset readline's completer to be our own."""
1686 1686 self.readline.set_completer(self.Completer.rlcomplete)
1687 1687
1688 1688 def set_completer_frame(self, frame=None):
1689 1689 """Set the frame of the completer."""
1690 1690 if frame:
1691 1691 self.Completer.namespace = frame.f_locals
1692 1692 self.Completer.global_namespace = frame.f_globals
1693 1693 else:
1694 1694 self.Completer.namespace = self.user_ns
1695 1695 self.Completer.global_namespace = self.user_global_ns
1696 1696
1697 1697 #-------------------------------------------------------------------------
1698 1698 # Things related to magics
1699 1699 #-------------------------------------------------------------------------
1700 1700
1701 1701 def init_magics(self):
1702 1702 # FIXME: Move the color initialization to the DisplayHook, which
1703 1703 # should be split into a prompt manager and displayhook. We probably
1704 1704 # even need a centralize colors management object.
1705 1705 self.magic_colors(self.colors)
1706 1706 # History was moved to a separate module
1707 1707 from . import history
1708 1708 history.init_ipython(self)
1709 1709
1710 1710 def magic(self,arg_s):
1711 1711 """Call a magic function by name.
1712 1712
1713 1713 Input: a string containing the name of the magic function to call and
1714 1714 any additional arguments to be passed to the magic.
1715 1715
1716 1716 magic('name -opt foo bar') is equivalent to typing at the ipython
1717 1717 prompt:
1718 1718
1719 1719 In[1]: %name -opt foo bar
1720 1720
1721 1721 To call a magic without arguments, simply use magic('name').
1722 1722
1723 1723 This provides a proper Python function to call IPython's magics in any
1724 1724 valid Python code you can type at the interpreter, including loops and
1725 1725 compound statements.
1726 1726 """
1727 1727 args = arg_s.split(' ',1)
1728 1728 magic_name = args[0]
1729 1729 magic_name = magic_name.lstrip(prefilter.ESC_MAGIC)
1730 1730
1731 1731 try:
1732 1732 magic_args = args[1]
1733 1733 except IndexError:
1734 1734 magic_args = ''
1735 1735 fn = getattr(self,'magic_'+magic_name,None)
1736 1736 if fn is None:
1737 1737 error("Magic function `%s` not found." % magic_name)
1738 1738 else:
1739 1739 magic_args = self.var_expand(magic_args,1)
1740 1740 with nested(self.builtin_trap,):
1741 1741 result = fn(magic_args)
1742 1742 return result
1743 1743
1744 1744 def define_magic(self, magicname, func):
1745 1745 """Expose own function as magic function for ipython
1746 1746
1747 1747 def foo_impl(self,parameter_s=''):
1748 1748 'My very own magic!. (Use docstrings, IPython reads them).'
1749 1749 print 'Magic function. Passed parameter is between < >:'
1750 1750 print '<%s>' % parameter_s
1751 1751 print 'The self object is:',self
1752 1752
1753 1753 self.define_magic('foo',foo_impl)
1754 1754 """
1755 1755
1756 1756 import new
1757 1757 im = types.MethodType(func,self)
1758 1758 old = getattr(self, "magic_" + magicname, None)
1759 1759 setattr(self, "magic_" + magicname, im)
1760 1760 return old
1761 1761
1762 1762 #-------------------------------------------------------------------------
1763 1763 # Things related to macros
1764 1764 #-------------------------------------------------------------------------
1765 1765
1766 1766 def define_macro(self, name, themacro):
1767 1767 """Define a new macro
1768 1768
1769 1769 Parameters
1770 1770 ----------
1771 1771 name : str
1772 1772 The name of the macro.
1773 1773 themacro : str or Macro
1774 1774 The action to do upon invoking the macro. If a string, a new
1775 1775 Macro object is created by passing the string to it.
1776 1776 """
1777 1777
1778 1778 from IPython.core import macro
1779 1779
1780 1780 if isinstance(themacro, basestring):
1781 1781 themacro = macro.Macro(themacro)
1782 1782 if not isinstance(themacro, macro.Macro):
1783 1783 raise ValueError('A macro must be a string or a Macro instance.')
1784 1784 self.user_ns[name] = themacro
1785 1785
1786 1786 #-------------------------------------------------------------------------
1787 1787 # Things related to the running of system commands
1788 1788 #-------------------------------------------------------------------------
1789 1789
1790 1790 def system(self, cmd):
1791 1791 """Call the given cmd in a subprocess.
1792 1792
1793 1793 Parameters
1794 1794 ----------
1795 1795 cmd : str
1796 1796 Command to execute (can not end in '&', as bacground processes are
1797 1797 not supported.
1798 1798 """
1799 1799 # We do not support backgrounding processes because we either use
1800 1800 # pexpect or pipes to read from. Users can always just call
1801 1801 # os.system() if they really want a background process.
1802 1802 if cmd.endswith('&'):
1803 1803 raise OSError("Background processes not supported.")
1804 1804
1805 1805 return system(self.var_expand(cmd, depth=2))
1806 1806
1807 1807 def getoutput(self, cmd, split=True):
1808 1808 """Get output (possibly including stderr) from a subprocess.
1809 1809
1810 1810 Parameters
1811 1811 ----------
1812 1812 cmd : str
1813 1813 Command to execute (can not end in '&', as background processes are
1814 1814 not supported.
1815 1815 split : bool, optional
1816 1816
1817 1817 If True, split the output into an IPython SList. Otherwise, an
1818 1818 IPython LSString is returned. These are objects similar to normal
1819 1819 lists and strings, with a few convenience attributes for easier
1820 1820 manipulation of line-based output. You can use '?' on them for
1821 1821 details.
1822 1822 """
1823 1823 if cmd.endswith('&'):
1824 1824 raise OSError("Background processes not supported.")
1825 1825 out = getoutput(self.var_expand(cmd, depth=2))
1826 1826 if split:
1827 1827 out = SList(out.splitlines())
1828 1828 else:
1829 1829 out = LSString(out)
1830 1830 return out
1831 1831
1832 1832 #-------------------------------------------------------------------------
1833 1833 # Things related to aliases
1834 1834 #-------------------------------------------------------------------------
1835 1835
1836 1836 def init_alias(self):
1837 1837 self.alias_manager = AliasManager(shell=self, config=self.config)
1838 1838 self.ns_table['alias'] = self.alias_manager.alias_table,
1839 1839
1840 1840 #-------------------------------------------------------------------------
1841 1841 # Things related to extensions and plugins
1842 1842 #-------------------------------------------------------------------------
1843 1843
1844 1844 def init_extension_manager(self):
1845 1845 self.extension_manager = ExtensionManager(shell=self, config=self.config)
1846 1846
1847 1847 def init_plugin_manager(self):
1848 1848 self.plugin_manager = PluginManager(config=self.config)
1849 1849
1850 1850 #-------------------------------------------------------------------------
1851 1851 # Things related to payloads
1852 1852 #-------------------------------------------------------------------------
1853 1853
1854 1854 def init_payload(self):
1855 1855 self.payload_manager = PayloadManager(config=self.config)
1856 1856
1857 1857 #-------------------------------------------------------------------------
1858 1858 # Things related to the prefilter
1859 1859 #-------------------------------------------------------------------------
1860 1860
1861 1861 def init_prefilter(self):
1862 1862 self.prefilter_manager = PrefilterManager(shell=self, config=self.config)
1863 1863 # Ultimately this will be refactored in the new interpreter code, but
1864 1864 # for now, we should expose the main prefilter method (there's legacy
1865 1865 # code out there that may rely on this).
1866 1866 self.prefilter = self.prefilter_manager.prefilter_lines
1867 1867
1868 1868 def auto_rewrite_input(self, cmd):
1869 1869 """Print to the screen the rewritten form of the user's command.
1870 1870
1871 1871 This shows visual feedback by rewriting input lines that cause
1872 1872 automatic calling to kick in, like::
1873 1873
1874 1874 /f x
1875 1875
1876 1876 into::
1877 1877
1878 1878 ------> f(x)
1879 1879
1880 1880 after the user's input prompt. This helps the user understand that the
1881 1881 input line was transformed automatically by IPython.
1882 1882 """
1883 1883 rw = self.displayhook.prompt1.auto_rewrite() + cmd
1884 1884
1885 1885 try:
1886 1886 # plain ascii works better w/ pyreadline, on some machines, so
1887 1887 # we use it and only print uncolored rewrite if we have unicode
1888 1888 rw = str(rw)
1889 1889 print >> IPython.utils.io.Term.cout, rw
1890 1890 except UnicodeEncodeError:
1891 1891 print "------> " + cmd
1892 1892
1893 1893 #-------------------------------------------------------------------------
1894 1894 # Things related to extracting values/expressions from kernel and user_ns
1895 1895 #-------------------------------------------------------------------------
1896 1896
1897 1897 def _simple_error(self):
1898 1898 etype, value = sys.exc_info()[:2]
1899 1899 return u'[ERROR] {e.__name__}: {v}'.format(e=etype, v=value)
1900 1900
1901 1901 def user_variables(self, names):
1902 1902 """Get a list of variable names from the user's namespace.
1903 1903
1904 1904 Parameters
1905 1905 ----------
1906 1906 names : list of strings
1907 1907 A list of names of variables to be read from the user namespace.
1908 1908
1909 1909 Returns
1910 1910 -------
1911 1911 A dict, keyed by the input names and with the repr() of each value.
1912 1912 """
1913 1913 out = {}
1914 1914 user_ns = self.user_ns
1915 1915 for varname in names:
1916 1916 try:
1917 1917 value = repr(user_ns[varname])
1918 1918 except:
1919 1919 value = self._simple_error()
1920 1920 out[varname] = value
1921 1921 return out
1922 1922
1923 1923 def user_expressions(self, expressions):
1924 1924 """Evaluate a dict of expressions in the user's namespace.
1925 1925
1926 1926 Parameters
1927 1927 ----------
1928 1928 expressions : dict
1929 1929 A dict with string keys and string values. The expression values
1930 1930 should be valid Python expressions, each of which will be evaluated
1931 1931 in the user namespace.
1932 1932
1933 1933 Returns
1934 1934 -------
1935 1935 A dict, keyed like the input expressions dict, with the repr() of each
1936 1936 value.
1937 1937 """
1938 1938 out = {}
1939 1939 user_ns = self.user_ns
1940 1940 global_ns = self.user_global_ns
1941 1941 for key, expr in expressions.iteritems():
1942 1942 try:
1943 1943 value = repr(eval(expr, global_ns, user_ns))
1944 1944 except:
1945 1945 value = self._simple_error()
1946 1946 out[key] = value
1947 1947 return out
1948 1948
1949 1949 #-------------------------------------------------------------------------
1950 1950 # Things related to the running of code
1951 1951 #-------------------------------------------------------------------------
1952 1952
1953 1953 def ex(self, cmd):
1954 1954 """Execute a normal python statement in user namespace."""
1955 1955 with nested(self.builtin_trap,):
1956 1956 exec cmd in self.user_global_ns, self.user_ns
1957 1957
1958 1958 def ev(self, expr):
1959 1959 """Evaluate python expression expr in user namespace.
1960 1960
1961 1961 Returns the result of evaluation
1962 1962 """
1963 1963 with nested(self.builtin_trap,):
1964 1964 return eval(expr, self.user_global_ns, self.user_ns)
1965 1965
1966 1966 def safe_execfile(self, fname, *where, **kw):
1967 1967 """A safe version of the builtin execfile().
1968 1968
1969 1969 This version will never throw an exception, but instead print
1970 1970 helpful error messages to the screen. This only works on pure
1971 1971 Python files with the .py extension.
1972 1972
1973 1973 Parameters
1974 1974 ----------
1975 1975 fname : string
1976 1976 The name of the file to be executed.
1977 1977 where : tuple
1978 1978 One or two namespaces, passed to execfile() as (globals,locals).
1979 1979 If only one is given, it is passed as both.
1980 1980 exit_ignore : bool (False)
1981 1981 If True, then silence SystemExit for non-zero status (it is always
1982 1982 silenced for zero status, as it is so common).
1983 1983 """
1984 1984 kw.setdefault('exit_ignore', False)
1985 1985
1986 1986 fname = os.path.abspath(os.path.expanduser(fname))
1987 1987
1988 1988 # Make sure we have a .py file
1989 1989 if not fname.endswith('.py'):
1990 1990 warn('File must end with .py to be run using execfile: <%s>' % fname)
1991 1991
1992 1992 # Make sure we can open the file
1993 1993 try:
1994 1994 with open(fname) as thefile:
1995 1995 pass
1996 1996 except:
1997 1997 warn('Could not open file <%s> for safe execution.' % fname)
1998 1998 return
1999 1999
2000 2000 # Find things also in current directory. This is needed to mimic the
2001 2001 # behavior of running a script from the system command line, where
2002 2002 # Python inserts the script's directory into sys.path
2003 2003 dname = os.path.dirname(fname)
2004 2004
2005 2005 with prepended_to_syspath(dname):
2006 2006 try:
2007 2007 execfile(fname,*where)
2008 2008 except SystemExit, status:
2009 2009 # If the call was made with 0 or None exit status (sys.exit(0)
2010 2010 # or sys.exit() ), don't bother showing a traceback, as both of
2011 2011 # these are considered normal by the OS:
2012 2012 # > python -c'import sys;sys.exit(0)'; echo $?
2013 2013 # 0
2014 2014 # > python -c'import sys;sys.exit()'; echo $?
2015 2015 # 0
2016 2016 # For other exit status, we show the exception unless
2017 2017 # explicitly silenced, but only in short form.
2018 2018 if status.code not in (0, None) and not kw['exit_ignore']:
2019 2019 self.showtraceback(exception_only=True)
2020 2020 except:
2021 2021 self.showtraceback()
2022 2022
2023 2023 def safe_execfile_ipy(self, fname):
2024 2024 """Like safe_execfile, but for .ipy files with IPython syntax.
2025 2025
2026 2026 Parameters
2027 2027 ----------
2028 2028 fname : str
2029 2029 The name of the file to execute. The filename must have a
2030 2030 .ipy extension.
2031 2031 """
2032 2032 fname = os.path.abspath(os.path.expanduser(fname))
2033 2033
2034 2034 # Make sure we have a .py file
2035 2035 if not fname.endswith('.ipy'):
2036 2036 warn('File must end with .py to be run using execfile: <%s>' % fname)
2037 2037
2038 2038 # Make sure we can open the file
2039 2039 try:
2040 2040 with open(fname) as thefile:
2041 2041 pass
2042 2042 except:
2043 2043 warn('Could not open file <%s> for safe execution.' % fname)
2044 2044 return
2045 2045
2046 2046 # Find things also in current directory. This is needed to mimic the
2047 2047 # behavior of running a script from the system command line, where
2048 2048 # Python inserts the script's directory into sys.path
2049 2049 dname = os.path.dirname(fname)
2050 2050
2051 2051 with prepended_to_syspath(dname):
2052 2052 try:
2053 2053 with open(fname) as thefile:
2054 2054 # self.run_cell currently captures all exceptions
2055 2055 # raised in user code. It would be nice if there were
2056 2056 # versions of runlines, execfile that did raise, so
2057 2057 # we could catch the errors.
2058 2058 self.run_cell(thefile.read())
2059 2059 except:
2060 2060 self.showtraceback()
2061 2061 warn('Unknown failure executing file: <%s>' % fname)
2062 2062
2063 2063 def run_cell(self, cell, store_history=True):
2064 2064 """Run the contents of an entire multiline 'cell' of code, and store it
2065 2065 in the history.
2066 2066
2067 2067 The cell is split into separate blocks which can be executed
2068 2068 individually. Then, based on how many blocks there are, they are
2069 2069 executed as follows:
2070 2070
2071 2071 - A single block: 'single' mode. If it is also a single line, dynamic
2072 2072 transformations, including automagic and macros, will be applied.
2073 2073
2074 2074 If there's more than one block, it depends:
2075 2075
2076 2076 - if the last one is no more than two lines long, run all but the last
2077 2077 in 'exec' mode and the very last one in 'single' mode. This makes it
2078 2078 easy to type simple expressions at the end to see computed values. -
2079 2079 otherwise (last one is also multiline), run all in 'exec' mode
2080 2080
2081 2081 When code is executed in 'single' mode, :func:`sys.displayhook` fires,
2082 2082 results are displayed and output prompts are computed. In 'exec' mode,
2083 2083 no results are displayed unless :func:`print` is called explicitly;
2084 2084 this mode is more akin to running a script.
2085 2085
2086 2086 Parameters
2087 2087 ----------
2088 2088 cell : str
2089 2089 A single or multiline string.
2090 2090 """
2091 2091 # Store the untransformed code
2092 2092 raw_cell = cell
2093 2093
2094 2094 # We only do dynamic transforms on a single line. We need to do this
2095 2095 # first, because a macro can be expanded to several lines, which then
2096 2096 # need to be split into blocks again.
2097 2097 if len(cell.splitlines()) <= 1:
2098 2098 temp = self.input_splitter.split_blocks(cell)
2099 2099 cell = self.prefilter_manager.prefilter_line(temp[0])
2100 2100
2101 2101 # We need to break up the input into executable blocks that can be run
2102 2102 # in 'single' mode, to provide comfortable user behavior.
2103 2103 blocks = self.input_splitter.split_blocks(cell)
2104 2104
2105 2105 if not blocks:
2106 2106 return
2107 2107
2108 2108 # Store the 'ipython' version of the cell as well, since that's what
2109 2109 # needs to go into the translated history and get executed (the
2110 2110 # original cell may contain non-python syntax).
2111 2111 cell = ''.join(blocks)
2112 2112
2113 2113 # Store raw and processed history
2114 2114 if store_history:
2115 2115 self.history_manager.store_inputs(self.execution_count,
2116 2116 cell, raw_cell)
2117 2117
2118 2118 self.logger.log(cell, raw_cell)
2119 2119
2120 2120 # All user code execution must happen with our context managers active
2121 2121 with nested(self.builtin_trap, self.display_trap):
2122 2122
2123 2123 # Single-block input should behave like an interactive prompt
2124 2124 if len(blocks) == 1:
2125 2125 out = self.run_source(blocks[0])
2126 2126 # Write output to the database. Does nothing unless
2127 2127 # history output logging is enabled.
2128 2128 if store_history:
2129 2129 self.history_manager.store_output(self.execution_count)
2130 2130 # since we return here, we need to update the execution count
2131 2131 self.execution_count += 1
2132 2132 return out
2133 2133
2134 2134 # In multi-block input, if the last block is a simple (one-two
2135 2135 # lines) expression, run it in single mode so it produces output.
2136 2136 # Otherwise just run it all in 'exec' mode. This seems like a
2137 2137 # reasonable usability design.
2138 2138 last = blocks[-1]
2139 2139 last_nlines = len(last.splitlines())
2140 2140
2141 2141 if last_nlines < 2:
2142 2142 # Here we consider the cell split between 'body' and 'last',
2143 2143 # store all history and execute 'body', and if successful, then
2144 2144 # proceed to execute 'last'.
2145 2145
2146 2146 # Get the main body to run as a cell
2147 2147 ipy_body = ''.join(blocks[:-1])
2148 2148 retcode = self.run_source(ipy_body, symbol='exec',
2149 2149 post_execute=False)
2150 2150 if retcode==0:
2151 2151 # Last expression compiled as 'single' so it produces output
2152 2152 self.run_source(last)
2153 2153 else:
2154 2154 # Run the whole cell as one entity, storing both raw and
2155 2155 # processed input in history
2156 2156 self.run_source(ipy_cell, symbol='exec')
2157 2157
2158 2158 # Write output to the database. Does nothing unless
2159 2159 # history output logging is enabled.
2160 2160 if store_history:
2161 2161 self.history_manager.store_output(self.execution_count)
2162 2162 # Each cell is a *single* input, regardless of how many lines it has
2163 2163 self.execution_count += 1
2164 2164
2165 2165 # PENDING REMOVAL: this method is slated for deletion, once our new
2166 2166 # input logic has been 100% moved to frontends and is stable.
2167 2167 def runlines(self, lines, clean=False):
2168 2168 """Run a string of one or more lines of source.
2169 2169
2170 2170 This method is capable of running a string containing multiple source
2171 2171 lines, as if they had been entered at the IPython prompt. Since it
2172 2172 exposes IPython's processing machinery, the given strings can contain
2173 2173 magic calls (%magic), special shell access (!cmd), etc.
2174 2174 """
2175 2175
2176 2176 if not isinstance(lines, (list, tuple)):
2177 2177 lines = lines.splitlines()
2178 2178
2179 2179 if clean:
2180 2180 lines = self._cleanup_ipy_script(lines)
2181 2181
2182 2182 # We must start with a clean buffer, in case this is run from an
2183 2183 # interactive IPython session (via a magic, for example).
2184 2184 self.reset_buffer()
2185 2185
2186 2186 # Since we will prefilter all lines, store the user's raw input too
2187 2187 # before we apply any transformations
2188 2188 self.buffer_raw[:] = [ l+'\n' for l in lines]
2189 2189
2190 2190 more = False
2191 2191 prefilter_lines = self.prefilter_manager.prefilter_lines
2192 2192 with nested(self.builtin_trap, self.display_trap):
2193 2193 for line in lines:
2194 2194 # skip blank lines so we don't mess up the prompt counter, but
2195 2195 # do NOT skip even a blank line if we are in a code block (more
2196 2196 # is true)
2197 2197
2198 2198 if line or more:
2199 2199 more = self.push_line(prefilter_lines(line, more))
2200 2200 # IPython's run_source returns None if there was an error
2201 2201 # compiling the code. This allows us to stop processing
2202 2202 # right away, so the user gets the error message at the
2203 2203 # right place.
2204 2204 if more is None:
2205 2205 break
2206 2206 # final newline in case the input didn't have it, so that the code
2207 2207 # actually does get executed
2208 2208 if more:
2209 2209 self.push_line('\n')
2210 2210
2211 2211 def run_source(self, source, filename=None,
2212 2212 symbol='single', post_execute=True):
2213 2213 """Compile and run some source in the interpreter.
2214 2214
2215 2215 Arguments are as for compile_command().
2216 2216
2217 2217 One several things can happen:
2218 2218
2219 2219 1) The input is incorrect; compile_command() raised an
2220 2220 exception (SyntaxError or OverflowError). A syntax traceback
2221 2221 will be printed by calling the showsyntaxerror() method.
2222 2222
2223 2223 2) The input is incomplete, and more input is required;
2224 2224 compile_command() returned None. Nothing happens.
2225 2225
2226 2226 3) The input is complete; compile_command() returned a code
2227 2227 object. The code is executed by calling self.run_code() (which
2228 2228 also handles run-time exceptions, except for SystemExit).
2229 2229
2230 2230 The return value is:
2231 2231
2232 2232 - True in case 2
2233 2233
2234 2234 - False in the other cases, unless an exception is raised, where
2235 2235 None is returned instead. This can be used by external callers to
2236 2236 know whether to continue feeding input or not.
2237 2237
2238 2238 The return value can be used to decide whether to use sys.ps1 or
2239 2239 sys.ps2 to prompt the next line."""
2240 2240
2241 2241 # We need to ensure that the source is unicode from here on.
2242 2242 if type(source)==str:
2243 2243 usource = source.decode(self.stdin_encoding)
2244 2244 else:
2245 2245 usource = source
2246 2246
2247 2247 if 0: # dbg
2248 2248 print 'Source:', repr(source) # dbg
2249 2249 print 'USource:', repr(usource) # dbg
2250 2250 print 'type:', type(source) # dbg
2251 2251 print 'encoding', self.stdin_encoding # dbg
2252 2252
2253 2253 try:
2254 2254 code = self.compile(usource, symbol, self.execution_count)
2255 2255 except (OverflowError, SyntaxError, ValueError, TypeError, MemoryError):
2256 2256 # Case 1
2257 2257 self.showsyntaxerror(filename)
2258 2258 return None
2259 2259
2260 2260 if code is None:
2261 2261 # Case 2
2262 2262 return True
2263 2263
2264 2264 # Case 3
2265 2265 # We store the code object so that threaded shells and
2266 2266 # custom exception handlers can access all this info if needed.
2267 2267 # The source corresponding to this can be obtained from the
2268 2268 # buffer attribute as '\n'.join(self.buffer).
2269 2269 self.code_to_run = code
2270 2270 # now actually execute the code object
2271 2271 if self.run_code(code, post_execute) == 0:
2272 2272 return False
2273 2273 else:
2274 2274 return None
2275 2275
2276 2276 # For backwards compatibility
2277 2277 runsource = run_source
2278 2278
2279 2279 def run_code(self, code_obj, post_execute=True):
2280 2280 """Execute a code object.
2281 2281
2282 2282 When an exception occurs, self.showtraceback() is called to display a
2283 2283 traceback.
2284 2284
2285 2285 Return value: a flag indicating whether the code to be run completed
2286 2286 successfully:
2287 2287
2288 2288 - 0: successful execution.
2289 2289 - 1: an error occurred.
2290 2290 """
2291 2291
2292 2292 # Set our own excepthook in case the user code tries to call it
2293 2293 # directly, so that the IPython crash handler doesn't get triggered
2294 2294 old_excepthook,sys.excepthook = sys.excepthook, self.excepthook
2295 2295
2296 2296 # we save the original sys.excepthook in the instance, in case config
2297 2297 # code (such as magics) needs access to it.
2298 2298 self.sys_excepthook = old_excepthook
2299 2299 outflag = 1 # happens in more places, so it's easier as default
2300 2300 try:
2301 2301 try:
2302 2302 self.hooks.pre_run_code_hook()
2303 2303 #rprint('Running code', repr(code_obj)) # dbg
2304 2304 exec code_obj in self.user_global_ns, self.user_ns
2305 2305 finally:
2306 2306 # Reset our crash handler in place
2307 2307 sys.excepthook = old_excepthook
2308 2308 except SystemExit:
2309 2309 self.reset_buffer()
2310 2310 self.showtraceback(exception_only=True)
2311 2311 warn("To exit: use any of 'exit', 'quit', %Exit or Ctrl-D.", level=1)
2312 2312 except self.custom_exceptions:
2313 2313 etype,value,tb = sys.exc_info()
2314 2314 self.CustomTB(etype,value,tb)
2315 2315 except:
2316 2316 self.showtraceback()
2317 2317 else:
2318 2318 outflag = 0
2319 2319 if softspace(sys.stdout, 0):
2320 2320 print
2321 2321
2322 2322 # Execute any registered post-execution functions. Here, any errors
2323 2323 # are reported only minimally and just on the terminal, because the
2324 2324 # main exception channel may be occupied with a user traceback.
2325 2325 # FIXME: we need to think this mechanism a little more carefully.
2326 2326 if post_execute:
2327 2327 for func in self._post_execute:
2328 2328 try:
2329 2329 func()
2330 2330 except:
2331 2331 head = '[ ERROR ] Evaluating post_execute function: %s' % \
2332 2332 func
2333 2333 print >> io.Term.cout, head
2334 2334 print >> io.Term.cout, self._simple_error()
2335 2335 print >> io.Term.cout, 'Removing from post_execute'
2336 2336 self._post_execute.remove(func)
2337 2337
2338 2338 # Flush out code object which has been run (and source)
2339 2339 self.code_to_run = None
2340 2340 return outflag
2341 2341
2342 2342 # For backwards compatibility
2343 2343 runcode = run_code
2344 2344
2345 2345 # PENDING REMOVAL: this method is slated for deletion, once our new
2346 2346 # input logic has been 100% moved to frontends and is stable.
2347 2347 def push_line(self, line):
2348 2348 """Push a line to the interpreter.
2349 2349
2350 2350 The line should not have a trailing newline; it may have
2351 2351 internal newlines. The line is appended to a buffer and the
2352 2352 interpreter's run_source() method is called with the
2353 2353 concatenated contents of the buffer as source. If this
2354 2354 indicates that the command was executed or invalid, the buffer
2355 2355 is reset; otherwise, the command is incomplete, and the buffer
2356 2356 is left as it was after the line was appended. The return
2357 2357 value is 1 if more input is required, 0 if the line was dealt
2358 2358 with in some way (this is the same as run_source()).
2359 2359 """
2360 2360
2361 2361 # autoindent management should be done here, and not in the
2362 2362 # interactive loop, since that one is only seen by keyboard input. We
2363 2363 # need this done correctly even for code run via runlines (which uses
2364 2364 # push).
2365 2365
2366 2366 #print 'push line: <%s>' % line # dbg
2367 2367 self.buffer.append(line)
2368 2368 full_source = '\n'.join(self.buffer)
2369 2369 more = self.run_source(full_source, self.filename)
2370 2370 if not more:
2371 2371 self.history_manager.store_inputs(self.execution_count,
2372 2372 '\n'.join(self.buffer_raw), full_source)
2373 2373 self.reset_buffer()
2374 2374 self.execution_count += 1
2375 2375 return more
2376 2376
2377 2377 def reset_buffer(self):
2378 2378 """Reset the input buffer."""
2379 2379 self.buffer[:] = []
2380 2380 self.buffer_raw[:] = []
2381 2381 self.input_splitter.reset()
2382 2382
2383 2383 # For backwards compatibility
2384 2384 resetbuffer = reset_buffer
2385 2385
2386 2386 def _is_secondary_block_start(self, s):
2387 2387 if not s.endswith(':'):
2388 2388 return False
2389 2389 if (s.startswith('elif') or
2390 2390 s.startswith('else') or
2391 2391 s.startswith('except') or
2392 2392 s.startswith('finally')):
2393 2393 return True
2394 2394
2395 2395 def _cleanup_ipy_script(self, script):
2396 2396 """Make a script safe for self.runlines()
2397 2397
2398 2398 Currently, IPython is lines based, with blocks being detected by
2399 2399 empty lines. This is a problem for block based scripts that may
2400 2400 not have empty lines after blocks. This script adds those empty
2401 2401 lines to make scripts safe for running in the current line based
2402 2402 IPython.
2403 2403 """
2404 2404 res = []
2405 2405 lines = script.splitlines()
2406 2406 level = 0
2407 2407
2408 2408 for l in lines:
2409 2409 lstripped = l.lstrip()
2410 2410 stripped = l.strip()
2411 2411 if not stripped:
2412 2412 continue
2413 2413 newlevel = len(l) - len(lstripped)
2414 2414 if level > 0 and newlevel == 0 and \
2415 2415 not self._is_secondary_block_start(stripped):
2416 2416 # add empty line
2417 2417 res.append('')
2418 2418 res.append(l)
2419 2419 level = newlevel
2420 2420
2421 2421 return '\n'.join(res) + '\n'
2422 2422
2423 2423 #-------------------------------------------------------------------------
2424 2424 # Things related to GUI support and pylab
2425 2425 #-------------------------------------------------------------------------
2426 2426
2427 2427 def enable_pylab(self, gui=None):
2428 2428 raise NotImplementedError('Implement enable_pylab in a subclass')
2429 2429
2430 2430 #-------------------------------------------------------------------------
2431 2431 # Utilities
2432 2432 #-------------------------------------------------------------------------
2433 2433
2434 2434 def var_expand(self,cmd,depth=0):
2435 2435 """Expand python variables in a string.
2436 2436
2437 2437 The depth argument indicates how many frames above the caller should
2438 2438 be walked to look for the local namespace where to expand variables.
2439 2439
2440 2440 The global namespace for expansion is always the user's interactive
2441 2441 namespace.
2442 2442 """
2443 2443
2444 2444 return str(ItplNS(cmd,
2445 2445 self.user_ns, # globals
2446 2446 # Skip our own frame in searching for locals:
2447 2447 sys._getframe(depth+1).f_locals # locals
2448 2448 ))
2449 2449
2450 2450 def mktempfile(self, data=None, prefix='ipython_edit_'):
2451 2451 """Make a new tempfile and return its filename.
2452 2452
2453 2453 This makes a call to tempfile.mktemp, but it registers the created
2454 2454 filename internally so ipython cleans it up at exit time.
2455 2455
2456 2456 Optional inputs:
2457 2457
2458 2458 - data(None): if data is given, it gets written out to the temp file
2459 2459 immediately, and the file is closed again."""
2460 2460
2461 2461 filename = tempfile.mktemp('.py', prefix)
2462 2462 self.tempfiles.append(filename)
2463 2463
2464 2464 if data:
2465 2465 tmp_file = open(filename,'w')
2466 2466 tmp_file.write(data)
2467 2467 tmp_file.close()
2468 2468 return filename
2469 2469
2470 2470 # TODO: This should be removed when Term is refactored.
2471 2471 def write(self,data):
2472 2472 """Write a string to the default output"""
2473 2473 io.Term.cout.write(data)
2474 2474
2475 2475 # TODO: This should be removed when Term is refactored.
2476 2476 def write_err(self,data):
2477 2477 """Write a string to the default error output"""
2478 2478 io.Term.cerr.write(data)
2479 2479
2480 2480 def ask_yes_no(self,prompt,default=True):
2481 2481 if self.quiet:
2482 2482 return True
2483 2483 return ask_yes_no(prompt,default)
2484 2484
2485 2485 def show_usage(self):
2486 2486 """Show a usage message"""
2487 2487 page.page(IPython.core.usage.interactive_usage)
2488 2488
2489 2489 #-------------------------------------------------------------------------
2490 2490 # Things related to IPython exiting
2491 2491 #-------------------------------------------------------------------------
2492 2492 def atexit_operations(self):
2493 2493 """This will be executed at the time of exit.
2494 2494
2495 2495 Cleanup operations and saving of persistent data that is done
2496 2496 unconditionally by IPython should be performed here.
2497 2497
2498 2498 For things that may depend on startup flags or platform specifics (such
2499 2499 as having readline or not), register a separate atexit function in the
2500 2500 code that has the appropriate information, rather than trying to
2501 2501 clutter
2502 2502 """
2503 2503 # Cleanup all tempfiles left around
2504 2504 for tfile in self.tempfiles:
2505 2505 try:
2506 2506 os.unlink(tfile)
2507 2507 except OSError:
2508 2508 pass
2509 2509
2510 2510 # Close the history session (this stores the end time and line count)
2511 2511 self.history_manager.end_session()
2512 2512
2513 2513 # Clear all user namespaces to release all references cleanly.
2514 2514 self.reset(new_session=False)
2515 2515
2516 2516 # Run user hooks
2517 2517 self.hooks.shutdown_hook()
2518 2518
2519 2519 def cleanup(self):
2520 2520 self.restore_sys_module_state()
2521 2521
2522 2522
2523 2523 class InteractiveShellABC(object):
2524 2524 """An abstract base class for InteractiveShell."""
2525 2525 __metaclass__ = abc.ABCMeta
2526 2526
2527 2527 InteractiveShellABC.register(InteractiveShell)
@@ -1,3456 +1,3450 b''
1 1 # encoding: utf-8
2 2 """Magic functions for InteractiveShell.
3 3 """
4 4
5 5 #-----------------------------------------------------------------------------
6 6 # Copyright (C) 2001 Janko Hauser <jhauser@zscout.de> and
7 7 # Copyright (C) 2001-2007 Fernando Perez <fperez@colorado.edu>
8 8 # Copyright (C) 2008-2009 The IPython Development Team
9 9
10 10 # Distributed under the terms of the BSD License. The full license is in
11 11 # the file COPYING, distributed as part of this software.
12 12 #-----------------------------------------------------------------------------
13 13
14 14 #-----------------------------------------------------------------------------
15 15 # Imports
16 16 #-----------------------------------------------------------------------------
17 17
18 18 import __builtin__
19 19 import __future__
20 20 import bdb
21 21 import inspect
22 22 import os
23 23 import sys
24 24 import shutil
25 25 import re
26 26 import time
27 27 import textwrap
28 28 import types
29 29 from cStringIO import StringIO
30 30 from getopt import getopt,GetoptError
31 31 from pprint import pformat
32 32
33 33 # cProfile was added in Python2.5
34 34 try:
35 35 import cProfile as profile
36 36 import pstats
37 37 except ImportError:
38 38 # profile isn't bundled by default in Debian for license reasons
39 39 try:
40 40 import profile,pstats
41 41 except ImportError:
42 42 profile = pstats = None
43 43
44 44 import IPython
45 45 from IPython.core import debugger, oinspect
46 46 from IPython.core.error import TryNext
47 47 from IPython.core.error import UsageError
48 48 from IPython.core.fakemodule import FakeModule
49 49 from IPython.core.macro import Macro
50 50 from IPython.core import page
51 51 from IPython.core.prefilter import ESC_MAGIC
52 52 from IPython.lib.pylabtools import mpl_runner
53 53 from IPython.external.Itpl import itpl, printpl
54 54 from IPython.testing import decorators as testdec
55 55 from IPython.utils.io import file_read, nlprint
56 56 import IPython.utils.io
57 57 from IPython.utils.path import get_py_filename
58 58 from IPython.utils.process import arg_split, abbrev_cwd
59 59 from IPython.utils.terminal import set_term_title
60 60 from IPython.utils.text import LSString, SList, format_screen
61 61 from IPython.utils.timing import clock, clock2
62 62 from IPython.utils.warn import warn, error
63 63 from IPython.utils.ipstruct import Struct
64 64 import IPython.utils.generics
65 65
66 66 #-----------------------------------------------------------------------------
67 67 # Utility functions
68 68 #-----------------------------------------------------------------------------
69 69
70 70 def on_off(tag):
71 71 """Return an ON/OFF string for a 1/0 input. Simple utility function."""
72 72 return ['OFF','ON'][tag]
73 73
74 74 class Bunch: pass
75 75
76 76 def compress_dhist(dh):
77 77 head, tail = dh[:-10], dh[-10:]
78 78
79 79 newhead = []
80 80 done = set()
81 81 for h in head:
82 82 if h in done:
83 83 continue
84 84 newhead.append(h)
85 85 done.add(h)
86 86
87 87 return newhead + tail
88 88
89 89
90 90 #***************************************************************************
91 91 # Main class implementing Magic functionality
92 92
93 93 # XXX - for some odd reason, if Magic is made a new-style class, we get errors
94 94 # on construction of the main InteractiveShell object. Something odd is going
95 95 # on with super() calls, Configurable and the MRO... For now leave it as-is, but
96 96 # eventually this needs to be clarified.
97 97 # BG: This is because InteractiveShell inherits from this, but is itself a
98 98 # Configurable. This messes up the MRO in some way. The fix is that we need to
99 99 # make Magic a configurable that InteractiveShell does not subclass.
100 100
101 101 class Magic:
102 102 """Magic functions for InteractiveShell.
103 103
104 104 Shell functions which can be reached as %function_name. All magic
105 105 functions should accept a string, which they can parse for their own
106 106 needs. This can make some functions easier to type, eg `%cd ../`
107 107 vs. `%cd("../")`
108 108
109 109 ALL definitions MUST begin with the prefix magic_. The user won't need it
110 110 at the command line, but it is is needed in the definition. """
111 111
112 112 # class globals
113 113 auto_status = ['Automagic is OFF, % prefix IS needed for magic functions.',
114 114 'Automagic is ON, % prefix NOT needed for magic functions.']
115 115
116 116 #......................................................................
117 117 # some utility functions
118 118
119 119 def __init__(self,shell):
120 120
121 121 self.options_table = {}
122 122 if profile is None:
123 123 self.magic_prun = self.profile_missing_notice
124 124 self.shell = shell
125 125
126 126 # namespace for holding state we may need
127 127 self._magic_state = Bunch()
128 128
129 129 def profile_missing_notice(self, *args, **kwargs):
130 130 error("""\
131 131 The profile module could not be found. It has been removed from the standard
132 132 python packages because of its non-free license. To use profiling, install the
133 133 python-profiler package from non-free.""")
134 134
135 135 def default_option(self,fn,optstr):
136 136 """Make an entry in the options_table for fn, with value optstr"""
137 137
138 138 if fn not in self.lsmagic():
139 139 error("%s is not a magic function" % fn)
140 140 self.options_table[fn] = optstr
141 141
142 142 def lsmagic(self):
143 143 """Return a list of currently available magic functions.
144 144
145 145 Gives a list of the bare names after mangling (['ls','cd', ...], not
146 146 ['magic_ls','magic_cd',...]"""
147 147
148 148 # FIXME. This needs a cleanup, in the way the magics list is built.
149 149
150 150 # magics in class definition
151 151 class_magic = lambda fn: fn.startswith('magic_') and \
152 152 callable(Magic.__dict__[fn])
153 153 # in instance namespace (run-time user additions)
154 154 inst_magic = lambda fn: fn.startswith('magic_') and \
155 155 callable(self.__dict__[fn])
156 156 # and bound magics by user (so they can access self):
157 157 inst_bound_magic = lambda fn: fn.startswith('magic_') and \
158 158 callable(self.__class__.__dict__[fn])
159 159 magics = filter(class_magic,Magic.__dict__.keys()) + \
160 160 filter(inst_magic,self.__dict__.keys()) + \
161 161 filter(inst_bound_magic,self.__class__.__dict__.keys())
162 162 out = []
163 163 for fn in set(magics):
164 164 out.append(fn.replace('magic_','',1))
165 165 out.sort()
166 166 return out
167 167
168 168 def extract_input_lines(self, range_str, raw=False):
169 169 """Return as a string a set of input history slices.
170 170
171 171 Inputs:
172 172
173 173 - range_str: the set of slices is given as a string, like
174 174 "~5/6-~4/2 4:8 9", since this function is for use by magic functions
175 175 which get their arguments as strings. The number before the / is the
176 176 session number: ~n goes n back from the current session.
177 177
178 178 Optional inputs:
179 179
180 180 - raw(False): by default, the processed input is used. If this is
181 181 true, the raw input history is used instead.
182 182
183 183 Note that slices can be called with two notations:
184 184
185 185 N:M -> standard python form, means including items N...(M-1).
186 186
187 187 N-M -> include items N..M (closed endpoint)."""
188 188 lines = self.shell.history_manager.\
189 get_hist_from_rangestr(range_str, raw=raw)
189 get_range_by_str(range_str, raw=raw)
190 190 return "\n".join(x for _, _, x in lines)
191 191
192 192 def arg_err(self,func):
193 193 """Print docstring if incorrect arguments were passed"""
194 194 print 'Error in arguments:'
195 195 print oinspect.getdoc(func)
196 196
197 197 def format_latex(self,strng):
198 198 """Format a string for latex inclusion."""
199 199
200 200 # Characters that need to be escaped for latex:
201 201 escape_re = re.compile(r'(%|_|\$|#|&)',re.MULTILINE)
202 202 # Magic command names as headers:
203 203 cmd_name_re = re.compile(r'^(%s.*?):' % ESC_MAGIC,
204 204 re.MULTILINE)
205 205 # Magic commands
206 206 cmd_re = re.compile(r'(?P<cmd>%s.+?\b)(?!\}\}:)' % ESC_MAGIC,
207 207 re.MULTILINE)
208 208 # Paragraph continue
209 209 par_re = re.compile(r'\\$',re.MULTILINE)
210 210
211 211 # The "\n" symbol
212 212 newline_re = re.compile(r'\\n')
213 213
214 214 # Now build the string for output:
215 215 #strng = cmd_name_re.sub(r'\n\\texttt{\\textsl{\\large \1}}:',strng)
216 216 strng = cmd_name_re.sub(r'\n\\bigskip\n\\texttt{\\textbf{ \1}}:',
217 217 strng)
218 218 strng = cmd_re.sub(r'\\texttt{\g<cmd>}',strng)
219 219 strng = par_re.sub(r'\\\\',strng)
220 220 strng = escape_re.sub(r'\\\1',strng)
221 221 strng = newline_re.sub(r'\\textbackslash{}n',strng)
222 222 return strng
223 223
224 224 def parse_options(self,arg_str,opt_str,*long_opts,**kw):
225 225 """Parse options passed to an argument string.
226 226
227 227 The interface is similar to that of getopt(), but it returns back a
228 228 Struct with the options as keys and the stripped argument string still
229 229 as a string.
230 230
231 231 arg_str is quoted as a true sys.argv vector by using shlex.split.
232 232 This allows us to easily expand variables, glob files, quote
233 233 arguments, etc.
234 234
235 235 Options:
236 236 -mode: default 'string'. If given as 'list', the argument string is
237 237 returned as a list (split on whitespace) instead of a string.
238 238
239 239 -list_all: put all option values in lists. Normally only options
240 240 appearing more than once are put in a list.
241 241
242 242 -posix (True): whether to split the input line in POSIX mode or not,
243 243 as per the conventions outlined in the shlex module from the
244 244 standard library."""
245 245
246 246 # inject default options at the beginning of the input line
247 247 caller = sys._getframe(1).f_code.co_name.replace('magic_','')
248 248 arg_str = '%s %s' % (self.options_table.get(caller,''),arg_str)
249 249
250 250 mode = kw.get('mode','string')
251 251 if mode not in ['string','list']:
252 252 raise ValueError,'incorrect mode given: %s' % mode
253 253 # Get options
254 254 list_all = kw.get('list_all',0)
255 255 posix = kw.get('posix', os.name == 'posix')
256 256
257 257 # Check if we have more than one argument to warrant extra processing:
258 258 odict = {} # Dictionary with options
259 259 args = arg_str.split()
260 260 if len(args) >= 1:
261 261 # If the list of inputs only has 0 or 1 thing in it, there's no
262 262 # need to look for options
263 263 argv = arg_split(arg_str,posix)
264 264 # Do regular option processing
265 265 try:
266 266 opts,args = getopt(argv,opt_str,*long_opts)
267 267 except GetoptError,e:
268 268 raise UsageError('%s ( allowed: "%s" %s)' % (e.msg,opt_str,
269 269 " ".join(long_opts)))
270 270 for o,a in opts:
271 271 if o.startswith('--'):
272 272 o = o[2:]
273 273 else:
274 274 o = o[1:]
275 275 try:
276 276 odict[o].append(a)
277 277 except AttributeError:
278 278 odict[o] = [odict[o],a]
279 279 except KeyError:
280 280 if list_all:
281 281 odict[o] = [a]
282 282 else:
283 283 odict[o] = a
284 284
285 285 # Prepare opts,args for return
286 286 opts = Struct(odict)
287 287 if mode == 'string':
288 288 args = ' '.join(args)
289 289
290 290 return opts,args
291 291
292 292 #......................................................................
293 293 # And now the actual magic functions
294 294
295 295 # Functions for IPython shell work (vars,funcs, config, etc)
296 296 def magic_lsmagic(self, parameter_s = ''):
297 297 """List currently available magic functions."""
298 298 mesc = ESC_MAGIC
299 299 print 'Available magic functions:\n'+mesc+\
300 300 (' '+mesc).join(self.lsmagic())
301 301 print '\n' + Magic.auto_status[self.shell.automagic]
302 302 return None
303 303
304 304 def magic_magic(self, parameter_s = ''):
305 305 """Print information about the magic function system.
306 306
307 307 Supported formats: -latex, -brief, -rest
308 308 """
309 309
310 310 mode = ''
311 311 try:
312 312 if parameter_s.split()[0] == '-latex':
313 313 mode = 'latex'
314 314 if parameter_s.split()[0] == '-brief':
315 315 mode = 'brief'
316 316 if parameter_s.split()[0] == '-rest':
317 317 mode = 'rest'
318 318 rest_docs = []
319 319 except:
320 320 pass
321 321
322 322 magic_docs = []
323 323 for fname in self.lsmagic():
324 324 mname = 'magic_' + fname
325 325 for space in (Magic,self,self.__class__):
326 326 try:
327 327 fn = space.__dict__[mname]
328 328 except KeyError:
329 329 pass
330 330 else:
331 331 break
332 332 if mode == 'brief':
333 333 # only first line
334 334 if fn.__doc__:
335 335 fndoc = fn.__doc__.split('\n',1)[0]
336 336 else:
337 337 fndoc = 'No documentation'
338 338 else:
339 339 if fn.__doc__:
340 340 fndoc = fn.__doc__.rstrip()
341 341 else:
342 342 fndoc = 'No documentation'
343 343
344 344
345 345 if mode == 'rest':
346 346 rest_docs.append('**%s%s**::\n\n\t%s\n\n' %(ESC_MAGIC,
347 347 fname,fndoc))
348 348
349 349 else:
350 350 magic_docs.append('%s%s:\n\t%s\n' %(ESC_MAGIC,
351 351 fname,fndoc))
352 352
353 353 magic_docs = ''.join(magic_docs)
354 354
355 355 if mode == 'rest':
356 356 return "".join(rest_docs)
357 357
358 358 if mode == 'latex':
359 359 print self.format_latex(magic_docs)
360 360 return
361 361 else:
362 362 magic_docs = format_screen(magic_docs)
363 363 if mode == 'brief':
364 364 return magic_docs
365 365
366 366 outmsg = """
367 367 IPython's 'magic' functions
368 368 ===========================
369 369
370 370 The magic function system provides a series of functions which allow you to
371 371 control the behavior of IPython itself, plus a lot of system-type
372 372 features. All these functions are prefixed with a % character, but parameters
373 373 are given without parentheses or quotes.
374 374
375 375 NOTE: If you have 'automagic' enabled (via the command line option or with the
376 376 %automagic function), you don't need to type in the % explicitly. By default,
377 377 IPython ships with automagic on, so you should only rarely need the % escape.
378 378
379 379 Example: typing '%cd mydir' (without the quotes) changes you working directory
380 380 to 'mydir', if it exists.
381 381
382 382 You can define your own magic functions to extend the system. See the supplied
383 383 ipythonrc and example-magic.py files for details (in your ipython
384 384 configuration directory, typically $HOME/.config/ipython on Linux or $HOME/.ipython elsewhere).
385 385
386 386 You can also define your own aliased names for magic functions. In your
387 387 ipythonrc file, placing a line like:
388 388
389 389 execute __IPYTHON__.magic_pf = __IPYTHON__.magic_profile
390 390
391 391 will define %pf as a new name for %profile.
392 392
393 393 You can also call magics in code using the magic() function, which IPython
394 394 automatically adds to the builtin namespace. Type 'magic?' for details.
395 395
396 396 For a list of the available magic functions, use %lsmagic. For a description
397 397 of any of them, type %magic_name?, e.g. '%cd?'.
398 398
399 399 Currently the magic system has the following functions:\n"""
400 400
401 401 mesc = ESC_MAGIC
402 402 outmsg = ("%s\n%s\n\nSummary of magic functions (from %slsmagic):"
403 403 "\n\n%s%s\n\n%s" % (outmsg,
404 404 magic_docs,mesc,mesc,
405 405 (' '+mesc).join(self.lsmagic()),
406 406 Magic.auto_status[self.shell.automagic] ) )
407 407 page.page(outmsg)
408 408
409 409 def magic_automagic(self, parameter_s = ''):
410 410 """Make magic functions callable without having to type the initial %.
411 411
412 412 Without argumentsl toggles on/off (when off, you must call it as
413 413 %automagic, of course). With arguments it sets the value, and you can
414 414 use any of (case insensitive):
415 415
416 416 - on,1,True: to activate
417 417
418 418 - off,0,False: to deactivate.
419 419
420 420 Note that magic functions have lowest priority, so if there's a
421 421 variable whose name collides with that of a magic fn, automagic won't
422 422 work for that function (you get the variable instead). However, if you
423 423 delete the variable (del var), the previously shadowed magic function
424 424 becomes visible to automagic again."""
425 425
426 426 arg = parameter_s.lower()
427 427 if parameter_s in ('on','1','true'):
428 428 self.shell.automagic = True
429 429 elif parameter_s in ('off','0','false'):
430 430 self.shell.automagic = False
431 431 else:
432 432 self.shell.automagic = not self.shell.automagic
433 433 print '\n' + Magic.auto_status[self.shell.automagic]
434 434
435 435 @testdec.skip_doctest
436 436 def magic_autocall(self, parameter_s = ''):
437 437 """Make functions callable without having to type parentheses.
438 438
439 439 Usage:
440 440
441 441 %autocall [mode]
442 442
443 443 The mode can be one of: 0->Off, 1->Smart, 2->Full. If not given, the
444 444 value is toggled on and off (remembering the previous state).
445 445
446 446 In more detail, these values mean:
447 447
448 448 0 -> fully disabled
449 449
450 450 1 -> active, but do not apply if there are no arguments on the line.
451 451
452 452 In this mode, you get:
453 453
454 454 In [1]: callable
455 455 Out[1]: <built-in function callable>
456 456
457 457 In [2]: callable 'hello'
458 458 ------> callable('hello')
459 459 Out[2]: False
460 460
461 461 2 -> Active always. Even if no arguments are present, the callable
462 462 object is called:
463 463
464 464 In [2]: float
465 465 ------> float()
466 466 Out[2]: 0.0
467 467
468 468 Note that even with autocall off, you can still use '/' at the start of
469 469 a line to treat the first argument on the command line as a function
470 470 and add parentheses to it:
471 471
472 472 In [8]: /str 43
473 473 ------> str(43)
474 474 Out[8]: '43'
475 475
476 476 # all-random (note for auto-testing)
477 477 """
478 478
479 479 if parameter_s:
480 480 arg = int(parameter_s)
481 481 else:
482 482 arg = 'toggle'
483 483
484 484 if not arg in (0,1,2,'toggle'):
485 485 error('Valid modes: (0->Off, 1->Smart, 2->Full')
486 486 return
487 487
488 488 if arg in (0,1,2):
489 489 self.shell.autocall = arg
490 490 else: # toggle
491 491 if self.shell.autocall:
492 492 self._magic_state.autocall_save = self.shell.autocall
493 493 self.shell.autocall = 0
494 494 else:
495 495 try:
496 496 self.shell.autocall = self._magic_state.autocall_save
497 497 except AttributeError:
498 498 self.shell.autocall = self._magic_state.autocall_save = 1
499 499
500 500 print "Automatic calling is:",['OFF','Smart','Full'][self.shell.autocall]
501 501
502 502
503 503 def magic_page(self, parameter_s=''):
504 504 """Pretty print the object and display it through a pager.
505 505
506 506 %page [options] OBJECT
507 507
508 508 If no object is given, use _ (last output).
509 509
510 510 Options:
511 511
512 512 -r: page str(object), don't pretty-print it."""
513 513
514 514 # After a function contributed by Olivier Aubert, slightly modified.
515 515
516 516 # Process options/args
517 517 opts,args = self.parse_options(parameter_s,'r')
518 518 raw = 'r' in opts
519 519
520 520 oname = args and args or '_'
521 521 info = self._ofind(oname)
522 522 if info['found']:
523 523 txt = (raw and str or pformat)( info['obj'] )
524 524 page.page(txt)
525 525 else:
526 526 print 'Object `%s` not found' % oname
527 527
528 528 def magic_profile(self, parameter_s=''):
529 529 """Print your currently active IPython profile."""
530 530 if self.shell.profile:
531 531 printpl('Current IPython profile: $self.shell.profile.')
532 532 else:
533 533 print 'No profile active.'
534 534
535 535 def magic_pinfo(self, parameter_s='', namespaces=None):
536 536 """Provide detailed information about an object.
537 537
538 538 '%pinfo object' is just a synonym for object? or ?object."""
539 539
540 540 #print 'pinfo par: <%s>' % parameter_s # dbg
541 541
542 542
543 543 # detail_level: 0 -> obj? , 1 -> obj??
544 544 detail_level = 0
545 545 # We need to detect if we got called as 'pinfo pinfo foo', which can
546 546 # happen if the user types 'pinfo foo?' at the cmd line.
547 547 pinfo,qmark1,oname,qmark2 = \
548 548 re.match('(pinfo )?(\?*)(.*?)(\??$)',parameter_s).groups()
549 549 if pinfo or qmark1 or qmark2:
550 550 detail_level = 1
551 551 if "*" in oname:
552 552 self.magic_psearch(oname)
553 553 else:
554 554 self.shell._inspect('pinfo', oname, detail_level=detail_level,
555 555 namespaces=namespaces)
556 556
557 557 def magic_pinfo2(self, parameter_s='', namespaces=None):
558 558 """Provide extra detailed information about an object.
559 559
560 560 '%pinfo2 object' is just a synonym for object?? or ??object."""
561 561 self.shell._inspect('pinfo', parameter_s, detail_level=1,
562 562 namespaces=namespaces)
563 563
564 564 @testdec.skip_doctest
565 565 def magic_pdef(self, parameter_s='', namespaces=None):
566 566 """Print the definition header for any callable object.
567 567
568 568 If the object is a class, print the constructor information.
569 569
570 570 Examples
571 571 --------
572 572 ::
573 573
574 574 In [3]: %pdef urllib.urlopen
575 575 urllib.urlopen(url, data=None, proxies=None)
576 576 """
577 577 self._inspect('pdef',parameter_s, namespaces)
578 578
579 579 def magic_pdoc(self, parameter_s='', namespaces=None):
580 580 """Print the docstring for an object.
581 581
582 582 If the given object is a class, it will print both the class and the
583 583 constructor docstrings."""
584 584 self._inspect('pdoc',parameter_s, namespaces)
585 585
586 586 def magic_psource(self, parameter_s='', namespaces=None):
587 587 """Print (or run through pager) the source code for an object."""
588 588 self._inspect('psource',parameter_s, namespaces)
589 589
590 590 def magic_pfile(self, parameter_s=''):
591 591 """Print (or run through pager) the file where an object is defined.
592 592
593 593 The file opens at the line where the object definition begins. IPython
594 594 will honor the environment variable PAGER if set, and otherwise will
595 595 do its best to print the file in a convenient form.
596 596
597 597 If the given argument is not an object currently defined, IPython will
598 598 try to interpret it as a filename (automatically adding a .py extension
599 599 if needed). You can thus use %pfile as a syntax highlighting code
600 600 viewer."""
601 601
602 602 # first interpret argument as an object name
603 603 out = self._inspect('pfile',parameter_s)
604 604 # if not, try the input as a filename
605 605 if out == 'not found':
606 606 try:
607 607 filename = get_py_filename(parameter_s)
608 608 except IOError,msg:
609 609 print msg
610 610 return
611 611 page.page(self.shell.inspector.format(file(filename).read()))
612 612
613 613 def magic_psearch(self, parameter_s=''):
614 614 """Search for object in namespaces by wildcard.
615 615
616 616 %psearch [options] PATTERN [OBJECT TYPE]
617 617
618 618 Note: ? can be used as a synonym for %psearch, at the beginning or at
619 619 the end: both a*? and ?a* are equivalent to '%psearch a*'. Still, the
620 620 rest of the command line must be unchanged (options come first), so
621 621 for example the following forms are equivalent
622 622
623 623 %psearch -i a* function
624 624 -i a* function?
625 625 ?-i a* function
626 626
627 627 Arguments:
628 628
629 629 PATTERN
630 630
631 631 where PATTERN is a string containing * as a wildcard similar to its
632 632 use in a shell. The pattern is matched in all namespaces on the
633 633 search path. By default objects starting with a single _ are not
634 634 matched, many IPython generated objects have a single
635 635 underscore. The default is case insensitive matching. Matching is
636 636 also done on the attributes of objects and not only on the objects
637 637 in a module.
638 638
639 639 [OBJECT TYPE]
640 640
641 641 Is the name of a python type from the types module. The name is
642 642 given in lowercase without the ending type, ex. StringType is
643 643 written string. By adding a type here only objects matching the
644 644 given type are matched. Using all here makes the pattern match all
645 645 types (this is the default).
646 646
647 647 Options:
648 648
649 649 -a: makes the pattern match even objects whose names start with a
650 650 single underscore. These names are normally ommitted from the
651 651 search.
652 652
653 653 -i/-c: make the pattern case insensitive/sensitive. If neither of
654 654 these options is given, the default is read from your ipythonrc
655 655 file. The option name which sets this value is
656 656 'wildcards_case_sensitive'. If this option is not specified in your
657 657 ipythonrc file, IPython's internal default is to do a case sensitive
658 658 search.
659 659
660 660 -e/-s NAMESPACE: exclude/search a given namespace. The pattern you
661 661 specifiy can be searched in any of the following namespaces:
662 662 'builtin', 'user', 'user_global','internal', 'alias', where
663 663 'builtin' and 'user' are the search defaults. Note that you should
664 664 not use quotes when specifying namespaces.
665 665
666 666 'Builtin' contains the python module builtin, 'user' contains all
667 667 user data, 'alias' only contain the shell aliases and no python
668 668 objects, 'internal' contains objects used by IPython. The
669 669 'user_global' namespace is only used by embedded IPython instances,
670 670 and it contains module-level globals. You can add namespaces to the
671 671 search with -s or exclude them with -e (these options can be given
672 672 more than once).
673 673
674 674 Examples:
675 675
676 676 %psearch a* -> objects beginning with an a
677 677 %psearch -e builtin a* -> objects NOT in the builtin space starting in a
678 678 %psearch a* function -> all functions beginning with an a
679 679 %psearch re.e* -> objects beginning with an e in module re
680 680 %psearch r*.e* -> objects that start with e in modules starting in r
681 681 %psearch r*.* string -> all strings in modules beginning with r
682 682
683 683 Case sensitve search:
684 684
685 685 %psearch -c a* list all object beginning with lower case a
686 686
687 687 Show objects beginning with a single _:
688 688
689 689 %psearch -a _* list objects beginning with a single underscore"""
690 690 try:
691 691 parameter_s = parameter_s.encode('ascii')
692 692 except UnicodeEncodeError:
693 693 print 'Python identifiers can only contain ascii characters.'
694 694 return
695 695
696 696 # default namespaces to be searched
697 697 def_search = ['user','builtin']
698 698
699 699 # Process options/args
700 700 opts,args = self.parse_options(parameter_s,'cias:e:',list_all=True)
701 701 opt = opts.get
702 702 shell = self.shell
703 703 psearch = shell.inspector.psearch
704 704
705 705 # select case options
706 706 if opts.has_key('i'):
707 707 ignore_case = True
708 708 elif opts.has_key('c'):
709 709 ignore_case = False
710 710 else:
711 711 ignore_case = not shell.wildcards_case_sensitive
712 712
713 713 # Build list of namespaces to search from user options
714 714 def_search.extend(opt('s',[]))
715 715 ns_exclude = ns_exclude=opt('e',[])
716 716 ns_search = [nm for nm in def_search if nm not in ns_exclude]
717 717
718 718 # Call the actual search
719 719 try:
720 720 psearch(args,shell.ns_table,ns_search,
721 721 show_all=opt('a'),ignore_case=ignore_case)
722 722 except:
723 723 shell.showtraceback()
724 724
725 725 @testdec.skip_doctest
726 726 def magic_who_ls(self, parameter_s=''):
727 727 """Return a sorted list of all interactive variables.
728 728
729 729 If arguments are given, only variables of types matching these
730 730 arguments are returned.
731 731
732 732 Examples
733 733 --------
734 734
735 735 Define two variables and list them with who_ls::
736 736
737 737 In [1]: alpha = 123
738 738
739 739 In [2]: beta = 'test'
740 740
741 741 In [3]: %who_ls
742 742 Out[3]: ['alpha', 'beta']
743 743
744 744 In [4]: %who_ls int
745 745 Out[4]: ['alpha']
746 746
747 747 In [5]: %who_ls str
748 748 Out[5]: ['beta']
749 749 """
750 750
751 751 user_ns = self.shell.user_ns
752 752 internal_ns = self.shell.internal_ns
753 753 user_ns_hidden = self.shell.user_ns_hidden
754 754 out = [ i for i in user_ns
755 755 if not i.startswith('_') \
756 756 and not (i in internal_ns or i in user_ns_hidden) ]
757 757
758 758 typelist = parameter_s.split()
759 759 if typelist:
760 760 typeset = set(typelist)
761 761 out = [i for i in out if type(user_ns[i]).__name__ in typeset]
762 762
763 763 out.sort()
764 764 return out
765 765
766 766 @testdec.skip_doctest
767 767 def magic_who(self, parameter_s=''):
768 768 """Print all interactive variables, with some minimal formatting.
769 769
770 770 If any arguments are given, only variables whose type matches one of
771 771 these are printed. For example:
772 772
773 773 %who function str
774 774
775 775 will only list functions and strings, excluding all other types of
776 776 variables. To find the proper type names, simply use type(var) at a
777 777 command line to see how python prints type names. For example:
778 778
779 779 In [1]: type('hello')\\
780 780 Out[1]: <type 'str'>
781 781
782 782 indicates that the type name for strings is 'str'.
783 783
784 784 %who always excludes executed names loaded through your configuration
785 785 file and things which are internal to IPython.
786 786
787 787 This is deliberate, as typically you may load many modules and the
788 788 purpose of %who is to show you only what you've manually defined.
789 789
790 790 Examples
791 791 --------
792 792
793 793 Define two variables and list them with who::
794 794
795 795 In [1]: alpha = 123
796 796
797 797 In [2]: beta = 'test'
798 798
799 799 In [3]: %who
800 800 alpha beta
801 801
802 802 In [4]: %who int
803 803 alpha
804 804
805 805 In [5]: %who str
806 806 beta
807 807 """
808 808
809 809 varlist = self.magic_who_ls(parameter_s)
810 810 if not varlist:
811 811 if parameter_s:
812 812 print 'No variables match your requested type.'
813 813 else:
814 814 print 'Interactive namespace is empty.'
815 815 return
816 816
817 817 # if we have variables, move on...
818 818 count = 0
819 819 for i in varlist:
820 820 print i+'\t',
821 821 count += 1
822 822 if count > 8:
823 823 count = 0
824 824 print
825 825 print
826 826
827 827 @testdec.skip_doctest
828 828 def magic_whos(self, parameter_s=''):
829 829 """Like %who, but gives some extra information about each variable.
830 830
831 831 The same type filtering of %who can be applied here.
832 832
833 833 For all variables, the type is printed. Additionally it prints:
834 834
835 835 - For {},[],(): their length.
836 836
837 837 - For numpy and Numeric arrays, a summary with shape, number of
838 838 elements, typecode and size in memory.
839 839
840 840 - Everything else: a string representation, snipping their middle if
841 841 too long.
842 842
843 843 Examples
844 844 --------
845 845
846 846 Define two variables and list them with whos::
847 847
848 848 In [1]: alpha = 123
849 849
850 850 In [2]: beta = 'test'
851 851
852 852 In [3]: %whos
853 853 Variable Type Data/Info
854 854 --------------------------------
855 855 alpha int 123
856 856 beta str test
857 857 """
858 858
859 859 varnames = self.magic_who_ls(parameter_s)
860 860 if not varnames:
861 861 if parameter_s:
862 862 print 'No variables match your requested type.'
863 863 else:
864 864 print 'Interactive namespace is empty.'
865 865 return
866 866
867 867 # if we have variables, move on...
868 868
869 869 # for these types, show len() instead of data:
870 870 seq_types = [types.DictType,types.ListType,types.TupleType]
871 871
872 872 # for numpy/Numeric arrays, display summary info
873 873 try:
874 874 import numpy
875 875 except ImportError:
876 876 ndarray_type = None
877 877 else:
878 878 ndarray_type = numpy.ndarray.__name__
879 879 try:
880 880 import Numeric
881 881 except ImportError:
882 882 array_type = None
883 883 else:
884 884 array_type = Numeric.ArrayType.__name__
885 885
886 886 # Find all variable names and types so we can figure out column sizes
887 887 def get_vars(i):
888 888 return self.shell.user_ns[i]
889 889
890 890 # some types are well known and can be shorter
891 891 abbrevs = {'IPython.core.macro.Macro' : 'Macro'}
892 892 def type_name(v):
893 893 tn = type(v).__name__
894 894 return abbrevs.get(tn,tn)
895 895
896 896 varlist = map(get_vars,varnames)
897 897
898 898 typelist = []
899 899 for vv in varlist:
900 900 tt = type_name(vv)
901 901
902 902 if tt=='instance':
903 903 typelist.append( abbrevs.get(str(vv.__class__),
904 904 str(vv.__class__)))
905 905 else:
906 906 typelist.append(tt)
907 907
908 908 # column labels and # of spaces as separator
909 909 varlabel = 'Variable'
910 910 typelabel = 'Type'
911 911 datalabel = 'Data/Info'
912 912 colsep = 3
913 913 # variable format strings
914 914 vformat = "$vname.ljust(varwidth)$vtype.ljust(typewidth)"
915 915 vfmt_short = '$vstr[:25]<...>$vstr[-25:]'
916 916 aformat = "%s: %s elems, type `%s`, %s bytes"
917 917 # find the size of the columns to format the output nicely
918 918 varwidth = max(max(map(len,varnames)), len(varlabel)) + colsep
919 919 typewidth = max(max(map(len,typelist)), len(typelabel)) + colsep
920 920 # table header
921 921 print varlabel.ljust(varwidth) + typelabel.ljust(typewidth) + \
922 922 ' '+datalabel+'\n' + '-'*(varwidth+typewidth+len(datalabel)+1)
923 923 # and the table itself
924 924 kb = 1024
925 925 Mb = 1048576 # kb**2
926 926 for vname,var,vtype in zip(varnames,varlist,typelist):
927 927 print itpl(vformat),
928 928 if vtype in seq_types:
929 929 print len(var)
930 930 elif vtype in [array_type,ndarray_type]:
931 931 vshape = str(var.shape).replace(',','').replace(' ','x')[1:-1]
932 932 if vtype==ndarray_type:
933 933 # numpy
934 934 vsize = var.size
935 935 vbytes = vsize*var.itemsize
936 936 vdtype = var.dtype
937 937 else:
938 938 # Numeric
939 939 vsize = Numeric.size(var)
940 940 vbytes = vsize*var.itemsize()
941 941 vdtype = var.typecode()
942 942
943 943 if vbytes < 100000:
944 944 print aformat % (vshape,vsize,vdtype,vbytes)
945 945 else:
946 946 print aformat % (vshape,vsize,vdtype,vbytes),
947 947 if vbytes < Mb:
948 948 print '(%s kb)' % (vbytes/kb,)
949 949 else:
950 950 print '(%s Mb)' % (vbytes/Mb,)
951 951 else:
952 952 try:
953 953 vstr = str(var)
954 954 except UnicodeEncodeError:
955 955 vstr = unicode(var).encode(sys.getdefaultencoding(),
956 956 'backslashreplace')
957 957 vstr = vstr.replace('\n','\\n')
958 958 if len(vstr) < 50:
959 959 print vstr
960 960 else:
961 961 printpl(vfmt_short)
962 962
963 963 def magic_reset(self, parameter_s=''):
964 964 """Resets the namespace by removing all names defined by the user.
965 965
966 966 Input/Output history are left around in case you need them.
967 967
968 968 Parameters
969 969 ----------
970 970 -y : force reset without asking for confirmation.
971 971
972 972 Examples
973 973 --------
974 974 In [6]: a = 1
975 975
976 976 In [7]: a
977 977 Out[7]: 1
978 978
979 979 In [8]: 'a' in _ip.user_ns
980 980 Out[8]: True
981 981
982 982 In [9]: %reset -f
983 983
984 984 In [10]: 'a' in _ip.user_ns
985 985 Out[10]: False
986 986 """
987 987
988 988 if parameter_s == '-f':
989 989 ans = True
990 990 else:
991 991 ans = self.shell.ask_yes_no(
992 992 "Once deleted, variables cannot be recovered. Proceed (y/[n])? ")
993 993 if not ans:
994 994 print 'Nothing done.'
995 995 return
996 996 user_ns = self.shell.user_ns
997 997 for i in self.magic_who_ls():
998 998 del(user_ns[i])
999 999
1000 1000 # Also flush the private list of module references kept for script
1001 1001 # execution protection
1002 1002 self.shell.clear_main_mod_cache()
1003 1003
1004 1004 def magic_reset_selective(self, parameter_s=''):
1005 1005 """Resets the namespace by removing names defined by the user.
1006 1006
1007 1007 Input/Output history are left around in case you need them.
1008 1008
1009 1009 %reset_selective [-f] regex
1010 1010
1011 1011 No action is taken if regex is not included
1012 1012
1013 1013 Options
1014 1014 -f : force reset without asking for confirmation.
1015 1015
1016 1016 Examples
1017 1017 --------
1018 1018
1019 1019 We first fully reset the namespace so your output looks identical to
1020 1020 this example for pedagogical reasons; in practice you do not need a
1021 1021 full reset.
1022 1022
1023 1023 In [1]: %reset -f
1024 1024
1025 1025 Now, with a clean namespace we can make a few variables and use
1026 1026 %reset_selective to only delete names that match our regexp:
1027 1027
1028 1028 In [2]: a=1; b=2; c=3; b1m=4; b2m=5; b3m=6; b4m=7; b2s=8
1029 1029
1030 1030 In [3]: who_ls
1031 1031 Out[3]: ['a', 'b', 'b1m', 'b2m', 'b2s', 'b3m', 'b4m', 'c']
1032 1032
1033 1033 In [4]: %reset_selective -f b[2-3]m
1034 1034
1035 1035 In [5]: who_ls
1036 1036 Out[5]: ['a', 'b', 'b1m', 'b2s', 'b4m', 'c']
1037 1037
1038 1038 In [6]: %reset_selective -f d
1039 1039
1040 1040 In [7]: who_ls
1041 1041 Out[7]: ['a', 'b', 'b1m', 'b2s', 'b4m', 'c']
1042 1042
1043 1043 In [8]: %reset_selective -f c
1044 1044
1045 1045 In [9]: who_ls
1046 1046 Out[9]: ['a', 'b', 'b1m', 'b2s', 'b4m']
1047 1047
1048 1048 In [10]: %reset_selective -f b
1049 1049
1050 1050 In [11]: who_ls
1051 1051 Out[11]: ['a']
1052 1052 """
1053 1053
1054 1054 opts, regex = self.parse_options(parameter_s,'f')
1055 1055
1056 1056 if opts.has_key('f'):
1057 1057 ans = True
1058 1058 else:
1059 1059 ans = self.shell.ask_yes_no(
1060 1060 "Once deleted, variables cannot be recovered. Proceed (y/[n])? ")
1061 1061 if not ans:
1062 1062 print 'Nothing done.'
1063 1063 return
1064 1064 user_ns = self.shell.user_ns
1065 1065 if not regex:
1066 1066 print 'No regex pattern specified. Nothing done.'
1067 1067 return
1068 1068 else:
1069 1069 try:
1070 1070 m = re.compile(regex)
1071 1071 except TypeError:
1072 1072 raise TypeError('regex must be a string or compiled pattern')
1073 1073 for i in self.magic_who_ls():
1074 1074 if m.search(i):
1075 1075 del(user_ns[i])
1076 1076
1077 1077 def magic_logstart(self,parameter_s=''):
1078 1078 """Start logging anywhere in a session.
1079 1079
1080 1080 %logstart [-o|-r|-t] [log_name [log_mode]]
1081 1081
1082 1082 If no name is given, it defaults to a file named 'ipython_log.py' in your
1083 1083 current directory, in 'rotate' mode (see below).
1084 1084
1085 1085 '%logstart name' saves to file 'name' in 'backup' mode. It saves your
1086 1086 history up to that point and then continues logging.
1087 1087
1088 1088 %logstart takes a second optional parameter: logging mode. This can be one
1089 1089 of (note that the modes are given unquoted):\\
1090 1090 append: well, that says it.\\
1091 1091 backup: rename (if exists) to name~ and start name.\\
1092 1092 global: single logfile in your home dir, appended to.\\
1093 1093 over : overwrite existing log.\\
1094 1094 rotate: create rotating logs name.1~, name.2~, etc.
1095 1095
1096 1096 Options:
1097 1097
1098 1098 -o: log also IPython's output. In this mode, all commands which
1099 1099 generate an Out[NN] prompt are recorded to the logfile, right after
1100 1100 their corresponding input line. The output lines are always
1101 1101 prepended with a '#[Out]# ' marker, so that the log remains valid
1102 1102 Python code.
1103 1103
1104 1104 Since this marker is always the same, filtering only the output from
1105 1105 a log is very easy, using for example a simple awk call:
1106 1106
1107 1107 awk -F'#\\[Out\\]# ' '{if($2) {print $2}}' ipython_log.py
1108 1108
1109 1109 -r: log 'raw' input. Normally, IPython's logs contain the processed
1110 1110 input, so that user lines are logged in their final form, converted
1111 1111 into valid Python. For example, %Exit is logged as
1112 1112 '_ip.magic("Exit"). If the -r flag is given, all input is logged
1113 1113 exactly as typed, with no transformations applied.
1114 1114
1115 1115 -t: put timestamps before each input line logged (these are put in
1116 1116 comments)."""
1117 1117
1118 1118 opts,par = self.parse_options(parameter_s,'ort')
1119 1119 log_output = 'o' in opts
1120 1120 log_raw_input = 'r' in opts
1121 1121 timestamp = 't' in opts
1122 1122
1123 1123 logger = self.shell.logger
1124 1124
1125 1125 # if no args are given, the defaults set in the logger constructor by
1126 1126 # ipytohn remain valid
1127 1127 if par:
1128 1128 try:
1129 1129 logfname,logmode = par.split()
1130 1130 except:
1131 1131 logfname = par
1132 1132 logmode = 'backup'
1133 1133 else:
1134 1134 logfname = logger.logfname
1135 1135 logmode = logger.logmode
1136 1136 # put logfname into rc struct as if it had been called on the command
1137 1137 # line, so it ends up saved in the log header Save it in case we need
1138 1138 # to restore it...
1139 1139 old_logfile = self.shell.logfile
1140 1140 if logfname:
1141 1141 logfname = os.path.expanduser(logfname)
1142 1142 self.shell.logfile = logfname
1143 1143
1144 1144 loghead = '# IPython log file\n\n'
1145 1145 try:
1146 1146 started = logger.logstart(logfname,loghead,logmode,
1147 1147 log_output,timestamp,log_raw_input)
1148 1148 except:
1149 1149 self.shell.logfile = old_logfile
1150 1150 warn("Couldn't start log: %s" % sys.exc_info()[1])
1151 1151 else:
1152 1152 # log input history up to this point, optionally interleaving
1153 1153 # output if requested
1154 1154
1155 1155 if timestamp:
1156 1156 # disable timestamping for the previous history, since we've
1157 1157 # lost those already (no time machine here).
1158 1158 logger.timestamp = False
1159 1159
1160 1160 if log_raw_input:
1161 1161 input_hist = self.shell.history_manager.input_hist_raw
1162 1162 else:
1163 1163 input_hist = self.shell.history_manager.input_hist_parsed
1164 1164
1165 1165 if log_output:
1166 1166 log_write = logger.log_write
1167 1167 output_hist = self.shell.history_manager.output_hist
1168 1168 for n in range(1,len(input_hist)-1):
1169 1169 log_write(input_hist[n].rstrip())
1170 1170 if n in output_hist:
1171 1171 log_write(repr(output_hist[n]),'output')
1172 1172 else:
1173 1173 logger.log_write(''.join(input_hist[1:]))
1174 1174 if timestamp:
1175 1175 # re-enable timestamping
1176 1176 logger.timestamp = True
1177 1177
1178 1178 print ('Activating auto-logging. '
1179 1179 'Current session state plus future input saved.')
1180 1180 logger.logstate()
1181 1181
1182 1182 def magic_logstop(self,parameter_s=''):
1183 1183 """Fully stop logging and close log file.
1184 1184
1185 1185 In order to start logging again, a new %logstart call needs to be made,
1186 1186 possibly (though not necessarily) with a new filename, mode and other
1187 1187 options."""
1188 1188 self.logger.logstop()
1189 1189
1190 1190 def magic_logoff(self,parameter_s=''):
1191 1191 """Temporarily stop logging.
1192 1192
1193 1193 You must have previously started logging."""
1194 1194 self.shell.logger.switch_log(0)
1195 1195
1196 1196 def magic_logon(self,parameter_s=''):
1197 1197 """Restart logging.
1198 1198
1199 1199 This function is for restarting logging which you've temporarily
1200 1200 stopped with %logoff. For starting logging for the first time, you
1201 1201 must use the %logstart function, which allows you to specify an
1202 1202 optional log filename."""
1203 1203
1204 1204 self.shell.logger.switch_log(1)
1205 1205
1206 1206 def magic_logstate(self,parameter_s=''):
1207 1207 """Print the status of the logging system."""
1208 1208
1209 1209 self.shell.logger.logstate()
1210 1210
1211 1211 def magic_pdb(self, parameter_s=''):
1212 1212 """Control the automatic calling of the pdb interactive debugger.
1213 1213
1214 1214 Call as '%pdb on', '%pdb 1', '%pdb off' or '%pdb 0'. If called without
1215 1215 argument it works as a toggle.
1216 1216
1217 1217 When an exception is triggered, IPython can optionally call the
1218 1218 interactive pdb debugger after the traceback printout. %pdb toggles
1219 1219 this feature on and off.
1220 1220
1221 1221 The initial state of this feature is set in your ipythonrc
1222 1222 configuration file (the variable is called 'pdb').
1223 1223
1224 1224 If you want to just activate the debugger AFTER an exception has fired,
1225 1225 without having to type '%pdb on' and rerunning your code, you can use
1226 1226 the %debug magic."""
1227 1227
1228 1228 par = parameter_s.strip().lower()
1229 1229
1230 1230 if par:
1231 1231 try:
1232 1232 new_pdb = {'off':0,'0':0,'on':1,'1':1}[par]
1233 1233 except KeyError:
1234 1234 print ('Incorrect argument. Use on/1, off/0, '
1235 1235 'or nothing for a toggle.')
1236 1236 return
1237 1237 else:
1238 1238 # toggle
1239 1239 new_pdb = not self.shell.call_pdb
1240 1240
1241 1241 # set on the shell
1242 1242 self.shell.call_pdb = new_pdb
1243 1243 print 'Automatic pdb calling has been turned',on_off(new_pdb)
1244 1244
1245 1245 def magic_debug(self, parameter_s=''):
1246 1246 """Activate the interactive debugger in post-mortem mode.
1247 1247
1248 1248 If an exception has just occurred, this lets you inspect its stack
1249 1249 frames interactively. Note that this will always work only on the last
1250 1250 traceback that occurred, so you must call this quickly after an
1251 1251 exception that you wish to inspect has fired, because if another one
1252 1252 occurs, it clobbers the previous one.
1253 1253
1254 1254 If you want IPython to automatically do this on every exception, see
1255 1255 the %pdb magic for more details.
1256 1256 """
1257 1257 self.shell.debugger(force=True)
1258 1258
1259 1259 @testdec.skip_doctest
1260 1260 def magic_prun(self, parameter_s ='',user_mode=1,
1261 1261 opts=None,arg_lst=None,prog_ns=None):
1262 1262
1263 1263 """Run a statement through the python code profiler.
1264 1264
1265 1265 Usage:
1266 1266 %prun [options] statement
1267 1267
1268 1268 The given statement (which doesn't require quote marks) is run via the
1269 1269 python profiler in a manner similar to the profile.run() function.
1270 1270 Namespaces are internally managed to work correctly; profile.run
1271 1271 cannot be used in IPython because it makes certain assumptions about
1272 1272 namespaces which do not hold under IPython.
1273 1273
1274 1274 Options:
1275 1275
1276 1276 -l <limit>: you can place restrictions on what or how much of the
1277 1277 profile gets printed. The limit value can be:
1278 1278
1279 1279 * A string: only information for function names containing this string
1280 1280 is printed.
1281 1281
1282 1282 * An integer: only these many lines are printed.
1283 1283
1284 1284 * A float (between 0 and 1): this fraction of the report is printed
1285 1285 (for example, use a limit of 0.4 to see the topmost 40% only).
1286 1286
1287 1287 You can combine several limits with repeated use of the option. For
1288 1288 example, '-l __init__ -l 5' will print only the topmost 5 lines of
1289 1289 information about class constructors.
1290 1290
1291 1291 -r: return the pstats.Stats object generated by the profiling. This
1292 1292 object has all the information about the profile in it, and you can
1293 1293 later use it for further analysis or in other functions.
1294 1294
1295 1295 -s <key>: sort profile by given key. You can provide more than one key
1296 1296 by using the option several times: '-s key1 -s key2 -s key3...'. The
1297 1297 default sorting key is 'time'.
1298 1298
1299 1299 The following is copied verbatim from the profile documentation
1300 1300 referenced below:
1301 1301
1302 1302 When more than one key is provided, additional keys are used as
1303 1303 secondary criteria when the there is equality in all keys selected
1304 1304 before them.
1305 1305
1306 1306 Abbreviations can be used for any key names, as long as the
1307 1307 abbreviation is unambiguous. The following are the keys currently
1308 1308 defined:
1309 1309
1310 1310 Valid Arg Meaning
1311 1311 "calls" call count
1312 1312 "cumulative" cumulative time
1313 1313 "file" file name
1314 1314 "module" file name
1315 1315 "pcalls" primitive call count
1316 1316 "line" line number
1317 1317 "name" function name
1318 1318 "nfl" name/file/line
1319 1319 "stdname" standard name
1320 1320 "time" internal time
1321 1321
1322 1322 Note that all sorts on statistics are in descending order (placing
1323 1323 most time consuming items first), where as name, file, and line number
1324 1324 searches are in ascending order (i.e., alphabetical). The subtle
1325 1325 distinction between "nfl" and "stdname" is that the standard name is a
1326 1326 sort of the name as printed, which means that the embedded line
1327 1327 numbers get compared in an odd way. For example, lines 3, 20, and 40
1328 1328 would (if the file names were the same) appear in the string order
1329 1329 "20" "3" and "40". In contrast, "nfl" does a numeric compare of the
1330 1330 line numbers. In fact, sort_stats("nfl") is the same as
1331 1331 sort_stats("name", "file", "line").
1332 1332
1333 1333 -T <filename>: save profile results as shown on screen to a text
1334 1334 file. The profile is still shown on screen.
1335 1335
1336 1336 -D <filename>: save (via dump_stats) profile statistics to given
1337 1337 filename. This data is in a format understod by the pstats module, and
1338 1338 is generated by a call to the dump_stats() method of profile
1339 1339 objects. The profile is still shown on screen.
1340 1340
1341 1341 If you want to run complete programs under the profiler's control, use
1342 1342 '%run -p [prof_opts] filename.py [args to program]' where prof_opts
1343 1343 contains profiler specific options as described here.
1344 1344
1345 1345 You can read the complete documentation for the profile module with::
1346 1346
1347 1347 In [1]: import profile; profile.help()
1348 1348 """
1349 1349
1350 1350 opts_def = Struct(D=[''],l=[],s=['time'],T=[''])
1351 1351 # protect user quote marks
1352 1352 parameter_s = parameter_s.replace('"',r'\"').replace("'",r"\'")
1353 1353
1354 1354 if user_mode: # regular user call
1355 1355 opts,arg_str = self.parse_options(parameter_s,'D:l:rs:T:',
1356 1356 list_all=1)
1357 1357 namespace = self.shell.user_ns
1358 1358 else: # called to run a program by %run -p
1359 1359 try:
1360 1360 filename = get_py_filename(arg_lst[0])
1361 1361 except IOError,msg:
1362 1362 error(msg)
1363 1363 return
1364 1364
1365 1365 arg_str = 'execfile(filename,prog_ns)'
1366 1366 namespace = locals()
1367 1367
1368 1368 opts.merge(opts_def)
1369 1369
1370 1370 prof = profile.Profile()
1371 1371 try:
1372 1372 prof = prof.runctx(arg_str,namespace,namespace)
1373 1373 sys_exit = ''
1374 1374 except SystemExit:
1375 1375 sys_exit = """*** SystemExit exception caught in code being profiled."""
1376 1376
1377 1377 stats = pstats.Stats(prof).strip_dirs().sort_stats(*opts.s)
1378 1378
1379 1379 lims = opts.l
1380 1380 if lims:
1381 1381 lims = [] # rebuild lims with ints/floats/strings
1382 1382 for lim in opts.l:
1383 1383 try:
1384 1384 lims.append(int(lim))
1385 1385 except ValueError:
1386 1386 try:
1387 1387 lims.append(float(lim))
1388 1388 except ValueError:
1389 1389 lims.append(lim)
1390 1390
1391 1391 # Trap output.
1392 1392 stdout_trap = StringIO()
1393 1393
1394 1394 if hasattr(stats,'stream'):
1395 1395 # In newer versions of python, the stats object has a 'stream'
1396 1396 # attribute to write into.
1397 1397 stats.stream = stdout_trap
1398 1398 stats.print_stats(*lims)
1399 1399 else:
1400 1400 # For older versions, we manually redirect stdout during printing
1401 1401 sys_stdout = sys.stdout
1402 1402 try:
1403 1403 sys.stdout = stdout_trap
1404 1404 stats.print_stats(*lims)
1405 1405 finally:
1406 1406 sys.stdout = sys_stdout
1407 1407
1408 1408 output = stdout_trap.getvalue()
1409 1409 output = output.rstrip()
1410 1410
1411 1411 page.page(output)
1412 1412 print sys_exit,
1413 1413
1414 1414 dump_file = opts.D[0]
1415 1415 text_file = opts.T[0]
1416 1416 if dump_file:
1417 1417 prof.dump_stats(dump_file)
1418 1418 print '\n*** Profile stats marshalled to file',\
1419 1419 `dump_file`+'.',sys_exit
1420 1420 if text_file:
1421 1421 pfile = file(text_file,'w')
1422 1422 pfile.write(output)
1423 1423 pfile.close()
1424 1424 print '\n*** Profile printout saved to text file',\
1425 1425 `text_file`+'.',sys_exit
1426 1426
1427 1427 if opts.has_key('r'):
1428 1428 return stats
1429 1429 else:
1430 1430 return None
1431 1431
1432 1432 @testdec.skip_doctest
1433 1433 def magic_run(self, parameter_s ='',runner=None,
1434 1434 file_finder=get_py_filename):
1435 1435 """Run the named file inside IPython as a program.
1436 1436
1437 1437 Usage:\\
1438 1438 %run [-n -i -t [-N<N>] -d [-b<N>] -p [profile options]] file [args]
1439 1439
1440 1440 Parameters after the filename are passed as command-line arguments to
1441 1441 the program (put in sys.argv). Then, control returns to IPython's
1442 1442 prompt.
1443 1443
1444 1444 This is similar to running at a system prompt:\\
1445 1445 $ python file args\\
1446 1446 but with the advantage of giving you IPython's tracebacks, and of
1447 1447 loading all variables into your interactive namespace for further use
1448 1448 (unless -p is used, see below).
1449 1449
1450 1450 The file is executed in a namespace initially consisting only of
1451 1451 __name__=='__main__' and sys.argv constructed as indicated. It thus
1452 1452 sees its environment as if it were being run as a stand-alone program
1453 1453 (except for sharing global objects such as previously imported
1454 1454 modules). But after execution, the IPython interactive namespace gets
1455 1455 updated with all variables defined in the program (except for __name__
1456 1456 and sys.argv). This allows for very convenient loading of code for
1457 1457 interactive work, while giving each program a 'clean sheet' to run in.
1458 1458
1459 1459 Options:
1460 1460
1461 1461 -n: __name__ is NOT set to '__main__', but to the running file's name
1462 1462 without extension (as python does under import). This allows running
1463 1463 scripts and reloading the definitions in them without calling code
1464 1464 protected by an ' if __name__ == "__main__" ' clause.
1465 1465
1466 1466 -i: run the file in IPython's namespace instead of an empty one. This
1467 1467 is useful if you are experimenting with code written in a text editor
1468 1468 which depends on variables defined interactively.
1469 1469
1470 1470 -e: ignore sys.exit() calls or SystemExit exceptions in the script
1471 1471 being run. This is particularly useful if IPython is being used to
1472 1472 run unittests, which always exit with a sys.exit() call. In such
1473 1473 cases you are interested in the output of the test results, not in
1474 1474 seeing a traceback of the unittest module.
1475 1475
1476 1476 -t: print timing information at the end of the run. IPython will give
1477 1477 you an estimated CPU time consumption for your script, which under
1478 1478 Unix uses the resource module to avoid the wraparound problems of
1479 1479 time.clock(). Under Unix, an estimate of time spent on system tasks
1480 1480 is also given (for Windows platforms this is reported as 0.0).
1481 1481
1482 1482 If -t is given, an additional -N<N> option can be given, where <N>
1483 1483 must be an integer indicating how many times you want the script to
1484 1484 run. The final timing report will include total and per run results.
1485 1485
1486 1486 For example (testing the script uniq_stable.py):
1487 1487
1488 1488 In [1]: run -t uniq_stable
1489 1489
1490 1490 IPython CPU timings (estimated):\\
1491 1491 User : 0.19597 s.\\
1492 1492 System: 0.0 s.\\
1493 1493
1494 1494 In [2]: run -t -N5 uniq_stable
1495 1495
1496 1496 IPython CPU timings (estimated):\\
1497 1497 Total runs performed: 5\\
1498 1498 Times : Total Per run\\
1499 1499 User : 0.910862 s, 0.1821724 s.\\
1500 1500 System: 0.0 s, 0.0 s.
1501 1501
1502 1502 -d: run your program under the control of pdb, the Python debugger.
1503 1503 This allows you to execute your program step by step, watch variables,
1504 1504 etc. Internally, what IPython does is similar to calling:
1505 1505
1506 1506 pdb.run('execfile("YOURFILENAME")')
1507 1507
1508 1508 with a breakpoint set on line 1 of your file. You can change the line
1509 1509 number for this automatic breakpoint to be <N> by using the -bN option
1510 1510 (where N must be an integer). For example:
1511 1511
1512 1512 %run -d -b40 myscript
1513 1513
1514 1514 will set the first breakpoint at line 40 in myscript.py. Note that
1515 1515 the first breakpoint must be set on a line which actually does
1516 1516 something (not a comment or docstring) for it to stop execution.
1517 1517
1518 1518 When the pdb debugger starts, you will see a (Pdb) prompt. You must
1519 1519 first enter 'c' (without qoutes) to start execution up to the first
1520 1520 breakpoint.
1521 1521
1522 1522 Entering 'help' gives information about the use of the debugger. You
1523 1523 can easily see pdb's full documentation with "import pdb;pdb.help()"
1524 1524 at a prompt.
1525 1525
1526 1526 -p: run program under the control of the Python profiler module (which
1527 1527 prints a detailed report of execution times, function calls, etc).
1528 1528
1529 1529 You can pass other options after -p which affect the behavior of the
1530 1530 profiler itself. See the docs for %prun for details.
1531 1531
1532 1532 In this mode, the program's variables do NOT propagate back to the
1533 1533 IPython interactive namespace (because they remain in the namespace
1534 1534 where the profiler executes them).
1535 1535
1536 1536 Internally this triggers a call to %prun, see its documentation for
1537 1537 details on the options available specifically for profiling.
1538 1538
1539 1539 There is one special usage for which the text above doesn't apply:
1540 1540 if the filename ends with .ipy, the file is run as ipython script,
1541 1541 just as if the commands were written on IPython prompt.
1542 1542 """
1543 1543
1544 1544 # get arguments and set sys.argv for program to be run.
1545 1545 opts,arg_lst = self.parse_options(parameter_s,'nidtN:b:pD:l:rs:T:e',
1546 1546 mode='list',list_all=1)
1547 1547
1548 1548 try:
1549 1549 filename = file_finder(arg_lst[0])
1550 1550 except IndexError:
1551 1551 warn('you must provide at least a filename.')
1552 1552 print '\n%run:\n',oinspect.getdoc(self.magic_run)
1553 1553 return
1554 1554 except IOError,msg:
1555 1555 error(msg)
1556 1556 return
1557 1557
1558 1558 if filename.lower().endswith('.ipy'):
1559 1559 self.shell.safe_execfile_ipy(filename)
1560 1560 return
1561 1561
1562 1562 # Control the response to exit() calls made by the script being run
1563 1563 exit_ignore = opts.has_key('e')
1564 1564
1565 1565 # Make sure that the running script gets a proper sys.argv as if it
1566 1566 # were run from a system shell.
1567 1567 save_argv = sys.argv # save it for later restoring
1568 1568 sys.argv = [filename]+ arg_lst[1:] # put in the proper filename
1569 1569
1570 1570 if opts.has_key('i'):
1571 1571 # Run in user's interactive namespace
1572 1572 prog_ns = self.shell.user_ns
1573 1573 __name__save = self.shell.user_ns['__name__']
1574 1574 prog_ns['__name__'] = '__main__'
1575 1575 main_mod = self.shell.new_main_mod(prog_ns)
1576 1576 else:
1577 1577 # Run in a fresh, empty namespace
1578 1578 if opts.has_key('n'):
1579 1579 name = os.path.splitext(os.path.basename(filename))[0]
1580 1580 else:
1581 1581 name = '__main__'
1582 1582
1583 1583 main_mod = self.shell.new_main_mod()
1584 1584 prog_ns = main_mod.__dict__
1585 1585 prog_ns['__name__'] = name
1586 1586
1587 1587 # Since '%run foo' emulates 'python foo.py' at the cmd line, we must
1588 1588 # set the __file__ global in the script's namespace
1589 1589 prog_ns['__file__'] = filename
1590 1590
1591 1591 # pickle fix. See interactiveshell for an explanation. But we need to make sure
1592 1592 # that, if we overwrite __main__, we replace it at the end
1593 1593 main_mod_name = prog_ns['__name__']
1594 1594
1595 1595 if main_mod_name == '__main__':
1596 1596 restore_main = sys.modules['__main__']
1597 1597 else:
1598 1598 restore_main = False
1599 1599
1600 1600 # This needs to be undone at the end to prevent holding references to
1601 1601 # every single object ever created.
1602 1602 sys.modules[main_mod_name] = main_mod
1603 1603
1604 1604 stats = None
1605 1605 try:
1606 1606 #self.shell.save_history()
1607 1607
1608 1608 if opts.has_key('p'):
1609 1609 stats = self.magic_prun('',0,opts,arg_lst,prog_ns)
1610 1610 else:
1611 1611 if opts.has_key('d'):
1612 1612 deb = debugger.Pdb(self.shell.colors)
1613 1613 # reset Breakpoint state, which is moronically kept
1614 1614 # in a class
1615 1615 bdb.Breakpoint.next = 1
1616 1616 bdb.Breakpoint.bplist = {}
1617 1617 bdb.Breakpoint.bpbynumber = [None]
1618 1618 # Set an initial breakpoint to stop execution
1619 1619 maxtries = 10
1620 1620 bp = int(opts.get('b',[1])[0])
1621 1621 checkline = deb.checkline(filename,bp)
1622 1622 if not checkline:
1623 1623 for bp in range(bp+1,bp+maxtries+1):
1624 1624 if deb.checkline(filename,bp):
1625 1625 break
1626 1626 else:
1627 1627 msg = ("\nI failed to find a valid line to set "
1628 1628 "a breakpoint\n"
1629 1629 "after trying up to line: %s.\n"
1630 1630 "Please set a valid breakpoint manually "
1631 1631 "with the -b option." % bp)
1632 1632 error(msg)
1633 1633 return
1634 1634 # if we find a good linenumber, set the breakpoint
1635 1635 deb.do_break('%s:%s' % (filename,bp))
1636 1636 # Start file run
1637 1637 print "NOTE: Enter 'c' at the",
1638 1638 print "%s prompt to start your script." % deb.prompt
1639 1639 try:
1640 1640 deb.run('execfile("%s")' % filename,prog_ns)
1641 1641
1642 1642 except:
1643 1643 etype, value, tb = sys.exc_info()
1644 1644 # Skip three frames in the traceback: the %run one,
1645 1645 # one inside bdb.py, and the command-line typed by the
1646 1646 # user (run by exec in pdb itself).
1647 1647 self.shell.InteractiveTB(etype,value,tb,tb_offset=3)
1648 1648 else:
1649 1649 if runner is None:
1650 1650 runner = self.shell.safe_execfile
1651 1651 if opts.has_key('t'):
1652 1652 # timed execution
1653 1653 try:
1654 1654 nruns = int(opts['N'][0])
1655 1655 if nruns < 1:
1656 1656 error('Number of runs must be >=1')
1657 1657 return
1658 1658 except (KeyError):
1659 1659 nruns = 1
1660 1660 if nruns == 1:
1661 1661 t0 = clock2()
1662 1662 runner(filename,prog_ns,prog_ns,
1663 1663 exit_ignore=exit_ignore)
1664 1664 t1 = clock2()
1665 1665 t_usr = t1[0]-t0[0]
1666 1666 t_sys = t1[1]-t0[1]
1667 1667 print "\nIPython CPU timings (estimated):"
1668 1668 print " User : %10s s." % t_usr
1669 1669 print " System: %10s s." % t_sys
1670 1670 else:
1671 1671 runs = range(nruns)
1672 1672 t0 = clock2()
1673 1673 for nr in runs:
1674 1674 runner(filename,prog_ns,prog_ns,
1675 1675 exit_ignore=exit_ignore)
1676 1676 t1 = clock2()
1677 1677 t_usr = t1[0]-t0[0]
1678 1678 t_sys = t1[1]-t0[1]
1679 1679 print "\nIPython CPU timings (estimated):"
1680 1680 print "Total runs performed:",nruns
1681 1681 print " Times : %10s %10s" % ('Total','Per run')
1682 1682 print " User : %10s s, %10s s." % (t_usr,t_usr/nruns)
1683 1683 print " System: %10s s, %10s s." % (t_sys,t_sys/nruns)
1684 1684
1685 1685 else:
1686 1686 # regular execution
1687 1687 runner(filename,prog_ns,prog_ns,exit_ignore=exit_ignore)
1688 1688
1689 1689 if opts.has_key('i'):
1690 1690 self.shell.user_ns['__name__'] = __name__save
1691 1691 else:
1692 1692 # The shell MUST hold a reference to prog_ns so after %run
1693 1693 # exits, the python deletion mechanism doesn't zero it out
1694 1694 # (leaving dangling references).
1695 1695 self.shell.cache_main_mod(prog_ns,filename)
1696 1696 # update IPython interactive namespace
1697 1697
1698 1698 # Some forms of read errors on the file may mean the
1699 1699 # __name__ key was never set; using pop we don't have to
1700 1700 # worry about a possible KeyError.
1701 1701 prog_ns.pop('__name__', None)
1702 1702
1703 1703 self.shell.user_ns.update(prog_ns)
1704 1704 finally:
1705 1705 # It's a bit of a mystery why, but __builtins__ can change from
1706 1706 # being a module to becoming a dict missing some key data after
1707 1707 # %run. As best I can see, this is NOT something IPython is doing
1708 1708 # at all, and similar problems have been reported before:
1709 1709 # http://coding.derkeiler.com/Archive/Python/comp.lang.python/2004-10/0188.html
1710 1710 # Since this seems to be done by the interpreter itself, the best
1711 1711 # we can do is to at least restore __builtins__ for the user on
1712 1712 # exit.
1713 1713 self.shell.user_ns['__builtins__'] = __builtin__
1714 1714
1715 1715 # Ensure key global structures are restored
1716 1716 sys.argv = save_argv
1717 1717 if restore_main:
1718 1718 sys.modules['__main__'] = restore_main
1719 1719 else:
1720 1720 # Remove from sys.modules the reference to main_mod we'd
1721 1721 # added. Otherwise it will trap references to objects
1722 1722 # contained therein.
1723 1723 del sys.modules[main_mod_name]
1724 1724
1725 1725 #self.shell.reload_history()
1726 1726
1727 1727 return stats
1728 1728
1729 1729 @testdec.skip_doctest
1730 1730 def magic_timeit(self, parameter_s =''):
1731 1731 """Time execution of a Python statement or expression
1732 1732
1733 1733 Usage:\\
1734 1734 %timeit [-n<N> -r<R> [-t|-c]] statement
1735 1735
1736 1736 Time execution of a Python statement or expression using the timeit
1737 1737 module.
1738 1738
1739 1739 Options:
1740 1740 -n<N>: execute the given statement <N> times in a loop. If this value
1741 1741 is not given, a fitting value is chosen.
1742 1742
1743 1743 -r<R>: repeat the loop iteration <R> times and take the best result.
1744 1744 Default: 3
1745 1745
1746 1746 -t: use time.time to measure the time, which is the default on Unix.
1747 1747 This function measures wall time.
1748 1748
1749 1749 -c: use time.clock to measure the time, which is the default on
1750 1750 Windows and measures wall time. On Unix, resource.getrusage is used
1751 1751 instead and returns the CPU user time.
1752 1752
1753 1753 -p<P>: use a precision of <P> digits to display the timing result.
1754 1754 Default: 3
1755 1755
1756 1756
1757 1757 Examples:
1758 1758
1759 1759 In [1]: %timeit pass
1760 1760 10000000 loops, best of 3: 53.3 ns per loop
1761 1761
1762 1762 In [2]: u = None
1763 1763
1764 1764 In [3]: %timeit u is None
1765 1765 10000000 loops, best of 3: 184 ns per loop
1766 1766
1767 1767 In [4]: %timeit -r 4 u == None
1768 1768 1000000 loops, best of 4: 242 ns per loop
1769 1769
1770 1770 In [5]: import time
1771 1771
1772 1772 In [6]: %timeit -n1 time.sleep(2)
1773 1773 1 loops, best of 3: 2 s per loop
1774 1774
1775 1775
1776 1776 The times reported by %timeit will be slightly higher than those
1777 1777 reported by the timeit.py script when variables are accessed. This is
1778 1778 due to the fact that %timeit executes the statement in the namespace
1779 1779 of the shell, compared with timeit.py, which uses a single setup
1780 1780 statement to import function or create variables. Generally, the bias
1781 1781 does not matter as long as results from timeit.py are not mixed with
1782 1782 those from %timeit."""
1783 1783
1784 1784 import timeit
1785 1785 import math
1786 1786
1787 1787 # XXX: Unfortunately the unicode 'micro' symbol can cause problems in
1788 1788 # certain terminals. Until we figure out a robust way of
1789 1789 # auto-detecting if the terminal can deal with it, use plain 'us' for
1790 1790 # microseconds. I am really NOT happy about disabling the proper
1791 1791 # 'micro' prefix, but crashing is worse... If anyone knows what the
1792 1792 # right solution for this is, I'm all ears...
1793 1793 #
1794 1794 # Note: using
1795 1795 #
1796 1796 # s = u'\xb5'
1797 1797 # s.encode(sys.getdefaultencoding())
1798 1798 #
1799 1799 # is not sufficient, as I've seen terminals where that fails but
1800 1800 # print s
1801 1801 #
1802 1802 # succeeds
1803 1803 #
1804 1804 # See bug: https://bugs.launchpad.net/ipython/+bug/348466
1805 1805
1806 1806 #units = [u"s", u"ms",u'\xb5',"ns"]
1807 1807 units = [u"s", u"ms",u'us',"ns"]
1808 1808
1809 1809 scaling = [1, 1e3, 1e6, 1e9]
1810 1810
1811 1811 opts, stmt = self.parse_options(parameter_s,'n:r:tcp:',
1812 1812 posix=False)
1813 1813 if stmt == "":
1814 1814 return
1815 1815 timefunc = timeit.default_timer
1816 1816 number = int(getattr(opts, "n", 0))
1817 1817 repeat = int(getattr(opts, "r", timeit.default_repeat))
1818 1818 precision = int(getattr(opts, "p", 3))
1819 1819 if hasattr(opts, "t"):
1820 1820 timefunc = time.time
1821 1821 if hasattr(opts, "c"):
1822 1822 timefunc = clock
1823 1823
1824 1824 timer = timeit.Timer(timer=timefunc)
1825 1825 # this code has tight coupling to the inner workings of timeit.Timer,
1826 1826 # but is there a better way to achieve that the code stmt has access
1827 1827 # to the shell namespace?
1828 1828
1829 1829 src = timeit.template % {'stmt': timeit.reindent(stmt, 8),
1830 1830 'setup': "pass"}
1831 1831 # Track compilation time so it can be reported if too long
1832 1832 # Minimum time above which compilation time will be reported
1833 1833 tc_min = 0.1
1834 1834
1835 1835 t0 = clock()
1836 1836 code = compile(src, "<magic-timeit>", "exec")
1837 1837 tc = clock()-t0
1838 1838
1839 1839 ns = {}
1840 1840 exec code in self.shell.user_ns, ns
1841 1841 timer.inner = ns["inner"]
1842 1842
1843 1843 if number == 0:
1844 1844 # determine number so that 0.2 <= total time < 2.0
1845 1845 number = 1
1846 1846 for i in range(1, 10):
1847 1847 if timer.timeit(number) >= 0.2:
1848 1848 break
1849 1849 number *= 10
1850 1850
1851 1851 best = min(timer.repeat(repeat, number)) / number
1852 1852
1853 1853 if best > 0.0 and best < 1000.0:
1854 1854 order = min(-int(math.floor(math.log10(best)) // 3), 3)
1855 1855 elif best >= 1000.0:
1856 1856 order = 0
1857 1857 else:
1858 1858 order = 3
1859 1859 print u"%d loops, best of %d: %.*g %s per loop" % (number, repeat,
1860 1860 precision,
1861 1861 best * scaling[order],
1862 1862 units[order])
1863 1863 if tc > tc_min:
1864 1864 print "Compiler time: %.2f s" % tc
1865 1865
1866 1866 @testdec.skip_doctest
1867 1867 def magic_time(self,parameter_s = ''):
1868 1868 """Time execution of a Python statement or expression.
1869 1869
1870 1870 The CPU and wall clock times are printed, and the value of the
1871 1871 expression (if any) is returned. Note that under Win32, system time
1872 1872 is always reported as 0, since it can not be measured.
1873 1873
1874 1874 This function provides very basic timing functionality. In Python
1875 1875 2.3, the timeit module offers more control and sophistication, so this
1876 1876 could be rewritten to use it (patches welcome).
1877 1877
1878 1878 Some examples:
1879 1879
1880 1880 In [1]: time 2**128
1881 1881 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1882 1882 Wall time: 0.00
1883 1883 Out[1]: 340282366920938463463374607431768211456L
1884 1884
1885 1885 In [2]: n = 1000000
1886 1886
1887 1887 In [3]: time sum(range(n))
1888 1888 CPU times: user 1.20 s, sys: 0.05 s, total: 1.25 s
1889 1889 Wall time: 1.37
1890 1890 Out[3]: 499999500000L
1891 1891
1892 1892 In [4]: time print 'hello world'
1893 1893 hello world
1894 1894 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1895 1895 Wall time: 0.00
1896 1896
1897 1897 Note that the time needed by Python to compile the given expression
1898 1898 will be reported if it is more than 0.1s. In this example, the
1899 1899 actual exponentiation is done by Python at compilation time, so while
1900 1900 the expression can take a noticeable amount of time to compute, that
1901 1901 time is purely due to the compilation:
1902 1902
1903 1903 In [5]: time 3**9999;
1904 1904 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1905 1905 Wall time: 0.00 s
1906 1906
1907 1907 In [6]: time 3**999999;
1908 1908 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1909 1909 Wall time: 0.00 s
1910 1910 Compiler : 0.78 s
1911 1911 """
1912 1912
1913 1913 # fail immediately if the given expression can't be compiled
1914 1914
1915 1915 expr = self.shell.prefilter(parameter_s,False)
1916 1916
1917 1917 # Minimum time above which compilation time will be reported
1918 1918 tc_min = 0.1
1919 1919
1920 1920 try:
1921 1921 mode = 'eval'
1922 1922 t0 = clock()
1923 1923 code = compile(expr,'<timed eval>',mode)
1924 1924 tc = clock()-t0
1925 1925 except SyntaxError:
1926 1926 mode = 'exec'
1927 1927 t0 = clock()
1928 1928 code = compile(expr,'<timed exec>',mode)
1929 1929 tc = clock()-t0
1930 1930 # skew measurement as little as possible
1931 1931 glob = self.shell.user_ns
1932 1932 clk = clock2
1933 1933 wtime = time.time
1934 1934 # time execution
1935 1935 wall_st = wtime()
1936 1936 if mode=='eval':
1937 1937 st = clk()
1938 1938 out = eval(code,glob)
1939 1939 end = clk()
1940 1940 else:
1941 1941 st = clk()
1942 1942 exec code in glob
1943 1943 end = clk()
1944 1944 out = None
1945 1945 wall_end = wtime()
1946 1946 # Compute actual times and report
1947 1947 wall_time = wall_end-wall_st
1948 1948 cpu_user = end[0]-st[0]
1949 1949 cpu_sys = end[1]-st[1]
1950 1950 cpu_tot = cpu_user+cpu_sys
1951 1951 print "CPU times: user %.2f s, sys: %.2f s, total: %.2f s" % \
1952 1952 (cpu_user,cpu_sys,cpu_tot)
1953 1953 print "Wall time: %.2f s" % wall_time
1954 1954 if tc > tc_min:
1955 1955 print "Compiler : %.2f s" % tc
1956 1956 return out
1957 1957
1958 1958 @testdec.skip_doctest
1959 1959 def magic_macro(self,parameter_s = ''):
1960 1960 """Define a set of input lines as a macro for future re-execution.
1961 1961
1962 1962 Usage:\\
1963 1963 %macro [options] name n1-n2 n3-n4 ... n5 .. n6 ...
1964 1964
1965 1965 Options:
1966 1966
1967 1967 -r: use 'raw' input. By default, the 'processed' history is used,
1968 1968 so that magics are loaded in their transformed version to valid
1969 1969 Python. If this option is given, the raw input as typed as the
1970 1970 command line is used instead.
1971 1971
1972 1972 This will define a global variable called `name` which is a string
1973 1973 made of joining the slices and lines you specify (n1,n2,... numbers
1974 1974 above) from your input history into a single string. This variable
1975 1975 acts like an automatic function which re-executes those lines as if
1976 1976 you had typed them. You just type 'name' at the prompt and the code
1977 1977 executes.
1978 1978
1979 The notation for indicating number ranges is: n1-n2 means 'use line
1980 numbers n1,...n2' (the endpoint is included). That is, '5-7' means
1981 using the lines numbered 5,6 and 7.
1979 The syntax for indicating input ranges is described in %history.
1982 1980
1983 1981 Note: as a 'hidden' feature, you can also use traditional python slice
1984 1982 notation, where N:M means numbers N through M-1.
1985 1983
1986 1984 For example, if your history contains (%hist prints it):
1987 1985
1988 1986 44: x=1
1989 1987 45: y=3
1990 1988 46: z=x+y
1991 1989 47: print x
1992 1990 48: a=5
1993 1991 49: print 'x',x,'y',y
1994 1992
1995 1993 you can create a macro with lines 44 through 47 (included) and line 49
1996 1994 called my_macro with:
1997 1995
1998 1996 In [55]: %macro my_macro 44-47 49
1999 1997
2000 1998 Now, typing `my_macro` (without quotes) will re-execute all this code
2001 1999 in one pass.
2002 2000
2003 2001 You don't need to give the line-numbers in order, and any given line
2004 2002 number can appear multiple times. You can assemble macros with any
2005 2003 lines from your input history in any order.
2006 2004
2007 2005 The macro is a simple object which holds its value in an attribute,
2008 2006 but IPython's display system checks for macros and executes them as
2009 2007 code instead of printing them when you type their name.
2010 2008
2011 2009 You can view a macro's contents by explicitly printing it with:
2012 2010
2013 2011 'print macro_name'.
2014 2012
2015 2013 For one-off cases which DON'T contain magic function calls in them you
2016 2014 can obtain similar results by explicitly executing slices from your
2017 2015 input history with:
2018 2016
2019 2017 In [60]: exec In[44:48]+In[49]"""
2020 2018
2021 2019 opts,args = self.parse_options(parameter_s,'r',mode='list')
2022 2020 if not args: # List existing macros
2023 2021 return sorted(k for k,v in self.shell.user_ns.iteritems() if\
2024 2022 isinstance(v, Macro))
2025 2023 if len(args) == 1:
2026 2024 raise UsageError(
2027 2025 "%macro insufficient args; usage '%macro name n1-n2 n3-4...")
2028 2026 name, ranges = args[0], " ".join(args[1:])
2029 2027
2030 2028 #print 'rng',ranges # dbg
2031 2029 lines = self.extract_input_lines(ranges,'r' in opts)
2032 2030 macro = Macro(lines)
2033 2031 self.shell.define_macro(name, macro)
2034 2032 print 'Macro `%s` created. To execute, type its name (without quotes).' % name
2035 2033 print 'Macro contents:'
2036 2034 print macro,
2037 2035
2038 2036 def magic_save(self,parameter_s = ''):
2039 2037 """Save a set of lines to a given filename.
2040 2038
2041 2039 Usage:\\
2042 2040 %save [options] filename n1-n2 n3-n4 ... n5 .. n6 ...
2043 2041
2044 2042 Options:
2045 2043
2046 2044 -r: use 'raw' input. By default, the 'processed' history is used,
2047 2045 so that magics are loaded in their transformed version to valid
2048 2046 Python. If this option is given, the raw input as typed as the
2049 2047 command line is used instead.
2050 2048
2051 This function uses the same syntax as %macro for line extraction, but
2052 instead of creating a macro it saves the resulting string to the
2053 filename you specify.
2049 This function uses the same syntax as %history for input ranges,
2050 then saves the lines to the filename you specify.
2054 2051
2055 2052 It adds a '.py' extension to the file if you don't do so yourself, and
2056 2053 it asks for confirmation before overwriting existing files."""
2057 2054
2058 2055 opts,args = self.parse_options(parameter_s,'r',mode='list')
2059 2056 fname,ranges = args[0], " ".join(args[1:])
2060 2057 if not fname.endswith('.py'):
2061 2058 fname += '.py'
2062 2059 if os.path.isfile(fname):
2063 2060 ans = raw_input('File `%s` exists. Overwrite (y/[N])? ' % fname)
2064 2061 if ans.lower() not in ['y','yes']:
2065 2062 print 'Operation cancelled.'
2066 2063 return
2067 2064 cmds = self.extract_input_lines(ranges, 'r' in opts)
2068 2065 with open(fname,'w') as f:
2069 2066 f.write(cmds)
2070 2067 print 'The following commands were written to file `%s`:' % fname
2071 2068 print cmds
2072 2069
2073 2070 def _edit_macro(self,mname,macro):
2074 2071 """open an editor with the macro data in a file"""
2075 2072 filename = self.shell.mktempfile(macro.value)
2076 2073 self.shell.hooks.editor(filename)
2077 2074
2078 2075 # and make a new macro object, to replace the old one
2079 2076 mfile = open(filename)
2080 2077 mvalue = mfile.read()
2081 2078 mfile.close()
2082 2079 self.shell.user_ns[mname] = Macro(mvalue)
2083 2080
2084 2081 def magic_ed(self,parameter_s=''):
2085 2082 """Alias to %edit."""
2086 2083 return self.magic_edit(parameter_s)
2087 2084
2088 2085 @testdec.skip_doctest
2089 2086 def magic_edit(self,parameter_s='',last_call=['','']):
2090 2087 """Bring up an editor and execute the resulting code.
2091 2088
2092 2089 Usage:
2093 2090 %edit [options] [args]
2094 2091
2095 2092 %edit runs IPython's editor hook. The default version of this hook is
2096 2093 set to call the __IPYTHON__.rc.editor command. This is read from your
2097 2094 environment variable $EDITOR. If this isn't found, it will default to
2098 2095 vi under Linux/Unix and to notepad under Windows. See the end of this
2099 2096 docstring for how to change the editor hook.
2100 2097
2101 2098 You can also set the value of this editor via the command line option
2102 2099 '-editor' or in your ipythonrc file. This is useful if you wish to use
2103 2100 specifically for IPython an editor different from your typical default
2104 2101 (and for Windows users who typically don't set environment variables).
2105 2102
2106 2103 This command allows you to conveniently edit multi-line code right in
2107 2104 your IPython session.
2108 2105
2109 2106 If called without arguments, %edit opens up an empty editor with a
2110 2107 temporary file and will execute the contents of this file when you
2111 2108 close it (don't forget to save it!).
2112 2109
2113 2110
2114 2111 Options:
2115 2112
2116 2113 -n <number>: open the editor at a specified line number. By default,
2117 2114 the IPython editor hook uses the unix syntax 'editor +N filename', but
2118 2115 you can configure this by providing your own modified hook if your
2119 2116 favorite editor supports line-number specifications with a different
2120 2117 syntax.
2121 2118
2122 2119 -p: this will call the editor with the same data as the previous time
2123 2120 it was used, regardless of how long ago (in your current session) it
2124 2121 was.
2125 2122
2126 2123 -r: use 'raw' input. This option only applies to input taken from the
2127 2124 user's history. By default, the 'processed' history is used, so that
2128 2125 magics are loaded in their transformed version to valid Python. If
2129 2126 this option is given, the raw input as typed as the command line is
2130 2127 used instead. When you exit the editor, it will be executed by
2131 2128 IPython's own processor.
2132 2129
2133 2130 -x: do not execute the edited code immediately upon exit. This is
2134 2131 mainly useful if you are editing programs which need to be called with
2135 2132 command line arguments, which you can then do using %run.
2136 2133
2137 2134
2138 2135 Arguments:
2139 2136
2140 2137 If arguments are given, the following possibilites exist:
2138
2139 - If the argument is a filename, IPython will load that into the
2140 editor. It will execute its contents with execfile() when you exit,
2141 loading any code in the file into your interactive namespace.
2141 2142
2142 - The arguments are numbers or pairs of colon-separated numbers (like
2143 1 4:8 9). These are interpreted as lines of previous input to be
2144 loaded into the editor. The syntax is the same of the %macro command.
2143 - The arguments are ranges of input history, e.g. "7 ~1/4-6".
2144 The syntax is the same as in the %history magic.
2145 2145
2146 - If the argument doesn't start with a number, it is evaluated as a
2147 variable and its contents loaded into the editor. You can thus edit
2148 any string which contains python code (including the result of
2149 previous edits).
2146 - If the argument is a string variable, its contents are loaded
2147 into the editor. You can thus edit any string which contains
2148 python code (including the result of previous edits).
2150 2149
2151 2150 - If the argument is the name of an object (other than a string),
2152 2151 IPython will try to locate the file where it was defined and open the
2153 2152 editor at the point where it is defined. You can use `%edit function`
2154 2153 to load an editor exactly at the point where 'function' is defined,
2155 2154 edit it and have the file be executed automatically.
2156 2155
2157 2156 If the object is a macro (see %macro for details), this opens up your
2158 2157 specified editor with a temporary file containing the macro's data.
2159 2158 Upon exit, the macro is reloaded with the contents of the file.
2160 2159
2161 2160 Note: opening at an exact line is only supported under Unix, and some
2162 2161 editors (like kedit and gedit up to Gnome 2.8) do not understand the
2163 2162 '+NUMBER' parameter necessary for this feature. Good editors like
2164 2163 (X)Emacs, vi, jed, pico and joe all do.
2165 2164
2166 - If the argument is not found as a variable, IPython will look for a
2167 file with that name (adding .py if necessary) and load it into the
2168 editor. It will execute its contents with execfile() when you exit,
2169 loading any code in the file into your interactive namespace.
2170
2171 2165 After executing your code, %edit will return as output the code you
2172 2166 typed in the editor (except when it was an existing file). This way
2173 2167 you can reload the code in further invocations of %edit as a variable,
2174 2168 via _<NUMBER> or Out[<NUMBER>], where <NUMBER> is the prompt number of
2175 2169 the output.
2176 2170
2177 2171 Note that %edit is also available through the alias %ed.
2178 2172
2179 2173 This is an example of creating a simple function inside the editor and
2180 2174 then modifying it. First, start up the editor:
2181 2175
2182 2176 In [1]: ed
2183 2177 Editing... done. Executing edited code...
2184 2178 Out[1]: 'def foo():n print "foo() was defined in an editing session"n'
2185 2179
2186 2180 We can then call the function foo():
2187 2181
2188 2182 In [2]: foo()
2189 2183 foo() was defined in an editing session
2190 2184
2191 2185 Now we edit foo. IPython automatically loads the editor with the
2192 2186 (temporary) file where foo() was previously defined:
2193 2187
2194 2188 In [3]: ed foo
2195 2189 Editing... done. Executing edited code...
2196 2190
2197 2191 And if we call foo() again we get the modified version:
2198 2192
2199 2193 In [4]: foo()
2200 2194 foo() has now been changed!
2201 2195
2202 2196 Here is an example of how to edit a code snippet successive
2203 2197 times. First we call the editor:
2204 2198
2205 2199 In [5]: ed
2206 2200 Editing... done. Executing edited code...
2207 2201 hello
2208 2202 Out[5]: "print 'hello'n"
2209 2203
2210 2204 Now we call it again with the previous output (stored in _):
2211 2205
2212 2206 In [6]: ed _
2213 2207 Editing... done. Executing edited code...
2214 2208 hello world
2215 2209 Out[6]: "print 'hello world'n"
2216 2210
2217 2211 Now we call it with the output #8 (stored in _8, also as Out[8]):
2218 2212
2219 2213 In [7]: ed _8
2220 2214 Editing... done. Executing edited code...
2221 2215 hello again
2222 2216 Out[7]: "print 'hello again'n"
2223 2217
2224 2218
2225 2219 Changing the default editor hook:
2226 2220
2227 2221 If you wish to write your own editor hook, you can put it in a
2228 2222 configuration file which you load at startup time. The default hook
2229 2223 is defined in the IPython.core.hooks module, and you can use that as a
2230 2224 starting example for further modifications. That file also has
2231 2225 general instructions on how to set a new hook for use once you've
2232 2226 defined it."""
2233 2227
2234 2228 # FIXME: This function has become a convoluted mess. It needs a
2235 2229 # ground-up rewrite with clean, simple logic.
2236 2230
2237 2231 def make_filename(arg):
2238 2232 "Make a filename from the given args"
2239 2233 try:
2240 2234 filename = get_py_filename(arg)
2241 2235 except IOError:
2242 2236 if args.endswith('.py'):
2243 2237 filename = arg
2244 2238 else:
2245 2239 filename = None
2246 2240 return filename
2247 2241
2248 2242 # custom exceptions
2249 2243 class DataIsObject(Exception): pass
2250 2244
2251 2245 opts,args = self.parse_options(parameter_s,'prxn:')
2252 2246 # Set a few locals from the options for convenience:
2253 2247 opts_prev = 'p' in opts
2254 2248 opts_raw = 'r' in opts
2255 2249
2256 2250 # Default line number value
2257 2251 lineno = opts.get('n',None)
2258 2252
2259 2253 if opts_prev:
2260 2254 args = '_%s' % last_call[0]
2261 2255 if not self.shell.user_ns.has_key(args):
2262 2256 args = last_call[1]
2263 2257
2264 2258 # use last_call to remember the state of the previous call, but don't
2265 2259 # let it be clobbered by successive '-p' calls.
2266 2260 try:
2267 2261 last_call[0] = self.shell.displayhook.prompt_count
2268 2262 if not opts_prev:
2269 2263 last_call[1] = parameter_s
2270 2264 except:
2271 2265 pass
2272 2266
2273 2267 # by default this is done with temp files, except when the given
2274 2268 # arg is a filename
2275 2269 use_temp = True
2276 2270
2277 2271 data = ''
2278 2272 if args.endswith('.py'):
2279 2273 filename = make_filename(args)
2280 2274 use_temp = False
2281 2275 elif args:
2282 2276 # Mode where user specifies ranges of lines, like in %macro.
2283 2277 data = self.extract_input_lines(args, opts_raw)
2284 2278 if not data:
2285 2279 try:
2286 2280 # Load the parameter given as a variable. If not a string,
2287 2281 # process it as an object instead (below)
2288 2282
2289 2283 #print '*** args',args,'type',type(args) # dbg
2290 2284 data = eval(args, self.shell.user_ns)
2291 2285 if not isinstance(data, basestring):
2292 2286 raise DataIsObject
2293 2287
2294 2288 except (NameError,SyntaxError):
2295 2289 # given argument is not a variable, try as a filename
2296 2290 filename = make_filename(args)
2297 2291 if filename is None:
2298 2292 warn("Argument given (%s) can't be found as a variable "
2299 2293 "or as a filename." % args)
2300 2294 return
2301 2295 use_temp = False
2302 2296
2303 2297 except DataIsObject:
2304 2298 # macros have a special edit function
2305 2299 if isinstance(data, Macro):
2306 2300 self._edit_macro(args,data)
2307 2301 return
2308 2302
2309 2303 # For objects, try to edit the file where they are defined
2310 2304 try:
2311 2305 filename = inspect.getabsfile(data)
2312 2306 if 'fakemodule' in filename.lower() and inspect.isclass(data):
2313 2307 # class created by %edit? Try to find source
2314 2308 # by looking for method definitions instead, the
2315 2309 # __module__ in those classes is FakeModule.
2316 2310 attrs = [getattr(data, aname) for aname in dir(data)]
2317 2311 for attr in attrs:
2318 2312 if not inspect.ismethod(attr):
2319 2313 continue
2320 2314 filename = inspect.getabsfile(attr)
2321 2315 if filename and 'fakemodule' not in filename.lower():
2322 2316 # change the attribute to be the edit target instead
2323 2317 data = attr
2324 2318 break
2325 2319
2326 2320 datafile = 1
2327 2321 except TypeError:
2328 2322 filename = make_filename(args)
2329 2323 datafile = 1
2330 2324 warn('Could not find file where `%s` is defined.\n'
2331 2325 'Opening a file named `%s`' % (args,filename))
2332 2326 # Now, make sure we can actually read the source (if it was in
2333 2327 # a temp file it's gone by now).
2334 2328 if datafile:
2335 2329 try:
2336 2330 if lineno is None:
2337 2331 lineno = inspect.getsourcelines(data)[1]
2338 2332 except IOError:
2339 2333 filename = make_filename(args)
2340 2334 if filename is None:
2341 2335 warn('The file `%s` where `%s` was defined cannot '
2342 2336 'be read.' % (filename,data))
2343 2337 return
2344 2338 use_temp = False
2345 2339
2346 2340 if use_temp:
2347 2341 filename = self.shell.mktempfile(data)
2348 2342 print 'IPython will make a temporary file named:',filename
2349 2343
2350 2344 # do actual editing here
2351 2345 print 'Editing...',
2352 2346 sys.stdout.flush()
2353 2347 try:
2354 2348 # Quote filenames that may have spaces in them
2355 2349 if ' ' in filename:
2356 2350 filename = "%s" % filename
2357 2351 self.shell.hooks.editor(filename,lineno)
2358 2352 except TryNext:
2359 2353 warn('Could not open editor')
2360 2354 return
2361 2355
2362 2356 # XXX TODO: should this be generalized for all string vars?
2363 2357 # For now, this is special-cased to blocks created by cpaste
2364 2358 if args.strip() == 'pasted_block':
2365 2359 self.shell.user_ns['pasted_block'] = file_read(filename)
2366 2360
2367 2361 if 'x' in opts: # -x prevents actual execution
2368 2362 print
2369 2363 else:
2370 2364 print 'done. Executing edited code...'
2371 2365 if opts_raw:
2372 2366 self.shell.run_cell(file_read(filename),
2373 2367 store_history=False)
2374 2368 else:
2375 2369 self.shell.safe_execfile(filename,self.shell.user_ns,
2376 2370 self.shell.user_ns)
2377 2371
2378 2372
2379 2373 if use_temp:
2380 2374 try:
2381 2375 return open(filename).read()
2382 2376 except IOError,msg:
2383 2377 if msg.filename == filename:
2384 2378 warn('File not found. Did you forget to save?')
2385 2379 return
2386 2380 else:
2387 2381 self.shell.showtraceback()
2388 2382
2389 2383 def magic_xmode(self,parameter_s = ''):
2390 2384 """Switch modes for the exception handlers.
2391 2385
2392 2386 Valid modes: Plain, Context and Verbose.
2393 2387
2394 2388 If called without arguments, acts as a toggle."""
2395 2389
2396 2390 def xmode_switch_err(name):
2397 2391 warn('Error changing %s exception modes.\n%s' %
2398 2392 (name,sys.exc_info()[1]))
2399 2393
2400 2394 shell = self.shell
2401 2395 new_mode = parameter_s.strip().capitalize()
2402 2396 try:
2403 2397 shell.InteractiveTB.set_mode(mode=new_mode)
2404 2398 print 'Exception reporting mode:',shell.InteractiveTB.mode
2405 2399 except:
2406 2400 xmode_switch_err('user')
2407 2401
2408 2402 def magic_colors(self,parameter_s = ''):
2409 2403 """Switch color scheme for prompts, info system and exception handlers.
2410 2404
2411 2405 Currently implemented schemes: NoColor, Linux, LightBG.
2412 2406
2413 2407 Color scheme names are not case-sensitive.
2414 2408
2415 2409 Examples
2416 2410 --------
2417 2411 To get a plain black and white terminal::
2418 2412
2419 2413 %colors nocolor
2420 2414 """
2421 2415
2422 2416 def color_switch_err(name):
2423 2417 warn('Error changing %s color schemes.\n%s' %
2424 2418 (name,sys.exc_info()[1]))
2425 2419
2426 2420
2427 2421 new_scheme = parameter_s.strip()
2428 2422 if not new_scheme:
2429 2423 raise UsageError(
2430 2424 "%colors: you must specify a color scheme. See '%colors?'")
2431 2425 return
2432 2426 # local shortcut
2433 2427 shell = self.shell
2434 2428
2435 2429 import IPython.utils.rlineimpl as readline
2436 2430
2437 2431 if not readline.have_readline and sys.platform == "win32":
2438 2432 msg = """\
2439 2433 Proper color support under MS Windows requires the pyreadline library.
2440 2434 You can find it at:
2441 2435 http://ipython.scipy.org/moin/PyReadline/Intro
2442 2436 Gary's readline needs the ctypes module, from:
2443 2437 http://starship.python.net/crew/theller/ctypes
2444 2438 (Note that ctypes is already part of Python versions 2.5 and newer).
2445 2439
2446 2440 Defaulting color scheme to 'NoColor'"""
2447 2441 new_scheme = 'NoColor'
2448 2442 warn(msg)
2449 2443
2450 2444 # readline option is 0
2451 2445 if not shell.has_readline:
2452 2446 new_scheme = 'NoColor'
2453 2447
2454 2448 # Set prompt colors
2455 2449 try:
2456 2450 shell.displayhook.set_colors(new_scheme)
2457 2451 except:
2458 2452 color_switch_err('prompt')
2459 2453 else:
2460 2454 shell.colors = \
2461 2455 shell.displayhook.color_table.active_scheme_name
2462 2456 # Set exception colors
2463 2457 try:
2464 2458 shell.InteractiveTB.set_colors(scheme = new_scheme)
2465 2459 shell.SyntaxTB.set_colors(scheme = new_scheme)
2466 2460 except:
2467 2461 color_switch_err('exception')
2468 2462
2469 2463 # Set info (for 'object?') colors
2470 2464 if shell.color_info:
2471 2465 try:
2472 2466 shell.inspector.set_active_scheme(new_scheme)
2473 2467 except:
2474 2468 color_switch_err('object inspector')
2475 2469 else:
2476 2470 shell.inspector.set_active_scheme('NoColor')
2477 2471
2478 2472 def magic_pprint(self, parameter_s=''):
2479 2473 """Toggle pretty printing on/off."""
2480 2474 ptformatter = self.shell.display_formatter.formatters['text/plain']
2481 2475 ptformatter.pprint = bool(1 - ptformatter.pprint)
2482 2476 print 'Pretty printing has been turned', \
2483 2477 ['OFF','ON'][ptformatter.pprint]
2484 2478
2485 2479 def magic_Exit(self, parameter_s=''):
2486 2480 """Exit IPython."""
2487 2481
2488 2482 self.shell.ask_exit()
2489 2483
2490 2484 # Add aliases as magics so all common forms work: exit, quit, Exit, Quit.
2491 2485 magic_exit = magic_quit = magic_Quit = magic_Exit
2492 2486
2493 2487 #......................................................................
2494 2488 # Functions to implement unix shell-type things
2495 2489
2496 2490 @testdec.skip_doctest
2497 2491 def magic_alias(self, parameter_s = ''):
2498 2492 """Define an alias for a system command.
2499 2493
2500 2494 '%alias alias_name cmd' defines 'alias_name' as an alias for 'cmd'
2501 2495
2502 2496 Then, typing 'alias_name params' will execute the system command 'cmd
2503 2497 params' (from your underlying operating system).
2504 2498
2505 2499 Aliases have lower precedence than magic functions and Python normal
2506 2500 variables, so if 'foo' is both a Python variable and an alias, the
2507 2501 alias can not be executed until 'del foo' removes the Python variable.
2508 2502
2509 2503 You can use the %l specifier in an alias definition to represent the
2510 2504 whole line when the alias is called. For example:
2511 2505
2512 2506 In [2]: alias bracket echo "Input in brackets: <%l>"
2513 2507 In [3]: bracket hello world
2514 2508 Input in brackets: <hello world>
2515 2509
2516 2510 You can also define aliases with parameters using %s specifiers (one
2517 2511 per parameter):
2518 2512
2519 2513 In [1]: alias parts echo first %s second %s
2520 2514 In [2]: %parts A B
2521 2515 first A second B
2522 2516 In [3]: %parts A
2523 2517 Incorrect number of arguments: 2 expected.
2524 2518 parts is an alias to: 'echo first %s second %s'
2525 2519
2526 2520 Note that %l and %s are mutually exclusive. You can only use one or
2527 2521 the other in your aliases.
2528 2522
2529 2523 Aliases expand Python variables just like system calls using ! or !!
2530 2524 do: all expressions prefixed with '$' get expanded. For details of
2531 2525 the semantic rules, see PEP-215:
2532 2526 http://www.python.org/peps/pep-0215.html. This is the library used by
2533 2527 IPython for variable expansion. If you want to access a true shell
2534 2528 variable, an extra $ is necessary to prevent its expansion by IPython:
2535 2529
2536 2530 In [6]: alias show echo
2537 2531 In [7]: PATH='A Python string'
2538 2532 In [8]: show $PATH
2539 2533 A Python string
2540 2534 In [9]: show $$PATH
2541 2535 /usr/local/lf9560/bin:/usr/local/intel/compiler70/ia32/bin:...
2542 2536
2543 2537 You can use the alias facility to acess all of $PATH. See the %rehash
2544 2538 and %rehashx functions, which automatically create aliases for the
2545 2539 contents of your $PATH.
2546 2540
2547 2541 If called with no parameters, %alias prints the current alias table."""
2548 2542
2549 2543 par = parameter_s.strip()
2550 2544 if not par:
2551 2545 stored = self.db.get('stored_aliases', {} )
2552 2546 aliases = sorted(self.shell.alias_manager.aliases)
2553 2547 # for k, v in stored:
2554 2548 # atab.append(k, v[0])
2555 2549
2556 2550 print "Total number of aliases:", len(aliases)
2557 2551 sys.stdout.flush()
2558 2552 return aliases
2559 2553
2560 2554 # Now try to define a new one
2561 2555 try:
2562 2556 alias,cmd = par.split(None, 1)
2563 2557 except:
2564 2558 print oinspect.getdoc(self.magic_alias)
2565 2559 else:
2566 2560 self.shell.alias_manager.soft_define_alias(alias, cmd)
2567 2561 # end magic_alias
2568 2562
2569 2563 def magic_unalias(self, parameter_s = ''):
2570 2564 """Remove an alias"""
2571 2565
2572 2566 aname = parameter_s.strip()
2573 2567 self.shell.alias_manager.undefine_alias(aname)
2574 2568 stored = self.db.get('stored_aliases', {} )
2575 2569 if aname in stored:
2576 2570 print "Removing %stored alias",aname
2577 2571 del stored[aname]
2578 2572 self.db['stored_aliases'] = stored
2579 2573
2580 2574 def magic_rehashx(self, parameter_s = ''):
2581 2575 """Update the alias table with all executable files in $PATH.
2582 2576
2583 2577 This version explicitly checks that every entry in $PATH is a file
2584 2578 with execute access (os.X_OK), so it is much slower than %rehash.
2585 2579
2586 2580 Under Windows, it checks executability as a match agains a
2587 2581 '|'-separated string of extensions, stored in the IPython config
2588 2582 variable win_exec_ext. This defaults to 'exe|com|bat'.
2589 2583
2590 2584 This function also resets the root module cache of module completer,
2591 2585 used on slow filesystems.
2592 2586 """
2593 2587 from IPython.core.alias import InvalidAliasError
2594 2588
2595 2589 # for the benefit of module completer in ipy_completers.py
2596 2590 del self.db['rootmodules']
2597 2591
2598 2592 path = [os.path.abspath(os.path.expanduser(p)) for p in
2599 2593 os.environ.get('PATH','').split(os.pathsep)]
2600 2594 path = filter(os.path.isdir,path)
2601 2595
2602 2596 syscmdlist = []
2603 2597 # Now define isexec in a cross platform manner.
2604 2598 if os.name == 'posix':
2605 2599 isexec = lambda fname:os.path.isfile(fname) and \
2606 2600 os.access(fname,os.X_OK)
2607 2601 else:
2608 2602 try:
2609 2603 winext = os.environ['pathext'].replace(';','|').replace('.','')
2610 2604 except KeyError:
2611 2605 winext = 'exe|com|bat|py'
2612 2606 if 'py' not in winext:
2613 2607 winext += '|py'
2614 2608 execre = re.compile(r'(.*)\.(%s)$' % winext,re.IGNORECASE)
2615 2609 isexec = lambda fname:os.path.isfile(fname) and execre.match(fname)
2616 2610 savedir = os.getcwd()
2617 2611
2618 2612 # Now walk the paths looking for executables to alias.
2619 2613 try:
2620 2614 # write the whole loop for posix/Windows so we don't have an if in
2621 2615 # the innermost part
2622 2616 if os.name == 'posix':
2623 2617 for pdir in path:
2624 2618 os.chdir(pdir)
2625 2619 for ff in os.listdir(pdir):
2626 2620 if isexec(ff):
2627 2621 try:
2628 2622 # Removes dots from the name since ipython
2629 2623 # will assume names with dots to be python.
2630 2624 self.shell.alias_manager.define_alias(
2631 2625 ff.replace('.',''), ff)
2632 2626 except InvalidAliasError:
2633 2627 pass
2634 2628 else:
2635 2629 syscmdlist.append(ff)
2636 2630 else:
2637 2631 no_alias = self.shell.alias_manager.no_alias
2638 2632 for pdir in path:
2639 2633 os.chdir(pdir)
2640 2634 for ff in os.listdir(pdir):
2641 2635 base, ext = os.path.splitext(ff)
2642 2636 if isexec(ff) and base.lower() not in no_alias:
2643 2637 if ext.lower() == '.exe':
2644 2638 ff = base
2645 2639 try:
2646 2640 # Removes dots from the name since ipython
2647 2641 # will assume names with dots to be python.
2648 2642 self.shell.alias_manager.define_alias(
2649 2643 base.lower().replace('.',''), ff)
2650 2644 except InvalidAliasError:
2651 2645 pass
2652 2646 syscmdlist.append(ff)
2653 2647 db = self.db
2654 2648 db['syscmdlist'] = syscmdlist
2655 2649 finally:
2656 2650 os.chdir(savedir)
2657 2651
2658 2652 @testdec.skip_doctest
2659 2653 def magic_pwd(self, parameter_s = ''):
2660 2654 """Return the current working directory path.
2661 2655
2662 2656 Examples
2663 2657 --------
2664 2658 ::
2665 2659
2666 2660 In [9]: pwd
2667 2661 Out[9]: '/home/tsuser/sprint/ipython'
2668 2662 """
2669 2663 return os.getcwd()
2670 2664
2671 2665 @testdec.skip_doctest
2672 2666 def magic_cd(self, parameter_s=''):
2673 2667 """Change the current working directory.
2674 2668
2675 2669 This command automatically maintains an internal list of directories
2676 2670 you visit during your IPython session, in the variable _dh. The
2677 2671 command %dhist shows this history nicely formatted. You can also
2678 2672 do 'cd -<tab>' to see directory history conveniently.
2679 2673
2680 2674 Usage:
2681 2675
2682 2676 cd 'dir': changes to directory 'dir'.
2683 2677
2684 2678 cd -: changes to the last visited directory.
2685 2679
2686 2680 cd -<n>: changes to the n-th directory in the directory history.
2687 2681
2688 2682 cd --foo: change to directory that matches 'foo' in history
2689 2683
2690 2684 cd -b <bookmark_name>: jump to a bookmark set by %bookmark
2691 2685 (note: cd <bookmark_name> is enough if there is no
2692 2686 directory <bookmark_name>, but a bookmark with the name exists.)
2693 2687 'cd -b <tab>' allows you to tab-complete bookmark names.
2694 2688
2695 2689 Options:
2696 2690
2697 2691 -q: quiet. Do not print the working directory after the cd command is
2698 2692 executed. By default IPython's cd command does print this directory,
2699 2693 since the default prompts do not display path information.
2700 2694
2701 2695 Note that !cd doesn't work for this purpose because the shell where
2702 2696 !command runs is immediately discarded after executing 'command'.
2703 2697
2704 2698 Examples
2705 2699 --------
2706 2700 ::
2707 2701
2708 2702 In [10]: cd parent/child
2709 2703 /home/tsuser/parent/child
2710 2704 """
2711 2705
2712 2706 parameter_s = parameter_s.strip()
2713 2707 #bkms = self.shell.persist.get("bookmarks",{})
2714 2708
2715 2709 oldcwd = os.getcwd()
2716 2710 numcd = re.match(r'(-)(\d+)$',parameter_s)
2717 2711 # jump in directory history by number
2718 2712 if numcd:
2719 2713 nn = int(numcd.group(2))
2720 2714 try:
2721 2715 ps = self.shell.user_ns['_dh'][nn]
2722 2716 except IndexError:
2723 2717 print 'The requested directory does not exist in history.'
2724 2718 return
2725 2719 else:
2726 2720 opts = {}
2727 2721 elif parameter_s.startswith('--'):
2728 2722 ps = None
2729 2723 fallback = None
2730 2724 pat = parameter_s[2:]
2731 2725 dh = self.shell.user_ns['_dh']
2732 2726 # first search only by basename (last component)
2733 2727 for ent in reversed(dh):
2734 2728 if pat in os.path.basename(ent) and os.path.isdir(ent):
2735 2729 ps = ent
2736 2730 break
2737 2731
2738 2732 if fallback is None and pat in ent and os.path.isdir(ent):
2739 2733 fallback = ent
2740 2734
2741 2735 # if we have no last part match, pick the first full path match
2742 2736 if ps is None:
2743 2737 ps = fallback
2744 2738
2745 2739 if ps is None:
2746 2740 print "No matching entry in directory history"
2747 2741 return
2748 2742 else:
2749 2743 opts = {}
2750 2744
2751 2745
2752 2746 else:
2753 2747 #turn all non-space-escaping backslashes to slashes,
2754 2748 # for c:\windows\directory\names\
2755 2749 parameter_s = re.sub(r'\\(?! )','/', parameter_s)
2756 2750 opts,ps = self.parse_options(parameter_s,'qb',mode='string')
2757 2751 # jump to previous
2758 2752 if ps == '-':
2759 2753 try:
2760 2754 ps = self.shell.user_ns['_dh'][-2]
2761 2755 except IndexError:
2762 2756 raise UsageError('%cd -: No previous directory to change to.')
2763 2757 # jump to bookmark if needed
2764 2758 else:
2765 2759 if not os.path.isdir(ps) or opts.has_key('b'):
2766 2760 bkms = self.db.get('bookmarks', {})
2767 2761
2768 2762 if bkms.has_key(ps):
2769 2763 target = bkms[ps]
2770 2764 print '(bookmark:%s) -> %s' % (ps,target)
2771 2765 ps = target
2772 2766 else:
2773 2767 if opts.has_key('b'):
2774 2768 raise UsageError("Bookmark '%s' not found. "
2775 2769 "Use '%%bookmark -l' to see your bookmarks." % ps)
2776 2770
2777 2771 # at this point ps should point to the target dir
2778 2772 if ps:
2779 2773 try:
2780 2774 os.chdir(os.path.expanduser(ps))
2781 2775 if hasattr(self.shell, 'term_title') and self.shell.term_title:
2782 2776 set_term_title('IPython: ' + abbrev_cwd())
2783 2777 except OSError:
2784 2778 print sys.exc_info()[1]
2785 2779 else:
2786 2780 cwd = os.getcwd()
2787 2781 dhist = self.shell.user_ns['_dh']
2788 2782 if oldcwd != cwd:
2789 2783 dhist.append(cwd)
2790 2784 self.db['dhist'] = compress_dhist(dhist)[-100:]
2791 2785
2792 2786 else:
2793 2787 os.chdir(self.shell.home_dir)
2794 2788 if hasattr(self.shell, 'term_title') and self.shell.term_title:
2795 2789 set_term_title('IPython: ' + '~')
2796 2790 cwd = os.getcwd()
2797 2791 dhist = self.shell.user_ns['_dh']
2798 2792
2799 2793 if oldcwd != cwd:
2800 2794 dhist.append(cwd)
2801 2795 self.db['dhist'] = compress_dhist(dhist)[-100:]
2802 2796 if not 'q' in opts and self.shell.user_ns['_dh']:
2803 2797 print self.shell.user_ns['_dh'][-1]
2804 2798
2805 2799
2806 2800 def magic_env(self, parameter_s=''):
2807 2801 """List environment variables."""
2808 2802
2809 2803 return os.environ.data
2810 2804
2811 2805 def magic_pushd(self, parameter_s=''):
2812 2806 """Place the current dir on stack and change directory.
2813 2807
2814 2808 Usage:\\
2815 2809 %pushd ['dirname']
2816 2810 """
2817 2811
2818 2812 dir_s = self.shell.dir_stack
2819 2813 tgt = os.path.expanduser(parameter_s)
2820 2814 cwd = os.getcwd().replace(self.home_dir,'~')
2821 2815 if tgt:
2822 2816 self.magic_cd(parameter_s)
2823 2817 dir_s.insert(0,cwd)
2824 2818 return self.magic_dirs()
2825 2819
2826 2820 def magic_popd(self, parameter_s=''):
2827 2821 """Change to directory popped off the top of the stack.
2828 2822 """
2829 2823 if not self.shell.dir_stack:
2830 2824 raise UsageError("%popd on empty stack")
2831 2825 top = self.shell.dir_stack.pop(0)
2832 2826 self.magic_cd(top)
2833 2827 print "popd ->",top
2834 2828
2835 2829 def magic_dirs(self, parameter_s=''):
2836 2830 """Return the current directory stack."""
2837 2831
2838 2832 return self.shell.dir_stack
2839 2833
2840 2834 def magic_dhist(self, parameter_s=''):
2841 2835 """Print your history of visited directories.
2842 2836
2843 2837 %dhist -> print full history\\
2844 2838 %dhist n -> print last n entries only\\
2845 2839 %dhist n1 n2 -> print entries between n1 and n2 (n1 not included)\\
2846 2840
2847 2841 This history is automatically maintained by the %cd command, and
2848 2842 always available as the global list variable _dh. You can use %cd -<n>
2849 2843 to go to directory number <n>.
2850 2844
2851 2845 Note that most of time, you should view directory history by entering
2852 2846 cd -<TAB>.
2853 2847
2854 2848 """
2855 2849
2856 2850 dh = self.shell.user_ns['_dh']
2857 2851 if parameter_s:
2858 2852 try:
2859 2853 args = map(int,parameter_s.split())
2860 2854 except:
2861 2855 self.arg_err(Magic.magic_dhist)
2862 2856 return
2863 2857 if len(args) == 1:
2864 2858 ini,fin = max(len(dh)-(args[0]),0),len(dh)
2865 2859 elif len(args) == 2:
2866 2860 ini,fin = args
2867 2861 else:
2868 2862 self.arg_err(Magic.magic_dhist)
2869 2863 return
2870 2864 else:
2871 2865 ini,fin = 0,len(dh)
2872 2866 nlprint(dh,
2873 2867 header = 'Directory history (kept in _dh)',
2874 2868 start=ini,stop=fin)
2875 2869
2876 2870 @testdec.skip_doctest
2877 2871 def magic_sc(self, parameter_s=''):
2878 2872 """Shell capture - execute a shell command and capture its output.
2879 2873
2880 2874 DEPRECATED. Suboptimal, retained for backwards compatibility.
2881 2875
2882 2876 You should use the form 'var = !command' instead. Example:
2883 2877
2884 2878 "%sc -l myfiles = ls ~" should now be written as
2885 2879
2886 2880 "myfiles = !ls ~"
2887 2881
2888 2882 myfiles.s, myfiles.l and myfiles.n still apply as documented
2889 2883 below.
2890 2884
2891 2885 --
2892 2886 %sc [options] varname=command
2893 2887
2894 2888 IPython will run the given command using commands.getoutput(), and
2895 2889 will then update the user's interactive namespace with a variable
2896 2890 called varname, containing the value of the call. Your command can
2897 2891 contain shell wildcards, pipes, etc.
2898 2892
2899 2893 The '=' sign in the syntax is mandatory, and the variable name you
2900 2894 supply must follow Python's standard conventions for valid names.
2901 2895
2902 2896 (A special format without variable name exists for internal use)
2903 2897
2904 2898 Options:
2905 2899
2906 2900 -l: list output. Split the output on newlines into a list before
2907 2901 assigning it to the given variable. By default the output is stored
2908 2902 as a single string.
2909 2903
2910 2904 -v: verbose. Print the contents of the variable.
2911 2905
2912 2906 In most cases you should not need to split as a list, because the
2913 2907 returned value is a special type of string which can automatically
2914 2908 provide its contents either as a list (split on newlines) or as a
2915 2909 space-separated string. These are convenient, respectively, either
2916 2910 for sequential processing or to be passed to a shell command.
2917 2911
2918 2912 For example:
2919 2913
2920 2914 # all-random
2921 2915
2922 2916 # Capture into variable a
2923 2917 In [1]: sc a=ls *py
2924 2918
2925 2919 # a is a string with embedded newlines
2926 2920 In [2]: a
2927 2921 Out[2]: 'setup.py\\nwin32_manual_post_install.py'
2928 2922
2929 2923 # which can be seen as a list:
2930 2924 In [3]: a.l
2931 2925 Out[3]: ['setup.py', 'win32_manual_post_install.py']
2932 2926
2933 2927 # or as a whitespace-separated string:
2934 2928 In [4]: a.s
2935 2929 Out[4]: 'setup.py win32_manual_post_install.py'
2936 2930
2937 2931 # a.s is useful to pass as a single command line:
2938 2932 In [5]: !wc -l $a.s
2939 2933 146 setup.py
2940 2934 130 win32_manual_post_install.py
2941 2935 276 total
2942 2936
2943 2937 # while the list form is useful to loop over:
2944 2938 In [6]: for f in a.l:
2945 2939 ...: !wc -l $f
2946 2940 ...:
2947 2941 146 setup.py
2948 2942 130 win32_manual_post_install.py
2949 2943
2950 2944 Similiarly, the lists returned by the -l option are also special, in
2951 2945 the sense that you can equally invoke the .s attribute on them to
2952 2946 automatically get a whitespace-separated string from their contents:
2953 2947
2954 2948 In [7]: sc -l b=ls *py
2955 2949
2956 2950 In [8]: b
2957 2951 Out[8]: ['setup.py', 'win32_manual_post_install.py']
2958 2952
2959 2953 In [9]: b.s
2960 2954 Out[9]: 'setup.py win32_manual_post_install.py'
2961 2955
2962 2956 In summary, both the lists and strings used for ouptut capture have
2963 2957 the following special attributes:
2964 2958
2965 2959 .l (or .list) : value as list.
2966 2960 .n (or .nlstr): value as newline-separated string.
2967 2961 .s (or .spstr): value as space-separated string.
2968 2962 """
2969 2963
2970 2964 opts,args = self.parse_options(parameter_s,'lv')
2971 2965 # Try to get a variable name and command to run
2972 2966 try:
2973 2967 # the variable name must be obtained from the parse_options
2974 2968 # output, which uses shlex.split to strip options out.
2975 2969 var,_ = args.split('=',1)
2976 2970 var = var.strip()
2977 2971 # But the the command has to be extracted from the original input
2978 2972 # parameter_s, not on what parse_options returns, to avoid the
2979 2973 # quote stripping which shlex.split performs on it.
2980 2974 _,cmd = parameter_s.split('=',1)
2981 2975 except ValueError:
2982 2976 var,cmd = '',''
2983 2977 # If all looks ok, proceed
2984 2978 split = 'l' in opts
2985 2979 out = self.shell.getoutput(cmd, split=split)
2986 2980 if opts.has_key('v'):
2987 2981 print '%s ==\n%s' % (var,pformat(out))
2988 2982 if var:
2989 2983 self.shell.user_ns.update({var:out})
2990 2984 else:
2991 2985 return out
2992 2986
2993 2987 def magic_sx(self, parameter_s=''):
2994 2988 """Shell execute - run a shell command and capture its output.
2995 2989
2996 2990 %sx command
2997 2991
2998 2992 IPython will run the given command using commands.getoutput(), and
2999 2993 return the result formatted as a list (split on '\\n'). Since the
3000 2994 output is _returned_, it will be stored in ipython's regular output
3001 2995 cache Out[N] and in the '_N' automatic variables.
3002 2996
3003 2997 Notes:
3004 2998
3005 2999 1) If an input line begins with '!!', then %sx is automatically
3006 3000 invoked. That is, while:
3007 3001 !ls
3008 3002 causes ipython to simply issue system('ls'), typing
3009 3003 !!ls
3010 3004 is a shorthand equivalent to:
3011 3005 %sx ls
3012 3006
3013 3007 2) %sx differs from %sc in that %sx automatically splits into a list,
3014 3008 like '%sc -l'. The reason for this is to make it as easy as possible
3015 3009 to process line-oriented shell output via further python commands.
3016 3010 %sc is meant to provide much finer control, but requires more
3017 3011 typing.
3018 3012
3019 3013 3) Just like %sc -l, this is a list with special attributes:
3020 3014
3021 3015 .l (or .list) : value as list.
3022 3016 .n (or .nlstr): value as newline-separated string.
3023 3017 .s (or .spstr): value as whitespace-separated string.
3024 3018
3025 3019 This is very useful when trying to use such lists as arguments to
3026 3020 system commands."""
3027 3021
3028 3022 if parameter_s:
3029 3023 return self.shell.getoutput(parameter_s)
3030 3024
3031 3025
3032 3026 def magic_bookmark(self, parameter_s=''):
3033 3027 """Manage IPython's bookmark system.
3034 3028
3035 3029 %bookmark <name> - set bookmark to current dir
3036 3030 %bookmark <name> <dir> - set bookmark to <dir>
3037 3031 %bookmark -l - list all bookmarks
3038 3032 %bookmark -d <name> - remove bookmark
3039 3033 %bookmark -r - remove all bookmarks
3040 3034
3041 3035 You can later on access a bookmarked folder with:
3042 3036 %cd -b <name>
3043 3037 or simply '%cd <name>' if there is no directory called <name> AND
3044 3038 there is such a bookmark defined.
3045 3039
3046 3040 Your bookmarks persist through IPython sessions, but they are
3047 3041 associated with each profile."""
3048 3042
3049 3043 opts,args = self.parse_options(parameter_s,'drl',mode='list')
3050 3044 if len(args) > 2:
3051 3045 raise UsageError("%bookmark: too many arguments")
3052 3046
3053 3047 bkms = self.db.get('bookmarks',{})
3054 3048
3055 3049 if opts.has_key('d'):
3056 3050 try:
3057 3051 todel = args[0]
3058 3052 except IndexError:
3059 3053 raise UsageError(
3060 3054 "%bookmark -d: must provide a bookmark to delete")
3061 3055 else:
3062 3056 try:
3063 3057 del bkms[todel]
3064 3058 except KeyError:
3065 3059 raise UsageError(
3066 3060 "%%bookmark -d: Can't delete bookmark '%s'" % todel)
3067 3061
3068 3062 elif opts.has_key('r'):
3069 3063 bkms = {}
3070 3064 elif opts.has_key('l'):
3071 3065 bks = bkms.keys()
3072 3066 bks.sort()
3073 3067 if bks:
3074 3068 size = max(map(len,bks))
3075 3069 else:
3076 3070 size = 0
3077 3071 fmt = '%-'+str(size)+'s -> %s'
3078 3072 print 'Current bookmarks:'
3079 3073 for bk in bks:
3080 3074 print fmt % (bk,bkms[bk])
3081 3075 else:
3082 3076 if not args:
3083 3077 raise UsageError("%bookmark: You must specify the bookmark name")
3084 3078 elif len(args)==1:
3085 3079 bkms[args[0]] = os.getcwd()
3086 3080 elif len(args)==2:
3087 3081 bkms[args[0]] = args[1]
3088 3082 self.db['bookmarks'] = bkms
3089 3083
3090 3084 def magic_pycat(self, parameter_s=''):
3091 3085 """Show a syntax-highlighted file through a pager.
3092 3086
3093 3087 This magic is similar to the cat utility, but it will assume the file
3094 3088 to be Python source and will show it with syntax highlighting. """
3095 3089
3096 3090 try:
3097 3091 filename = get_py_filename(parameter_s)
3098 3092 cont = file_read(filename)
3099 3093 except IOError:
3100 3094 try:
3101 3095 cont = eval(parameter_s,self.user_ns)
3102 3096 except NameError:
3103 3097 cont = None
3104 3098 if cont is None:
3105 3099 print "Error: no such file or variable"
3106 3100 return
3107 3101
3108 3102 page.page(self.shell.pycolorize(cont))
3109 3103
3110 3104 def _rerun_pasted(self):
3111 3105 """ Rerun a previously pasted command.
3112 3106 """
3113 3107 b = self.user_ns.get('pasted_block', None)
3114 3108 if b is None:
3115 3109 raise UsageError('No previous pasted block available')
3116 3110 print "Re-executing '%s...' (%d chars)"% (b.split('\n',1)[0], len(b))
3117 3111 exec b in self.user_ns
3118 3112
3119 3113 def _get_pasted_lines(self, sentinel):
3120 3114 """ Yield pasted lines until the user enters the given sentinel value.
3121 3115 """
3122 3116 from IPython.core import interactiveshell
3123 3117 print "Pasting code; enter '%s' alone on the line to stop." % sentinel
3124 3118 while True:
3125 3119 l = interactiveshell.raw_input_original(':')
3126 3120 if l == sentinel:
3127 3121 return
3128 3122 else:
3129 3123 yield l
3130 3124
3131 3125 def _strip_pasted_lines_for_code(self, raw_lines):
3132 3126 """ Strip non-code parts of a sequence of lines to return a block of
3133 3127 code.
3134 3128 """
3135 3129 # Regular expressions that declare text we strip from the input:
3136 3130 strip_re = [r'^\s*In \[\d+\]:', # IPython input prompt
3137 3131 r'^\s*(\s?>)+', # Python input prompt
3138 3132 r'^\s*\.{3,}', # Continuation prompts
3139 3133 r'^\++',
3140 3134 ]
3141 3135
3142 3136 strip_from_start = map(re.compile,strip_re)
3143 3137
3144 3138 lines = []
3145 3139 for l in raw_lines:
3146 3140 for pat in strip_from_start:
3147 3141 l = pat.sub('',l)
3148 3142 lines.append(l)
3149 3143
3150 3144 block = "\n".join(lines) + '\n'
3151 3145 #print "block:\n",block
3152 3146 return block
3153 3147
3154 3148 def _execute_block(self, block, par):
3155 3149 """ Execute a block, or store it in a variable, per the user's request.
3156 3150 """
3157 3151 if not par:
3158 3152 b = textwrap.dedent(block)
3159 3153 self.user_ns['pasted_block'] = b
3160 3154 exec b in self.user_ns
3161 3155 else:
3162 3156 self.user_ns[par] = SList(block.splitlines())
3163 3157 print "Block assigned to '%s'" % par
3164 3158
3165 3159 def magic_quickref(self,arg):
3166 3160 """ Show a quick reference sheet """
3167 3161 import IPython.core.usage
3168 3162 qr = IPython.core.usage.quick_reference + self.magic_magic('-brief')
3169 3163
3170 3164 page.page(qr)
3171 3165
3172 3166 def magic_doctest_mode(self,parameter_s=''):
3173 3167 """Toggle doctest mode on and off.
3174 3168
3175 3169 This mode is intended to make IPython behave as much as possible like a
3176 3170 plain Python shell, from the perspective of how its prompts, exceptions
3177 3171 and output look. This makes it easy to copy and paste parts of a
3178 3172 session into doctests. It does so by:
3179 3173
3180 3174 - Changing the prompts to the classic ``>>>`` ones.
3181 3175 - Changing the exception reporting mode to 'Plain'.
3182 3176 - Disabling pretty-printing of output.
3183 3177
3184 3178 Note that IPython also supports the pasting of code snippets that have
3185 3179 leading '>>>' and '...' prompts in them. This means that you can paste
3186 3180 doctests from files or docstrings (even if they have leading
3187 3181 whitespace), and the code will execute correctly. You can then use
3188 3182 '%history -t' to see the translated history; this will give you the
3189 3183 input after removal of all the leading prompts and whitespace, which
3190 3184 can be pasted back into an editor.
3191 3185
3192 3186 With these features, you can switch into this mode easily whenever you
3193 3187 need to do testing and changes to doctests, without having to leave
3194 3188 your existing IPython session.
3195 3189 """
3196 3190
3197 3191 from IPython.utils.ipstruct import Struct
3198 3192
3199 3193 # Shorthands
3200 3194 shell = self.shell
3201 3195 oc = shell.displayhook
3202 3196 meta = shell.meta
3203 3197 disp_formatter = self.shell.display_formatter
3204 3198 ptformatter = disp_formatter.formatters['text/plain']
3205 3199 # dstore is a data store kept in the instance metadata bag to track any
3206 3200 # changes we make, so we can undo them later.
3207 3201 dstore = meta.setdefault('doctest_mode',Struct())
3208 3202 save_dstore = dstore.setdefault
3209 3203
3210 3204 # save a few values we'll need to recover later
3211 3205 mode = save_dstore('mode',False)
3212 3206 save_dstore('rc_pprint',ptformatter.pprint)
3213 3207 save_dstore('xmode',shell.InteractiveTB.mode)
3214 3208 save_dstore('rc_separate_out',shell.separate_out)
3215 3209 save_dstore('rc_separate_out2',shell.separate_out2)
3216 3210 save_dstore('rc_prompts_pad_left',shell.prompts_pad_left)
3217 3211 save_dstore('rc_separate_in',shell.separate_in)
3218 3212 save_dstore('rc_plain_text_only',disp_formatter.plain_text_only)
3219 3213
3220 3214 if mode == False:
3221 3215 # turn on
3222 3216 oc.prompt1.p_template = '>>> '
3223 3217 oc.prompt2.p_template = '... '
3224 3218 oc.prompt_out.p_template = ''
3225 3219
3226 3220 # Prompt separators like plain python
3227 3221 oc.input_sep = oc.prompt1.sep = ''
3228 3222 oc.output_sep = ''
3229 3223 oc.output_sep2 = ''
3230 3224
3231 3225 oc.prompt1.pad_left = oc.prompt2.pad_left = \
3232 3226 oc.prompt_out.pad_left = False
3233 3227
3234 3228 ptformatter.pprint = False
3235 3229 disp_formatter.plain_text_only = True
3236 3230
3237 3231 shell.magic_xmode('Plain')
3238 3232 else:
3239 3233 # turn off
3240 3234 oc.prompt1.p_template = shell.prompt_in1
3241 3235 oc.prompt2.p_template = shell.prompt_in2
3242 3236 oc.prompt_out.p_template = shell.prompt_out
3243 3237
3244 3238 oc.input_sep = oc.prompt1.sep = dstore.rc_separate_in
3245 3239
3246 3240 oc.output_sep = dstore.rc_separate_out
3247 3241 oc.output_sep2 = dstore.rc_separate_out2
3248 3242
3249 3243 oc.prompt1.pad_left = oc.prompt2.pad_left = \
3250 3244 oc.prompt_out.pad_left = dstore.rc_prompts_pad_left
3251 3245
3252 3246 ptformatter.pprint = dstore.rc_pprint
3253 3247 disp_formatter.plain_text_only = dstore.rc_plain_text_only
3254 3248
3255 3249 shell.magic_xmode(dstore.xmode)
3256 3250
3257 3251 # Store new mode and inform
3258 3252 dstore.mode = bool(1-int(mode))
3259 3253 mode_label = ['OFF','ON'][dstore.mode]
3260 3254 print 'Doctest mode is:', mode_label
3261 3255
3262 3256 def magic_gui(self, parameter_s=''):
3263 3257 """Enable or disable IPython GUI event loop integration.
3264 3258
3265 3259 %gui [GUINAME]
3266 3260
3267 3261 This magic replaces IPython's threaded shells that were activated
3268 3262 using the (pylab/wthread/etc.) command line flags. GUI toolkits
3269 3263 can now be enabled, disabled and swtiched at runtime and keyboard
3270 3264 interrupts should work without any problems. The following toolkits
3271 3265 are supported: wxPython, PyQt4, PyGTK, and Tk::
3272 3266
3273 3267 %gui wx # enable wxPython event loop integration
3274 3268 %gui qt4|qt # enable PyQt4 event loop integration
3275 3269 %gui gtk # enable PyGTK event loop integration
3276 3270 %gui tk # enable Tk event loop integration
3277 3271 %gui # disable all event loop integration
3278 3272
3279 3273 WARNING: after any of these has been called you can simply create
3280 3274 an application object, but DO NOT start the event loop yourself, as
3281 3275 we have already handled that.
3282 3276 """
3283 3277 from IPython.lib.inputhook import enable_gui
3284 3278 opts, arg = self.parse_options(parameter_s, '')
3285 3279 if arg=='': arg = None
3286 3280 return enable_gui(arg)
3287 3281
3288 3282 def magic_load_ext(self, module_str):
3289 3283 """Load an IPython extension by its module name."""
3290 3284 return self.extension_manager.load_extension(module_str)
3291 3285
3292 3286 def magic_unload_ext(self, module_str):
3293 3287 """Unload an IPython extension by its module name."""
3294 3288 self.extension_manager.unload_extension(module_str)
3295 3289
3296 3290 def magic_reload_ext(self, module_str):
3297 3291 """Reload an IPython extension by its module name."""
3298 3292 self.extension_manager.reload_extension(module_str)
3299 3293
3300 3294 @testdec.skip_doctest
3301 3295 def magic_install_profiles(self, s):
3302 3296 """Install the default IPython profiles into the .ipython dir.
3303 3297
3304 3298 If the default profiles have already been installed, they will not
3305 3299 be overwritten. You can force overwriting them by using the ``-o``
3306 3300 option::
3307 3301
3308 3302 In [1]: %install_profiles -o
3309 3303 """
3310 3304 if '-o' in s:
3311 3305 overwrite = True
3312 3306 else:
3313 3307 overwrite = False
3314 3308 from IPython.config import profile
3315 3309 profile_dir = os.path.split(profile.__file__)[0]
3316 3310 ipython_dir = self.ipython_dir
3317 3311 files = os.listdir(profile_dir)
3318 3312
3319 3313 to_install = []
3320 3314 for f in files:
3321 3315 if f.startswith('ipython_config'):
3322 3316 src = os.path.join(profile_dir, f)
3323 3317 dst = os.path.join(ipython_dir, f)
3324 3318 if (not os.path.isfile(dst)) or overwrite:
3325 3319 to_install.append((f, src, dst))
3326 3320 if len(to_install)>0:
3327 3321 print "Installing profiles to: ", ipython_dir
3328 3322 for (f, src, dst) in to_install:
3329 3323 shutil.copy(src, dst)
3330 3324 print " %s" % f
3331 3325
3332 3326 def magic_install_default_config(self, s):
3333 3327 """Install IPython's default config file into the .ipython dir.
3334 3328
3335 3329 If the default config file (:file:`ipython_config.py`) is already
3336 3330 installed, it will not be overwritten. You can force overwriting
3337 3331 by using the ``-o`` option::
3338 3332
3339 3333 In [1]: %install_default_config
3340 3334 """
3341 3335 if '-o' in s:
3342 3336 overwrite = True
3343 3337 else:
3344 3338 overwrite = False
3345 3339 from IPython.config import default
3346 3340 config_dir = os.path.split(default.__file__)[0]
3347 3341 ipython_dir = self.ipython_dir
3348 3342 default_config_file_name = 'ipython_config.py'
3349 3343 src = os.path.join(config_dir, default_config_file_name)
3350 3344 dst = os.path.join(ipython_dir, default_config_file_name)
3351 3345 if (not os.path.isfile(dst)) or overwrite:
3352 3346 shutil.copy(src, dst)
3353 3347 print "Installing default config file: %s" % dst
3354 3348
3355 3349 # Pylab support: simple wrappers that activate pylab, load gui input
3356 3350 # handling and modify slightly %run
3357 3351
3358 3352 @testdec.skip_doctest
3359 3353 def _pylab_magic_run(self, parameter_s=''):
3360 3354 Magic.magic_run(self, parameter_s,
3361 3355 runner=mpl_runner(self.shell.safe_execfile))
3362 3356
3363 3357 _pylab_magic_run.__doc__ = magic_run.__doc__
3364 3358
3365 3359 @testdec.skip_doctest
3366 3360 def magic_pylab(self, s):
3367 3361 """Load numpy and matplotlib to work interactively.
3368 3362
3369 3363 %pylab [GUINAME]
3370 3364
3371 3365 This function lets you activate pylab (matplotlib, numpy and
3372 3366 interactive support) at any point during an IPython session.
3373 3367
3374 3368 It will import at the top level numpy as np, pyplot as plt, matplotlib,
3375 3369 pylab and mlab, as well as all names from numpy and pylab.
3376 3370
3377 3371 Parameters
3378 3372 ----------
3379 3373 guiname : optional
3380 3374 One of the valid arguments to the %gui magic ('qt', 'wx', 'gtk' or
3381 3375 'tk'). If given, the corresponding Matplotlib backend is used,
3382 3376 otherwise matplotlib's default (which you can override in your
3383 3377 matplotlib config file) is used.
3384 3378
3385 3379 Examples
3386 3380 --------
3387 3381 In this case, where the MPL default is TkAgg:
3388 3382 In [2]: %pylab
3389 3383
3390 3384 Welcome to pylab, a matplotlib-based Python environment.
3391 3385 Backend in use: TkAgg
3392 3386 For more information, type 'help(pylab)'.
3393 3387
3394 3388 But you can explicitly request a different backend:
3395 3389 In [3]: %pylab qt
3396 3390
3397 3391 Welcome to pylab, a matplotlib-based Python environment.
3398 3392 Backend in use: Qt4Agg
3399 3393 For more information, type 'help(pylab)'.
3400 3394 """
3401 3395 self.shell.enable_pylab(s)
3402 3396
3403 3397 def magic_tb(self, s):
3404 3398 """Print the last traceback with the currently active exception mode.
3405 3399
3406 3400 See %xmode for changing exception reporting modes."""
3407 3401 self.shell.showtraceback()
3408 3402
3409 3403 @testdec.skip_doctest
3410 3404 def magic_precision(self, s=''):
3411 3405 """Set floating point precision for pretty printing.
3412 3406
3413 3407 Can set either integer precision or a format string.
3414 3408
3415 3409 If numpy has been imported and precision is an int,
3416 3410 numpy display precision will also be set, via ``numpy.set_printoptions``.
3417 3411
3418 3412 If no argument is given, defaults will be restored.
3419 3413
3420 3414 Examples
3421 3415 --------
3422 3416 ::
3423 3417
3424 3418 In [1]: from math import pi
3425 3419
3426 3420 In [2]: %precision 3
3427 3421 Out[2]: '%.3f'
3428 3422
3429 3423 In [3]: pi
3430 3424 Out[3]: 3.142
3431 3425
3432 3426 In [4]: %precision %i
3433 3427 Out[4]: '%i'
3434 3428
3435 3429 In [5]: pi
3436 3430 Out[5]: 3
3437 3431
3438 3432 In [6]: %precision %e
3439 3433 Out[6]: '%e'
3440 3434
3441 3435 In [7]: pi**10
3442 3436 Out[7]: 9.364805e+04
3443 3437
3444 3438 In [8]: %precision
3445 3439 Out[8]: '%r'
3446 3440
3447 3441 In [9]: pi**10
3448 3442 Out[9]: 93648.047476082982
3449 3443
3450 3444 """
3451 3445
3452 3446 ptformatter = self.shell.display_formatter.formatters['text/plain']
3453 3447 ptformatter.float_precision = s
3454 3448 return ptformatter.float_format
3455 3449
3456 3450 # end Magic
@@ -1,109 +1,109 b''
1 1 """Tests for the IPython tab-completion machinery.
2 2 """
3 3 #-----------------------------------------------------------------------------
4 4 # Module imports
5 5 #-----------------------------------------------------------------------------
6 6
7 7 # stdlib
8 8 import os
9 9 import sys
10 10 import unittest
11 11
12 12 # third party
13 13 import nose.tools as nt
14 14
15 15 # our own packages
16 16 from IPython.utils.tempdir import TemporaryDirectory
17 17 from IPython.core.history import HistoryManager, extract_hist_ranges
18 18
19 19 def test_history():
20 20
21 21 ip = get_ipython()
22 22 with TemporaryDirectory() as tmpdir:
23 23 #tmpdir = '/software/temp'
24 24 histfile = os.path.realpath(os.path.join(tmpdir, 'history.sqlite'))
25 25 # Ensure that we restore the history management that we mess with in
26 26 # this test doesn't affect the IPython instance used by the test suite
27 27 # beyond this test.
28 28 hist_manager_ori = ip.history_manager
29 29 try:
30 30 ip.history_manager = HistoryManager(shell=ip)
31 31 ip.history_manager.hist_file = histfile
32 32 ip.history_manager.init_db() # Has to be called after changing file
33 33 ip.history_manager.reset()
34 34 print 'test',histfile
35 35 hist = ['a=1', 'def f():\n test = 1\n return test', 'b=2']
36 36 for i, h in enumerate(hist, start=1):
37 37 ip.history_manager.store_inputs(i, h)
38 38
39 39 ip.history_manager.db_log_output = True
40 40 # Doesn't match the input, but we'll just check it's stored.
41 41 ip.history_manager.output_hist_reprs[3].append("spam")
42 42 ip.history_manager.store_output(3)
43 43
44 44 nt.assert_equal(ip.history_manager.input_hist_raw, [''] + hist)
45 45
46 46 # Check lines were written to DB
47 47 c = ip.history_manager.db.execute("SELECT source_raw FROM history")
48 48 nt.assert_equal([x for x, in c], hist)
49 49
50 50 # New session
51 51 ip.history_manager.reset()
52 52 newcmds = ["z=5","class X(object):\n pass", "k='p'"]
53 53 for i, cmd in enumerate(newcmds, start=1):
54 54 ip.history_manager.store_inputs(i, cmd)
55 gothist = ip.history_manager.get_history(start=1, stop=4)
55 gothist = ip.history_manager.get_range(start=1, stop=4)
56 56 nt.assert_equal(list(gothist), zip([0,0,0],[1,2,3], newcmds))
57 57 # Previous session:
58 gothist = ip.history_manager.get_history(-1, 1, 4)
58 gothist = ip.history_manager.get_range(-1, 1, 4)
59 59 nt.assert_equal(list(gothist), zip([1,1,1],[1,2,3], hist))
60 60
61 61 # Check get_hist_tail
62 gothist = ip.history_manager.get_hist_tail(4, output=True,
62 gothist = ip.history_manager.get_tail(4, output=True,
63 63 include_latest=True)
64 64 expected = [(1, 3, (hist[-1], ["spam"])),
65 65 (2, 1, (newcmds[0], None)),
66 66 (2, 2, (newcmds[1], None)),
67 67 (2, 3, (newcmds[2], None)),]
68 68 nt.assert_equal(list(gothist), expected)
69 69
70 gothist = ip.history_manager.get_hist_tail(2)
70 gothist = ip.history_manager.get_tail(2)
71 71 expected = [(2, 1, newcmds[0]),
72 72 (2, 2, newcmds[1])]
73 73 nt.assert_equal(list(gothist), expected)
74 74
75 75 # Check get_hist_search
76 gothist = ip.history_manager.get_hist_search("*test*")
76 gothist = ip.history_manager.search("*test*")
77 77 nt.assert_equal(list(gothist), [(1,2,hist[1])] )
78 gothist = ip.history_manager.get_hist_search("b*", output=True)
78 gothist = ip.history_manager.search("b*", output=True)
79 79 nt.assert_equal(list(gothist), [(1,3,(hist[2],["spam"]))] )
80 80
81 81 # Cross testing: check that magic %save can get previous session.
82 82 testfilename = os.path.realpath(os.path.join(tmpdir, "test.py"))
83 83 ip.magic_save(testfilename + " ~1/1-3")
84 84 testfile = open(testfilename, "r")
85 85 nt.assert_equal(testfile.read(), "\n".join(hist))
86 86 finally:
87 87 # Restore history manager
88 88 ip.history_manager = hist_manager_ori
89 89
90 90 def test_extract_hist_ranges():
91 91 instr = "1 2/3 ~4/5-6 ~4/7-~4/9 ~9/2-~7/5"
92 92 expected = [(0, 1, 2), # 0 == current session
93 93 (2, 3, 4),
94 94 (-4, 5, 7),
95 95 (-4, 7, 10),
96 96 (-9, 2, None), # None == to end
97 97 (-8, 1, None),
98 98 (-7, 1, 6)]
99 99 actual = list(extract_hist_ranges(instr))
100 100 nt.assert_equal(actual, expected)
101 101
102 102 def test_magic_rerun():
103 103 """Simple test for %rerun (no args -> rerun last line)"""
104 104 ip = get_ipython()
105 105 ip.run_cell("a = 10")
106 106 ip.run_cell("a += 1")
107 107 nt.assert_equal(ip.user_ns["a"], 11)
108 108 ip.run_cell("%rerun")
109 109 nt.assert_equal(ip.user_ns["a"], 12)
@@ -1,643 +1,643 b''
1 1 #!/usr/bin/env python
2 2 """A simple interactive kernel that talks to a frontend over 0MQ.
3 3
4 4 Things to do:
5 5
6 6 * Implement `set_parent` logic. Right before doing exec, the Kernel should
7 7 call set_parent on all the PUB objects with the message about to be executed.
8 8 * Implement random port and security key logic.
9 9 * Implement control messages.
10 10 * Implement event loop and poll version.
11 11 """
12 12
13 13 #-----------------------------------------------------------------------------
14 14 # Imports
15 15 #-----------------------------------------------------------------------------
16 16 from __future__ import print_function
17 17
18 18 # Standard library imports.
19 19 import __builtin__
20 20 import atexit
21 21 import sys
22 22 import time
23 23 import traceback
24 24 import logging
25 25 # System library imports.
26 26 import zmq
27 27
28 28 # Local imports.
29 29 from IPython.config.configurable import Configurable
30 30 from IPython.utils import io
31 31 from IPython.utils.jsonutil import json_clean
32 32 from IPython.lib import pylabtools
33 33 from IPython.utils.traitlets import Instance, Float
34 34 from entry_point import (base_launch_kernel, make_argument_parser, make_kernel,
35 35 start_kernel)
36 36 from iostream import OutStream
37 37 from session import Session, Message
38 38 from zmqshell import ZMQInteractiveShell
39 39
40 40 #-----------------------------------------------------------------------------
41 41 # Globals
42 42 #-----------------------------------------------------------------------------
43 43
44 44 # Module-level logger
45 45 logger = logging.getLogger(__name__)
46 46
47 47 #-----------------------------------------------------------------------------
48 48 # Main kernel class
49 49 #-----------------------------------------------------------------------------
50 50
51 51 class Kernel(Configurable):
52 52
53 53 #---------------------------------------------------------------------------
54 54 # Kernel interface
55 55 #---------------------------------------------------------------------------
56 56
57 57 shell = Instance('IPython.core.interactiveshell.InteractiveShellABC')
58 58 session = Instance(Session)
59 59 reply_socket = Instance('zmq.Socket')
60 60 pub_socket = Instance('zmq.Socket')
61 61 req_socket = Instance('zmq.Socket')
62 62
63 63 # Private interface
64 64
65 65 # Time to sleep after flushing the stdout/err buffers in each execute
66 66 # cycle. While this introduces a hard limit on the minimal latency of the
67 67 # execute cycle, it helps prevent output synchronization problems for
68 68 # clients.
69 69 # Units are in seconds. The minimum zmq latency on local host is probably
70 70 # ~150 microseconds, set this to 500us for now. We may need to increase it
71 71 # a little if it's not enough after more interactive testing.
72 72 _execute_sleep = Float(0.0005, config=True)
73 73
74 74 # Frequency of the kernel's event loop.
75 75 # Units are in seconds, kernel subclasses for GUI toolkits may need to
76 76 # adapt to milliseconds.
77 77 _poll_interval = Float(0.05, config=True)
78 78
79 79 # If the shutdown was requested over the network, we leave here the
80 80 # necessary reply message so it can be sent by our registered atexit
81 81 # handler. This ensures that the reply is only sent to clients truly at
82 82 # the end of our shutdown process (which happens after the underlying
83 83 # IPython shell's own shutdown).
84 84 _shutdown_message = None
85 85
86 86 # This is a dict of port number that the kernel is listening on. It is set
87 87 # by record_ports and used by connect_request.
88 88 _recorded_ports = None
89 89
90 90
91 91 def __init__(self, **kwargs):
92 92 super(Kernel, self).__init__(**kwargs)
93 93
94 94 # Before we even start up the shell, register *first* our exit handlers
95 95 # so they come before the shell's
96 96 atexit.register(self._at_shutdown)
97 97
98 98 # Initialize the InteractiveShell subclass
99 99 self.shell = ZMQInteractiveShell.instance()
100 100 self.shell.displayhook.session = self.session
101 101 self.shell.displayhook.pub_socket = self.pub_socket
102 102 self.shell.display_pub.session = self.session
103 103 self.shell.display_pub.pub_socket = self.pub_socket
104 104
105 105 # TMP - hack while developing
106 106 self.shell._reply_content = None
107 107
108 108 # Build dict of handlers for message types
109 109 msg_types = [ 'execute_request', 'complete_request',
110 110 'object_info_request', 'history_tail_request',
111 111 'connect_request', 'shutdown_request']
112 112 self.handlers = {}
113 113 for msg_type in msg_types:
114 114 self.handlers[msg_type] = getattr(self, msg_type)
115 115
116 116 def do_one_iteration(self):
117 117 """Do one iteration of the kernel's evaluation loop.
118 118 """
119 119 ident,msg = self.session.recv(self.reply_socket, zmq.NOBLOCK)
120 120 if msg is None:
121 121 return
122 122
123 123 # This assert will raise in versions of zeromq 2.0.7 and lesser.
124 124 # We now require 2.0.8 or above, so we can uncomment for safety.
125 125 # print(ident,msg, file=sys.__stdout__)
126 126 assert ident is not None, "Missing message part."
127 127
128 128 # Print some info about this message and leave a '--->' marker, so it's
129 129 # easier to trace visually the message chain when debugging. Each
130 130 # handler prints its message at the end.
131 131 # Eventually we'll move these from stdout to a logger.
132 132 logger.debug('\n*** MESSAGE TYPE:'+str(msg['msg_type'])+'***')
133 133 logger.debug(' Content: '+str(msg['content'])+'\n --->\n ')
134 134
135 135 # Find and call actual handler for message
136 136 handler = self.handlers.get(msg['msg_type'], None)
137 137 if handler is None:
138 138 logger.error("UNKNOWN MESSAGE TYPE:" +str(msg))
139 139 else:
140 140 handler(ident, msg)
141 141
142 142 # Check whether we should exit, in case the incoming message set the
143 143 # exit flag on
144 144 if self.shell.exit_now:
145 145 logger.debug('\nExiting IPython kernel...')
146 146 # We do a normal, clean exit, which allows any actions registered
147 147 # via atexit (such as history saving) to take place.
148 148 sys.exit(0)
149 149
150 150
151 151 def start(self):
152 152 """ Start the kernel main loop.
153 153 """
154 154 while True:
155 155 time.sleep(self._poll_interval)
156 156 self.do_one_iteration()
157 157
158 158 def record_ports(self, xrep_port, pub_port, req_port, hb_port):
159 159 """Record the ports that this kernel is using.
160 160
161 161 The creator of the Kernel instance must call this methods if they
162 162 want the :meth:`connect_request` method to return the port numbers.
163 163 """
164 164 self._recorded_ports = {
165 165 'xrep_port' : xrep_port,
166 166 'pub_port' : pub_port,
167 167 'req_port' : req_port,
168 168 'hb_port' : hb_port
169 169 }
170 170
171 171 #---------------------------------------------------------------------------
172 172 # Kernel request handlers
173 173 #---------------------------------------------------------------------------
174 174
175 175 def _publish_pyin(self, code, parent):
176 176 """Publish the code request on the pyin stream."""
177 177
178 178 pyin_msg = self.session.send(self.pub_socket, u'pyin',{u'code':code}, parent=parent)
179 179
180 180 def execute_request(self, ident, parent):
181 181
182 182 status_msg = self.session.send(self.pub_socket,
183 183 u'status',
184 184 {u'execution_state':u'busy'},
185 185 parent=parent
186 186 )
187 187
188 188 try:
189 189 content = parent[u'content']
190 190 code = content[u'code']
191 191 silent = content[u'silent']
192 192 except:
193 193 logger.error("Got bad msg: ")
194 194 logger.error(str(Message(parent)))
195 195 return
196 196
197 197 shell = self.shell # we'll need this a lot here
198 198
199 199 # Replace raw_input. Note that is not sufficient to replace
200 200 # raw_input in the user namespace.
201 201 raw_input = lambda prompt='': self._raw_input(prompt, ident, parent)
202 202 __builtin__.raw_input = raw_input
203 203
204 204 # Set the parent message of the display hook and out streams.
205 205 shell.displayhook.set_parent(parent)
206 206 shell.display_pub.set_parent(parent)
207 207 sys.stdout.set_parent(parent)
208 208 sys.stderr.set_parent(parent)
209 209
210 210 # Re-broadcast our input for the benefit of listening clients, and
211 211 # start computing output
212 212 if not silent:
213 213 self._publish_pyin(code, parent)
214 214
215 215 reply_content = {}
216 216 try:
217 217 if silent:
218 218 # run_code uses 'exec' mode, so no displayhook will fire, and it
219 219 # doesn't call logging or history manipulations. Print
220 220 # statements in that code will obviously still execute.
221 221 shell.run_code(code)
222 222 else:
223 223 # FIXME: the shell calls the exception handler itself.
224 224 shell._reply_content = None
225 225 shell.run_cell(code)
226 226 except:
227 227 status = u'error'
228 228 # FIXME: this code right now isn't being used yet by default,
229 229 # because the runlines() call above directly fires off exception
230 230 # reporting. This code, therefore, is only active in the scenario
231 231 # where runlines itself has an unhandled exception. We need to
232 232 # uniformize this, for all exception construction to come from a
233 233 # single location in the codbase.
234 234 etype, evalue, tb = sys.exc_info()
235 235 tb_list = traceback.format_exception(etype, evalue, tb)
236 236 reply_content.update(shell._showtraceback(etype, evalue, tb_list))
237 237 else:
238 238 status = u'ok'
239 239
240 240 reply_content[u'status'] = status
241 241
242 242 # Return the execution counter so clients can display prompts
243 243 reply_content['execution_count'] = shell.execution_count -1
244 244
245 245 # FIXME - fish exception info out of shell, possibly left there by
246 246 # runlines. We'll need to clean up this logic later.
247 247 if shell._reply_content is not None:
248 248 reply_content.update(shell._reply_content)
249 249
250 250 # At this point, we can tell whether the main code execution succeeded
251 251 # or not. If it did, we proceed to evaluate user_variables/expressions
252 252 if reply_content['status'] == 'ok':
253 253 reply_content[u'user_variables'] = \
254 254 shell.user_variables(content[u'user_variables'])
255 255 reply_content[u'user_expressions'] = \
256 256 shell.user_expressions(content[u'user_expressions'])
257 257 else:
258 258 # If there was an error, don't even try to compute variables or
259 259 # expressions
260 260 reply_content[u'user_variables'] = {}
261 261 reply_content[u'user_expressions'] = {}
262 262
263 263 # Payloads should be retrieved regardless of outcome, so we can both
264 264 # recover partial output (that could have been generated early in a
265 265 # block, before an error) and clear the payload system always.
266 266 reply_content[u'payload'] = shell.payload_manager.read_payload()
267 267 # Be agressive about clearing the payload because we don't want
268 268 # it to sit in memory until the next execute_request comes in.
269 269 shell.payload_manager.clear_payload()
270 270
271 271 # Flush output before sending the reply.
272 272 sys.stdout.flush()
273 273 sys.stderr.flush()
274 274 # FIXME: on rare occasions, the flush doesn't seem to make it to the
275 275 # clients... This seems to mitigate the problem, but we definitely need
276 276 # to better understand what's going on.
277 277 if self._execute_sleep:
278 278 time.sleep(self._execute_sleep)
279 279
280 280 # Send the reply.
281 281 reply_msg = self.session.send(self.reply_socket, u'execute_reply',
282 282 reply_content, parent, ident=ident)
283 283 logger.debug(str(reply_msg))
284 284
285 285 if reply_msg['content']['status'] == u'error':
286 286 self._abort_queue()
287 287
288 288 status_msg = self.session.send(self.pub_socket,
289 289 u'status',
290 290 {u'execution_state':u'idle'},
291 291 parent=parent
292 292 )
293 293
294 294 def complete_request(self, ident, parent):
295 295 txt, matches = self._complete(parent)
296 296 matches = {'matches' : matches,
297 297 'matched_text' : txt,
298 298 'status' : 'ok'}
299 299 completion_msg = self.session.send(self.reply_socket, 'complete_reply',
300 300 matches, parent, ident)
301 301 logger.debug(str(completion_msg))
302 302
303 303 def object_info_request(self, ident, parent):
304 304 object_info = self.shell.object_inspect(parent['content']['oname'])
305 305 # Before we send this object over, we scrub it for JSON usage
306 306 oinfo = json_clean(object_info)
307 307 msg = self.session.send(self.reply_socket, 'object_info_reply',
308 308 oinfo, parent, ident)
309 309 logger.debug(msg)
310 310
311 311 def history_tail_request(self, ident, parent):
312 312 # We need to pull these out, as passing **kwargs doesn't work with
313 313 # unicode keys before Python 2.6.5.
314 314 n = parent['content']['n']
315 315 raw = parent['content']['raw']
316 316 output = parent['content']['output']
317 hist = self.shell.history_manager.get_hist_tail(n, raw=raw, output=output)
317 hist = self.shell.history_manager.get_tail(n, raw=raw, output=output)
318 318 content = {'history' : list(hist)}
319 319 msg = self.session.send(self.reply_socket, 'history_tail_reply',
320 320 content, parent, ident)
321 321 logger.debug(str(msg))
322 322
323 323 def connect_request(self, ident, parent):
324 324 if self._recorded_ports is not None:
325 325 content = self._recorded_ports.copy()
326 326 else:
327 327 content = {}
328 328 msg = self.session.send(self.reply_socket, 'connect_reply',
329 329 content, parent, ident)
330 330 logger.debug(msg)
331 331
332 332 def shutdown_request(self, ident, parent):
333 333 self.shell.exit_now = True
334 334 self._shutdown_message = self.session.msg(u'shutdown_reply', parent['content'], parent)
335 335 sys.exit(0)
336 336
337 337 #---------------------------------------------------------------------------
338 338 # Protected interface
339 339 #---------------------------------------------------------------------------
340 340
341 341 def _abort_queue(self):
342 342 while True:
343 343 ident,msg = self.session.recv(self.reply_socket, zmq.NOBLOCK)
344 344 if msg is None:
345 345 break
346 346 else:
347 347 assert ident is not None, \
348 348 "Unexpected missing message part."
349 349
350 350 logger.debug("Aborting:\n"+str(Message(msg)))
351 351 msg_type = msg['msg_type']
352 352 reply_type = msg_type.split('_')[0] + '_reply'
353 353 reply_msg = self.session.send(self.reply_socket, reply_type,
354 354 {'status' : 'aborted'}, msg, ident=ident)
355 355 logger.debug(reply_msg)
356 356 # We need to wait a bit for requests to come in. This can probably
357 357 # be set shorter for true asynchronous clients.
358 358 time.sleep(0.1)
359 359
360 360 def _raw_input(self, prompt, ident, parent):
361 361 # Flush output before making the request.
362 362 sys.stderr.flush()
363 363 sys.stdout.flush()
364 364
365 365 # Send the input request.
366 366 content = dict(prompt=prompt)
367 367 msg = self.session.send(self.req_socket, u'input_request', content, parent)
368 368
369 369 # Await a response.
370 370 ident, reply = self.session.recv(self.req_socket, 0)
371 371 try:
372 372 value = reply['content']['value']
373 373 except:
374 374 logger.error("Got bad raw_input reply: ")
375 375 logger.error(str(Message(parent)))
376 376 value = ''
377 377 return value
378 378
379 379 def _complete(self, msg):
380 380 c = msg['content']
381 381 try:
382 382 cpos = int(c['cursor_pos'])
383 383 except:
384 384 # If we don't get something that we can convert to an integer, at
385 385 # least attempt the completion guessing the cursor is at the end of
386 386 # the text, if there's any, and otherwise of the line
387 387 cpos = len(c['text'])
388 388 if cpos==0:
389 389 cpos = len(c['line'])
390 390 return self.shell.complete(c['text'], c['line'], cpos)
391 391
392 392 def _object_info(self, context):
393 393 symbol, leftover = self._symbol_from_context(context)
394 394 if symbol is not None and not leftover:
395 395 doc = getattr(symbol, '__doc__', '')
396 396 else:
397 397 doc = ''
398 398 object_info = dict(docstring = doc)
399 399 return object_info
400 400
401 401 def _symbol_from_context(self, context):
402 402 if not context:
403 403 return None, context
404 404
405 405 base_symbol_string = context[0]
406 406 symbol = self.shell.user_ns.get(base_symbol_string, None)
407 407 if symbol is None:
408 408 symbol = __builtin__.__dict__.get(base_symbol_string, None)
409 409 if symbol is None:
410 410 return None, context
411 411
412 412 context = context[1:]
413 413 for i, name in enumerate(context):
414 414 new_symbol = getattr(symbol, name, None)
415 415 if new_symbol is None:
416 416 return symbol, context[i:]
417 417 else:
418 418 symbol = new_symbol
419 419
420 420 return symbol, []
421 421
422 422 def _at_shutdown(self):
423 423 """Actions taken at shutdown by the kernel, called by python's atexit.
424 424 """
425 425 # io.rprint("Kernel at_shutdown") # dbg
426 426 if self._shutdown_message is not None:
427 427 self.session.send(self.reply_socket, self._shutdown_message)
428 428 self.session.send(self.pub_socket, self._shutdown_message)
429 429 logger.debug(str(self._shutdown_message))
430 430 # A very short sleep to give zmq time to flush its message buffers
431 431 # before Python truly shuts down.
432 432 time.sleep(0.01)
433 433
434 434
435 435 class QtKernel(Kernel):
436 436 """A Kernel subclass with Qt support."""
437 437
438 438 def start(self):
439 439 """Start a kernel with QtPy4 event loop integration."""
440 440
441 441 from PyQt4 import QtCore
442 442 from IPython.lib.guisupport import get_app_qt4, start_event_loop_qt4
443 443
444 444 self.app = get_app_qt4([" "])
445 445 self.app.setQuitOnLastWindowClosed(False)
446 446 self.timer = QtCore.QTimer()
447 447 self.timer.timeout.connect(self.do_one_iteration)
448 448 # Units for the timer are in milliseconds
449 449 self.timer.start(1000*self._poll_interval)
450 450 start_event_loop_qt4(self.app)
451 451
452 452
453 453 class WxKernel(Kernel):
454 454 """A Kernel subclass with Wx support."""
455 455
456 456 def start(self):
457 457 """Start a kernel with wx event loop support."""
458 458
459 459 import wx
460 460 from IPython.lib.guisupport import start_event_loop_wx
461 461
462 462 doi = self.do_one_iteration
463 463 # Wx uses milliseconds
464 464 poll_interval = int(1000*self._poll_interval)
465 465
466 466 # We have to put the wx.Timer in a wx.Frame for it to fire properly.
467 467 # We make the Frame hidden when we create it in the main app below.
468 468 class TimerFrame(wx.Frame):
469 469 def __init__(self, func):
470 470 wx.Frame.__init__(self, None, -1)
471 471 self.timer = wx.Timer(self)
472 472 # Units for the timer are in milliseconds
473 473 self.timer.Start(poll_interval)
474 474 self.Bind(wx.EVT_TIMER, self.on_timer)
475 475 self.func = func
476 476
477 477 def on_timer(self, event):
478 478 self.func()
479 479
480 480 # We need a custom wx.App to create our Frame subclass that has the
481 481 # wx.Timer to drive the ZMQ event loop.
482 482 class IPWxApp(wx.App):
483 483 def OnInit(self):
484 484 self.frame = TimerFrame(doi)
485 485 self.frame.Show(False)
486 486 return True
487 487
488 488 # The redirect=False here makes sure that wx doesn't replace
489 489 # sys.stdout/stderr with its own classes.
490 490 self.app = IPWxApp(redirect=False)
491 491 start_event_loop_wx(self.app)
492 492
493 493
494 494 class TkKernel(Kernel):
495 495 """A Kernel subclass with Tk support."""
496 496
497 497 def start(self):
498 498 """Start a Tk enabled event loop."""
499 499
500 500 import Tkinter
501 501 doi = self.do_one_iteration
502 502 # Tk uses milliseconds
503 503 poll_interval = int(1000*self._poll_interval)
504 504 # For Tkinter, we create a Tk object and call its withdraw method.
505 505 class Timer(object):
506 506 def __init__(self, func):
507 507 self.app = Tkinter.Tk()
508 508 self.app.withdraw()
509 509 self.func = func
510 510
511 511 def on_timer(self):
512 512 self.func()
513 513 self.app.after(poll_interval, self.on_timer)
514 514
515 515 def start(self):
516 516 self.on_timer() # Call it once to get things going.
517 517 self.app.mainloop()
518 518
519 519 self.timer = Timer(doi)
520 520 self.timer.start()
521 521
522 522
523 523 class GTKKernel(Kernel):
524 524 """A Kernel subclass with GTK support."""
525 525
526 526 def start(self):
527 527 """Start the kernel, coordinating with the GTK event loop"""
528 528 from .gui.gtkembed import GTKEmbed
529 529
530 530 gtk_kernel = GTKEmbed(self)
531 531 gtk_kernel.start()
532 532
533 533
534 534 #-----------------------------------------------------------------------------
535 535 # Kernel main and launch functions
536 536 #-----------------------------------------------------------------------------
537 537
538 538 def launch_kernel(ip=None, xrep_port=0, pub_port=0, req_port=0, hb_port=0,
539 539 independent=False, pylab=False, colors=None):
540 540 """Launches a localhost kernel, binding to the specified ports.
541 541
542 542 Parameters
543 543 ----------
544 544 ip : str, optional
545 545 The ip address the kernel will bind to.
546 546
547 547 xrep_port : int, optional
548 548 The port to use for XREP channel.
549 549
550 550 pub_port : int, optional
551 551 The port to use for the SUB channel.
552 552
553 553 req_port : int, optional
554 554 The port to use for the REQ (raw input) channel.
555 555
556 556 hb_port : int, optional
557 557 The port to use for the hearbeat REP channel.
558 558
559 559 independent : bool, optional (default False)
560 560 If set, the kernel process is guaranteed to survive if this process
561 561 dies. If not set, an effort is made to ensure that the kernel is killed
562 562 when this process dies. Note that in this case it is still good practice
563 563 to kill kernels manually before exiting.
564 564
565 565 pylab : bool or string, optional (default False)
566 566 If not False, the kernel will be launched with pylab enabled. If a
567 567 string is passed, matplotlib will use the specified backend. Otherwise,
568 568 matplotlib's default backend will be used.
569 569
570 570 colors : None or string, optional (default None)
571 571 If not None, specify the color scheme. One of (NoColor, LightBG, Linux)
572 572
573 573 Returns
574 574 -------
575 575 A tuple of form:
576 576 (kernel_process, xrep_port, pub_port, req_port)
577 577 where kernel_process is a Popen object and the ports are integers.
578 578 """
579 579 extra_arguments = []
580 580 if pylab:
581 581 extra_arguments.append('--pylab')
582 582 if isinstance(pylab, basestring):
583 583 extra_arguments.append(pylab)
584 584 if ip is not None:
585 585 extra_arguments.append('--ip')
586 586 if isinstance(ip, basestring):
587 587 extra_arguments.append(ip)
588 588 if colors is not None:
589 589 extra_arguments.append('--colors')
590 590 extra_arguments.append(colors)
591 591 return base_launch_kernel('from IPython.zmq.ipkernel import main; main()',
592 592 xrep_port, pub_port, req_port, hb_port,
593 593 independent, extra_arguments)
594 594
595 595
596 596 def main():
597 597 """ The IPython kernel main entry point.
598 598 """
599 599 parser = make_argument_parser()
600 600 parser.add_argument('--pylab', type=str, metavar='GUI', nargs='?',
601 601 const='auto', help = \
602 602 "Pre-load matplotlib and numpy for interactive use. If GUI is not \
603 603 given, the GUI backend is matplotlib's, otherwise use one of: \
604 604 ['tk', 'gtk', 'qt', 'wx', 'inline'].")
605 605 parser.add_argument('--colors',
606 606 type=str, dest='colors',
607 607 help="Set the color scheme (NoColor, Linux, and LightBG).",
608 608 metavar='ZMQInteractiveShell.colors')
609 609 namespace = parser.parse_args()
610 610
611 611 kernel_class = Kernel
612 612
613 613 kernel_classes = {
614 614 'qt' : QtKernel,
615 615 'qt4': QtKernel,
616 616 'inline': Kernel,
617 617 'wx' : WxKernel,
618 618 'tk' : TkKernel,
619 619 'gtk': GTKKernel,
620 620 }
621 621 if namespace.pylab:
622 622 if namespace.pylab == 'auto':
623 623 gui, backend = pylabtools.find_gui_and_backend()
624 624 else:
625 625 gui, backend = pylabtools.find_gui_and_backend(namespace.pylab)
626 626 kernel_class = kernel_classes.get(gui)
627 627 if kernel_class is None:
628 628 raise ValueError('GUI is not supported: %r' % gui)
629 629 pylabtools.activate_matplotlib(backend)
630 630 if namespace.colors:
631 631 ZMQInteractiveShell.colors=namespace.colors
632 632
633 633 kernel = make_kernel(namespace, kernel_class, OutStream)
634 634
635 635 if namespace.pylab:
636 636 pylabtools.import_pylab(kernel.shell.user_ns, backend,
637 637 shell=kernel.shell)
638 638
639 639 start_kernel(namespace, kernel)
640 640
641 641
642 642 if __name__ == '__main__':
643 643 main()
General Comments 0
You need to be logged in to leave comments. Login now