##// END OF EJS Templates
better update for the docs
koryd -
Show More
@@ -1,1426 +1,1427 b''
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:run
191 Usage, in line mode:run
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 double quotes `"` is required.
510 On Windows systems, the use of single quotes `'` when specifing
511 a file is not supported. Use double quotes `"`.
511
512
512 Options:
513 Options:
513
514
514 -n
515 -n
515 __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
516 without extension (as python does under import). This allows running
517 without extension (as python does under import). This allows running
517 scripts and reloading the definitions in them without calling code
518 scripts and reloading the definitions in them without calling code
518 protected by an ``if __name__ == "__main__"`` clause.
519 protected by an ``if __name__ == "__main__"`` clause.
519
520
520 -i
521 -i
521 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
522 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
523 which depends on variables defined interactively.
524 which depends on variables defined interactively.
524
525
525 -e
526 -e
526 ignore sys.exit() calls or SystemExit exceptions in the script
527 ignore sys.exit() calls or SystemExit exceptions in the script
527 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
528 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
529 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
530 seeing a traceback of the unittest module.
531 seeing a traceback of the unittest module.
531
532
532 -t
533 -t
533 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
534 you an estimated CPU time consumption for your script, which under
535 you an estimated CPU time consumption for your script, which under
535 Unix uses the resource module to avoid the wraparound problems of
536 Unix uses the resource module to avoid the wraparound problems of
536 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
537 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).
538
539
539 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>
540 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
541 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.
542
543
543 For example (testing the script uniq_stable.py)::
544 For example (testing the script uniq_stable.py)::
544
545
545 In [1]: run -t uniq_stable
546 In [1]: run -t uniq_stable
546
547
547 IPython CPU timings (estimated):
548 IPython CPU timings (estimated):
548 User : 0.19597 s.
549 User : 0.19597 s.
549 System: 0.0 s.
550 System: 0.0 s.
550
551
551 In [2]: run -t -N5 uniq_stable
552 In [2]: run -t -N5 uniq_stable
552
553
553 IPython CPU timings (estimated):
554 IPython CPU timings (estimated):
554 Total runs performed: 5
555 Total runs performed: 5
555 Times : Total Per run
556 Times : Total Per run
556 User : 0.910862 s, 0.1821724 s.
557 User : 0.910862 s, 0.1821724 s.
557 System: 0.0 s, 0.0 s.
558 System: 0.0 s, 0.0 s.
558
559
559 -d
560 -d
560 run your program under the control of pdb, the Python debugger.
561 run your program under the control of pdb, the Python debugger.
561 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,
562 etc. Internally, what IPython does is similar to calling::
563 etc. Internally, what IPython does is similar to calling::
563
564
564 pdb.run('execfile("YOURFILENAME")')
565 pdb.run('execfile("YOURFILENAME")')
565
566
566 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
567 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
568 (where N must be an integer). For example::
569 (where N must be an integer). For example::
569
570
570 %run -d -b40 myscript
571 %run -d -b40 myscript
571
572
572 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
573 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
574 something (not a comment or docstring) for it to stop execution.
575 something (not a comment or docstring) for it to stop execution.
575
576
576 Or you can specify a breakpoint in a different file::
577 Or you can specify a breakpoint in a different file::
577
578
578 %run -d -b myotherfile.py:20 myscript
579 %run -d -b myotherfile.py:20 myscript
579
580
580 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
581 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
582 breakpoint.
583 breakpoint.
583
584
584 Entering 'help' gives information about the use of the debugger. You
585 Entering 'help' gives information about the use of the debugger. You
585 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()"
586 at a prompt.
587 at a prompt.
587
588
588 -p
589 -p
589 run program under the control of the Python profiler module (which
590 run program under the control of the Python profiler module (which
590 prints a detailed report of execution times, function calls, etc).
591 prints a detailed report of execution times, function calls, etc).
591
592
592 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
593 profiler itself. See the docs for %prun for details.
594 profiler itself. See the docs for %prun for details.
594
595
595 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
596 IPython interactive namespace (because they remain in the namespace
597 IPython interactive namespace (because they remain in the namespace
597 where the profiler executes them).
598 where the profiler executes them).
598
599
599 Internally this triggers a call to %prun, see its documentation for
600 Internally this triggers a call to %prun, see its documentation for
600 details on the options available specifically for profiling.
601 details on the options available specifically for profiling.
601
602
602 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:
603 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,
604 just as if the commands were written on IPython prompt.
605 just as if the commands were written on IPython prompt.
605
606
606 -m
607 -m
607 specify module name to load instead of script path. Similar to
608 specify module name to load instead of script path. Similar to
608 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
609 want to combine with other %run options. Unlike the python interpreter
610 want to combine with other %run options. Unlike the python interpreter
610 only source modules are allowed no .pyc or .pyo files.
611 only source modules are allowed no .pyc or .pyo files.
611 For example::
612 For example::
612
613
613 %run -m example
614 %run -m example
614
615
615 will run the example module.
616 will run the example module.
616
617
617 -G
618 -G
618 disable shell-like glob expansion of arguments.
619 disable shell-like glob expansion of arguments.
619
620
620 """
621 """
621
622
622 # Logic to handle issue #3664
623 # Logic to handle issue #3664
623 # 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.
624 if '-m' in parameter_s and '--' not in parameter_s:
625 if '-m' in parameter_s and '--' not in parameter_s:
625 argv = shlex.split(parameter_s, posix=(os.name == 'posix'))
626 argv = shlex.split(parameter_s, posix=(os.name == 'posix'))
626 for idx, arg in enumerate(argv):
627 for idx, arg in enumerate(argv):
627 if arg and arg.startswith('-') and arg != '-':
628 if arg and arg.startswith('-') and arg != '-':
628 if arg == '-m':
629 if arg == '-m':
629 argv.insert(idx + 2, '--')
630 argv.insert(idx + 2, '--')
630 break
631 break
631 else:
632 else:
632 # Positional arg, break
633 # Positional arg, break
633 break
634 break
634 parameter_s = ' '.join(shlex.quote(arg) for arg in argv)
635 parameter_s = ' '.join(shlex.quote(arg) for arg in argv)
635
636
636 # get arguments and set sys.argv for program to be run.
637 # get arguments and set sys.argv for program to be run.
637 opts, arg_lst = self.parse_options(parameter_s,
638 opts, arg_lst = self.parse_options(parameter_s,
638 'nidtN:b:pD:l:rs:T:em:G',
639 'nidtN:b:pD:l:rs:T:em:G',
639 mode='list', list_all=1)
640 mode='list', list_all=1)
640 if "m" in opts:
641 if "m" in opts:
641 modulename = opts["m"][0]
642 modulename = opts["m"][0]
642 modpath = find_mod(modulename)
643 modpath = find_mod(modulename)
643 if modpath is None:
644 if modpath is None:
644 warn('%r is not a valid modulename on sys.path'%modulename)
645 warn('%r is not a valid modulename on sys.path'%modulename)
645 return
646 return
646 arg_lst = [modpath] + arg_lst
647 arg_lst = [modpath] + arg_lst
647 try:
648 try:
648 filename = file_finder(arg_lst[0])
649 filename = file_finder(arg_lst[0])
649 except IndexError:
650 except IndexError:
650 warn('you must provide at least a filename.')
651 warn('you must provide at least a filename.')
651 print('\n%run:\n', oinspect.getdoc(self.run))
652 print('\n%run:\n', oinspect.getdoc(self.run))
652 return
653 return
653 except IOError as e:
654 except IOError as e:
654 try:
655 try:
655 msg = str(e)
656 msg = str(e)
656 except UnicodeError:
657 except UnicodeError:
657 msg = e.message
658 msg = e.message
658 error(msg)
659 error(msg)
659 return
660 return
660
661
661 if filename.lower().endswith(('.ipy', '.ipynb')):
662 if filename.lower().endswith(('.ipy', '.ipynb')):
662 with preserve_keys(self.shell.user_ns, '__file__'):
663 with preserve_keys(self.shell.user_ns, '__file__'):
663 self.shell.user_ns['__file__'] = filename
664 self.shell.user_ns['__file__'] = filename
664 self.shell.safe_execfile_ipy(filename)
665 self.shell.safe_execfile_ipy(filename)
665 return
666 return
666
667
667 # 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
668 exit_ignore = 'e' in opts
669 exit_ignore = 'e' in opts
669
670
670 # 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
671 # were run from a system shell.
672 # were run from a system shell.
672 save_argv = sys.argv # save it for later restoring
673 save_argv = sys.argv # save it for later restoring
673
674
674 if 'G' in opts:
675 if 'G' in opts:
675 args = arg_lst[1:]
676 args = arg_lst[1:]
676 else:
677 else:
677 # tilde and glob expansion
678 # tilde and glob expansion
678 args = shellglob(map(os.path.expanduser, arg_lst[1:]))
679 args = shellglob(map(os.path.expanduser, arg_lst[1:]))
679
680
680 sys.argv = [filename] + args # put in the proper filename
681 sys.argv = [filename] + args # put in the proper filename
681
682
682 if 'n' in opts:
683 if 'n' in opts:
683 name = os.path.splitext(os.path.basename(filename))[0]
684 name = os.path.splitext(os.path.basename(filename))[0]
684 else:
685 else:
685 name = '__main__'
686 name = '__main__'
686
687
687 if 'i' in opts:
688 if 'i' in opts:
688 # Run in user's interactive namespace
689 # Run in user's interactive namespace
689 prog_ns = self.shell.user_ns
690 prog_ns = self.shell.user_ns
690 __name__save = self.shell.user_ns['__name__']
691 __name__save = self.shell.user_ns['__name__']
691 prog_ns['__name__'] = name
692 prog_ns['__name__'] = name
692 main_mod = self.shell.user_module
693 main_mod = self.shell.user_module
693
694
694 # 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
695 # set the __file__ global in the script's namespace
696 # set the __file__ global in the script's namespace
696 # TK: Is this necessary in interactive mode?
697 # TK: Is this necessary in interactive mode?
697 prog_ns['__file__'] = filename
698 prog_ns['__file__'] = filename
698 else:
699 else:
699 # Run in a fresh, empty namespace
700 # Run in a fresh, empty namespace
700
701
701 # 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
702 # exits, the python deletion mechanism doesn't zero it out
703 # exits, the python deletion mechanism doesn't zero it out
703 # (leaving dangling references). See interactiveshell for details
704 # (leaving dangling references). See interactiveshell for details
704 main_mod = self.shell.new_main_mod(filename, name)
705 main_mod = self.shell.new_main_mod(filename, name)
705 prog_ns = main_mod.__dict__
706 prog_ns = main_mod.__dict__
706
707
707 # pickle fix. See interactiveshell for an explanation. But we need to
708 # pickle fix. See interactiveshell for an explanation. But we need to
708 # 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
709 main_mod_name = prog_ns['__name__']
710 main_mod_name = prog_ns['__name__']
710
711
711 if main_mod_name == '__main__':
712 if main_mod_name == '__main__':
712 restore_main = sys.modules['__main__']
713 restore_main = sys.modules['__main__']
713 else:
714 else:
714 restore_main = False
715 restore_main = False
715
716
716 # 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
717 # every single object ever created.
718 # every single object ever created.
718 sys.modules[main_mod_name] = main_mod
719 sys.modules[main_mod_name] = main_mod
719
720
720 if 'p' in opts or 'd' in opts:
721 if 'p' in opts or 'd' in opts:
721 if 'm' in opts:
722 if 'm' in opts:
722 code = 'run_module(modulename, prog_ns)'
723 code = 'run_module(modulename, prog_ns)'
723 code_ns = {
724 code_ns = {
724 'run_module': self.shell.safe_run_module,
725 'run_module': self.shell.safe_run_module,
725 'prog_ns': prog_ns,
726 'prog_ns': prog_ns,
726 'modulename': modulename,
727 'modulename': modulename,
727 }
728 }
728 else:
729 else:
729 if 'd' in opts:
730 if 'd' in opts:
730 # allow exceptions to raise in debug mode
731 # allow exceptions to raise in debug mode
731 code = 'execfile(filename, prog_ns, raise_exceptions=True)'
732 code = 'execfile(filename, prog_ns, raise_exceptions=True)'
732 else:
733 else:
733 code = 'execfile(filename, prog_ns)'
734 code = 'execfile(filename, prog_ns)'
734 code_ns = {
735 code_ns = {
735 'execfile': self.shell.safe_execfile,
736 'execfile': self.shell.safe_execfile,
736 'prog_ns': prog_ns,
737 'prog_ns': prog_ns,
737 'filename': get_py_filename(filename),
738 'filename': get_py_filename(filename),
738 }
739 }
739
740
740 try:
741 try:
741 stats = None
742 stats = None
742 if 'p' in opts:
743 if 'p' in opts:
743 stats = self._run_with_profiler(code, opts, code_ns)
744 stats = self._run_with_profiler(code, opts, code_ns)
744 else:
745 else:
745 if 'd' in opts:
746 if 'd' in opts:
746 bp_file, bp_line = parse_breakpoint(
747 bp_file, bp_line = parse_breakpoint(
747 opts.get('b', ['1'])[0], filename)
748 opts.get('b', ['1'])[0], filename)
748 self._run_with_debugger(
749 self._run_with_debugger(
749 code, code_ns, filename, bp_line, bp_file)
750 code, code_ns, filename, bp_line, bp_file)
750 else:
751 else:
751 if 'm' in opts:
752 if 'm' in opts:
752 def run():
753 def run():
753 self.shell.safe_run_module(modulename, prog_ns)
754 self.shell.safe_run_module(modulename, prog_ns)
754 else:
755 else:
755 if runner is None:
756 if runner is None:
756 runner = self.default_runner
757 runner = self.default_runner
757 if runner is None:
758 if runner is None:
758 runner = self.shell.safe_execfile
759 runner = self.shell.safe_execfile
759
760
760 def run():
761 def run():
761 runner(filename, prog_ns, prog_ns,
762 runner(filename, prog_ns, prog_ns,
762 exit_ignore=exit_ignore)
763 exit_ignore=exit_ignore)
763
764
764 if 't' in opts:
765 if 't' in opts:
765 # timed execution
766 # timed execution
766 try:
767 try:
767 nruns = int(opts['N'][0])
768 nruns = int(opts['N'][0])
768 if nruns < 1:
769 if nruns < 1:
769 error('Number of runs must be >=1')
770 error('Number of runs must be >=1')
770 return
771 return
771 except (KeyError):
772 except (KeyError):
772 nruns = 1
773 nruns = 1
773 self._run_with_timing(run, nruns)
774 self._run_with_timing(run, nruns)
774 else:
775 else:
775 # regular execution
776 # regular execution
776 run()
777 run()
777
778
778 if 'i' in opts:
779 if 'i' in opts:
779 self.shell.user_ns['__name__'] = __name__save
780 self.shell.user_ns['__name__'] = __name__save
780 else:
781 else:
781 # update IPython interactive namespace
782 # update IPython interactive namespace
782
783
783 # Some forms of read errors on the file may mean the
784 # Some forms of read errors on the file may mean the
784 # __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
785 # worry about a possible KeyError.
786 # worry about a possible KeyError.
786 prog_ns.pop('__name__', None)
787 prog_ns.pop('__name__', None)
787
788
788 with preserve_keys(self.shell.user_ns, '__file__'):
789 with preserve_keys(self.shell.user_ns, '__file__'):
789 self.shell.user_ns.update(prog_ns)
790 self.shell.user_ns.update(prog_ns)
790 finally:
791 finally:
791 # 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
792 # 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
793 # %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
794 # at all, and similar problems have been reported before:
795 # at all, and similar problems have been reported before:
795 # 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
796 # 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
797 # 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
798 # exit.
799 # exit.
799 self.shell.user_ns['__builtins__'] = builtin_mod
800 self.shell.user_ns['__builtins__'] = builtin_mod
800
801
801 # Ensure key global structures are restored
802 # Ensure key global structures are restored
802 sys.argv = save_argv
803 sys.argv = save_argv
803 if restore_main:
804 if restore_main:
804 sys.modules['__main__'] = restore_main
805 sys.modules['__main__'] = restore_main
805 else:
806 else:
806 # Remove from sys.modules the reference to main_mod we'd
807 # Remove from sys.modules the reference to main_mod we'd
807 # added. Otherwise it will trap references to objects
808 # added. Otherwise it will trap references to objects
808 # contained therein.
809 # contained therein.
809 del sys.modules[main_mod_name]
810 del sys.modules[main_mod_name]
810
811
811 return stats
812 return stats
812
813
813 def _run_with_debugger(self, code, code_ns, filename=None,
814 def _run_with_debugger(self, code, code_ns, filename=None,
814 bp_line=None, bp_file=None):
815 bp_line=None, bp_file=None):
815 """
816 """
816 Run `code` in debugger with a break point.
817 Run `code` in debugger with a break point.
817
818
818 Parameters
819 Parameters
819 ----------
820 ----------
820 code : str
821 code : str
821 Code to execute.
822 Code to execute.
822 code_ns : dict
823 code_ns : dict
823 A namespace in which `code` is executed.
824 A namespace in which `code` is executed.
824 filename : str
825 filename : str
825 `code` is ran as if it is in `filename`.
826 `code` is ran as if it is in `filename`.
826 bp_line : int, optional
827 bp_line : int, optional
827 Line number of the break point.
828 Line number of the break point.
828 bp_file : str, optional
829 bp_file : str, optional
829 Path to the file in which break point is specified.
830 Path to the file in which break point is specified.
830 `filename` is used if not given.
831 `filename` is used if not given.
831
832
832 Raises
833 Raises
833 ------
834 ------
834 UsageError
835 UsageError
835 If the break point given by `bp_line` is not valid.
836 If the break point given by `bp_line` is not valid.
836
837
837 """
838 """
838 deb = self.shell.InteractiveTB.pdb
839 deb = self.shell.InteractiveTB.pdb
839 if not deb:
840 if not deb:
840 self.shell.InteractiveTB.pdb = self.shell.InteractiveTB.debugger_cls()
841 self.shell.InteractiveTB.pdb = self.shell.InteractiveTB.debugger_cls()
841 deb = self.shell.InteractiveTB.pdb
842 deb = self.shell.InteractiveTB.pdb
842
843
843 # 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
844 # handle it not existing. https://github.com/ipython/ipython/issues/10028
845 # handle it not existing. https://github.com/ipython/ipython/issues/10028
845 if hasattr(deb, 'curframe'):
846 if hasattr(deb, 'curframe'):
846 del deb.curframe
847 del deb.curframe
847
848
848 # reset Breakpoint state, which is moronically kept
849 # reset Breakpoint state, which is moronically kept
849 # in a class
850 # in a class
850 bdb.Breakpoint.next = 1
851 bdb.Breakpoint.next = 1
851 bdb.Breakpoint.bplist = {}
852 bdb.Breakpoint.bplist = {}
852 bdb.Breakpoint.bpbynumber = [None]
853 bdb.Breakpoint.bpbynumber = [None]
853 deb.clear_all_breaks()
854 deb.clear_all_breaks()
854 if bp_line is not None:
855 if bp_line is not None:
855 # Set an initial breakpoint to stop execution
856 # Set an initial breakpoint to stop execution
856 maxtries = 10
857 maxtries = 10
857 bp_file = bp_file or filename
858 bp_file = bp_file or filename
858 checkline = deb.checkline(bp_file, bp_line)
859 checkline = deb.checkline(bp_file, bp_line)
859 if not checkline:
860 if not checkline:
860 for bp in range(bp_line + 1, bp_line + maxtries + 1):
861 for bp in range(bp_line + 1, bp_line + maxtries + 1):
861 if deb.checkline(bp_file, bp):
862 if deb.checkline(bp_file, bp):
862 break
863 break
863 else:
864 else:
864 msg = ("\nI failed to find a valid line to set "
865 msg = ("\nI failed to find a valid line to set "
865 "a breakpoint\n"
866 "a breakpoint\n"
866 "after trying up to line: %s.\n"
867 "after trying up to line: %s.\n"
867 "Please set a valid breakpoint manually "
868 "Please set a valid breakpoint manually "
868 "with the -b option." % bp)
869 "with the -b option." % bp)
869 raise UsageError(msg)
870 raise UsageError(msg)
870 # if we find a good linenumber, set the breakpoint
871 # if we find a good linenumber, set the breakpoint
871 deb.do_break('%s:%s' % (bp_file, bp_line))
872 deb.do_break('%s:%s' % (bp_file, bp_line))
872
873
873 if filename:
874 if filename:
874 # Mimic Pdb._runscript(...)
875 # Mimic Pdb._runscript(...)
875 deb._wait_for_mainpyfile = True
876 deb._wait_for_mainpyfile = True
876 deb.mainpyfile = deb.canonic(filename)
877 deb.mainpyfile = deb.canonic(filename)
877
878
878 # Start file run
879 # Start file run
879 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)
880 try:
881 try:
881 if filename:
882 if filename:
882 # 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
883 deb._exec_filename = filename
884 deb._exec_filename = filename
884 while True:
885 while True:
885 try:
886 try:
886 deb.run(code, code_ns)
887 deb.run(code, code_ns)
887 except Restart:
888 except Restart:
888 print("Restarting")
889 print("Restarting")
889 if filename:
890 if filename:
890 deb._wait_for_mainpyfile = True
891 deb._wait_for_mainpyfile = True
891 deb.mainpyfile = deb.canonic(filename)
892 deb.mainpyfile = deb.canonic(filename)
892 continue
893 continue
893 else:
894 else:
894 break
895 break
895
896
896
897
897 except:
898 except:
898 etype, value, tb = sys.exc_info()
899 etype, value, tb = sys.exc_info()
899 # Skip three frames in the traceback: the %run one,
900 # Skip three frames in the traceback: the %run one,
900 # one inside bdb.py, and the command-line typed by the
901 # one inside bdb.py, and the command-line typed by the
901 # user (run by exec in pdb itself).
902 # user (run by exec in pdb itself).
902 self.shell.InteractiveTB(etype, value, tb, tb_offset=3)
903 self.shell.InteractiveTB(etype, value, tb, tb_offset=3)
903
904
904 @staticmethod
905 @staticmethod
905 def _run_with_timing(run, nruns):
906 def _run_with_timing(run, nruns):
906 """
907 """
907 Run function `run` and print timing information.
908 Run function `run` and print timing information.
908
909
909 Parameters
910 Parameters
910 ----------
911 ----------
911 run : callable
912 run : callable
912 Any callable object which takes no argument.
913 Any callable object which takes no argument.
913 nruns : int
914 nruns : int
914 Number of times to execute `run`.
915 Number of times to execute `run`.
915
916
916 """
917 """
917 twall0 = time.time()
918 twall0 = time.time()
918 if nruns == 1:
919 if nruns == 1:
919 t0 = clock2()
920 t0 = clock2()
920 run()
921 run()
921 t1 = clock2()
922 t1 = clock2()
922 t_usr = t1[0] - t0[0]
923 t_usr = t1[0] - t0[0]
923 t_sys = t1[1] - t0[1]
924 t_sys = t1[1] - t0[1]
924 print("\nIPython CPU timings (estimated):")
925 print("\nIPython CPU timings (estimated):")
925 print(" User : %10.2f s." % t_usr)
926 print(" User : %10.2f s." % t_usr)
926 print(" System : %10.2f s." % t_sys)
927 print(" System : %10.2f s." % t_sys)
927 else:
928 else:
928 runs = range(nruns)
929 runs = range(nruns)
929 t0 = clock2()
930 t0 = clock2()
930 for nr in runs:
931 for nr in runs:
931 run()
932 run()
932 t1 = clock2()
933 t1 = clock2()
933 t_usr = t1[0] - t0[0]
934 t_usr = t1[0] - t0[0]
934 t_sys = t1[1] - t0[1]
935 t_sys = t1[1] - t0[1]
935 print("\nIPython CPU timings (estimated):")
936 print("\nIPython CPU timings (estimated):")
936 print("Total runs performed:", nruns)
937 print("Total runs performed:", nruns)
937 print(" Times : %10s %10s" % ('Total', 'Per run'))
938 print(" Times : %10s %10s" % ('Total', 'Per run'))
938 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))
939 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))
940 twall1 = time.time()
941 twall1 = time.time()
941 print("Wall time: %10.2f s." % (twall1 - twall0))
942 print("Wall time: %10.2f s." % (twall1 - twall0))
942
943
943 @skip_doctest
944 @skip_doctest
944 @line_cell_magic
945 @line_cell_magic
945 @needs_local_scope
946 @needs_local_scope
946 def timeit(self, line='', cell=None, local_ns=None):
947 def timeit(self, line='', cell=None, local_ns=None):
947 """Time execution of a Python statement or expression
948 """Time execution of a Python statement or expression
948
949
949 Usage, in line mode:
950 Usage, in line mode:
950 %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
951 or in cell mode:
952 or in cell mode:
952 %%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
953 code
954 code
954 code...
955 code...
955
956
956 Time execution of a Python statement or expression using the timeit
957 Time execution of a Python statement or expression using the timeit
957 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:
958
959
959 - 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
960 ones can be chained with using semicolons).
961 ones can be chained with using semicolons).
961
962
962 - 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
963 (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
964 body has access to any variables created in the setup code.
965 body has access to any variables created in the setup code.
965
966
966 Options:
967 Options:
967 -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
968 provided, <N> is determined so as to get sufficient accuracy.
969 provided, <N> is determined so as to get sufficient accuracy.
969
970
970 -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
971 best result.
972 best result.
972 Default: 7
973 Default: 7
973
974
974 -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.
975 This function measures wall time.
976 This function measures wall time.
976
977
977 -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
978 Windows and measures wall time. On Unix, resource.getrusage is used
979 Windows and measures wall time. On Unix, resource.getrusage is used
979 instead and returns the CPU user time.
980 instead and returns the CPU user time.
980
981
981 -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.
982 Default: 3
983 Default: 3
983
984
984 -q: Quiet, do not print result.
985 -q: Quiet, do not print result.
985
986
986 -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
987 the result in more details.
988 the result in more details.
988
989
989
990
990 Examples
991 Examples
991 --------
992 --------
992 ::
993 ::
993
994
994 In [1]: %timeit pass
995 In [1]: %timeit pass
995 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)
996
997
997 In [2]: u = None
998 In [2]: u = None
998
999
999 In [3]: %timeit u is None
1000 In [3]: %timeit u is None
1000 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)
1001
1002
1002 In [4]: %timeit -r 4 u == None
1003 In [4]: %timeit -r 4 u == None
1003
1004
1004 In [5]: import time
1005 In [5]: import time
1005
1006
1006 In [6]: %timeit -n1 time.sleep(2)
1007 In [6]: %timeit -n1 time.sleep(2)
1007
1008
1008
1009
1009 The times reported by %timeit will be slightly higher than those
1010 The times reported by %timeit will be slightly higher than those
1010 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
1011 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
1012 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
1013 statement to import function or create variables. Generally, the bias
1014 statement to import function or create variables. Generally, the bias
1014 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
1015 those from %timeit."""
1016 those from %timeit."""
1016
1017
1017 opts, stmt = self.parse_options(line,'n:r:tcp:qo',
1018 opts, stmt = self.parse_options(line,'n:r:tcp:qo',
1018 posix=False, strict=False)
1019 posix=False, strict=False)
1019 if stmt == "" and cell is None:
1020 if stmt == "" and cell is None:
1020 return
1021 return
1021
1022
1022 timefunc = timeit.default_timer
1023 timefunc = timeit.default_timer
1023 number = int(getattr(opts, "n", 0))
1024 number = int(getattr(opts, "n", 0))
1024 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
1025 repeat = int(getattr(opts, "r", default_repeat))
1026 repeat = int(getattr(opts, "r", default_repeat))
1026 precision = int(getattr(opts, "p", 3))
1027 precision = int(getattr(opts, "p", 3))
1027 quiet = 'q' in opts
1028 quiet = 'q' in opts
1028 return_result = 'o' in opts
1029 return_result = 'o' in opts
1029 if hasattr(opts, "t"):
1030 if hasattr(opts, "t"):
1030 timefunc = time.time
1031 timefunc = time.time
1031 if hasattr(opts, "c"):
1032 if hasattr(opts, "c"):
1032 timefunc = clock
1033 timefunc = clock
1033
1034
1034 timer = Timer(timer=timefunc)
1035 timer = Timer(timer=timefunc)
1035 # 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,
1036 # 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
1037 # to the shell namespace?
1038 # to the shell namespace?
1038 transform = self.shell.transform_cell
1039 transform = self.shell.transform_cell
1039
1040
1040 if cell is None:
1041 if cell is None:
1041 # called as line magic
1042 # called as line magic
1042 ast_setup = self.shell.compile.ast_parse("pass")
1043 ast_setup = self.shell.compile.ast_parse("pass")
1043 ast_stmt = self.shell.compile.ast_parse(transform(stmt))
1044 ast_stmt = self.shell.compile.ast_parse(transform(stmt))
1044 else:
1045 else:
1045 ast_setup = self.shell.compile.ast_parse(transform(stmt))
1046 ast_setup = self.shell.compile.ast_parse(transform(stmt))
1046 ast_stmt = self.shell.compile.ast_parse(transform(cell))
1047 ast_stmt = self.shell.compile.ast_parse(transform(cell))
1047
1048
1048 ast_setup = self.shell.transform_ast(ast_setup)
1049 ast_setup = self.shell.transform_ast(ast_setup)
1049 ast_stmt = self.shell.transform_ast(ast_stmt)
1050 ast_stmt = self.shell.transform_ast(ast_stmt)
1050
1051
1051 # 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
1052 # Invalid code may become valid when put inside the function & loop,
1053 # Invalid code may become valid when put inside the function & loop,
1053 # which messes up error messages.
1054 # which messes up error messages.
1054 # https://github.com/ipython/ipython/issues/10636
1055 # https://github.com/ipython/ipython/issues/10636
1055 self.shell.compile(ast_setup, "<magic-timeit-setup>", "exec")
1056 self.shell.compile(ast_setup, "<magic-timeit-setup>", "exec")
1056 self.shell.compile(ast_stmt, "<magic-timeit-stmt>", "exec")
1057 self.shell.compile(ast_stmt, "<magic-timeit-stmt>", "exec")
1057
1058
1058 # 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
1059 # 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
1060 # without affecting the timing code.
1061 # without affecting the timing code.
1061 timeit_ast_template = ast.parse('def inner(_it, _timer):\n'
1062 timeit_ast_template = ast.parse('def inner(_it, _timer):\n'
1062 ' setup\n'
1063 ' setup\n'
1063 ' _t0 = _timer()\n'
1064 ' _t0 = _timer()\n'
1064 ' for _i in _it:\n'
1065 ' for _i in _it:\n'
1065 ' stmt\n'
1066 ' stmt\n'
1066 ' _t1 = _timer()\n'
1067 ' _t1 = _timer()\n'
1067 ' return _t1 - _t0\n')
1068 ' return _t1 - _t0\n')
1068
1069
1069 timeit_ast = TimeitTemplateFiller(ast_setup, ast_stmt).visit(timeit_ast_template)
1070 timeit_ast = TimeitTemplateFiller(ast_setup, ast_stmt).visit(timeit_ast_template)
1070 timeit_ast = ast.fix_missing_locations(timeit_ast)
1071 timeit_ast = ast.fix_missing_locations(timeit_ast)
1071
1072
1072 # Track compilation time so it can be reported if too long
1073 # Track compilation time so it can be reported if too long
1073 # Minimum time above which compilation time will be reported
1074 # Minimum time above which compilation time will be reported
1074 tc_min = 0.1
1075 tc_min = 0.1
1075
1076
1076 t0 = clock()
1077 t0 = clock()
1077 code = self.shell.compile(timeit_ast, "<magic-timeit>", "exec")
1078 code = self.shell.compile(timeit_ast, "<magic-timeit>", "exec")
1078 tc = clock()-t0
1079 tc = clock()-t0
1079
1080
1080 ns = {}
1081 ns = {}
1081 glob = self.shell.user_ns
1082 glob = self.shell.user_ns
1082 # 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.
1083 if local_ns is not None:
1084 if local_ns is not None:
1084 conflict_globs = {}
1085 conflict_globs = {}
1085 for var_name, var_val in glob.items():
1086 for var_name, var_val in glob.items():
1086 if var_name in local_ns:
1087 if var_name in local_ns:
1087 conflict_globs[var_name] = var_val
1088 conflict_globs[var_name] = var_val
1088 glob.update(local_ns)
1089 glob.update(local_ns)
1089
1090
1090 exec(code, glob, ns)
1091 exec(code, glob, ns)
1091 timer.inner = ns["inner"]
1092 timer.inner = ns["inner"]
1092
1093
1093 # 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
1094 # best and worst timings.
1095 # best and worst timings.
1095 # Issue: https://github.com/ipython/ipython/issues/6471
1096 # Issue: https://github.com/ipython/ipython/issues/6471
1096 if number == 0:
1097 if number == 0:
1097 # determine number so that 0.2 <= total time < 2.0
1098 # determine number so that 0.2 <= total time < 2.0
1098 for index in range(0, 10):
1099 for index in range(0, 10):
1099 number = 10 ** index
1100 number = 10 ** index
1100 time_number = timer.timeit(number)
1101 time_number = timer.timeit(number)
1101 if time_number >= 0.2:
1102 if time_number >= 0.2:
1102 break
1103 break
1103
1104
1104 all_runs = timer.repeat(repeat, number)
1105 all_runs = timer.repeat(repeat, number)
1105 best = min(all_runs) / number
1106 best = min(all_runs) / number
1106 worst = max(all_runs) / number
1107 worst = max(all_runs) / number
1107 timeit_result = TimeitResult(number, repeat, best, worst, all_runs, tc, precision)
1108 timeit_result = TimeitResult(number, repeat, best, worst, all_runs, tc, precision)
1108
1109
1109 # Restore global vars from conflict_globs
1110 # Restore global vars from conflict_globs
1110 if local_ns is not None:
1111 if local_ns is not None:
1111 if len(conflict_globs) > 0:
1112 if len(conflict_globs) > 0:
1112 glob.update(conflict_globs)
1113 glob.update(conflict_globs)
1113
1114
1114 if not quiet :
1115 if not quiet :
1115 # Check best timing is greater than zero to avoid a
1116 # Check best timing is greater than zero to avoid a
1116 # ZeroDivisionError.
1117 # ZeroDivisionError.
1117 # In cases where the slowest timing is lesser than a micosecond
1118 # In cases where the slowest timing is lesser than a micosecond
1118 # we assume that it does not really matter if the fastest
1119 # we assume that it does not really matter if the fastest
1119 # timing is 4 times faster than the slowest timing or not.
1120 # timing is 4 times faster than the slowest timing or not.
1120 if worst > 4 * best and best > 0 and worst > 1e-6:
1121 if worst > 4 * best and best > 0 and worst > 1e-6:
1121 print("The slowest run took %0.2f times longer than the "
1122 print("The slowest run took %0.2f times longer than the "
1122 "fastest. This could mean that an intermediate result "
1123 "fastest. This could mean that an intermediate result "
1123 "is being cached." % (worst / best))
1124 "is being cached." % (worst / best))
1124
1125
1125 print( timeit_result )
1126 print( timeit_result )
1126
1127
1127 if tc > tc_min:
1128 if tc > tc_min:
1128 print("Compiler time: %.2f s" % tc)
1129 print("Compiler time: %.2f s" % tc)
1129 if return_result:
1130 if return_result:
1130 return timeit_result
1131 return timeit_result
1131
1132
1132 @skip_doctest
1133 @skip_doctest
1133 @needs_local_scope
1134 @needs_local_scope
1134 @line_cell_magic
1135 @line_cell_magic
1135 def time(self,line='', cell=None, local_ns=None):
1136 def time(self,line='', cell=None, local_ns=None):
1136 """Time execution of a Python statement or expression.
1137 """Time execution of a Python statement or expression.
1137
1138
1138 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
1139 expression (if any) is returned. Note that under Win32, system time
1140 expression (if any) is returned. Note that under Win32, system time
1140 is always reported as 0, since it can not be measured.
1141 is always reported as 0, since it can not be measured.
1141
1142
1142 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:
1143
1144
1144 - 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
1145 ones can be chained with using semicolons).
1146 ones can be chained with using semicolons).
1146
1147
1147 - In cell mode, you can time the cell body (a directly
1148 - In cell mode, you can time the cell body (a directly
1148 following statement raises an error).
1149 following statement raises an error).
1149
1150
1150 This function provides very basic timing functionality. Use the timeit
1151 This function provides very basic timing functionality. Use the timeit
1151 magic for more control over the measurement.
1152 magic for more control over the measurement.
1152
1153
1153 Examples
1154 Examples
1154 --------
1155 --------
1155 ::
1156 ::
1156
1157
1157 In [1]: %time 2**128
1158 In [1]: %time 2**128
1158 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
1159 Wall time: 0.00
1160 Wall time: 0.00
1160 Out[1]: 340282366920938463463374607431768211456L
1161 Out[1]: 340282366920938463463374607431768211456L
1161
1162
1162 In [2]: n = 1000000
1163 In [2]: n = 1000000
1163
1164
1164 In [3]: %time sum(range(n))
1165 In [3]: %time sum(range(n))
1165 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
1166 Wall time: 1.37
1167 Wall time: 1.37
1167 Out[3]: 499999500000L
1168 Out[3]: 499999500000L
1168
1169
1169 In [4]: %time print 'hello world'
1170 In [4]: %time print 'hello world'
1170 hello world
1171 hello world
1171 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
1172 Wall time: 0.00
1173 Wall time: 0.00
1173
1174
1174 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
1175 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
1176 actual exponentiation is done by Python at compilation time, so while
1177 actual exponentiation is done by Python at compilation time, so while
1177 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
1178 time is purely due to the compilation:
1179 time is purely due to the compilation:
1179
1180
1180 In [5]: %time 3**9999;
1181 In [5]: %time 3**9999;
1181 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
1182 Wall time: 0.00 s
1183 Wall time: 0.00 s
1183
1184
1184 In [6]: %time 3**999999;
1185 In [6]: %time 3**999999;
1185 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
1186 Wall time: 0.00 s
1187 Wall time: 0.00 s
1187 Compiler : 0.78 s
1188 Compiler : 0.78 s
1188 """
1189 """
1189
1190
1190 # fail immediately if the given expression can't be compiled
1191 # fail immediately if the given expression can't be compiled
1191
1192
1192 if line and cell:
1193 if line and cell:
1193 raise UsageError("Can't use statement directly after '%%time'!")
1194 raise UsageError("Can't use statement directly after '%%time'!")
1194
1195
1195 if cell:
1196 if cell:
1196 expr = self.shell.transform_cell(cell)
1197 expr = self.shell.transform_cell(cell)
1197 else:
1198 else:
1198 expr = self.shell.transform_cell(line)
1199 expr = self.shell.transform_cell(line)
1199
1200
1200 # Minimum time above which parse time will be reported
1201 # Minimum time above which parse time will be reported
1201 tp_min = 0.1
1202 tp_min = 0.1
1202
1203
1203 t0 = clock()
1204 t0 = clock()
1204 expr_ast = self.shell.compile.ast_parse(expr)
1205 expr_ast = self.shell.compile.ast_parse(expr)
1205 tp = clock()-t0
1206 tp = clock()-t0
1206
1207
1207 # Apply AST transformations
1208 # Apply AST transformations
1208 expr_ast = self.shell.transform_ast(expr_ast)
1209 expr_ast = self.shell.transform_ast(expr_ast)
1209
1210
1210 # Minimum time above which compilation time will be reported
1211 # Minimum time above which compilation time will be reported
1211 tc_min = 0.1
1212 tc_min = 0.1
1212
1213
1213 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):
1214 mode = 'eval'
1215 mode = 'eval'
1215 source = '<timed eval>'
1216 source = '<timed eval>'
1216 expr_ast = ast.Expression(expr_ast.body[0].value)
1217 expr_ast = ast.Expression(expr_ast.body[0].value)
1217 else:
1218 else:
1218 mode = 'exec'
1219 mode = 'exec'
1219 source = '<timed exec>'
1220 source = '<timed exec>'
1220 t0 = clock()
1221 t0 = clock()
1221 code = self.shell.compile(expr_ast, source, mode)
1222 code = self.shell.compile(expr_ast, source, mode)
1222 tc = clock()-t0
1223 tc = clock()-t0
1223
1224
1224 # skew measurement as little as possible
1225 # skew measurement as little as possible
1225 glob = self.shell.user_ns
1226 glob = self.shell.user_ns
1226 wtime = time.time
1227 wtime = time.time
1227 # time execution
1228 # time execution
1228 wall_st = wtime()
1229 wall_st = wtime()
1229 if mode=='eval':
1230 if mode=='eval':
1230 st = clock2()
1231 st = clock2()
1231 try:
1232 try:
1232 out = eval(code, glob, local_ns)
1233 out = eval(code, glob, local_ns)
1233 except:
1234 except:
1234 self.shell.showtraceback()
1235 self.shell.showtraceback()
1235 return
1236 return
1236 end = clock2()
1237 end = clock2()
1237 else:
1238 else:
1238 st = clock2()
1239 st = clock2()
1239 try:
1240 try:
1240 exec(code, glob, local_ns)
1241 exec(code, glob, local_ns)
1241 except:
1242 except:
1242 self.shell.showtraceback()
1243 self.shell.showtraceback()
1243 return
1244 return
1244 end = clock2()
1245 end = clock2()
1245 out = None
1246 out = None
1246 wall_end = wtime()
1247 wall_end = wtime()
1247 # Compute actual times and report
1248 # Compute actual times and report
1248 wall_time = wall_end-wall_st
1249 wall_time = wall_end-wall_st
1249 cpu_user = end[0]-st[0]
1250 cpu_user = end[0]-st[0]
1250 cpu_sys = end[1]-st[1]
1251 cpu_sys = end[1]-st[1]
1251 cpu_tot = cpu_user+cpu_sys
1252 cpu_tot = cpu_user+cpu_sys
1252 # 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
1253 if sys.platform != 'win32':
1254 if sys.platform != 'win32':
1254 print("CPU times: user %s, sys: %s, total: %s" % \
1255 print("CPU times: user %s, sys: %s, total: %s" % \
1255 (_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)))
1256 print("Wall time: %s" % _format_time(wall_time))
1257 print("Wall time: %s" % _format_time(wall_time))
1257 if tc > tc_min:
1258 if tc > tc_min:
1258 print("Compiler : %s" % _format_time(tc))
1259 print("Compiler : %s" % _format_time(tc))
1259 if tp > tp_min:
1260 if tp > tp_min:
1260 print("Parser : %s" % _format_time(tp))
1261 print("Parser : %s" % _format_time(tp))
1261 return out
1262 return out
1262
1263
1263 @skip_doctest
1264 @skip_doctest
1264 @line_magic
1265 @line_magic
1265 def macro(self, parameter_s=''):
1266 def macro(self, parameter_s=''):
1266 """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,
1267 filenames or string objects.
1268 filenames or string objects.
1268
1269
1269 Usage:\\
1270 Usage:\\
1270 %macro [options] name n1-n2 n3-n4 ... n5 .. n6 ...
1271 %macro [options] name n1-n2 n3-n4 ... n5 .. n6 ...
1271
1272
1272 Options:
1273 Options:
1273
1274
1274 -r: use 'raw' input. By default, the 'processed' history is used,
1275 -r: use 'raw' input. By default, the 'processed' history is used,
1275 so that magics are loaded in their transformed version to valid
1276 so that magics are loaded in their transformed version to valid
1276 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
1277 command line is used instead.
1278 command line is used instead.
1278
1279
1279 -q: quiet macro definition. By default, a tag line is printed
1280 -q: quiet macro definition. By default, a tag line is printed
1280 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
1281 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
1282 is produced once the macro is created.
1283 is produced once the macro is created.
1283
1284
1284 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
1285 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
1286 above) from your input history into a single string. This variable
1287 above) from your input history into a single string. This variable
1287 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
1288 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
1289 executes.
1290 executes.
1290
1291
1291 The syntax for indicating input ranges is described in %history.
1292 The syntax for indicating input ranges is described in %history.
1292
1293
1293 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
1294 notation, where N:M means numbers N through M-1.
1295 notation, where N:M means numbers N through M-1.
1295
1296
1296 For example, if your history contains (print using %hist -n )::
1297 For example, if your history contains (print using %hist -n )::
1297
1298
1298 44: x=1
1299 44: x=1
1299 45: y=3
1300 45: y=3
1300 46: z=x+y
1301 46: z=x+y
1301 47: print x
1302 47: print x
1302 48: a=5
1303 48: a=5
1303 49: print 'x',x,'y',y
1304 49: print 'x',x,'y',y
1304
1305
1305 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
1306 called my_macro with::
1307 called my_macro with::
1307
1308
1308 In [55]: %macro my_macro 44-47 49
1309 In [55]: %macro my_macro 44-47 49
1309
1310
1310 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
1311 in one pass.
1312 in one pass.
1312
1313
1313 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
1314 number can appear multiple times. You can assemble macros with any
1315 number can appear multiple times. You can assemble macros with any
1315 lines from your input history in any order.
1316 lines from your input history in any order.
1316
1317
1317 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,
1318 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
1319 code instead of printing them when you type their name.
1320 code instead of printing them when you type their name.
1320
1321
1321 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::
1322
1323
1323 print macro_name
1324 print macro_name
1324
1325
1325 """
1326 """
1326 opts,args = self.parse_options(parameter_s,'rq',mode='list')
1327 opts,args = self.parse_options(parameter_s,'rq',mode='list')
1327 if not args: # List existing macros
1328 if not args: # List existing macros
1328 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))
1329 if len(args) == 1:
1330 if len(args) == 1:
1330 raise UsageError(
1331 raise UsageError(
1331 "%macro insufficient args; usage '%macro name n1-n2 n3-4...")
1332 "%macro insufficient args; usage '%macro name n1-n2 n3-4...")
1332 name, codefrom = args[0], " ".join(args[1:])
1333 name, codefrom = args[0], " ".join(args[1:])
1333
1334
1334 #print 'rng',ranges # dbg
1335 #print 'rng',ranges # dbg
1335 try:
1336 try:
1336 lines = self.shell.find_user_code(codefrom, 'r' in opts)
1337 lines = self.shell.find_user_code(codefrom, 'r' in opts)
1337 except (ValueError, TypeError) as e:
1338 except (ValueError, TypeError) as e:
1338 print(e.args[0])
1339 print(e.args[0])
1339 return
1340 return
1340 macro = Macro(lines)
1341 macro = Macro(lines)
1341 self.shell.define_macro(name, macro)
1342 self.shell.define_macro(name, macro)
1342 if not ( 'q' in opts) :
1343 if not ( 'q' in opts) :
1343 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)
1344 print('=== Macro contents: ===')
1345 print('=== Macro contents: ===')
1345 print(macro, end=' ')
1346 print(macro, end=' ')
1346
1347
1347 @magic_arguments.magic_arguments()
1348 @magic_arguments.magic_arguments()
1348 @magic_arguments.argument('output', type=str, default='', nargs='?',
1349 @magic_arguments.argument('output', type=str, default='', nargs='?',
1349 help="""The name of the variable in which to store output.
1350 help="""The name of the variable in which to store output.
1350 This is a utils.io.CapturedIO object with stdout/err attributes
1351 This is a utils.io.CapturedIO object with stdout/err attributes
1351 for the text of the captured output.
1352 for the text of the captured output.
1352
1353
1353 CapturedOutput also has a show() method for displaying the output,
1354 CapturedOutput also has a show() method for displaying the output,
1354 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
1355 output.
1356 output.
1356
1357
1357 If unspecified, captured output is discarded.
1358 If unspecified, captured output is discarded.
1358 """
1359 """
1359 )
1360 )
1360 @magic_arguments.argument('--no-stderr', action="store_true",
1361 @magic_arguments.argument('--no-stderr', action="store_true",
1361 help="""Don't capture stderr."""
1362 help="""Don't capture stderr."""
1362 )
1363 )
1363 @magic_arguments.argument('--no-stdout', action="store_true",
1364 @magic_arguments.argument('--no-stdout', action="store_true",
1364 help="""Don't capture stdout."""
1365 help="""Don't capture stdout."""
1365 )
1366 )
1366 @magic_arguments.argument('--no-display', action="store_true",
1367 @magic_arguments.argument('--no-display', action="store_true",
1367 help="""Don't capture IPython's rich display."""
1368 help="""Don't capture IPython's rich display."""
1368 )
1369 )
1369 @cell_magic
1370 @cell_magic
1370 def capture(self, line, cell):
1371 def capture(self, line, cell):
1371 """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."""
1372 args = magic_arguments.parse_argstring(self.capture, line)
1373 args = magic_arguments.parse_argstring(self.capture, line)
1373 out = not args.no_stdout
1374 out = not args.no_stdout
1374 err = not args.no_stderr
1375 err = not args.no_stderr
1375 disp = not args.no_display
1376 disp = not args.no_display
1376 with capture_output(out, err, disp) as io:
1377 with capture_output(out, err, disp) as io:
1377 self.shell.run_cell(cell)
1378 self.shell.run_cell(cell)
1378 if args.output:
1379 if args.output:
1379 self.shell.user_ns[args.output] = io
1380 self.shell.user_ns[args.output] = io
1380
1381
1381 def parse_breakpoint(text, current_file):
1382 def parse_breakpoint(text, current_file):
1382 '''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'''
1383 colon = text.find(':')
1384 colon = text.find(':')
1384 if colon == -1:
1385 if colon == -1:
1385 return current_file, int(text)
1386 return current_file, int(text)
1386 else:
1387 else:
1387 return text[:colon], int(text[colon+1:])
1388 return text[:colon], int(text[colon+1:])
1388
1389
1389 def _format_time(timespan, precision=3):
1390 def _format_time(timespan, precision=3):
1390 """Formats the timespan in a human readable form"""
1391 """Formats the timespan in a human readable form"""
1391
1392
1392 if timespan >= 60.0:
1393 if timespan >= 60.0:
1393 # 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
1394 # Idea from http://snipplr.com/view/5713/
1395 # Idea from http://snipplr.com/view/5713/
1395 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)]
1396 time = []
1397 time = []
1397 leftover = timespan
1398 leftover = timespan
1398 for suffix, length in parts:
1399 for suffix, length in parts:
1399 value = int(leftover / length)
1400 value = int(leftover / length)
1400 if value > 0:
1401 if value > 0:
1401 leftover = leftover % length
1402 leftover = leftover % length
1402 time.append(u'%s%s' % (str(value), suffix))
1403 time.append(u'%s%s' % (str(value), suffix))
1403 if leftover < 1:
1404 if leftover < 1:
1404 break
1405 break
1405 return " ".join(time)
1406 return " ".join(time)
1406
1407
1407
1408
1408 # Unfortunately the unicode 'micro' symbol can cause problems in
1409 # Unfortunately the unicode 'micro' symbol can cause problems in
1409 # certain terminals.
1410 # certain terminals.
1410 # See bug: https://bugs.launchpad.net/ipython/+bug/348466
1411 # See bug: https://bugs.launchpad.net/ipython/+bug/348466
1411 # 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
1412 # 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.
1413 units = [u"s", u"ms",u'us',"ns"] # the save value
1414 units = [u"s", u"ms",u'us',"ns"] # the save value
1414 if hasattr(sys.stdout, 'encoding') and sys.stdout.encoding:
1415 if hasattr(sys.stdout, 'encoding') and sys.stdout.encoding:
1415 try:
1416 try:
1416 u'\xb5'.encode(sys.stdout.encoding)
1417 u'\xb5'.encode(sys.stdout.encoding)
1417 units = [u"s", u"ms",u'\xb5s',"ns"]
1418 units = [u"s", u"ms",u'\xb5s',"ns"]
1418 except:
1419 except:
1419 pass
1420 pass
1420 scaling = [1, 1e3, 1e6, 1e9]
1421 scaling = [1, 1e3, 1e6, 1e9]
1421
1422
1422 if timespan > 0.0:
1423 if timespan > 0.0:
1423 order = min(-int(math.floor(math.log10(timespan)) // 3), 3)
1424 order = min(-int(math.floor(math.log10(timespan)) // 3), 3)
1424 else:
1425 else:
1425 order = 3
1426 order = 3
1426 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