##// END OF EJS Templates
Initial implementation of %pastebin magic command.
Thomas Kluyver -
Show More
@@ -1,3456 +1,3476 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 from cStringIO import StringIO
29 29 from getopt import getopt,GetoptError
30 30 from pprint import pformat
31 31
32 32 # cProfile was added in Python2.5
33 33 try:
34 34 import cProfile as profile
35 35 import pstats
36 36 except ImportError:
37 37 # profile isn't bundled by default in Debian for license reasons
38 38 try:
39 39 import profile,pstats
40 40 except ImportError:
41 41 profile = pstats = None
42 42
43 43 import IPython
44 44 from IPython.core import debugger, oinspect
45 45 from IPython.core.error import TryNext
46 46 from IPython.core.error import UsageError
47 47 from IPython.core.fakemodule import FakeModule
48 48 from IPython.core.macro import Macro
49 49 from IPython.core import page
50 50 from IPython.core.prefilter import ESC_MAGIC
51 51 from IPython.lib.pylabtools import mpl_runner
52 52 from IPython.external.Itpl import itpl, printpl
53 53 from IPython.testing import decorators as testdec
54 54 from IPython.utils.io import file_read, nlprint
55 55 import IPython.utils.io
56 56 from IPython.utils.path import get_py_filename
57 57 from IPython.utils.process import arg_split, abbrev_cwd
58 58 from IPython.utils.terminal import set_term_title
59 59 from IPython.utils.text import LSString, SList, format_screen
60 60 from IPython.utils.timing import clock, clock2
61 61 from IPython.utils.warn import warn, error
62 62 from IPython.utils.ipstruct import Struct
63 63 import IPython.utils.generics
64 64
65 65 #-----------------------------------------------------------------------------
66 66 # Utility functions
67 67 #-----------------------------------------------------------------------------
68 68
69 69 def on_off(tag):
70 70 """Return an ON/OFF string for a 1/0 input. Simple utility function."""
71 71 return ['OFF','ON'][tag]
72 72
73 73 class Bunch: pass
74 74
75 75 def compress_dhist(dh):
76 76 head, tail = dh[:-10], dh[-10:]
77 77
78 78 newhead = []
79 79 done = set()
80 80 for h in head:
81 81 if h in done:
82 82 continue
83 83 newhead.append(h)
84 84 done.add(h)
85 85
86 86 return newhead + tail
87 87
88 88 def needs_local_scope(func):
89 89 """Decorator to mark magic functions which need to local scope to run."""
90 90 func.needs_local_scope = True
91 91 return func
92 92
93 93 #***************************************************************************
94 94 # Main class implementing Magic functionality
95 95
96 96 # XXX - for some odd reason, if Magic is made a new-style class, we get errors
97 97 # on construction of the main InteractiveShell object. Something odd is going
98 98 # on with super() calls, Configurable and the MRO... For now leave it as-is, but
99 99 # eventually this needs to be clarified.
100 100 # BG: This is because InteractiveShell inherits from this, but is itself a
101 101 # Configurable. This messes up the MRO in some way. The fix is that we need to
102 102 # make Magic a configurable that InteractiveShell does not subclass.
103 103
104 104 class Magic:
105 105 """Magic functions for InteractiveShell.
106 106
107 107 Shell functions which can be reached as %function_name. All magic
108 108 functions should accept a string, which they can parse for their own
109 109 needs. This can make some functions easier to type, eg `%cd ../`
110 110 vs. `%cd("../")`
111 111
112 112 ALL definitions MUST begin with the prefix magic_. The user won't need it
113 113 at the command line, but it is is needed in the definition. """
114 114
115 115 # class globals
116 116 auto_status = ['Automagic is OFF, % prefix IS needed for magic functions.',
117 117 'Automagic is ON, % prefix NOT needed for magic functions.']
118 118
119 119 #......................................................................
120 120 # some utility functions
121 121
122 122 def __init__(self,shell):
123 123
124 124 self.options_table = {}
125 125 if profile is None:
126 126 self.magic_prun = self.profile_missing_notice
127 127 self.shell = shell
128 128
129 129 # namespace for holding state we may need
130 130 self._magic_state = Bunch()
131 131
132 132 def profile_missing_notice(self, *args, **kwargs):
133 133 error("""\
134 134 The profile module could not be found. It has been removed from the standard
135 135 python packages because of its non-free license. To use profiling, install the
136 136 python-profiler package from non-free.""")
137 137
138 138 def default_option(self,fn,optstr):
139 139 """Make an entry in the options_table for fn, with value optstr"""
140 140
141 141 if fn not in self.lsmagic():
142 142 error("%s is not a magic function" % fn)
143 143 self.options_table[fn] = optstr
144 144
145 145 def lsmagic(self):
146 146 """Return a list of currently available magic functions.
147 147
148 148 Gives a list of the bare names after mangling (['ls','cd', ...], not
149 149 ['magic_ls','magic_cd',...]"""
150 150
151 151 # FIXME. This needs a cleanup, in the way the magics list is built.
152 152
153 153 # magics in class definition
154 154 class_magic = lambda fn: fn.startswith('magic_') and \
155 155 callable(Magic.__dict__[fn])
156 156 # in instance namespace (run-time user additions)
157 157 inst_magic = lambda fn: fn.startswith('magic_') and \
158 158 callable(self.__dict__[fn])
159 159 # and bound magics by user (so they can access self):
160 160 inst_bound_magic = lambda fn: fn.startswith('magic_') and \
161 161 callable(self.__class__.__dict__[fn])
162 162 magics = filter(class_magic,Magic.__dict__.keys()) + \
163 163 filter(inst_magic,self.__dict__.keys()) + \
164 164 filter(inst_bound_magic,self.__class__.__dict__.keys())
165 165 out = []
166 166 for fn in set(magics):
167 167 out.append(fn.replace('magic_','',1))
168 168 out.sort()
169 169 return out
170 170
171 171 def extract_input_lines(self, range_str, raw=False):
172 172 """Return as a string a set of input history slices.
173 173
174 174 Inputs:
175 175
176 176 - range_str: the set of slices is given as a string, like
177 177 "~5/6-~4/2 4:8 9", since this function is for use by magic functions
178 178 which get their arguments as strings. The number before the / is the
179 179 session number: ~n goes n back from the current session.
180 180
181 181 Optional inputs:
182 182
183 183 - raw(False): by default, the processed input is used. If this is
184 184 true, the raw input history is used instead.
185 185
186 186 Note that slices can be called with two notations:
187 187
188 188 N:M -> standard python form, means including items N...(M-1).
189 189
190 190 N-M -> include items N..M (closed endpoint)."""
191 191 lines = self.shell.history_manager.\
192 192 get_range_by_str(range_str, raw=raw)
193 193 return "\n".join(x for _, _, x in lines)
194 194
195 195 def arg_err(self,func):
196 196 """Print docstring if incorrect arguments were passed"""
197 197 print 'Error in arguments:'
198 198 print oinspect.getdoc(func)
199 199
200 200 def format_latex(self,strng):
201 201 """Format a string for latex inclusion."""
202 202
203 203 # Characters that need to be escaped for latex:
204 204 escape_re = re.compile(r'(%|_|\$|#|&)',re.MULTILINE)
205 205 # Magic command names as headers:
206 206 cmd_name_re = re.compile(r'^(%s.*?):' % ESC_MAGIC,
207 207 re.MULTILINE)
208 208 # Magic commands
209 209 cmd_re = re.compile(r'(?P<cmd>%s.+?\b)(?!\}\}:)' % ESC_MAGIC,
210 210 re.MULTILINE)
211 211 # Paragraph continue
212 212 par_re = re.compile(r'\\$',re.MULTILINE)
213 213
214 214 # The "\n" symbol
215 215 newline_re = re.compile(r'\\n')
216 216
217 217 # Now build the string for output:
218 218 #strng = cmd_name_re.sub(r'\n\\texttt{\\textsl{\\large \1}}:',strng)
219 219 strng = cmd_name_re.sub(r'\n\\bigskip\n\\texttt{\\textbf{ \1}}:',
220 220 strng)
221 221 strng = cmd_re.sub(r'\\texttt{\g<cmd>}',strng)
222 222 strng = par_re.sub(r'\\\\',strng)
223 223 strng = escape_re.sub(r'\\\1',strng)
224 224 strng = newline_re.sub(r'\\textbackslash{}n',strng)
225 225 return strng
226 226
227 227 def parse_options(self,arg_str,opt_str,*long_opts,**kw):
228 228 """Parse options passed to an argument string.
229 229
230 230 The interface is similar to that of getopt(), but it returns back a
231 231 Struct with the options as keys and the stripped argument string still
232 232 as a string.
233 233
234 234 arg_str is quoted as a true sys.argv vector by using shlex.split.
235 235 This allows us to easily expand variables, glob files, quote
236 236 arguments, etc.
237 237
238 238 Options:
239 239 -mode: default 'string'. If given as 'list', the argument string is
240 240 returned as a list (split on whitespace) instead of a string.
241 241
242 242 -list_all: put all option values in lists. Normally only options
243 243 appearing more than once are put in a list.
244 244
245 245 -posix (True): whether to split the input line in POSIX mode or not,
246 246 as per the conventions outlined in the shlex module from the
247 247 standard library."""
248 248
249 249 # inject default options at the beginning of the input line
250 250 caller = sys._getframe(1).f_code.co_name.replace('magic_','')
251 251 arg_str = '%s %s' % (self.options_table.get(caller,''),arg_str)
252 252
253 253 mode = kw.get('mode','string')
254 254 if mode not in ['string','list']:
255 255 raise ValueError,'incorrect mode given: %s' % mode
256 256 # Get options
257 257 list_all = kw.get('list_all',0)
258 258 posix = kw.get('posix', os.name == 'posix')
259 259
260 260 # Check if we have more than one argument to warrant extra processing:
261 261 odict = {} # Dictionary with options
262 262 args = arg_str.split()
263 263 if len(args) >= 1:
264 264 # If the list of inputs only has 0 or 1 thing in it, there's no
265 265 # need to look for options
266 266 argv = arg_split(arg_str,posix)
267 267 # Do regular option processing
268 268 try:
269 269 opts,args = getopt(argv,opt_str,*long_opts)
270 270 except GetoptError,e:
271 271 raise UsageError('%s ( allowed: "%s" %s)' % (e.msg,opt_str,
272 272 " ".join(long_opts)))
273 273 for o,a in opts:
274 274 if o.startswith('--'):
275 275 o = o[2:]
276 276 else:
277 277 o = o[1:]
278 278 try:
279 279 odict[o].append(a)
280 280 except AttributeError:
281 281 odict[o] = [odict[o],a]
282 282 except KeyError:
283 283 if list_all:
284 284 odict[o] = [a]
285 285 else:
286 286 odict[o] = a
287 287
288 288 # Prepare opts,args for return
289 289 opts = Struct(odict)
290 290 if mode == 'string':
291 291 args = ' '.join(args)
292 292
293 293 return opts,args
294 294
295 295 #......................................................................
296 296 # And now the actual magic functions
297 297
298 298 # Functions for IPython shell work (vars,funcs, config, etc)
299 299 def magic_lsmagic(self, parameter_s = ''):
300 300 """List currently available magic functions."""
301 301 mesc = ESC_MAGIC
302 302 print 'Available magic functions:\n'+mesc+\
303 303 (' '+mesc).join(self.lsmagic())
304 304 print '\n' + Magic.auto_status[self.shell.automagic]
305 305 return None
306 306
307 307 def magic_magic(self, parameter_s = ''):
308 308 """Print information about the magic function system.
309 309
310 310 Supported formats: -latex, -brief, -rest
311 311 """
312 312
313 313 mode = ''
314 314 try:
315 315 if parameter_s.split()[0] == '-latex':
316 316 mode = 'latex'
317 317 if parameter_s.split()[0] == '-brief':
318 318 mode = 'brief'
319 319 if parameter_s.split()[0] == '-rest':
320 320 mode = 'rest'
321 321 rest_docs = []
322 322 except:
323 323 pass
324 324
325 325 magic_docs = []
326 326 for fname in self.lsmagic():
327 327 mname = 'magic_' + fname
328 328 for space in (Magic,self,self.__class__):
329 329 try:
330 330 fn = space.__dict__[mname]
331 331 except KeyError:
332 332 pass
333 333 else:
334 334 break
335 335 if mode == 'brief':
336 336 # only first line
337 337 if fn.__doc__:
338 338 fndoc = fn.__doc__.split('\n',1)[0]
339 339 else:
340 340 fndoc = 'No documentation'
341 341 else:
342 342 if fn.__doc__:
343 343 fndoc = fn.__doc__.rstrip()
344 344 else:
345 345 fndoc = 'No documentation'
346 346
347 347
348 348 if mode == 'rest':
349 349 rest_docs.append('**%s%s**::\n\n\t%s\n\n' %(ESC_MAGIC,
350 350 fname,fndoc))
351 351
352 352 else:
353 353 magic_docs.append('%s%s:\n\t%s\n' %(ESC_MAGIC,
354 354 fname,fndoc))
355 355
356 356 magic_docs = ''.join(magic_docs)
357 357
358 358 if mode == 'rest':
359 359 return "".join(rest_docs)
360 360
361 361 if mode == 'latex':
362 362 print self.format_latex(magic_docs)
363 363 return
364 364 else:
365 365 magic_docs = format_screen(magic_docs)
366 366 if mode == 'brief':
367 367 return magic_docs
368 368
369 369 outmsg = """
370 370 IPython's 'magic' functions
371 371 ===========================
372 372
373 373 The magic function system provides a series of functions which allow you to
374 374 control the behavior of IPython itself, plus a lot of system-type
375 375 features. All these functions are prefixed with a % character, but parameters
376 376 are given without parentheses or quotes.
377 377
378 378 NOTE: If you have 'automagic' enabled (via the command line option or with the
379 379 %automagic function), you don't need to type in the % explicitly. By default,
380 380 IPython ships with automagic on, so you should only rarely need the % escape.
381 381
382 382 Example: typing '%cd mydir' (without the quotes) changes you working directory
383 383 to 'mydir', if it exists.
384 384
385 385 You can define your own magic functions to extend the system. See the supplied
386 386 ipythonrc and example-magic.py files for details (in your ipython
387 387 configuration directory, typically $HOME/.config/ipython on Linux or $HOME/.ipython elsewhere).
388 388
389 389 You can also define your own aliased names for magic functions. In your
390 390 ipythonrc file, placing a line like:
391 391
392 392 execute __IPYTHON__.magic_pf = __IPYTHON__.magic_profile
393 393
394 394 will define %pf as a new name for %profile.
395 395
396 396 You can also call magics in code using the magic() function, which IPython
397 397 automatically adds to the builtin namespace. Type 'magic?' for details.
398 398
399 399 For a list of the available magic functions, use %lsmagic. For a description
400 400 of any of them, type %magic_name?, e.g. '%cd?'.
401 401
402 402 Currently the magic system has the following functions:\n"""
403 403
404 404 mesc = ESC_MAGIC
405 405 outmsg = ("%s\n%s\n\nSummary of magic functions (from %slsmagic):"
406 406 "\n\n%s%s\n\n%s" % (outmsg,
407 407 magic_docs,mesc,mesc,
408 408 (' '+mesc).join(self.lsmagic()),
409 409 Magic.auto_status[self.shell.automagic] ) )
410 410 page.page(outmsg)
411 411
412 412 def magic_automagic(self, parameter_s = ''):
413 413 """Make magic functions callable without having to type the initial %.
414 414
415 415 Without argumentsl toggles on/off (when off, you must call it as
416 416 %automagic, of course). With arguments it sets the value, and you can
417 417 use any of (case insensitive):
418 418
419 419 - on,1,True: to activate
420 420
421 421 - off,0,False: to deactivate.
422 422
423 423 Note that magic functions have lowest priority, so if there's a
424 424 variable whose name collides with that of a magic fn, automagic won't
425 425 work for that function (you get the variable instead). However, if you
426 426 delete the variable (del var), the previously shadowed magic function
427 427 becomes visible to automagic again."""
428 428
429 429 arg = parameter_s.lower()
430 430 if parameter_s in ('on','1','true'):
431 431 self.shell.automagic = True
432 432 elif parameter_s in ('off','0','false'):
433 433 self.shell.automagic = False
434 434 else:
435 435 self.shell.automagic = not self.shell.automagic
436 436 print '\n' + Magic.auto_status[self.shell.automagic]
437 437
438 438 @testdec.skip_doctest
439 439 def magic_autocall(self, parameter_s = ''):
440 440 """Make functions callable without having to type parentheses.
441 441
442 442 Usage:
443 443
444 444 %autocall [mode]
445 445
446 446 The mode can be one of: 0->Off, 1->Smart, 2->Full. If not given, the
447 447 value is toggled on and off (remembering the previous state).
448 448
449 449 In more detail, these values mean:
450 450
451 451 0 -> fully disabled
452 452
453 453 1 -> active, but do not apply if there are no arguments on the line.
454 454
455 455 In this mode, you get:
456 456
457 457 In [1]: callable
458 458 Out[1]: <built-in function callable>
459 459
460 460 In [2]: callable 'hello'
461 461 ------> callable('hello')
462 462 Out[2]: False
463 463
464 464 2 -> Active always. Even if no arguments are present, the callable
465 465 object is called:
466 466
467 467 In [2]: float
468 468 ------> float()
469 469 Out[2]: 0.0
470 470
471 471 Note that even with autocall off, you can still use '/' at the start of
472 472 a line to treat the first argument on the command line as a function
473 473 and add parentheses to it:
474 474
475 475 In [8]: /str 43
476 476 ------> str(43)
477 477 Out[8]: '43'
478 478
479 479 # all-random (note for auto-testing)
480 480 """
481 481
482 482 if parameter_s:
483 483 arg = int(parameter_s)
484 484 else:
485 485 arg = 'toggle'
486 486
487 487 if not arg in (0,1,2,'toggle'):
488 488 error('Valid modes: (0->Off, 1->Smart, 2->Full')
489 489 return
490 490
491 491 if arg in (0,1,2):
492 492 self.shell.autocall = arg
493 493 else: # toggle
494 494 if self.shell.autocall:
495 495 self._magic_state.autocall_save = self.shell.autocall
496 496 self.shell.autocall = 0
497 497 else:
498 498 try:
499 499 self.shell.autocall = self._magic_state.autocall_save
500 500 except AttributeError:
501 501 self.shell.autocall = self._magic_state.autocall_save = 1
502 502
503 503 print "Automatic calling is:",['OFF','Smart','Full'][self.shell.autocall]
504 504
505 505
506 506 def magic_page(self, parameter_s=''):
507 507 """Pretty print the object and display it through a pager.
508 508
509 509 %page [options] OBJECT
510 510
511 511 If no object is given, use _ (last output).
512 512
513 513 Options:
514 514
515 515 -r: page str(object), don't pretty-print it."""
516 516
517 517 # After a function contributed by Olivier Aubert, slightly modified.
518 518
519 519 # Process options/args
520 520 opts,args = self.parse_options(parameter_s,'r')
521 521 raw = 'r' in opts
522 522
523 523 oname = args and args or '_'
524 524 info = self._ofind(oname)
525 525 if info['found']:
526 526 txt = (raw and str or pformat)( info['obj'] )
527 527 page.page(txt)
528 528 else:
529 529 print 'Object `%s` not found' % oname
530 530
531 531 def magic_profile(self, parameter_s=''):
532 532 """Print your currently active IPython profile."""
533 533 if self.shell.profile:
534 534 printpl('Current IPython profile: $self.shell.profile.')
535 535 else:
536 536 print 'No profile active.'
537 537
538 538 def magic_pinfo(self, parameter_s='', namespaces=None):
539 539 """Provide detailed information about an object.
540 540
541 541 '%pinfo object' is just a synonym for object? or ?object."""
542 542
543 543 #print 'pinfo par: <%s>' % parameter_s # dbg
544 544
545 545
546 546 # detail_level: 0 -> obj? , 1 -> obj??
547 547 detail_level = 0
548 548 # We need to detect if we got called as 'pinfo pinfo foo', which can
549 549 # happen if the user types 'pinfo foo?' at the cmd line.
550 550 pinfo,qmark1,oname,qmark2 = \
551 551 re.match('(pinfo )?(\?*)(.*?)(\??$)',parameter_s).groups()
552 552 if pinfo or qmark1 or qmark2:
553 553 detail_level = 1
554 554 if "*" in oname:
555 555 self.magic_psearch(oname)
556 556 else:
557 557 self.shell._inspect('pinfo', oname, detail_level=detail_level,
558 558 namespaces=namespaces)
559 559
560 560 def magic_pinfo2(self, parameter_s='', namespaces=None):
561 561 """Provide extra detailed information about an object.
562 562
563 563 '%pinfo2 object' is just a synonym for object?? or ??object."""
564 564 self.shell._inspect('pinfo', parameter_s, detail_level=1,
565 565 namespaces=namespaces)
566 566
567 567 @testdec.skip_doctest
568 568 def magic_pdef(self, parameter_s='', namespaces=None):
569 569 """Print the definition header for any callable object.
570 570
571 571 If the object is a class, print the constructor information.
572 572
573 573 Examples
574 574 --------
575 575 ::
576 576
577 577 In [3]: %pdef urllib.urlopen
578 578 urllib.urlopen(url, data=None, proxies=None)
579 579 """
580 580 self._inspect('pdef',parameter_s, namespaces)
581 581
582 582 def magic_pdoc(self, parameter_s='', namespaces=None):
583 583 """Print the docstring for an object.
584 584
585 585 If the given object is a class, it will print both the class and the
586 586 constructor docstrings."""
587 587 self._inspect('pdoc',parameter_s, namespaces)
588 588
589 589 def magic_psource(self, parameter_s='', namespaces=None):
590 590 """Print (or run through pager) the source code for an object."""
591 591 self._inspect('psource',parameter_s, namespaces)
592 592
593 593 def magic_pfile(self, parameter_s=''):
594 594 """Print (or run through pager) the file where an object is defined.
595 595
596 596 The file opens at the line where the object definition begins. IPython
597 597 will honor the environment variable PAGER if set, and otherwise will
598 598 do its best to print the file in a convenient form.
599 599
600 600 If the given argument is not an object currently defined, IPython will
601 601 try to interpret it as a filename (automatically adding a .py extension
602 602 if needed). You can thus use %pfile as a syntax highlighting code
603 603 viewer."""
604 604
605 605 # first interpret argument as an object name
606 606 out = self._inspect('pfile',parameter_s)
607 607 # if not, try the input as a filename
608 608 if out == 'not found':
609 609 try:
610 610 filename = get_py_filename(parameter_s)
611 611 except IOError,msg:
612 612 print msg
613 613 return
614 614 page.page(self.shell.inspector.format(file(filename).read()))
615 615
616 616 def magic_psearch(self, parameter_s=''):
617 617 """Search for object in namespaces by wildcard.
618 618
619 619 %psearch [options] PATTERN [OBJECT TYPE]
620 620
621 621 Note: ? can be used as a synonym for %psearch, at the beginning or at
622 622 the end: both a*? and ?a* are equivalent to '%psearch a*'. Still, the
623 623 rest of the command line must be unchanged (options come first), so
624 624 for example the following forms are equivalent
625 625
626 626 %psearch -i a* function
627 627 -i a* function?
628 628 ?-i a* function
629 629
630 630 Arguments:
631 631
632 632 PATTERN
633 633
634 634 where PATTERN is a string containing * as a wildcard similar to its
635 635 use in a shell. The pattern is matched in all namespaces on the
636 636 search path. By default objects starting with a single _ are not
637 637 matched, many IPython generated objects have a single
638 638 underscore. The default is case insensitive matching. Matching is
639 639 also done on the attributes of objects and not only on the objects
640 640 in a module.
641 641
642 642 [OBJECT TYPE]
643 643
644 644 Is the name of a python type from the types module. The name is
645 645 given in lowercase without the ending type, ex. StringType is
646 646 written string. By adding a type here only objects matching the
647 647 given type are matched. Using all here makes the pattern match all
648 648 types (this is the default).
649 649
650 650 Options:
651 651
652 652 -a: makes the pattern match even objects whose names start with a
653 653 single underscore. These names are normally ommitted from the
654 654 search.
655 655
656 656 -i/-c: make the pattern case insensitive/sensitive. If neither of
657 657 these options is given, the default is read from your ipythonrc
658 658 file. The option name which sets this value is
659 659 'wildcards_case_sensitive'. If this option is not specified in your
660 660 ipythonrc file, IPython's internal default is to do a case sensitive
661 661 search.
662 662
663 663 -e/-s NAMESPACE: exclude/search a given namespace. The pattern you
664 664 specifiy can be searched in any of the following namespaces:
665 665 'builtin', 'user', 'user_global','internal', 'alias', where
666 666 'builtin' and 'user' are the search defaults. Note that you should
667 667 not use quotes when specifying namespaces.
668 668
669 669 'Builtin' contains the python module builtin, 'user' contains all
670 670 user data, 'alias' only contain the shell aliases and no python
671 671 objects, 'internal' contains objects used by IPython. The
672 672 'user_global' namespace is only used by embedded IPython instances,
673 673 and it contains module-level globals. You can add namespaces to the
674 674 search with -s or exclude them with -e (these options can be given
675 675 more than once).
676 676
677 677 Examples:
678 678
679 679 %psearch a* -> objects beginning with an a
680 680 %psearch -e builtin a* -> objects NOT in the builtin space starting in a
681 681 %psearch a* function -> all functions beginning with an a
682 682 %psearch re.e* -> objects beginning with an e in module re
683 683 %psearch r*.e* -> objects that start with e in modules starting in r
684 684 %psearch r*.* string -> all strings in modules beginning with r
685 685
686 686 Case sensitve search:
687 687
688 688 %psearch -c a* list all object beginning with lower case a
689 689
690 690 Show objects beginning with a single _:
691 691
692 692 %psearch -a _* list objects beginning with a single underscore"""
693 693 try:
694 694 parameter_s = parameter_s.encode('ascii')
695 695 except UnicodeEncodeError:
696 696 print 'Python identifiers can only contain ascii characters.'
697 697 return
698 698
699 699 # default namespaces to be searched
700 700 def_search = ['user','builtin']
701 701
702 702 # Process options/args
703 703 opts,args = self.parse_options(parameter_s,'cias:e:',list_all=True)
704 704 opt = opts.get
705 705 shell = self.shell
706 706 psearch = shell.inspector.psearch
707 707
708 708 # select case options
709 709 if opts.has_key('i'):
710 710 ignore_case = True
711 711 elif opts.has_key('c'):
712 712 ignore_case = False
713 713 else:
714 714 ignore_case = not shell.wildcards_case_sensitive
715 715
716 716 # Build list of namespaces to search from user options
717 717 def_search.extend(opt('s',[]))
718 718 ns_exclude = ns_exclude=opt('e',[])
719 719 ns_search = [nm for nm in def_search if nm not in ns_exclude]
720 720
721 721 # Call the actual search
722 722 try:
723 723 psearch(args,shell.ns_table,ns_search,
724 724 show_all=opt('a'),ignore_case=ignore_case)
725 725 except:
726 726 shell.showtraceback()
727 727
728 728 @testdec.skip_doctest
729 729 def magic_who_ls(self, parameter_s=''):
730 730 """Return a sorted list of all interactive variables.
731 731
732 732 If arguments are given, only variables of types matching these
733 733 arguments are returned.
734 734
735 735 Examples
736 736 --------
737 737
738 738 Define two variables and list them with who_ls::
739 739
740 740 In [1]: alpha = 123
741 741
742 742 In [2]: beta = 'test'
743 743
744 744 In [3]: %who_ls
745 745 Out[3]: ['alpha', 'beta']
746 746
747 747 In [4]: %who_ls int
748 748 Out[4]: ['alpha']
749 749
750 750 In [5]: %who_ls str
751 751 Out[5]: ['beta']
752 752 """
753 753
754 754 user_ns = self.shell.user_ns
755 755 internal_ns = self.shell.internal_ns
756 756 user_ns_hidden = self.shell.user_ns_hidden
757 757 out = [ i for i in user_ns
758 758 if not i.startswith('_') \
759 759 and not (i in internal_ns or i in user_ns_hidden) ]
760 760
761 761 typelist = parameter_s.split()
762 762 if typelist:
763 763 typeset = set(typelist)
764 764 out = [i for i in out if type(user_ns[i]).__name__ in typeset]
765 765
766 766 out.sort()
767 767 return out
768 768
769 769 @testdec.skip_doctest
770 770 def magic_who(self, parameter_s=''):
771 771 """Print all interactive variables, with some minimal formatting.
772 772
773 773 If any arguments are given, only variables whose type matches one of
774 774 these are printed. For example:
775 775
776 776 %who function str
777 777
778 778 will only list functions and strings, excluding all other types of
779 779 variables. To find the proper type names, simply use type(var) at a
780 780 command line to see how python prints type names. For example:
781 781
782 782 In [1]: type('hello')\\
783 783 Out[1]: <type 'str'>
784 784
785 785 indicates that the type name for strings is 'str'.
786 786
787 787 %who always excludes executed names loaded through your configuration
788 788 file and things which are internal to IPython.
789 789
790 790 This is deliberate, as typically you may load many modules and the
791 791 purpose of %who is to show you only what you've manually defined.
792 792
793 793 Examples
794 794 --------
795 795
796 796 Define two variables and list them with who::
797 797
798 798 In [1]: alpha = 123
799 799
800 800 In [2]: beta = 'test'
801 801
802 802 In [3]: %who
803 803 alpha beta
804 804
805 805 In [4]: %who int
806 806 alpha
807 807
808 808 In [5]: %who str
809 809 beta
810 810 """
811 811
812 812 varlist = self.magic_who_ls(parameter_s)
813 813 if not varlist:
814 814 if parameter_s:
815 815 print 'No variables match your requested type.'
816 816 else:
817 817 print 'Interactive namespace is empty.'
818 818 return
819 819
820 820 # if we have variables, move on...
821 821 count = 0
822 822 for i in varlist:
823 823 print i+'\t',
824 824 count += 1
825 825 if count > 8:
826 826 count = 0
827 827 print
828 828 print
829 829
830 830 @testdec.skip_doctest
831 831 def magic_whos(self, parameter_s=''):
832 832 """Like %who, but gives some extra information about each variable.
833 833
834 834 The same type filtering of %who can be applied here.
835 835
836 836 For all variables, the type is printed. Additionally it prints:
837 837
838 838 - For {},[],(): their length.
839 839
840 840 - For numpy arrays, a summary with shape, number of
841 841 elements, typecode and size in memory.
842 842
843 843 - Everything else: a string representation, snipping their middle if
844 844 too long.
845 845
846 846 Examples
847 847 --------
848 848
849 849 Define two variables and list them with whos::
850 850
851 851 In [1]: alpha = 123
852 852
853 853 In [2]: beta = 'test'
854 854
855 855 In [3]: %whos
856 856 Variable Type Data/Info
857 857 --------------------------------
858 858 alpha int 123
859 859 beta str test
860 860 """
861 861
862 862 varnames = self.magic_who_ls(parameter_s)
863 863 if not varnames:
864 864 if parameter_s:
865 865 print 'No variables match your requested type.'
866 866 else:
867 867 print 'Interactive namespace is empty.'
868 868 return
869 869
870 870 # if we have variables, move on...
871 871
872 872 # for these types, show len() instead of data:
873 873 seq_types = ['dict', 'list', 'tuple']
874 874
875 875 # for numpy/Numeric arrays, display summary info
876 876 try:
877 877 import numpy
878 878 except ImportError:
879 879 ndarray_type = None
880 880 else:
881 881 ndarray_type = numpy.ndarray.__name__
882 882 try:
883 883 import Numeric
884 884 except ImportError:
885 885 array_type = None
886 886 else:
887 887 array_type = Numeric.ArrayType.__name__
888 888
889 889 # Find all variable names and types so we can figure out column sizes
890 890 def get_vars(i):
891 891 return self.shell.user_ns[i]
892 892
893 893 # some types are well known and can be shorter
894 894 abbrevs = {'IPython.core.macro.Macro' : 'Macro'}
895 895 def type_name(v):
896 896 tn = type(v).__name__
897 897 return abbrevs.get(tn,tn)
898 898
899 899 varlist = map(get_vars,varnames)
900 900
901 901 typelist = []
902 902 for vv in varlist:
903 903 tt = type_name(vv)
904 904
905 905 if tt=='instance':
906 906 typelist.append( abbrevs.get(str(vv.__class__),
907 907 str(vv.__class__)))
908 908 else:
909 909 typelist.append(tt)
910 910
911 911 # column labels and # of spaces as separator
912 912 varlabel = 'Variable'
913 913 typelabel = 'Type'
914 914 datalabel = 'Data/Info'
915 915 colsep = 3
916 916 # variable format strings
917 917 vformat = "$vname.ljust(varwidth)$vtype.ljust(typewidth)"
918 918 vfmt_short = '$vstr[:25]<...>$vstr[-25:]'
919 919 aformat = "%s: %s elems, type `%s`, %s bytes"
920 920 # find the size of the columns to format the output nicely
921 921 varwidth = max(max(map(len,varnames)), len(varlabel)) + colsep
922 922 typewidth = max(max(map(len,typelist)), len(typelabel)) + colsep
923 923 # table header
924 924 print varlabel.ljust(varwidth) + typelabel.ljust(typewidth) + \
925 925 ' '+datalabel+'\n' + '-'*(varwidth+typewidth+len(datalabel)+1)
926 926 # and the table itself
927 927 kb = 1024
928 928 Mb = 1048576 # kb**2
929 929 for vname,var,vtype in zip(varnames,varlist,typelist):
930 930 print itpl(vformat),
931 931 if vtype in seq_types:
932 932 print "n="+str(len(var))
933 933 elif vtype in [array_type,ndarray_type]:
934 934 vshape = str(var.shape).replace(',','').replace(' ','x')[1:-1]
935 935 if vtype==ndarray_type:
936 936 # numpy
937 937 vsize = var.size
938 938 vbytes = vsize*var.itemsize
939 939 vdtype = var.dtype
940 940 else:
941 941 # Numeric
942 942 vsize = Numeric.size(var)
943 943 vbytes = vsize*var.itemsize()
944 944 vdtype = var.typecode()
945 945
946 946 if vbytes < 100000:
947 947 print aformat % (vshape,vsize,vdtype,vbytes)
948 948 else:
949 949 print aformat % (vshape,vsize,vdtype,vbytes),
950 950 if vbytes < Mb:
951 951 print '(%s kb)' % (vbytes/kb,)
952 952 else:
953 953 print '(%s Mb)' % (vbytes/Mb,)
954 954 else:
955 955 try:
956 956 vstr = str(var)
957 957 except UnicodeEncodeError:
958 958 vstr = unicode(var).encode(sys.getdefaultencoding(),
959 959 'backslashreplace')
960 960 vstr = vstr.replace('\n','\\n')
961 961 if len(vstr) < 50:
962 962 print vstr
963 963 else:
964 964 printpl(vfmt_short)
965 965
966 966 def magic_reset(self, parameter_s=''):
967 967 """Resets the namespace by removing all names defined by the user.
968 968
969 969 Input/Output history are left around in case you need them.
970 970
971 971 Parameters
972 972 ----------
973 973 -f : force reset without asking for confirmation.
974 974
975 975 Examples
976 976 --------
977 977 In [6]: a = 1
978 978
979 979 In [7]: a
980 980 Out[7]: 1
981 981
982 982 In [8]: 'a' in _ip.user_ns
983 983 Out[8]: True
984 984
985 985 In [9]: %reset -f
986 986
987 987 In [10]: 'a' in _ip.user_ns
988 988 Out[10]: False
989 989 """
990 990
991 991 if parameter_s == '-f':
992 992 ans = True
993 993 else:
994 994 ans = self.shell.ask_yes_no(
995 995 "Once deleted, variables cannot be recovered. Proceed (y/[n])? ")
996 996 if not ans:
997 997 print 'Nothing done.'
998 998 return
999 999 user_ns = self.shell.user_ns
1000 1000 for i in self.magic_who_ls():
1001 1001 del(user_ns[i])
1002 1002
1003 1003 # Also flush the private list of module references kept for script
1004 1004 # execution protection
1005 1005 self.shell.clear_main_mod_cache()
1006 1006
1007 1007 def magic_reset_selective(self, parameter_s=''):
1008 1008 """Resets the namespace by removing names defined by the user.
1009 1009
1010 1010 Input/Output history are left around in case you need them.
1011 1011
1012 1012 %reset_selective [-f] regex
1013 1013
1014 1014 No action is taken if regex is not included
1015 1015
1016 1016 Options
1017 1017 -f : force reset without asking for confirmation.
1018 1018
1019 1019 Examples
1020 1020 --------
1021 1021
1022 1022 We first fully reset the namespace so your output looks identical to
1023 1023 this example for pedagogical reasons; in practice you do not need a
1024 1024 full reset.
1025 1025
1026 1026 In [1]: %reset -f
1027 1027
1028 1028 Now, with a clean namespace we can make a few variables and use
1029 1029 %reset_selective to only delete names that match our regexp:
1030 1030
1031 1031 In [2]: a=1; b=2; c=3; b1m=4; b2m=5; b3m=6; b4m=7; b2s=8
1032 1032
1033 1033 In [3]: who_ls
1034 1034 Out[3]: ['a', 'b', 'b1m', 'b2m', 'b2s', 'b3m', 'b4m', 'c']
1035 1035
1036 1036 In [4]: %reset_selective -f b[2-3]m
1037 1037
1038 1038 In [5]: who_ls
1039 1039 Out[5]: ['a', 'b', 'b1m', 'b2s', 'b4m', 'c']
1040 1040
1041 1041 In [6]: %reset_selective -f d
1042 1042
1043 1043 In [7]: who_ls
1044 1044 Out[7]: ['a', 'b', 'b1m', 'b2s', 'b4m', 'c']
1045 1045
1046 1046 In [8]: %reset_selective -f c
1047 1047
1048 1048 In [9]: who_ls
1049 1049 Out[9]: ['a', 'b', 'b1m', 'b2s', 'b4m']
1050 1050
1051 1051 In [10]: %reset_selective -f b
1052 1052
1053 1053 In [11]: who_ls
1054 1054 Out[11]: ['a']
1055 1055 """
1056 1056
1057 1057 opts, regex = self.parse_options(parameter_s,'f')
1058 1058
1059 1059 if opts.has_key('f'):
1060 1060 ans = True
1061 1061 else:
1062 1062 ans = self.shell.ask_yes_no(
1063 1063 "Once deleted, variables cannot be recovered. Proceed (y/[n])? ")
1064 1064 if not ans:
1065 1065 print 'Nothing done.'
1066 1066 return
1067 1067 user_ns = self.shell.user_ns
1068 1068 if not regex:
1069 1069 print 'No regex pattern specified. Nothing done.'
1070 1070 return
1071 1071 else:
1072 1072 try:
1073 1073 m = re.compile(regex)
1074 1074 except TypeError:
1075 1075 raise TypeError('regex must be a string or compiled pattern')
1076 1076 for i in self.magic_who_ls():
1077 1077 if m.search(i):
1078 1078 del(user_ns[i])
1079 1079
1080 1080 def magic_logstart(self,parameter_s=''):
1081 1081 """Start logging anywhere in a session.
1082 1082
1083 1083 %logstart [-o|-r|-t] [log_name [log_mode]]
1084 1084
1085 1085 If no name is given, it defaults to a file named 'ipython_log.py' in your
1086 1086 current directory, in 'rotate' mode (see below).
1087 1087
1088 1088 '%logstart name' saves to file 'name' in 'backup' mode. It saves your
1089 1089 history up to that point and then continues logging.
1090 1090
1091 1091 %logstart takes a second optional parameter: logging mode. This can be one
1092 1092 of (note that the modes are given unquoted):\\
1093 1093 append: well, that says it.\\
1094 1094 backup: rename (if exists) to name~ and start name.\\
1095 1095 global: single logfile in your home dir, appended to.\\
1096 1096 over : overwrite existing log.\\
1097 1097 rotate: create rotating logs name.1~, name.2~, etc.
1098 1098
1099 1099 Options:
1100 1100
1101 1101 -o: log also IPython's output. In this mode, all commands which
1102 1102 generate an Out[NN] prompt are recorded to the logfile, right after
1103 1103 their corresponding input line. The output lines are always
1104 1104 prepended with a '#[Out]# ' marker, so that the log remains valid
1105 1105 Python code.
1106 1106
1107 1107 Since this marker is always the same, filtering only the output from
1108 1108 a log is very easy, using for example a simple awk call:
1109 1109
1110 1110 awk -F'#\\[Out\\]# ' '{if($2) {print $2}}' ipython_log.py
1111 1111
1112 1112 -r: log 'raw' input. Normally, IPython's logs contain the processed
1113 1113 input, so that user lines are logged in their final form, converted
1114 1114 into valid Python. For example, %Exit is logged as
1115 1115 '_ip.magic("Exit"). If the -r flag is given, all input is logged
1116 1116 exactly as typed, with no transformations applied.
1117 1117
1118 1118 -t: put timestamps before each input line logged (these are put in
1119 1119 comments)."""
1120 1120
1121 1121 opts,par = self.parse_options(parameter_s,'ort')
1122 1122 log_output = 'o' in opts
1123 1123 log_raw_input = 'r' in opts
1124 1124 timestamp = 't' in opts
1125 1125
1126 1126 logger = self.shell.logger
1127 1127
1128 1128 # if no args are given, the defaults set in the logger constructor by
1129 1129 # ipytohn remain valid
1130 1130 if par:
1131 1131 try:
1132 1132 logfname,logmode = par.split()
1133 1133 except:
1134 1134 logfname = par
1135 1135 logmode = 'backup'
1136 1136 else:
1137 1137 logfname = logger.logfname
1138 1138 logmode = logger.logmode
1139 1139 # put logfname into rc struct as if it had been called on the command
1140 1140 # line, so it ends up saved in the log header Save it in case we need
1141 1141 # to restore it...
1142 1142 old_logfile = self.shell.logfile
1143 1143 if logfname:
1144 1144 logfname = os.path.expanduser(logfname)
1145 1145 self.shell.logfile = logfname
1146 1146
1147 1147 loghead = '# IPython log file\n\n'
1148 1148 try:
1149 1149 started = logger.logstart(logfname,loghead,logmode,
1150 1150 log_output,timestamp,log_raw_input)
1151 1151 except:
1152 1152 self.shell.logfile = old_logfile
1153 1153 warn("Couldn't start log: %s" % sys.exc_info()[1])
1154 1154 else:
1155 1155 # log input history up to this point, optionally interleaving
1156 1156 # output if requested
1157 1157
1158 1158 if timestamp:
1159 1159 # disable timestamping for the previous history, since we've
1160 1160 # lost those already (no time machine here).
1161 1161 logger.timestamp = False
1162 1162
1163 1163 if log_raw_input:
1164 1164 input_hist = self.shell.history_manager.input_hist_raw
1165 1165 else:
1166 1166 input_hist = self.shell.history_manager.input_hist_parsed
1167 1167
1168 1168 if log_output:
1169 1169 log_write = logger.log_write
1170 1170 output_hist = self.shell.history_manager.output_hist
1171 1171 for n in range(1,len(input_hist)-1):
1172 1172 log_write(input_hist[n].rstrip())
1173 1173 if n in output_hist:
1174 1174 log_write(repr(output_hist[n]),'output')
1175 1175 else:
1176 1176 logger.log_write(''.join(input_hist[1:]))
1177 1177 if timestamp:
1178 1178 # re-enable timestamping
1179 1179 logger.timestamp = True
1180 1180
1181 1181 print ('Activating auto-logging. '
1182 1182 'Current session state plus future input saved.')
1183 1183 logger.logstate()
1184 1184
1185 1185 def magic_logstop(self,parameter_s=''):
1186 1186 """Fully stop logging and close log file.
1187 1187
1188 1188 In order to start logging again, a new %logstart call needs to be made,
1189 1189 possibly (though not necessarily) with a new filename, mode and other
1190 1190 options."""
1191 1191 self.logger.logstop()
1192 1192
1193 1193 def magic_logoff(self,parameter_s=''):
1194 1194 """Temporarily stop logging.
1195 1195
1196 1196 You must have previously started logging."""
1197 1197 self.shell.logger.switch_log(0)
1198 1198
1199 1199 def magic_logon(self,parameter_s=''):
1200 1200 """Restart logging.
1201 1201
1202 1202 This function is for restarting logging which you've temporarily
1203 1203 stopped with %logoff. For starting logging for the first time, you
1204 1204 must use the %logstart function, which allows you to specify an
1205 1205 optional log filename."""
1206 1206
1207 1207 self.shell.logger.switch_log(1)
1208 1208
1209 1209 def magic_logstate(self,parameter_s=''):
1210 1210 """Print the status of the logging system."""
1211 1211
1212 1212 self.shell.logger.logstate()
1213 1213
1214 1214 def magic_pdb(self, parameter_s=''):
1215 1215 """Control the automatic calling of the pdb interactive debugger.
1216 1216
1217 1217 Call as '%pdb on', '%pdb 1', '%pdb off' or '%pdb 0'. If called without
1218 1218 argument it works as a toggle.
1219 1219
1220 1220 When an exception is triggered, IPython can optionally call the
1221 1221 interactive pdb debugger after the traceback printout. %pdb toggles
1222 1222 this feature on and off.
1223 1223
1224 1224 The initial state of this feature is set in your ipythonrc
1225 1225 configuration file (the variable is called 'pdb').
1226 1226
1227 1227 If you want to just activate the debugger AFTER an exception has fired,
1228 1228 without having to type '%pdb on' and rerunning your code, you can use
1229 1229 the %debug magic."""
1230 1230
1231 1231 par = parameter_s.strip().lower()
1232 1232
1233 1233 if par:
1234 1234 try:
1235 1235 new_pdb = {'off':0,'0':0,'on':1,'1':1}[par]
1236 1236 except KeyError:
1237 1237 print ('Incorrect argument. Use on/1, off/0, '
1238 1238 'or nothing for a toggle.')
1239 1239 return
1240 1240 else:
1241 1241 # toggle
1242 1242 new_pdb = not self.shell.call_pdb
1243 1243
1244 1244 # set on the shell
1245 1245 self.shell.call_pdb = new_pdb
1246 1246 print 'Automatic pdb calling has been turned',on_off(new_pdb)
1247 1247
1248 1248 def magic_debug(self, parameter_s=''):
1249 1249 """Activate the interactive debugger in post-mortem mode.
1250 1250
1251 1251 If an exception has just occurred, this lets you inspect its stack
1252 1252 frames interactively. Note that this will always work only on the last
1253 1253 traceback that occurred, so you must call this quickly after an
1254 1254 exception that you wish to inspect has fired, because if another one
1255 1255 occurs, it clobbers the previous one.
1256 1256
1257 1257 If you want IPython to automatically do this on every exception, see
1258 1258 the %pdb magic for more details.
1259 1259 """
1260 1260 self.shell.debugger(force=True)
1261 1261
1262 1262 @testdec.skip_doctest
1263 1263 def magic_prun(self, parameter_s ='',user_mode=1,
1264 1264 opts=None,arg_lst=None,prog_ns=None):
1265 1265
1266 1266 """Run a statement through the python code profiler.
1267 1267
1268 1268 Usage:
1269 1269 %prun [options] statement
1270 1270
1271 1271 The given statement (which doesn't require quote marks) is run via the
1272 1272 python profiler in a manner similar to the profile.run() function.
1273 1273 Namespaces are internally managed to work correctly; profile.run
1274 1274 cannot be used in IPython because it makes certain assumptions about
1275 1275 namespaces which do not hold under IPython.
1276 1276
1277 1277 Options:
1278 1278
1279 1279 -l <limit>: you can place restrictions on what or how much of the
1280 1280 profile gets printed. The limit value can be:
1281 1281
1282 1282 * A string: only information for function names containing this string
1283 1283 is printed.
1284 1284
1285 1285 * An integer: only these many lines are printed.
1286 1286
1287 1287 * A float (between 0 and 1): this fraction of the report is printed
1288 1288 (for example, use a limit of 0.4 to see the topmost 40% only).
1289 1289
1290 1290 You can combine several limits with repeated use of the option. For
1291 1291 example, '-l __init__ -l 5' will print only the topmost 5 lines of
1292 1292 information about class constructors.
1293 1293
1294 1294 -r: return the pstats.Stats object generated by the profiling. This
1295 1295 object has all the information about the profile in it, and you can
1296 1296 later use it for further analysis or in other functions.
1297 1297
1298 1298 -s <key>: sort profile by given key. You can provide more than one key
1299 1299 by using the option several times: '-s key1 -s key2 -s key3...'. The
1300 1300 default sorting key is 'time'.
1301 1301
1302 1302 The following is copied verbatim from the profile documentation
1303 1303 referenced below:
1304 1304
1305 1305 When more than one key is provided, additional keys are used as
1306 1306 secondary criteria when the there is equality in all keys selected
1307 1307 before them.
1308 1308
1309 1309 Abbreviations can be used for any key names, as long as the
1310 1310 abbreviation is unambiguous. The following are the keys currently
1311 1311 defined:
1312 1312
1313 1313 Valid Arg Meaning
1314 1314 "calls" call count
1315 1315 "cumulative" cumulative time
1316 1316 "file" file name
1317 1317 "module" file name
1318 1318 "pcalls" primitive call count
1319 1319 "line" line number
1320 1320 "name" function name
1321 1321 "nfl" name/file/line
1322 1322 "stdname" standard name
1323 1323 "time" internal time
1324 1324
1325 1325 Note that all sorts on statistics are in descending order (placing
1326 1326 most time consuming items first), where as name, file, and line number
1327 1327 searches are in ascending order (i.e., alphabetical). The subtle
1328 1328 distinction between "nfl" and "stdname" is that the standard name is a
1329 1329 sort of the name as printed, which means that the embedded line
1330 1330 numbers get compared in an odd way. For example, lines 3, 20, and 40
1331 1331 would (if the file names were the same) appear in the string order
1332 1332 "20" "3" and "40". In contrast, "nfl" does a numeric compare of the
1333 1333 line numbers. In fact, sort_stats("nfl") is the same as
1334 1334 sort_stats("name", "file", "line").
1335 1335
1336 1336 -T <filename>: save profile results as shown on screen to a text
1337 1337 file. The profile is still shown on screen.
1338 1338
1339 1339 -D <filename>: save (via dump_stats) profile statistics to given
1340 1340 filename. This data is in a format understod by the pstats module, and
1341 1341 is generated by a call to the dump_stats() method of profile
1342 1342 objects. The profile is still shown on screen.
1343 1343
1344 1344 If you want to run complete programs under the profiler's control, use
1345 1345 '%run -p [prof_opts] filename.py [args to program]' where prof_opts
1346 1346 contains profiler specific options as described here.
1347 1347
1348 1348 You can read the complete documentation for the profile module with::
1349 1349
1350 1350 In [1]: import profile; profile.help()
1351 1351 """
1352 1352
1353 1353 opts_def = Struct(D=[''],l=[],s=['time'],T=[''])
1354 1354 # protect user quote marks
1355 1355 parameter_s = parameter_s.replace('"',r'\"').replace("'",r"\'")
1356 1356
1357 1357 if user_mode: # regular user call
1358 1358 opts,arg_str = self.parse_options(parameter_s,'D:l:rs:T:',
1359 1359 list_all=1)
1360 1360 namespace = self.shell.user_ns
1361 1361 else: # called to run a program by %run -p
1362 1362 try:
1363 1363 filename = get_py_filename(arg_lst[0])
1364 1364 except IOError,msg:
1365 1365 error(msg)
1366 1366 return
1367 1367
1368 1368 arg_str = 'execfile(filename,prog_ns)'
1369 1369 namespace = locals()
1370 1370
1371 1371 opts.merge(opts_def)
1372 1372
1373 1373 prof = profile.Profile()
1374 1374 try:
1375 1375 prof = prof.runctx(arg_str,namespace,namespace)
1376 1376 sys_exit = ''
1377 1377 except SystemExit:
1378 1378 sys_exit = """*** SystemExit exception caught in code being profiled."""
1379 1379
1380 1380 stats = pstats.Stats(prof).strip_dirs().sort_stats(*opts.s)
1381 1381
1382 1382 lims = opts.l
1383 1383 if lims:
1384 1384 lims = [] # rebuild lims with ints/floats/strings
1385 1385 for lim in opts.l:
1386 1386 try:
1387 1387 lims.append(int(lim))
1388 1388 except ValueError:
1389 1389 try:
1390 1390 lims.append(float(lim))
1391 1391 except ValueError:
1392 1392 lims.append(lim)
1393 1393
1394 1394 # Trap output.
1395 1395 stdout_trap = StringIO()
1396 1396
1397 1397 if hasattr(stats,'stream'):
1398 1398 # In newer versions of python, the stats object has a 'stream'
1399 1399 # attribute to write into.
1400 1400 stats.stream = stdout_trap
1401 1401 stats.print_stats(*lims)
1402 1402 else:
1403 1403 # For older versions, we manually redirect stdout during printing
1404 1404 sys_stdout = sys.stdout
1405 1405 try:
1406 1406 sys.stdout = stdout_trap
1407 1407 stats.print_stats(*lims)
1408 1408 finally:
1409 1409 sys.stdout = sys_stdout
1410 1410
1411 1411 output = stdout_trap.getvalue()
1412 1412 output = output.rstrip()
1413 1413
1414 1414 page.page(output)
1415 1415 print sys_exit,
1416 1416
1417 1417 dump_file = opts.D[0]
1418 1418 text_file = opts.T[0]
1419 1419 if dump_file:
1420 1420 prof.dump_stats(dump_file)
1421 1421 print '\n*** Profile stats marshalled to file',\
1422 1422 `dump_file`+'.',sys_exit
1423 1423 if text_file:
1424 1424 pfile = file(text_file,'w')
1425 1425 pfile.write(output)
1426 1426 pfile.close()
1427 1427 print '\n*** Profile printout saved to text file',\
1428 1428 `text_file`+'.',sys_exit
1429 1429
1430 1430 if opts.has_key('r'):
1431 1431 return stats
1432 1432 else:
1433 1433 return None
1434 1434
1435 1435 @testdec.skip_doctest
1436 1436 def magic_run(self, parameter_s ='',runner=None,
1437 1437 file_finder=get_py_filename):
1438 1438 """Run the named file inside IPython as a program.
1439 1439
1440 1440 Usage:\\
1441 1441 %run [-n -i -t [-N<N>] -d [-b<N>] -p [profile options]] file [args]
1442 1442
1443 1443 Parameters after the filename are passed as command-line arguments to
1444 1444 the program (put in sys.argv). Then, control returns to IPython's
1445 1445 prompt.
1446 1446
1447 1447 This is similar to running at a system prompt:\\
1448 1448 $ python file args\\
1449 1449 but with the advantage of giving you IPython's tracebacks, and of
1450 1450 loading all variables into your interactive namespace for further use
1451 1451 (unless -p is used, see below).
1452 1452
1453 1453 The file is executed in a namespace initially consisting only of
1454 1454 __name__=='__main__' and sys.argv constructed as indicated. It thus
1455 1455 sees its environment as if it were being run as a stand-alone program
1456 1456 (except for sharing global objects such as previously imported
1457 1457 modules). But after execution, the IPython interactive namespace gets
1458 1458 updated with all variables defined in the program (except for __name__
1459 1459 and sys.argv). This allows for very convenient loading of code for
1460 1460 interactive work, while giving each program a 'clean sheet' to run in.
1461 1461
1462 1462 Options:
1463 1463
1464 1464 -n: __name__ is NOT set to '__main__', but to the running file's name
1465 1465 without extension (as python does under import). This allows running
1466 1466 scripts and reloading the definitions in them without calling code
1467 1467 protected by an ' if __name__ == "__main__" ' clause.
1468 1468
1469 1469 -i: run the file in IPython's namespace instead of an empty one. This
1470 1470 is useful if you are experimenting with code written in a text editor
1471 1471 which depends on variables defined interactively.
1472 1472
1473 1473 -e: ignore sys.exit() calls or SystemExit exceptions in the script
1474 1474 being run. This is particularly useful if IPython is being used to
1475 1475 run unittests, which always exit with a sys.exit() call. In such
1476 1476 cases you are interested in the output of the test results, not in
1477 1477 seeing a traceback of the unittest module.
1478 1478
1479 1479 -t: print timing information at the end of the run. IPython will give
1480 1480 you an estimated CPU time consumption for your script, which under
1481 1481 Unix uses the resource module to avoid the wraparound problems of
1482 1482 time.clock(). Under Unix, an estimate of time spent on system tasks
1483 1483 is also given (for Windows platforms this is reported as 0.0).
1484 1484
1485 1485 If -t is given, an additional -N<N> option can be given, where <N>
1486 1486 must be an integer indicating how many times you want the script to
1487 1487 run. The final timing report will include total and per run results.
1488 1488
1489 1489 For example (testing the script uniq_stable.py):
1490 1490
1491 1491 In [1]: run -t uniq_stable
1492 1492
1493 1493 IPython CPU timings (estimated):\\
1494 1494 User : 0.19597 s.\\
1495 1495 System: 0.0 s.\\
1496 1496
1497 1497 In [2]: run -t -N5 uniq_stable
1498 1498
1499 1499 IPython CPU timings (estimated):\\
1500 1500 Total runs performed: 5\\
1501 1501 Times : Total Per run\\
1502 1502 User : 0.910862 s, 0.1821724 s.\\
1503 1503 System: 0.0 s, 0.0 s.
1504 1504
1505 1505 -d: run your program under the control of pdb, the Python debugger.
1506 1506 This allows you to execute your program step by step, watch variables,
1507 1507 etc. Internally, what IPython does is similar to calling:
1508 1508
1509 1509 pdb.run('execfile("YOURFILENAME")')
1510 1510
1511 1511 with a breakpoint set on line 1 of your file. You can change the line
1512 1512 number for this automatic breakpoint to be <N> by using the -bN option
1513 1513 (where N must be an integer). For example:
1514 1514
1515 1515 %run -d -b40 myscript
1516 1516
1517 1517 will set the first breakpoint at line 40 in myscript.py. Note that
1518 1518 the first breakpoint must be set on a line which actually does
1519 1519 something (not a comment or docstring) for it to stop execution.
1520 1520
1521 1521 When the pdb debugger starts, you will see a (Pdb) prompt. You must
1522 1522 first enter 'c' (without qoutes) to start execution up to the first
1523 1523 breakpoint.
1524 1524
1525 1525 Entering 'help' gives information about the use of the debugger. You
1526 1526 can easily see pdb's full documentation with "import pdb;pdb.help()"
1527 1527 at a prompt.
1528 1528
1529 1529 -p: run program under the control of the Python profiler module (which
1530 1530 prints a detailed report of execution times, function calls, etc).
1531 1531
1532 1532 You can pass other options after -p which affect the behavior of the
1533 1533 profiler itself. See the docs for %prun for details.
1534 1534
1535 1535 In this mode, the program's variables do NOT propagate back to the
1536 1536 IPython interactive namespace (because they remain in the namespace
1537 1537 where the profiler executes them).
1538 1538
1539 1539 Internally this triggers a call to %prun, see its documentation for
1540 1540 details on the options available specifically for profiling.
1541 1541
1542 1542 There is one special usage for which the text above doesn't apply:
1543 1543 if the filename ends with .ipy, the file is run as ipython script,
1544 1544 just as if the commands were written on IPython prompt.
1545 1545 """
1546 1546
1547 1547 # get arguments and set sys.argv for program to be run.
1548 1548 opts,arg_lst = self.parse_options(parameter_s,'nidtN:b:pD:l:rs:T:e',
1549 1549 mode='list',list_all=1)
1550 1550
1551 1551 try:
1552 1552 filename = file_finder(arg_lst[0])
1553 1553 except IndexError:
1554 1554 warn('you must provide at least a filename.')
1555 1555 print '\n%run:\n',oinspect.getdoc(self.magic_run)
1556 1556 return
1557 1557 except IOError,msg:
1558 1558 error(msg)
1559 1559 return
1560 1560
1561 1561 if filename.lower().endswith('.ipy'):
1562 1562 self.shell.safe_execfile_ipy(filename)
1563 1563 return
1564 1564
1565 1565 # Control the response to exit() calls made by the script being run
1566 1566 exit_ignore = opts.has_key('e')
1567 1567
1568 1568 # Make sure that the running script gets a proper sys.argv as if it
1569 1569 # were run from a system shell.
1570 1570 save_argv = sys.argv # save it for later restoring
1571 1571 sys.argv = [filename]+ arg_lst[1:] # put in the proper filename
1572 1572
1573 1573 if opts.has_key('i'):
1574 1574 # Run in user's interactive namespace
1575 1575 prog_ns = self.shell.user_ns
1576 1576 __name__save = self.shell.user_ns['__name__']
1577 1577 prog_ns['__name__'] = '__main__'
1578 1578 main_mod = self.shell.new_main_mod(prog_ns)
1579 1579 else:
1580 1580 # Run in a fresh, empty namespace
1581 1581 if opts.has_key('n'):
1582 1582 name = os.path.splitext(os.path.basename(filename))[0]
1583 1583 else:
1584 1584 name = '__main__'
1585 1585
1586 1586 main_mod = self.shell.new_main_mod()
1587 1587 prog_ns = main_mod.__dict__
1588 1588 prog_ns['__name__'] = name
1589 1589
1590 1590 # Since '%run foo' emulates 'python foo.py' at the cmd line, we must
1591 1591 # set the __file__ global in the script's namespace
1592 1592 prog_ns['__file__'] = filename
1593 1593
1594 1594 # pickle fix. See interactiveshell for an explanation. But we need to make sure
1595 1595 # that, if we overwrite __main__, we replace it at the end
1596 1596 main_mod_name = prog_ns['__name__']
1597 1597
1598 1598 if main_mod_name == '__main__':
1599 1599 restore_main = sys.modules['__main__']
1600 1600 else:
1601 1601 restore_main = False
1602 1602
1603 1603 # This needs to be undone at the end to prevent holding references to
1604 1604 # every single object ever created.
1605 1605 sys.modules[main_mod_name] = main_mod
1606 1606
1607 1607 stats = None
1608 1608 try:
1609 1609 #self.shell.save_history()
1610 1610
1611 1611 if opts.has_key('p'):
1612 1612 stats = self.magic_prun('',0,opts,arg_lst,prog_ns)
1613 1613 else:
1614 1614 if opts.has_key('d'):
1615 1615 deb = debugger.Pdb(self.shell.colors)
1616 1616 # reset Breakpoint state, which is moronically kept
1617 1617 # in a class
1618 1618 bdb.Breakpoint.next = 1
1619 1619 bdb.Breakpoint.bplist = {}
1620 1620 bdb.Breakpoint.bpbynumber = [None]
1621 1621 # Set an initial breakpoint to stop execution
1622 1622 maxtries = 10
1623 1623 bp = int(opts.get('b',[1])[0])
1624 1624 checkline = deb.checkline(filename,bp)
1625 1625 if not checkline:
1626 1626 for bp in range(bp+1,bp+maxtries+1):
1627 1627 if deb.checkline(filename,bp):
1628 1628 break
1629 1629 else:
1630 1630 msg = ("\nI failed to find a valid line to set "
1631 1631 "a breakpoint\n"
1632 1632 "after trying up to line: %s.\n"
1633 1633 "Please set a valid breakpoint manually "
1634 1634 "with the -b option." % bp)
1635 1635 error(msg)
1636 1636 return
1637 1637 # if we find a good linenumber, set the breakpoint
1638 1638 deb.do_break('%s:%s' % (filename,bp))
1639 1639 # Start file run
1640 1640 print "NOTE: Enter 'c' at the",
1641 1641 print "%s prompt to start your script." % deb.prompt
1642 1642 try:
1643 1643 deb.run('execfile("%s")' % filename,prog_ns)
1644 1644
1645 1645 except:
1646 1646 etype, value, tb = sys.exc_info()
1647 1647 # Skip three frames in the traceback: the %run one,
1648 1648 # one inside bdb.py, and the command-line typed by the
1649 1649 # user (run by exec in pdb itself).
1650 1650 self.shell.InteractiveTB(etype,value,tb,tb_offset=3)
1651 1651 else:
1652 1652 if runner is None:
1653 1653 runner = self.shell.safe_execfile
1654 1654 if opts.has_key('t'):
1655 1655 # timed execution
1656 1656 try:
1657 1657 nruns = int(opts['N'][0])
1658 1658 if nruns < 1:
1659 1659 error('Number of runs must be >=1')
1660 1660 return
1661 1661 except (KeyError):
1662 1662 nruns = 1
1663 1663 if nruns == 1:
1664 1664 t0 = clock2()
1665 1665 runner(filename,prog_ns,prog_ns,
1666 1666 exit_ignore=exit_ignore)
1667 1667 t1 = clock2()
1668 1668 t_usr = t1[0]-t0[0]
1669 1669 t_sys = t1[1]-t0[1]
1670 1670 print "\nIPython CPU timings (estimated):"
1671 1671 print " User : %10s s." % t_usr
1672 1672 print " System: %10s s." % t_sys
1673 1673 else:
1674 1674 runs = range(nruns)
1675 1675 t0 = clock2()
1676 1676 for nr in runs:
1677 1677 runner(filename,prog_ns,prog_ns,
1678 1678 exit_ignore=exit_ignore)
1679 1679 t1 = clock2()
1680 1680 t_usr = t1[0]-t0[0]
1681 1681 t_sys = t1[1]-t0[1]
1682 1682 print "\nIPython CPU timings (estimated):"
1683 1683 print "Total runs performed:",nruns
1684 1684 print " Times : %10s %10s" % ('Total','Per run')
1685 1685 print " User : %10s s, %10s s." % (t_usr,t_usr/nruns)
1686 1686 print " System: %10s s, %10s s." % (t_sys,t_sys/nruns)
1687 1687
1688 1688 else:
1689 1689 # regular execution
1690 1690 runner(filename,prog_ns,prog_ns,exit_ignore=exit_ignore)
1691 1691
1692 1692 if opts.has_key('i'):
1693 1693 self.shell.user_ns['__name__'] = __name__save
1694 1694 else:
1695 1695 # The shell MUST hold a reference to prog_ns so after %run
1696 1696 # exits, the python deletion mechanism doesn't zero it out
1697 1697 # (leaving dangling references).
1698 1698 self.shell.cache_main_mod(prog_ns,filename)
1699 1699 # update IPython interactive namespace
1700 1700
1701 1701 # Some forms of read errors on the file may mean the
1702 1702 # __name__ key was never set; using pop we don't have to
1703 1703 # worry about a possible KeyError.
1704 1704 prog_ns.pop('__name__', None)
1705 1705
1706 1706 self.shell.user_ns.update(prog_ns)
1707 1707 finally:
1708 1708 # It's a bit of a mystery why, but __builtins__ can change from
1709 1709 # being a module to becoming a dict missing some key data after
1710 1710 # %run. As best I can see, this is NOT something IPython is doing
1711 1711 # at all, and similar problems have been reported before:
1712 1712 # http://coding.derkeiler.com/Archive/Python/comp.lang.python/2004-10/0188.html
1713 1713 # Since this seems to be done by the interpreter itself, the best
1714 1714 # we can do is to at least restore __builtins__ for the user on
1715 1715 # exit.
1716 1716 self.shell.user_ns['__builtins__'] = __builtin__
1717 1717
1718 1718 # Ensure key global structures are restored
1719 1719 sys.argv = save_argv
1720 1720 if restore_main:
1721 1721 sys.modules['__main__'] = restore_main
1722 1722 else:
1723 1723 # Remove from sys.modules the reference to main_mod we'd
1724 1724 # added. Otherwise it will trap references to objects
1725 1725 # contained therein.
1726 1726 del sys.modules[main_mod_name]
1727 1727
1728 1728 #self.shell.reload_history()
1729 1729
1730 1730 return stats
1731 1731
1732 1732 @testdec.skip_doctest
1733 1733 def magic_timeit(self, parameter_s =''):
1734 1734 """Time execution of a Python statement or expression
1735 1735
1736 1736 Usage:\\
1737 1737 %timeit [-n<N> -r<R> [-t|-c]] statement
1738 1738
1739 1739 Time execution of a Python statement or expression using the timeit
1740 1740 module.
1741 1741
1742 1742 Options:
1743 1743 -n<N>: execute the given statement <N> times in a loop. If this value
1744 1744 is not given, a fitting value is chosen.
1745 1745
1746 1746 -r<R>: repeat the loop iteration <R> times and take the best result.
1747 1747 Default: 3
1748 1748
1749 1749 -t: use time.time to measure the time, which is the default on Unix.
1750 1750 This function measures wall time.
1751 1751
1752 1752 -c: use time.clock to measure the time, which is the default on
1753 1753 Windows and measures wall time. On Unix, resource.getrusage is used
1754 1754 instead and returns the CPU user time.
1755 1755
1756 1756 -p<P>: use a precision of <P> digits to display the timing result.
1757 1757 Default: 3
1758 1758
1759 1759
1760 1760 Examples:
1761 1761
1762 1762 In [1]: %timeit pass
1763 1763 10000000 loops, best of 3: 53.3 ns per loop
1764 1764
1765 1765 In [2]: u = None
1766 1766
1767 1767 In [3]: %timeit u is None
1768 1768 10000000 loops, best of 3: 184 ns per loop
1769 1769
1770 1770 In [4]: %timeit -r 4 u == None
1771 1771 1000000 loops, best of 4: 242 ns per loop
1772 1772
1773 1773 In [5]: import time
1774 1774
1775 1775 In [6]: %timeit -n1 time.sleep(2)
1776 1776 1 loops, best of 3: 2 s per loop
1777 1777
1778 1778
1779 1779 The times reported by %timeit will be slightly higher than those
1780 1780 reported by the timeit.py script when variables are accessed. This is
1781 1781 due to the fact that %timeit executes the statement in the namespace
1782 1782 of the shell, compared with timeit.py, which uses a single setup
1783 1783 statement to import function or create variables. Generally, the bias
1784 1784 does not matter as long as results from timeit.py are not mixed with
1785 1785 those from %timeit."""
1786 1786
1787 1787 import timeit
1788 1788 import math
1789 1789
1790 1790 # XXX: Unfortunately the unicode 'micro' symbol can cause problems in
1791 1791 # certain terminals. Until we figure out a robust way of
1792 1792 # auto-detecting if the terminal can deal with it, use plain 'us' for
1793 1793 # microseconds. I am really NOT happy about disabling the proper
1794 1794 # 'micro' prefix, but crashing is worse... If anyone knows what the
1795 1795 # right solution for this is, I'm all ears...
1796 1796 #
1797 1797 # Note: using
1798 1798 #
1799 1799 # s = u'\xb5'
1800 1800 # s.encode(sys.getdefaultencoding())
1801 1801 #
1802 1802 # is not sufficient, as I've seen terminals where that fails but
1803 1803 # print s
1804 1804 #
1805 1805 # succeeds
1806 1806 #
1807 1807 # See bug: https://bugs.launchpad.net/ipython/+bug/348466
1808 1808
1809 1809 #units = [u"s", u"ms",u'\xb5',"ns"]
1810 1810 units = [u"s", u"ms",u'us',"ns"]
1811 1811
1812 1812 scaling = [1, 1e3, 1e6, 1e9]
1813 1813
1814 1814 opts, stmt = self.parse_options(parameter_s,'n:r:tcp:',
1815 1815 posix=False)
1816 1816 if stmt == "":
1817 1817 return
1818 1818 timefunc = timeit.default_timer
1819 1819 number = int(getattr(opts, "n", 0))
1820 1820 repeat = int(getattr(opts, "r", timeit.default_repeat))
1821 1821 precision = int(getattr(opts, "p", 3))
1822 1822 if hasattr(opts, "t"):
1823 1823 timefunc = time.time
1824 1824 if hasattr(opts, "c"):
1825 1825 timefunc = clock
1826 1826
1827 1827 timer = timeit.Timer(timer=timefunc)
1828 1828 # this code has tight coupling to the inner workings of timeit.Timer,
1829 1829 # but is there a better way to achieve that the code stmt has access
1830 1830 # to the shell namespace?
1831 1831
1832 1832 src = timeit.template % {'stmt': timeit.reindent(stmt, 8),
1833 1833 'setup': "pass"}
1834 1834 # Track compilation time so it can be reported if too long
1835 1835 # Minimum time above which compilation time will be reported
1836 1836 tc_min = 0.1
1837 1837
1838 1838 t0 = clock()
1839 1839 code = compile(src, "<magic-timeit>", "exec")
1840 1840 tc = clock()-t0
1841 1841
1842 1842 ns = {}
1843 1843 exec code in self.shell.user_ns, ns
1844 1844 timer.inner = ns["inner"]
1845 1845
1846 1846 if number == 0:
1847 1847 # determine number so that 0.2 <= total time < 2.0
1848 1848 number = 1
1849 1849 for i in range(1, 10):
1850 1850 if timer.timeit(number) >= 0.2:
1851 1851 break
1852 1852 number *= 10
1853 1853
1854 1854 best = min(timer.repeat(repeat, number)) / number
1855 1855
1856 1856 if best > 0.0 and best < 1000.0:
1857 1857 order = min(-int(math.floor(math.log10(best)) // 3), 3)
1858 1858 elif best >= 1000.0:
1859 1859 order = 0
1860 1860 else:
1861 1861 order = 3
1862 1862 print u"%d loops, best of %d: %.*g %s per loop" % (number, repeat,
1863 1863 precision,
1864 1864 best * scaling[order],
1865 1865 units[order])
1866 1866 if tc > tc_min:
1867 1867 print "Compiler time: %.2f s" % tc
1868 1868
1869 1869 @testdec.skip_doctest
1870 1870 @needs_local_scope
1871 1871 def magic_time(self,parameter_s = ''):
1872 1872 """Time execution of a Python statement or expression.
1873 1873
1874 1874 The CPU and wall clock times are printed, and the value of the
1875 1875 expression (if any) is returned. Note that under Win32, system time
1876 1876 is always reported as 0, since it can not be measured.
1877 1877
1878 1878 This function provides very basic timing functionality. In Python
1879 1879 2.3, the timeit module offers more control and sophistication, so this
1880 1880 could be rewritten to use it (patches welcome).
1881 1881
1882 1882 Some examples:
1883 1883
1884 1884 In [1]: time 2**128
1885 1885 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1886 1886 Wall time: 0.00
1887 1887 Out[1]: 340282366920938463463374607431768211456L
1888 1888
1889 1889 In [2]: n = 1000000
1890 1890
1891 1891 In [3]: time sum(range(n))
1892 1892 CPU times: user 1.20 s, sys: 0.05 s, total: 1.25 s
1893 1893 Wall time: 1.37
1894 1894 Out[3]: 499999500000L
1895 1895
1896 1896 In [4]: time print 'hello world'
1897 1897 hello world
1898 1898 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1899 1899 Wall time: 0.00
1900 1900
1901 1901 Note that the time needed by Python to compile the given expression
1902 1902 will be reported if it is more than 0.1s. In this example, the
1903 1903 actual exponentiation is done by Python at compilation time, so while
1904 1904 the expression can take a noticeable amount of time to compute, that
1905 1905 time is purely due to the compilation:
1906 1906
1907 1907 In [5]: time 3**9999;
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
1911 1911 In [6]: time 3**999999;
1912 1912 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1913 1913 Wall time: 0.00 s
1914 1914 Compiler : 0.78 s
1915 1915 """
1916 1916
1917 1917 # fail immediately if the given expression can't be compiled
1918 1918
1919 1919 expr = self.shell.prefilter(parameter_s,False)
1920 1920
1921 1921 # Minimum time above which compilation time will be reported
1922 1922 tc_min = 0.1
1923 1923
1924 1924 try:
1925 1925 mode = 'eval'
1926 1926 t0 = clock()
1927 1927 code = compile(expr,'<timed eval>',mode)
1928 1928 tc = clock()-t0
1929 1929 except SyntaxError:
1930 1930 mode = 'exec'
1931 1931 t0 = clock()
1932 1932 code = compile(expr,'<timed exec>',mode)
1933 1933 tc = clock()-t0
1934 1934 # skew measurement as little as possible
1935 1935 glob = self.shell.user_ns
1936 1936 locs = self._magic_locals
1937 1937 clk = clock2
1938 1938 wtime = time.time
1939 1939 # time execution
1940 1940 wall_st = wtime()
1941 1941 if mode=='eval':
1942 1942 st = clk()
1943 1943 out = eval(code, glob, locs)
1944 1944 end = clk()
1945 1945 else:
1946 1946 st = clk()
1947 1947 exec code in glob, locs
1948 1948 end = clk()
1949 1949 out = None
1950 1950 wall_end = wtime()
1951 1951 # Compute actual times and report
1952 1952 wall_time = wall_end-wall_st
1953 1953 cpu_user = end[0]-st[0]
1954 1954 cpu_sys = end[1]-st[1]
1955 1955 cpu_tot = cpu_user+cpu_sys
1956 1956 print "CPU times: user %.2f s, sys: %.2f s, total: %.2f s" % \
1957 1957 (cpu_user,cpu_sys,cpu_tot)
1958 1958 print "Wall time: %.2f s" % wall_time
1959 1959 if tc > tc_min:
1960 1960 print "Compiler : %.2f s" % tc
1961 1961 return out
1962 1962
1963 1963 @testdec.skip_doctest
1964 1964 def magic_macro(self,parameter_s = ''):
1965 1965 """Define a set of input lines as a macro for future re-execution.
1966 1966
1967 1967 Usage:\\
1968 1968 %macro [options] name n1-n2 n3-n4 ... n5 .. n6 ...
1969 1969
1970 1970 Options:
1971 1971
1972 1972 -r: use 'raw' input. By default, the 'processed' history is used,
1973 1973 so that magics are loaded in their transformed version to valid
1974 1974 Python. If this option is given, the raw input as typed as the
1975 1975 command line is used instead.
1976 1976
1977 1977 This will define a global variable called `name` which is a string
1978 1978 made of joining the slices and lines you specify (n1,n2,... numbers
1979 1979 above) from your input history into a single string. This variable
1980 1980 acts like an automatic function which re-executes those lines as if
1981 1981 you had typed them. You just type 'name' at the prompt and the code
1982 1982 executes.
1983 1983
1984 1984 The syntax for indicating input ranges is described in %history.
1985 1985
1986 1986 Note: as a 'hidden' feature, you can also use traditional python slice
1987 1987 notation, where N:M means numbers N through M-1.
1988 1988
1989 1989 For example, if your history contains (%hist prints it):
1990 1990
1991 1991 44: x=1
1992 1992 45: y=3
1993 1993 46: z=x+y
1994 1994 47: print x
1995 1995 48: a=5
1996 1996 49: print 'x',x,'y',y
1997 1997
1998 1998 you can create a macro with lines 44 through 47 (included) and line 49
1999 1999 called my_macro with:
2000 2000
2001 2001 In [55]: %macro my_macro 44-47 49
2002 2002
2003 2003 Now, typing `my_macro` (without quotes) will re-execute all this code
2004 2004 in one pass.
2005 2005
2006 2006 You don't need to give the line-numbers in order, and any given line
2007 2007 number can appear multiple times. You can assemble macros with any
2008 2008 lines from your input history in any order.
2009 2009
2010 2010 The macro is a simple object which holds its value in an attribute,
2011 2011 but IPython's display system checks for macros and executes them as
2012 2012 code instead of printing them when you type their name.
2013 2013
2014 2014 You can view a macro's contents by explicitly printing it with:
2015 2015
2016 2016 'print macro_name'.
2017 2017
2018 2018 For one-off cases which DON'T contain magic function calls in them you
2019 2019 can obtain similar results by explicitly executing slices from your
2020 2020 input history with:
2021 2021
2022 2022 In [60]: exec In[44:48]+In[49]"""
2023 2023
2024 2024 opts,args = self.parse_options(parameter_s,'r',mode='list')
2025 2025 if not args: # List existing macros
2026 2026 return sorted(k for k,v in self.shell.user_ns.iteritems() if\
2027 2027 isinstance(v, Macro))
2028 2028 if len(args) == 1:
2029 2029 raise UsageError(
2030 2030 "%macro insufficient args; usage '%macro name n1-n2 n3-4...")
2031 2031 name, ranges = args[0], " ".join(args[1:])
2032 2032
2033 2033 #print 'rng',ranges # dbg
2034 2034 lines = self.extract_input_lines(ranges,'r' in opts)
2035 2035 macro = Macro(lines)
2036 2036 self.shell.define_macro(name, macro)
2037 2037 print 'Macro `%s` created. To execute, type its name (without quotes).' % name
2038 2038 print 'Macro contents:'
2039 2039 print macro,
2040 2040
2041 2041 def magic_save(self,parameter_s = ''):
2042 2042 """Save a set of lines to a given filename.
2043 2043
2044 2044 Usage:\\
2045 2045 %save [options] filename n1-n2 n3-n4 ... n5 .. n6 ...
2046 2046
2047 2047 Options:
2048 2048
2049 2049 -r: use 'raw' input. By default, the 'processed' history is used,
2050 2050 so that magics are loaded in their transformed version to valid
2051 2051 Python. If this option is given, the raw input as typed as the
2052 2052 command line is used instead.
2053 2053
2054 2054 This function uses the same syntax as %history for input ranges,
2055 2055 then saves the lines to the filename you specify.
2056 2056
2057 2057 It adds a '.py' extension to the file if you don't do so yourself, and
2058 2058 it asks for confirmation before overwriting existing files."""
2059 2059
2060 2060 opts,args = self.parse_options(parameter_s,'r',mode='list')
2061 2061 fname,ranges = args[0], " ".join(args[1:])
2062 2062 if not fname.endswith('.py'):
2063 2063 fname += '.py'
2064 2064 if os.path.isfile(fname):
2065 2065 ans = raw_input('File `%s` exists. Overwrite (y/[N])? ' % fname)
2066 2066 if ans.lower() not in ['y','yes']:
2067 2067 print 'Operation cancelled.'
2068 2068 return
2069 2069 cmds = self.extract_input_lines(ranges, 'r' in opts)
2070 2070 with open(fname,'w') as f:
2071 2071 f.write("# coding: utf-8\n")
2072 2072 f.write(cmds.encode("utf-8"))
2073 2073 print 'The following commands were written to file `%s`:' % fname
2074 2074 print cmds
2075
2076 def magic_pastebin(self, parameter_s = ''):
2077 """Upload code to the 'Lodge it' paste bin, returning the URL."""
2078 pbserver = ServerProxy('http://paste.pocoo.org/xmlrpc/')
2079 code = self.extract_input_lines(parameter_s)
2080 if not code:
2081 try:
2082 codeobj = eval(parameter_s, self.shell.user_ns)
2083 except Exception:
2084 codeobj = None
2085 if isinstance(codeobj, str):
2086 code = codeobj
2087 elif isinstance(codeobj, Macro):
2088 code = codeobj.value
2089 else:
2090 print parameter_s, ("was not recognised as a history range, nor"
2091 " as a string or macro.")
2092 return
2093 id = pbserver.pastes.newPaste("python", code)
2094 return "http://paste.pocoo.org/show/" + id
2075 2095
2076 2096 def _edit_macro(self,mname,macro):
2077 2097 """open an editor with the macro data in a file"""
2078 2098 filename = self.shell.mktempfile(macro.value)
2079 2099 self.shell.hooks.editor(filename)
2080 2100
2081 2101 # and make a new macro object, to replace the old one
2082 2102 mfile = open(filename)
2083 2103 mvalue = mfile.read()
2084 2104 mfile.close()
2085 2105 self.shell.user_ns[mname] = Macro(mvalue)
2086 2106
2087 2107 def magic_ed(self,parameter_s=''):
2088 2108 """Alias to %edit."""
2089 2109 return self.magic_edit(parameter_s)
2090 2110
2091 2111 @testdec.skip_doctest
2092 2112 def magic_edit(self,parameter_s='',last_call=['','']):
2093 2113 """Bring up an editor and execute the resulting code.
2094 2114
2095 2115 Usage:
2096 2116 %edit [options] [args]
2097 2117
2098 2118 %edit runs IPython's editor hook. The default version of this hook is
2099 2119 set to call the __IPYTHON__.rc.editor command. This is read from your
2100 2120 environment variable $EDITOR. If this isn't found, it will default to
2101 2121 vi under Linux/Unix and to notepad under Windows. See the end of this
2102 2122 docstring for how to change the editor hook.
2103 2123
2104 2124 You can also set the value of this editor via the command line option
2105 2125 '-editor' or in your ipythonrc file. This is useful if you wish to use
2106 2126 specifically for IPython an editor different from your typical default
2107 2127 (and for Windows users who typically don't set environment variables).
2108 2128
2109 2129 This command allows you to conveniently edit multi-line code right in
2110 2130 your IPython session.
2111 2131
2112 2132 If called without arguments, %edit opens up an empty editor with a
2113 2133 temporary file and will execute the contents of this file when you
2114 2134 close it (don't forget to save it!).
2115 2135
2116 2136
2117 2137 Options:
2118 2138
2119 2139 -n <number>: open the editor at a specified line number. By default,
2120 2140 the IPython editor hook uses the unix syntax 'editor +N filename', but
2121 2141 you can configure this by providing your own modified hook if your
2122 2142 favorite editor supports line-number specifications with a different
2123 2143 syntax.
2124 2144
2125 2145 -p: this will call the editor with the same data as the previous time
2126 2146 it was used, regardless of how long ago (in your current session) it
2127 2147 was.
2128 2148
2129 2149 -r: use 'raw' input. This option only applies to input taken from the
2130 2150 user's history. By default, the 'processed' history is used, so that
2131 2151 magics are loaded in their transformed version to valid Python. If
2132 2152 this option is given, the raw input as typed as the command line is
2133 2153 used instead. When you exit the editor, it will be executed by
2134 2154 IPython's own processor.
2135 2155
2136 2156 -x: do not execute the edited code immediately upon exit. This is
2137 2157 mainly useful if you are editing programs which need to be called with
2138 2158 command line arguments, which you can then do using %run.
2139 2159
2140 2160
2141 2161 Arguments:
2142 2162
2143 2163 If arguments are given, the following possibilites exist:
2144 2164
2145 2165 - If the argument is a filename, IPython will load that into the
2146 2166 editor. It will execute its contents with execfile() when you exit,
2147 2167 loading any code in the file into your interactive namespace.
2148 2168
2149 2169 - The arguments are ranges of input history, e.g. "7 ~1/4-6".
2150 2170 The syntax is the same as in the %history magic.
2151 2171
2152 2172 - If the argument is a string variable, its contents are loaded
2153 2173 into the editor. You can thus edit any string which contains
2154 2174 python code (including the result of previous edits).
2155 2175
2156 2176 - If the argument is the name of an object (other than a string),
2157 2177 IPython will try to locate the file where it was defined and open the
2158 2178 editor at the point where it is defined. You can use `%edit function`
2159 2179 to load an editor exactly at the point where 'function' is defined,
2160 2180 edit it and have the file be executed automatically.
2161 2181
2162 2182 If the object is a macro (see %macro for details), this opens up your
2163 2183 specified editor with a temporary file containing the macro's data.
2164 2184 Upon exit, the macro is reloaded with the contents of the file.
2165 2185
2166 2186 Note: opening at an exact line is only supported under Unix, and some
2167 2187 editors (like kedit and gedit up to Gnome 2.8) do not understand the
2168 2188 '+NUMBER' parameter necessary for this feature. Good editors like
2169 2189 (X)Emacs, vi, jed, pico and joe all do.
2170 2190
2171 2191 After executing your code, %edit will return as output the code you
2172 2192 typed in the editor (except when it was an existing file). This way
2173 2193 you can reload the code in further invocations of %edit as a variable,
2174 2194 via _<NUMBER> or Out[<NUMBER>], where <NUMBER> is the prompt number of
2175 2195 the output.
2176 2196
2177 2197 Note that %edit is also available through the alias %ed.
2178 2198
2179 2199 This is an example of creating a simple function inside the editor and
2180 2200 then modifying it. First, start up the editor:
2181 2201
2182 2202 In [1]: ed
2183 2203 Editing... done. Executing edited code...
2184 2204 Out[1]: 'def foo():n print "foo() was defined in an editing session"n'
2185 2205
2186 2206 We can then call the function foo():
2187 2207
2188 2208 In [2]: foo()
2189 2209 foo() was defined in an editing session
2190 2210
2191 2211 Now we edit foo. IPython automatically loads the editor with the
2192 2212 (temporary) file where foo() was previously defined:
2193 2213
2194 2214 In [3]: ed foo
2195 2215 Editing... done. Executing edited code...
2196 2216
2197 2217 And if we call foo() again we get the modified version:
2198 2218
2199 2219 In [4]: foo()
2200 2220 foo() has now been changed!
2201 2221
2202 2222 Here is an example of how to edit a code snippet successive
2203 2223 times. First we call the editor:
2204 2224
2205 2225 In [5]: ed
2206 2226 Editing... done. Executing edited code...
2207 2227 hello
2208 2228 Out[5]: "print 'hello'n"
2209 2229
2210 2230 Now we call it again with the previous output (stored in _):
2211 2231
2212 2232 In [6]: ed _
2213 2233 Editing... done. Executing edited code...
2214 2234 hello world
2215 2235 Out[6]: "print 'hello world'n"
2216 2236
2217 2237 Now we call it with the output #8 (stored in _8, also as Out[8]):
2218 2238
2219 2239 In [7]: ed _8
2220 2240 Editing... done. Executing edited code...
2221 2241 hello again
2222 2242 Out[7]: "print 'hello again'n"
2223 2243
2224 2244
2225 2245 Changing the default editor hook:
2226 2246
2227 2247 If you wish to write your own editor hook, you can put it in a
2228 2248 configuration file which you load at startup time. The default hook
2229 2249 is defined in the IPython.core.hooks module, and you can use that as a
2230 2250 starting example for further modifications. That file also has
2231 2251 general instructions on how to set a new hook for use once you've
2232 2252 defined it."""
2233 2253
2234 2254 # FIXME: This function has become a convoluted mess. It needs a
2235 2255 # ground-up rewrite with clean, simple logic.
2236 2256
2237 2257 def make_filename(arg):
2238 2258 "Make a filename from the given args"
2239 2259 try:
2240 2260 filename = get_py_filename(arg)
2241 2261 except IOError:
2242 2262 if args.endswith('.py'):
2243 2263 filename = arg
2244 2264 else:
2245 2265 filename = None
2246 2266 return filename
2247 2267
2248 2268 # custom exceptions
2249 2269 class DataIsObject(Exception): pass
2250 2270
2251 2271 opts,args = self.parse_options(parameter_s,'prxn:')
2252 2272 # Set a few locals from the options for convenience:
2253 2273 opts_prev = 'p' in opts
2254 2274 opts_raw = 'r' in opts
2255 2275
2256 2276 # Default line number value
2257 2277 lineno = opts.get('n',None)
2258 2278
2259 2279 if opts_prev:
2260 2280 args = '_%s' % last_call[0]
2261 2281 if not self.shell.user_ns.has_key(args):
2262 2282 args = last_call[1]
2263 2283
2264 2284 # use last_call to remember the state of the previous call, but don't
2265 2285 # let it be clobbered by successive '-p' calls.
2266 2286 try:
2267 2287 last_call[0] = self.shell.displayhook.prompt_count
2268 2288 if not opts_prev:
2269 2289 last_call[1] = parameter_s
2270 2290 except:
2271 2291 pass
2272 2292
2273 2293 # by default this is done with temp files, except when the given
2274 2294 # arg is a filename
2275 2295 use_temp = True
2276 2296
2277 2297 data = ''
2278 2298 if args.endswith('.py'):
2279 2299 filename = make_filename(args)
2280 2300 use_temp = False
2281 2301 elif args:
2282 2302 # Mode where user specifies ranges of lines, like in %macro.
2283 2303 data = self.extract_input_lines(args, opts_raw)
2284 2304 if not data:
2285 2305 try:
2286 2306 # Load the parameter given as a variable. If not a string,
2287 2307 # process it as an object instead (below)
2288 2308
2289 2309 #print '*** args',args,'type',type(args) # dbg
2290 2310 data = eval(args, self.shell.user_ns)
2291 2311 if not isinstance(data, basestring):
2292 2312 raise DataIsObject
2293 2313
2294 2314 except (NameError,SyntaxError):
2295 2315 # given argument is not a variable, try as a filename
2296 2316 filename = make_filename(args)
2297 2317 if filename is None:
2298 2318 warn("Argument given (%s) can't be found as a variable "
2299 2319 "or as a filename." % args)
2300 2320 return
2301 2321 use_temp = False
2302 2322
2303 2323 except DataIsObject:
2304 2324 # macros have a special edit function
2305 2325 if isinstance(data, Macro):
2306 2326 self._edit_macro(args,data)
2307 2327 return
2308 2328
2309 2329 # For objects, try to edit the file where they are defined
2310 2330 try:
2311 2331 filename = inspect.getabsfile(data)
2312 2332 if 'fakemodule' in filename.lower() and inspect.isclass(data):
2313 2333 # class created by %edit? Try to find source
2314 2334 # by looking for method definitions instead, the
2315 2335 # __module__ in those classes is FakeModule.
2316 2336 attrs = [getattr(data, aname) for aname in dir(data)]
2317 2337 for attr in attrs:
2318 2338 if not inspect.ismethod(attr):
2319 2339 continue
2320 2340 filename = inspect.getabsfile(attr)
2321 2341 if filename and 'fakemodule' not in filename.lower():
2322 2342 # change the attribute to be the edit target instead
2323 2343 data = attr
2324 2344 break
2325 2345
2326 2346 datafile = 1
2327 2347 except TypeError:
2328 2348 filename = make_filename(args)
2329 2349 datafile = 1
2330 2350 warn('Could not find file where `%s` is defined.\n'
2331 2351 'Opening a file named `%s`' % (args,filename))
2332 2352 # Now, make sure we can actually read the source (if it was in
2333 2353 # a temp file it's gone by now).
2334 2354 if datafile:
2335 2355 try:
2336 2356 if lineno is None:
2337 2357 lineno = inspect.getsourcelines(data)[1]
2338 2358 except IOError:
2339 2359 filename = make_filename(args)
2340 2360 if filename is None:
2341 2361 warn('The file `%s` where `%s` was defined cannot '
2342 2362 'be read.' % (filename,data))
2343 2363 return
2344 2364 use_temp = False
2345 2365
2346 2366 if use_temp:
2347 2367 filename = self.shell.mktempfile(data)
2348 2368 print 'IPython will make a temporary file named:',filename
2349 2369
2350 2370 # do actual editing here
2351 2371 print 'Editing...',
2352 2372 sys.stdout.flush()
2353 2373 try:
2354 2374 # Quote filenames that may have spaces in them
2355 2375 if ' ' in filename:
2356 2376 filename = "%s" % filename
2357 2377 self.shell.hooks.editor(filename,lineno)
2358 2378 except TryNext:
2359 2379 warn('Could not open editor')
2360 2380 return
2361 2381
2362 2382 # XXX TODO: should this be generalized for all string vars?
2363 2383 # For now, this is special-cased to blocks created by cpaste
2364 2384 if args.strip() == 'pasted_block':
2365 2385 self.shell.user_ns['pasted_block'] = file_read(filename)
2366 2386
2367 2387 if 'x' in opts: # -x prevents actual execution
2368 2388 print
2369 2389 else:
2370 2390 print 'done. Executing edited code...'
2371 2391 if opts_raw:
2372 2392 self.shell.run_cell(file_read(filename),
2373 2393 store_history=False)
2374 2394 else:
2375 2395 self.shell.safe_execfile(filename,self.shell.user_ns,
2376 2396 self.shell.user_ns)
2377 2397
2378 2398
2379 2399 if use_temp:
2380 2400 try:
2381 2401 return open(filename).read()
2382 2402 except IOError,msg:
2383 2403 if msg.filename == filename:
2384 2404 warn('File not found. Did you forget to save?')
2385 2405 return
2386 2406 else:
2387 2407 self.shell.showtraceback()
2388 2408
2389 2409 def magic_xmode(self,parameter_s = ''):
2390 2410 """Switch modes for the exception handlers.
2391 2411
2392 2412 Valid modes: Plain, Context and Verbose.
2393 2413
2394 2414 If called without arguments, acts as a toggle."""
2395 2415
2396 2416 def xmode_switch_err(name):
2397 2417 warn('Error changing %s exception modes.\n%s' %
2398 2418 (name,sys.exc_info()[1]))
2399 2419
2400 2420 shell = self.shell
2401 2421 new_mode = parameter_s.strip().capitalize()
2402 2422 try:
2403 2423 shell.InteractiveTB.set_mode(mode=new_mode)
2404 2424 print 'Exception reporting mode:',shell.InteractiveTB.mode
2405 2425 except:
2406 2426 xmode_switch_err('user')
2407 2427
2408 2428 def magic_colors(self,parameter_s = ''):
2409 2429 """Switch color scheme for prompts, info system and exception handlers.
2410 2430
2411 2431 Currently implemented schemes: NoColor, Linux, LightBG.
2412 2432
2413 2433 Color scheme names are not case-sensitive.
2414 2434
2415 2435 Examples
2416 2436 --------
2417 2437 To get a plain black and white terminal::
2418 2438
2419 2439 %colors nocolor
2420 2440 """
2421 2441
2422 2442 def color_switch_err(name):
2423 2443 warn('Error changing %s color schemes.\n%s' %
2424 2444 (name,sys.exc_info()[1]))
2425 2445
2426 2446
2427 2447 new_scheme = parameter_s.strip()
2428 2448 if not new_scheme:
2429 2449 raise UsageError(
2430 2450 "%colors: you must specify a color scheme. See '%colors?'")
2431 2451 return
2432 2452 # local shortcut
2433 2453 shell = self.shell
2434 2454
2435 2455 import IPython.utils.rlineimpl as readline
2436 2456
2437 2457 if not readline.have_readline and sys.platform == "win32":
2438 2458 msg = """\
2439 2459 Proper color support under MS Windows requires the pyreadline library.
2440 2460 You can find it at:
2441 2461 http://ipython.scipy.org/moin/PyReadline/Intro
2442 2462 Gary's readline needs the ctypes module, from:
2443 2463 http://starship.python.net/crew/theller/ctypes
2444 2464 (Note that ctypes is already part of Python versions 2.5 and newer).
2445 2465
2446 2466 Defaulting color scheme to 'NoColor'"""
2447 2467 new_scheme = 'NoColor'
2448 2468 warn(msg)
2449 2469
2450 2470 # readline option is 0
2451 2471 if not shell.has_readline:
2452 2472 new_scheme = 'NoColor'
2453 2473
2454 2474 # Set prompt colors
2455 2475 try:
2456 2476 shell.displayhook.set_colors(new_scheme)
2457 2477 except:
2458 2478 color_switch_err('prompt')
2459 2479 else:
2460 2480 shell.colors = \
2461 2481 shell.displayhook.color_table.active_scheme_name
2462 2482 # Set exception colors
2463 2483 try:
2464 2484 shell.InteractiveTB.set_colors(scheme = new_scheme)
2465 2485 shell.SyntaxTB.set_colors(scheme = new_scheme)
2466 2486 except:
2467 2487 color_switch_err('exception')
2468 2488
2469 2489 # Set info (for 'object?') colors
2470 2490 if shell.color_info:
2471 2491 try:
2472 2492 shell.inspector.set_active_scheme(new_scheme)
2473 2493 except:
2474 2494 color_switch_err('object inspector')
2475 2495 else:
2476 2496 shell.inspector.set_active_scheme('NoColor')
2477 2497
2478 2498 def magic_pprint(self, parameter_s=''):
2479 2499 """Toggle pretty printing on/off."""
2480 2500 ptformatter = self.shell.display_formatter.formatters['text/plain']
2481 2501 ptformatter.pprint = bool(1 - ptformatter.pprint)
2482 2502 print 'Pretty printing has been turned', \
2483 2503 ['OFF','ON'][ptformatter.pprint]
2484 2504
2485 2505 def magic_Exit(self, parameter_s=''):
2486 2506 """Exit IPython."""
2487 2507
2488 2508 self.shell.ask_exit()
2489 2509
2490 2510 # Add aliases as magics so all common forms work: exit, quit, Exit, Quit.
2491 2511 magic_exit = magic_quit = magic_Quit = magic_Exit
2492 2512
2493 2513 #......................................................................
2494 2514 # Functions to implement unix shell-type things
2495 2515
2496 2516 @testdec.skip_doctest
2497 2517 def magic_alias(self, parameter_s = ''):
2498 2518 """Define an alias for a system command.
2499 2519
2500 2520 '%alias alias_name cmd' defines 'alias_name' as an alias for 'cmd'
2501 2521
2502 2522 Then, typing 'alias_name params' will execute the system command 'cmd
2503 2523 params' (from your underlying operating system).
2504 2524
2505 2525 Aliases have lower precedence than magic functions and Python normal
2506 2526 variables, so if 'foo' is both a Python variable and an alias, the
2507 2527 alias can not be executed until 'del foo' removes the Python variable.
2508 2528
2509 2529 You can use the %l specifier in an alias definition to represent the
2510 2530 whole line when the alias is called. For example:
2511 2531
2512 2532 In [2]: alias bracket echo "Input in brackets: <%l>"
2513 2533 In [3]: bracket hello world
2514 2534 Input in brackets: <hello world>
2515 2535
2516 2536 You can also define aliases with parameters using %s specifiers (one
2517 2537 per parameter):
2518 2538
2519 2539 In [1]: alias parts echo first %s second %s
2520 2540 In [2]: %parts A B
2521 2541 first A second B
2522 2542 In [3]: %parts A
2523 2543 Incorrect number of arguments: 2 expected.
2524 2544 parts is an alias to: 'echo first %s second %s'
2525 2545
2526 2546 Note that %l and %s are mutually exclusive. You can only use one or
2527 2547 the other in your aliases.
2528 2548
2529 2549 Aliases expand Python variables just like system calls using ! or !!
2530 2550 do: all expressions prefixed with '$' get expanded. For details of
2531 2551 the semantic rules, see PEP-215:
2532 2552 http://www.python.org/peps/pep-0215.html. This is the library used by
2533 2553 IPython for variable expansion. If you want to access a true shell
2534 2554 variable, an extra $ is necessary to prevent its expansion by IPython:
2535 2555
2536 2556 In [6]: alias show echo
2537 2557 In [7]: PATH='A Python string'
2538 2558 In [8]: show $PATH
2539 2559 A Python string
2540 2560 In [9]: show $$PATH
2541 2561 /usr/local/lf9560/bin:/usr/local/intel/compiler70/ia32/bin:...
2542 2562
2543 2563 You can use the alias facility to acess all of $PATH. See the %rehash
2544 2564 and %rehashx functions, which automatically create aliases for the
2545 2565 contents of your $PATH.
2546 2566
2547 2567 If called with no parameters, %alias prints the current alias table."""
2548 2568
2549 2569 par = parameter_s.strip()
2550 2570 if not par:
2551 2571 stored = self.db.get('stored_aliases', {} )
2552 2572 aliases = sorted(self.shell.alias_manager.aliases)
2553 2573 # for k, v in stored:
2554 2574 # atab.append(k, v[0])
2555 2575
2556 2576 print "Total number of aliases:", len(aliases)
2557 2577 sys.stdout.flush()
2558 2578 return aliases
2559 2579
2560 2580 # Now try to define a new one
2561 2581 try:
2562 2582 alias,cmd = par.split(None, 1)
2563 2583 except:
2564 2584 print oinspect.getdoc(self.magic_alias)
2565 2585 else:
2566 2586 self.shell.alias_manager.soft_define_alias(alias, cmd)
2567 2587 # end magic_alias
2568 2588
2569 2589 def magic_unalias(self, parameter_s = ''):
2570 2590 """Remove an alias"""
2571 2591
2572 2592 aname = parameter_s.strip()
2573 2593 self.shell.alias_manager.undefine_alias(aname)
2574 2594 stored = self.db.get('stored_aliases', {} )
2575 2595 if aname in stored:
2576 2596 print "Removing %stored alias",aname
2577 2597 del stored[aname]
2578 2598 self.db['stored_aliases'] = stored
2579 2599
2580 2600 def magic_rehashx(self, parameter_s = ''):
2581 2601 """Update the alias table with all executable files in $PATH.
2582 2602
2583 2603 This version explicitly checks that every entry in $PATH is a file
2584 2604 with execute access (os.X_OK), so it is much slower than %rehash.
2585 2605
2586 2606 Under Windows, it checks executability as a match agains a
2587 2607 '|'-separated string of extensions, stored in the IPython config
2588 2608 variable win_exec_ext. This defaults to 'exe|com|bat'.
2589 2609
2590 2610 This function also resets the root module cache of module completer,
2591 2611 used on slow filesystems.
2592 2612 """
2593 2613 from IPython.core.alias import InvalidAliasError
2594 2614
2595 2615 # for the benefit of module completer in ipy_completers.py
2596 2616 del self.db['rootmodules']
2597 2617
2598 2618 path = [os.path.abspath(os.path.expanduser(p)) for p in
2599 2619 os.environ.get('PATH','').split(os.pathsep)]
2600 2620 path = filter(os.path.isdir,path)
2601 2621
2602 2622 syscmdlist = []
2603 2623 # Now define isexec in a cross platform manner.
2604 2624 if os.name == 'posix':
2605 2625 isexec = lambda fname:os.path.isfile(fname) and \
2606 2626 os.access(fname,os.X_OK)
2607 2627 else:
2608 2628 try:
2609 2629 winext = os.environ['pathext'].replace(';','|').replace('.','')
2610 2630 except KeyError:
2611 2631 winext = 'exe|com|bat|py'
2612 2632 if 'py' not in winext:
2613 2633 winext += '|py'
2614 2634 execre = re.compile(r'(.*)\.(%s)$' % winext,re.IGNORECASE)
2615 2635 isexec = lambda fname:os.path.isfile(fname) and execre.match(fname)
2616 2636 savedir = os.getcwd()
2617 2637
2618 2638 # Now walk the paths looking for executables to alias.
2619 2639 try:
2620 2640 # write the whole loop for posix/Windows so we don't have an if in
2621 2641 # the innermost part
2622 2642 if os.name == 'posix':
2623 2643 for pdir in path:
2624 2644 os.chdir(pdir)
2625 2645 for ff in os.listdir(pdir):
2626 2646 if isexec(ff):
2627 2647 try:
2628 2648 # Removes dots from the name since ipython
2629 2649 # will assume names with dots to be python.
2630 2650 self.shell.alias_manager.define_alias(
2631 2651 ff.replace('.',''), ff)
2632 2652 except InvalidAliasError:
2633 2653 pass
2634 2654 else:
2635 2655 syscmdlist.append(ff)
2636 2656 else:
2637 2657 no_alias = self.shell.alias_manager.no_alias
2638 2658 for pdir in path:
2639 2659 os.chdir(pdir)
2640 2660 for ff in os.listdir(pdir):
2641 2661 base, ext = os.path.splitext(ff)
2642 2662 if isexec(ff) and base.lower() not in no_alias:
2643 2663 if ext.lower() == '.exe':
2644 2664 ff = base
2645 2665 try:
2646 2666 # Removes dots from the name since ipython
2647 2667 # will assume names with dots to be python.
2648 2668 self.shell.alias_manager.define_alias(
2649 2669 base.lower().replace('.',''), ff)
2650 2670 except InvalidAliasError:
2651 2671 pass
2652 2672 syscmdlist.append(ff)
2653 2673 db = self.db
2654 2674 db['syscmdlist'] = syscmdlist
2655 2675 finally:
2656 2676 os.chdir(savedir)
2657 2677
2658 2678 @testdec.skip_doctest
2659 2679 def magic_pwd(self, parameter_s = ''):
2660 2680 """Return the current working directory path.
2661 2681
2662 2682 Examples
2663 2683 --------
2664 2684 ::
2665 2685
2666 2686 In [9]: pwd
2667 2687 Out[9]: '/home/tsuser/sprint/ipython'
2668 2688 """
2669 2689 return os.getcwd()
2670 2690
2671 2691 @testdec.skip_doctest
2672 2692 def magic_cd(self, parameter_s=''):
2673 2693 """Change the current working directory.
2674 2694
2675 2695 This command automatically maintains an internal list of directories
2676 2696 you visit during your IPython session, in the variable _dh. The
2677 2697 command %dhist shows this history nicely formatted. You can also
2678 2698 do 'cd -<tab>' to see directory history conveniently.
2679 2699
2680 2700 Usage:
2681 2701
2682 2702 cd 'dir': changes to directory 'dir'.
2683 2703
2684 2704 cd -: changes to the last visited directory.
2685 2705
2686 2706 cd -<n>: changes to the n-th directory in the directory history.
2687 2707
2688 2708 cd --foo: change to directory that matches 'foo' in history
2689 2709
2690 2710 cd -b <bookmark_name>: jump to a bookmark set by %bookmark
2691 2711 (note: cd <bookmark_name> is enough if there is no
2692 2712 directory <bookmark_name>, but a bookmark with the name exists.)
2693 2713 'cd -b <tab>' allows you to tab-complete bookmark names.
2694 2714
2695 2715 Options:
2696 2716
2697 2717 -q: quiet. Do not print the working directory after the cd command is
2698 2718 executed. By default IPython's cd command does print this directory,
2699 2719 since the default prompts do not display path information.
2700 2720
2701 2721 Note that !cd doesn't work for this purpose because the shell where
2702 2722 !command runs is immediately discarded after executing 'command'.
2703 2723
2704 2724 Examples
2705 2725 --------
2706 2726 ::
2707 2727
2708 2728 In [10]: cd parent/child
2709 2729 /home/tsuser/parent/child
2710 2730 """
2711 2731
2712 2732 parameter_s = parameter_s.strip()
2713 2733 #bkms = self.shell.persist.get("bookmarks",{})
2714 2734
2715 2735 oldcwd = os.getcwd()
2716 2736 numcd = re.match(r'(-)(\d+)$',parameter_s)
2717 2737 # jump in directory history by number
2718 2738 if numcd:
2719 2739 nn = int(numcd.group(2))
2720 2740 try:
2721 2741 ps = self.shell.user_ns['_dh'][nn]
2722 2742 except IndexError:
2723 2743 print 'The requested directory does not exist in history.'
2724 2744 return
2725 2745 else:
2726 2746 opts = {}
2727 2747 elif parameter_s.startswith('--'):
2728 2748 ps = None
2729 2749 fallback = None
2730 2750 pat = parameter_s[2:]
2731 2751 dh = self.shell.user_ns['_dh']
2732 2752 # first search only by basename (last component)
2733 2753 for ent in reversed(dh):
2734 2754 if pat in os.path.basename(ent) and os.path.isdir(ent):
2735 2755 ps = ent
2736 2756 break
2737 2757
2738 2758 if fallback is None and pat in ent and os.path.isdir(ent):
2739 2759 fallback = ent
2740 2760
2741 2761 # if we have no last part match, pick the first full path match
2742 2762 if ps is None:
2743 2763 ps = fallback
2744 2764
2745 2765 if ps is None:
2746 2766 print "No matching entry in directory history"
2747 2767 return
2748 2768 else:
2749 2769 opts = {}
2750 2770
2751 2771
2752 2772 else:
2753 2773 #turn all non-space-escaping backslashes to slashes,
2754 2774 # for c:\windows\directory\names\
2755 2775 parameter_s = re.sub(r'\\(?! )','/', parameter_s)
2756 2776 opts,ps = self.parse_options(parameter_s,'qb',mode='string')
2757 2777 # jump to previous
2758 2778 if ps == '-':
2759 2779 try:
2760 2780 ps = self.shell.user_ns['_dh'][-2]
2761 2781 except IndexError:
2762 2782 raise UsageError('%cd -: No previous directory to change to.')
2763 2783 # jump to bookmark if needed
2764 2784 else:
2765 2785 if not os.path.isdir(ps) or opts.has_key('b'):
2766 2786 bkms = self.db.get('bookmarks', {})
2767 2787
2768 2788 if bkms.has_key(ps):
2769 2789 target = bkms[ps]
2770 2790 print '(bookmark:%s) -> %s' % (ps,target)
2771 2791 ps = target
2772 2792 else:
2773 2793 if opts.has_key('b'):
2774 2794 raise UsageError("Bookmark '%s' not found. "
2775 2795 "Use '%%bookmark -l' to see your bookmarks." % ps)
2776 2796
2777 2797 # at this point ps should point to the target dir
2778 2798 if ps:
2779 2799 try:
2780 2800 os.chdir(os.path.expanduser(ps))
2781 2801 if hasattr(self.shell, 'term_title') and self.shell.term_title:
2782 2802 set_term_title('IPython: ' + abbrev_cwd())
2783 2803 except OSError:
2784 2804 print sys.exc_info()[1]
2785 2805 else:
2786 2806 cwd = os.getcwd()
2787 2807 dhist = self.shell.user_ns['_dh']
2788 2808 if oldcwd != cwd:
2789 2809 dhist.append(cwd)
2790 2810 self.db['dhist'] = compress_dhist(dhist)[-100:]
2791 2811
2792 2812 else:
2793 2813 os.chdir(self.shell.home_dir)
2794 2814 if hasattr(self.shell, 'term_title') and self.shell.term_title:
2795 2815 set_term_title('IPython: ' + '~')
2796 2816 cwd = os.getcwd()
2797 2817 dhist = self.shell.user_ns['_dh']
2798 2818
2799 2819 if oldcwd != cwd:
2800 2820 dhist.append(cwd)
2801 2821 self.db['dhist'] = compress_dhist(dhist)[-100:]
2802 2822 if not 'q' in opts and self.shell.user_ns['_dh']:
2803 2823 print self.shell.user_ns['_dh'][-1]
2804 2824
2805 2825
2806 2826 def magic_env(self, parameter_s=''):
2807 2827 """List environment variables."""
2808 2828
2809 2829 return os.environ.data
2810 2830
2811 2831 def magic_pushd(self, parameter_s=''):
2812 2832 """Place the current dir on stack and change directory.
2813 2833
2814 2834 Usage:\\
2815 2835 %pushd ['dirname']
2816 2836 """
2817 2837
2818 2838 dir_s = self.shell.dir_stack
2819 2839 tgt = os.path.expanduser(parameter_s)
2820 2840 cwd = os.getcwd().replace(self.home_dir,'~')
2821 2841 if tgt:
2822 2842 self.magic_cd(parameter_s)
2823 2843 dir_s.insert(0,cwd)
2824 2844 return self.magic_dirs()
2825 2845
2826 2846 def magic_popd(self, parameter_s=''):
2827 2847 """Change to directory popped off the top of the stack.
2828 2848 """
2829 2849 if not self.shell.dir_stack:
2830 2850 raise UsageError("%popd on empty stack")
2831 2851 top = self.shell.dir_stack.pop(0)
2832 2852 self.magic_cd(top)
2833 2853 print "popd ->",top
2834 2854
2835 2855 def magic_dirs(self, parameter_s=''):
2836 2856 """Return the current directory stack."""
2837 2857
2838 2858 return self.shell.dir_stack
2839 2859
2840 2860 def magic_dhist(self, parameter_s=''):
2841 2861 """Print your history of visited directories.
2842 2862
2843 2863 %dhist -> print full history\\
2844 2864 %dhist n -> print last n entries only\\
2845 2865 %dhist n1 n2 -> print entries between n1 and n2 (n1 not included)\\
2846 2866
2847 2867 This history is automatically maintained by the %cd command, and
2848 2868 always available as the global list variable _dh. You can use %cd -<n>
2849 2869 to go to directory number <n>.
2850 2870
2851 2871 Note that most of time, you should view directory history by entering
2852 2872 cd -<TAB>.
2853 2873
2854 2874 """
2855 2875
2856 2876 dh = self.shell.user_ns['_dh']
2857 2877 if parameter_s:
2858 2878 try:
2859 2879 args = map(int,parameter_s.split())
2860 2880 except:
2861 2881 self.arg_err(Magic.magic_dhist)
2862 2882 return
2863 2883 if len(args) == 1:
2864 2884 ini,fin = max(len(dh)-(args[0]),0),len(dh)
2865 2885 elif len(args) == 2:
2866 2886 ini,fin = args
2867 2887 else:
2868 2888 self.arg_err(Magic.magic_dhist)
2869 2889 return
2870 2890 else:
2871 2891 ini,fin = 0,len(dh)
2872 2892 nlprint(dh,
2873 2893 header = 'Directory history (kept in _dh)',
2874 2894 start=ini,stop=fin)
2875 2895
2876 2896 @testdec.skip_doctest
2877 2897 def magic_sc(self, parameter_s=''):
2878 2898 """Shell capture - execute a shell command and capture its output.
2879 2899
2880 2900 DEPRECATED. Suboptimal, retained for backwards compatibility.
2881 2901
2882 2902 You should use the form 'var = !command' instead. Example:
2883 2903
2884 2904 "%sc -l myfiles = ls ~" should now be written as
2885 2905
2886 2906 "myfiles = !ls ~"
2887 2907
2888 2908 myfiles.s, myfiles.l and myfiles.n still apply as documented
2889 2909 below.
2890 2910
2891 2911 --
2892 2912 %sc [options] varname=command
2893 2913
2894 2914 IPython will run the given command using commands.getoutput(), and
2895 2915 will then update the user's interactive namespace with a variable
2896 2916 called varname, containing the value of the call. Your command can
2897 2917 contain shell wildcards, pipes, etc.
2898 2918
2899 2919 The '=' sign in the syntax is mandatory, and the variable name you
2900 2920 supply must follow Python's standard conventions for valid names.
2901 2921
2902 2922 (A special format without variable name exists for internal use)
2903 2923
2904 2924 Options:
2905 2925
2906 2926 -l: list output. Split the output on newlines into a list before
2907 2927 assigning it to the given variable. By default the output is stored
2908 2928 as a single string.
2909 2929
2910 2930 -v: verbose. Print the contents of the variable.
2911 2931
2912 2932 In most cases you should not need to split as a list, because the
2913 2933 returned value is a special type of string which can automatically
2914 2934 provide its contents either as a list (split on newlines) or as a
2915 2935 space-separated string. These are convenient, respectively, either
2916 2936 for sequential processing or to be passed to a shell command.
2917 2937
2918 2938 For example:
2919 2939
2920 2940 # all-random
2921 2941
2922 2942 # Capture into variable a
2923 2943 In [1]: sc a=ls *py
2924 2944
2925 2945 # a is a string with embedded newlines
2926 2946 In [2]: a
2927 2947 Out[2]: 'setup.py\\nwin32_manual_post_install.py'
2928 2948
2929 2949 # which can be seen as a list:
2930 2950 In [3]: a.l
2931 2951 Out[3]: ['setup.py', 'win32_manual_post_install.py']
2932 2952
2933 2953 # or as a whitespace-separated string:
2934 2954 In [4]: a.s
2935 2955 Out[4]: 'setup.py win32_manual_post_install.py'
2936 2956
2937 2957 # a.s is useful to pass as a single command line:
2938 2958 In [5]: !wc -l $a.s
2939 2959 146 setup.py
2940 2960 130 win32_manual_post_install.py
2941 2961 276 total
2942 2962
2943 2963 # while the list form is useful to loop over:
2944 2964 In [6]: for f in a.l:
2945 2965 ...: !wc -l $f
2946 2966 ...:
2947 2967 146 setup.py
2948 2968 130 win32_manual_post_install.py
2949 2969
2950 2970 Similiarly, the lists returned by the -l option are also special, in
2951 2971 the sense that you can equally invoke the .s attribute on them to
2952 2972 automatically get a whitespace-separated string from their contents:
2953 2973
2954 2974 In [7]: sc -l b=ls *py
2955 2975
2956 2976 In [8]: b
2957 2977 Out[8]: ['setup.py', 'win32_manual_post_install.py']
2958 2978
2959 2979 In [9]: b.s
2960 2980 Out[9]: 'setup.py win32_manual_post_install.py'
2961 2981
2962 2982 In summary, both the lists and strings used for ouptut capture have
2963 2983 the following special attributes:
2964 2984
2965 2985 .l (or .list) : value as list.
2966 2986 .n (or .nlstr): value as newline-separated string.
2967 2987 .s (or .spstr): value as space-separated string.
2968 2988 """
2969 2989
2970 2990 opts,args = self.parse_options(parameter_s,'lv')
2971 2991 # Try to get a variable name and command to run
2972 2992 try:
2973 2993 # the variable name must be obtained from the parse_options
2974 2994 # output, which uses shlex.split to strip options out.
2975 2995 var,_ = args.split('=',1)
2976 2996 var = var.strip()
2977 2997 # But the the command has to be extracted from the original input
2978 2998 # parameter_s, not on what parse_options returns, to avoid the
2979 2999 # quote stripping which shlex.split performs on it.
2980 3000 _,cmd = parameter_s.split('=',1)
2981 3001 except ValueError:
2982 3002 var,cmd = '',''
2983 3003 # If all looks ok, proceed
2984 3004 split = 'l' in opts
2985 3005 out = self.shell.getoutput(cmd, split=split)
2986 3006 if opts.has_key('v'):
2987 3007 print '%s ==\n%s' % (var,pformat(out))
2988 3008 if var:
2989 3009 self.shell.user_ns.update({var:out})
2990 3010 else:
2991 3011 return out
2992 3012
2993 3013 def magic_sx(self, parameter_s=''):
2994 3014 """Shell execute - run a shell command and capture its output.
2995 3015
2996 3016 %sx command
2997 3017
2998 3018 IPython will run the given command using commands.getoutput(), and
2999 3019 return the result formatted as a list (split on '\\n'). Since the
3000 3020 output is _returned_, it will be stored in ipython's regular output
3001 3021 cache Out[N] and in the '_N' automatic variables.
3002 3022
3003 3023 Notes:
3004 3024
3005 3025 1) If an input line begins with '!!', then %sx is automatically
3006 3026 invoked. That is, while:
3007 3027 !ls
3008 3028 causes ipython to simply issue system('ls'), typing
3009 3029 !!ls
3010 3030 is a shorthand equivalent to:
3011 3031 %sx ls
3012 3032
3013 3033 2) %sx differs from %sc in that %sx automatically splits into a list,
3014 3034 like '%sc -l'. The reason for this is to make it as easy as possible
3015 3035 to process line-oriented shell output via further python commands.
3016 3036 %sc is meant to provide much finer control, but requires more
3017 3037 typing.
3018 3038
3019 3039 3) Just like %sc -l, this is a list with special attributes:
3020 3040
3021 3041 .l (or .list) : value as list.
3022 3042 .n (or .nlstr): value as newline-separated string.
3023 3043 .s (or .spstr): value as whitespace-separated string.
3024 3044
3025 3045 This is very useful when trying to use such lists as arguments to
3026 3046 system commands."""
3027 3047
3028 3048 if parameter_s:
3029 3049 return self.shell.getoutput(parameter_s)
3030 3050
3031 3051
3032 3052 def magic_bookmark(self, parameter_s=''):
3033 3053 """Manage IPython's bookmark system.
3034 3054
3035 3055 %bookmark <name> - set bookmark to current dir
3036 3056 %bookmark <name> <dir> - set bookmark to <dir>
3037 3057 %bookmark -l - list all bookmarks
3038 3058 %bookmark -d <name> - remove bookmark
3039 3059 %bookmark -r - remove all bookmarks
3040 3060
3041 3061 You can later on access a bookmarked folder with:
3042 3062 %cd -b <name>
3043 3063 or simply '%cd <name>' if there is no directory called <name> AND
3044 3064 there is such a bookmark defined.
3045 3065
3046 3066 Your bookmarks persist through IPython sessions, but they are
3047 3067 associated with each profile."""
3048 3068
3049 3069 opts,args = self.parse_options(parameter_s,'drl',mode='list')
3050 3070 if len(args) > 2:
3051 3071 raise UsageError("%bookmark: too many arguments")
3052 3072
3053 3073 bkms = self.db.get('bookmarks',{})
3054 3074
3055 3075 if opts.has_key('d'):
3056 3076 try:
3057 3077 todel = args[0]
3058 3078 except IndexError:
3059 3079 raise UsageError(
3060 3080 "%bookmark -d: must provide a bookmark to delete")
3061 3081 else:
3062 3082 try:
3063 3083 del bkms[todel]
3064 3084 except KeyError:
3065 3085 raise UsageError(
3066 3086 "%%bookmark -d: Can't delete bookmark '%s'" % todel)
3067 3087
3068 3088 elif opts.has_key('r'):
3069 3089 bkms = {}
3070 3090 elif opts.has_key('l'):
3071 3091 bks = bkms.keys()
3072 3092 bks.sort()
3073 3093 if bks:
3074 3094 size = max(map(len,bks))
3075 3095 else:
3076 3096 size = 0
3077 3097 fmt = '%-'+str(size)+'s -> %s'
3078 3098 print 'Current bookmarks:'
3079 3099 for bk in bks:
3080 3100 print fmt % (bk,bkms[bk])
3081 3101 else:
3082 3102 if not args:
3083 3103 raise UsageError("%bookmark: You must specify the bookmark name")
3084 3104 elif len(args)==1:
3085 3105 bkms[args[0]] = os.getcwd()
3086 3106 elif len(args)==2:
3087 3107 bkms[args[0]] = args[1]
3088 3108 self.db['bookmarks'] = bkms
3089 3109
3090 3110 def magic_pycat(self, parameter_s=''):
3091 3111 """Show a syntax-highlighted file through a pager.
3092 3112
3093 3113 This magic is similar to the cat utility, but it will assume the file
3094 3114 to be Python source and will show it with syntax highlighting. """
3095 3115
3096 3116 try:
3097 3117 filename = get_py_filename(parameter_s)
3098 3118 cont = file_read(filename)
3099 3119 except IOError:
3100 3120 try:
3101 3121 cont = eval(parameter_s,self.user_ns)
3102 3122 except NameError:
3103 3123 cont = None
3104 3124 if cont is None:
3105 3125 print "Error: no such file or variable"
3106 3126 return
3107 3127
3108 3128 page.page(self.shell.pycolorize(cont))
3109 3129
3110 3130 def _rerun_pasted(self):
3111 3131 """ Rerun a previously pasted command.
3112 3132 """
3113 3133 b = self.user_ns.get('pasted_block', None)
3114 3134 if b is None:
3115 3135 raise UsageError('No previous pasted block available')
3116 3136 print "Re-executing '%s...' (%d chars)"% (b.split('\n',1)[0], len(b))
3117 3137 exec b in self.user_ns
3118 3138
3119 3139 def _get_pasted_lines(self, sentinel):
3120 3140 """ Yield pasted lines until the user enters the given sentinel value.
3121 3141 """
3122 3142 from IPython.core import interactiveshell
3123 3143 print "Pasting code; enter '%s' alone on the line to stop." % sentinel
3124 3144 while True:
3125 3145 l = interactiveshell.raw_input_original(':')
3126 3146 if l == sentinel:
3127 3147 return
3128 3148 else:
3129 3149 yield l
3130 3150
3131 3151 def _strip_pasted_lines_for_code(self, raw_lines):
3132 3152 """ Strip non-code parts of a sequence of lines to return a block of
3133 3153 code.
3134 3154 """
3135 3155 # Regular expressions that declare text we strip from the input:
3136 3156 strip_re = [r'^\s*In \[\d+\]:', # IPython input prompt
3137 3157 r'^\s*(\s?>)+', # Python input prompt
3138 3158 r'^\s*\.{3,}', # Continuation prompts
3139 3159 r'^\++',
3140 3160 ]
3141 3161
3142 3162 strip_from_start = map(re.compile,strip_re)
3143 3163
3144 3164 lines = []
3145 3165 for l in raw_lines:
3146 3166 for pat in strip_from_start:
3147 3167 l = pat.sub('',l)
3148 3168 lines.append(l)
3149 3169
3150 3170 block = "\n".join(lines) + '\n'
3151 3171 #print "block:\n",block
3152 3172 return block
3153 3173
3154 3174 def _execute_block(self, block, par):
3155 3175 """ Execute a block, or store it in a variable, per the user's request.
3156 3176 """
3157 3177 if not par:
3158 3178 b = textwrap.dedent(block)
3159 3179 self.user_ns['pasted_block'] = b
3160 3180 exec b in self.user_ns
3161 3181 else:
3162 3182 self.user_ns[par] = SList(block.splitlines())
3163 3183 print "Block assigned to '%s'" % par
3164 3184
3165 3185 def magic_quickref(self,arg):
3166 3186 """ Show a quick reference sheet """
3167 3187 import IPython.core.usage
3168 3188 qr = IPython.core.usage.quick_reference + self.magic_magic('-brief')
3169 3189
3170 3190 page.page(qr)
3171 3191
3172 3192 def magic_doctest_mode(self,parameter_s=''):
3173 3193 """Toggle doctest mode on and off.
3174 3194
3175 3195 This mode is intended to make IPython behave as much as possible like a
3176 3196 plain Python shell, from the perspective of how its prompts, exceptions
3177 3197 and output look. This makes it easy to copy and paste parts of a
3178 3198 session into doctests. It does so by:
3179 3199
3180 3200 - Changing the prompts to the classic ``>>>`` ones.
3181 3201 - Changing the exception reporting mode to 'Plain'.
3182 3202 - Disabling pretty-printing of output.
3183 3203
3184 3204 Note that IPython also supports the pasting of code snippets that have
3185 3205 leading '>>>' and '...' prompts in them. This means that you can paste
3186 3206 doctests from files or docstrings (even if they have leading
3187 3207 whitespace), and the code will execute correctly. You can then use
3188 3208 '%history -t' to see the translated history; this will give you the
3189 3209 input after removal of all the leading prompts and whitespace, which
3190 3210 can be pasted back into an editor.
3191 3211
3192 3212 With these features, you can switch into this mode easily whenever you
3193 3213 need to do testing and changes to doctests, without having to leave
3194 3214 your existing IPython session.
3195 3215 """
3196 3216
3197 3217 from IPython.utils.ipstruct import Struct
3198 3218
3199 3219 # Shorthands
3200 3220 shell = self.shell
3201 3221 oc = shell.displayhook
3202 3222 meta = shell.meta
3203 3223 disp_formatter = self.shell.display_formatter
3204 3224 ptformatter = disp_formatter.formatters['text/plain']
3205 3225 # dstore is a data store kept in the instance metadata bag to track any
3206 3226 # changes we make, so we can undo them later.
3207 3227 dstore = meta.setdefault('doctest_mode',Struct())
3208 3228 save_dstore = dstore.setdefault
3209 3229
3210 3230 # save a few values we'll need to recover later
3211 3231 mode = save_dstore('mode',False)
3212 3232 save_dstore('rc_pprint',ptformatter.pprint)
3213 3233 save_dstore('xmode',shell.InteractiveTB.mode)
3214 3234 save_dstore('rc_separate_out',shell.separate_out)
3215 3235 save_dstore('rc_separate_out2',shell.separate_out2)
3216 3236 save_dstore('rc_prompts_pad_left',shell.prompts_pad_left)
3217 3237 save_dstore('rc_separate_in',shell.separate_in)
3218 3238 save_dstore('rc_plain_text_only',disp_formatter.plain_text_only)
3219 3239
3220 3240 if mode == False:
3221 3241 # turn on
3222 3242 oc.prompt1.p_template = '>>> '
3223 3243 oc.prompt2.p_template = '... '
3224 3244 oc.prompt_out.p_template = ''
3225 3245
3226 3246 # Prompt separators like plain python
3227 3247 oc.input_sep = oc.prompt1.sep = ''
3228 3248 oc.output_sep = ''
3229 3249 oc.output_sep2 = ''
3230 3250
3231 3251 oc.prompt1.pad_left = oc.prompt2.pad_left = \
3232 3252 oc.prompt_out.pad_left = False
3233 3253
3234 3254 ptformatter.pprint = False
3235 3255 disp_formatter.plain_text_only = True
3236 3256
3237 3257 shell.magic_xmode('Plain')
3238 3258 else:
3239 3259 # turn off
3240 3260 oc.prompt1.p_template = shell.prompt_in1
3241 3261 oc.prompt2.p_template = shell.prompt_in2
3242 3262 oc.prompt_out.p_template = shell.prompt_out
3243 3263
3244 3264 oc.input_sep = oc.prompt1.sep = dstore.rc_separate_in
3245 3265
3246 3266 oc.output_sep = dstore.rc_separate_out
3247 3267 oc.output_sep2 = dstore.rc_separate_out2
3248 3268
3249 3269 oc.prompt1.pad_left = oc.prompt2.pad_left = \
3250 3270 oc.prompt_out.pad_left = dstore.rc_prompts_pad_left
3251 3271
3252 3272 ptformatter.pprint = dstore.rc_pprint
3253 3273 disp_formatter.plain_text_only = dstore.rc_plain_text_only
3254 3274
3255 3275 shell.magic_xmode(dstore.xmode)
3256 3276
3257 3277 # Store new mode and inform
3258 3278 dstore.mode = bool(1-int(mode))
3259 3279 mode_label = ['OFF','ON'][dstore.mode]
3260 3280 print 'Doctest mode is:', mode_label
3261 3281
3262 3282 def magic_gui(self, parameter_s=''):
3263 3283 """Enable or disable IPython GUI event loop integration.
3264 3284
3265 3285 %gui [GUINAME]
3266 3286
3267 3287 This magic replaces IPython's threaded shells that were activated
3268 3288 using the (pylab/wthread/etc.) command line flags. GUI toolkits
3269 3289 can now be enabled, disabled and swtiched at runtime and keyboard
3270 3290 interrupts should work without any problems. The following toolkits
3271 3291 are supported: wxPython, PyQt4, PyGTK, and Tk::
3272 3292
3273 3293 %gui wx # enable wxPython event loop integration
3274 3294 %gui qt4|qt # enable PyQt4 event loop integration
3275 3295 %gui gtk # enable PyGTK event loop integration
3276 3296 %gui tk # enable Tk event loop integration
3277 3297 %gui # disable all event loop integration
3278 3298
3279 3299 WARNING: after any of these has been called you can simply create
3280 3300 an application object, but DO NOT start the event loop yourself, as
3281 3301 we have already handled that.
3282 3302 """
3283 3303 from IPython.lib.inputhook import enable_gui
3284 3304 opts, arg = self.parse_options(parameter_s, '')
3285 3305 if arg=='': arg = None
3286 3306 return enable_gui(arg)
3287 3307
3288 3308 def magic_load_ext(self, module_str):
3289 3309 """Load an IPython extension by its module name."""
3290 3310 return self.extension_manager.load_extension(module_str)
3291 3311
3292 3312 def magic_unload_ext(self, module_str):
3293 3313 """Unload an IPython extension by its module name."""
3294 3314 self.extension_manager.unload_extension(module_str)
3295 3315
3296 3316 def magic_reload_ext(self, module_str):
3297 3317 """Reload an IPython extension by its module name."""
3298 3318 self.extension_manager.reload_extension(module_str)
3299 3319
3300 3320 @testdec.skip_doctest
3301 3321 def magic_install_profiles(self, s):
3302 3322 """Install the default IPython profiles into the .ipython dir.
3303 3323
3304 3324 If the default profiles have already been installed, they will not
3305 3325 be overwritten. You can force overwriting them by using the ``-o``
3306 3326 option::
3307 3327
3308 3328 In [1]: %install_profiles -o
3309 3329 """
3310 3330 if '-o' in s:
3311 3331 overwrite = True
3312 3332 else:
3313 3333 overwrite = False
3314 3334 from IPython.config import profile
3315 3335 profile_dir = os.path.split(profile.__file__)[0]
3316 3336 ipython_dir = self.ipython_dir
3317 3337 files = os.listdir(profile_dir)
3318 3338
3319 3339 to_install = []
3320 3340 for f in files:
3321 3341 if f.startswith('ipython_config'):
3322 3342 src = os.path.join(profile_dir, f)
3323 3343 dst = os.path.join(ipython_dir, f)
3324 3344 if (not os.path.isfile(dst)) or overwrite:
3325 3345 to_install.append((f, src, dst))
3326 3346 if len(to_install)>0:
3327 3347 print "Installing profiles to: ", ipython_dir
3328 3348 for (f, src, dst) in to_install:
3329 3349 shutil.copy(src, dst)
3330 3350 print " %s" % f
3331 3351
3332 3352 def magic_install_default_config(self, s):
3333 3353 """Install IPython's default config file into the .ipython dir.
3334 3354
3335 3355 If the default config file (:file:`ipython_config.py`) is already
3336 3356 installed, it will not be overwritten. You can force overwriting
3337 3357 by using the ``-o`` option::
3338 3358
3339 3359 In [1]: %install_default_config
3340 3360 """
3341 3361 if '-o' in s:
3342 3362 overwrite = True
3343 3363 else:
3344 3364 overwrite = False
3345 3365 from IPython.config import default
3346 3366 config_dir = os.path.split(default.__file__)[0]
3347 3367 ipython_dir = self.ipython_dir
3348 3368 default_config_file_name = 'ipython_config.py'
3349 3369 src = os.path.join(config_dir, default_config_file_name)
3350 3370 dst = os.path.join(ipython_dir, default_config_file_name)
3351 3371 if (not os.path.isfile(dst)) or overwrite:
3352 3372 shutil.copy(src, dst)
3353 3373 print "Installing default config file: %s" % dst
3354 3374
3355 3375 # Pylab support: simple wrappers that activate pylab, load gui input
3356 3376 # handling and modify slightly %run
3357 3377
3358 3378 @testdec.skip_doctest
3359 3379 def _pylab_magic_run(self, parameter_s=''):
3360 3380 Magic.magic_run(self, parameter_s,
3361 3381 runner=mpl_runner(self.shell.safe_execfile))
3362 3382
3363 3383 _pylab_magic_run.__doc__ = magic_run.__doc__
3364 3384
3365 3385 @testdec.skip_doctest
3366 3386 def magic_pylab(self, s):
3367 3387 """Load numpy and matplotlib to work interactively.
3368 3388
3369 3389 %pylab [GUINAME]
3370 3390
3371 3391 This function lets you activate pylab (matplotlib, numpy and
3372 3392 interactive support) at any point during an IPython session.
3373 3393
3374 3394 It will import at the top level numpy as np, pyplot as plt, matplotlib,
3375 3395 pylab and mlab, as well as all names from numpy and pylab.
3376 3396
3377 3397 Parameters
3378 3398 ----------
3379 3399 guiname : optional
3380 3400 One of the valid arguments to the %gui magic ('qt', 'wx', 'gtk', 'osx' or
3381 3401 'tk'). If given, the corresponding Matplotlib backend is used,
3382 3402 otherwise matplotlib's default (which you can override in your
3383 3403 matplotlib config file) is used.
3384 3404
3385 3405 Examples
3386 3406 --------
3387 3407 In this case, where the MPL default is TkAgg:
3388 3408 In [2]: %pylab
3389 3409
3390 3410 Welcome to pylab, a matplotlib-based Python environment.
3391 3411 Backend in use: TkAgg
3392 3412 For more information, type 'help(pylab)'.
3393 3413
3394 3414 But you can explicitly request a different backend:
3395 3415 In [3]: %pylab qt
3396 3416
3397 3417 Welcome to pylab, a matplotlib-based Python environment.
3398 3418 Backend in use: Qt4Agg
3399 3419 For more information, type 'help(pylab)'.
3400 3420 """
3401 3421 self.shell.enable_pylab(s)
3402 3422
3403 3423 def magic_tb(self, s):
3404 3424 """Print the last traceback with the currently active exception mode.
3405 3425
3406 3426 See %xmode for changing exception reporting modes."""
3407 3427 self.shell.showtraceback()
3408 3428
3409 3429 @testdec.skip_doctest
3410 3430 def magic_precision(self, s=''):
3411 3431 """Set floating point precision for pretty printing.
3412 3432
3413 3433 Can set either integer precision or a format string.
3414 3434
3415 3435 If numpy has been imported and precision is an int,
3416 3436 numpy display precision will also be set, via ``numpy.set_printoptions``.
3417 3437
3418 3438 If no argument is given, defaults will be restored.
3419 3439
3420 3440 Examples
3421 3441 --------
3422 3442 ::
3423 3443
3424 3444 In [1]: from math import pi
3425 3445
3426 3446 In [2]: %precision 3
3427 3447 Out[2]: '%.3f'
3428 3448
3429 3449 In [3]: pi
3430 3450 Out[3]: 3.142
3431 3451
3432 3452 In [4]: %precision %i
3433 3453 Out[4]: '%i'
3434 3454
3435 3455 In [5]: pi
3436 3456 Out[5]: 3
3437 3457
3438 3458 In [6]: %precision %e
3439 3459 Out[6]: '%e'
3440 3460
3441 3461 In [7]: pi**10
3442 3462 Out[7]: 9.364805e+04
3443 3463
3444 3464 In [8]: %precision
3445 3465 Out[8]: '%r'
3446 3466
3447 3467 In [9]: pi**10
3448 3468 Out[9]: 93648.047476082982
3449 3469
3450 3470 """
3451 3471
3452 3472 ptformatter = self.shell.display_formatter.formatters['text/plain']
3453 3473 ptformatter.float_precision = s
3454 3474 return ptformatter.float_format
3455 3475
3456 3476 # end Magic
General Comments 0
You need to be logged in to leave comments. Login now