##// END OF EJS Templates
%unalias no longer croaks if there are no stored_aliases
vivainio -
Show More
@@ -1,3024 +1,3024 b''
1 1 # -*- coding: utf-8 -*-
2 2 """Magic functions for InteractiveShell.
3 3
4 $Id: Magic.py 1825 2006-10-13 21:29:33Z vivainio $"""
4 $Id: Magic.py 1829 2006-10-16 08:04:11Z vivainio $"""
5 5
6 6 #*****************************************************************************
7 7 # Copyright (C) 2001 Janko Hauser <jhauser@zscout.de> and
8 8 # Copyright (C) 2001-2006 Fernando Perez <fperez@colorado.edu>
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 # Modules and globals
16 16
17 17 from IPython import Release
18 18 __author__ = '%s <%s>\n%s <%s>' % \
19 19 ( Release.authors['Janko'] + Release.authors['Fernando'] )
20 20 __license__ = Release.license
21 21
22 22 # Python standard modules
23 23 import __builtin__
24 24 import bdb
25 25 import inspect
26 26 import os
27 27 import pdb
28 28 import pydoc
29 29 import shlex
30 30 import sys
31 31 import re
32 32 import tempfile
33 33 import time
34 34 import cPickle as pickle
35 35 import textwrap
36 36 from cStringIO import StringIO
37 37 from getopt import getopt,GetoptError
38 38 from pprint import pprint, pformat
39 39
40 40 # profile isn't bundled by default in Debian for license reasons
41 41 try:
42 42 import profile,pstats
43 43 except ImportError:
44 44 profile = pstats = None
45 45
46 46 # Homebrewed
47 47 import IPython
48 48 from IPython import Debugger, OInspect, wildcard
49 49 from IPython.FakeModule import FakeModule
50 50 from IPython.Itpl import Itpl, itpl, printpl,itplns
51 51 from IPython.PyColorize import Parser
52 52 from IPython.ipstruct import Struct
53 53 from IPython.macro import Macro
54 54 from IPython.genutils import *
55 55 from IPython import platutils
56 56
57 57 #***************************************************************************
58 58 # Utility functions
59 59 def on_off(tag):
60 60 """Return an ON/OFF string for a 1/0 input. Simple utility function."""
61 61 return ['OFF','ON'][tag]
62 62
63 63 class Bunch: pass
64 64
65 65 def arg_split(s,posix=True):
66 66 """Split a command line's arguments in a shell-like manner.
67 67
68 68 This is a modified version of the standard library's shlex.split()
69 69 function, but with a default of posix=False for splitting, so that quotes
70 70 in inputs are respected."""
71 71
72 72 lex = shlex.shlex(s, posix=posix)
73 73 lex.whitespace_split = True
74 74 return list(lex)
75 75
76 76 #***************************************************************************
77 77 # Main class implementing Magic functionality
78 78 class Magic:
79 79 """Magic functions for InteractiveShell.
80 80
81 81 Shell functions which can be reached as %function_name. All magic
82 82 functions should accept a string, which they can parse for their own
83 83 needs. This can make some functions easier to type, eg `%cd ../`
84 84 vs. `%cd("../")`
85 85
86 86 ALL definitions MUST begin with the prefix magic_. The user won't need it
87 87 at the command line, but it is is needed in the definition. """
88 88
89 89 # class globals
90 90 auto_status = ['Automagic is OFF, % prefix IS needed for magic functions.',
91 91 'Automagic is ON, % prefix NOT needed for magic functions.']
92 92
93 93 #......................................................................
94 94 # some utility functions
95 95
96 96 def __init__(self,shell):
97 97
98 98 self.options_table = {}
99 99 if profile is None:
100 100 self.magic_prun = self.profile_missing_notice
101 101 self.shell = shell
102 102
103 103 # namespace for holding state we may need
104 104 self._magic_state = Bunch()
105 105
106 106 def profile_missing_notice(self, *args, **kwargs):
107 107 error("""\
108 108 The profile module could not be found. If you are a Debian user,
109 109 it has been removed from the standard Debian package because of its non-free
110 110 license. To use profiling, please install"python2.3-profiler" from non-free.""")
111 111
112 112 def default_option(self,fn,optstr):
113 113 """Make an entry in the options_table for fn, with value optstr"""
114 114
115 115 if fn not in self.lsmagic():
116 116 error("%s is not a magic function" % fn)
117 117 self.options_table[fn] = optstr
118 118
119 119 def lsmagic(self):
120 120 """Return a list of currently available magic functions.
121 121
122 122 Gives a list of the bare names after mangling (['ls','cd', ...], not
123 123 ['magic_ls','magic_cd',...]"""
124 124
125 125 # FIXME. This needs a cleanup, in the way the magics list is built.
126 126
127 127 # magics in class definition
128 128 class_magic = lambda fn: fn.startswith('magic_') and \
129 129 callable(Magic.__dict__[fn])
130 130 # in instance namespace (run-time user additions)
131 131 inst_magic = lambda fn: fn.startswith('magic_') and \
132 132 callable(self.__dict__[fn])
133 133 # and bound magics by user (so they can access self):
134 134 inst_bound_magic = lambda fn: fn.startswith('magic_') and \
135 135 callable(self.__class__.__dict__[fn])
136 136 magics = filter(class_magic,Magic.__dict__.keys()) + \
137 137 filter(inst_magic,self.__dict__.keys()) + \
138 138 filter(inst_bound_magic,self.__class__.__dict__.keys())
139 139 out = []
140 140 for fn in magics:
141 141 out.append(fn.replace('magic_','',1))
142 142 out.sort()
143 143 return out
144 144
145 145 def extract_input_slices(self,slices,raw=False):
146 146 """Return as a string a set of input history slices.
147 147
148 148 Inputs:
149 149
150 150 - slices: the set of slices is given as a list of strings (like
151 151 ['1','4:8','9'], since this function is for use by magic functions
152 152 which get their arguments as strings.
153 153
154 154 Optional inputs:
155 155
156 156 - raw(False): by default, the processed input is used. If this is
157 157 true, the raw input history is used instead.
158 158
159 159 Note that slices can be called with two notations:
160 160
161 161 N:M -> standard python form, means including items N...(M-1).
162 162
163 163 N-M -> include items N..M (closed endpoint)."""
164 164
165 165 if raw:
166 166 hist = self.shell.input_hist_raw
167 167 else:
168 168 hist = self.shell.input_hist
169 169
170 170 cmds = []
171 171 for chunk in slices:
172 172 if ':' in chunk:
173 173 ini,fin = map(int,chunk.split(':'))
174 174 elif '-' in chunk:
175 175 ini,fin = map(int,chunk.split('-'))
176 176 fin += 1
177 177 else:
178 178 ini = int(chunk)
179 179 fin = ini+1
180 180 cmds.append(hist[ini:fin])
181 181 return cmds
182 182
183 183 def _ofind(self, oname, namespaces=None):
184 184 """Find an object in the available namespaces.
185 185
186 186 self._ofind(oname) -> dict with keys: found,obj,ospace,ismagic
187 187
188 188 Has special code to detect magic functions.
189 189 """
190 190
191 191 oname = oname.strip()
192 192
193 193 alias_ns = None
194 194 if namespaces is None:
195 195 # Namespaces to search in:
196 196 # Put them in a list. The order is important so that we
197 197 # find things in the same order that Python finds them.
198 198 namespaces = [ ('Interactive', self.shell.user_ns),
199 199 ('IPython internal', self.shell.internal_ns),
200 200 ('Python builtin', __builtin__.__dict__),
201 201 ('Alias', self.shell.alias_table),
202 202 ]
203 203 alias_ns = self.shell.alias_table
204 204
205 205 # initialize results to 'null'
206 206 found = 0; obj = None; ospace = None; ds = None;
207 207 ismagic = 0; isalias = 0; parent = None
208 208
209 209 # Look for the given name by splitting it in parts. If the head is
210 210 # found, then we look for all the remaining parts as members, and only
211 211 # declare success if we can find them all.
212 212 oname_parts = oname.split('.')
213 213 oname_head, oname_rest = oname_parts[0],oname_parts[1:]
214 214 for nsname,ns in namespaces:
215 215 try:
216 216 obj = ns[oname_head]
217 217 except KeyError:
218 218 continue
219 219 else:
220 220 for part in oname_rest:
221 221 try:
222 222 parent = obj
223 223 obj = getattr(obj,part)
224 224 except:
225 225 # Blanket except b/c some badly implemented objects
226 226 # allow __getattr__ to raise exceptions other than
227 227 # AttributeError, which then crashes IPython.
228 228 break
229 229 else:
230 230 # If we finish the for loop (no break), we got all members
231 231 found = 1
232 232 ospace = nsname
233 233 if ns == alias_ns:
234 234 isalias = 1
235 235 break # namespace loop
236 236
237 237 # Try to see if it's magic
238 238 if not found:
239 239 if oname.startswith(self.shell.ESC_MAGIC):
240 240 oname = oname[1:]
241 241 obj = getattr(self,'magic_'+oname,None)
242 242 if obj is not None:
243 243 found = 1
244 244 ospace = 'IPython internal'
245 245 ismagic = 1
246 246
247 247 # Last try: special-case some literals like '', [], {}, etc:
248 248 if not found and oname_head in ["''",'""','[]','{}','()']:
249 249 obj = eval(oname_head)
250 250 found = 1
251 251 ospace = 'Interactive'
252 252
253 253 return {'found':found, 'obj':obj, 'namespace':ospace,
254 254 'ismagic':ismagic, 'isalias':isalias, 'parent':parent}
255 255
256 256 def arg_err(self,func):
257 257 """Print docstring if incorrect arguments were passed"""
258 258 print 'Error in arguments:'
259 259 print OInspect.getdoc(func)
260 260
261 261 def format_latex(self,strng):
262 262 """Format a string for latex inclusion."""
263 263
264 264 # Characters that need to be escaped for latex:
265 265 escape_re = re.compile(r'(%|_|\$|#|&)',re.MULTILINE)
266 266 # Magic command names as headers:
267 267 cmd_name_re = re.compile(r'^(%s.*?):' % self.shell.ESC_MAGIC,
268 268 re.MULTILINE)
269 269 # Magic commands
270 270 cmd_re = re.compile(r'(?P<cmd>%s.+?\b)(?!\}\}:)' % self.shell.ESC_MAGIC,
271 271 re.MULTILINE)
272 272 # Paragraph continue
273 273 par_re = re.compile(r'\\$',re.MULTILINE)
274 274
275 275 # The "\n" symbol
276 276 newline_re = re.compile(r'\\n')
277 277
278 278 # Now build the string for output:
279 279 #strng = cmd_name_re.sub(r'\n\\texttt{\\textsl{\\large \1}}:',strng)
280 280 strng = cmd_name_re.sub(r'\n\\bigskip\n\\texttt{\\textbf{ \1}}:',
281 281 strng)
282 282 strng = cmd_re.sub(r'\\texttt{\g<cmd>}',strng)
283 283 strng = par_re.sub(r'\\\\',strng)
284 284 strng = escape_re.sub(r'\\\1',strng)
285 285 strng = newline_re.sub(r'\\textbackslash{}n',strng)
286 286 return strng
287 287
288 288 def format_screen(self,strng):
289 289 """Format a string for screen printing.
290 290
291 291 This removes some latex-type format codes."""
292 292 # Paragraph continue
293 293 par_re = re.compile(r'\\$',re.MULTILINE)
294 294 strng = par_re.sub('',strng)
295 295 return strng
296 296
297 297 def parse_options(self,arg_str,opt_str,*long_opts,**kw):
298 298 """Parse options passed to an argument string.
299 299
300 300 The interface is similar to that of getopt(), but it returns back a
301 301 Struct with the options as keys and the stripped argument string still
302 302 as a string.
303 303
304 304 arg_str is quoted as a true sys.argv vector by using shlex.split.
305 305 This allows us to easily expand variables, glob files, quote
306 306 arguments, etc.
307 307
308 308 Options:
309 309 -mode: default 'string'. If given as 'list', the argument string is
310 310 returned as a list (split on whitespace) instead of a string.
311 311
312 312 -list_all: put all option values in lists. Normally only options
313 313 appearing more than once are put in a list.
314 314
315 315 -posix (True): whether to split the input line in POSIX mode or not,
316 316 as per the conventions outlined in the shlex module from the
317 317 standard library."""
318 318
319 319 # inject default options at the beginning of the input line
320 320 caller = sys._getframe(1).f_code.co_name.replace('magic_','')
321 321 arg_str = '%s %s' % (self.options_table.get(caller,''),arg_str)
322 322
323 323 mode = kw.get('mode','string')
324 324 if mode not in ['string','list']:
325 325 raise ValueError,'incorrect mode given: %s' % mode
326 326 # Get options
327 327 list_all = kw.get('list_all',0)
328 328 posix = kw.get('posix',True)
329 329
330 330 # Check if we have more than one argument to warrant extra processing:
331 331 odict = {} # Dictionary with options
332 332 args = arg_str.split()
333 333 if len(args) >= 1:
334 334 # If the list of inputs only has 0 or 1 thing in it, there's no
335 335 # need to look for options
336 336 argv = arg_split(arg_str,posix)
337 337 # Do regular option processing
338 338 try:
339 339 opts,args = getopt(argv,opt_str,*long_opts)
340 340 except GetoptError,e:
341 341 raise GetoptError('%s ( allowed: "%s" %s)' % (e.msg,opt_str,
342 342 " ".join(long_opts)))
343 343 for o,a in opts:
344 344 if o.startswith('--'):
345 345 o = o[2:]
346 346 else:
347 347 o = o[1:]
348 348 try:
349 349 odict[o].append(a)
350 350 except AttributeError:
351 351 odict[o] = [odict[o],a]
352 352 except KeyError:
353 353 if list_all:
354 354 odict[o] = [a]
355 355 else:
356 356 odict[o] = a
357 357
358 358 # Prepare opts,args for return
359 359 opts = Struct(odict)
360 360 if mode == 'string':
361 361 args = ' '.join(args)
362 362
363 363 return opts,args
364 364
365 365 #......................................................................
366 366 # And now the actual magic functions
367 367
368 368 # Functions for IPython shell work (vars,funcs, config, etc)
369 369 def magic_lsmagic(self, parameter_s = ''):
370 370 """List currently available magic functions."""
371 371 mesc = self.shell.ESC_MAGIC
372 372 print 'Available magic functions:\n'+mesc+\
373 373 (' '+mesc).join(self.lsmagic())
374 374 print '\n' + Magic.auto_status[self.shell.rc.automagic]
375 375 return None
376 376
377 377 def magic_magic(self, parameter_s = ''):
378 378 """Print information about the magic function system."""
379 379
380 380 mode = ''
381 381 try:
382 382 if parameter_s.split()[0] == '-latex':
383 383 mode = 'latex'
384 384 if parameter_s.split()[0] == '-brief':
385 385 mode = 'brief'
386 386 except:
387 387 pass
388 388
389 389 magic_docs = []
390 390 for fname in self.lsmagic():
391 391 mname = 'magic_' + fname
392 392 for space in (Magic,self,self.__class__):
393 393 try:
394 394 fn = space.__dict__[mname]
395 395 except KeyError:
396 396 pass
397 397 else:
398 398 break
399 399 if mode == 'brief':
400 400 # only first line
401 401 fndoc = fn.__doc__.split('\n',1)[0]
402 402 else:
403 403 fndoc = fn.__doc__
404 404
405 405 magic_docs.append('%s%s:\n\t%s\n' %(self.shell.ESC_MAGIC,
406 406 fname,fndoc))
407 407 magic_docs = ''.join(magic_docs)
408 408
409 409 if mode == 'latex':
410 410 print self.format_latex(magic_docs)
411 411 return
412 412 else:
413 413 magic_docs = self.format_screen(magic_docs)
414 414 if mode == 'brief':
415 415 return magic_docs
416 416
417 417 outmsg = """
418 418 IPython's 'magic' functions
419 419 ===========================
420 420
421 421 The magic function system provides a series of functions which allow you to
422 422 control the behavior of IPython itself, plus a lot of system-type
423 423 features. All these functions are prefixed with a % character, but parameters
424 424 are given without parentheses or quotes.
425 425
426 426 NOTE: If you have 'automagic' enabled (via the command line option or with the
427 427 %automagic function), you don't need to type in the % explicitly. By default,
428 428 IPython ships with automagic on, so you should only rarely need the % escape.
429 429
430 430 Example: typing '%cd mydir' (without the quotes) changes you working directory
431 431 to 'mydir', if it exists.
432 432
433 433 You can define your own magic functions to extend the system. See the supplied
434 434 ipythonrc and example-magic.py files for details (in your ipython
435 435 configuration directory, typically $HOME/.ipython/).
436 436
437 437 You can also define your own aliased names for magic functions. In your
438 438 ipythonrc file, placing a line like:
439 439
440 440 execute __IPYTHON__.magic_pf = __IPYTHON__.magic_profile
441 441
442 442 will define %pf as a new name for %profile.
443 443
444 444 You can also call magics in code using the ipmagic() function, which IPython
445 445 automatically adds to the builtin namespace. Type 'ipmagic?' for details.
446 446
447 447 For a list of the available magic functions, use %lsmagic. For a description
448 448 of any of them, type %magic_name?, e.g. '%cd?'.
449 449
450 450 Currently the magic system has the following functions:\n"""
451 451
452 452 mesc = self.shell.ESC_MAGIC
453 453 outmsg = ("%s\n%s\n\nSummary of magic functions (from %slsmagic):"
454 454 "\n\n%s%s\n\n%s" % (outmsg,
455 455 magic_docs,mesc,mesc,
456 456 (' '+mesc).join(self.lsmagic()),
457 457 Magic.auto_status[self.shell.rc.automagic] ) )
458 458
459 459 page(outmsg,screen_lines=self.shell.rc.screen_length)
460 460
461 461 def magic_automagic(self, parameter_s = ''):
462 462 """Make magic functions callable without having to type the initial %.
463 463
464 464 Toggles on/off (when off, you must call it as %automagic, of
465 465 course). Note that magic functions have lowest priority, so if there's
466 466 a variable whose name collides with that of a magic fn, automagic
467 467 won't work for that function (you get the variable instead). However,
468 468 if you delete the variable (del var), the previously shadowed magic
469 469 function becomes visible to automagic again."""
470 470
471 471 rc = self.shell.rc
472 472 rc.automagic = not rc.automagic
473 473 print '\n' + Magic.auto_status[rc.automagic]
474 474
475 475 def magic_autocall(self, parameter_s = ''):
476 476 """Make functions callable without having to type parentheses.
477 477
478 478 Usage:
479 479
480 480 %autocall [mode]
481 481
482 482 The mode can be one of: 0->Off, 1->Smart, 2->Full. If not given, the
483 483 value is toggled on and off (remembering the previous state)."""
484 484
485 485 rc = self.shell.rc
486 486
487 487 if parameter_s:
488 488 arg = int(parameter_s)
489 489 else:
490 490 arg = 'toggle'
491 491
492 492 if not arg in (0,1,2,'toggle'):
493 493 error('Valid modes: (0->Off, 1->Smart, 2->Full')
494 494 return
495 495
496 496 if arg in (0,1,2):
497 497 rc.autocall = arg
498 498 else: # toggle
499 499 if rc.autocall:
500 500 self._magic_state.autocall_save = rc.autocall
501 501 rc.autocall = 0
502 502 else:
503 503 try:
504 504 rc.autocall = self._magic_state.autocall_save
505 505 except AttributeError:
506 506 rc.autocall = self._magic_state.autocall_save = 1
507 507
508 508 print "Automatic calling is:",['OFF','Smart','Full'][rc.autocall]
509 509
510 510 def magic_autoindent(self, parameter_s = ''):
511 511 """Toggle autoindent on/off (if available)."""
512 512
513 513 self.shell.set_autoindent()
514 514 print "Automatic indentation is:",['OFF','ON'][self.shell.autoindent]
515 515
516 516 def magic_system_verbose(self, parameter_s = ''):
517 517 """Toggle verbose printing of system calls on/off."""
518 518
519 519 self.shell.rc_set_toggle('system_verbose')
520 520 print "System verbose printing is:",\
521 521 ['OFF','ON'][self.shell.rc.system_verbose]
522 522
523 523 def magic_history(self, parameter_s = ''):
524 524 """Print input history (_i<n> variables), with most recent last.
525 525
526 526 %history -> print at most 40 inputs (some may be multi-line)\\
527 527 %history n -> print at most n inputs\\
528 528 %history n1 n2 -> print inputs between n1 and n2 (n2 not included)\\
529 529
530 530 Each input's number <n> is shown, and is accessible as the
531 531 automatically generated variable _i<n>. Multi-line statements are
532 532 printed starting at a new line for easy copy/paste.
533 533
534 534
535 535 Options:
536 536
537 537 -n: do NOT print line numbers. This is useful if you want to get a
538 538 printout of many lines which can be directly pasted into a text
539 539 editor.
540 540
541 541 This feature is only available if numbered prompts are in use.
542 542
543 543 -r: print the 'raw' history. IPython filters your input and
544 544 converts it all into valid Python source before executing it (things
545 545 like magics or aliases are turned into function calls, for
546 546 example). With this option, you'll see the unfiltered history
547 547 instead of the filtered version: '%cd /' will be seen as '%cd /'
548 548 instead of '_ip.magic("%cd /")'.
549 549 """
550 550
551 551 shell = self.shell
552 552 if not shell.outputcache.do_full_cache:
553 553 print 'This feature is only available if numbered prompts are in use.'
554 554 return
555 555 opts,args = self.parse_options(parameter_s,'nr',mode='list')
556 556
557 557 if opts.has_key('r'):
558 558 input_hist = shell.input_hist_raw
559 559 else:
560 560 input_hist = shell.input_hist
561 561
562 562 default_length = 40
563 563 if len(args) == 0:
564 564 final = len(input_hist)
565 565 init = max(1,final-default_length)
566 566 elif len(args) == 1:
567 567 final = len(input_hist)
568 568 init = max(1,final-int(args[0]))
569 569 elif len(args) == 2:
570 570 init,final = map(int,args)
571 571 else:
572 572 warn('%hist takes 0, 1 or 2 arguments separated by spaces.')
573 573 print self.magic_hist.__doc__
574 574 return
575 575 width = len(str(final))
576 576 line_sep = ['','\n']
577 577 print_nums = not opts.has_key('n')
578 578 for in_num in range(init,final):
579 579 inline = input_hist[in_num]
580 580 multiline = int(inline.count('\n') > 1)
581 581 if print_nums:
582 582 print '%s:%s' % (str(in_num).ljust(width),line_sep[multiline]),
583 583 print inline,
584 584
585 585 def magic_hist(self, parameter_s=''):
586 586 """Alternate name for %history."""
587 587 return self.magic_history(parameter_s)
588 588
589 589 def magic_p(self, parameter_s=''):
590 590 """Just a short alias for Python's 'print'."""
591 591 exec 'print ' + parameter_s in self.shell.user_ns
592 592
593 593 def magic_r(self, parameter_s=''):
594 594 """Repeat previous input.
595 595
596 596 If given an argument, repeats the previous command which starts with
597 597 the same string, otherwise it just repeats the previous input.
598 598
599 599 Shell escaped commands (with ! as first character) are not recognized
600 600 by this system, only pure python code and magic commands.
601 601 """
602 602
603 603 start = parameter_s.strip()
604 604 esc_magic = self.shell.ESC_MAGIC
605 605 # Identify magic commands even if automagic is on (which means
606 606 # the in-memory version is different from that typed by the user).
607 607 if self.shell.rc.automagic:
608 608 start_magic = esc_magic+start
609 609 else:
610 610 start_magic = start
611 611 # Look through the input history in reverse
612 612 for n in range(len(self.shell.input_hist)-2,0,-1):
613 613 input = self.shell.input_hist[n]
614 614 # skip plain 'r' lines so we don't recurse to infinity
615 615 if input != '_ip.magic("r")\n' and \
616 616 (input.startswith(start) or input.startswith(start_magic)):
617 617 #print 'match',`input` # dbg
618 618 print 'Executing:',input,
619 619 self.shell.runlines(input)
620 620 return
621 621 print 'No previous input matching `%s` found.' % start
622 622
623 623 def magic_page(self, parameter_s=''):
624 624 """Pretty print the object and display it through a pager.
625 625
626 626 If no parameter is given, use _ (last output)."""
627 627 # After a function contributed by Olivier Aubert, slightly modified.
628 628
629 629 oname = parameter_s and parameter_s or '_'
630 630 info = self._ofind(oname)
631 631 if info['found']:
632 632 page(pformat(info['obj']))
633 633 else:
634 634 print 'Object `%s` not found' % oname
635 635
636 636 def magic_profile(self, parameter_s=''):
637 637 """Print your currently active IPyhton profile."""
638 638 if self.shell.rc.profile:
639 639 printpl('Current IPython profile: $self.shell.rc.profile.')
640 640 else:
641 641 print 'No profile active.'
642 642
643 643 def _inspect(self,meth,oname,namespaces=None,**kw):
644 644 """Generic interface to the inspector system.
645 645
646 646 This function is meant to be called by pdef, pdoc & friends."""
647 647
648 648 oname = oname.strip()
649 649 info = Struct(self._ofind(oname, namespaces))
650 650
651 651 if info.found:
652 652 # Get the docstring of the class property if it exists.
653 653 path = oname.split('.')
654 654 root = '.'.join(path[:-1])
655 655 if info.parent is not None:
656 656 try:
657 657 target = getattr(info.parent, '__class__')
658 658 # The object belongs to a class instance.
659 659 try:
660 660 target = getattr(target, path[-1])
661 661 # The class defines the object.
662 662 if isinstance(target, property):
663 663 oname = root + '.__class__.' + path[-1]
664 664 info = Struct(self._ofind(oname))
665 665 except AttributeError: pass
666 666 except AttributeError: pass
667 667
668 668 pmethod = getattr(self.shell.inspector,meth)
669 669 formatter = info.ismagic and self.format_screen or None
670 670 if meth == 'pdoc':
671 671 pmethod(info.obj,oname,formatter)
672 672 elif meth == 'pinfo':
673 673 pmethod(info.obj,oname,formatter,info,**kw)
674 674 else:
675 675 pmethod(info.obj,oname)
676 676 else:
677 677 print 'Object `%s` not found.' % oname
678 678 return 'not found' # so callers can take other action
679 679
680 680 def magic_pdef(self, parameter_s='', namespaces=None):
681 681 """Print the definition header for any callable object.
682 682
683 683 If the object is a class, print the constructor information."""
684 684 print "+++"
685 685 self._inspect('pdef',parameter_s, namespaces)
686 686
687 687 def magic_pdoc(self, parameter_s='', namespaces=None):
688 688 """Print the docstring for an object.
689 689
690 690 If the given object is a class, it will print both the class and the
691 691 constructor docstrings."""
692 692 self._inspect('pdoc',parameter_s, namespaces)
693 693
694 694 def magic_psource(self, parameter_s='', namespaces=None):
695 695 """Print (or run through pager) the source code for an object."""
696 696 self._inspect('psource',parameter_s, namespaces)
697 697
698 698 def magic_pfile(self, parameter_s=''):
699 699 """Print (or run through pager) the file where an object is defined.
700 700
701 701 The file opens at the line where the object definition begins. IPython
702 702 will honor the environment variable PAGER if set, and otherwise will
703 703 do its best to print the file in a convenient form.
704 704
705 705 If the given argument is not an object currently defined, IPython will
706 706 try to interpret it as a filename (automatically adding a .py extension
707 707 if needed). You can thus use %pfile as a syntax highlighting code
708 708 viewer."""
709 709
710 710 # first interpret argument as an object name
711 711 out = self._inspect('pfile',parameter_s)
712 712 # if not, try the input as a filename
713 713 if out == 'not found':
714 714 try:
715 715 filename = get_py_filename(parameter_s)
716 716 except IOError,msg:
717 717 print msg
718 718 return
719 719 page(self.shell.inspector.format(file(filename).read()))
720 720
721 721 def magic_pinfo(self, parameter_s='', namespaces=None):
722 722 """Provide detailed information about an object.
723 723
724 724 '%pinfo object' is just a synonym for object? or ?object."""
725 725
726 726 #print 'pinfo par: <%s>' % parameter_s # dbg
727 727
728 728 # detail_level: 0 -> obj? , 1 -> obj??
729 729 detail_level = 0
730 730 # We need to detect if we got called as 'pinfo pinfo foo', which can
731 731 # happen if the user types 'pinfo foo?' at the cmd line.
732 732 pinfo,qmark1,oname,qmark2 = \
733 733 re.match('(pinfo )?(\?*)(.*?)(\??$)',parameter_s).groups()
734 734 if pinfo or qmark1 or qmark2:
735 735 detail_level = 1
736 736 if "*" in oname:
737 737 self.magic_psearch(oname)
738 738 else:
739 739 self._inspect('pinfo', oname, detail_level=detail_level,
740 740 namespaces=namespaces)
741 741
742 742 def magic_psearch(self, parameter_s=''):
743 743 """Search for object in namespaces by wildcard.
744 744
745 745 %psearch [options] PATTERN [OBJECT TYPE]
746 746
747 747 Note: ? can be used as a synonym for %psearch, at the beginning or at
748 748 the end: both a*? and ?a* are equivalent to '%psearch a*'. Still, the
749 749 rest of the command line must be unchanged (options come first), so
750 750 for example the following forms are equivalent
751 751
752 752 %psearch -i a* function
753 753 -i a* function?
754 754 ?-i a* function
755 755
756 756 Arguments:
757 757
758 758 PATTERN
759 759
760 760 where PATTERN is a string containing * as a wildcard similar to its
761 761 use in a shell. The pattern is matched in all namespaces on the
762 762 search path. By default objects starting with a single _ are not
763 763 matched, many IPython generated objects have a single
764 764 underscore. The default is case insensitive matching. Matching is
765 765 also done on the attributes of objects and not only on the objects
766 766 in a module.
767 767
768 768 [OBJECT TYPE]
769 769
770 770 Is the name of a python type from the types module. The name is
771 771 given in lowercase without the ending type, ex. StringType is
772 772 written string. By adding a type here only objects matching the
773 773 given type are matched. Using all here makes the pattern match all
774 774 types (this is the default).
775 775
776 776 Options:
777 777
778 778 -a: makes the pattern match even objects whose names start with a
779 779 single underscore. These names are normally ommitted from the
780 780 search.
781 781
782 782 -i/-c: make the pattern case insensitive/sensitive. If neither of
783 783 these options is given, the default is read from your ipythonrc
784 784 file. The option name which sets this value is
785 785 'wildcards_case_sensitive'. If this option is not specified in your
786 786 ipythonrc file, IPython's internal default is to do a case sensitive
787 787 search.
788 788
789 789 -e/-s NAMESPACE: exclude/search a given namespace. The pattern you
790 790 specifiy can be searched in any of the following namespaces:
791 791 'builtin', 'user', 'user_global','internal', 'alias', where
792 792 'builtin' and 'user' are the search defaults. Note that you should
793 793 not use quotes when specifying namespaces.
794 794
795 795 'Builtin' contains the python module builtin, 'user' contains all
796 796 user data, 'alias' only contain the shell aliases and no python
797 797 objects, 'internal' contains objects used by IPython. The
798 798 'user_global' namespace is only used by embedded IPython instances,
799 799 and it contains module-level globals. You can add namespaces to the
800 800 search with -s or exclude them with -e (these options can be given
801 801 more than once).
802 802
803 803 Examples:
804 804
805 805 %psearch a* -> objects beginning with an a
806 806 %psearch -e builtin a* -> objects NOT in the builtin space starting in a
807 807 %psearch a* function -> all functions beginning with an a
808 808 %psearch re.e* -> objects beginning with an e in module re
809 809 %psearch r*.e* -> objects that start with e in modules starting in r
810 810 %psearch r*.* string -> all strings in modules beginning with r
811 811
812 812 Case sensitve search:
813 813
814 814 %psearch -c a* list all object beginning with lower case a
815 815
816 816 Show objects beginning with a single _:
817 817
818 818 %psearch -a _* list objects beginning with a single underscore"""
819 819
820 820 # default namespaces to be searched
821 821 def_search = ['user','builtin']
822 822
823 823 # Process options/args
824 824 opts,args = self.parse_options(parameter_s,'cias:e:',list_all=True)
825 825 opt = opts.get
826 826 shell = self.shell
827 827 psearch = shell.inspector.psearch
828 828
829 829 # select case options
830 830 if opts.has_key('i'):
831 831 ignore_case = True
832 832 elif opts.has_key('c'):
833 833 ignore_case = False
834 834 else:
835 835 ignore_case = not shell.rc.wildcards_case_sensitive
836 836
837 837 # Build list of namespaces to search from user options
838 838 def_search.extend(opt('s',[]))
839 839 ns_exclude = ns_exclude=opt('e',[])
840 840 ns_search = [nm for nm in def_search if nm not in ns_exclude]
841 841
842 842 # Call the actual search
843 843 try:
844 844 psearch(args,shell.ns_table,ns_search,
845 845 show_all=opt('a'),ignore_case=ignore_case)
846 846 except:
847 847 shell.showtraceback()
848 848
849 849 def magic_who_ls(self, parameter_s=''):
850 850 """Return a sorted list of all interactive variables.
851 851
852 852 If arguments are given, only variables of types matching these
853 853 arguments are returned."""
854 854
855 855 user_ns = self.shell.user_ns
856 856 internal_ns = self.shell.internal_ns
857 857 user_config_ns = self.shell.user_config_ns
858 858 out = []
859 859 typelist = parameter_s.split()
860 860
861 861 for i in user_ns:
862 862 if not (i.startswith('_') or i.startswith('_i')) \
863 863 and not (i in internal_ns or i in user_config_ns):
864 864 if typelist:
865 865 if type(user_ns[i]).__name__ in typelist:
866 866 out.append(i)
867 867 else:
868 868 out.append(i)
869 869 out.sort()
870 870 return out
871 871
872 872 def magic_who(self, parameter_s=''):
873 873 """Print all interactive variables, with some minimal formatting.
874 874
875 875 If any arguments are given, only variables whose type matches one of
876 876 these are printed. For example:
877 877
878 878 %who function str
879 879
880 880 will only list functions and strings, excluding all other types of
881 881 variables. To find the proper type names, simply use type(var) at a
882 882 command line to see how python prints type names. For example:
883 883
884 884 In [1]: type('hello')\\
885 885 Out[1]: <type 'str'>
886 886
887 887 indicates that the type name for strings is 'str'.
888 888
889 889 %who always excludes executed names loaded through your configuration
890 890 file and things which are internal to IPython.
891 891
892 892 This is deliberate, as typically you may load many modules and the
893 893 purpose of %who is to show you only what you've manually defined."""
894 894
895 895 varlist = self.magic_who_ls(parameter_s)
896 896 if not varlist:
897 897 print 'Interactive namespace is empty.'
898 898 return
899 899
900 900 # if we have variables, move on...
901 901
902 902 # stupid flushing problem: when prompts have no separators, stdout is
903 903 # getting lost. I'm starting to think this is a python bug. I'm having
904 904 # to force a flush with a print because even a sys.stdout.flush
905 905 # doesn't seem to do anything!
906 906
907 907 count = 0
908 908 for i in varlist:
909 909 print i+'\t',
910 910 count += 1
911 911 if count > 8:
912 912 count = 0
913 913 print
914 914 sys.stdout.flush() # FIXME. Why the hell isn't this flushing???
915 915
916 916 print # well, this does force a flush at the expense of an extra \n
917 917
918 918 def magic_whos(self, parameter_s=''):
919 919 """Like %who, but gives some extra information about each variable.
920 920
921 921 The same type filtering of %who can be applied here.
922 922
923 923 For all variables, the type is printed. Additionally it prints:
924 924
925 925 - For {},[],(): their length.
926 926
927 927 - For Numeric arrays, a summary with shape, number of elements,
928 928 typecode and size in memory.
929 929
930 930 - Everything else: a string representation, snipping their middle if
931 931 too long."""
932 932
933 933 varnames = self.magic_who_ls(parameter_s)
934 934 if not varnames:
935 935 print 'Interactive namespace is empty.'
936 936 return
937 937
938 938 # if we have variables, move on...
939 939
940 940 # for these types, show len() instead of data:
941 941 seq_types = [types.DictType,types.ListType,types.TupleType]
942 942
943 943 # for Numeric arrays, display summary info
944 944 try:
945 945 import Numeric
946 946 except ImportError:
947 947 array_type = None
948 948 else:
949 949 array_type = Numeric.ArrayType.__name__
950 950
951 951 # Find all variable names and types so we can figure out column sizes
952 952 get_vars = lambda i: self.shell.user_ns[i]
953 953 type_name = lambda v: type(v).__name__
954 954 varlist = map(get_vars,varnames)
955 955
956 956 typelist = []
957 957 for vv in varlist:
958 958 tt = type_name(vv)
959 959 if tt=='instance':
960 960 typelist.append(str(vv.__class__))
961 961 else:
962 962 typelist.append(tt)
963 963
964 964 # column labels and # of spaces as separator
965 965 varlabel = 'Variable'
966 966 typelabel = 'Type'
967 967 datalabel = 'Data/Info'
968 968 colsep = 3
969 969 # variable format strings
970 970 vformat = "$vname.ljust(varwidth)$vtype.ljust(typewidth)"
971 971 vfmt_short = '$vstr[:25]<...>$vstr[-25:]'
972 972 aformat = "%s: %s elems, type `%s`, %s bytes"
973 973 # find the size of the columns to format the output nicely
974 974 varwidth = max(max(map(len,varnames)), len(varlabel)) + colsep
975 975 typewidth = max(max(map(len,typelist)), len(typelabel)) + colsep
976 976 # table header
977 977 print varlabel.ljust(varwidth) + typelabel.ljust(typewidth) + \
978 978 ' '+datalabel+'\n' + '-'*(varwidth+typewidth+len(datalabel)+1)
979 979 # and the table itself
980 980 kb = 1024
981 981 Mb = 1048576 # kb**2
982 982 for vname,var,vtype in zip(varnames,varlist,typelist):
983 983 print itpl(vformat),
984 984 if vtype in seq_types:
985 985 print len(var)
986 986 elif vtype==array_type:
987 987 vshape = str(var.shape).replace(',','').replace(' ','x')[1:-1]
988 988 vsize = Numeric.size(var)
989 989 vbytes = vsize*var.itemsize()
990 990 if vbytes < 100000:
991 991 print aformat % (vshape,vsize,var.typecode(),vbytes)
992 992 else:
993 993 print aformat % (vshape,vsize,var.typecode(),vbytes),
994 994 if vbytes < Mb:
995 995 print '(%s kb)' % (vbytes/kb,)
996 996 else:
997 997 print '(%s Mb)' % (vbytes/Mb,)
998 998 else:
999 999 vstr = str(var).replace('\n','\\n')
1000 1000 if len(vstr) < 50:
1001 1001 print vstr
1002 1002 else:
1003 1003 printpl(vfmt_short)
1004 1004
1005 1005 def magic_reset(self, parameter_s=''):
1006 1006 """Resets the namespace by removing all names defined by the user.
1007 1007
1008 1008 Input/Output history are left around in case you need them."""
1009 1009
1010 1010 ans = self.shell.ask_yes_no(
1011 1011 "Once deleted, variables cannot be recovered. Proceed (y/[n])? ")
1012 1012 if not ans:
1013 1013 print 'Nothing done.'
1014 1014 return
1015 1015 user_ns = self.shell.user_ns
1016 1016 for i in self.magic_who_ls():
1017 1017 del(user_ns[i])
1018 1018
1019 1019 def magic_config(self,parameter_s=''):
1020 1020 """Show IPython's internal configuration."""
1021 1021
1022 1022 page('Current configuration structure:\n'+
1023 1023 pformat(self.shell.rc.dict()))
1024 1024
1025 1025 def magic_logstart(self,parameter_s=''):
1026 1026 """Start logging anywhere in a session.
1027 1027
1028 1028 %logstart [-o|-r|-t] [log_name [log_mode]]
1029 1029
1030 1030 If no name is given, it defaults to a file named 'ipython_log.py' in your
1031 1031 current directory, in 'rotate' mode (see below).
1032 1032
1033 1033 '%logstart name' saves to file 'name' in 'backup' mode. It saves your
1034 1034 history up to that point and then continues logging.
1035 1035
1036 1036 %logstart takes a second optional parameter: logging mode. This can be one
1037 1037 of (note that the modes are given unquoted):\\
1038 1038 append: well, that says it.\\
1039 1039 backup: rename (if exists) to name~ and start name.\\
1040 1040 global: single logfile in your home dir, appended to.\\
1041 1041 over : overwrite existing log.\\
1042 1042 rotate: create rotating logs name.1~, name.2~, etc.
1043 1043
1044 1044 Options:
1045 1045
1046 1046 -o: log also IPython's output. In this mode, all commands which
1047 1047 generate an Out[NN] prompt are recorded to the logfile, right after
1048 1048 their corresponding input line. The output lines are always
1049 1049 prepended with a '#[Out]# ' marker, so that the log remains valid
1050 1050 Python code.
1051 1051
1052 1052 Since this marker is always the same, filtering only the output from
1053 1053 a log is very easy, using for example a simple awk call:
1054 1054
1055 1055 awk -F'#\\[Out\\]# ' '{if($2) {print $2}}' ipython_log.py
1056 1056
1057 1057 -r: log 'raw' input. Normally, IPython's logs contain the processed
1058 1058 input, so that user lines are logged in their final form, converted
1059 1059 into valid Python. For example, %Exit is logged as
1060 1060 '_ip.magic("Exit"). If the -r flag is given, all input is logged
1061 1061 exactly as typed, with no transformations applied.
1062 1062
1063 1063 -t: put timestamps before each input line logged (these are put in
1064 1064 comments)."""
1065 1065
1066 1066 opts,par = self.parse_options(parameter_s,'ort')
1067 1067 log_output = 'o' in opts
1068 1068 log_raw_input = 'r' in opts
1069 1069 timestamp = 't' in opts
1070 1070
1071 1071 rc = self.shell.rc
1072 1072 logger = self.shell.logger
1073 1073
1074 1074 # if no args are given, the defaults set in the logger constructor by
1075 1075 # ipytohn remain valid
1076 1076 if par:
1077 1077 try:
1078 1078 logfname,logmode = par.split()
1079 1079 except:
1080 1080 logfname = par
1081 1081 logmode = 'backup'
1082 1082 else:
1083 1083 logfname = logger.logfname
1084 1084 logmode = logger.logmode
1085 1085 # put logfname into rc struct as if it had been called on the command
1086 1086 # line, so it ends up saved in the log header Save it in case we need
1087 1087 # to restore it...
1088 1088 old_logfile = rc.opts.get('logfile','')
1089 1089 if logfname:
1090 1090 logfname = os.path.expanduser(logfname)
1091 1091 rc.opts.logfile = logfname
1092 1092 loghead = self.shell.loghead_tpl % (rc.opts,rc.args)
1093 1093 try:
1094 1094 started = logger.logstart(logfname,loghead,logmode,
1095 1095 log_output,timestamp,log_raw_input)
1096 1096 except:
1097 1097 rc.opts.logfile = old_logfile
1098 1098 warn("Couldn't start log: %s" % sys.exc_info()[1])
1099 1099 else:
1100 1100 # log input history up to this point, optionally interleaving
1101 1101 # output if requested
1102 1102
1103 1103 if timestamp:
1104 1104 # disable timestamping for the previous history, since we've
1105 1105 # lost those already (no time machine here).
1106 1106 logger.timestamp = False
1107 1107
1108 1108 if log_raw_input:
1109 1109 input_hist = self.shell.input_hist_raw
1110 1110 else:
1111 1111 input_hist = self.shell.input_hist
1112 1112
1113 1113 if log_output:
1114 1114 log_write = logger.log_write
1115 1115 output_hist = self.shell.output_hist
1116 1116 for n in range(1,len(input_hist)-1):
1117 1117 log_write(input_hist[n].rstrip())
1118 1118 if n in output_hist:
1119 1119 log_write(repr(output_hist[n]),'output')
1120 1120 else:
1121 1121 logger.log_write(input_hist[1:])
1122 1122 if timestamp:
1123 1123 # re-enable timestamping
1124 1124 logger.timestamp = True
1125 1125
1126 1126 print ('Activating auto-logging. '
1127 1127 'Current session state plus future input saved.')
1128 1128 logger.logstate()
1129 1129
1130 1130 def magic_logoff(self,parameter_s=''):
1131 1131 """Temporarily stop logging.
1132 1132
1133 1133 You must have previously started logging."""
1134 1134 self.shell.logger.switch_log(0)
1135 1135
1136 1136 def magic_logon(self,parameter_s=''):
1137 1137 """Restart logging.
1138 1138
1139 1139 This function is for restarting logging which you've temporarily
1140 1140 stopped with %logoff. For starting logging for the first time, you
1141 1141 must use the %logstart function, which allows you to specify an
1142 1142 optional log filename."""
1143 1143
1144 1144 self.shell.logger.switch_log(1)
1145 1145
1146 1146 def magic_logstate(self,parameter_s=''):
1147 1147 """Print the status of the logging system."""
1148 1148
1149 1149 self.shell.logger.logstate()
1150 1150
1151 1151 def magic_pdb(self, parameter_s=''):
1152 1152 """Control the calling of the pdb interactive debugger.
1153 1153
1154 1154 Call as '%pdb on', '%pdb 1', '%pdb off' or '%pdb 0'. If called without
1155 1155 argument it works as a toggle.
1156 1156
1157 1157 When an exception is triggered, IPython can optionally call the
1158 1158 interactive pdb debugger after the traceback printout. %pdb toggles
1159 1159 this feature on and off."""
1160 1160
1161 1161 par = parameter_s.strip().lower()
1162 1162
1163 1163 if par:
1164 1164 try:
1165 1165 new_pdb = {'off':0,'0':0,'on':1,'1':1}[par]
1166 1166 except KeyError:
1167 1167 print ('Incorrect argument. Use on/1, off/0, '
1168 1168 'or nothing for a toggle.')
1169 1169 return
1170 1170 else:
1171 1171 # toggle
1172 1172 new_pdb = not self.shell.InteractiveTB.call_pdb
1173 1173
1174 1174 # set on the shell
1175 1175 self.shell.call_pdb = new_pdb
1176 1176 print 'Automatic pdb calling has been turned',on_off(new_pdb)
1177 1177
1178 1178 def magic_prun(self, parameter_s ='',user_mode=1,
1179 1179 opts=None,arg_lst=None,prog_ns=None):
1180 1180
1181 1181 """Run a statement through the python code profiler.
1182 1182
1183 1183 Usage:\\
1184 1184 %prun [options] statement
1185 1185
1186 1186 The given statement (which doesn't require quote marks) is run via the
1187 1187 python profiler in a manner similar to the profile.run() function.
1188 1188 Namespaces are internally managed to work correctly; profile.run
1189 1189 cannot be used in IPython because it makes certain assumptions about
1190 1190 namespaces which do not hold under IPython.
1191 1191
1192 1192 Options:
1193 1193
1194 1194 -l <limit>: you can place restrictions on what or how much of the
1195 1195 profile gets printed. The limit value can be:
1196 1196
1197 1197 * A string: only information for function names containing this string
1198 1198 is printed.
1199 1199
1200 1200 * An integer: only these many lines are printed.
1201 1201
1202 1202 * A float (between 0 and 1): this fraction of the report is printed
1203 1203 (for example, use a limit of 0.4 to see the topmost 40% only).
1204 1204
1205 1205 You can combine several limits with repeated use of the option. For
1206 1206 example, '-l __init__ -l 5' will print only the topmost 5 lines of
1207 1207 information about class constructors.
1208 1208
1209 1209 -r: return the pstats.Stats object generated by the profiling. This
1210 1210 object has all the information about the profile in it, and you can
1211 1211 later use it for further analysis or in other functions.
1212 1212
1213 1213 Since magic functions have a particular form of calling which prevents
1214 1214 you from writing something like:\\
1215 1215 In [1]: p = %prun -r print 4 # invalid!\\
1216 1216 you must instead use IPython's automatic variables to assign this:\\
1217 1217 In [1]: %prun -r print 4 \\
1218 1218 Out[1]: <pstats.Stats instance at 0x8222cec>\\
1219 1219 In [2]: stats = _
1220 1220
1221 1221 If you really need to assign this value via an explicit function call,
1222 1222 you can always tap directly into the true name of the magic function
1223 1223 by using the _ip.magic function:\\
1224 1224 In [3]: stats = _ip.magic('prun','-r print 4')
1225 1225
1226 1226 You can type _ip.magic? for more details.
1227 1227
1228 1228 -s <key>: sort profile by given key. You can provide more than one key
1229 1229 by using the option several times: '-s key1 -s key2 -s key3...'. The
1230 1230 default sorting key is 'time'.
1231 1231
1232 1232 The following is copied verbatim from the profile documentation
1233 1233 referenced below:
1234 1234
1235 1235 When more than one key is provided, additional keys are used as
1236 1236 secondary criteria when the there is equality in all keys selected
1237 1237 before them.
1238 1238
1239 1239 Abbreviations can be used for any key names, as long as the
1240 1240 abbreviation is unambiguous. The following are the keys currently
1241 1241 defined:
1242 1242
1243 1243 Valid Arg Meaning\\
1244 1244 "calls" call count\\
1245 1245 "cumulative" cumulative time\\
1246 1246 "file" file name\\
1247 1247 "module" file name\\
1248 1248 "pcalls" primitive call count\\
1249 1249 "line" line number\\
1250 1250 "name" function name\\
1251 1251 "nfl" name/file/line\\
1252 1252 "stdname" standard name\\
1253 1253 "time" internal time
1254 1254
1255 1255 Note that all sorts on statistics are in descending order (placing
1256 1256 most time consuming items first), where as name, file, and line number
1257 1257 searches are in ascending order (i.e., alphabetical). The subtle
1258 1258 distinction between "nfl" and "stdname" is that the standard name is a
1259 1259 sort of the name as printed, which means that the embedded line
1260 1260 numbers get compared in an odd way. For example, lines 3, 20, and 40
1261 1261 would (if the file names were the same) appear in the string order
1262 1262 "20" "3" and "40". In contrast, "nfl" does a numeric compare of the
1263 1263 line numbers. In fact, sort_stats("nfl") is the same as
1264 1264 sort_stats("name", "file", "line").
1265 1265
1266 1266 -T <filename>: save profile results as shown on screen to a text
1267 1267 file. The profile is still shown on screen.
1268 1268
1269 1269 -D <filename>: save (via dump_stats) profile statistics to given
1270 1270 filename. This data is in a format understod by the pstats module, and
1271 1271 is generated by a call to the dump_stats() method of profile
1272 1272 objects. The profile is still shown on screen.
1273 1273
1274 1274 If you want to run complete programs under the profiler's control, use
1275 1275 '%run -p [prof_opts] filename.py [args to program]' where prof_opts
1276 1276 contains profiler specific options as described here.
1277 1277
1278 1278 You can read the complete documentation for the profile module with:\\
1279 1279 In [1]: import profile; profile.help() """
1280 1280
1281 1281 opts_def = Struct(D=[''],l=[],s=['time'],T=[''])
1282 1282 # protect user quote marks
1283 1283 parameter_s = parameter_s.replace('"',r'\"').replace("'",r"\'")
1284 1284
1285 1285 if user_mode: # regular user call
1286 1286 opts,arg_str = self.parse_options(parameter_s,'D:l:rs:T:',
1287 1287 list_all=1)
1288 1288 namespace = self.shell.user_ns
1289 1289 else: # called to run a program by %run -p
1290 1290 try:
1291 1291 filename = get_py_filename(arg_lst[0])
1292 1292 except IOError,msg:
1293 1293 error(msg)
1294 1294 return
1295 1295
1296 1296 arg_str = 'execfile(filename,prog_ns)'
1297 1297 namespace = locals()
1298 1298
1299 1299 opts.merge(opts_def)
1300 1300
1301 1301 prof = profile.Profile()
1302 1302 try:
1303 1303 prof = prof.runctx(arg_str,namespace,namespace)
1304 1304 sys_exit = ''
1305 1305 except SystemExit:
1306 1306 sys_exit = """*** SystemExit exception caught in code being profiled."""
1307 1307
1308 1308 stats = pstats.Stats(prof).strip_dirs().sort_stats(*opts.s)
1309 1309
1310 1310 lims = opts.l
1311 1311 if lims:
1312 1312 lims = [] # rebuild lims with ints/floats/strings
1313 1313 for lim in opts.l:
1314 1314 try:
1315 1315 lims.append(int(lim))
1316 1316 except ValueError:
1317 1317 try:
1318 1318 lims.append(float(lim))
1319 1319 except ValueError:
1320 1320 lims.append(lim)
1321 1321
1322 1322 # trap output
1323 1323 sys_stdout = sys.stdout
1324 1324 stdout_trap = StringIO()
1325 1325 try:
1326 1326 sys.stdout = stdout_trap
1327 1327 stats.print_stats(*lims)
1328 1328 finally:
1329 1329 sys.stdout = sys_stdout
1330 1330 output = stdout_trap.getvalue()
1331 1331 output = output.rstrip()
1332 1332
1333 1333 page(output,screen_lines=self.shell.rc.screen_length)
1334 1334 print sys_exit,
1335 1335
1336 1336 dump_file = opts.D[0]
1337 1337 text_file = opts.T[0]
1338 1338 if dump_file:
1339 1339 prof.dump_stats(dump_file)
1340 1340 print '\n*** Profile stats marshalled to file',\
1341 1341 `dump_file`+'.',sys_exit
1342 1342 if text_file:
1343 1343 file(text_file,'w').write(output)
1344 1344 print '\n*** Profile printout saved to text file',\
1345 1345 `text_file`+'.',sys_exit
1346 1346
1347 1347 if opts.has_key('r'):
1348 1348 return stats
1349 1349 else:
1350 1350 return None
1351 1351
1352 1352 def magic_run(self, parameter_s ='',runner=None):
1353 1353 """Run the named file inside IPython as a program.
1354 1354
1355 1355 Usage:\\
1356 1356 %run [-n -i -t [-N<N>] -d [-b<N>] -p [profile options]] file [args]
1357 1357
1358 1358 Parameters after the filename are passed as command-line arguments to
1359 1359 the program (put in sys.argv). Then, control returns to IPython's
1360 1360 prompt.
1361 1361
1362 1362 This is similar to running at a system prompt:\\
1363 1363 $ python file args\\
1364 1364 but with the advantage of giving you IPython's tracebacks, and of
1365 1365 loading all variables into your interactive namespace for further use
1366 1366 (unless -p is used, see below).
1367 1367
1368 1368 The file is executed in a namespace initially consisting only of
1369 1369 __name__=='__main__' and sys.argv constructed as indicated. It thus
1370 1370 sees its environment as if it were being run as a stand-alone
1371 1371 program. But after execution, the IPython interactive namespace gets
1372 1372 updated with all variables defined in the program (except for __name__
1373 1373 and sys.argv). This allows for very convenient loading of code for
1374 1374 interactive work, while giving each program a 'clean sheet' to run in.
1375 1375
1376 1376 Options:
1377 1377
1378 1378 -n: __name__ is NOT set to '__main__', but to the running file's name
1379 1379 without extension (as python does under import). This allows running
1380 1380 scripts and reloading the definitions in them without calling code
1381 1381 protected by an ' if __name__ == "__main__" ' clause.
1382 1382
1383 1383 -i: run the file in IPython's namespace instead of an empty one. This
1384 1384 is useful if you are experimenting with code written in a text editor
1385 1385 which depends on variables defined interactively.
1386 1386
1387 1387 -e: ignore sys.exit() calls or SystemExit exceptions in the script
1388 1388 being run. This is particularly useful if IPython is being used to
1389 1389 run unittests, which always exit with a sys.exit() call. In such
1390 1390 cases you are interested in the output of the test results, not in
1391 1391 seeing a traceback of the unittest module.
1392 1392
1393 1393 -t: print timing information at the end of the run. IPython will give
1394 1394 you an estimated CPU time consumption for your script, which under
1395 1395 Unix uses the resource module to avoid the wraparound problems of
1396 1396 time.clock(). Under Unix, an estimate of time spent on system tasks
1397 1397 is also given (for Windows platforms this is reported as 0.0).
1398 1398
1399 1399 If -t is given, an additional -N<N> option can be given, where <N>
1400 1400 must be an integer indicating how many times you want the script to
1401 1401 run. The final timing report will include total and per run results.
1402 1402
1403 1403 For example (testing the script uniq_stable.py):
1404 1404
1405 1405 In [1]: run -t uniq_stable
1406 1406
1407 1407 IPython CPU timings (estimated):\\
1408 1408 User : 0.19597 s.\\
1409 1409 System: 0.0 s.\\
1410 1410
1411 1411 In [2]: run -t -N5 uniq_stable
1412 1412
1413 1413 IPython CPU timings (estimated):\\
1414 1414 Total runs performed: 5\\
1415 1415 Times : Total Per run\\
1416 1416 User : 0.910862 s, 0.1821724 s.\\
1417 1417 System: 0.0 s, 0.0 s.
1418 1418
1419 1419 -d: run your program under the control of pdb, the Python debugger.
1420 1420 This allows you to execute your program step by step, watch variables,
1421 1421 etc. Internally, what IPython does is similar to calling:
1422 1422
1423 1423 pdb.run('execfile("YOURFILENAME")')
1424 1424
1425 1425 with a breakpoint set on line 1 of your file. You can change the line
1426 1426 number for this automatic breakpoint to be <N> by using the -bN option
1427 1427 (where N must be an integer). For example:
1428 1428
1429 1429 %run -d -b40 myscript
1430 1430
1431 1431 will set the first breakpoint at line 40 in myscript.py. Note that
1432 1432 the first breakpoint must be set on a line which actually does
1433 1433 something (not a comment or docstring) for it to stop execution.
1434 1434
1435 1435 When the pdb debugger starts, you will see a (Pdb) prompt. You must
1436 1436 first enter 'c' (without qoutes) to start execution up to the first
1437 1437 breakpoint.
1438 1438
1439 1439 Entering 'help' gives information about the use of the debugger. You
1440 1440 can easily see pdb's full documentation with "import pdb;pdb.help()"
1441 1441 at a prompt.
1442 1442
1443 1443 -p: run program under the control of the Python profiler module (which
1444 1444 prints a detailed report of execution times, function calls, etc).
1445 1445
1446 1446 You can pass other options after -p which affect the behavior of the
1447 1447 profiler itself. See the docs for %prun for details.
1448 1448
1449 1449 In this mode, the program's variables do NOT propagate back to the
1450 1450 IPython interactive namespace (because they remain in the namespace
1451 1451 where the profiler executes them).
1452 1452
1453 1453 Internally this triggers a call to %prun, see its documentation for
1454 1454 details on the options available specifically for profiling."""
1455 1455
1456 1456 # get arguments and set sys.argv for program to be run.
1457 1457 opts,arg_lst = self.parse_options(parameter_s,'nidtN:b:pD:l:rs:T:e',
1458 1458 mode='list',list_all=1)
1459 1459
1460 1460 try:
1461 1461 filename = get_py_filename(arg_lst[0])
1462 1462 except IndexError:
1463 1463 warn('you must provide at least a filename.')
1464 1464 print '\n%run:\n',OInspect.getdoc(self.magic_run)
1465 1465 return
1466 1466 except IOError,msg:
1467 1467 error(msg)
1468 1468 return
1469 1469
1470 1470 # Control the response to exit() calls made by the script being run
1471 1471 exit_ignore = opts.has_key('e')
1472 1472
1473 1473 # Make sure that the running script gets a proper sys.argv as if it
1474 1474 # were run from a system shell.
1475 1475 save_argv = sys.argv # save it for later restoring
1476 1476 sys.argv = [filename]+ arg_lst[1:] # put in the proper filename
1477 1477
1478 1478 if opts.has_key('i'):
1479 1479 prog_ns = self.shell.user_ns
1480 1480 __name__save = self.shell.user_ns['__name__']
1481 1481 prog_ns['__name__'] = '__main__'
1482 1482 else:
1483 1483 if opts.has_key('n'):
1484 1484 name = os.path.splitext(os.path.basename(filename))[0]
1485 1485 else:
1486 1486 name = '__main__'
1487 1487 prog_ns = {'__name__':name}
1488 1488
1489 1489 # Since '%run foo' emulates 'python foo.py' at the cmd line, we must
1490 1490 # set the __file__ global in the script's namespace
1491 1491 prog_ns['__file__'] = filename
1492 1492
1493 1493 # pickle fix. See iplib for an explanation. But we need to make sure
1494 1494 # that, if we overwrite __main__, we replace it at the end
1495 1495 if prog_ns['__name__'] == '__main__':
1496 1496 restore_main = sys.modules['__main__']
1497 1497 else:
1498 1498 restore_main = False
1499 1499
1500 1500 sys.modules[prog_ns['__name__']] = FakeModule(prog_ns)
1501 1501
1502 1502 stats = None
1503 1503 try:
1504 1504 if opts.has_key('p'):
1505 1505 stats = self.magic_prun('',0,opts,arg_lst,prog_ns)
1506 1506 else:
1507 1507 if opts.has_key('d'):
1508 1508 deb = Debugger.Pdb(self.shell.rc.colors)
1509 1509 # reset Breakpoint state, which is moronically kept
1510 1510 # in a class
1511 1511 bdb.Breakpoint.next = 1
1512 1512 bdb.Breakpoint.bplist = {}
1513 1513 bdb.Breakpoint.bpbynumber = [None]
1514 1514 # Set an initial breakpoint to stop execution
1515 1515 maxtries = 10
1516 1516 bp = int(opts.get('b',[1])[0])
1517 1517 checkline = deb.checkline(filename,bp)
1518 1518 if not checkline:
1519 1519 for bp in range(bp+1,bp+maxtries+1):
1520 1520 if deb.checkline(filename,bp):
1521 1521 break
1522 1522 else:
1523 1523 msg = ("\nI failed to find a valid line to set "
1524 1524 "a breakpoint\n"
1525 1525 "after trying up to line: %s.\n"
1526 1526 "Please set a valid breakpoint manually "
1527 1527 "with the -b option." % bp)
1528 1528 error(msg)
1529 1529 return
1530 1530 # if we find a good linenumber, set the breakpoint
1531 1531 deb.do_break('%s:%s' % (filename,bp))
1532 1532 # Start file run
1533 1533 print "NOTE: Enter 'c' at the",
1534 1534 print "ipdb> prompt to start your script."
1535 1535 try:
1536 1536 deb.run('execfile("%s")' % filename,prog_ns)
1537 1537 except:
1538 1538 etype, value, tb = sys.exc_info()
1539 1539 # Skip three frames in the traceback: the %run one,
1540 1540 # one inside bdb.py, and the command-line typed by the
1541 1541 # user (run by exec in pdb itself).
1542 1542 self.shell.InteractiveTB(etype,value,tb,tb_offset=3)
1543 1543 else:
1544 1544 if runner is None:
1545 1545 runner = self.shell.safe_execfile
1546 1546 if opts.has_key('t'):
1547 1547 try:
1548 1548 nruns = int(opts['N'][0])
1549 1549 if nruns < 1:
1550 1550 error('Number of runs must be >=1')
1551 1551 return
1552 1552 except (KeyError):
1553 1553 nruns = 1
1554 1554 if nruns == 1:
1555 1555 t0 = clock2()
1556 1556 runner(filename,prog_ns,prog_ns,exit_ignore=exit_ignore)
1557 1557 t1 = clock2()
1558 1558 t_usr = t1[0]-t0[0]
1559 1559 t_sys = t1[1]-t1[1]
1560 1560 print "\nIPython CPU timings (estimated):"
1561 1561 print " User : %10s s." % t_usr
1562 1562 print " System: %10s s." % t_sys
1563 1563 else:
1564 1564 runs = range(nruns)
1565 1565 t0 = clock2()
1566 1566 for nr in runs:
1567 1567 runner(filename,prog_ns,prog_ns,exit_ignore=exit_ignore)
1568 1568 t1 = clock2()
1569 1569 t_usr = t1[0]-t0[0]
1570 1570 t_sys = t1[1]-t1[1]
1571 1571 print "\nIPython CPU timings (estimated):"
1572 1572 print "Total runs performed:",nruns
1573 1573 print " Times : %10s %10s" % ('Total','Per run')
1574 1574 print " User : %10s s, %10s s." % (t_usr,t_usr/nruns)
1575 1575 print " System: %10s s, %10s s." % (t_sys,t_sys/nruns)
1576 1576
1577 1577 else:
1578 1578 runner(filename,prog_ns,prog_ns,exit_ignore=exit_ignore)
1579 1579 if opts.has_key('i'):
1580 1580 self.shell.user_ns['__name__'] = __name__save
1581 1581 else:
1582 1582 # update IPython interactive namespace
1583 1583 del prog_ns['__name__']
1584 1584 self.shell.user_ns.update(prog_ns)
1585 1585 finally:
1586 1586 sys.argv = save_argv
1587 1587 if restore_main:
1588 1588 sys.modules['__main__'] = restore_main
1589 1589 return stats
1590 1590
1591 1591 def magic_runlog(self, parameter_s =''):
1592 1592 """Run files as logs.
1593 1593
1594 1594 Usage:\\
1595 1595 %runlog file1 file2 ...
1596 1596
1597 1597 Run the named files (treating them as log files) in sequence inside
1598 1598 the interpreter, and return to the prompt. This is much slower than
1599 1599 %run because each line is executed in a try/except block, but it
1600 1600 allows running files with syntax errors in them.
1601 1601
1602 1602 Normally IPython will guess when a file is one of its own logfiles, so
1603 1603 you can typically use %run even for logs. This shorthand allows you to
1604 1604 force any file to be treated as a log file."""
1605 1605
1606 1606 for f in parameter_s.split():
1607 1607 self.shell.safe_execfile(f,self.shell.user_ns,
1608 1608 self.shell.user_ns,islog=1)
1609 1609
1610 1610 def magic_timeit(self, parameter_s =''):
1611 1611 """Time execution of a Python statement or expression
1612 1612
1613 1613 Usage:\\
1614 1614 %timeit [-n<N> -r<R> [-t|-c]] statement
1615 1615
1616 1616 Time execution of a Python statement or expression using the timeit
1617 1617 module.
1618 1618
1619 1619 Options:
1620 1620 -n<N>: execute the given statement <N> times in a loop. If this value
1621 1621 is not given, a fitting value is chosen.
1622 1622
1623 1623 -r<R>: repeat the loop iteration <R> times and take the best result.
1624 1624 Default: 3
1625 1625
1626 1626 -t: use time.time to measure the time, which is the default on Unix.
1627 1627 This function measures wall time.
1628 1628
1629 1629 -c: use time.clock to measure the time, which is the default on
1630 1630 Windows and measures wall time. On Unix, resource.getrusage is used
1631 1631 instead and returns the CPU user time.
1632 1632
1633 1633 -p<P>: use a precision of <P> digits to display the timing result.
1634 1634 Default: 3
1635 1635
1636 1636
1637 1637 Examples:\\
1638 1638 In [1]: %timeit pass
1639 1639 10000000 loops, best of 3: 53.3 ns per loop
1640 1640
1641 1641 In [2]: u = None
1642 1642
1643 1643 In [3]: %timeit u is None
1644 1644 10000000 loops, best of 3: 184 ns per loop
1645 1645
1646 1646 In [4]: %timeit -r 4 u == None
1647 1647 1000000 loops, best of 4: 242 ns per loop
1648 1648
1649 1649 In [5]: import time
1650 1650
1651 1651 In [6]: %timeit -n1 time.sleep(2)
1652 1652 1 loops, best of 3: 2 s per loop
1653 1653
1654 1654
1655 1655 The times reported by %timeit will be slightly higher than those
1656 1656 reported by the timeit.py script when variables are accessed. This is
1657 1657 due to the fact that %timeit executes the statement in the namespace
1658 1658 of the shell, compared with timeit.py, which uses a single setup
1659 1659 statement to import function or create variables. Generally, the bias
1660 1660 does not matter as long as results from timeit.py are not mixed with
1661 1661 those from %timeit."""
1662 1662
1663 1663 import timeit
1664 1664 import math
1665 1665
1666 1666 units = ["s", "ms", "\xc2\xb5s", "ns"]
1667 1667 scaling = [1, 1e3, 1e6, 1e9]
1668 1668
1669 1669 opts, stmt = self.parse_options(parameter_s,'n:r:tcp:',
1670 1670 posix=False)
1671 1671 if stmt == "":
1672 1672 return
1673 1673 timefunc = timeit.default_timer
1674 1674 number = int(getattr(opts, "n", 0))
1675 1675 repeat = int(getattr(opts, "r", timeit.default_repeat))
1676 1676 precision = int(getattr(opts, "p", 3))
1677 1677 if hasattr(opts, "t"):
1678 1678 timefunc = time.time
1679 1679 if hasattr(opts, "c"):
1680 1680 timefunc = clock
1681 1681
1682 1682 timer = timeit.Timer(timer=timefunc)
1683 1683 # this code has tight coupling to the inner workings of timeit.Timer,
1684 1684 # but is there a better way to achieve that the code stmt has access
1685 1685 # to the shell namespace?
1686 1686
1687 1687 src = timeit.template % {'stmt': timeit.reindent(stmt, 8),
1688 1688 'setup': "pass"}
1689 1689 code = compile(src, "<magic-timeit>", "exec")
1690 1690 ns = {}
1691 1691 exec code in self.shell.user_ns, ns
1692 1692 timer.inner = ns["inner"]
1693 1693
1694 1694 if number == 0:
1695 1695 # determine number so that 0.2 <= total time < 2.0
1696 1696 number = 1
1697 1697 for i in range(1, 10):
1698 1698 number *= 10
1699 1699 if timer.timeit(number) >= 0.2:
1700 1700 break
1701 1701
1702 1702 best = min(timer.repeat(repeat, number)) / number
1703 1703
1704 1704 if best > 0.0:
1705 1705 order = min(-int(math.floor(math.log10(best)) // 3), 3)
1706 1706 else:
1707 1707 order = 3
1708 1708 print "%d loops, best of %d: %.*g %s per loop" % (number, repeat,
1709 1709 precision,
1710 1710 best * scaling[order],
1711 1711 units[order])
1712 1712
1713 1713 def magic_time(self,parameter_s = ''):
1714 1714 """Time execution of a Python statement or expression.
1715 1715
1716 1716 The CPU and wall clock times are printed, and the value of the
1717 1717 expression (if any) is returned. Note that under Win32, system time
1718 1718 is always reported as 0, since it can not be measured.
1719 1719
1720 1720 This function provides very basic timing functionality. In Python
1721 1721 2.3, the timeit module offers more control and sophistication, so this
1722 1722 could be rewritten to use it (patches welcome).
1723 1723
1724 1724 Some examples:
1725 1725
1726 1726 In [1]: time 2**128
1727 1727 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1728 1728 Wall time: 0.00
1729 1729 Out[1]: 340282366920938463463374607431768211456L
1730 1730
1731 1731 In [2]: n = 1000000
1732 1732
1733 1733 In [3]: time sum(range(n))
1734 1734 CPU times: user 1.20 s, sys: 0.05 s, total: 1.25 s
1735 1735 Wall time: 1.37
1736 1736 Out[3]: 499999500000L
1737 1737
1738 1738 In [4]: time print 'hello world'
1739 1739 hello world
1740 1740 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1741 1741 Wall time: 0.00
1742 1742 """
1743 1743
1744 1744 # fail immediately if the given expression can't be compiled
1745 1745 try:
1746 1746 mode = 'eval'
1747 1747 code = compile(parameter_s,'<timed eval>',mode)
1748 1748 except SyntaxError:
1749 1749 mode = 'exec'
1750 1750 code = compile(parameter_s,'<timed exec>',mode)
1751 1751 # skew measurement as little as possible
1752 1752 glob = self.shell.user_ns
1753 1753 clk = clock2
1754 1754 wtime = time.time
1755 1755 # time execution
1756 1756 wall_st = wtime()
1757 1757 if mode=='eval':
1758 1758 st = clk()
1759 1759 out = eval(code,glob)
1760 1760 end = clk()
1761 1761 else:
1762 1762 st = clk()
1763 1763 exec code in glob
1764 1764 end = clk()
1765 1765 out = None
1766 1766 wall_end = wtime()
1767 1767 # Compute actual times and report
1768 1768 wall_time = wall_end-wall_st
1769 1769 cpu_user = end[0]-st[0]
1770 1770 cpu_sys = end[1]-st[1]
1771 1771 cpu_tot = cpu_user+cpu_sys
1772 1772 print "CPU times: user %.2f s, sys: %.2f s, total: %.2f s" % \
1773 1773 (cpu_user,cpu_sys,cpu_tot)
1774 1774 print "Wall time: %.2f" % wall_time
1775 1775 return out
1776 1776
1777 1777 def magic_macro(self,parameter_s = ''):
1778 1778 """Define a set of input lines as a macro for future re-execution.
1779 1779
1780 1780 Usage:\\
1781 1781 %macro [options] name n1-n2 n3-n4 ... n5 .. n6 ...
1782 1782
1783 1783 Options:
1784 1784
1785 1785 -r: use 'raw' input. By default, the 'processed' history is used,
1786 1786 so that magics are loaded in their transformed version to valid
1787 1787 Python. If this option is given, the raw input as typed as the
1788 1788 command line is used instead.
1789 1789
1790 1790 This will define a global variable called `name` which is a string
1791 1791 made of joining the slices and lines you specify (n1,n2,... numbers
1792 1792 above) from your input history into a single string. This variable
1793 1793 acts like an automatic function which re-executes those lines as if
1794 1794 you had typed them. You just type 'name' at the prompt and the code
1795 1795 executes.
1796 1796
1797 1797 The notation for indicating number ranges is: n1-n2 means 'use line
1798 1798 numbers n1,...n2' (the endpoint is included). That is, '5-7' means
1799 1799 using the lines numbered 5,6 and 7.
1800 1800
1801 1801 Note: as a 'hidden' feature, you can also use traditional python slice
1802 1802 notation, where N:M means numbers N through M-1.
1803 1803
1804 1804 For example, if your history contains (%hist prints it):
1805 1805
1806 1806 44: x=1\\
1807 1807 45: y=3\\
1808 1808 46: z=x+y\\
1809 1809 47: print x\\
1810 1810 48: a=5\\
1811 1811 49: print 'x',x,'y',y\\
1812 1812
1813 1813 you can create a macro with lines 44 through 47 (included) and line 49
1814 1814 called my_macro with:
1815 1815
1816 1816 In [51]: %macro my_macro 44-47 49
1817 1817
1818 1818 Now, typing `my_macro` (without quotes) will re-execute all this code
1819 1819 in one pass.
1820 1820
1821 1821 You don't need to give the line-numbers in order, and any given line
1822 1822 number can appear multiple times. You can assemble macros with any
1823 1823 lines from your input history in any order.
1824 1824
1825 1825 The macro is a simple object which holds its value in an attribute,
1826 1826 but IPython's display system checks for macros and executes them as
1827 1827 code instead of printing them when you type their name.
1828 1828
1829 1829 You can view a macro's contents by explicitly printing it with:
1830 1830
1831 1831 'print macro_name'.
1832 1832
1833 1833 For one-off cases which DON'T contain magic function calls in them you
1834 1834 can obtain similar results by explicitly executing slices from your
1835 1835 input history with:
1836 1836
1837 1837 In [60]: exec In[44:48]+In[49]"""
1838 1838
1839 1839 opts,args = self.parse_options(parameter_s,'r',mode='list')
1840 1840 name,ranges = args[0], args[1:]
1841 1841 #print 'rng',ranges # dbg
1842 1842 lines = self.extract_input_slices(ranges,opts.has_key('r'))
1843 1843 macro = Macro(lines)
1844 1844 self.shell.user_ns.update({name:macro})
1845 1845 print 'Macro `%s` created. To execute, type its name (without quotes).' % name
1846 1846 print 'Macro contents:'
1847 1847 print macro,
1848 1848
1849 1849 def magic_save(self,parameter_s = ''):
1850 1850 """Save a set of lines to a given filename.
1851 1851
1852 1852 Usage:\\
1853 1853 %save [options] filename n1-n2 n3-n4 ... n5 .. n6 ...
1854 1854
1855 1855 Options:
1856 1856
1857 1857 -r: use 'raw' input. By default, the 'processed' history is used,
1858 1858 so that magics are loaded in their transformed version to valid
1859 1859 Python. If this option is given, the raw input as typed as the
1860 1860 command line is used instead.
1861 1861
1862 1862 This function uses the same syntax as %macro for line extraction, but
1863 1863 instead of creating a macro it saves the resulting string to the
1864 1864 filename you specify.
1865 1865
1866 1866 It adds a '.py' extension to the file if you don't do so yourself, and
1867 1867 it asks for confirmation before overwriting existing files."""
1868 1868
1869 1869 opts,args = self.parse_options(parameter_s,'r',mode='list')
1870 1870 fname,ranges = args[0], args[1:]
1871 1871 if not fname.endswith('.py'):
1872 1872 fname += '.py'
1873 1873 if os.path.isfile(fname):
1874 1874 ans = raw_input('File `%s` exists. Overwrite (y/[N])? ' % fname)
1875 1875 if ans.lower() not in ['y','yes']:
1876 1876 print 'Operation cancelled.'
1877 1877 return
1878 1878 cmds = ''.join(self.extract_input_slices(ranges,opts.has_key('r')))
1879 1879 f = file(fname,'w')
1880 1880 f.write(cmds)
1881 1881 f.close()
1882 1882 print 'The following commands were written to file `%s`:' % fname
1883 1883 print cmds
1884 1884
1885 1885 def _edit_macro(self,mname,macro):
1886 1886 """open an editor with the macro data in a file"""
1887 1887 filename = self.shell.mktempfile(macro.value)
1888 1888 self.shell.hooks.editor(filename)
1889 1889
1890 1890 # and make a new macro object, to replace the old one
1891 1891 mfile = open(filename)
1892 1892 mvalue = mfile.read()
1893 1893 mfile.close()
1894 1894 self.shell.user_ns[mname] = Macro(mvalue)
1895 1895
1896 1896 def magic_ed(self,parameter_s=''):
1897 1897 """Alias to %edit."""
1898 1898 return self.magic_edit(parameter_s)
1899 1899
1900 1900 def magic_edit(self,parameter_s='',last_call=['','']):
1901 1901 """Bring up an editor and execute the resulting code.
1902 1902
1903 1903 Usage:
1904 1904 %edit [options] [args]
1905 1905
1906 1906 %edit runs IPython's editor hook. The default version of this hook is
1907 1907 set to call the __IPYTHON__.rc.editor command. This is read from your
1908 1908 environment variable $EDITOR. If this isn't found, it will default to
1909 1909 vi under Linux/Unix and to notepad under Windows. See the end of this
1910 1910 docstring for how to change the editor hook.
1911 1911
1912 1912 You can also set the value of this editor via the command line option
1913 1913 '-editor' or in your ipythonrc file. This is useful if you wish to use
1914 1914 specifically for IPython an editor different from your typical default
1915 1915 (and for Windows users who typically don't set environment variables).
1916 1916
1917 1917 This command allows you to conveniently edit multi-line code right in
1918 1918 your IPython session.
1919 1919
1920 1920 If called without arguments, %edit opens up an empty editor with a
1921 1921 temporary file and will execute the contents of this file when you
1922 1922 close it (don't forget to save it!).
1923 1923
1924 1924
1925 1925 Options:
1926 1926
1927 1927 -n <number>: open the editor at a specified line number. By default,
1928 1928 the IPython editor hook uses the unix syntax 'editor +N filename', but
1929 1929 you can configure this by providing your own modified hook if your
1930 1930 favorite editor supports line-number specifications with a different
1931 1931 syntax.
1932 1932
1933 1933 -p: this will call the editor with the same data as the previous time
1934 1934 it was used, regardless of how long ago (in your current session) it
1935 1935 was.
1936 1936
1937 1937 -r: use 'raw' input. This option only applies to input taken from the
1938 1938 user's history. By default, the 'processed' history is used, so that
1939 1939 magics are loaded in their transformed version to valid Python. If
1940 1940 this option is given, the raw input as typed as the command line is
1941 1941 used instead. When you exit the editor, it will be executed by
1942 1942 IPython's own processor.
1943 1943
1944 1944 -x: do not execute the edited code immediately upon exit. This is
1945 1945 mainly useful if you are editing programs which need to be called with
1946 1946 command line arguments, which you can then do using %run.
1947 1947
1948 1948
1949 1949 Arguments:
1950 1950
1951 1951 If arguments are given, the following possibilites exist:
1952 1952
1953 1953 - The arguments are numbers or pairs of colon-separated numbers (like
1954 1954 1 4:8 9). These are interpreted as lines of previous input to be
1955 1955 loaded into the editor. The syntax is the same of the %macro command.
1956 1956
1957 1957 - If the argument doesn't start with a number, it is evaluated as a
1958 1958 variable and its contents loaded into the editor. You can thus edit
1959 1959 any string which contains python code (including the result of
1960 1960 previous edits).
1961 1961
1962 1962 - If the argument is the name of an object (other than a string),
1963 1963 IPython will try to locate the file where it was defined and open the
1964 1964 editor at the point where it is defined. You can use `%edit function`
1965 1965 to load an editor exactly at the point where 'function' is defined,
1966 1966 edit it and have the file be executed automatically.
1967 1967
1968 1968 If the object is a macro (see %macro for details), this opens up your
1969 1969 specified editor with a temporary file containing the macro's data.
1970 1970 Upon exit, the macro is reloaded with the contents of the file.
1971 1971
1972 1972 Note: opening at an exact line is only supported under Unix, and some
1973 1973 editors (like kedit and gedit up to Gnome 2.8) do not understand the
1974 1974 '+NUMBER' parameter necessary for this feature. Good editors like
1975 1975 (X)Emacs, vi, jed, pico and joe all do.
1976 1976
1977 1977 - If the argument is not found as a variable, IPython will look for a
1978 1978 file with that name (adding .py if necessary) and load it into the
1979 1979 editor. It will execute its contents with execfile() when you exit,
1980 1980 loading any code in the file into your interactive namespace.
1981 1981
1982 1982 After executing your code, %edit will return as output the code you
1983 1983 typed in the editor (except when it was an existing file). This way
1984 1984 you can reload the code in further invocations of %edit as a variable,
1985 1985 via _<NUMBER> or Out[<NUMBER>], where <NUMBER> is the prompt number of
1986 1986 the output.
1987 1987
1988 1988 Note that %edit is also available through the alias %ed.
1989 1989
1990 1990 This is an example of creating a simple function inside the editor and
1991 1991 then modifying it. First, start up the editor:
1992 1992
1993 1993 In [1]: ed\\
1994 1994 Editing... done. Executing edited code...\\
1995 1995 Out[1]: 'def foo():\\n print "foo() was defined in an editing session"\\n'
1996 1996
1997 1997 We can then call the function foo():
1998 1998
1999 1999 In [2]: foo()\\
2000 2000 foo() was defined in an editing session
2001 2001
2002 2002 Now we edit foo. IPython automatically loads the editor with the
2003 2003 (temporary) file where foo() was previously defined:
2004 2004
2005 2005 In [3]: ed foo\\
2006 2006 Editing... done. Executing edited code...
2007 2007
2008 2008 And if we call foo() again we get the modified version:
2009 2009
2010 2010 In [4]: foo()\\
2011 2011 foo() has now been changed!
2012 2012
2013 2013 Here is an example of how to edit a code snippet successive
2014 2014 times. First we call the editor:
2015 2015
2016 2016 In [8]: ed\\
2017 2017 Editing... done. Executing edited code...\\
2018 2018 hello\\
2019 2019 Out[8]: "print 'hello'\\n"
2020 2020
2021 2021 Now we call it again with the previous output (stored in _):
2022 2022
2023 2023 In [9]: ed _\\
2024 2024 Editing... done. Executing edited code...\\
2025 2025 hello world\\
2026 2026 Out[9]: "print 'hello world'\\n"
2027 2027
2028 2028 Now we call it with the output #8 (stored in _8, also as Out[8]):
2029 2029
2030 2030 In [10]: ed _8\\
2031 2031 Editing... done. Executing edited code...\\
2032 2032 hello again\\
2033 2033 Out[10]: "print 'hello again'\\n"
2034 2034
2035 2035
2036 2036 Changing the default editor hook:
2037 2037
2038 2038 If you wish to write your own editor hook, you can put it in a
2039 2039 configuration file which you load at startup time. The default hook
2040 2040 is defined in the IPython.hooks module, and you can use that as a
2041 2041 starting example for further modifications. That file also has
2042 2042 general instructions on how to set a new hook for use once you've
2043 2043 defined it."""
2044 2044
2045 2045 # FIXME: This function has become a convoluted mess. It needs a
2046 2046 # ground-up rewrite with clean, simple logic.
2047 2047
2048 2048 def make_filename(arg):
2049 2049 "Make a filename from the given args"
2050 2050 try:
2051 2051 filename = get_py_filename(arg)
2052 2052 except IOError:
2053 2053 if args.endswith('.py'):
2054 2054 filename = arg
2055 2055 else:
2056 2056 filename = None
2057 2057 return filename
2058 2058
2059 2059 # custom exceptions
2060 2060 class DataIsObject(Exception): pass
2061 2061
2062 2062 opts,args = self.parse_options(parameter_s,'prxn:')
2063 2063 # Set a few locals from the options for convenience:
2064 2064 opts_p = opts.has_key('p')
2065 2065 opts_r = opts.has_key('r')
2066 2066
2067 2067 # Default line number value
2068 2068 lineno = opts.get('n',None)
2069 2069
2070 2070 if opts_p:
2071 2071 args = '_%s' % last_call[0]
2072 2072 if not self.shell.user_ns.has_key(args):
2073 2073 args = last_call[1]
2074 2074
2075 2075 # use last_call to remember the state of the previous call, but don't
2076 2076 # let it be clobbered by successive '-p' calls.
2077 2077 try:
2078 2078 last_call[0] = self.shell.outputcache.prompt_count
2079 2079 if not opts_p:
2080 2080 last_call[1] = parameter_s
2081 2081 except:
2082 2082 pass
2083 2083
2084 2084 # by default this is done with temp files, except when the given
2085 2085 # arg is a filename
2086 2086 use_temp = 1
2087 2087
2088 2088 if re.match(r'\d',args):
2089 2089 # Mode where user specifies ranges of lines, like in %macro.
2090 2090 # This means that you can't edit files whose names begin with
2091 2091 # numbers this way. Tough.
2092 2092 ranges = args.split()
2093 2093 data = ''.join(self.extract_input_slices(ranges,opts_r))
2094 2094 elif args.endswith('.py'):
2095 2095 filename = make_filename(args)
2096 2096 data = ''
2097 2097 use_temp = 0
2098 2098 elif args:
2099 2099 try:
2100 2100 # Load the parameter given as a variable. If not a string,
2101 2101 # process it as an object instead (below)
2102 2102
2103 2103 #print '*** args',args,'type',type(args) # dbg
2104 2104 data = eval(args,self.shell.user_ns)
2105 2105 if not type(data) in StringTypes:
2106 2106 raise DataIsObject
2107 2107
2108 2108 except (NameError,SyntaxError):
2109 2109 # given argument is not a variable, try as a filename
2110 2110 filename = make_filename(args)
2111 2111 if filename is None:
2112 2112 warn("Argument given (%s) can't be found as a variable "
2113 2113 "or as a filename." % args)
2114 2114 return
2115 2115
2116 2116 data = ''
2117 2117 use_temp = 0
2118 2118 except DataIsObject:
2119 2119
2120 2120 # macros have a special edit function
2121 2121 if isinstance(data,Macro):
2122 2122 self._edit_macro(args,data)
2123 2123 return
2124 2124
2125 2125 # For objects, try to edit the file where they are defined
2126 2126 try:
2127 2127 filename = inspect.getabsfile(data)
2128 2128 datafile = 1
2129 2129 except TypeError:
2130 2130 filename = make_filename(args)
2131 2131 datafile = 1
2132 2132 warn('Could not find file where `%s` is defined.\n'
2133 2133 'Opening a file named `%s`' % (args,filename))
2134 2134 # Now, make sure we can actually read the source (if it was in
2135 2135 # a temp file it's gone by now).
2136 2136 if datafile:
2137 2137 try:
2138 2138 if lineno is None:
2139 2139 lineno = inspect.getsourcelines(data)[1]
2140 2140 except IOError:
2141 2141 filename = make_filename(args)
2142 2142 if filename is None:
2143 2143 warn('The file `%s` where `%s` was defined cannot '
2144 2144 'be read.' % (filename,data))
2145 2145 return
2146 2146 use_temp = 0
2147 2147 else:
2148 2148 data = ''
2149 2149
2150 2150 if use_temp:
2151 2151 filename = self.shell.mktempfile(data)
2152 2152 print 'IPython will make a temporary file named:',filename
2153 2153
2154 2154 # do actual editing here
2155 2155 print 'Editing...',
2156 2156 sys.stdout.flush()
2157 2157 self.shell.hooks.editor(filename,lineno)
2158 2158 if opts.has_key('x'): # -x prevents actual execution
2159 2159 print
2160 2160 else:
2161 2161 print 'done. Executing edited code...'
2162 2162 if opts_r:
2163 2163 self.shell.runlines(file_read(filename))
2164 2164 else:
2165 2165 self.shell.safe_execfile(filename,self.shell.user_ns)
2166 2166 if use_temp:
2167 2167 try:
2168 2168 return open(filename).read()
2169 2169 except IOError,msg:
2170 2170 if msg.filename == filename:
2171 2171 warn('File not found. Did you forget to save?')
2172 2172 return
2173 2173 else:
2174 2174 self.shell.showtraceback()
2175 2175
2176 2176 def magic_xmode(self,parameter_s = ''):
2177 2177 """Switch modes for the exception handlers.
2178 2178
2179 2179 Valid modes: Plain, Context and Verbose.
2180 2180
2181 2181 If called without arguments, acts as a toggle."""
2182 2182
2183 2183 def xmode_switch_err(name):
2184 2184 warn('Error changing %s exception modes.\n%s' %
2185 2185 (name,sys.exc_info()[1]))
2186 2186
2187 2187 shell = self.shell
2188 2188 new_mode = parameter_s.strip().capitalize()
2189 2189 try:
2190 2190 shell.InteractiveTB.set_mode(mode=new_mode)
2191 2191 print 'Exception reporting mode:',shell.InteractiveTB.mode
2192 2192 except:
2193 2193 xmode_switch_err('user')
2194 2194
2195 2195 # threaded shells use a special handler in sys.excepthook
2196 2196 if shell.isthreaded:
2197 2197 try:
2198 2198 shell.sys_excepthook.set_mode(mode=new_mode)
2199 2199 except:
2200 2200 xmode_switch_err('threaded')
2201 2201
2202 2202 def magic_colors(self,parameter_s = ''):
2203 2203 """Switch color scheme for prompts, info system and exception handlers.
2204 2204
2205 2205 Currently implemented schemes: NoColor, Linux, LightBG.
2206 2206
2207 2207 Color scheme names are not case-sensitive."""
2208 2208
2209 2209 def color_switch_err(name):
2210 2210 warn('Error changing %s color schemes.\n%s' %
2211 2211 (name,sys.exc_info()[1]))
2212 2212
2213 2213
2214 2214 new_scheme = parameter_s.strip()
2215 2215 if not new_scheme:
2216 2216 print 'You must specify a color scheme.'
2217 2217 return
2218 2218 import IPython.rlineimpl as readline
2219 2219 if not readline.have_readline:
2220 2220 msg = """\
2221 2221 Proper color support under MS Windows requires the pyreadline library.
2222 2222 You can find it at:
2223 2223 http://ipython.scipy.org/moin/PyReadline/Intro
2224 2224 Gary's readline needs the ctypes module, from:
2225 2225 http://starship.python.net/crew/theller/ctypes
2226 2226 (Note that ctypes is already part of Python versions 2.5 and newer).
2227 2227
2228 2228 Defaulting color scheme to 'NoColor'"""
2229 2229 new_scheme = 'NoColor'
2230 2230 warn(msg)
2231 2231 # local shortcut
2232 2232 shell = self.shell
2233 2233
2234 2234 # Set prompt colors
2235 2235 try:
2236 2236 shell.outputcache.set_colors(new_scheme)
2237 2237 except:
2238 2238 color_switch_err('prompt')
2239 2239 else:
2240 2240 shell.rc.colors = \
2241 2241 shell.outputcache.color_table.active_scheme_name
2242 2242 # Set exception colors
2243 2243 try:
2244 2244 shell.InteractiveTB.set_colors(scheme = new_scheme)
2245 2245 shell.SyntaxTB.set_colors(scheme = new_scheme)
2246 2246 except:
2247 2247 color_switch_err('exception')
2248 2248
2249 2249 # threaded shells use a verbose traceback in sys.excepthook
2250 2250 if shell.isthreaded:
2251 2251 try:
2252 2252 shell.sys_excepthook.set_colors(scheme=new_scheme)
2253 2253 except:
2254 2254 color_switch_err('system exception handler')
2255 2255
2256 2256 # Set info (for 'object?') colors
2257 2257 if shell.rc.color_info:
2258 2258 try:
2259 2259 shell.inspector.set_active_scheme(new_scheme)
2260 2260 except:
2261 2261 color_switch_err('object inspector')
2262 2262 else:
2263 2263 shell.inspector.set_active_scheme('NoColor')
2264 2264
2265 2265 def magic_color_info(self,parameter_s = ''):
2266 2266 """Toggle color_info.
2267 2267
2268 2268 The color_info configuration parameter controls whether colors are
2269 2269 used for displaying object details (by things like %psource, %pfile or
2270 2270 the '?' system). This function toggles this value with each call.
2271 2271
2272 2272 Note that unless you have a fairly recent pager (less works better
2273 2273 than more) in your system, using colored object information displays
2274 2274 will not work properly. Test it and see."""
2275 2275
2276 2276 self.shell.rc.color_info = 1 - self.shell.rc.color_info
2277 2277 self.magic_colors(self.shell.rc.colors)
2278 2278 print 'Object introspection functions have now coloring:',
2279 2279 print ['OFF','ON'][self.shell.rc.color_info]
2280 2280
2281 2281 def magic_Pprint(self, parameter_s=''):
2282 2282 """Toggle pretty printing on/off."""
2283 2283
2284 2284 self.shell.rc.pprint = 1 - self.shell.rc.pprint
2285 2285 print 'Pretty printing has been turned', \
2286 2286 ['OFF','ON'][self.shell.rc.pprint]
2287 2287
2288 2288 def magic_exit(self, parameter_s=''):
2289 2289 """Exit IPython, confirming if configured to do so.
2290 2290
2291 2291 You can configure whether IPython asks for confirmation upon exit by
2292 2292 setting the confirm_exit flag in the ipythonrc file."""
2293 2293
2294 2294 self.shell.exit()
2295 2295
2296 2296 def magic_quit(self, parameter_s=''):
2297 2297 """Exit IPython, confirming if configured to do so (like %exit)"""
2298 2298
2299 2299 self.shell.exit()
2300 2300
2301 2301 def magic_Exit(self, parameter_s=''):
2302 2302 """Exit IPython without confirmation."""
2303 2303
2304 2304 self.shell.exit_now = True
2305 2305
2306 2306 def magic_Quit(self, parameter_s=''):
2307 2307 """Exit IPython without confirmation (like %Exit)."""
2308 2308
2309 2309 self.shell.exit_now = True
2310 2310
2311 2311 #......................................................................
2312 2312 # Functions to implement unix shell-type things
2313 2313
2314 2314 def magic_alias(self, parameter_s = ''):
2315 2315 """Define an alias for a system command.
2316 2316
2317 2317 '%alias alias_name cmd' defines 'alias_name' as an alias for 'cmd'
2318 2318
2319 2319 Then, typing 'alias_name params' will execute the system command 'cmd
2320 2320 params' (from your underlying operating system).
2321 2321
2322 2322 Aliases have lower precedence than magic functions and Python normal
2323 2323 variables, so if 'foo' is both a Python variable and an alias, the
2324 2324 alias can not be executed until 'del foo' removes the Python variable.
2325 2325
2326 2326 You can use the %l specifier in an alias definition to represent the
2327 2327 whole line when the alias is called. For example:
2328 2328
2329 2329 In [2]: alias all echo "Input in brackets: <%l>"\\
2330 2330 In [3]: all hello world\\
2331 2331 Input in brackets: <hello world>
2332 2332
2333 2333 You can also define aliases with parameters using %s specifiers (one
2334 2334 per parameter):
2335 2335
2336 2336 In [1]: alias parts echo first %s second %s\\
2337 2337 In [2]: %parts A B\\
2338 2338 first A second B\\
2339 2339 In [3]: %parts A\\
2340 2340 Incorrect number of arguments: 2 expected.\\
2341 2341 parts is an alias to: 'echo first %s second %s'
2342 2342
2343 2343 Note that %l and %s are mutually exclusive. You can only use one or
2344 2344 the other in your aliases.
2345 2345
2346 2346 Aliases expand Python variables just like system calls using ! or !!
2347 2347 do: all expressions prefixed with '$' get expanded. For details of
2348 2348 the semantic rules, see PEP-215:
2349 2349 http://www.python.org/peps/pep-0215.html. This is the library used by
2350 2350 IPython for variable expansion. If you want to access a true shell
2351 2351 variable, an extra $ is necessary to prevent its expansion by IPython:
2352 2352
2353 2353 In [6]: alias show echo\\
2354 2354 In [7]: PATH='A Python string'\\
2355 2355 In [8]: show $PATH\\
2356 2356 A Python string\\
2357 2357 In [9]: show $$PATH\\
2358 2358 /usr/local/lf9560/bin:/usr/local/intel/compiler70/ia32/bin:...
2359 2359
2360 2360 You can use the alias facility to acess all of $PATH. See the %rehash
2361 2361 and %rehashx functions, which automatically create aliases for the
2362 2362 contents of your $PATH.
2363 2363
2364 2364 If called with no parameters, %alias prints the current alias table."""
2365 2365
2366 2366 par = parameter_s.strip()
2367 2367 if not par:
2368 stored = self.db['stored_aliases']
2368 stored = self.db.get('stored_aliases', {} )
2369 2369 atab = self.shell.alias_table
2370 2370 aliases = atab.keys()
2371 2371 aliases.sort()
2372 2372 res = []
2373 2373 showlast = []
2374 2374 for alias in aliases:
2375 2375 tgt = atab[alias][1]
2376 2376 # 'interesting' aliases
2377 2377 if (alias in stored or
2378 2378 alias != os.path.splitext(tgt)[0] or
2379 2379 ' ' in tgt):
2380 2380 showlast.append((alias, tgt))
2381 2381 else:
2382 2382 res.append((alias, tgt ))
2383 2383
2384 2384 # show most interesting aliases last
2385 2385 res.extend(showlast)
2386 2386 print "Total number of aliases:",len(aliases)
2387 2387 return res
2388 2388 try:
2389 2389 alias,cmd = par.split(None,1)
2390 2390 except:
2391 2391 print OInspect.getdoc(self.magic_alias)
2392 2392 else:
2393 2393 nargs = cmd.count('%s')
2394 2394 if nargs>0 and cmd.find('%l')>=0:
2395 2395 error('The %s and %l specifiers are mutually exclusive '
2396 2396 'in alias definitions.')
2397 2397 else: # all looks OK
2398 2398 self.shell.alias_table[alias] = (nargs,cmd)
2399 2399 self.shell.alias_table_validate(verbose=0)
2400 2400 # end magic_alias
2401 2401
2402 2402 def magic_unalias(self, parameter_s = ''):
2403 2403 """Remove an alias"""
2404 2404
2405 2405 aname = parameter_s.strip()
2406 2406 if aname in self.shell.alias_table:
2407 2407 del self.shell.alias_table[aname]
2408 stored = self.db['stored_aliases']
2408 stored = self.db.get('stored_aliases', {} )
2409 2409 if aname in stored:
2410 2410 print "Removing %stored alias",aname
2411 2411 del stored[aname]
2412 2412 self.db['stored_aliases'] = stored
2413 2413
2414 2414 def magic_rehash(self, parameter_s = ''):
2415 2415 """Update the alias table with all entries in $PATH.
2416 2416
2417 2417 This version does no checks on execute permissions or whether the
2418 2418 contents of $PATH are truly files (instead of directories or something
2419 2419 else). For such a safer (but slower) version, use %rehashx."""
2420 2420
2421 2421 # This function (and rehashx) manipulate the alias_table directly
2422 2422 # rather than calling magic_alias, for speed reasons. A rehash on a
2423 2423 # typical Linux box involves several thousand entries, so efficiency
2424 2424 # here is a top concern.
2425 2425
2426 2426 path = filter(os.path.isdir,os.environ['PATH'].split(os.pathsep))
2427 2427 alias_table = self.shell.alias_table
2428 2428 for pdir in path:
2429 2429 for ff in os.listdir(pdir):
2430 2430 # each entry in the alias table must be (N,name), where
2431 2431 # N is the number of positional arguments of the alias.
2432 2432 alias_table[ff] = (0,ff)
2433 2433 # Make sure the alias table doesn't contain keywords or builtins
2434 2434 self.shell.alias_table_validate()
2435 2435 # Call again init_auto_alias() so we get 'rm -i' and other modified
2436 2436 # aliases since %rehash will probably clobber them
2437 2437 self.shell.init_auto_alias()
2438 2438
2439 2439 def magic_rehashx(self, parameter_s = ''):
2440 2440 """Update the alias table with all executable files in $PATH.
2441 2441
2442 2442 This version explicitly checks that every entry in $PATH is a file
2443 2443 with execute access (os.X_OK), so it is much slower than %rehash.
2444 2444
2445 2445 Under Windows, it checks executability as a match agains a
2446 2446 '|'-separated string of extensions, stored in the IPython config
2447 2447 variable win_exec_ext. This defaults to 'exe|com|bat'. """
2448 2448
2449 2449 path = [os.path.abspath(os.path.expanduser(p)) for p in
2450 2450 os.environ['PATH'].split(os.pathsep)]
2451 2451 path = filter(os.path.isdir,path)
2452 2452
2453 2453 alias_table = self.shell.alias_table
2454 2454 syscmdlist = []
2455 2455 if os.name == 'posix':
2456 2456 isexec = lambda fname:os.path.isfile(fname) and \
2457 2457 os.access(fname,os.X_OK)
2458 2458 else:
2459 2459
2460 2460 try:
2461 2461 winext = os.environ['pathext'].replace(';','|').replace('.','')
2462 2462 except KeyError:
2463 2463 winext = 'exe|com|bat|py'
2464 2464 if 'py' not in winext:
2465 2465 winext += '|py'
2466 2466 execre = re.compile(r'(.*)\.(%s)$' % winext,re.IGNORECASE)
2467 2467 isexec = lambda fname:os.path.isfile(fname) and execre.match(fname)
2468 2468 savedir = os.getcwd()
2469 2469 try:
2470 2470 # write the whole loop for posix/Windows so we don't have an if in
2471 2471 # the innermost part
2472 2472 if os.name == 'posix':
2473 2473 for pdir in path:
2474 2474 os.chdir(pdir)
2475 2475 for ff in os.listdir(pdir):
2476 2476 if isexec(ff) and ff not in self.shell.no_alias:
2477 2477 # each entry in the alias table must be (N,name),
2478 2478 # where N is the number of positional arguments of the
2479 2479 # alias.
2480 2480 alias_table[ff] = (0,ff)
2481 2481 syscmdlist.append(ff)
2482 2482 else:
2483 2483 for pdir in path:
2484 2484 os.chdir(pdir)
2485 2485 for ff in os.listdir(pdir):
2486 2486 if isexec(ff) and os.path.splitext(ff)[0] not in self.shell.no_alias:
2487 2487 alias_table[execre.sub(r'\1',ff)] = (0,ff)
2488 2488 syscmdlist.append(ff)
2489 2489 # Make sure the alias table doesn't contain keywords or builtins
2490 2490 self.shell.alias_table_validate()
2491 2491 # Call again init_auto_alias() so we get 'rm -i' and other
2492 2492 # modified aliases since %rehashx will probably clobber them
2493 2493 self.shell.init_auto_alias()
2494 2494 db = self.getapi().db
2495 2495 db['syscmdlist'] = syscmdlist
2496 2496 finally:
2497 2497 os.chdir(savedir)
2498 2498
2499 2499 def magic_pwd(self, parameter_s = ''):
2500 2500 """Return the current working directory path."""
2501 2501 return os.getcwd()
2502 2502
2503 2503 def magic_cd(self, parameter_s=''):
2504 2504 """Change the current working directory.
2505 2505
2506 2506 This command automatically maintains an internal list of directories
2507 2507 you visit during your IPython session, in the variable _dh. The
2508 2508 command %dhist shows this history nicely formatted.
2509 2509
2510 2510 Usage:
2511 2511
2512 2512 cd 'dir': changes to directory 'dir'.
2513 2513
2514 2514 cd -: changes to the last visited directory.
2515 2515
2516 2516 cd -<n>: changes to the n-th directory in the directory history.
2517 2517
2518 2518 cd -b <bookmark_name>: jump to a bookmark set by %bookmark
2519 2519 (note: cd <bookmark_name> is enough if there is no
2520 2520 directory <bookmark_name>, but a bookmark with the name exists.)
2521 2521
2522 2522 Options:
2523 2523
2524 2524 -q: quiet. Do not print the working directory after the cd command is
2525 2525 executed. By default IPython's cd command does print this directory,
2526 2526 since the default prompts do not display path information.
2527 2527
2528 2528 Note that !cd doesn't work for this purpose because the shell where
2529 2529 !command runs is immediately discarded after executing 'command'."""
2530 2530
2531 2531 parameter_s = parameter_s.strip()
2532 2532 #bkms = self.shell.persist.get("bookmarks",{})
2533 2533
2534 2534 numcd = re.match(r'(-)(\d+)$',parameter_s)
2535 2535 # jump in directory history by number
2536 2536 if numcd:
2537 2537 nn = int(numcd.group(2))
2538 2538 try:
2539 2539 ps = self.shell.user_ns['_dh'][nn]
2540 2540 except IndexError:
2541 2541 print 'The requested directory does not exist in history.'
2542 2542 return
2543 2543 else:
2544 2544 opts = {}
2545 2545 else:
2546 2546 #turn all non-space-escaping backslashes to slashes,
2547 2547 # for c:\windows\directory\names\
2548 2548 parameter_s = re.sub(r'\\(?! )','/', parameter_s)
2549 2549 opts,ps = self.parse_options(parameter_s,'qb',mode='string')
2550 2550 # jump to previous
2551 2551 if ps == '-':
2552 2552 try:
2553 2553 ps = self.shell.user_ns['_dh'][-2]
2554 2554 except IndexError:
2555 2555 print 'No previous directory to change to.'
2556 2556 return
2557 2557 # jump to bookmark if needed
2558 2558 else:
2559 2559 if not os.path.isdir(ps) or opts.has_key('b'):
2560 2560 bkms = self.db.get('bookmarks', {})
2561 2561
2562 2562 if bkms.has_key(ps):
2563 2563 target = bkms[ps]
2564 2564 print '(bookmark:%s) -> %s' % (ps,target)
2565 2565 ps = target
2566 2566 else:
2567 2567 if opts.has_key('b'):
2568 2568 error("Bookmark '%s' not found. "
2569 2569 "Use '%%bookmark -l' to see your bookmarks." % ps)
2570 2570 return
2571 2571
2572 2572 # at this point ps should point to the target dir
2573 2573 if ps:
2574 2574 try:
2575 2575 os.chdir(os.path.expanduser(ps))
2576 2576 ttitle = ("IPy:" + (
2577 2577 os.getcwd() == '/' and '/' or os.path.basename(os.getcwd())))
2578 2578 platutils.set_term_title(ttitle)
2579 2579 except OSError:
2580 2580 print sys.exc_info()[1]
2581 2581 else:
2582 2582 self.shell.user_ns['_dh'].append(os.getcwd())
2583 2583 else:
2584 2584 os.chdir(self.shell.home_dir)
2585 2585 platutils.set_term_title("IPy:~")
2586 2586 self.shell.user_ns['_dh'].append(os.getcwd())
2587 2587 if not 'q' in opts:
2588 2588 print self.shell.user_ns['_dh'][-1]
2589 2589
2590 2590 def magic_dhist(self, parameter_s=''):
2591 2591 """Print your history of visited directories.
2592 2592
2593 2593 %dhist -> print full history\\
2594 2594 %dhist n -> print last n entries only\\
2595 2595 %dhist n1 n2 -> print entries between n1 and n2 (n1 not included)\\
2596 2596
2597 2597 This history is automatically maintained by the %cd command, and
2598 2598 always available as the global list variable _dh. You can use %cd -<n>
2599 2599 to go to directory number <n>."""
2600 2600
2601 2601 dh = self.shell.user_ns['_dh']
2602 2602 if parameter_s:
2603 2603 try:
2604 2604 args = map(int,parameter_s.split())
2605 2605 except:
2606 2606 self.arg_err(Magic.magic_dhist)
2607 2607 return
2608 2608 if len(args) == 1:
2609 2609 ini,fin = max(len(dh)-(args[0]),0),len(dh)
2610 2610 elif len(args) == 2:
2611 2611 ini,fin = args
2612 2612 else:
2613 2613 self.arg_err(Magic.magic_dhist)
2614 2614 return
2615 2615 else:
2616 2616 ini,fin = 0,len(dh)
2617 2617 nlprint(dh,
2618 2618 header = 'Directory history (kept in _dh)',
2619 2619 start=ini,stop=fin)
2620 2620
2621 2621 def magic_env(self, parameter_s=''):
2622 2622 """List environment variables."""
2623 2623
2624 2624 return os.environ.data
2625 2625
2626 2626 def magic_pushd(self, parameter_s=''):
2627 2627 """Place the current dir on stack and change directory.
2628 2628
2629 2629 Usage:\\
2630 2630 %pushd ['dirname']
2631 2631
2632 2632 %pushd with no arguments does a %pushd to your home directory.
2633 2633 """
2634 2634 if parameter_s == '': parameter_s = '~'
2635 2635 dir_s = self.shell.dir_stack
2636 2636 if len(dir_s)>0 and os.path.expanduser(parameter_s) != \
2637 2637 os.path.expanduser(self.shell.dir_stack[0]):
2638 2638 try:
2639 2639 self.magic_cd(parameter_s)
2640 2640 dir_s.insert(0,os.getcwd().replace(self.home_dir,'~'))
2641 2641 self.magic_dirs()
2642 2642 except:
2643 2643 print 'Invalid directory'
2644 2644 else:
2645 2645 print 'You are already there!'
2646 2646
2647 2647 def magic_popd(self, parameter_s=''):
2648 2648 """Change to directory popped off the top of the stack.
2649 2649 """
2650 2650 if len (self.shell.dir_stack) > 1:
2651 2651 self.shell.dir_stack.pop(0)
2652 2652 self.magic_cd(self.shell.dir_stack[0])
2653 2653 print self.shell.dir_stack[0]
2654 2654 else:
2655 2655 print "You can't remove the starting directory from the stack:",\
2656 2656 self.shell.dir_stack
2657 2657
2658 2658 def magic_dirs(self, parameter_s=''):
2659 2659 """Return the current directory stack."""
2660 2660
2661 2661 return self.shell.dir_stack[:]
2662 2662
2663 2663 def magic_sc(self, parameter_s=''):
2664 2664 """Shell capture - execute a shell command and capture its output.
2665 2665
2666 2666 DEPRECATED. Suboptimal, retained for backwards compatibility.
2667 2667
2668 2668 You should use the form 'var = !command' instead. Example:
2669 2669
2670 2670 "%sc -l myfiles = ls ~" should now be written as
2671 2671
2672 2672 "myfiles = !ls ~"
2673 2673
2674 2674 myfiles.s, myfiles.l and myfiles.n still apply as documented
2675 2675 below.
2676 2676
2677 2677 --
2678 2678 %sc [options] varname=command
2679 2679
2680 2680 IPython will run the given command using commands.getoutput(), and
2681 2681 will then update the user's interactive namespace with a variable
2682 2682 called varname, containing the value of the call. Your command can
2683 2683 contain shell wildcards, pipes, etc.
2684 2684
2685 2685 The '=' sign in the syntax is mandatory, and the variable name you
2686 2686 supply must follow Python's standard conventions for valid names.
2687 2687
2688 2688 (A special format without variable name exists for internal use)
2689 2689
2690 2690 Options:
2691 2691
2692 2692 -l: list output. Split the output on newlines into a list before
2693 2693 assigning it to the given variable. By default the output is stored
2694 2694 as a single string.
2695 2695
2696 2696 -v: verbose. Print the contents of the variable.
2697 2697
2698 2698 In most cases you should not need to split as a list, because the
2699 2699 returned value is a special type of string which can automatically
2700 2700 provide its contents either as a list (split on newlines) or as a
2701 2701 space-separated string. These are convenient, respectively, either
2702 2702 for sequential processing or to be passed to a shell command.
2703 2703
2704 2704 For example:
2705 2705
2706 2706 # Capture into variable a
2707 2707 In [9]: sc a=ls *py
2708 2708
2709 2709 # a is a string with embedded newlines
2710 2710 In [10]: a
2711 2711 Out[10]: 'setup.py\nwin32_manual_post_install.py'
2712 2712
2713 2713 # which can be seen as a list:
2714 2714 In [11]: a.l
2715 2715 Out[11]: ['setup.py', 'win32_manual_post_install.py']
2716 2716
2717 2717 # or as a whitespace-separated string:
2718 2718 In [12]: a.s
2719 2719 Out[12]: 'setup.py win32_manual_post_install.py'
2720 2720
2721 2721 # a.s is useful to pass as a single command line:
2722 2722 In [13]: !wc -l $a.s
2723 2723 146 setup.py
2724 2724 130 win32_manual_post_install.py
2725 2725 276 total
2726 2726
2727 2727 # while the list form is useful to loop over:
2728 2728 In [14]: for f in a.l:
2729 2729 ....: !wc -l $f
2730 2730 ....:
2731 2731 146 setup.py
2732 2732 130 win32_manual_post_install.py
2733 2733
2734 2734 Similiarly, the lists returned by the -l option are also special, in
2735 2735 the sense that you can equally invoke the .s attribute on them to
2736 2736 automatically get a whitespace-separated string from their contents:
2737 2737
2738 2738 In [1]: sc -l b=ls *py
2739 2739
2740 2740 In [2]: b
2741 2741 Out[2]: ['setup.py', 'win32_manual_post_install.py']
2742 2742
2743 2743 In [3]: b.s
2744 2744 Out[3]: 'setup.py win32_manual_post_install.py'
2745 2745
2746 2746 In summary, both the lists and strings used for ouptut capture have
2747 2747 the following special attributes:
2748 2748
2749 2749 .l (or .list) : value as list.
2750 2750 .n (or .nlstr): value as newline-separated string.
2751 2751 .s (or .spstr): value as space-separated string.
2752 2752 """
2753 2753
2754 2754 opts,args = self.parse_options(parameter_s,'lv')
2755 2755 # Try to get a variable name and command to run
2756 2756 try:
2757 2757 # the variable name must be obtained from the parse_options
2758 2758 # output, which uses shlex.split to strip options out.
2759 2759 var,_ = args.split('=',1)
2760 2760 var = var.strip()
2761 2761 # But the the command has to be extracted from the original input
2762 2762 # parameter_s, not on what parse_options returns, to avoid the
2763 2763 # quote stripping which shlex.split performs on it.
2764 2764 _,cmd = parameter_s.split('=',1)
2765 2765 except ValueError:
2766 2766 var,cmd = '',''
2767 2767 # If all looks ok, proceed
2768 2768 out,err = self.shell.getoutputerror(cmd)
2769 2769 if err:
2770 2770 print >> Term.cerr,err
2771 2771 if opts.has_key('l'):
2772 2772 out = SList(out.split('\n'))
2773 2773 else:
2774 2774 out = LSString(out)
2775 2775 if opts.has_key('v'):
2776 2776 print '%s ==\n%s' % (var,pformat(out))
2777 2777 if var:
2778 2778 self.shell.user_ns.update({var:out})
2779 2779 else:
2780 2780 return out
2781 2781
2782 2782 def magic_sx(self, parameter_s=''):
2783 2783 """Shell execute - run a shell command and capture its output.
2784 2784
2785 2785 %sx command
2786 2786
2787 2787 IPython will run the given command using commands.getoutput(), and
2788 2788 return the result formatted as a list (split on '\\n'). Since the
2789 2789 output is _returned_, it will be stored in ipython's regular output
2790 2790 cache Out[N] and in the '_N' automatic variables.
2791 2791
2792 2792 Notes:
2793 2793
2794 2794 1) If an input line begins with '!!', then %sx is automatically
2795 2795 invoked. That is, while:
2796 2796 !ls
2797 2797 causes ipython to simply issue system('ls'), typing
2798 2798 !!ls
2799 2799 is a shorthand equivalent to:
2800 2800 %sx ls
2801 2801
2802 2802 2) %sx differs from %sc in that %sx automatically splits into a list,
2803 2803 like '%sc -l'. The reason for this is to make it as easy as possible
2804 2804 to process line-oriented shell output via further python commands.
2805 2805 %sc is meant to provide much finer control, but requires more
2806 2806 typing.
2807 2807
2808 2808 3) Just like %sc -l, this is a list with special attributes:
2809 2809
2810 2810 .l (or .list) : value as list.
2811 2811 .n (or .nlstr): value as newline-separated string.
2812 2812 .s (or .spstr): value as whitespace-separated string.
2813 2813
2814 2814 This is very useful when trying to use such lists as arguments to
2815 2815 system commands."""
2816 2816
2817 2817 if parameter_s:
2818 2818 out,err = self.shell.getoutputerror(parameter_s)
2819 2819 if err:
2820 2820 print >> Term.cerr,err
2821 2821 return SList(out.split('\n'))
2822 2822
2823 2823 def magic_bg(self, parameter_s=''):
2824 2824 """Run a job in the background, in a separate thread.
2825 2825
2826 2826 For example,
2827 2827
2828 2828 %bg myfunc(x,y,z=1)
2829 2829
2830 2830 will execute 'myfunc(x,y,z=1)' in a background thread. As soon as the
2831 2831 execution starts, a message will be printed indicating the job
2832 2832 number. If your job number is 5, you can use
2833 2833
2834 2834 myvar = jobs.result(5) or myvar = jobs[5].result
2835 2835
2836 2836 to assign this result to variable 'myvar'.
2837 2837
2838 2838 IPython has a job manager, accessible via the 'jobs' object. You can
2839 2839 type jobs? to get more information about it, and use jobs.<TAB> to see
2840 2840 its attributes. All attributes not starting with an underscore are
2841 2841 meant for public use.
2842 2842
2843 2843 In particular, look at the jobs.new() method, which is used to create
2844 2844 new jobs. This magic %bg function is just a convenience wrapper
2845 2845 around jobs.new(), for expression-based jobs. If you want to create a
2846 2846 new job with an explicit function object and arguments, you must call
2847 2847 jobs.new() directly.
2848 2848
2849 2849 The jobs.new docstring also describes in detail several important
2850 2850 caveats associated with a thread-based model for background job
2851 2851 execution. Type jobs.new? for details.
2852 2852
2853 2853 You can check the status of all jobs with jobs.status().
2854 2854
2855 2855 The jobs variable is set by IPython into the Python builtin namespace.
2856 2856 If you ever declare a variable named 'jobs', you will shadow this
2857 2857 name. You can either delete your global jobs variable to regain
2858 2858 access to the job manager, or make a new name and assign it manually
2859 2859 to the manager (stored in IPython's namespace). For example, to
2860 2860 assign the job manager to the Jobs name, use:
2861 2861
2862 2862 Jobs = __builtins__.jobs"""
2863 2863
2864 2864 self.shell.jobs.new(parameter_s,self.shell.user_ns)
2865 2865
2866 2866
2867 2867 def magic_bookmark(self, parameter_s=''):
2868 2868 """Manage IPython's bookmark system.
2869 2869
2870 2870 %bookmark <name> - set bookmark to current dir
2871 2871 %bookmark <name> <dir> - set bookmark to <dir>
2872 2872 %bookmark -l - list all bookmarks
2873 2873 %bookmark -d <name> - remove bookmark
2874 2874 %bookmark -r - remove all bookmarks
2875 2875
2876 2876 You can later on access a bookmarked folder with:
2877 2877 %cd -b <name>
2878 2878 or simply '%cd <name>' if there is no directory called <name> AND
2879 2879 there is such a bookmark defined.
2880 2880
2881 2881 Your bookmarks persist through IPython sessions, but they are
2882 2882 associated with each profile."""
2883 2883
2884 2884 opts,args = self.parse_options(parameter_s,'drl',mode='list')
2885 2885 if len(args) > 2:
2886 2886 error('You can only give at most two arguments')
2887 2887 return
2888 2888
2889 2889 bkms = self.db.get('bookmarks',{})
2890 2890
2891 2891 if opts.has_key('d'):
2892 2892 try:
2893 2893 todel = args[0]
2894 2894 except IndexError:
2895 2895 error('You must provide a bookmark to delete')
2896 2896 else:
2897 2897 try:
2898 2898 del bkms[todel]
2899 2899 except:
2900 2900 error("Can't delete bookmark '%s'" % todel)
2901 2901 elif opts.has_key('r'):
2902 2902 bkms = {}
2903 2903 elif opts.has_key('l'):
2904 2904 bks = bkms.keys()
2905 2905 bks.sort()
2906 2906 if bks:
2907 2907 size = max(map(len,bks))
2908 2908 else:
2909 2909 size = 0
2910 2910 fmt = '%-'+str(size)+'s -> %s'
2911 2911 print 'Current bookmarks:'
2912 2912 for bk in bks:
2913 2913 print fmt % (bk,bkms[bk])
2914 2914 else:
2915 2915 if not args:
2916 2916 error("You must specify the bookmark name")
2917 2917 elif len(args)==1:
2918 2918 bkms[args[0]] = os.getcwd()
2919 2919 elif len(args)==2:
2920 2920 bkms[args[0]] = args[1]
2921 2921 self.db['bookmarks'] = bkms
2922 2922
2923 2923 def magic_pycat(self, parameter_s=''):
2924 2924 """Show a syntax-highlighted file through a pager.
2925 2925
2926 2926 This magic is similar to the cat utility, but it will assume the file
2927 2927 to be Python source and will show it with syntax highlighting. """
2928 2928
2929 2929 try:
2930 2930 filename = get_py_filename(parameter_s)
2931 2931 cont = file_read(filename)
2932 2932 except IOError:
2933 2933 try:
2934 2934 cont = eval(parameter_s,self.user_ns)
2935 2935 except NameError:
2936 2936 cont = None
2937 2937 if cont is None:
2938 2938 print "Error: no such file or variable"
2939 2939 return
2940 2940
2941 2941 page(self.shell.pycolorize(cont),
2942 2942 screen_lines=self.shell.rc.screen_length)
2943 2943
2944 2944 def magic_cpaste(self, parameter_s=''):
2945 2945 """Allows you to paste & execute a pre-formatted code block from clipboard
2946 2946
2947 2947 You must terminate the block with '--' (two minus-signs) alone on the
2948 2948 line. You can also provide your own sentinel with '%paste -s %%' ('%%'
2949 2949 is the new sentinel for this operation)
2950 2950
2951 2951 The block is dedented prior to execution to enable execution of
2952 2952 method definitions. '>' characters at the beginning of a line is
2953 2953 ignored, to allow pasting directly from e-mails. The executed block
2954 2954 is also assigned to variable named 'pasted_block' for later editing
2955 2955 with '%edit pasted_block'.
2956 2956
2957 2957 You can also pass a variable name as an argument, e.g. '%cpaste foo'.
2958 2958 This assigns the pasted block to variable 'foo' as string, without
2959 2959 dedenting or executing it.
2960 2960
2961 2961 Do not be alarmed by garbled output on Windows (it's a readline bug).
2962 2962 Just press enter and type -- (and press enter again) and the block
2963 2963 will be what was just pasted.
2964 2964
2965 2965 IPython statements (magics, shell escapes) are not supported (yet).
2966 2966 """
2967 2967 opts,args = self.parse_options(parameter_s,'s:',mode='string')
2968 2968 par = args.strip()
2969 2969 sentinel = opts.get('s','--')
2970 2970
2971 2971 from IPython import iplib
2972 2972 lines = []
2973 2973 print "Pasting code; enter '%s' alone on the line to stop." % sentinel
2974 2974 while 1:
2975 2975 l = iplib.raw_input_original(':')
2976 2976 if l ==sentinel:
2977 2977 break
2978 2978 lines.append(l.lstrip('>'))
2979 2979 block = "\n".join(lines) + '\n'
2980 2980 #print "block:\n",block
2981 2981 if not par:
2982 2982 b = textwrap.dedent(block)
2983 2983 exec b in self.user_ns
2984 2984 self.user_ns['pasted_block'] = b
2985 2985 else:
2986 2986 self.user_ns[par] = block
2987 2987 print "Block assigned to '%s'" % par
2988 2988
2989 2989 def magic_quickref(self,arg):
2990 2990 """ Show a quick reference sheet """
2991 2991 import IPython.usage
2992 2992 qr = IPython.usage.quick_reference + self.magic_magic('-brief')
2993 2993
2994 2994 page(qr)
2995 2995
2996 2996 def magic_upgrade(self,arg):
2997 2997 """ Upgrade your IPython installation
2998 2998
2999 2999 This will copy the config files that don't yet exist in your
3000 3000 ipython dir from the system config dir. Use this after upgrading
3001 3001 IPython if you don't wish to delete your .ipython dir.
3002 3002
3003 3003 Call with -nolegacy to get rid of ipythonrc* files (recommended for
3004 3004 new users)
3005 3005
3006 3006 """
3007 3007 ip = self.getapi()
3008 3008 ipinstallation = path(IPython.__file__).dirname()
3009 3009 upgrade_script = '%s "%s"' % (sys.executable,ipinstallation / 'upgrade_dir.py')
3010 3010 src_config = ipinstallation / 'UserConfig'
3011 3011 userdir = path(ip.options.ipythondir)
3012 3012 cmd = '%s "%s" "%s"' % (upgrade_script, src_config, userdir)
3013 3013 print ">",cmd
3014 3014 shell(cmd)
3015 3015 if arg == '-nolegacy':
3016 3016 legacy = userdir.files('ipythonrc*')
3017 3017 print "Nuking legacy files:",legacy
3018 3018
3019 3019 [p.remove() for p in legacy]
3020 3020 suffix = (sys.platform == 'win32' and '.ini' or '')
3021 3021 (userdir / ('ipythonrc' + suffix)).write_text('# Empty, see ipy_user_conf.py\n')
3022 3022
3023 3023
3024 3024 # end Magic
General Comments 0
You need to be logged in to leave comments. Login now