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