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