##// END OF EJS Templates
Merge pull request #11377 from kd2718/magic_run_documentation...
Matthias Bussonnier -
r24693:06023599 merge
parent child Browse files
Show More
@@ -1,1424 +1,1427
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2 """Implementation of execution-related magic functions."""
2 """Implementation of execution-related magic functions."""
3
3
4 # Copyright (c) IPython Development Team.
4 # Copyright (c) IPython Development Team.
5 # Distributed under the terms of the Modified BSD License.
5 # Distributed under the terms of the Modified BSD License.
6
6
7
7
8 import ast
8 import ast
9 import bdb
9 import bdb
10 import builtins as builtin_mod
10 import builtins as builtin_mod
11 import gc
11 import gc
12 import itertools
12 import itertools
13 import os
13 import os
14 import shlex
14 import shlex
15 import sys
15 import sys
16 import time
16 import time
17 import timeit
17 import timeit
18 import math
18 import math
19 from pdb import Restart
19 from pdb import Restart
20
20
21 # cProfile was added in Python2.5
21 # cProfile was added in Python2.5
22 try:
22 try:
23 import cProfile as profile
23 import cProfile as profile
24 import pstats
24 import pstats
25 except ImportError:
25 except ImportError:
26 # profile isn't bundled by default in Debian for license reasons
26 # profile isn't bundled by default in Debian for license reasons
27 try:
27 try:
28 import profile, pstats
28 import profile, pstats
29 except ImportError:
29 except ImportError:
30 profile = pstats = None
30 profile = pstats = None
31
31
32 from IPython.core import oinspect
32 from IPython.core import oinspect
33 from IPython.core import magic_arguments
33 from IPython.core import magic_arguments
34 from IPython.core import page
34 from IPython.core import page
35 from IPython.core.error import UsageError
35 from IPython.core.error import UsageError
36 from IPython.core.macro import Macro
36 from IPython.core.macro import Macro
37 from IPython.core.magic import (Magics, magics_class, line_magic, cell_magic,
37 from IPython.core.magic import (Magics, magics_class, line_magic, cell_magic,
38 line_cell_magic, on_off, needs_local_scope)
38 line_cell_magic, on_off, needs_local_scope)
39 from IPython.testing.skipdoctest import skip_doctest
39 from IPython.testing.skipdoctest import skip_doctest
40 from IPython.utils.contexts import preserve_keys
40 from IPython.utils.contexts import preserve_keys
41 from IPython.utils.capture import capture_output
41 from IPython.utils.capture import capture_output
42 from IPython.utils.ipstruct import Struct
42 from IPython.utils.ipstruct import Struct
43 from IPython.utils.module_paths import find_mod
43 from IPython.utils.module_paths import find_mod
44 from IPython.utils.path import get_py_filename, shellglob
44 from IPython.utils.path import get_py_filename, shellglob
45 from IPython.utils.timing import clock, clock2
45 from IPython.utils.timing import clock, clock2
46 from warnings import warn
46 from warnings import warn
47 from logging import error
47 from logging import error
48 from io import StringIO
48 from io import StringIO
49
49
50
50
51 #-----------------------------------------------------------------------------
51 #-----------------------------------------------------------------------------
52 # Magic implementation classes
52 # Magic implementation classes
53 #-----------------------------------------------------------------------------
53 #-----------------------------------------------------------------------------
54
54
55
55
56 class TimeitResult(object):
56 class TimeitResult(object):
57 """
57 """
58 Object returned by the timeit magic with info about the run.
58 Object returned by the timeit magic with info about the run.
59
59
60 Contains the following attributes :
60 Contains the following attributes :
61
61
62 loops: (int) number of loops done per measurement
62 loops: (int) number of loops done per measurement
63 repeat: (int) number of times the measurement has been repeated
63 repeat: (int) number of times the measurement has been repeated
64 best: (float) best execution time / number
64 best: (float) best execution time / number
65 all_runs: (list of float) execution time of each run (in s)
65 all_runs: (list of float) execution time of each run (in s)
66 compile_time: (float) time of statement compilation (s)
66 compile_time: (float) time of statement compilation (s)
67
67
68 """
68 """
69 def __init__(self, loops, repeat, best, worst, all_runs, compile_time, precision):
69 def __init__(self, loops, repeat, best, worst, all_runs, compile_time, precision):
70 self.loops = loops
70 self.loops = loops
71 self.repeat = repeat
71 self.repeat = repeat
72 self.best = best
72 self.best = best
73 self.worst = worst
73 self.worst = worst
74 self.all_runs = all_runs
74 self.all_runs = all_runs
75 self.compile_time = compile_time
75 self.compile_time = compile_time
76 self._precision = precision
76 self._precision = precision
77 self.timings = [ dt / self.loops for dt in all_runs]
77 self.timings = [ dt / self.loops for dt in all_runs]
78
78
79 @property
79 @property
80 def average(self):
80 def average(self):
81 return math.fsum(self.timings) / len(self.timings)
81 return math.fsum(self.timings) / len(self.timings)
82
82
83 @property
83 @property
84 def stdev(self):
84 def stdev(self):
85 mean = self.average
85 mean = self.average
86 return (math.fsum([(x - mean) ** 2 for x in self.timings]) / len(self.timings)) ** 0.5
86 return (math.fsum([(x - mean) ** 2 for x in self.timings]) / len(self.timings)) ** 0.5
87
87
88 def __str__(self):
88 def __str__(self):
89 pm = '+-'
89 pm = '+-'
90 if hasattr(sys.stdout, 'encoding') and sys.stdout.encoding:
90 if hasattr(sys.stdout, 'encoding') and sys.stdout.encoding:
91 try:
91 try:
92 u'\xb1'.encode(sys.stdout.encoding)
92 u'\xb1'.encode(sys.stdout.encoding)
93 pm = u'\xb1'
93 pm = u'\xb1'
94 except:
94 except:
95 pass
95 pass
96 return (
96 return (
97 u"{mean} {pm} {std} per loop (mean {pm} std. dev. of {runs} run{run_plural}, {loops} loop{loop_plural} each)"
97 u"{mean} {pm} {std} per loop (mean {pm} std. dev. of {runs} run{run_plural}, {loops} loop{loop_plural} each)"
98 .format(
98 .format(
99 pm = pm,
99 pm = pm,
100 runs = self.repeat,
100 runs = self.repeat,
101 loops = self.loops,
101 loops = self.loops,
102 loop_plural = "" if self.loops == 1 else "s",
102 loop_plural = "" if self.loops == 1 else "s",
103 run_plural = "" if self.repeat == 1 else "s",
103 run_plural = "" if self.repeat == 1 else "s",
104 mean = _format_time(self.average, self._precision),
104 mean = _format_time(self.average, self._precision),
105 std = _format_time(self.stdev, self._precision))
105 std = _format_time(self.stdev, self._precision))
106 )
106 )
107
107
108 def _repr_pretty_(self, p , cycle):
108 def _repr_pretty_(self, p , cycle):
109 unic = self.__str__()
109 unic = self.__str__()
110 p.text(u'<TimeitResult : '+unic+u'>')
110 p.text(u'<TimeitResult : '+unic+u'>')
111
111
112
112
113 class TimeitTemplateFiller(ast.NodeTransformer):
113 class TimeitTemplateFiller(ast.NodeTransformer):
114 """Fill in the AST template for timing execution.
114 """Fill in the AST template for timing execution.
115
115
116 This is quite closely tied to the template definition, which is in
116 This is quite closely tied to the template definition, which is in
117 :meth:`ExecutionMagics.timeit`.
117 :meth:`ExecutionMagics.timeit`.
118 """
118 """
119 def __init__(self, ast_setup, ast_stmt):
119 def __init__(self, ast_setup, ast_stmt):
120 self.ast_setup = ast_setup
120 self.ast_setup = ast_setup
121 self.ast_stmt = ast_stmt
121 self.ast_stmt = ast_stmt
122
122
123 def visit_FunctionDef(self, node):
123 def visit_FunctionDef(self, node):
124 "Fill in the setup statement"
124 "Fill in the setup statement"
125 self.generic_visit(node)
125 self.generic_visit(node)
126 if node.name == "inner":
126 if node.name == "inner":
127 node.body[:1] = self.ast_setup.body
127 node.body[:1] = self.ast_setup.body
128
128
129 return node
129 return node
130
130
131 def visit_For(self, node):
131 def visit_For(self, node):
132 "Fill in the statement to be timed"
132 "Fill in the statement to be timed"
133 if getattr(getattr(node.body[0], 'value', None), 'id', None) == 'stmt':
133 if getattr(getattr(node.body[0], 'value', None), 'id', None) == 'stmt':
134 node.body = self.ast_stmt.body
134 node.body = self.ast_stmt.body
135 return node
135 return node
136
136
137
137
138 class Timer(timeit.Timer):
138 class Timer(timeit.Timer):
139 """Timer class that explicitly uses self.inner
139 """Timer class that explicitly uses self.inner
140
140
141 which is an undocumented implementation detail of CPython,
141 which is an undocumented implementation detail of CPython,
142 not shared by PyPy.
142 not shared by PyPy.
143 """
143 """
144 # Timer.timeit copied from CPython 3.4.2
144 # Timer.timeit copied from CPython 3.4.2
145 def timeit(self, number=timeit.default_number):
145 def timeit(self, number=timeit.default_number):
146 """Time 'number' executions of the main statement.
146 """Time 'number' executions of the main statement.
147
147
148 To be precise, this executes the setup statement once, and
148 To be precise, this executes the setup statement once, and
149 then returns the time it takes to execute the main statement
149 then returns the time it takes to execute the main statement
150 a number of times, as a float measured in seconds. The
150 a number of times, as a float measured in seconds. The
151 argument is the number of times through the loop, defaulting
151 argument is the number of times through the loop, defaulting
152 to one million. The main statement, the setup statement and
152 to one million. The main statement, the setup statement and
153 the timer function to be used are passed to the constructor.
153 the timer function to be used are passed to the constructor.
154 """
154 """
155 it = itertools.repeat(None, number)
155 it = itertools.repeat(None, number)
156 gcold = gc.isenabled()
156 gcold = gc.isenabled()
157 gc.disable()
157 gc.disable()
158 try:
158 try:
159 timing = self.inner(it, self.timer)
159 timing = self.inner(it, self.timer)
160 finally:
160 finally:
161 if gcold:
161 if gcold:
162 gc.enable()
162 gc.enable()
163 return timing
163 return timing
164
164
165
165
166 @magics_class
166 @magics_class
167 class ExecutionMagics(Magics):
167 class ExecutionMagics(Magics):
168 """Magics related to code execution, debugging, profiling, etc.
168 """Magics related to code execution, debugging, profiling, etc.
169
169
170 """
170 """
171
171
172 def __init__(self, shell):
172 def __init__(self, shell):
173 super(ExecutionMagics, self).__init__(shell)
173 super(ExecutionMagics, self).__init__(shell)
174 if profile is None:
174 if profile is None:
175 self.prun = self.profile_missing_notice
175 self.prun = self.profile_missing_notice
176 # Default execution function used to actually run user code.
176 # Default execution function used to actually run user code.
177 self.default_runner = None
177 self.default_runner = None
178
178
179 def profile_missing_notice(self, *args, **kwargs):
179 def profile_missing_notice(self, *args, **kwargs):
180 error("""\
180 error("""\
181 The profile module could not be found. It has been removed from the standard
181 The profile module could not be found. It has been removed from the standard
182 python packages because of its non-free license. To use profiling, install the
182 python packages because of its non-free license. To use profiling, install the
183 python-profiler package from non-free.""")
183 python-profiler package from non-free.""")
184
184
185 @skip_doctest
185 @skip_doctest
186 @line_cell_magic
186 @line_cell_magic
187 def prun(self, parameter_s='', cell=None):
187 def prun(self, parameter_s='', cell=None):
188
188
189 """Run a statement through the python code profiler.
189 """Run a statement through the python code profiler.
190
190
191 Usage, in line mode:
191 Usage, in line mode:
192 %prun [options] statement
192 %prun [options] statement
193
193
194 Usage, in cell mode:
194 Usage, in cell mode:
195 %%prun [options] [statement]
195 %%prun [options] [statement]
196 code...
196 code...
197 code...
197 code...
198
198
199 In cell mode, the additional code lines are appended to the (possibly
199 In cell mode, the additional code lines are appended to the (possibly
200 empty) statement in the first line. Cell mode allows you to easily
200 empty) statement in the first line. Cell mode allows you to easily
201 profile multiline blocks without having to put them in a separate
201 profile multiline blocks without having to put them in a separate
202 function.
202 function.
203
203
204 The given statement (which doesn't require quote marks) is run via the
204 The given statement (which doesn't require quote marks) is run via the
205 python profiler in a manner similar to the profile.run() function.
205 python profiler in a manner similar to the profile.run() function.
206 Namespaces are internally managed to work correctly; profile.run
206 Namespaces are internally managed to work correctly; profile.run
207 cannot be used in IPython because it makes certain assumptions about
207 cannot be used in IPython because it makes certain assumptions about
208 namespaces which do not hold under IPython.
208 namespaces which do not hold under IPython.
209
209
210 Options:
210 Options:
211
211
212 -l <limit>
212 -l <limit>
213 you can place restrictions on what or how much of the
213 you can place restrictions on what or how much of the
214 profile gets printed. The limit value can be:
214 profile gets printed. The limit value can be:
215
215
216 * A string: only information for function names containing this string
216 * A string: only information for function names containing this string
217 is printed.
217 is printed.
218
218
219 * An integer: only these many lines are printed.
219 * An integer: only these many lines are printed.
220
220
221 * A float (between 0 and 1): this fraction of the report is printed
221 * A float (between 0 and 1): this fraction of the report is printed
222 (for example, use a limit of 0.4 to see the topmost 40% only).
222 (for example, use a limit of 0.4 to see the topmost 40% only).
223
223
224 You can combine several limits with repeated use of the option. For
224 You can combine several limits with repeated use of the option. For
225 example, ``-l __init__ -l 5`` will print only the topmost 5 lines of
225 example, ``-l __init__ -l 5`` will print only the topmost 5 lines of
226 information about class constructors.
226 information about class constructors.
227
227
228 -r
228 -r
229 return the pstats.Stats object generated by the profiling. This
229 return the pstats.Stats object generated by the profiling. This
230 object has all the information about the profile in it, and you can
230 object has all the information about the profile in it, and you can
231 later use it for further analysis or in other functions.
231 later use it for further analysis or in other functions.
232
232
233 -s <key>
233 -s <key>
234 sort profile by given key. You can provide more than one key
234 sort profile by given key. You can provide more than one key
235 by using the option several times: '-s key1 -s key2 -s key3...'. The
235 by using the option several times: '-s key1 -s key2 -s key3...'. The
236 default sorting key is 'time'.
236 default sorting key is 'time'.
237
237
238 The following is copied verbatim from the profile documentation
238 The following is copied verbatim from the profile documentation
239 referenced below:
239 referenced below:
240
240
241 When more than one key is provided, additional keys are used as
241 When more than one key is provided, additional keys are used as
242 secondary criteria when the there is equality in all keys selected
242 secondary criteria when the there is equality in all keys selected
243 before them.
243 before them.
244
244
245 Abbreviations can be used for any key names, as long as the
245 Abbreviations can be used for any key names, as long as the
246 abbreviation is unambiguous. The following are the keys currently
246 abbreviation is unambiguous. The following are the keys currently
247 defined:
247 defined:
248
248
249 ============ =====================
249 ============ =====================
250 Valid Arg Meaning
250 Valid Arg Meaning
251 ============ =====================
251 ============ =====================
252 "calls" call count
252 "calls" call count
253 "cumulative" cumulative time
253 "cumulative" cumulative time
254 "file" file name
254 "file" file name
255 "module" file name
255 "module" file name
256 "pcalls" primitive call count
256 "pcalls" primitive call count
257 "line" line number
257 "line" line number
258 "name" function name
258 "name" function name
259 "nfl" name/file/line
259 "nfl" name/file/line
260 "stdname" standard name
260 "stdname" standard name
261 "time" internal time
261 "time" internal time
262 ============ =====================
262 ============ =====================
263
263
264 Note that all sorts on statistics are in descending order (placing
264 Note that all sorts on statistics are in descending order (placing
265 most time consuming items first), where as name, file, and line number
265 most time consuming items first), where as name, file, and line number
266 searches are in ascending order (i.e., alphabetical). The subtle
266 searches are in ascending order (i.e., alphabetical). The subtle
267 distinction between "nfl" and "stdname" is that the standard name is a
267 distinction between "nfl" and "stdname" is that the standard name is a
268 sort of the name as printed, which means that the embedded line
268 sort of the name as printed, which means that the embedded line
269 numbers get compared in an odd way. For example, lines 3, 20, and 40
269 numbers get compared in an odd way. For example, lines 3, 20, and 40
270 would (if the file names were the same) appear in the string order
270 would (if the file names were the same) appear in the string order
271 "20" "3" and "40". In contrast, "nfl" does a numeric compare of the
271 "20" "3" and "40". In contrast, "nfl" does a numeric compare of the
272 line numbers. In fact, sort_stats("nfl") is the same as
272 line numbers. In fact, sort_stats("nfl") is the same as
273 sort_stats("name", "file", "line").
273 sort_stats("name", "file", "line").
274
274
275 -T <filename>
275 -T <filename>
276 save profile results as shown on screen to a text
276 save profile results as shown on screen to a text
277 file. The profile is still shown on screen.
277 file. The profile is still shown on screen.
278
278
279 -D <filename>
279 -D <filename>
280 save (via dump_stats) profile statistics to given
280 save (via dump_stats) profile statistics to given
281 filename. This data is in a format understood by the pstats module, and
281 filename. This data is in a format understood by the pstats module, and
282 is generated by a call to the dump_stats() method of profile
282 is generated by a call to the dump_stats() method of profile
283 objects. The profile is still shown on screen.
283 objects. The profile is still shown on screen.
284
284
285 -q
285 -q
286 suppress output to the pager. Best used with -T and/or -D above.
286 suppress output to the pager. Best used with -T and/or -D above.
287
287
288 If you want to run complete programs under the profiler's control, use
288 If you want to run complete programs under the profiler's control, use
289 ``%run -p [prof_opts] filename.py [args to program]`` where prof_opts
289 ``%run -p [prof_opts] filename.py [args to program]`` where prof_opts
290 contains profiler specific options as described here.
290 contains profiler specific options as described here.
291
291
292 You can read the complete documentation for the profile module with::
292 You can read the complete documentation for the profile module with::
293
293
294 In [1]: import profile; profile.help()
294 In [1]: import profile; profile.help()
295 """
295 """
296 opts, arg_str = self.parse_options(parameter_s, 'D:l:rs:T:q',
296 opts, arg_str = self.parse_options(parameter_s, 'D:l:rs:T:q',
297 list_all=True, posix=False)
297 list_all=True, posix=False)
298 if cell is not None:
298 if cell is not None:
299 arg_str += '\n' + cell
299 arg_str += '\n' + cell
300 arg_str = self.shell.transform_cell(arg_str)
300 arg_str = self.shell.transform_cell(arg_str)
301 return self._run_with_profiler(arg_str, opts, self.shell.user_ns)
301 return self._run_with_profiler(arg_str, opts, self.shell.user_ns)
302
302
303 def _run_with_profiler(self, code, opts, namespace):
303 def _run_with_profiler(self, code, opts, namespace):
304 """
304 """
305 Run `code` with profiler. Used by ``%prun`` and ``%run -p``.
305 Run `code` with profiler. Used by ``%prun`` and ``%run -p``.
306
306
307 Parameters
307 Parameters
308 ----------
308 ----------
309 code : str
309 code : str
310 Code to be executed.
310 Code to be executed.
311 opts : Struct
311 opts : Struct
312 Options parsed by `self.parse_options`.
312 Options parsed by `self.parse_options`.
313 namespace : dict
313 namespace : dict
314 A dictionary for Python namespace (e.g., `self.shell.user_ns`).
314 A dictionary for Python namespace (e.g., `self.shell.user_ns`).
315
315
316 """
316 """
317
317
318 # Fill default values for unspecified options:
318 # Fill default values for unspecified options:
319 opts.merge(Struct(D=[''], l=[], s=['time'], T=['']))
319 opts.merge(Struct(D=[''], l=[], s=['time'], T=['']))
320
320
321 prof = profile.Profile()
321 prof = profile.Profile()
322 try:
322 try:
323 prof = prof.runctx(code, namespace, namespace)
323 prof = prof.runctx(code, namespace, namespace)
324 sys_exit = ''
324 sys_exit = ''
325 except SystemExit:
325 except SystemExit:
326 sys_exit = """*** SystemExit exception caught in code being profiled."""
326 sys_exit = """*** SystemExit exception caught in code being profiled."""
327
327
328 stats = pstats.Stats(prof).strip_dirs().sort_stats(*opts.s)
328 stats = pstats.Stats(prof).strip_dirs().sort_stats(*opts.s)
329
329
330 lims = opts.l
330 lims = opts.l
331 if lims:
331 if lims:
332 lims = [] # rebuild lims with ints/floats/strings
332 lims = [] # rebuild lims with ints/floats/strings
333 for lim in opts.l:
333 for lim in opts.l:
334 try:
334 try:
335 lims.append(int(lim))
335 lims.append(int(lim))
336 except ValueError:
336 except ValueError:
337 try:
337 try:
338 lims.append(float(lim))
338 lims.append(float(lim))
339 except ValueError:
339 except ValueError:
340 lims.append(lim)
340 lims.append(lim)
341
341
342 # Trap output.
342 # Trap output.
343 stdout_trap = StringIO()
343 stdout_trap = StringIO()
344 stats_stream = stats.stream
344 stats_stream = stats.stream
345 try:
345 try:
346 stats.stream = stdout_trap
346 stats.stream = stdout_trap
347 stats.print_stats(*lims)
347 stats.print_stats(*lims)
348 finally:
348 finally:
349 stats.stream = stats_stream
349 stats.stream = stats_stream
350
350
351 output = stdout_trap.getvalue()
351 output = stdout_trap.getvalue()
352 output = output.rstrip()
352 output = output.rstrip()
353
353
354 if 'q' not in opts:
354 if 'q' not in opts:
355 page.page(output)
355 page.page(output)
356 print(sys_exit, end=' ')
356 print(sys_exit, end=' ')
357
357
358 dump_file = opts.D[0]
358 dump_file = opts.D[0]
359 text_file = opts.T[0]
359 text_file = opts.T[0]
360 if dump_file:
360 if dump_file:
361 prof.dump_stats(dump_file)
361 prof.dump_stats(dump_file)
362 print('\n*** Profile stats marshalled to file',\
362 print('\n*** Profile stats marshalled to file',\
363 repr(dump_file)+'.',sys_exit)
363 repr(dump_file)+'.',sys_exit)
364 if text_file:
364 if text_file:
365 pfile = open(text_file,'w')
365 pfile = open(text_file,'w')
366 pfile.write(output)
366 pfile.write(output)
367 pfile.close()
367 pfile.close()
368 print('\n*** Profile printout saved to text file',\
368 print('\n*** Profile printout saved to text file',\
369 repr(text_file)+'.',sys_exit)
369 repr(text_file)+'.',sys_exit)
370
370
371 if 'r' in opts:
371 if 'r' in opts:
372 return stats
372 return stats
373 else:
373 else:
374 return None
374 return None
375
375
376 @line_magic
376 @line_magic
377 def pdb(self, parameter_s=''):
377 def pdb(self, parameter_s=''):
378 """Control the automatic calling of the pdb interactive debugger.
378 """Control the automatic calling of the pdb interactive debugger.
379
379
380 Call as '%pdb on', '%pdb 1', '%pdb off' or '%pdb 0'. If called without
380 Call as '%pdb on', '%pdb 1', '%pdb off' or '%pdb 0'. If called without
381 argument it works as a toggle.
381 argument it works as a toggle.
382
382
383 When an exception is triggered, IPython can optionally call the
383 When an exception is triggered, IPython can optionally call the
384 interactive pdb debugger after the traceback printout. %pdb toggles
384 interactive pdb debugger after the traceback printout. %pdb toggles
385 this feature on and off.
385 this feature on and off.
386
386
387 The initial state of this feature is set in your configuration
387 The initial state of this feature is set in your configuration
388 file (the option is ``InteractiveShell.pdb``).
388 file (the option is ``InteractiveShell.pdb``).
389
389
390 If you want to just activate the debugger AFTER an exception has fired,
390 If you want to just activate the debugger AFTER an exception has fired,
391 without having to type '%pdb on' and rerunning your code, you can use
391 without having to type '%pdb on' and rerunning your code, you can use
392 the %debug magic."""
392 the %debug magic."""
393
393
394 par = parameter_s.strip().lower()
394 par = parameter_s.strip().lower()
395
395
396 if par:
396 if par:
397 try:
397 try:
398 new_pdb = {'off':0,'0':0,'on':1,'1':1}[par]
398 new_pdb = {'off':0,'0':0,'on':1,'1':1}[par]
399 except KeyError:
399 except KeyError:
400 print ('Incorrect argument. Use on/1, off/0, '
400 print ('Incorrect argument. Use on/1, off/0, '
401 'or nothing for a toggle.')
401 'or nothing for a toggle.')
402 return
402 return
403 else:
403 else:
404 # toggle
404 # toggle
405 new_pdb = not self.shell.call_pdb
405 new_pdb = not self.shell.call_pdb
406
406
407 # set on the shell
407 # set on the shell
408 self.shell.call_pdb = new_pdb
408 self.shell.call_pdb = new_pdb
409 print('Automatic pdb calling has been turned',on_off(new_pdb))
409 print('Automatic pdb calling has been turned',on_off(new_pdb))
410
410
411 @skip_doctest
411 @skip_doctest
412 @magic_arguments.magic_arguments()
412 @magic_arguments.magic_arguments()
413 @magic_arguments.argument('--breakpoint', '-b', metavar='FILE:LINE',
413 @magic_arguments.argument('--breakpoint', '-b', metavar='FILE:LINE',
414 help="""
414 help="""
415 Set break point at LINE in FILE.
415 Set break point at LINE in FILE.
416 """
416 """
417 )
417 )
418 @magic_arguments.argument('statement', nargs='*',
418 @magic_arguments.argument('statement', nargs='*',
419 help="""
419 help="""
420 Code to run in debugger.
420 Code to run in debugger.
421 You can omit this in cell magic mode.
421 You can omit this in cell magic mode.
422 """
422 """
423 )
423 )
424 @line_cell_magic
424 @line_cell_magic
425 def debug(self, line='', cell=None):
425 def debug(self, line='', cell=None):
426 """Activate the interactive debugger.
426 """Activate the interactive debugger.
427
427
428 This magic command support two ways of activating debugger.
428 This magic command support two ways of activating debugger.
429 One is to activate debugger before executing code. This way, you
429 One is to activate debugger before executing code. This way, you
430 can set a break point, to step through the code from the point.
430 can set a break point, to step through the code from the point.
431 You can use this mode by giving statements to execute and optionally
431 You can use this mode by giving statements to execute and optionally
432 a breakpoint.
432 a breakpoint.
433
433
434 The other one is to activate debugger in post-mortem mode. You can
434 The other one is to activate debugger in post-mortem mode. You can
435 activate this mode simply running %debug without any argument.
435 activate this mode simply running %debug without any argument.
436 If an exception has just occurred, this lets you inspect its stack
436 If an exception has just occurred, this lets you inspect its stack
437 frames interactively. Note that this will always work only on the last
437 frames interactively. Note that this will always work only on the last
438 traceback that occurred, so you must call this quickly after an
438 traceback that occurred, so you must call this quickly after an
439 exception that you wish to inspect has fired, because if another one
439 exception that you wish to inspect has fired, because if another one
440 occurs, it clobbers the previous one.
440 occurs, it clobbers the previous one.
441
441
442 If you want IPython to automatically do this on every exception, see
442 If you want IPython to automatically do this on every exception, see
443 the %pdb magic for more details.
443 the %pdb magic for more details.
444 """
444 """
445 args = magic_arguments.parse_argstring(self.debug, line)
445 args = magic_arguments.parse_argstring(self.debug, line)
446
446
447 if not (args.breakpoint or args.statement or cell):
447 if not (args.breakpoint or args.statement or cell):
448 self._debug_post_mortem()
448 self._debug_post_mortem()
449 else:
449 else:
450 code = "\n".join(args.statement)
450 code = "\n".join(args.statement)
451 if cell:
451 if cell:
452 code += "\n" + cell
452 code += "\n" + cell
453 self._debug_exec(code, args.breakpoint)
453 self._debug_exec(code, args.breakpoint)
454
454
455 def _debug_post_mortem(self):
455 def _debug_post_mortem(self):
456 self.shell.debugger(force=True)
456 self.shell.debugger(force=True)
457
457
458 def _debug_exec(self, code, breakpoint):
458 def _debug_exec(self, code, breakpoint):
459 if breakpoint:
459 if breakpoint:
460 (filename, bp_line) = breakpoint.rsplit(':', 1)
460 (filename, bp_line) = breakpoint.rsplit(':', 1)
461 bp_line = int(bp_line)
461 bp_line = int(bp_line)
462 else:
462 else:
463 (filename, bp_line) = (None, None)
463 (filename, bp_line) = (None, None)
464 self._run_with_debugger(code, self.shell.user_ns, filename, bp_line)
464 self._run_with_debugger(code, self.shell.user_ns, filename, bp_line)
465
465
466 @line_magic
466 @line_magic
467 def tb(self, s):
467 def tb(self, s):
468 """Print the last traceback with the currently active exception mode.
468 """Print the last traceback with the currently active exception mode.
469
469
470 See %xmode for changing exception reporting modes."""
470 See %xmode for changing exception reporting modes."""
471 self.shell.showtraceback()
471 self.shell.showtraceback()
472
472
473 @skip_doctest
473 @skip_doctest
474 @line_magic
474 @line_magic
475 def run(self, parameter_s='', runner=None,
475 def run(self, parameter_s='', runner=None,
476 file_finder=get_py_filename):
476 file_finder=get_py_filename):
477 """Run the named file inside IPython as a program.
477 """Run the named file inside IPython as a program.
478
478
479 Usage::
479 Usage::
480
480
481 %run [-n -i -e -G]
481 %run [-n -i -e -G]
482 [( -t [-N<N>] | -d [-b<N>] | -p [profile options] )]
482 [( -t [-N<N>] | -d [-b<N>] | -p [profile options] )]
483 ( -m mod | file ) [args]
483 ( -m mod | file ) [args]
484
484
485 Parameters after the filename are passed as command-line arguments to
485 Parameters after the filename are passed as command-line arguments to
486 the program (put in sys.argv). Then, control returns to IPython's
486 the program (put in sys.argv). Then, control returns to IPython's
487 prompt.
487 prompt.
488
488
489 This is similar to running at a system prompt ``python file args``,
489 This is similar to running at a system prompt ``python file args``,
490 but with the advantage of giving you IPython's tracebacks, and of
490 but with the advantage of giving you IPython's tracebacks, and of
491 loading all variables into your interactive namespace for further use
491 loading all variables into your interactive namespace for further use
492 (unless -p is used, see below).
492 (unless -p is used, see below).
493
493
494 The file is executed in a namespace initially consisting only of
494 The file is executed in a namespace initially consisting only of
495 ``__name__=='__main__'`` and sys.argv constructed as indicated. It thus
495 ``__name__=='__main__'`` and sys.argv constructed as indicated. It thus
496 sees its environment as if it were being run as a stand-alone program
496 sees its environment as if it were being run as a stand-alone program
497 (except for sharing global objects such as previously imported
497 (except for sharing global objects such as previously imported
498 modules). But after execution, the IPython interactive namespace gets
498 modules). But after execution, the IPython interactive namespace gets
499 updated with all variables defined in the program (except for __name__
499 updated with all variables defined in the program (except for __name__
500 and sys.argv). This allows for very convenient loading of code for
500 and sys.argv). This allows for very convenient loading of code for
501 interactive work, while giving each program a 'clean sheet' to run in.
501 interactive work, while giving each program a 'clean sheet' to run in.
502
502
503 Arguments are expanded using shell-like glob match. Patterns
503 Arguments are expanded using shell-like glob match. Patterns
504 '*', '?', '[seq]' and '[!seq]' can be used. Additionally,
504 '*', '?', '[seq]' and '[!seq]' can be used. Additionally,
505 tilde '~' will be expanded into user's home directory. Unlike
505 tilde '~' will be expanded into user's home directory. Unlike
506 real shells, quotation does not suppress expansions. Use
506 real shells, quotation does not suppress expansions. Use
507 *two* back slashes (e.g. ``\\\\*``) to suppress expansions.
507 *two* back slashes (e.g. ``\\\\*``) to suppress expansions.
508 To completely disable these expansions, you can use -G flag.
508 To completely disable these expansions, you can use -G flag.
509
509
510 On Windows systems, the use of single quotes `'` when specifing
511 a file is not supported. Use double quotes `"`.
512
510 Options:
513 Options:
511
514
512 -n
515 -n
513 __name__ is NOT set to '__main__', but to the running file's name
516 __name__ is NOT set to '__main__', but to the running file's name
514 without extension (as python does under import). This allows running
517 without extension (as python does under import). This allows running
515 scripts and reloading the definitions in them without calling code
518 scripts and reloading the definitions in them without calling code
516 protected by an ``if __name__ == "__main__"`` clause.
519 protected by an ``if __name__ == "__main__"`` clause.
517
520
518 -i
521 -i
519 run the file in IPython's namespace instead of an empty one. This
522 run the file in IPython's namespace instead of an empty one. This
520 is useful if you are experimenting with code written in a text editor
523 is useful if you are experimenting with code written in a text editor
521 which depends on variables defined interactively.
524 which depends on variables defined interactively.
522
525
523 -e
526 -e
524 ignore sys.exit() calls or SystemExit exceptions in the script
527 ignore sys.exit() calls or SystemExit exceptions in the script
525 being run. This is particularly useful if IPython is being used to
528 being run. This is particularly useful if IPython is being used to
526 run unittests, which always exit with a sys.exit() call. In such
529 run unittests, which always exit with a sys.exit() call. In such
527 cases you are interested in the output of the test results, not in
530 cases you are interested in the output of the test results, not in
528 seeing a traceback of the unittest module.
531 seeing a traceback of the unittest module.
529
532
530 -t
533 -t
531 print timing information at the end of the run. IPython will give
534 print timing information at the end of the run. IPython will give
532 you an estimated CPU time consumption for your script, which under
535 you an estimated CPU time consumption for your script, which under
533 Unix uses the resource module to avoid the wraparound problems of
536 Unix uses the resource module to avoid the wraparound problems of
534 time.clock(). Under Unix, an estimate of time spent on system tasks
537 time.clock(). Under Unix, an estimate of time spent on system tasks
535 is also given (for Windows platforms this is reported as 0.0).
538 is also given (for Windows platforms this is reported as 0.0).
536
539
537 If -t is given, an additional ``-N<N>`` option can be given, where <N>
540 If -t is given, an additional ``-N<N>`` option can be given, where <N>
538 must be an integer indicating how many times you want the script to
541 must be an integer indicating how many times you want the script to
539 run. The final timing report will include total and per run results.
542 run. The final timing report will include total and per run results.
540
543
541 For example (testing the script uniq_stable.py)::
544 For example (testing the script uniq_stable.py)::
542
545
543 In [1]: run -t uniq_stable
546 In [1]: run -t uniq_stable
544
547
545 IPython CPU timings (estimated):
548 IPython CPU timings (estimated):
546 User : 0.19597 s.
549 User : 0.19597 s.
547 System: 0.0 s.
550 System: 0.0 s.
548
551
549 In [2]: run -t -N5 uniq_stable
552 In [2]: run -t -N5 uniq_stable
550
553
551 IPython CPU timings (estimated):
554 IPython CPU timings (estimated):
552 Total runs performed: 5
555 Total runs performed: 5
553 Times : Total Per run
556 Times : Total Per run
554 User : 0.910862 s, 0.1821724 s.
557 User : 0.910862 s, 0.1821724 s.
555 System: 0.0 s, 0.0 s.
558 System: 0.0 s, 0.0 s.
556
559
557 -d
560 -d
558 run your program under the control of pdb, the Python debugger.
561 run your program under the control of pdb, the Python debugger.
559 This allows you to execute your program step by step, watch variables,
562 This allows you to execute your program step by step, watch variables,
560 etc. Internally, what IPython does is similar to calling::
563 etc. Internally, what IPython does is similar to calling::
561
564
562 pdb.run('execfile("YOURFILENAME")')
565 pdb.run('execfile("YOURFILENAME")')
563
566
564 with a breakpoint set on line 1 of your file. You can change the line
567 with a breakpoint set on line 1 of your file. You can change the line
565 number for this automatic breakpoint to be <N> by using the -bN option
568 number for this automatic breakpoint to be <N> by using the -bN option
566 (where N must be an integer). For example::
569 (where N must be an integer). For example::
567
570
568 %run -d -b40 myscript
571 %run -d -b40 myscript
569
572
570 will set the first breakpoint at line 40 in myscript.py. Note that
573 will set the first breakpoint at line 40 in myscript.py. Note that
571 the first breakpoint must be set on a line which actually does
574 the first breakpoint must be set on a line which actually does
572 something (not a comment or docstring) for it to stop execution.
575 something (not a comment or docstring) for it to stop execution.
573
576
574 Or you can specify a breakpoint in a different file::
577 Or you can specify a breakpoint in a different file::
575
578
576 %run -d -b myotherfile.py:20 myscript
579 %run -d -b myotherfile.py:20 myscript
577
580
578 When the pdb debugger starts, you will see a (Pdb) prompt. You must
581 When the pdb debugger starts, you will see a (Pdb) prompt. You must
579 first enter 'c' (without quotes) to start execution up to the first
582 first enter 'c' (without quotes) to start execution up to the first
580 breakpoint.
583 breakpoint.
581
584
582 Entering 'help' gives information about the use of the debugger. You
585 Entering 'help' gives information about the use of the debugger. You
583 can easily see pdb's full documentation with "import pdb;pdb.help()"
586 can easily see pdb's full documentation with "import pdb;pdb.help()"
584 at a prompt.
587 at a prompt.
585
588
586 -p
589 -p
587 run program under the control of the Python profiler module (which
590 run program under the control of the Python profiler module (which
588 prints a detailed report of execution times, function calls, etc).
591 prints a detailed report of execution times, function calls, etc).
589
592
590 You can pass other options after -p which affect the behavior of the
593 You can pass other options after -p which affect the behavior of the
591 profiler itself. See the docs for %prun for details.
594 profiler itself. See the docs for %prun for details.
592
595
593 In this mode, the program's variables do NOT propagate back to the
596 In this mode, the program's variables do NOT propagate back to the
594 IPython interactive namespace (because they remain in the namespace
597 IPython interactive namespace (because they remain in the namespace
595 where the profiler executes them).
598 where the profiler executes them).
596
599
597 Internally this triggers a call to %prun, see its documentation for
600 Internally this triggers a call to %prun, see its documentation for
598 details on the options available specifically for profiling.
601 details on the options available specifically for profiling.
599
602
600 There is one special usage for which the text above doesn't apply:
603 There is one special usage for which the text above doesn't apply:
601 if the filename ends with .ipy[nb], the file is run as ipython script,
604 if the filename ends with .ipy[nb], the file is run as ipython script,
602 just as if the commands were written on IPython prompt.
605 just as if the commands were written on IPython prompt.
603
606
604 -m
607 -m
605 specify module name to load instead of script path. Similar to
608 specify module name to load instead of script path. Similar to
606 the -m option for the python interpreter. Use this option last if you
609 the -m option for the python interpreter. Use this option last if you
607 want to combine with other %run options. Unlike the python interpreter
610 want to combine with other %run options. Unlike the python interpreter
608 only source modules are allowed no .pyc or .pyo files.
611 only source modules are allowed no .pyc or .pyo files.
609 For example::
612 For example::
610
613
611 %run -m example
614 %run -m example
612
615
613 will run the example module.
616 will run the example module.
614
617
615 -G
618 -G
616 disable shell-like glob expansion of arguments.
619 disable shell-like glob expansion of arguments.
617
620
618 """
621 """
619
622
620 # Logic to handle issue #3664
623 # Logic to handle issue #3664
621 # Add '--' after '-m <module_name>' to ignore additional args passed to a module.
624 # Add '--' after '-m <module_name>' to ignore additional args passed to a module.
622 if '-m' in parameter_s and '--' not in parameter_s:
625 if '-m' in parameter_s and '--' not in parameter_s:
623 argv = shlex.split(parameter_s, posix=(os.name == 'posix'))
626 argv = shlex.split(parameter_s, posix=(os.name == 'posix'))
624 for idx, arg in enumerate(argv):
627 for idx, arg in enumerate(argv):
625 if arg and arg.startswith('-') and arg != '-':
628 if arg and arg.startswith('-') and arg != '-':
626 if arg == '-m':
629 if arg == '-m':
627 argv.insert(idx + 2, '--')
630 argv.insert(idx + 2, '--')
628 break
631 break
629 else:
632 else:
630 # Positional arg, break
633 # Positional arg, break
631 break
634 break
632 parameter_s = ' '.join(shlex.quote(arg) for arg in argv)
635 parameter_s = ' '.join(shlex.quote(arg) for arg in argv)
633
636
634 # get arguments and set sys.argv for program to be run.
637 # get arguments and set sys.argv for program to be run.
635 opts, arg_lst = self.parse_options(parameter_s,
638 opts, arg_lst = self.parse_options(parameter_s,
636 'nidtN:b:pD:l:rs:T:em:G',
639 'nidtN:b:pD:l:rs:T:em:G',
637 mode='list', list_all=1)
640 mode='list', list_all=1)
638 if "m" in opts:
641 if "m" in opts:
639 modulename = opts["m"][0]
642 modulename = opts["m"][0]
640 modpath = find_mod(modulename)
643 modpath = find_mod(modulename)
641 if modpath is None:
644 if modpath is None:
642 warn('%r is not a valid modulename on sys.path'%modulename)
645 warn('%r is not a valid modulename on sys.path'%modulename)
643 return
646 return
644 arg_lst = [modpath] + arg_lst
647 arg_lst = [modpath] + arg_lst
645 try:
648 try:
646 filename = file_finder(arg_lst[0])
649 filename = file_finder(arg_lst[0])
647 except IndexError:
650 except IndexError:
648 warn('you must provide at least a filename.')
651 warn('you must provide at least a filename.')
649 print('\n%run:\n', oinspect.getdoc(self.run))
652 print('\n%run:\n', oinspect.getdoc(self.run))
650 return
653 return
651 except IOError as e:
654 except IOError as e:
652 try:
655 try:
653 msg = str(e)
656 msg = str(e)
654 except UnicodeError:
657 except UnicodeError:
655 msg = e.message
658 msg = e.message
656 error(msg)
659 error(msg)
657 return
660 return
658
661
659 if filename.lower().endswith(('.ipy', '.ipynb')):
662 if filename.lower().endswith(('.ipy', '.ipynb')):
660 with preserve_keys(self.shell.user_ns, '__file__'):
663 with preserve_keys(self.shell.user_ns, '__file__'):
661 self.shell.user_ns['__file__'] = filename
664 self.shell.user_ns['__file__'] = filename
662 self.shell.safe_execfile_ipy(filename)
665 self.shell.safe_execfile_ipy(filename)
663 return
666 return
664
667
665 # Control the response to exit() calls made by the script being run
668 # Control the response to exit() calls made by the script being run
666 exit_ignore = 'e' in opts
669 exit_ignore = 'e' in opts
667
670
668 # Make sure that the running script gets a proper sys.argv as if it
671 # Make sure that the running script gets a proper sys.argv as if it
669 # were run from a system shell.
672 # were run from a system shell.
670 save_argv = sys.argv # save it for later restoring
673 save_argv = sys.argv # save it for later restoring
671
674
672 if 'G' in opts:
675 if 'G' in opts:
673 args = arg_lst[1:]
676 args = arg_lst[1:]
674 else:
677 else:
675 # tilde and glob expansion
678 # tilde and glob expansion
676 args = shellglob(map(os.path.expanduser, arg_lst[1:]))
679 args = shellglob(map(os.path.expanduser, arg_lst[1:]))
677
680
678 sys.argv = [filename] + args # put in the proper filename
681 sys.argv = [filename] + args # put in the proper filename
679
682
680 if 'n' in opts:
683 if 'n' in opts:
681 name = os.path.splitext(os.path.basename(filename))[0]
684 name = os.path.splitext(os.path.basename(filename))[0]
682 else:
685 else:
683 name = '__main__'
686 name = '__main__'
684
687
685 if 'i' in opts:
688 if 'i' in opts:
686 # Run in user's interactive namespace
689 # Run in user's interactive namespace
687 prog_ns = self.shell.user_ns
690 prog_ns = self.shell.user_ns
688 __name__save = self.shell.user_ns['__name__']
691 __name__save = self.shell.user_ns['__name__']
689 prog_ns['__name__'] = name
692 prog_ns['__name__'] = name
690 main_mod = self.shell.user_module
693 main_mod = self.shell.user_module
691
694
692 # Since '%run foo' emulates 'python foo.py' at the cmd line, we must
695 # Since '%run foo' emulates 'python foo.py' at the cmd line, we must
693 # set the __file__ global in the script's namespace
696 # set the __file__ global in the script's namespace
694 # TK: Is this necessary in interactive mode?
697 # TK: Is this necessary in interactive mode?
695 prog_ns['__file__'] = filename
698 prog_ns['__file__'] = filename
696 else:
699 else:
697 # Run in a fresh, empty namespace
700 # Run in a fresh, empty namespace
698
701
699 # The shell MUST hold a reference to prog_ns so after %run
702 # The shell MUST hold a reference to prog_ns so after %run
700 # exits, the python deletion mechanism doesn't zero it out
703 # exits, the python deletion mechanism doesn't zero it out
701 # (leaving dangling references). See interactiveshell for details
704 # (leaving dangling references). See interactiveshell for details
702 main_mod = self.shell.new_main_mod(filename, name)
705 main_mod = self.shell.new_main_mod(filename, name)
703 prog_ns = main_mod.__dict__
706 prog_ns = main_mod.__dict__
704
707
705 # pickle fix. See interactiveshell for an explanation. But we need to
708 # pickle fix. See interactiveshell for an explanation. But we need to
706 # make sure that, if we overwrite __main__, we replace it at the end
709 # make sure that, if we overwrite __main__, we replace it at the end
707 main_mod_name = prog_ns['__name__']
710 main_mod_name = prog_ns['__name__']
708
711
709 if main_mod_name == '__main__':
712 if main_mod_name == '__main__':
710 restore_main = sys.modules['__main__']
713 restore_main = sys.modules['__main__']
711 else:
714 else:
712 restore_main = False
715 restore_main = False
713
716
714 # This needs to be undone at the end to prevent holding references to
717 # This needs to be undone at the end to prevent holding references to
715 # every single object ever created.
718 # every single object ever created.
716 sys.modules[main_mod_name] = main_mod
719 sys.modules[main_mod_name] = main_mod
717
720
718 if 'p' in opts or 'd' in opts:
721 if 'p' in opts or 'd' in opts:
719 if 'm' in opts:
722 if 'm' in opts:
720 code = 'run_module(modulename, prog_ns)'
723 code = 'run_module(modulename, prog_ns)'
721 code_ns = {
724 code_ns = {
722 'run_module': self.shell.safe_run_module,
725 'run_module': self.shell.safe_run_module,
723 'prog_ns': prog_ns,
726 'prog_ns': prog_ns,
724 'modulename': modulename,
727 'modulename': modulename,
725 }
728 }
726 else:
729 else:
727 if 'd' in opts:
730 if 'd' in opts:
728 # allow exceptions to raise in debug mode
731 # allow exceptions to raise in debug mode
729 code = 'execfile(filename, prog_ns, raise_exceptions=True)'
732 code = 'execfile(filename, prog_ns, raise_exceptions=True)'
730 else:
733 else:
731 code = 'execfile(filename, prog_ns)'
734 code = 'execfile(filename, prog_ns)'
732 code_ns = {
735 code_ns = {
733 'execfile': self.shell.safe_execfile,
736 'execfile': self.shell.safe_execfile,
734 'prog_ns': prog_ns,
737 'prog_ns': prog_ns,
735 'filename': get_py_filename(filename),
738 'filename': get_py_filename(filename),
736 }
739 }
737
740
738 try:
741 try:
739 stats = None
742 stats = None
740 if 'p' in opts:
743 if 'p' in opts:
741 stats = self._run_with_profiler(code, opts, code_ns)
744 stats = self._run_with_profiler(code, opts, code_ns)
742 else:
745 else:
743 if 'd' in opts:
746 if 'd' in opts:
744 bp_file, bp_line = parse_breakpoint(
747 bp_file, bp_line = parse_breakpoint(
745 opts.get('b', ['1'])[0], filename)
748 opts.get('b', ['1'])[0], filename)
746 self._run_with_debugger(
749 self._run_with_debugger(
747 code, code_ns, filename, bp_line, bp_file)
750 code, code_ns, filename, bp_line, bp_file)
748 else:
751 else:
749 if 'm' in opts:
752 if 'm' in opts:
750 def run():
753 def run():
751 self.shell.safe_run_module(modulename, prog_ns)
754 self.shell.safe_run_module(modulename, prog_ns)
752 else:
755 else:
753 if runner is None:
756 if runner is None:
754 runner = self.default_runner
757 runner = self.default_runner
755 if runner is None:
758 if runner is None:
756 runner = self.shell.safe_execfile
759 runner = self.shell.safe_execfile
757
760
758 def run():
761 def run():
759 runner(filename, prog_ns, prog_ns,
762 runner(filename, prog_ns, prog_ns,
760 exit_ignore=exit_ignore)
763 exit_ignore=exit_ignore)
761
764
762 if 't' in opts:
765 if 't' in opts:
763 # timed execution
766 # timed execution
764 try:
767 try:
765 nruns = int(opts['N'][0])
768 nruns = int(opts['N'][0])
766 if nruns < 1:
769 if nruns < 1:
767 error('Number of runs must be >=1')
770 error('Number of runs must be >=1')
768 return
771 return
769 except (KeyError):
772 except (KeyError):
770 nruns = 1
773 nruns = 1
771 self._run_with_timing(run, nruns)
774 self._run_with_timing(run, nruns)
772 else:
775 else:
773 # regular execution
776 # regular execution
774 run()
777 run()
775
778
776 if 'i' in opts:
779 if 'i' in opts:
777 self.shell.user_ns['__name__'] = __name__save
780 self.shell.user_ns['__name__'] = __name__save
778 else:
781 else:
779 # update IPython interactive namespace
782 # update IPython interactive namespace
780
783
781 # Some forms of read errors on the file may mean the
784 # Some forms of read errors on the file may mean the
782 # __name__ key was never set; using pop we don't have to
785 # __name__ key was never set; using pop we don't have to
783 # worry about a possible KeyError.
786 # worry about a possible KeyError.
784 prog_ns.pop('__name__', None)
787 prog_ns.pop('__name__', None)
785
788
786 with preserve_keys(self.shell.user_ns, '__file__'):
789 with preserve_keys(self.shell.user_ns, '__file__'):
787 self.shell.user_ns.update(prog_ns)
790 self.shell.user_ns.update(prog_ns)
788 finally:
791 finally:
789 # It's a bit of a mystery why, but __builtins__ can change from
792 # It's a bit of a mystery why, but __builtins__ can change from
790 # being a module to becoming a dict missing some key data after
793 # being a module to becoming a dict missing some key data after
791 # %run. As best I can see, this is NOT something IPython is doing
794 # %run. As best I can see, this is NOT something IPython is doing
792 # at all, and similar problems have been reported before:
795 # at all, and similar problems have been reported before:
793 # http://coding.derkeiler.com/Archive/Python/comp.lang.python/2004-10/0188.html
796 # http://coding.derkeiler.com/Archive/Python/comp.lang.python/2004-10/0188.html
794 # Since this seems to be done by the interpreter itself, the best
797 # Since this seems to be done by the interpreter itself, the best
795 # we can do is to at least restore __builtins__ for the user on
798 # we can do is to at least restore __builtins__ for the user on
796 # exit.
799 # exit.
797 self.shell.user_ns['__builtins__'] = builtin_mod
800 self.shell.user_ns['__builtins__'] = builtin_mod
798
801
799 # Ensure key global structures are restored
802 # Ensure key global structures are restored
800 sys.argv = save_argv
803 sys.argv = save_argv
801 if restore_main:
804 if restore_main:
802 sys.modules['__main__'] = restore_main
805 sys.modules['__main__'] = restore_main
803 else:
806 else:
804 # Remove from sys.modules the reference to main_mod we'd
807 # Remove from sys.modules the reference to main_mod we'd
805 # added. Otherwise it will trap references to objects
808 # added. Otherwise it will trap references to objects
806 # contained therein.
809 # contained therein.
807 del sys.modules[main_mod_name]
810 del sys.modules[main_mod_name]
808
811
809 return stats
812 return stats
810
813
811 def _run_with_debugger(self, code, code_ns, filename=None,
814 def _run_with_debugger(self, code, code_ns, filename=None,
812 bp_line=None, bp_file=None):
815 bp_line=None, bp_file=None):
813 """
816 """
814 Run `code` in debugger with a break point.
817 Run `code` in debugger with a break point.
815
818
816 Parameters
819 Parameters
817 ----------
820 ----------
818 code : str
821 code : str
819 Code to execute.
822 Code to execute.
820 code_ns : dict
823 code_ns : dict
821 A namespace in which `code` is executed.
824 A namespace in which `code` is executed.
822 filename : str
825 filename : str
823 `code` is ran as if it is in `filename`.
826 `code` is ran as if it is in `filename`.
824 bp_line : int, optional
827 bp_line : int, optional
825 Line number of the break point.
828 Line number of the break point.
826 bp_file : str, optional
829 bp_file : str, optional
827 Path to the file in which break point is specified.
830 Path to the file in which break point is specified.
828 `filename` is used if not given.
831 `filename` is used if not given.
829
832
830 Raises
833 Raises
831 ------
834 ------
832 UsageError
835 UsageError
833 If the break point given by `bp_line` is not valid.
836 If the break point given by `bp_line` is not valid.
834
837
835 """
838 """
836 deb = self.shell.InteractiveTB.pdb
839 deb = self.shell.InteractiveTB.pdb
837 if not deb:
840 if not deb:
838 self.shell.InteractiveTB.pdb = self.shell.InteractiveTB.debugger_cls()
841 self.shell.InteractiveTB.pdb = self.shell.InteractiveTB.debugger_cls()
839 deb = self.shell.InteractiveTB.pdb
842 deb = self.shell.InteractiveTB.pdb
840
843
841 # deb.checkline() fails if deb.curframe exists but is None; it can
844 # deb.checkline() fails if deb.curframe exists but is None; it can
842 # handle it not existing. https://github.com/ipython/ipython/issues/10028
845 # handle it not existing. https://github.com/ipython/ipython/issues/10028
843 if hasattr(deb, 'curframe'):
846 if hasattr(deb, 'curframe'):
844 del deb.curframe
847 del deb.curframe
845
848
846 # reset Breakpoint state, which is moronically kept
849 # reset Breakpoint state, which is moronically kept
847 # in a class
850 # in a class
848 bdb.Breakpoint.next = 1
851 bdb.Breakpoint.next = 1
849 bdb.Breakpoint.bplist = {}
852 bdb.Breakpoint.bplist = {}
850 bdb.Breakpoint.bpbynumber = [None]
853 bdb.Breakpoint.bpbynumber = [None]
851 deb.clear_all_breaks()
854 deb.clear_all_breaks()
852 if bp_line is not None:
855 if bp_line is not None:
853 # Set an initial breakpoint to stop execution
856 # Set an initial breakpoint to stop execution
854 maxtries = 10
857 maxtries = 10
855 bp_file = bp_file or filename
858 bp_file = bp_file or filename
856 checkline = deb.checkline(bp_file, bp_line)
859 checkline = deb.checkline(bp_file, bp_line)
857 if not checkline:
860 if not checkline:
858 for bp in range(bp_line + 1, bp_line + maxtries + 1):
861 for bp in range(bp_line + 1, bp_line + maxtries + 1):
859 if deb.checkline(bp_file, bp):
862 if deb.checkline(bp_file, bp):
860 break
863 break
861 else:
864 else:
862 msg = ("\nI failed to find a valid line to set "
865 msg = ("\nI failed to find a valid line to set "
863 "a breakpoint\n"
866 "a breakpoint\n"
864 "after trying up to line: %s.\n"
867 "after trying up to line: %s.\n"
865 "Please set a valid breakpoint manually "
868 "Please set a valid breakpoint manually "
866 "with the -b option." % bp)
869 "with the -b option." % bp)
867 raise UsageError(msg)
870 raise UsageError(msg)
868 # if we find a good linenumber, set the breakpoint
871 # if we find a good linenumber, set the breakpoint
869 deb.do_break('%s:%s' % (bp_file, bp_line))
872 deb.do_break('%s:%s' % (bp_file, bp_line))
870
873
871 if filename:
874 if filename:
872 # Mimic Pdb._runscript(...)
875 # Mimic Pdb._runscript(...)
873 deb._wait_for_mainpyfile = True
876 deb._wait_for_mainpyfile = True
874 deb.mainpyfile = deb.canonic(filename)
877 deb.mainpyfile = deb.canonic(filename)
875
878
876 # Start file run
879 # Start file run
877 print("NOTE: Enter 'c' at the %s prompt to continue execution." % deb.prompt)
880 print("NOTE: Enter 'c' at the %s prompt to continue execution." % deb.prompt)
878 try:
881 try:
879 if filename:
882 if filename:
880 # save filename so it can be used by methods on the deb object
883 # save filename so it can be used by methods on the deb object
881 deb._exec_filename = filename
884 deb._exec_filename = filename
882 while True:
885 while True:
883 try:
886 try:
884 deb.run(code, code_ns)
887 deb.run(code, code_ns)
885 except Restart:
888 except Restart:
886 print("Restarting")
889 print("Restarting")
887 if filename:
890 if filename:
888 deb._wait_for_mainpyfile = True
891 deb._wait_for_mainpyfile = True
889 deb.mainpyfile = deb.canonic(filename)
892 deb.mainpyfile = deb.canonic(filename)
890 continue
893 continue
891 else:
894 else:
892 break
895 break
893
896
894
897
895 except:
898 except:
896 etype, value, tb = sys.exc_info()
899 etype, value, tb = sys.exc_info()
897 # Skip three frames in the traceback: the %run one,
900 # Skip three frames in the traceback: the %run one,
898 # one inside bdb.py, and the command-line typed by the
901 # one inside bdb.py, and the command-line typed by the
899 # user (run by exec in pdb itself).
902 # user (run by exec in pdb itself).
900 self.shell.InteractiveTB(etype, value, tb, tb_offset=3)
903 self.shell.InteractiveTB(etype, value, tb, tb_offset=3)
901
904
902 @staticmethod
905 @staticmethod
903 def _run_with_timing(run, nruns):
906 def _run_with_timing(run, nruns):
904 """
907 """
905 Run function `run` and print timing information.
908 Run function `run` and print timing information.
906
909
907 Parameters
910 Parameters
908 ----------
911 ----------
909 run : callable
912 run : callable
910 Any callable object which takes no argument.
913 Any callable object which takes no argument.
911 nruns : int
914 nruns : int
912 Number of times to execute `run`.
915 Number of times to execute `run`.
913
916
914 """
917 """
915 twall0 = time.time()
918 twall0 = time.time()
916 if nruns == 1:
919 if nruns == 1:
917 t0 = clock2()
920 t0 = clock2()
918 run()
921 run()
919 t1 = clock2()
922 t1 = clock2()
920 t_usr = t1[0] - t0[0]
923 t_usr = t1[0] - t0[0]
921 t_sys = t1[1] - t0[1]
924 t_sys = t1[1] - t0[1]
922 print("\nIPython CPU timings (estimated):")
925 print("\nIPython CPU timings (estimated):")
923 print(" User : %10.2f s." % t_usr)
926 print(" User : %10.2f s." % t_usr)
924 print(" System : %10.2f s." % t_sys)
927 print(" System : %10.2f s." % t_sys)
925 else:
928 else:
926 runs = range(nruns)
929 runs = range(nruns)
927 t0 = clock2()
930 t0 = clock2()
928 for nr in runs:
931 for nr in runs:
929 run()
932 run()
930 t1 = clock2()
933 t1 = clock2()
931 t_usr = t1[0] - t0[0]
934 t_usr = t1[0] - t0[0]
932 t_sys = t1[1] - t0[1]
935 t_sys = t1[1] - t0[1]
933 print("\nIPython CPU timings (estimated):")
936 print("\nIPython CPU timings (estimated):")
934 print("Total runs performed:", nruns)
937 print("Total runs performed:", nruns)
935 print(" Times : %10s %10s" % ('Total', 'Per run'))
938 print(" Times : %10s %10s" % ('Total', 'Per run'))
936 print(" User : %10.2f s, %10.2f s." % (t_usr, t_usr / nruns))
939 print(" User : %10.2f s, %10.2f s." % (t_usr, t_usr / nruns))
937 print(" System : %10.2f s, %10.2f s." % (t_sys, t_sys / nruns))
940 print(" System : %10.2f s, %10.2f s." % (t_sys, t_sys / nruns))
938 twall1 = time.time()
941 twall1 = time.time()
939 print("Wall time: %10.2f s." % (twall1 - twall0))
942 print("Wall time: %10.2f s." % (twall1 - twall0))
940
943
941 @skip_doctest
944 @skip_doctest
942 @line_cell_magic
945 @line_cell_magic
943 @needs_local_scope
946 @needs_local_scope
944 def timeit(self, line='', cell=None, local_ns=None):
947 def timeit(self, line='', cell=None, local_ns=None):
945 """Time execution of a Python statement or expression
948 """Time execution of a Python statement or expression
946
949
947 Usage, in line mode:
950 Usage, in line mode:
948 %timeit [-n<N> -r<R> [-t|-c] -q -p<P> -o] statement
951 %timeit [-n<N> -r<R> [-t|-c] -q -p<P> -o] statement
949 or in cell mode:
952 or in cell mode:
950 %%timeit [-n<N> -r<R> [-t|-c] -q -p<P> -o] setup_code
953 %%timeit [-n<N> -r<R> [-t|-c] -q -p<P> -o] setup_code
951 code
954 code
952 code...
955 code...
953
956
954 Time execution of a Python statement or expression using the timeit
957 Time execution of a Python statement or expression using the timeit
955 module. This function can be used both as a line and cell magic:
958 module. This function can be used both as a line and cell magic:
956
959
957 - In line mode you can time a single-line statement (though multiple
960 - In line mode you can time a single-line statement (though multiple
958 ones can be chained with using semicolons).
961 ones can be chained with using semicolons).
959
962
960 - In cell mode, the statement in the first line is used as setup code
963 - In cell mode, the statement in the first line is used as setup code
961 (executed but not timed) and the body of the cell is timed. The cell
964 (executed but not timed) and the body of the cell is timed. The cell
962 body has access to any variables created in the setup code.
965 body has access to any variables created in the setup code.
963
966
964 Options:
967 Options:
965 -n<N>: execute the given statement <N> times in a loop. If <N> is not
968 -n<N>: execute the given statement <N> times in a loop. If <N> is not
966 provided, <N> is determined so as to get sufficient accuracy.
969 provided, <N> is determined so as to get sufficient accuracy.
967
970
968 -r<R>: number of repeats <R>, each consisting of <N> loops, and take the
971 -r<R>: number of repeats <R>, each consisting of <N> loops, and take the
969 best result.
972 best result.
970 Default: 7
973 Default: 7
971
974
972 -t: use time.time to measure the time, which is the default on Unix.
975 -t: use time.time to measure the time, which is the default on Unix.
973 This function measures wall time.
976 This function measures wall time.
974
977
975 -c: use time.clock to measure the time, which is the default on
978 -c: use time.clock to measure the time, which is the default on
976 Windows and measures wall time. On Unix, resource.getrusage is used
979 Windows and measures wall time. On Unix, resource.getrusage is used
977 instead and returns the CPU user time.
980 instead and returns the CPU user time.
978
981
979 -p<P>: use a precision of <P> digits to display the timing result.
982 -p<P>: use a precision of <P> digits to display the timing result.
980 Default: 3
983 Default: 3
981
984
982 -q: Quiet, do not print result.
985 -q: Quiet, do not print result.
983
986
984 -o: return a TimeitResult that can be stored in a variable to inspect
987 -o: return a TimeitResult that can be stored in a variable to inspect
985 the result in more details.
988 the result in more details.
986
989
987
990
988 Examples
991 Examples
989 --------
992 --------
990 ::
993 ::
991
994
992 In [1]: %timeit pass
995 In [1]: %timeit pass
993 8.26 ns ± 0.12 ns per loop (mean ± std. dev. of 7 runs, 100000000 loops each)
996 8.26 ns ± 0.12 ns per loop (mean ± std. dev. of 7 runs, 100000000 loops each)
994
997
995 In [2]: u = None
998 In [2]: u = None
996
999
997 In [3]: %timeit u is None
1000 In [3]: %timeit u is None
998 29.9 ns ± 0.643 ns per loop (mean ± std. dev. of 7 runs, 10000000 loops each)
1001 29.9 ns ± 0.643 ns per loop (mean ± std. dev. of 7 runs, 10000000 loops each)
999
1002
1000 In [4]: %timeit -r 4 u == None
1003 In [4]: %timeit -r 4 u == None
1001
1004
1002 In [5]: import time
1005 In [5]: import time
1003
1006
1004 In [6]: %timeit -n1 time.sleep(2)
1007 In [6]: %timeit -n1 time.sleep(2)
1005
1008
1006
1009
1007 The times reported by %timeit will be slightly higher than those
1010 The times reported by %timeit will be slightly higher than those
1008 reported by the timeit.py script when variables are accessed. This is
1011 reported by the timeit.py script when variables are accessed. This is
1009 due to the fact that %timeit executes the statement in the namespace
1012 due to the fact that %timeit executes the statement in the namespace
1010 of the shell, compared with timeit.py, which uses a single setup
1013 of the shell, compared with timeit.py, which uses a single setup
1011 statement to import function or create variables. Generally, the bias
1014 statement to import function or create variables. Generally, the bias
1012 does not matter as long as results from timeit.py are not mixed with
1015 does not matter as long as results from timeit.py are not mixed with
1013 those from %timeit."""
1016 those from %timeit."""
1014
1017
1015 opts, stmt = self.parse_options(line,'n:r:tcp:qo',
1018 opts, stmt = self.parse_options(line,'n:r:tcp:qo',
1016 posix=False, strict=False)
1019 posix=False, strict=False)
1017 if stmt == "" and cell is None:
1020 if stmt == "" and cell is None:
1018 return
1021 return
1019
1022
1020 timefunc = timeit.default_timer
1023 timefunc = timeit.default_timer
1021 number = int(getattr(opts, "n", 0))
1024 number = int(getattr(opts, "n", 0))
1022 default_repeat = 7 if timeit.default_repeat < 7 else timeit.default_repeat
1025 default_repeat = 7 if timeit.default_repeat < 7 else timeit.default_repeat
1023 repeat = int(getattr(opts, "r", default_repeat))
1026 repeat = int(getattr(opts, "r", default_repeat))
1024 precision = int(getattr(opts, "p", 3))
1027 precision = int(getattr(opts, "p", 3))
1025 quiet = 'q' in opts
1028 quiet = 'q' in opts
1026 return_result = 'o' in opts
1029 return_result = 'o' in opts
1027 if hasattr(opts, "t"):
1030 if hasattr(opts, "t"):
1028 timefunc = time.time
1031 timefunc = time.time
1029 if hasattr(opts, "c"):
1032 if hasattr(opts, "c"):
1030 timefunc = clock
1033 timefunc = clock
1031
1034
1032 timer = Timer(timer=timefunc)
1035 timer = Timer(timer=timefunc)
1033 # this code has tight coupling to the inner workings of timeit.Timer,
1036 # this code has tight coupling to the inner workings of timeit.Timer,
1034 # but is there a better way to achieve that the code stmt has access
1037 # but is there a better way to achieve that the code stmt has access
1035 # to the shell namespace?
1038 # to the shell namespace?
1036 transform = self.shell.transform_cell
1039 transform = self.shell.transform_cell
1037
1040
1038 if cell is None:
1041 if cell is None:
1039 # called as line magic
1042 # called as line magic
1040 ast_setup = self.shell.compile.ast_parse("pass")
1043 ast_setup = self.shell.compile.ast_parse("pass")
1041 ast_stmt = self.shell.compile.ast_parse(transform(stmt))
1044 ast_stmt = self.shell.compile.ast_parse(transform(stmt))
1042 else:
1045 else:
1043 ast_setup = self.shell.compile.ast_parse(transform(stmt))
1046 ast_setup = self.shell.compile.ast_parse(transform(stmt))
1044 ast_stmt = self.shell.compile.ast_parse(transform(cell))
1047 ast_stmt = self.shell.compile.ast_parse(transform(cell))
1045
1048
1046 ast_setup = self.shell.transform_ast(ast_setup)
1049 ast_setup = self.shell.transform_ast(ast_setup)
1047 ast_stmt = self.shell.transform_ast(ast_stmt)
1050 ast_stmt = self.shell.transform_ast(ast_stmt)
1048
1051
1049 # Check that these compile to valid Python code *outside* the timer func
1052 # Check that these compile to valid Python code *outside* the timer func
1050 # Invalid code may become valid when put inside the function & loop,
1053 # Invalid code may become valid when put inside the function & loop,
1051 # which messes up error messages.
1054 # which messes up error messages.
1052 # https://github.com/ipython/ipython/issues/10636
1055 # https://github.com/ipython/ipython/issues/10636
1053 self.shell.compile(ast_setup, "<magic-timeit-setup>", "exec")
1056 self.shell.compile(ast_setup, "<magic-timeit-setup>", "exec")
1054 self.shell.compile(ast_stmt, "<magic-timeit-stmt>", "exec")
1057 self.shell.compile(ast_stmt, "<magic-timeit-stmt>", "exec")
1055
1058
1056 # This codestring is taken from timeit.template - we fill it in as an
1059 # This codestring is taken from timeit.template - we fill it in as an
1057 # AST, so that we can apply our AST transformations to the user code
1060 # AST, so that we can apply our AST transformations to the user code
1058 # without affecting the timing code.
1061 # without affecting the timing code.
1059 timeit_ast_template = ast.parse('def inner(_it, _timer):\n'
1062 timeit_ast_template = ast.parse('def inner(_it, _timer):\n'
1060 ' setup\n'
1063 ' setup\n'
1061 ' _t0 = _timer()\n'
1064 ' _t0 = _timer()\n'
1062 ' for _i in _it:\n'
1065 ' for _i in _it:\n'
1063 ' stmt\n'
1066 ' stmt\n'
1064 ' _t1 = _timer()\n'
1067 ' _t1 = _timer()\n'
1065 ' return _t1 - _t0\n')
1068 ' return _t1 - _t0\n')
1066
1069
1067 timeit_ast = TimeitTemplateFiller(ast_setup, ast_stmt).visit(timeit_ast_template)
1070 timeit_ast = TimeitTemplateFiller(ast_setup, ast_stmt).visit(timeit_ast_template)
1068 timeit_ast = ast.fix_missing_locations(timeit_ast)
1071 timeit_ast = ast.fix_missing_locations(timeit_ast)
1069
1072
1070 # Track compilation time so it can be reported if too long
1073 # Track compilation time so it can be reported if too long
1071 # Minimum time above which compilation time will be reported
1074 # Minimum time above which compilation time will be reported
1072 tc_min = 0.1
1075 tc_min = 0.1
1073
1076
1074 t0 = clock()
1077 t0 = clock()
1075 code = self.shell.compile(timeit_ast, "<magic-timeit>", "exec")
1078 code = self.shell.compile(timeit_ast, "<magic-timeit>", "exec")
1076 tc = clock()-t0
1079 tc = clock()-t0
1077
1080
1078 ns = {}
1081 ns = {}
1079 glob = self.shell.user_ns
1082 glob = self.shell.user_ns
1080 # handles global vars with same name as local vars. We store them in conflict_globs.
1083 # handles global vars with same name as local vars. We store them in conflict_globs.
1081 if local_ns is not None:
1084 if local_ns is not None:
1082 conflict_globs = {}
1085 conflict_globs = {}
1083 for var_name, var_val in glob.items():
1086 for var_name, var_val in glob.items():
1084 if var_name in local_ns:
1087 if var_name in local_ns:
1085 conflict_globs[var_name] = var_val
1088 conflict_globs[var_name] = var_val
1086 glob.update(local_ns)
1089 glob.update(local_ns)
1087
1090
1088 exec(code, glob, ns)
1091 exec(code, glob, ns)
1089 timer.inner = ns["inner"]
1092 timer.inner = ns["inner"]
1090
1093
1091 # This is used to check if there is a huge difference between the
1094 # This is used to check if there is a huge difference between the
1092 # best and worst timings.
1095 # best and worst timings.
1093 # Issue: https://github.com/ipython/ipython/issues/6471
1096 # Issue: https://github.com/ipython/ipython/issues/6471
1094 if number == 0:
1097 if number == 0:
1095 # determine number so that 0.2 <= total time < 2.0
1098 # determine number so that 0.2 <= total time < 2.0
1096 for index in range(0, 10):
1099 for index in range(0, 10):
1097 number = 10 ** index
1100 number = 10 ** index
1098 time_number = timer.timeit(number)
1101 time_number = timer.timeit(number)
1099 if time_number >= 0.2:
1102 if time_number >= 0.2:
1100 break
1103 break
1101
1104
1102 all_runs = timer.repeat(repeat, number)
1105 all_runs = timer.repeat(repeat, number)
1103 best = min(all_runs) / number
1106 best = min(all_runs) / number
1104 worst = max(all_runs) / number
1107 worst = max(all_runs) / number
1105 timeit_result = TimeitResult(number, repeat, best, worst, all_runs, tc, precision)
1108 timeit_result = TimeitResult(number, repeat, best, worst, all_runs, tc, precision)
1106
1109
1107 # Restore global vars from conflict_globs
1110 # Restore global vars from conflict_globs
1108 if local_ns is not None:
1111 if local_ns is not None:
1109 if len(conflict_globs) > 0:
1112 if len(conflict_globs) > 0:
1110 glob.update(conflict_globs)
1113 glob.update(conflict_globs)
1111
1114
1112 if not quiet :
1115 if not quiet :
1113 # Check best timing is greater than zero to avoid a
1116 # Check best timing is greater than zero to avoid a
1114 # ZeroDivisionError.
1117 # ZeroDivisionError.
1115 # In cases where the slowest timing is lesser than a micosecond
1118 # In cases where the slowest timing is lesser than a micosecond
1116 # we assume that it does not really matter if the fastest
1119 # we assume that it does not really matter if the fastest
1117 # timing is 4 times faster than the slowest timing or not.
1120 # timing is 4 times faster than the slowest timing or not.
1118 if worst > 4 * best and best > 0 and worst > 1e-6:
1121 if worst > 4 * best and best > 0 and worst > 1e-6:
1119 print("The slowest run took %0.2f times longer than the "
1122 print("The slowest run took %0.2f times longer than the "
1120 "fastest. This could mean that an intermediate result "
1123 "fastest. This could mean that an intermediate result "
1121 "is being cached." % (worst / best))
1124 "is being cached." % (worst / best))
1122
1125
1123 print( timeit_result )
1126 print( timeit_result )
1124
1127
1125 if tc > tc_min:
1128 if tc > tc_min:
1126 print("Compiler time: %.2f s" % tc)
1129 print("Compiler time: %.2f s" % tc)
1127 if return_result:
1130 if return_result:
1128 return timeit_result
1131 return timeit_result
1129
1132
1130 @skip_doctest
1133 @skip_doctest
1131 @needs_local_scope
1134 @needs_local_scope
1132 @line_cell_magic
1135 @line_cell_magic
1133 def time(self,line='', cell=None, local_ns=None):
1136 def time(self,line='', cell=None, local_ns=None):
1134 """Time execution of a Python statement or expression.
1137 """Time execution of a Python statement or expression.
1135
1138
1136 The CPU and wall clock times are printed, and the value of the
1139 The CPU and wall clock times are printed, and the value of the
1137 expression (if any) is returned. Note that under Win32, system time
1140 expression (if any) is returned. Note that under Win32, system time
1138 is always reported as 0, since it can not be measured.
1141 is always reported as 0, since it can not be measured.
1139
1142
1140 This function can be used both as a line and cell magic:
1143 This function can be used both as a line and cell magic:
1141
1144
1142 - In line mode you can time a single-line statement (though multiple
1145 - In line mode you can time a single-line statement (though multiple
1143 ones can be chained with using semicolons).
1146 ones can be chained with using semicolons).
1144
1147
1145 - In cell mode, you can time the cell body (a directly
1148 - In cell mode, you can time the cell body (a directly
1146 following statement raises an error).
1149 following statement raises an error).
1147
1150
1148 This function provides very basic timing functionality. Use the timeit
1151 This function provides very basic timing functionality. Use the timeit
1149 magic for more control over the measurement.
1152 magic for more control over the measurement.
1150
1153
1151 Examples
1154 Examples
1152 --------
1155 --------
1153 ::
1156 ::
1154
1157
1155 In [1]: %time 2**128
1158 In [1]: %time 2**128
1156 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1159 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1157 Wall time: 0.00
1160 Wall time: 0.00
1158 Out[1]: 340282366920938463463374607431768211456L
1161 Out[1]: 340282366920938463463374607431768211456L
1159
1162
1160 In [2]: n = 1000000
1163 In [2]: n = 1000000
1161
1164
1162 In [3]: %time sum(range(n))
1165 In [3]: %time sum(range(n))
1163 CPU times: user 1.20 s, sys: 0.05 s, total: 1.25 s
1166 CPU times: user 1.20 s, sys: 0.05 s, total: 1.25 s
1164 Wall time: 1.37
1167 Wall time: 1.37
1165 Out[3]: 499999500000L
1168 Out[3]: 499999500000L
1166
1169
1167 In [4]: %time print 'hello world'
1170 In [4]: %time print 'hello world'
1168 hello world
1171 hello world
1169 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1172 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1170 Wall time: 0.00
1173 Wall time: 0.00
1171
1174
1172 Note that the time needed by Python to compile the given expression
1175 Note that the time needed by Python to compile the given expression
1173 will be reported if it is more than 0.1s. In this example, the
1176 will be reported if it is more than 0.1s. In this example, the
1174 actual exponentiation is done by Python at compilation time, so while
1177 actual exponentiation is done by Python at compilation time, so while
1175 the expression can take a noticeable amount of time to compute, that
1178 the expression can take a noticeable amount of time to compute, that
1176 time is purely due to the compilation:
1179 time is purely due to the compilation:
1177
1180
1178 In [5]: %time 3**9999;
1181 In [5]: %time 3**9999;
1179 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1182 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1180 Wall time: 0.00 s
1183 Wall time: 0.00 s
1181
1184
1182 In [6]: %time 3**999999;
1185 In [6]: %time 3**999999;
1183 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1186 CPU times: user 0.00 s, sys: 0.00 s, total: 0.00 s
1184 Wall time: 0.00 s
1187 Wall time: 0.00 s
1185 Compiler : 0.78 s
1188 Compiler : 0.78 s
1186 """
1189 """
1187
1190
1188 # fail immediately if the given expression can't be compiled
1191 # fail immediately if the given expression can't be compiled
1189
1192
1190 if line and cell:
1193 if line and cell:
1191 raise UsageError("Can't use statement directly after '%%time'!")
1194 raise UsageError("Can't use statement directly after '%%time'!")
1192
1195
1193 if cell:
1196 if cell:
1194 expr = self.shell.transform_cell(cell)
1197 expr = self.shell.transform_cell(cell)
1195 else:
1198 else:
1196 expr = self.shell.transform_cell(line)
1199 expr = self.shell.transform_cell(line)
1197
1200
1198 # Minimum time above which parse time will be reported
1201 # Minimum time above which parse time will be reported
1199 tp_min = 0.1
1202 tp_min = 0.1
1200
1203
1201 t0 = clock()
1204 t0 = clock()
1202 expr_ast = self.shell.compile.ast_parse(expr)
1205 expr_ast = self.shell.compile.ast_parse(expr)
1203 tp = clock()-t0
1206 tp = clock()-t0
1204
1207
1205 # Apply AST transformations
1208 # Apply AST transformations
1206 expr_ast = self.shell.transform_ast(expr_ast)
1209 expr_ast = self.shell.transform_ast(expr_ast)
1207
1210
1208 # Minimum time above which compilation time will be reported
1211 # Minimum time above which compilation time will be reported
1209 tc_min = 0.1
1212 tc_min = 0.1
1210
1213
1211 if len(expr_ast.body)==1 and isinstance(expr_ast.body[0], ast.Expr):
1214 if len(expr_ast.body)==1 and isinstance(expr_ast.body[0], ast.Expr):
1212 mode = 'eval'
1215 mode = 'eval'
1213 source = '<timed eval>'
1216 source = '<timed eval>'
1214 expr_ast = ast.Expression(expr_ast.body[0].value)
1217 expr_ast = ast.Expression(expr_ast.body[0].value)
1215 else:
1218 else:
1216 mode = 'exec'
1219 mode = 'exec'
1217 source = '<timed exec>'
1220 source = '<timed exec>'
1218 t0 = clock()
1221 t0 = clock()
1219 code = self.shell.compile(expr_ast, source, mode)
1222 code = self.shell.compile(expr_ast, source, mode)
1220 tc = clock()-t0
1223 tc = clock()-t0
1221
1224
1222 # skew measurement as little as possible
1225 # skew measurement as little as possible
1223 glob = self.shell.user_ns
1226 glob = self.shell.user_ns
1224 wtime = time.time
1227 wtime = time.time
1225 # time execution
1228 # time execution
1226 wall_st = wtime()
1229 wall_st = wtime()
1227 if mode=='eval':
1230 if mode=='eval':
1228 st = clock2()
1231 st = clock2()
1229 try:
1232 try:
1230 out = eval(code, glob, local_ns)
1233 out = eval(code, glob, local_ns)
1231 except:
1234 except:
1232 self.shell.showtraceback()
1235 self.shell.showtraceback()
1233 return
1236 return
1234 end = clock2()
1237 end = clock2()
1235 else:
1238 else:
1236 st = clock2()
1239 st = clock2()
1237 try:
1240 try:
1238 exec(code, glob, local_ns)
1241 exec(code, glob, local_ns)
1239 except:
1242 except:
1240 self.shell.showtraceback()
1243 self.shell.showtraceback()
1241 return
1244 return
1242 end = clock2()
1245 end = clock2()
1243 out = None
1246 out = None
1244 wall_end = wtime()
1247 wall_end = wtime()
1245 # Compute actual times and report
1248 # Compute actual times and report
1246 wall_time = wall_end-wall_st
1249 wall_time = wall_end-wall_st
1247 cpu_user = end[0]-st[0]
1250 cpu_user = end[0]-st[0]
1248 cpu_sys = end[1]-st[1]
1251 cpu_sys = end[1]-st[1]
1249 cpu_tot = cpu_user+cpu_sys
1252 cpu_tot = cpu_user+cpu_sys
1250 # On windows cpu_sys is always zero, so no new information to the next print
1253 # On windows cpu_sys is always zero, so no new information to the next print
1251 if sys.platform != 'win32':
1254 if sys.platform != 'win32':
1252 print("CPU times: user %s, sys: %s, total: %s" % \
1255 print("CPU times: user %s, sys: %s, total: %s" % \
1253 (_format_time(cpu_user),_format_time(cpu_sys),_format_time(cpu_tot)))
1256 (_format_time(cpu_user),_format_time(cpu_sys),_format_time(cpu_tot)))
1254 print("Wall time: %s" % _format_time(wall_time))
1257 print("Wall time: %s" % _format_time(wall_time))
1255 if tc > tc_min:
1258 if tc > tc_min:
1256 print("Compiler : %s" % _format_time(tc))
1259 print("Compiler : %s" % _format_time(tc))
1257 if tp > tp_min:
1260 if tp > tp_min:
1258 print("Parser : %s" % _format_time(tp))
1261 print("Parser : %s" % _format_time(tp))
1259 return out
1262 return out
1260
1263
1261 @skip_doctest
1264 @skip_doctest
1262 @line_magic
1265 @line_magic
1263 def macro(self, parameter_s=''):
1266 def macro(self, parameter_s=''):
1264 """Define a macro for future re-execution. It accepts ranges of history,
1267 """Define a macro for future re-execution. It accepts ranges of history,
1265 filenames or string objects.
1268 filenames or string objects.
1266
1269
1267 Usage:\\
1270 Usage:\\
1268 %macro [options] name n1-n2 n3-n4 ... n5 .. n6 ...
1271 %macro [options] name n1-n2 n3-n4 ... n5 .. n6 ...
1269
1272
1270 Options:
1273 Options:
1271
1274
1272 -r: use 'raw' input. By default, the 'processed' history is used,
1275 -r: use 'raw' input. By default, the 'processed' history is used,
1273 so that magics are loaded in their transformed version to valid
1276 so that magics are loaded in their transformed version to valid
1274 Python. If this option is given, the raw input as typed at the
1277 Python. If this option is given, the raw input as typed at the
1275 command line is used instead.
1278 command line is used instead.
1276
1279
1277 -q: quiet macro definition. By default, a tag line is printed
1280 -q: quiet macro definition. By default, a tag line is printed
1278 to indicate the macro has been created, and then the contents of
1281 to indicate the macro has been created, and then the contents of
1279 the macro are printed. If this option is given, then no printout
1282 the macro are printed. If this option is given, then no printout
1280 is produced once the macro is created.
1283 is produced once the macro is created.
1281
1284
1282 This will define a global variable called `name` which is a string
1285 This will define a global variable called `name` which is a string
1283 made of joining the slices and lines you specify (n1,n2,... numbers
1286 made of joining the slices and lines you specify (n1,n2,... numbers
1284 above) from your input history into a single string. This variable
1287 above) from your input history into a single string. This variable
1285 acts like an automatic function which re-executes those lines as if
1288 acts like an automatic function which re-executes those lines as if
1286 you had typed them. You just type 'name' at the prompt and the code
1289 you had typed them. You just type 'name' at the prompt and the code
1287 executes.
1290 executes.
1288
1291
1289 The syntax for indicating input ranges is described in %history.
1292 The syntax for indicating input ranges is described in %history.
1290
1293
1291 Note: as a 'hidden' feature, you can also use traditional python slice
1294 Note: as a 'hidden' feature, you can also use traditional python slice
1292 notation, where N:M means numbers N through M-1.
1295 notation, where N:M means numbers N through M-1.
1293
1296
1294 For example, if your history contains (print using %hist -n )::
1297 For example, if your history contains (print using %hist -n )::
1295
1298
1296 44: x=1
1299 44: x=1
1297 45: y=3
1300 45: y=3
1298 46: z=x+y
1301 46: z=x+y
1299 47: print x
1302 47: print x
1300 48: a=5
1303 48: a=5
1301 49: print 'x',x,'y',y
1304 49: print 'x',x,'y',y
1302
1305
1303 you can create a macro with lines 44 through 47 (included) and line 49
1306 you can create a macro with lines 44 through 47 (included) and line 49
1304 called my_macro with::
1307 called my_macro with::
1305
1308
1306 In [55]: %macro my_macro 44-47 49
1309 In [55]: %macro my_macro 44-47 49
1307
1310
1308 Now, typing `my_macro` (without quotes) will re-execute all this code
1311 Now, typing `my_macro` (without quotes) will re-execute all this code
1309 in one pass.
1312 in one pass.
1310
1313
1311 You don't need to give the line-numbers in order, and any given line
1314 You don't need to give the line-numbers in order, and any given line
1312 number can appear multiple times. You can assemble macros with any
1315 number can appear multiple times. You can assemble macros with any
1313 lines from your input history in any order.
1316 lines from your input history in any order.
1314
1317
1315 The macro is a simple object which holds its value in an attribute,
1318 The macro is a simple object which holds its value in an attribute,
1316 but IPython's display system checks for macros and executes them as
1319 but IPython's display system checks for macros and executes them as
1317 code instead of printing them when you type their name.
1320 code instead of printing them when you type their name.
1318
1321
1319 You can view a macro's contents by explicitly printing it with::
1322 You can view a macro's contents by explicitly printing it with::
1320
1323
1321 print macro_name
1324 print macro_name
1322
1325
1323 """
1326 """
1324 opts,args = self.parse_options(parameter_s,'rq',mode='list')
1327 opts,args = self.parse_options(parameter_s,'rq',mode='list')
1325 if not args: # List existing macros
1328 if not args: # List existing macros
1326 return sorted(k for k,v in self.shell.user_ns.items() if isinstance(v, Macro))
1329 return sorted(k for k,v in self.shell.user_ns.items() if isinstance(v, Macro))
1327 if len(args) == 1:
1330 if len(args) == 1:
1328 raise UsageError(
1331 raise UsageError(
1329 "%macro insufficient args; usage '%macro name n1-n2 n3-4...")
1332 "%macro insufficient args; usage '%macro name n1-n2 n3-4...")
1330 name, codefrom = args[0], " ".join(args[1:])
1333 name, codefrom = args[0], " ".join(args[1:])
1331
1334
1332 #print 'rng',ranges # dbg
1335 #print 'rng',ranges # dbg
1333 try:
1336 try:
1334 lines = self.shell.find_user_code(codefrom, 'r' in opts)
1337 lines = self.shell.find_user_code(codefrom, 'r' in opts)
1335 except (ValueError, TypeError) as e:
1338 except (ValueError, TypeError) as e:
1336 print(e.args[0])
1339 print(e.args[0])
1337 return
1340 return
1338 macro = Macro(lines)
1341 macro = Macro(lines)
1339 self.shell.define_macro(name, macro)
1342 self.shell.define_macro(name, macro)
1340 if not ( 'q' in opts) :
1343 if not ( 'q' in opts) :
1341 print('Macro `%s` created. To execute, type its name (without quotes).' % name)
1344 print('Macro `%s` created. To execute, type its name (without quotes).' % name)
1342 print('=== Macro contents: ===')
1345 print('=== Macro contents: ===')
1343 print(macro, end=' ')
1346 print(macro, end=' ')
1344
1347
1345 @magic_arguments.magic_arguments()
1348 @magic_arguments.magic_arguments()
1346 @magic_arguments.argument('output', type=str, default='', nargs='?',
1349 @magic_arguments.argument('output', type=str, default='', nargs='?',
1347 help="""The name of the variable in which to store output.
1350 help="""The name of the variable in which to store output.
1348 This is a utils.io.CapturedIO object with stdout/err attributes
1351 This is a utils.io.CapturedIO object with stdout/err attributes
1349 for the text of the captured output.
1352 for the text of the captured output.
1350
1353
1351 CapturedOutput also has a show() method for displaying the output,
1354 CapturedOutput also has a show() method for displaying the output,
1352 and __call__ as well, so you can use that to quickly display the
1355 and __call__ as well, so you can use that to quickly display the
1353 output.
1356 output.
1354
1357
1355 If unspecified, captured output is discarded.
1358 If unspecified, captured output is discarded.
1356 """
1359 """
1357 )
1360 )
1358 @magic_arguments.argument('--no-stderr', action="store_true",
1361 @magic_arguments.argument('--no-stderr', action="store_true",
1359 help="""Don't capture stderr."""
1362 help="""Don't capture stderr."""
1360 )
1363 )
1361 @magic_arguments.argument('--no-stdout', action="store_true",
1364 @magic_arguments.argument('--no-stdout', action="store_true",
1362 help="""Don't capture stdout."""
1365 help="""Don't capture stdout."""
1363 )
1366 )
1364 @magic_arguments.argument('--no-display', action="store_true",
1367 @magic_arguments.argument('--no-display', action="store_true",
1365 help="""Don't capture IPython's rich display."""
1368 help="""Don't capture IPython's rich display."""
1366 )
1369 )
1367 @cell_magic
1370 @cell_magic
1368 def capture(self, line, cell):
1371 def capture(self, line, cell):
1369 """run the cell, capturing stdout, stderr, and IPython's rich display() calls."""
1372 """run the cell, capturing stdout, stderr, and IPython's rich display() calls."""
1370 args = magic_arguments.parse_argstring(self.capture, line)
1373 args = magic_arguments.parse_argstring(self.capture, line)
1371 out = not args.no_stdout
1374 out = not args.no_stdout
1372 err = not args.no_stderr
1375 err = not args.no_stderr
1373 disp = not args.no_display
1376 disp = not args.no_display
1374 with capture_output(out, err, disp) as io:
1377 with capture_output(out, err, disp) as io:
1375 self.shell.run_cell(cell)
1378 self.shell.run_cell(cell)
1376 if args.output:
1379 if args.output:
1377 self.shell.user_ns[args.output] = io
1380 self.shell.user_ns[args.output] = io
1378
1381
1379 def parse_breakpoint(text, current_file):
1382 def parse_breakpoint(text, current_file):
1380 '''Returns (file, line) for file:line and (current_file, line) for line'''
1383 '''Returns (file, line) for file:line and (current_file, line) for line'''
1381 colon = text.find(':')
1384 colon = text.find(':')
1382 if colon == -1:
1385 if colon == -1:
1383 return current_file, int(text)
1386 return current_file, int(text)
1384 else:
1387 else:
1385 return text[:colon], int(text[colon+1:])
1388 return text[:colon], int(text[colon+1:])
1386
1389
1387 def _format_time(timespan, precision=3):
1390 def _format_time(timespan, precision=3):
1388 """Formats the timespan in a human readable form"""
1391 """Formats the timespan in a human readable form"""
1389
1392
1390 if timespan >= 60.0:
1393 if timespan >= 60.0:
1391 # we have more than a minute, format that in a human readable form
1394 # we have more than a minute, format that in a human readable form
1392 # Idea from http://snipplr.com/view/5713/
1395 # Idea from http://snipplr.com/view/5713/
1393 parts = [("d", 60*60*24),("h", 60*60),("min", 60), ("s", 1)]
1396 parts = [("d", 60*60*24),("h", 60*60),("min", 60), ("s", 1)]
1394 time = []
1397 time = []
1395 leftover = timespan
1398 leftover = timespan
1396 for suffix, length in parts:
1399 for suffix, length in parts:
1397 value = int(leftover / length)
1400 value = int(leftover / length)
1398 if value > 0:
1401 if value > 0:
1399 leftover = leftover % length
1402 leftover = leftover % length
1400 time.append(u'%s%s' % (str(value), suffix))
1403 time.append(u'%s%s' % (str(value), suffix))
1401 if leftover < 1:
1404 if leftover < 1:
1402 break
1405 break
1403 return " ".join(time)
1406 return " ".join(time)
1404
1407
1405
1408
1406 # Unfortunately the unicode 'micro' symbol can cause problems in
1409 # Unfortunately the unicode 'micro' symbol can cause problems in
1407 # certain terminals.
1410 # certain terminals.
1408 # See bug: https://bugs.launchpad.net/ipython/+bug/348466
1411 # See bug: https://bugs.launchpad.net/ipython/+bug/348466
1409 # Try to prevent crashes by being more secure than it needs to
1412 # Try to prevent crashes by being more secure than it needs to
1410 # E.g. eclipse is able to print a µ, but has no sys.stdout.encoding set.
1413 # E.g. eclipse is able to print a µ, but has no sys.stdout.encoding set.
1411 units = [u"s", u"ms",u'us',"ns"] # the save value
1414 units = [u"s", u"ms",u'us',"ns"] # the save value
1412 if hasattr(sys.stdout, 'encoding') and sys.stdout.encoding:
1415 if hasattr(sys.stdout, 'encoding') and sys.stdout.encoding:
1413 try:
1416 try:
1414 u'\xb5'.encode(sys.stdout.encoding)
1417 u'\xb5'.encode(sys.stdout.encoding)
1415 units = [u"s", u"ms",u'\xb5s',"ns"]
1418 units = [u"s", u"ms",u'\xb5s',"ns"]
1416 except:
1419 except:
1417 pass
1420 pass
1418 scaling = [1, 1e3, 1e6, 1e9]
1421 scaling = [1, 1e3, 1e6, 1e9]
1419
1422
1420 if timespan > 0.0:
1423 if timespan > 0.0:
1421 order = min(-int(math.floor(math.log10(timespan)) // 3), 3)
1424 order = min(-int(math.floor(math.log10(timespan)) // 3), 3)
1422 else:
1425 else:
1423 order = 3
1426 order = 3
1424 return u"%.*g %s" % (precision, timespan * scaling[order], units[order])
1427 return u"%.*g %s" % (precision, timespan * scaling[order], units[order])
General Comments 0
You need to be logged in to leave comments. Login now