##// END OF EJS Templates
fix execution count when custom exception is caught during compile...
Min RK -
Show More
@@ -1,3330 +1,3332 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2 """Main IPython class."""
2 """Main IPython class."""
3
3
4 #-----------------------------------------------------------------------------
4 #-----------------------------------------------------------------------------
5 # Copyright (C) 2001 Janko Hauser <jhauser@zscout.de>
5 # Copyright (C) 2001 Janko Hauser <jhauser@zscout.de>
6 # Copyright (C) 2001-2007 Fernando Perez. <fperez@colorado.edu>
6 # Copyright (C) 2001-2007 Fernando Perez. <fperez@colorado.edu>
7 # Copyright (C) 2008-2011 The IPython Development Team
7 # Copyright (C) 2008-2011 The IPython Development Team
8 #
8 #
9 # Distributed under the terms of the BSD License. The full license is in
9 # Distributed under the terms of the BSD License. The full license is in
10 # the file COPYING, distributed as part of this software.
10 # the file COPYING, distributed as part of this software.
11 #-----------------------------------------------------------------------------
11 #-----------------------------------------------------------------------------
12
12
13
13
14 import abc
14 import abc
15 import ast
15 import ast
16 import atexit
16 import atexit
17 import builtins as builtin_mod
17 import builtins as builtin_mod
18 import functools
18 import functools
19 import os
19 import os
20 import re
20 import re
21 import runpy
21 import runpy
22 import sys
22 import sys
23 import tempfile
23 import tempfile
24 import traceback
24 import traceback
25 import types
25 import types
26 import subprocess
26 import subprocess
27 import warnings
27 import warnings
28 from io import open as io_open
28 from io import open as io_open
29
29
30 from pickleshare import PickleShareDB
30 from pickleshare import PickleShareDB
31
31
32 from traitlets.config.configurable import SingletonConfigurable
32 from traitlets.config.configurable import SingletonConfigurable
33 from IPython.core import oinspect
33 from IPython.core import oinspect
34 from IPython.core import magic
34 from IPython.core import magic
35 from IPython.core import page
35 from IPython.core import page
36 from IPython.core import prefilter
36 from IPython.core import prefilter
37 from IPython.core import ultratb
37 from IPython.core import ultratb
38 from IPython.core.alias import Alias, AliasManager
38 from IPython.core.alias import Alias, AliasManager
39 from IPython.core.autocall import ExitAutocall
39 from IPython.core.autocall import ExitAutocall
40 from IPython.core.builtin_trap import BuiltinTrap
40 from IPython.core.builtin_trap import BuiltinTrap
41 from IPython.core.events import EventManager, available_events
41 from IPython.core.events import EventManager, available_events
42 from IPython.core.compilerop import CachingCompiler, check_linecache_ipython
42 from IPython.core.compilerop import CachingCompiler, check_linecache_ipython
43 from IPython.core.debugger import Pdb
43 from IPython.core.debugger import Pdb
44 from IPython.core.display_trap import DisplayTrap
44 from IPython.core.display_trap import DisplayTrap
45 from IPython.core.displayhook import DisplayHook
45 from IPython.core.displayhook import DisplayHook
46 from IPython.core.displaypub import DisplayPublisher
46 from IPython.core.displaypub import DisplayPublisher
47 from IPython.core.error import InputRejected, UsageError
47 from IPython.core.error import InputRejected, UsageError
48 from IPython.core.extensions import ExtensionManager
48 from IPython.core.extensions import ExtensionManager
49 from IPython.core.formatters import DisplayFormatter
49 from IPython.core.formatters import DisplayFormatter
50 from IPython.core.history import HistoryManager
50 from IPython.core.history import HistoryManager
51 from IPython.core.inputsplitter import ESC_MAGIC, ESC_MAGIC2
51 from IPython.core.inputsplitter import ESC_MAGIC, ESC_MAGIC2
52 from IPython.core.logger import Logger
52 from IPython.core.logger import Logger
53 from IPython.core.macro import Macro
53 from IPython.core.macro import Macro
54 from IPython.core.payload import PayloadManager
54 from IPython.core.payload import PayloadManager
55 from IPython.core.prefilter import PrefilterManager
55 from IPython.core.prefilter import PrefilterManager
56 from IPython.core.profiledir import ProfileDir
56 from IPython.core.profiledir import ProfileDir
57 from IPython.core.usage import default_banner
57 from IPython.core.usage import default_banner
58 from IPython.display import display
58 from IPython.display import display
59 from IPython.testing.skipdoctest import skip_doctest
59 from IPython.testing.skipdoctest import skip_doctest
60 from IPython.utils import PyColorize
60 from IPython.utils import PyColorize
61 from IPython.utils import io
61 from IPython.utils import io
62 from IPython.utils import py3compat
62 from IPython.utils import py3compat
63 from IPython.utils import openpy
63 from IPython.utils import openpy
64 from IPython.utils.decorators import undoc
64 from IPython.utils.decorators import undoc
65 from IPython.utils.io import ask_yes_no
65 from IPython.utils.io import ask_yes_no
66 from IPython.utils.ipstruct import Struct
66 from IPython.utils.ipstruct import Struct
67 from IPython.paths import get_ipython_dir
67 from IPython.paths import get_ipython_dir
68 from IPython.utils.path import get_home_dir, get_py_filename, ensure_dir_exists
68 from IPython.utils.path import get_home_dir, get_py_filename, ensure_dir_exists
69 from IPython.utils.process import system, getoutput
69 from IPython.utils.process import system, getoutput
70 from IPython.utils.strdispatch import StrDispatch
70 from IPython.utils.strdispatch import StrDispatch
71 from IPython.utils.syspathcontext import prepended_to_syspath
71 from IPython.utils.syspathcontext import prepended_to_syspath
72 from IPython.utils.text import format_screen, LSString, SList, DollarFormatter
72 from IPython.utils.text import format_screen, LSString, SList, DollarFormatter
73 from IPython.utils.tempdir import TemporaryDirectory
73 from IPython.utils.tempdir import TemporaryDirectory
74 from traitlets import (
74 from traitlets import (
75 Integer, Bool, CaselessStrEnum, Enum, List, Dict, Unicode, Instance, Type,
75 Integer, Bool, CaselessStrEnum, Enum, List, Dict, Unicode, Instance, Type,
76 observe, default,
76 observe, default,
77 )
77 )
78 from warnings import warn
78 from warnings import warn
79 from logging import error
79 from logging import error
80 import IPython.core.hooks
80 import IPython.core.hooks
81
81
82 from typing import List as ListType
82 from typing import List as ListType
83 from ast import AST
83 from ast import AST
84
84
85 # NoOpContext is deprecated, but ipykernel imports it from here.
85 # NoOpContext is deprecated, but ipykernel imports it from here.
86 # See https://github.com/ipython/ipykernel/issues/157
86 # See https://github.com/ipython/ipykernel/issues/157
87 from IPython.utils.contexts import NoOpContext
87 from IPython.utils.contexts import NoOpContext
88
88
89 try:
89 try:
90 import docrepr.sphinxify as sphx
90 import docrepr.sphinxify as sphx
91
91
92 def sphinxify(doc):
92 def sphinxify(doc):
93 with TemporaryDirectory() as dirname:
93 with TemporaryDirectory() as dirname:
94 return {
94 return {
95 'text/html': sphx.sphinxify(doc, dirname),
95 'text/html': sphx.sphinxify(doc, dirname),
96 'text/plain': doc
96 'text/plain': doc
97 }
97 }
98 except ImportError:
98 except ImportError:
99 sphinxify = None
99 sphinxify = None
100
100
101
101
102 class ProvisionalWarning(DeprecationWarning):
102 class ProvisionalWarning(DeprecationWarning):
103 """
103 """
104 Warning class for unstable features
104 Warning class for unstable features
105 """
105 """
106 pass
106 pass
107
107
108 if sys.version_info > (3,6):
108 if sys.version_info > (3,6):
109 _assign_nodes = (ast.AugAssign, ast.AnnAssign, ast.Assign)
109 _assign_nodes = (ast.AugAssign, ast.AnnAssign, ast.Assign)
110 _single_targets_nodes = (ast.AugAssign, ast.AnnAssign)
110 _single_targets_nodes = (ast.AugAssign, ast.AnnAssign)
111 else:
111 else:
112 _assign_nodes = (ast.AugAssign, ast.Assign )
112 _assign_nodes = (ast.AugAssign, ast.Assign )
113 _single_targets_nodes = (ast.AugAssign, )
113 _single_targets_nodes = (ast.AugAssign, )
114
114
115 #-----------------------------------------------------------------------------
115 #-----------------------------------------------------------------------------
116 # Globals
116 # Globals
117 #-----------------------------------------------------------------------------
117 #-----------------------------------------------------------------------------
118
118
119 # compiled regexps for autoindent management
119 # compiled regexps for autoindent management
120 dedent_re = re.compile(r'^\s+raise|^\s+return|^\s+pass')
120 dedent_re = re.compile(r'^\s+raise|^\s+return|^\s+pass')
121
121
122 #-----------------------------------------------------------------------------
122 #-----------------------------------------------------------------------------
123 # Utilities
123 # Utilities
124 #-----------------------------------------------------------------------------
124 #-----------------------------------------------------------------------------
125
125
126 @undoc
126 @undoc
127 def softspace(file, newvalue):
127 def softspace(file, newvalue):
128 """Copied from code.py, to remove the dependency"""
128 """Copied from code.py, to remove the dependency"""
129
129
130 oldvalue = 0
130 oldvalue = 0
131 try:
131 try:
132 oldvalue = file.softspace
132 oldvalue = file.softspace
133 except AttributeError:
133 except AttributeError:
134 pass
134 pass
135 try:
135 try:
136 file.softspace = newvalue
136 file.softspace = newvalue
137 except (AttributeError, TypeError):
137 except (AttributeError, TypeError):
138 # "attribute-less object" or "read-only attributes"
138 # "attribute-less object" or "read-only attributes"
139 pass
139 pass
140 return oldvalue
140 return oldvalue
141
141
142 @undoc
142 @undoc
143 def no_op(*a, **kw):
143 def no_op(*a, **kw):
144 pass
144 pass
145
145
146
146
147 class SpaceInInput(Exception): pass
147 class SpaceInInput(Exception): pass
148
148
149
149
150 def get_default_colors():
150 def get_default_colors():
151 "DEPRECATED"
151 "DEPRECATED"
152 warn('get_default_color is deprecated since IPython 5.0, and returns `Neutral` on all platforms.',
152 warn('get_default_color is deprecated since IPython 5.0, and returns `Neutral` on all platforms.',
153 DeprecationWarning, stacklevel=2)
153 DeprecationWarning, stacklevel=2)
154 return 'Neutral'
154 return 'Neutral'
155
155
156
156
157 class SeparateUnicode(Unicode):
157 class SeparateUnicode(Unicode):
158 r"""A Unicode subclass to validate separate_in, separate_out, etc.
158 r"""A Unicode subclass to validate separate_in, separate_out, etc.
159
159
160 This is a Unicode based trait that converts '0'->'' and ``'\\n'->'\n'``.
160 This is a Unicode based trait that converts '0'->'' and ``'\\n'->'\n'``.
161 """
161 """
162
162
163 def validate(self, obj, value):
163 def validate(self, obj, value):
164 if value == '0': value = ''
164 if value == '0': value = ''
165 value = value.replace('\\n','\n')
165 value = value.replace('\\n','\n')
166 return super(SeparateUnicode, self).validate(obj, value)
166 return super(SeparateUnicode, self).validate(obj, value)
167
167
168
168
169 @undoc
169 @undoc
170 class DummyMod(object):
170 class DummyMod(object):
171 """A dummy module used for IPython's interactive module when
171 """A dummy module used for IPython's interactive module when
172 a namespace must be assigned to the module's __dict__."""
172 a namespace must be assigned to the module's __dict__."""
173 pass
173 pass
174
174
175
175
176 class ExecutionInfo(object):
176 class ExecutionInfo(object):
177 """The arguments used for a call to :meth:`InteractiveShell.run_cell`
177 """The arguments used for a call to :meth:`InteractiveShell.run_cell`
178
178
179 Stores information about what is going to happen.
179 Stores information about what is going to happen.
180 """
180 """
181 raw_cell = None
181 raw_cell = None
182 store_history = False
182 store_history = False
183 silent = False
183 silent = False
184 shell_futures = True
184 shell_futures = True
185
185
186 def __init__(self, raw_cell, store_history, silent, shell_futures):
186 def __init__(self, raw_cell, store_history, silent, shell_futures):
187 self.raw_cell = raw_cell
187 self.raw_cell = raw_cell
188 self.store_history = store_history
188 self.store_history = store_history
189 self.silent = silent
189 self.silent = silent
190 self.shell_futures = shell_futures
190 self.shell_futures = shell_futures
191
191
192 def __repr__(self):
192 def __repr__(self):
193 name = self.__class__.__qualname__
193 name = self.__class__.__qualname__
194 raw_cell = ((self.raw_cell[:50] + '..')
194 raw_cell = ((self.raw_cell[:50] + '..')
195 if len(self.raw_cell) > 50 else self.raw_cell)
195 if len(self.raw_cell) > 50 else self.raw_cell)
196 return '<%s object at %x, raw_cell="%s" store_history=%s silent=%s shell_futures=%s>' %\
196 return '<%s object at %x, raw_cell="%s" store_history=%s silent=%s shell_futures=%s>' %\
197 (name, id(self), raw_cell, self.store_history, self.silent, self.shell_futures)
197 (name, id(self), raw_cell, self.store_history, self.silent, self.shell_futures)
198
198
199
199
200 class ExecutionResult(object):
200 class ExecutionResult(object):
201 """The result of a call to :meth:`InteractiveShell.run_cell`
201 """The result of a call to :meth:`InteractiveShell.run_cell`
202
202
203 Stores information about what took place.
203 Stores information about what took place.
204 """
204 """
205 execution_count = None
205 execution_count = None
206 error_before_exec = None
206 error_before_exec = None
207 error_in_exec = None
207 error_in_exec = None
208 info = None
208 info = None
209 result = None
209 result = None
210
210
211 def __init__(self, info):
211 def __init__(self, info):
212 self.info = info
212 self.info = info
213
213
214 @property
214 @property
215 def success(self):
215 def success(self):
216 return (self.error_before_exec is None) and (self.error_in_exec is None)
216 return (self.error_before_exec is None) and (self.error_in_exec is None)
217
217
218 def raise_error(self):
218 def raise_error(self):
219 """Reraises error if `success` is `False`, otherwise does nothing"""
219 """Reraises error if `success` is `False`, otherwise does nothing"""
220 if self.error_before_exec is not None:
220 if self.error_before_exec is not None:
221 raise self.error_before_exec
221 raise self.error_before_exec
222 if self.error_in_exec is not None:
222 if self.error_in_exec is not None:
223 raise self.error_in_exec
223 raise self.error_in_exec
224
224
225 def __repr__(self):
225 def __repr__(self):
226 name = self.__class__.__qualname__
226 name = self.__class__.__qualname__
227 return '<%s object at %x, execution_count=%s error_before_exec=%s error_in_exec=%s info=%s result=%s>' %\
227 return '<%s object at %x, execution_count=%s error_before_exec=%s error_in_exec=%s info=%s result=%s>' %\
228 (name, id(self), self.execution_count, self.error_before_exec, self.error_in_exec, repr(self.info), repr(self.result))
228 (name, id(self), self.execution_count, self.error_before_exec, self.error_in_exec, repr(self.info), repr(self.result))
229
229
230
230
231 class InteractiveShell(SingletonConfigurable):
231 class InteractiveShell(SingletonConfigurable):
232 """An enhanced, interactive shell for Python."""
232 """An enhanced, interactive shell for Python."""
233
233
234 _instance = None
234 _instance = None
235
235
236 ast_transformers = List([], help=
236 ast_transformers = List([], help=
237 """
237 """
238 A list of ast.NodeTransformer subclass instances, which will be applied
238 A list of ast.NodeTransformer subclass instances, which will be applied
239 to user input before code is run.
239 to user input before code is run.
240 """
240 """
241 ).tag(config=True)
241 ).tag(config=True)
242
242
243 autocall = Enum((0,1,2), default_value=0, help=
243 autocall = Enum((0,1,2), default_value=0, help=
244 """
244 """
245 Make IPython automatically call any callable object even if you didn't
245 Make IPython automatically call any callable object even if you didn't
246 type explicit parentheses. For example, 'str 43' becomes 'str(43)'
246 type explicit parentheses. For example, 'str 43' becomes 'str(43)'
247 automatically. The value can be '0' to disable the feature, '1' for
247 automatically. The value can be '0' to disable the feature, '1' for
248 'smart' autocall, where it is not applied if there are no more
248 'smart' autocall, where it is not applied if there are no more
249 arguments on the line, and '2' for 'full' autocall, where all callable
249 arguments on the line, and '2' for 'full' autocall, where all callable
250 objects are automatically called (even if no arguments are present).
250 objects are automatically called (even if no arguments are present).
251 """
251 """
252 ).tag(config=True)
252 ).tag(config=True)
253 # TODO: remove all autoindent logic and put into frontends.
253 # TODO: remove all autoindent logic and put into frontends.
254 # We can't do this yet because even runlines uses the autoindent.
254 # We can't do this yet because even runlines uses the autoindent.
255 autoindent = Bool(True, help=
255 autoindent = Bool(True, help=
256 """
256 """
257 Autoindent IPython code entered interactively.
257 Autoindent IPython code entered interactively.
258 """
258 """
259 ).tag(config=True)
259 ).tag(config=True)
260
260
261 automagic = Bool(True, help=
261 automagic = Bool(True, help=
262 """
262 """
263 Enable magic commands to be called without the leading %.
263 Enable magic commands to be called without the leading %.
264 """
264 """
265 ).tag(config=True)
265 ).tag(config=True)
266
266
267 banner1 = Unicode(default_banner,
267 banner1 = Unicode(default_banner,
268 help="""The part of the banner to be printed before the profile"""
268 help="""The part of the banner to be printed before the profile"""
269 ).tag(config=True)
269 ).tag(config=True)
270 banner2 = Unicode('',
270 banner2 = Unicode('',
271 help="""The part of the banner to be printed after the profile"""
271 help="""The part of the banner to be printed after the profile"""
272 ).tag(config=True)
272 ).tag(config=True)
273
273
274 cache_size = Integer(1000, help=
274 cache_size = Integer(1000, help=
275 """
275 """
276 Set the size of the output cache. The default is 1000, you can
276 Set the size of the output cache. The default is 1000, you can
277 change it permanently in your config file. Setting it to 0 completely
277 change it permanently in your config file. Setting it to 0 completely
278 disables the caching system, and the minimum value accepted is 3 (if
278 disables the caching system, and the minimum value accepted is 3 (if
279 you provide a value less than 3, it is reset to 0 and a warning is
279 you provide a value less than 3, it is reset to 0 and a warning is
280 issued). This limit is defined because otherwise you'll spend more
280 issued). This limit is defined because otherwise you'll spend more
281 time re-flushing a too small cache than working
281 time re-flushing a too small cache than working
282 """
282 """
283 ).tag(config=True)
283 ).tag(config=True)
284 color_info = Bool(True, help=
284 color_info = Bool(True, help=
285 """
285 """
286 Use colors for displaying information about objects. Because this
286 Use colors for displaying information about objects. Because this
287 information is passed through a pager (like 'less'), and some pagers
287 information is passed through a pager (like 'less'), and some pagers
288 get confused with color codes, this capability can be turned off.
288 get confused with color codes, this capability can be turned off.
289 """
289 """
290 ).tag(config=True)
290 ).tag(config=True)
291 colors = CaselessStrEnum(('Neutral', 'NoColor','LightBG','Linux'),
291 colors = CaselessStrEnum(('Neutral', 'NoColor','LightBG','Linux'),
292 default_value='Neutral',
292 default_value='Neutral',
293 help="Set the color scheme (NoColor, Neutral, Linux, or LightBG)."
293 help="Set the color scheme (NoColor, Neutral, Linux, or LightBG)."
294 ).tag(config=True)
294 ).tag(config=True)
295 debug = Bool(False).tag(config=True)
295 debug = Bool(False).tag(config=True)
296 disable_failing_post_execute = Bool(False,
296 disable_failing_post_execute = Bool(False,
297 help="Don't call post-execute functions that have failed in the past."
297 help="Don't call post-execute functions that have failed in the past."
298 ).tag(config=True)
298 ).tag(config=True)
299 display_formatter = Instance(DisplayFormatter, allow_none=True)
299 display_formatter = Instance(DisplayFormatter, allow_none=True)
300 displayhook_class = Type(DisplayHook)
300 displayhook_class = Type(DisplayHook)
301 display_pub_class = Type(DisplayPublisher)
301 display_pub_class = Type(DisplayPublisher)
302
302
303 sphinxify_docstring = Bool(False, help=
303 sphinxify_docstring = Bool(False, help=
304 """
304 """
305 Enables rich html representation of docstrings. (This requires the
305 Enables rich html representation of docstrings. (This requires the
306 docrepr module).
306 docrepr module).
307 """).tag(config=True)
307 """).tag(config=True)
308
308
309 @observe("sphinxify_docstring")
309 @observe("sphinxify_docstring")
310 def _sphinxify_docstring_changed(self, change):
310 def _sphinxify_docstring_changed(self, change):
311 if change['new']:
311 if change['new']:
312 warn("`sphinxify_docstring` is provisional since IPython 5.0 and might change in future versions." , ProvisionalWarning)
312 warn("`sphinxify_docstring` is provisional since IPython 5.0 and might change in future versions." , ProvisionalWarning)
313
313
314 enable_html_pager = Bool(False, help=
314 enable_html_pager = Bool(False, help=
315 """
315 """
316 (Provisional API) enables html representation in mime bundles sent
316 (Provisional API) enables html representation in mime bundles sent
317 to pagers.
317 to pagers.
318 """).tag(config=True)
318 """).tag(config=True)
319
319
320 @observe("enable_html_pager")
320 @observe("enable_html_pager")
321 def _enable_html_pager_changed(self, change):
321 def _enable_html_pager_changed(self, change):
322 if change['new']:
322 if change['new']:
323 warn("`enable_html_pager` is provisional since IPython 5.0 and might change in future versions.", ProvisionalWarning)
323 warn("`enable_html_pager` is provisional since IPython 5.0 and might change in future versions.", ProvisionalWarning)
324
324
325 data_pub_class = None
325 data_pub_class = None
326
326
327 exit_now = Bool(False)
327 exit_now = Bool(False)
328 exiter = Instance(ExitAutocall)
328 exiter = Instance(ExitAutocall)
329 @default('exiter')
329 @default('exiter')
330 def _exiter_default(self):
330 def _exiter_default(self):
331 return ExitAutocall(self)
331 return ExitAutocall(self)
332 # Monotonically increasing execution counter
332 # Monotonically increasing execution counter
333 execution_count = Integer(1)
333 execution_count = Integer(1)
334 filename = Unicode("<ipython console>")
334 filename = Unicode("<ipython console>")
335 ipython_dir= Unicode('').tag(config=True) # Set to get_ipython_dir() in __init__
335 ipython_dir= Unicode('').tag(config=True) # Set to get_ipython_dir() in __init__
336
336
337 # Input splitter, to transform input line by line and detect when a block
337 # Input splitter, to transform input line by line and detect when a block
338 # is ready to be executed.
338 # is ready to be executed.
339 input_splitter = Instance('IPython.core.inputsplitter.IPythonInputSplitter',
339 input_splitter = Instance('IPython.core.inputsplitter.IPythonInputSplitter',
340 (), {'line_input_checker': True})
340 (), {'line_input_checker': True})
341
341
342 # This InputSplitter instance is used to transform completed cells before
342 # This InputSplitter instance is used to transform completed cells before
343 # running them. It allows cell magics to contain blank lines.
343 # running them. It allows cell magics to contain blank lines.
344 input_transformer_manager = Instance('IPython.core.inputsplitter.IPythonInputSplitter',
344 input_transformer_manager = Instance('IPython.core.inputsplitter.IPythonInputSplitter',
345 (), {'line_input_checker': False})
345 (), {'line_input_checker': False})
346
346
347 logstart = Bool(False, help=
347 logstart = Bool(False, help=
348 """
348 """
349 Start logging to the default log file in overwrite mode.
349 Start logging to the default log file in overwrite mode.
350 Use `logappend` to specify a log file to **append** logs to.
350 Use `logappend` to specify a log file to **append** logs to.
351 """
351 """
352 ).tag(config=True)
352 ).tag(config=True)
353 logfile = Unicode('', help=
353 logfile = Unicode('', help=
354 """
354 """
355 The name of the logfile to use.
355 The name of the logfile to use.
356 """
356 """
357 ).tag(config=True)
357 ).tag(config=True)
358 logappend = Unicode('', help=
358 logappend = Unicode('', help=
359 """
359 """
360 Start logging to the given file in append mode.
360 Start logging to the given file in append mode.
361 Use `logfile` to specify a log file to **overwrite** logs to.
361 Use `logfile` to specify a log file to **overwrite** logs to.
362 """
362 """
363 ).tag(config=True)
363 ).tag(config=True)
364 object_info_string_level = Enum((0,1,2), default_value=0,
364 object_info_string_level = Enum((0,1,2), default_value=0,
365 ).tag(config=True)
365 ).tag(config=True)
366 pdb = Bool(False, help=
366 pdb = Bool(False, help=
367 """
367 """
368 Automatically call the pdb debugger after every exception.
368 Automatically call the pdb debugger after every exception.
369 """
369 """
370 ).tag(config=True)
370 ).tag(config=True)
371 display_page = Bool(False,
371 display_page = Bool(False,
372 help="""If True, anything that would be passed to the pager
372 help="""If True, anything that would be passed to the pager
373 will be displayed as regular output instead."""
373 will be displayed as regular output instead."""
374 ).tag(config=True)
374 ).tag(config=True)
375
375
376 # deprecated prompt traits:
376 # deprecated prompt traits:
377
377
378 prompt_in1 = Unicode('In [\\#]: ',
378 prompt_in1 = Unicode('In [\\#]: ',
379 help="Deprecated since IPython 4.0 and ignored since 5.0, set TerminalInteractiveShell.prompts object directly."
379 help="Deprecated since IPython 4.0 and ignored since 5.0, set TerminalInteractiveShell.prompts object directly."
380 ).tag(config=True)
380 ).tag(config=True)
381 prompt_in2 = Unicode(' .\\D.: ',
381 prompt_in2 = Unicode(' .\\D.: ',
382 help="Deprecated since IPython 4.0 and ignored since 5.0, set TerminalInteractiveShell.prompts object directly."
382 help="Deprecated since IPython 4.0 and ignored since 5.0, set TerminalInteractiveShell.prompts object directly."
383 ).tag(config=True)
383 ).tag(config=True)
384 prompt_out = Unicode('Out[\\#]: ',
384 prompt_out = Unicode('Out[\\#]: ',
385 help="Deprecated since IPython 4.0 and ignored since 5.0, set TerminalInteractiveShell.prompts object directly."
385 help="Deprecated since IPython 4.0 and ignored since 5.0, set TerminalInteractiveShell.prompts object directly."
386 ).tag(config=True)
386 ).tag(config=True)
387 prompts_pad_left = Bool(True,
387 prompts_pad_left = Bool(True,
388 help="Deprecated since IPython 4.0 and ignored since 5.0, set TerminalInteractiveShell.prompts object directly."
388 help="Deprecated since IPython 4.0 and ignored since 5.0, set TerminalInteractiveShell.prompts object directly."
389 ).tag(config=True)
389 ).tag(config=True)
390
390
391 @observe('prompt_in1', 'prompt_in2', 'prompt_out', 'prompt_pad_left')
391 @observe('prompt_in1', 'prompt_in2', 'prompt_out', 'prompt_pad_left')
392 def _prompt_trait_changed(self, change):
392 def _prompt_trait_changed(self, change):
393 name = change['name']
393 name = change['name']
394 warn("InteractiveShell.{name} is deprecated since IPython 4.0"
394 warn("InteractiveShell.{name} is deprecated since IPython 4.0"
395 " and ignored since 5.0, set TerminalInteractiveShell.prompts"
395 " and ignored since 5.0, set TerminalInteractiveShell.prompts"
396 " object directly.".format(name=name))
396 " object directly.".format(name=name))
397
397
398 # protect against weird cases where self.config may not exist:
398 # protect against weird cases where self.config may not exist:
399
399
400 show_rewritten_input = Bool(True,
400 show_rewritten_input = Bool(True,
401 help="Show rewritten input, e.g. for autocall."
401 help="Show rewritten input, e.g. for autocall."
402 ).tag(config=True)
402 ).tag(config=True)
403
403
404 quiet = Bool(False).tag(config=True)
404 quiet = Bool(False).tag(config=True)
405
405
406 history_length = Integer(10000,
406 history_length = Integer(10000,
407 help='Total length of command history'
407 help='Total length of command history'
408 ).tag(config=True)
408 ).tag(config=True)
409
409
410 history_load_length = Integer(1000, help=
410 history_load_length = Integer(1000, help=
411 """
411 """
412 The number of saved history entries to be loaded
412 The number of saved history entries to be loaded
413 into the history buffer at startup.
413 into the history buffer at startup.
414 """
414 """
415 ).tag(config=True)
415 ).tag(config=True)
416
416
417 ast_node_interactivity = Enum(['all', 'last', 'last_expr', 'none', 'last_expr_or_assign'],
417 ast_node_interactivity = Enum(['all', 'last', 'last_expr', 'none', 'last_expr_or_assign'],
418 default_value='last_expr',
418 default_value='last_expr',
419 help="""
419 help="""
420 'all', 'last', 'last_expr' or 'none', 'last_expr_or_assign' specifying
420 'all', 'last', 'last_expr' or 'none', 'last_expr_or_assign' specifying
421 which nodes should be run interactively (displaying output from expressions).
421 which nodes should be run interactively (displaying output from expressions).
422 """
422 """
423 ).tag(config=True)
423 ).tag(config=True)
424
424
425 # TODO: this part of prompt management should be moved to the frontends.
425 # TODO: this part of prompt management should be moved to the frontends.
426 # Use custom TraitTypes that convert '0'->'' and '\\n'->'\n'
426 # Use custom TraitTypes that convert '0'->'' and '\\n'->'\n'
427 separate_in = SeparateUnicode('\n').tag(config=True)
427 separate_in = SeparateUnicode('\n').tag(config=True)
428 separate_out = SeparateUnicode('').tag(config=True)
428 separate_out = SeparateUnicode('').tag(config=True)
429 separate_out2 = SeparateUnicode('').tag(config=True)
429 separate_out2 = SeparateUnicode('').tag(config=True)
430 wildcards_case_sensitive = Bool(True).tag(config=True)
430 wildcards_case_sensitive = Bool(True).tag(config=True)
431 xmode = CaselessStrEnum(('Context','Plain', 'Verbose'),
431 xmode = CaselessStrEnum(('Context','Plain', 'Verbose'),
432 default_value='Context',
432 default_value='Context',
433 help="Switch modes for the IPython exception handlers."
433 help="Switch modes for the IPython exception handlers."
434 ).tag(config=True)
434 ).tag(config=True)
435
435
436 # Subcomponents of InteractiveShell
436 # Subcomponents of InteractiveShell
437 alias_manager = Instance('IPython.core.alias.AliasManager', allow_none=True)
437 alias_manager = Instance('IPython.core.alias.AliasManager', allow_none=True)
438 prefilter_manager = Instance('IPython.core.prefilter.PrefilterManager', allow_none=True)
438 prefilter_manager = Instance('IPython.core.prefilter.PrefilterManager', allow_none=True)
439 builtin_trap = Instance('IPython.core.builtin_trap.BuiltinTrap', allow_none=True)
439 builtin_trap = Instance('IPython.core.builtin_trap.BuiltinTrap', allow_none=True)
440 display_trap = Instance('IPython.core.display_trap.DisplayTrap', allow_none=True)
440 display_trap = Instance('IPython.core.display_trap.DisplayTrap', allow_none=True)
441 extension_manager = Instance('IPython.core.extensions.ExtensionManager', allow_none=True)
441 extension_manager = Instance('IPython.core.extensions.ExtensionManager', allow_none=True)
442 payload_manager = Instance('IPython.core.payload.PayloadManager', allow_none=True)
442 payload_manager = Instance('IPython.core.payload.PayloadManager', allow_none=True)
443 history_manager = Instance('IPython.core.history.HistoryAccessorBase', allow_none=True)
443 history_manager = Instance('IPython.core.history.HistoryAccessorBase', allow_none=True)
444 magics_manager = Instance('IPython.core.magic.MagicsManager', allow_none=True)
444 magics_manager = Instance('IPython.core.magic.MagicsManager', allow_none=True)
445
445
446 profile_dir = Instance('IPython.core.application.ProfileDir', allow_none=True)
446 profile_dir = Instance('IPython.core.application.ProfileDir', allow_none=True)
447 @property
447 @property
448 def profile(self):
448 def profile(self):
449 if self.profile_dir is not None:
449 if self.profile_dir is not None:
450 name = os.path.basename(self.profile_dir.location)
450 name = os.path.basename(self.profile_dir.location)
451 return name.replace('profile_','')
451 return name.replace('profile_','')
452
452
453
453
454 # Private interface
454 # Private interface
455 _post_execute = Dict()
455 _post_execute = Dict()
456
456
457 # Tracks any GUI loop loaded for pylab
457 # Tracks any GUI loop loaded for pylab
458 pylab_gui_select = None
458 pylab_gui_select = None
459
459
460 last_execution_succeeded = Bool(True, help='Did last executed command succeeded')
460 last_execution_succeeded = Bool(True, help='Did last executed command succeeded')
461
461
462 last_execution_result = Instance('IPython.core.interactiveshell.ExecutionResult', help='Result of executing the last command', allow_none=True)
462 last_execution_result = Instance('IPython.core.interactiveshell.ExecutionResult', help='Result of executing the last command', allow_none=True)
463
463
464 def __init__(self, ipython_dir=None, profile_dir=None,
464 def __init__(self, ipython_dir=None, profile_dir=None,
465 user_module=None, user_ns=None,
465 user_module=None, user_ns=None,
466 custom_exceptions=((), None), **kwargs):
466 custom_exceptions=((), None), **kwargs):
467
467
468 # This is where traits with a config_key argument are updated
468 # This is where traits with a config_key argument are updated
469 # from the values on config.
469 # from the values on config.
470 super(InteractiveShell, self).__init__(**kwargs)
470 super(InteractiveShell, self).__init__(**kwargs)
471 if 'PromptManager' in self.config:
471 if 'PromptManager' in self.config:
472 warn('As of IPython 5.0 `PromptManager` config will have no effect'
472 warn('As of IPython 5.0 `PromptManager` config will have no effect'
473 ' and has been replaced by TerminalInteractiveShell.prompts_class')
473 ' and has been replaced by TerminalInteractiveShell.prompts_class')
474 self.configurables = [self]
474 self.configurables = [self]
475
475
476 # These are relatively independent and stateless
476 # These are relatively independent and stateless
477 self.init_ipython_dir(ipython_dir)
477 self.init_ipython_dir(ipython_dir)
478 self.init_profile_dir(profile_dir)
478 self.init_profile_dir(profile_dir)
479 self.init_instance_attrs()
479 self.init_instance_attrs()
480 self.init_environment()
480 self.init_environment()
481
481
482 # Check if we're in a virtualenv, and set up sys.path.
482 # Check if we're in a virtualenv, and set up sys.path.
483 self.init_virtualenv()
483 self.init_virtualenv()
484
484
485 # Create namespaces (user_ns, user_global_ns, etc.)
485 # Create namespaces (user_ns, user_global_ns, etc.)
486 self.init_create_namespaces(user_module, user_ns)
486 self.init_create_namespaces(user_module, user_ns)
487 # This has to be done after init_create_namespaces because it uses
487 # This has to be done after init_create_namespaces because it uses
488 # something in self.user_ns, but before init_sys_modules, which
488 # something in self.user_ns, but before init_sys_modules, which
489 # is the first thing to modify sys.
489 # is the first thing to modify sys.
490 # TODO: When we override sys.stdout and sys.stderr before this class
490 # TODO: When we override sys.stdout and sys.stderr before this class
491 # is created, we are saving the overridden ones here. Not sure if this
491 # is created, we are saving the overridden ones here. Not sure if this
492 # is what we want to do.
492 # is what we want to do.
493 self.save_sys_module_state()
493 self.save_sys_module_state()
494 self.init_sys_modules()
494 self.init_sys_modules()
495
495
496 # While we're trying to have each part of the code directly access what
496 # While we're trying to have each part of the code directly access what
497 # it needs without keeping redundant references to objects, we have too
497 # it needs without keeping redundant references to objects, we have too
498 # much legacy code that expects ip.db to exist.
498 # much legacy code that expects ip.db to exist.
499 self.db = PickleShareDB(os.path.join(self.profile_dir.location, 'db'))
499 self.db = PickleShareDB(os.path.join(self.profile_dir.location, 'db'))
500
500
501 self.init_history()
501 self.init_history()
502 self.init_encoding()
502 self.init_encoding()
503 self.init_prefilter()
503 self.init_prefilter()
504
504
505 self.init_syntax_highlighting()
505 self.init_syntax_highlighting()
506 self.init_hooks()
506 self.init_hooks()
507 self.init_events()
507 self.init_events()
508 self.init_pushd_popd_magic()
508 self.init_pushd_popd_magic()
509 self.init_user_ns()
509 self.init_user_ns()
510 self.init_logger()
510 self.init_logger()
511 self.init_builtins()
511 self.init_builtins()
512
512
513 # The following was in post_config_initialization
513 # The following was in post_config_initialization
514 self.init_inspector()
514 self.init_inspector()
515 self.raw_input_original = input
515 self.raw_input_original = input
516 self.init_completer()
516 self.init_completer()
517 # TODO: init_io() needs to happen before init_traceback handlers
517 # TODO: init_io() needs to happen before init_traceback handlers
518 # because the traceback handlers hardcode the stdout/stderr streams.
518 # because the traceback handlers hardcode the stdout/stderr streams.
519 # This logic in in debugger.Pdb and should eventually be changed.
519 # This logic in in debugger.Pdb and should eventually be changed.
520 self.init_io()
520 self.init_io()
521 self.init_traceback_handlers(custom_exceptions)
521 self.init_traceback_handlers(custom_exceptions)
522 self.init_prompts()
522 self.init_prompts()
523 self.init_display_formatter()
523 self.init_display_formatter()
524 self.init_display_pub()
524 self.init_display_pub()
525 self.init_data_pub()
525 self.init_data_pub()
526 self.init_displayhook()
526 self.init_displayhook()
527 self.init_magics()
527 self.init_magics()
528 self.init_alias()
528 self.init_alias()
529 self.init_logstart()
529 self.init_logstart()
530 self.init_pdb()
530 self.init_pdb()
531 self.init_extension_manager()
531 self.init_extension_manager()
532 self.init_payload()
532 self.init_payload()
533 self.init_deprecation_warnings()
533 self.init_deprecation_warnings()
534 self.hooks.late_startup_hook()
534 self.hooks.late_startup_hook()
535 self.events.trigger('shell_initialized', self)
535 self.events.trigger('shell_initialized', self)
536 atexit.register(self.atexit_operations)
536 atexit.register(self.atexit_operations)
537
537
538 def get_ipython(self):
538 def get_ipython(self):
539 """Return the currently running IPython instance."""
539 """Return the currently running IPython instance."""
540 return self
540 return self
541
541
542 #-------------------------------------------------------------------------
542 #-------------------------------------------------------------------------
543 # Trait changed handlers
543 # Trait changed handlers
544 #-------------------------------------------------------------------------
544 #-------------------------------------------------------------------------
545 @observe('ipython_dir')
545 @observe('ipython_dir')
546 def _ipython_dir_changed(self, change):
546 def _ipython_dir_changed(self, change):
547 ensure_dir_exists(change['new'])
547 ensure_dir_exists(change['new'])
548
548
549 def set_autoindent(self,value=None):
549 def set_autoindent(self,value=None):
550 """Set the autoindent flag.
550 """Set the autoindent flag.
551
551
552 If called with no arguments, it acts as a toggle."""
552 If called with no arguments, it acts as a toggle."""
553 if value is None:
553 if value is None:
554 self.autoindent = not self.autoindent
554 self.autoindent = not self.autoindent
555 else:
555 else:
556 self.autoindent = value
556 self.autoindent = value
557
557
558 #-------------------------------------------------------------------------
558 #-------------------------------------------------------------------------
559 # init_* methods called by __init__
559 # init_* methods called by __init__
560 #-------------------------------------------------------------------------
560 #-------------------------------------------------------------------------
561
561
562 def init_ipython_dir(self, ipython_dir):
562 def init_ipython_dir(self, ipython_dir):
563 if ipython_dir is not None:
563 if ipython_dir is not None:
564 self.ipython_dir = ipython_dir
564 self.ipython_dir = ipython_dir
565 return
565 return
566
566
567 self.ipython_dir = get_ipython_dir()
567 self.ipython_dir = get_ipython_dir()
568
568
569 def init_profile_dir(self, profile_dir):
569 def init_profile_dir(self, profile_dir):
570 if profile_dir is not None:
570 if profile_dir is not None:
571 self.profile_dir = profile_dir
571 self.profile_dir = profile_dir
572 return
572 return
573 self.profile_dir =\
573 self.profile_dir =\
574 ProfileDir.create_profile_dir_by_name(self.ipython_dir, 'default')
574 ProfileDir.create_profile_dir_by_name(self.ipython_dir, 'default')
575
575
576 def init_instance_attrs(self):
576 def init_instance_attrs(self):
577 self.more = False
577 self.more = False
578
578
579 # command compiler
579 # command compiler
580 self.compile = CachingCompiler()
580 self.compile = CachingCompiler()
581
581
582 # Make an empty namespace, which extension writers can rely on both
582 # Make an empty namespace, which extension writers can rely on both
583 # existing and NEVER being used by ipython itself. This gives them a
583 # existing and NEVER being used by ipython itself. This gives them a
584 # convenient location for storing additional information and state
584 # convenient location for storing additional information and state
585 # their extensions may require, without fear of collisions with other
585 # their extensions may require, without fear of collisions with other
586 # ipython names that may develop later.
586 # ipython names that may develop later.
587 self.meta = Struct()
587 self.meta = Struct()
588
588
589 # Temporary files used for various purposes. Deleted at exit.
589 # Temporary files used for various purposes. Deleted at exit.
590 self.tempfiles = []
590 self.tempfiles = []
591 self.tempdirs = []
591 self.tempdirs = []
592
592
593 # keep track of where we started running (mainly for crash post-mortem)
593 # keep track of where we started running (mainly for crash post-mortem)
594 # This is not being used anywhere currently.
594 # This is not being used anywhere currently.
595 self.starting_dir = os.getcwd()
595 self.starting_dir = os.getcwd()
596
596
597 # Indentation management
597 # Indentation management
598 self.indent_current_nsp = 0
598 self.indent_current_nsp = 0
599
599
600 # Dict to track post-execution functions that have been registered
600 # Dict to track post-execution functions that have been registered
601 self._post_execute = {}
601 self._post_execute = {}
602
602
603 def init_environment(self):
603 def init_environment(self):
604 """Any changes we need to make to the user's environment."""
604 """Any changes we need to make to the user's environment."""
605 pass
605 pass
606
606
607 def init_encoding(self):
607 def init_encoding(self):
608 # Get system encoding at startup time. Certain terminals (like Emacs
608 # Get system encoding at startup time. Certain terminals (like Emacs
609 # under Win32 have it set to None, and we need to have a known valid
609 # under Win32 have it set to None, and we need to have a known valid
610 # encoding to use in the raw_input() method
610 # encoding to use in the raw_input() method
611 try:
611 try:
612 self.stdin_encoding = sys.stdin.encoding or 'ascii'
612 self.stdin_encoding = sys.stdin.encoding or 'ascii'
613 except AttributeError:
613 except AttributeError:
614 self.stdin_encoding = 'ascii'
614 self.stdin_encoding = 'ascii'
615
615
616
616
617 @observe('colors')
617 @observe('colors')
618 def init_syntax_highlighting(self, changes=None):
618 def init_syntax_highlighting(self, changes=None):
619 # Python source parser/formatter for syntax highlighting
619 # Python source parser/formatter for syntax highlighting
620 pyformat = PyColorize.Parser(style=self.colors, parent=self).format
620 pyformat = PyColorize.Parser(style=self.colors, parent=self).format
621 self.pycolorize = lambda src: pyformat(src,'str')
621 self.pycolorize = lambda src: pyformat(src,'str')
622
622
623 def refresh_style(self):
623 def refresh_style(self):
624 # No-op here, used in subclass
624 # No-op here, used in subclass
625 pass
625 pass
626
626
627 def init_pushd_popd_magic(self):
627 def init_pushd_popd_magic(self):
628 # for pushd/popd management
628 # for pushd/popd management
629 self.home_dir = get_home_dir()
629 self.home_dir = get_home_dir()
630
630
631 self.dir_stack = []
631 self.dir_stack = []
632
632
633 def init_logger(self):
633 def init_logger(self):
634 self.logger = Logger(self.home_dir, logfname='ipython_log.py',
634 self.logger = Logger(self.home_dir, logfname='ipython_log.py',
635 logmode='rotate')
635 logmode='rotate')
636
636
637 def init_logstart(self):
637 def init_logstart(self):
638 """Initialize logging in case it was requested at the command line.
638 """Initialize logging in case it was requested at the command line.
639 """
639 """
640 if self.logappend:
640 if self.logappend:
641 self.magic('logstart %s append' % self.logappend)
641 self.magic('logstart %s append' % self.logappend)
642 elif self.logfile:
642 elif self.logfile:
643 self.magic('logstart %s' % self.logfile)
643 self.magic('logstart %s' % self.logfile)
644 elif self.logstart:
644 elif self.logstart:
645 self.magic('logstart')
645 self.magic('logstart')
646
646
647 def init_deprecation_warnings(self):
647 def init_deprecation_warnings(self):
648 """
648 """
649 register default filter for deprecation warning.
649 register default filter for deprecation warning.
650
650
651 This will allow deprecation warning of function used interactively to show
651 This will allow deprecation warning of function used interactively to show
652 warning to users, and still hide deprecation warning from libraries import.
652 warning to users, and still hide deprecation warning from libraries import.
653 """
653 """
654 warnings.filterwarnings("default", category=DeprecationWarning, module=self.user_ns.get("__name__"))
654 warnings.filterwarnings("default", category=DeprecationWarning, module=self.user_ns.get("__name__"))
655
655
656 def init_builtins(self):
656 def init_builtins(self):
657 # A single, static flag that we set to True. Its presence indicates
657 # A single, static flag that we set to True. Its presence indicates
658 # that an IPython shell has been created, and we make no attempts at
658 # that an IPython shell has been created, and we make no attempts at
659 # removing on exit or representing the existence of more than one
659 # removing on exit or representing the existence of more than one
660 # IPython at a time.
660 # IPython at a time.
661 builtin_mod.__dict__['__IPYTHON__'] = True
661 builtin_mod.__dict__['__IPYTHON__'] = True
662 builtin_mod.__dict__['display'] = display
662 builtin_mod.__dict__['display'] = display
663
663
664 self.builtin_trap = BuiltinTrap(shell=self)
664 self.builtin_trap = BuiltinTrap(shell=self)
665
665
666 @observe('colors')
666 @observe('colors')
667 def init_inspector(self, changes=None):
667 def init_inspector(self, changes=None):
668 # Object inspector
668 # Object inspector
669 self.inspector = oinspect.Inspector(oinspect.InspectColors,
669 self.inspector = oinspect.Inspector(oinspect.InspectColors,
670 PyColorize.ANSICodeColors,
670 PyColorize.ANSICodeColors,
671 self.colors,
671 self.colors,
672 self.object_info_string_level)
672 self.object_info_string_level)
673
673
674 def init_io(self):
674 def init_io(self):
675 # This will just use sys.stdout and sys.stderr. If you want to
675 # This will just use sys.stdout and sys.stderr. If you want to
676 # override sys.stdout and sys.stderr themselves, you need to do that
676 # override sys.stdout and sys.stderr themselves, you need to do that
677 # *before* instantiating this class, because io holds onto
677 # *before* instantiating this class, because io holds onto
678 # references to the underlying streams.
678 # references to the underlying streams.
679 # io.std* are deprecated, but don't show our own deprecation warnings
679 # io.std* are deprecated, but don't show our own deprecation warnings
680 # during initialization of the deprecated API.
680 # during initialization of the deprecated API.
681 with warnings.catch_warnings():
681 with warnings.catch_warnings():
682 warnings.simplefilter('ignore', DeprecationWarning)
682 warnings.simplefilter('ignore', DeprecationWarning)
683 io.stdout = io.IOStream(sys.stdout)
683 io.stdout = io.IOStream(sys.stdout)
684 io.stderr = io.IOStream(sys.stderr)
684 io.stderr = io.IOStream(sys.stderr)
685
685
686 def init_prompts(self):
686 def init_prompts(self):
687 # Set system prompts, so that scripts can decide if they are running
687 # Set system prompts, so that scripts can decide if they are running
688 # interactively.
688 # interactively.
689 sys.ps1 = 'In : '
689 sys.ps1 = 'In : '
690 sys.ps2 = '...: '
690 sys.ps2 = '...: '
691 sys.ps3 = 'Out: '
691 sys.ps3 = 'Out: '
692
692
693 def init_display_formatter(self):
693 def init_display_formatter(self):
694 self.display_formatter = DisplayFormatter(parent=self)
694 self.display_formatter = DisplayFormatter(parent=self)
695 self.configurables.append(self.display_formatter)
695 self.configurables.append(self.display_formatter)
696
696
697 def init_display_pub(self):
697 def init_display_pub(self):
698 self.display_pub = self.display_pub_class(parent=self)
698 self.display_pub = self.display_pub_class(parent=self)
699 self.configurables.append(self.display_pub)
699 self.configurables.append(self.display_pub)
700
700
701 def init_data_pub(self):
701 def init_data_pub(self):
702 if not self.data_pub_class:
702 if not self.data_pub_class:
703 self.data_pub = None
703 self.data_pub = None
704 return
704 return
705 self.data_pub = self.data_pub_class(parent=self)
705 self.data_pub = self.data_pub_class(parent=self)
706 self.configurables.append(self.data_pub)
706 self.configurables.append(self.data_pub)
707
707
708 def init_displayhook(self):
708 def init_displayhook(self):
709 # Initialize displayhook, set in/out prompts and printing system
709 # Initialize displayhook, set in/out prompts and printing system
710 self.displayhook = self.displayhook_class(
710 self.displayhook = self.displayhook_class(
711 parent=self,
711 parent=self,
712 shell=self,
712 shell=self,
713 cache_size=self.cache_size,
713 cache_size=self.cache_size,
714 )
714 )
715 self.configurables.append(self.displayhook)
715 self.configurables.append(self.displayhook)
716 # This is a context manager that installs/revmoes the displayhook at
716 # This is a context manager that installs/revmoes the displayhook at
717 # the appropriate time.
717 # the appropriate time.
718 self.display_trap = DisplayTrap(hook=self.displayhook)
718 self.display_trap = DisplayTrap(hook=self.displayhook)
719
719
720 def init_virtualenv(self):
720 def init_virtualenv(self):
721 """Add a virtualenv to sys.path so the user can import modules from it.
721 """Add a virtualenv to sys.path so the user can import modules from it.
722 This isn't perfect: it doesn't use the Python interpreter with which the
722 This isn't perfect: it doesn't use the Python interpreter with which the
723 virtualenv was built, and it ignores the --no-site-packages option. A
723 virtualenv was built, and it ignores the --no-site-packages option. A
724 warning will appear suggesting the user installs IPython in the
724 warning will appear suggesting the user installs IPython in the
725 virtualenv, but for many cases, it probably works well enough.
725 virtualenv, but for many cases, it probably works well enough.
726
726
727 Adapted from code snippets online.
727 Adapted from code snippets online.
728
728
729 http://blog.ufsoft.org/2009/1/29/ipython-and-virtualenv
729 http://blog.ufsoft.org/2009/1/29/ipython-and-virtualenv
730 """
730 """
731 if 'VIRTUAL_ENV' not in os.environ:
731 if 'VIRTUAL_ENV' not in os.environ:
732 # Not in a virtualenv
732 # Not in a virtualenv
733 return
733 return
734
734
735 p = os.path.normcase(sys.executable)
735 p = os.path.normcase(sys.executable)
736 p_venv = os.path.normcase(os.environ['VIRTUAL_ENV'])
736 p_venv = os.path.normcase(os.environ['VIRTUAL_ENV'])
737
737
738 # executable path should end like /bin/python or \\scripts\\python.exe
738 # executable path should end like /bin/python or \\scripts\\python.exe
739 p_exe_up2 = os.path.dirname(os.path.dirname(p))
739 p_exe_up2 = os.path.dirname(os.path.dirname(p))
740 if p_exe_up2 and os.path.samefile(p_exe_up2, p_venv):
740 if p_exe_up2 and os.path.samefile(p_exe_up2, p_venv):
741 # Our exe is inside the virtualenv, don't need to do anything.
741 # Our exe is inside the virtualenv, don't need to do anything.
742 return
742 return
743
743
744 # fallback venv detection:
744 # fallback venv detection:
745 # stdlib venv may symlink sys.executable, so we can't use realpath.
745 # stdlib venv may symlink sys.executable, so we can't use realpath.
746 # but others can symlink *to* the venv Python, so we can't just use sys.executable.
746 # but others can symlink *to* the venv Python, so we can't just use sys.executable.
747 # So we just check every item in the symlink tree (generally <= 3)
747 # So we just check every item in the symlink tree (generally <= 3)
748 paths = [p]
748 paths = [p]
749 while os.path.islink(p):
749 while os.path.islink(p):
750 p = os.path.normcase(os.path.join(os.path.dirname(p), os.readlink(p)))
750 p = os.path.normcase(os.path.join(os.path.dirname(p), os.readlink(p)))
751 paths.append(p)
751 paths.append(p)
752
752
753 # In Cygwin paths like "c:\..." and '\cygdrive\c\...' are possible
753 # In Cygwin paths like "c:\..." and '\cygdrive\c\...' are possible
754 if p_venv.startswith('\\cygdrive'):
754 if p_venv.startswith('\\cygdrive'):
755 p_venv = p_venv[11:]
755 p_venv = p_venv[11:]
756 elif len(p_venv) >= 2 and p_venv[1] == ':':
756 elif len(p_venv) >= 2 and p_venv[1] == ':':
757 p_venv = p_venv[2:]
757 p_venv = p_venv[2:]
758
758
759 if any(p_venv in p for p in paths):
759 if any(p_venv in p for p in paths):
760 # Running properly in the virtualenv, don't need to do anything
760 # Running properly in the virtualenv, don't need to do anything
761 return
761 return
762
762
763 warn("Attempting to work in a virtualenv. If you encounter problems, please "
763 warn("Attempting to work in a virtualenv. If you encounter problems, please "
764 "install IPython inside the virtualenv.")
764 "install IPython inside the virtualenv.")
765 if sys.platform == "win32":
765 if sys.platform == "win32":
766 virtual_env = os.path.join(os.environ['VIRTUAL_ENV'], 'Lib', 'site-packages')
766 virtual_env = os.path.join(os.environ['VIRTUAL_ENV'], 'Lib', 'site-packages')
767 else:
767 else:
768 virtual_env = os.path.join(os.environ['VIRTUAL_ENV'], 'lib',
768 virtual_env = os.path.join(os.environ['VIRTUAL_ENV'], 'lib',
769 'python%d.%d' % sys.version_info[:2], 'site-packages')
769 'python%d.%d' % sys.version_info[:2], 'site-packages')
770
770
771 import site
771 import site
772 sys.path.insert(0, virtual_env)
772 sys.path.insert(0, virtual_env)
773 site.addsitedir(virtual_env)
773 site.addsitedir(virtual_env)
774
774
775 #-------------------------------------------------------------------------
775 #-------------------------------------------------------------------------
776 # Things related to injections into the sys module
776 # Things related to injections into the sys module
777 #-------------------------------------------------------------------------
777 #-------------------------------------------------------------------------
778
778
779 def save_sys_module_state(self):
779 def save_sys_module_state(self):
780 """Save the state of hooks in the sys module.
780 """Save the state of hooks in the sys module.
781
781
782 This has to be called after self.user_module is created.
782 This has to be called after self.user_module is created.
783 """
783 """
784 self._orig_sys_module_state = {'stdin': sys.stdin,
784 self._orig_sys_module_state = {'stdin': sys.stdin,
785 'stdout': sys.stdout,
785 'stdout': sys.stdout,
786 'stderr': sys.stderr,
786 'stderr': sys.stderr,
787 'excepthook': sys.excepthook}
787 'excepthook': sys.excepthook}
788 self._orig_sys_modules_main_name = self.user_module.__name__
788 self._orig_sys_modules_main_name = self.user_module.__name__
789 self._orig_sys_modules_main_mod = sys.modules.get(self.user_module.__name__)
789 self._orig_sys_modules_main_mod = sys.modules.get(self.user_module.__name__)
790
790
791 def restore_sys_module_state(self):
791 def restore_sys_module_state(self):
792 """Restore the state of the sys module."""
792 """Restore the state of the sys module."""
793 try:
793 try:
794 for k, v in self._orig_sys_module_state.items():
794 for k, v in self._orig_sys_module_state.items():
795 setattr(sys, k, v)
795 setattr(sys, k, v)
796 except AttributeError:
796 except AttributeError:
797 pass
797 pass
798 # Reset what what done in self.init_sys_modules
798 # Reset what what done in self.init_sys_modules
799 if self._orig_sys_modules_main_mod is not None:
799 if self._orig_sys_modules_main_mod is not None:
800 sys.modules[self._orig_sys_modules_main_name] = self._orig_sys_modules_main_mod
800 sys.modules[self._orig_sys_modules_main_name] = self._orig_sys_modules_main_mod
801
801
802 #-------------------------------------------------------------------------
802 #-------------------------------------------------------------------------
803 # Things related to the banner
803 # Things related to the banner
804 #-------------------------------------------------------------------------
804 #-------------------------------------------------------------------------
805
805
806 @property
806 @property
807 def banner(self):
807 def banner(self):
808 banner = self.banner1
808 banner = self.banner1
809 if self.profile and self.profile != 'default':
809 if self.profile and self.profile != 'default':
810 banner += '\nIPython profile: %s\n' % self.profile
810 banner += '\nIPython profile: %s\n' % self.profile
811 if self.banner2:
811 if self.banner2:
812 banner += '\n' + self.banner2
812 banner += '\n' + self.banner2
813 return banner
813 return banner
814
814
815 def show_banner(self, banner=None):
815 def show_banner(self, banner=None):
816 if banner is None:
816 if banner is None:
817 banner = self.banner
817 banner = self.banner
818 sys.stdout.write(banner)
818 sys.stdout.write(banner)
819
819
820 #-------------------------------------------------------------------------
820 #-------------------------------------------------------------------------
821 # Things related to hooks
821 # Things related to hooks
822 #-------------------------------------------------------------------------
822 #-------------------------------------------------------------------------
823
823
824 def init_hooks(self):
824 def init_hooks(self):
825 # hooks holds pointers used for user-side customizations
825 # hooks holds pointers used for user-side customizations
826 self.hooks = Struct()
826 self.hooks = Struct()
827
827
828 self.strdispatchers = {}
828 self.strdispatchers = {}
829
829
830 # Set all default hooks, defined in the IPython.hooks module.
830 # Set all default hooks, defined in the IPython.hooks module.
831 hooks = IPython.core.hooks
831 hooks = IPython.core.hooks
832 for hook_name in hooks.__all__:
832 for hook_name in hooks.__all__:
833 # default hooks have priority 100, i.e. low; user hooks should have
833 # default hooks have priority 100, i.e. low; user hooks should have
834 # 0-100 priority
834 # 0-100 priority
835 self.set_hook(hook_name,getattr(hooks,hook_name), 100, _warn_deprecated=False)
835 self.set_hook(hook_name,getattr(hooks,hook_name), 100, _warn_deprecated=False)
836
836
837 if self.display_page:
837 if self.display_page:
838 self.set_hook('show_in_pager', page.as_hook(page.display_page), 90)
838 self.set_hook('show_in_pager', page.as_hook(page.display_page), 90)
839
839
840 def set_hook(self,name,hook, priority=50, str_key=None, re_key=None,
840 def set_hook(self,name,hook, priority=50, str_key=None, re_key=None,
841 _warn_deprecated=True):
841 _warn_deprecated=True):
842 """set_hook(name,hook) -> sets an internal IPython hook.
842 """set_hook(name,hook) -> sets an internal IPython hook.
843
843
844 IPython exposes some of its internal API as user-modifiable hooks. By
844 IPython exposes some of its internal API as user-modifiable hooks. By
845 adding your function to one of these hooks, you can modify IPython's
845 adding your function to one of these hooks, you can modify IPython's
846 behavior to call at runtime your own routines."""
846 behavior to call at runtime your own routines."""
847
847
848 # At some point in the future, this should validate the hook before it
848 # At some point in the future, this should validate the hook before it
849 # accepts it. Probably at least check that the hook takes the number
849 # accepts it. Probably at least check that the hook takes the number
850 # of args it's supposed to.
850 # of args it's supposed to.
851
851
852 f = types.MethodType(hook,self)
852 f = types.MethodType(hook,self)
853
853
854 # check if the hook is for strdispatcher first
854 # check if the hook is for strdispatcher first
855 if str_key is not None:
855 if str_key is not None:
856 sdp = self.strdispatchers.get(name, StrDispatch())
856 sdp = self.strdispatchers.get(name, StrDispatch())
857 sdp.add_s(str_key, f, priority )
857 sdp.add_s(str_key, f, priority )
858 self.strdispatchers[name] = sdp
858 self.strdispatchers[name] = sdp
859 return
859 return
860 if re_key is not None:
860 if re_key is not None:
861 sdp = self.strdispatchers.get(name, StrDispatch())
861 sdp = self.strdispatchers.get(name, StrDispatch())
862 sdp.add_re(re.compile(re_key), f, priority )
862 sdp.add_re(re.compile(re_key), f, priority )
863 self.strdispatchers[name] = sdp
863 self.strdispatchers[name] = sdp
864 return
864 return
865
865
866 dp = getattr(self.hooks, name, None)
866 dp = getattr(self.hooks, name, None)
867 if name not in IPython.core.hooks.__all__:
867 if name not in IPython.core.hooks.__all__:
868 print("Warning! Hook '%s' is not one of %s" % \
868 print("Warning! Hook '%s' is not one of %s" % \
869 (name, IPython.core.hooks.__all__ ))
869 (name, IPython.core.hooks.__all__ ))
870
870
871 if _warn_deprecated and (name in IPython.core.hooks.deprecated):
871 if _warn_deprecated and (name in IPython.core.hooks.deprecated):
872 alternative = IPython.core.hooks.deprecated[name]
872 alternative = IPython.core.hooks.deprecated[name]
873 warn("Hook {} is deprecated. Use {} instead.".format(name, alternative), stacklevel=2)
873 warn("Hook {} is deprecated. Use {} instead.".format(name, alternative), stacklevel=2)
874
874
875 if not dp:
875 if not dp:
876 dp = IPython.core.hooks.CommandChainDispatcher()
876 dp = IPython.core.hooks.CommandChainDispatcher()
877
877
878 try:
878 try:
879 dp.add(f,priority)
879 dp.add(f,priority)
880 except AttributeError:
880 except AttributeError:
881 # it was not commandchain, plain old func - replace
881 # it was not commandchain, plain old func - replace
882 dp = f
882 dp = f
883
883
884 setattr(self.hooks,name, dp)
884 setattr(self.hooks,name, dp)
885
885
886 #-------------------------------------------------------------------------
886 #-------------------------------------------------------------------------
887 # Things related to events
887 # Things related to events
888 #-------------------------------------------------------------------------
888 #-------------------------------------------------------------------------
889
889
890 def init_events(self):
890 def init_events(self):
891 self.events = EventManager(self, available_events)
891 self.events = EventManager(self, available_events)
892
892
893 self.events.register("pre_execute", self._clear_warning_registry)
893 self.events.register("pre_execute", self._clear_warning_registry)
894
894
895 def register_post_execute(self, func):
895 def register_post_execute(self, func):
896 """DEPRECATED: Use ip.events.register('post_run_cell', func)
896 """DEPRECATED: Use ip.events.register('post_run_cell', func)
897
897
898 Register a function for calling after code execution.
898 Register a function for calling after code execution.
899 """
899 """
900 warn("ip.register_post_execute is deprecated, use "
900 warn("ip.register_post_execute is deprecated, use "
901 "ip.events.register('post_run_cell', func) instead.", stacklevel=2)
901 "ip.events.register('post_run_cell', func) instead.", stacklevel=2)
902 self.events.register('post_run_cell', func)
902 self.events.register('post_run_cell', func)
903
903
904 def _clear_warning_registry(self):
904 def _clear_warning_registry(self):
905 # clear the warning registry, so that different code blocks with
905 # clear the warning registry, so that different code blocks with
906 # overlapping line number ranges don't cause spurious suppression of
906 # overlapping line number ranges don't cause spurious suppression of
907 # warnings (see gh-6611 for details)
907 # warnings (see gh-6611 for details)
908 if "__warningregistry__" in self.user_global_ns:
908 if "__warningregistry__" in self.user_global_ns:
909 del self.user_global_ns["__warningregistry__"]
909 del self.user_global_ns["__warningregistry__"]
910
910
911 #-------------------------------------------------------------------------
911 #-------------------------------------------------------------------------
912 # Things related to the "main" module
912 # Things related to the "main" module
913 #-------------------------------------------------------------------------
913 #-------------------------------------------------------------------------
914
914
915 def new_main_mod(self, filename, modname):
915 def new_main_mod(self, filename, modname):
916 """Return a new 'main' module object for user code execution.
916 """Return a new 'main' module object for user code execution.
917
917
918 ``filename`` should be the path of the script which will be run in the
918 ``filename`` should be the path of the script which will be run in the
919 module. Requests with the same filename will get the same module, with
919 module. Requests with the same filename will get the same module, with
920 its namespace cleared.
920 its namespace cleared.
921
921
922 ``modname`` should be the module name - normally either '__main__' or
922 ``modname`` should be the module name - normally either '__main__' or
923 the basename of the file without the extension.
923 the basename of the file without the extension.
924
924
925 When scripts are executed via %run, we must keep a reference to their
925 When scripts are executed via %run, we must keep a reference to their
926 __main__ module around so that Python doesn't
926 __main__ module around so that Python doesn't
927 clear it, rendering references to module globals useless.
927 clear it, rendering references to module globals useless.
928
928
929 This method keeps said reference in a private dict, keyed by the
929 This method keeps said reference in a private dict, keyed by the
930 absolute path of the script. This way, for multiple executions of the
930 absolute path of the script. This way, for multiple executions of the
931 same script we only keep one copy of the namespace (the last one),
931 same script we only keep one copy of the namespace (the last one),
932 thus preventing memory leaks from old references while allowing the
932 thus preventing memory leaks from old references while allowing the
933 objects from the last execution to be accessible.
933 objects from the last execution to be accessible.
934 """
934 """
935 filename = os.path.abspath(filename)
935 filename = os.path.abspath(filename)
936 try:
936 try:
937 main_mod = self._main_mod_cache[filename]
937 main_mod = self._main_mod_cache[filename]
938 except KeyError:
938 except KeyError:
939 main_mod = self._main_mod_cache[filename] = types.ModuleType(
939 main_mod = self._main_mod_cache[filename] = types.ModuleType(
940 modname,
940 modname,
941 doc="Module created for script run in IPython")
941 doc="Module created for script run in IPython")
942 else:
942 else:
943 main_mod.__dict__.clear()
943 main_mod.__dict__.clear()
944 main_mod.__name__ = modname
944 main_mod.__name__ = modname
945
945
946 main_mod.__file__ = filename
946 main_mod.__file__ = filename
947 # It seems pydoc (and perhaps others) needs any module instance to
947 # It seems pydoc (and perhaps others) needs any module instance to
948 # implement a __nonzero__ method
948 # implement a __nonzero__ method
949 main_mod.__nonzero__ = lambda : True
949 main_mod.__nonzero__ = lambda : True
950
950
951 return main_mod
951 return main_mod
952
952
953 def clear_main_mod_cache(self):
953 def clear_main_mod_cache(self):
954 """Clear the cache of main modules.
954 """Clear the cache of main modules.
955
955
956 Mainly for use by utilities like %reset.
956 Mainly for use by utilities like %reset.
957
957
958 Examples
958 Examples
959 --------
959 --------
960
960
961 In [15]: import IPython
961 In [15]: import IPython
962
962
963 In [16]: m = _ip.new_main_mod(IPython.__file__, 'IPython')
963 In [16]: m = _ip.new_main_mod(IPython.__file__, 'IPython')
964
964
965 In [17]: len(_ip._main_mod_cache) > 0
965 In [17]: len(_ip._main_mod_cache) > 0
966 Out[17]: True
966 Out[17]: True
967
967
968 In [18]: _ip.clear_main_mod_cache()
968 In [18]: _ip.clear_main_mod_cache()
969
969
970 In [19]: len(_ip._main_mod_cache) == 0
970 In [19]: len(_ip._main_mod_cache) == 0
971 Out[19]: True
971 Out[19]: True
972 """
972 """
973 self._main_mod_cache.clear()
973 self._main_mod_cache.clear()
974
974
975 #-------------------------------------------------------------------------
975 #-------------------------------------------------------------------------
976 # Things related to debugging
976 # Things related to debugging
977 #-------------------------------------------------------------------------
977 #-------------------------------------------------------------------------
978
978
979 def init_pdb(self):
979 def init_pdb(self):
980 # Set calling of pdb on exceptions
980 # Set calling of pdb on exceptions
981 # self.call_pdb is a property
981 # self.call_pdb is a property
982 self.call_pdb = self.pdb
982 self.call_pdb = self.pdb
983
983
984 def _get_call_pdb(self):
984 def _get_call_pdb(self):
985 return self._call_pdb
985 return self._call_pdb
986
986
987 def _set_call_pdb(self,val):
987 def _set_call_pdb(self,val):
988
988
989 if val not in (0,1,False,True):
989 if val not in (0,1,False,True):
990 raise ValueError('new call_pdb value must be boolean')
990 raise ValueError('new call_pdb value must be boolean')
991
991
992 # store value in instance
992 # store value in instance
993 self._call_pdb = val
993 self._call_pdb = val
994
994
995 # notify the actual exception handlers
995 # notify the actual exception handlers
996 self.InteractiveTB.call_pdb = val
996 self.InteractiveTB.call_pdb = val
997
997
998 call_pdb = property(_get_call_pdb,_set_call_pdb,None,
998 call_pdb = property(_get_call_pdb,_set_call_pdb,None,
999 'Control auto-activation of pdb at exceptions')
999 'Control auto-activation of pdb at exceptions')
1000
1000
1001 def debugger(self,force=False):
1001 def debugger(self,force=False):
1002 """Call the pdb debugger.
1002 """Call the pdb debugger.
1003
1003
1004 Keywords:
1004 Keywords:
1005
1005
1006 - force(False): by default, this routine checks the instance call_pdb
1006 - force(False): by default, this routine checks the instance call_pdb
1007 flag and does not actually invoke the debugger if the flag is false.
1007 flag and does not actually invoke the debugger if the flag is false.
1008 The 'force' option forces the debugger to activate even if the flag
1008 The 'force' option forces the debugger to activate even if the flag
1009 is false.
1009 is false.
1010 """
1010 """
1011
1011
1012 if not (force or self.call_pdb):
1012 if not (force or self.call_pdb):
1013 return
1013 return
1014
1014
1015 if not hasattr(sys,'last_traceback'):
1015 if not hasattr(sys,'last_traceback'):
1016 error('No traceback has been produced, nothing to debug.')
1016 error('No traceback has been produced, nothing to debug.')
1017 return
1017 return
1018
1018
1019 self.InteractiveTB.debugger(force=True)
1019 self.InteractiveTB.debugger(force=True)
1020
1020
1021 #-------------------------------------------------------------------------
1021 #-------------------------------------------------------------------------
1022 # Things related to IPython's various namespaces
1022 # Things related to IPython's various namespaces
1023 #-------------------------------------------------------------------------
1023 #-------------------------------------------------------------------------
1024 default_user_namespaces = True
1024 default_user_namespaces = True
1025
1025
1026 def init_create_namespaces(self, user_module=None, user_ns=None):
1026 def init_create_namespaces(self, user_module=None, user_ns=None):
1027 # Create the namespace where the user will operate. user_ns is
1027 # Create the namespace where the user will operate. user_ns is
1028 # normally the only one used, and it is passed to the exec calls as
1028 # normally the only one used, and it is passed to the exec calls as
1029 # the locals argument. But we do carry a user_global_ns namespace
1029 # the locals argument. But we do carry a user_global_ns namespace
1030 # given as the exec 'globals' argument, This is useful in embedding
1030 # given as the exec 'globals' argument, This is useful in embedding
1031 # situations where the ipython shell opens in a context where the
1031 # situations where the ipython shell opens in a context where the
1032 # distinction between locals and globals is meaningful. For
1032 # distinction between locals and globals is meaningful. For
1033 # non-embedded contexts, it is just the same object as the user_ns dict.
1033 # non-embedded contexts, it is just the same object as the user_ns dict.
1034
1034
1035 # FIXME. For some strange reason, __builtins__ is showing up at user
1035 # FIXME. For some strange reason, __builtins__ is showing up at user
1036 # level as a dict instead of a module. This is a manual fix, but I
1036 # level as a dict instead of a module. This is a manual fix, but I
1037 # should really track down where the problem is coming from. Alex
1037 # should really track down where the problem is coming from. Alex
1038 # Schmolck reported this problem first.
1038 # Schmolck reported this problem first.
1039
1039
1040 # A useful post by Alex Martelli on this topic:
1040 # A useful post by Alex Martelli on this topic:
1041 # Re: inconsistent value from __builtins__
1041 # Re: inconsistent value from __builtins__
1042 # Von: Alex Martelli <aleaxit@yahoo.com>
1042 # Von: Alex Martelli <aleaxit@yahoo.com>
1043 # Datum: Freitag 01 Oktober 2004 04:45:34 nachmittags/abends
1043 # Datum: Freitag 01 Oktober 2004 04:45:34 nachmittags/abends
1044 # Gruppen: comp.lang.python
1044 # Gruppen: comp.lang.python
1045
1045
1046 # Michael Hohn <hohn@hooknose.lbl.gov> wrote:
1046 # Michael Hohn <hohn@hooknose.lbl.gov> wrote:
1047 # > >>> print type(builtin_check.get_global_binding('__builtins__'))
1047 # > >>> print type(builtin_check.get_global_binding('__builtins__'))
1048 # > <type 'dict'>
1048 # > <type 'dict'>
1049 # > >>> print type(__builtins__)
1049 # > >>> print type(__builtins__)
1050 # > <type 'module'>
1050 # > <type 'module'>
1051 # > Is this difference in return value intentional?
1051 # > Is this difference in return value intentional?
1052
1052
1053 # Well, it's documented that '__builtins__' can be either a dictionary
1053 # Well, it's documented that '__builtins__' can be either a dictionary
1054 # or a module, and it's been that way for a long time. Whether it's
1054 # or a module, and it's been that way for a long time. Whether it's
1055 # intentional (or sensible), I don't know. In any case, the idea is
1055 # intentional (or sensible), I don't know. In any case, the idea is
1056 # that if you need to access the built-in namespace directly, you
1056 # that if you need to access the built-in namespace directly, you
1057 # should start with "import __builtin__" (note, no 's') which will
1057 # should start with "import __builtin__" (note, no 's') which will
1058 # definitely give you a module. Yeah, it's somewhat confusing:-(.
1058 # definitely give you a module. Yeah, it's somewhat confusing:-(.
1059
1059
1060 # These routines return a properly built module and dict as needed by
1060 # These routines return a properly built module and dict as needed by
1061 # the rest of the code, and can also be used by extension writers to
1061 # the rest of the code, and can also be used by extension writers to
1062 # generate properly initialized namespaces.
1062 # generate properly initialized namespaces.
1063 if (user_ns is not None) or (user_module is not None):
1063 if (user_ns is not None) or (user_module is not None):
1064 self.default_user_namespaces = False
1064 self.default_user_namespaces = False
1065 self.user_module, self.user_ns = self.prepare_user_module(user_module, user_ns)
1065 self.user_module, self.user_ns = self.prepare_user_module(user_module, user_ns)
1066
1066
1067 # A record of hidden variables we have added to the user namespace, so
1067 # A record of hidden variables we have added to the user namespace, so
1068 # we can list later only variables defined in actual interactive use.
1068 # we can list later only variables defined in actual interactive use.
1069 self.user_ns_hidden = {}
1069 self.user_ns_hidden = {}
1070
1070
1071 # Now that FakeModule produces a real module, we've run into a nasty
1071 # Now that FakeModule produces a real module, we've run into a nasty
1072 # problem: after script execution (via %run), the module where the user
1072 # problem: after script execution (via %run), the module where the user
1073 # code ran is deleted. Now that this object is a true module (needed
1073 # code ran is deleted. Now that this object is a true module (needed
1074 # so doctest and other tools work correctly), the Python module
1074 # so doctest and other tools work correctly), the Python module
1075 # teardown mechanism runs over it, and sets to None every variable
1075 # teardown mechanism runs over it, and sets to None every variable
1076 # present in that module. Top-level references to objects from the
1076 # present in that module. Top-level references to objects from the
1077 # script survive, because the user_ns is updated with them. However,
1077 # script survive, because the user_ns is updated with them. However,
1078 # calling functions defined in the script that use other things from
1078 # calling functions defined in the script that use other things from
1079 # the script will fail, because the function's closure had references
1079 # the script will fail, because the function's closure had references
1080 # to the original objects, which are now all None. So we must protect
1080 # to the original objects, which are now all None. So we must protect
1081 # these modules from deletion by keeping a cache.
1081 # these modules from deletion by keeping a cache.
1082 #
1082 #
1083 # To avoid keeping stale modules around (we only need the one from the
1083 # To avoid keeping stale modules around (we only need the one from the
1084 # last run), we use a dict keyed with the full path to the script, so
1084 # last run), we use a dict keyed with the full path to the script, so
1085 # only the last version of the module is held in the cache. Note,
1085 # only the last version of the module is held in the cache. Note,
1086 # however, that we must cache the module *namespace contents* (their
1086 # however, that we must cache the module *namespace contents* (their
1087 # __dict__). Because if we try to cache the actual modules, old ones
1087 # __dict__). Because if we try to cache the actual modules, old ones
1088 # (uncached) could be destroyed while still holding references (such as
1088 # (uncached) could be destroyed while still holding references (such as
1089 # those held by GUI objects that tend to be long-lived)>
1089 # those held by GUI objects that tend to be long-lived)>
1090 #
1090 #
1091 # The %reset command will flush this cache. See the cache_main_mod()
1091 # The %reset command will flush this cache. See the cache_main_mod()
1092 # and clear_main_mod_cache() methods for details on use.
1092 # and clear_main_mod_cache() methods for details on use.
1093
1093
1094 # This is the cache used for 'main' namespaces
1094 # This is the cache used for 'main' namespaces
1095 self._main_mod_cache = {}
1095 self._main_mod_cache = {}
1096
1096
1097 # A table holding all the namespaces IPython deals with, so that
1097 # A table holding all the namespaces IPython deals with, so that
1098 # introspection facilities can search easily.
1098 # introspection facilities can search easily.
1099 self.ns_table = {'user_global':self.user_module.__dict__,
1099 self.ns_table = {'user_global':self.user_module.__dict__,
1100 'user_local':self.user_ns,
1100 'user_local':self.user_ns,
1101 'builtin':builtin_mod.__dict__
1101 'builtin':builtin_mod.__dict__
1102 }
1102 }
1103
1103
1104 @property
1104 @property
1105 def user_global_ns(self):
1105 def user_global_ns(self):
1106 return self.user_module.__dict__
1106 return self.user_module.__dict__
1107
1107
1108 def prepare_user_module(self, user_module=None, user_ns=None):
1108 def prepare_user_module(self, user_module=None, user_ns=None):
1109 """Prepare the module and namespace in which user code will be run.
1109 """Prepare the module and namespace in which user code will be run.
1110
1110
1111 When IPython is started normally, both parameters are None: a new module
1111 When IPython is started normally, both parameters are None: a new module
1112 is created automatically, and its __dict__ used as the namespace.
1112 is created automatically, and its __dict__ used as the namespace.
1113
1113
1114 If only user_module is provided, its __dict__ is used as the namespace.
1114 If only user_module is provided, its __dict__ is used as the namespace.
1115 If only user_ns is provided, a dummy module is created, and user_ns
1115 If only user_ns is provided, a dummy module is created, and user_ns
1116 becomes the global namespace. If both are provided (as they may be
1116 becomes the global namespace. If both are provided (as they may be
1117 when embedding), user_ns is the local namespace, and user_module
1117 when embedding), user_ns is the local namespace, and user_module
1118 provides the global namespace.
1118 provides the global namespace.
1119
1119
1120 Parameters
1120 Parameters
1121 ----------
1121 ----------
1122 user_module : module, optional
1122 user_module : module, optional
1123 The current user module in which IPython is being run. If None,
1123 The current user module in which IPython is being run. If None,
1124 a clean module will be created.
1124 a clean module will be created.
1125 user_ns : dict, optional
1125 user_ns : dict, optional
1126 A namespace in which to run interactive commands.
1126 A namespace in which to run interactive commands.
1127
1127
1128 Returns
1128 Returns
1129 -------
1129 -------
1130 A tuple of user_module and user_ns, each properly initialised.
1130 A tuple of user_module and user_ns, each properly initialised.
1131 """
1131 """
1132 if user_module is None and user_ns is not None:
1132 if user_module is None and user_ns is not None:
1133 user_ns.setdefault("__name__", "__main__")
1133 user_ns.setdefault("__name__", "__main__")
1134 user_module = DummyMod()
1134 user_module = DummyMod()
1135 user_module.__dict__ = user_ns
1135 user_module.__dict__ = user_ns
1136
1136
1137 if user_module is None:
1137 if user_module is None:
1138 user_module = types.ModuleType("__main__",
1138 user_module = types.ModuleType("__main__",
1139 doc="Automatically created module for IPython interactive environment")
1139 doc="Automatically created module for IPython interactive environment")
1140
1140
1141 # We must ensure that __builtin__ (without the final 's') is always
1141 # We must ensure that __builtin__ (without the final 's') is always
1142 # available and pointing to the __builtin__ *module*. For more details:
1142 # available and pointing to the __builtin__ *module*. For more details:
1143 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
1143 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
1144 user_module.__dict__.setdefault('__builtin__', builtin_mod)
1144 user_module.__dict__.setdefault('__builtin__', builtin_mod)
1145 user_module.__dict__.setdefault('__builtins__', builtin_mod)
1145 user_module.__dict__.setdefault('__builtins__', builtin_mod)
1146
1146
1147 if user_ns is None:
1147 if user_ns is None:
1148 user_ns = user_module.__dict__
1148 user_ns = user_module.__dict__
1149
1149
1150 return user_module, user_ns
1150 return user_module, user_ns
1151
1151
1152 def init_sys_modules(self):
1152 def init_sys_modules(self):
1153 # We need to insert into sys.modules something that looks like a
1153 # We need to insert into sys.modules something that looks like a
1154 # module but which accesses the IPython namespace, for shelve and
1154 # module but which accesses the IPython namespace, for shelve and
1155 # pickle to work interactively. Normally they rely on getting
1155 # pickle to work interactively. Normally they rely on getting
1156 # everything out of __main__, but for embedding purposes each IPython
1156 # everything out of __main__, but for embedding purposes each IPython
1157 # instance has its own private namespace, so we can't go shoving
1157 # instance has its own private namespace, so we can't go shoving
1158 # everything into __main__.
1158 # everything into __main__.
1159
1159
1160 # note, however, that we should only do this for non-embedded
1160 # note, however, that we should only do this for non-embedded
1161 # ipythons, which really mimic the __main__.__dict__ with their own
1161 # ipythons, which really mimic the __main__.__dict__ with their own
1162 # namespace. Embedded instances, on the other hand, should not do
1162 # namespace. Embedded instances, on the other hand, should not do
1163 # this because they need to manage the user local/global namespaces
1163 # this because they need to manage the user local/global namespaces
1164 # only, but they live within a 'normal' __main__ (meaning, they
1164 # only, but they live within a 'normal' __main__ (meaning, they
1165 # shouldn't overtake the execution environment of the script they're
1165 # shouldn't overtake the execution environment of the script they're
1166 # embedded in).
1166 # embedded in).
1167
1167
1168 # This is overridden in the InteractiveShellEmbed subclass to a no-op.
1168 # This is overridden in the InteractiveShellEmbed subclass to a no-op.
1169 main_name = self.user_module.__name__
1169 main_name = self.user_module.__name__
1170 sys.modules[main_name] = self.user_module
1170 sys.modules[main_name] = self.user_module
1171
1171
1172 def init_user_ns(self):
1172 def init_user_ns(self):
1173 """Initialize all user-visible namespaces to their minimum defaults.
1173 """Initialize all user-visible namespaces to their minimum defaults.
1174
1174
1175 Certain history lists are also initialized here, as they effectively
1175 Certain history lists are also initialized here, as they effectively
1176 act as user namespaces.
1176 act as user namespaces.
1177
1177
1178 Notes
1178 Notes
1179 -----
1179 -----
1180 All data structures here are only filled in, they are NOT reset by this
1180 All data structures here are only filled in, they are NOT reset by this
1181 method. If they were not empty before, data will simply be added to
1181 method. If they were not empty before, data will simply be added to
1182 therm.
1182 therm.
1183 """
1183 """
1184 # This function works in two parts: first we put a few things in
1184 # This function works in two parts: first we put a few things in
1185 # user_ns, and we sync that contents into user_ns_hidden so that these
1185 # user_ns, and we sync that contents into user_ns_hidden so that these
1186 # initial variables aren't shown by %who. After the sync, we add the
1186 # initial variables aren't shown by %who. After the sync, we add the
1187 # rest of what we *do* want the user to see with %who even on a new
1187 # rest of what we *do* want the user to see with %who even on a new
1188 # session (probably nothing, so they really only see their own stuff)
1188 # session (probably nothing, so they really only see their own stuff)
1189
1189
1190 # The user dict must *always* have a __builtin__ reference to the
1190 # The user dict must *always* have a __builtin__ reference to the
1191 # Python standard __builtin__ namespace, which must be imported.
1191 # Python standard __builtin__ namespace, which must be imported.
1192 # This is so that certain operations in prompt evaluation can be
1192 # This is so that certain operations in prompt evaluation can be
1193 # reliably executed with builtins. Note that we can NOT use
1193 # reliably executed with builtins. Note that we can NOT use
1194 # __builtins__ (note the 's'), because that can either be a dict or a
1194 # __builtins__ (note the 's'), because that can either be a dict or a
1195 # module, and can even mutate at runtime, depending on the context
1195 # module, and can even mutate at runtime, depending on the context
1196 # (Python makes no guarantees on it). In contrast, __builtin__ is
1196 # (Python makes no guarantees on it). In contrast, __builtin__ is
1197 # always a module object, though it must be explicitly imported.
1197 # always a module object, though it must be explicitly imported.
1198
1198
1199 # For more details:
1199 # For more details:
1200 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
1200 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
1201 ns = {}
1201 ns = {}
1202
1202
1203 # make global variables for user access to the histories
1203 # make global variables for user access to the histories
1204 ns['_ih'] = self.history_manager.input_hist_parsed
1204 ns['_ih'] = self.history_manager.input_hist_parsed
1205 ns['_oh'] = self.history_manager.output_hist
1205 ns['_oh'] = self.history_manager.output_hist
1206 ns['_dh'] = self.history_manager.dir_hist
1206 ns['_dh'] = self.history_manager.dir_hist
1207
1207
1208 # user aliases to input and output histories. These shouldn't show up
1208 # user aliases to input and output histories. These shouldn't show up
1209 # in %who, as they can have very large reprs.
1209 # in %who, as they can have very large reprs.
1210 ns['In'] = self.history_manager.input_hist_parsed
1210 ns['In'] = self.history_manager.input_hist_parsed
1211 ns['Out'] = self.history_manager.output_hist
1211 ns['Out'] = self.history_manager.output_hist
1212
1212
1213 # Store myself as the public api!!!
1213 # Store myself as the public api!!!
1214 ns['get_ipython'] = self.get_ipython
1214 ns['get_ipython'] = self.get_ipython
1215
1215
1216 ns['exit'] = self.exiter
1216 ns['exit'] = self.exiter
1217 ns['quit'] = self.exiter
1217 ns['quit'] = self.exiter
1218
1218
1219 # Sync what we've added so far to user_ns_hidden so these aren't seen
1219 # Sync what we've added so far to user_ns_hidden so these aren't seen
1220 # by %who
1220 # by %who
1221 self.user_ns_hidden.update(ns)
1221 self.user_ns_hidden.update(ns)
1222
1222
1223 # Anything put into ns now would show up in %who. Think twice before
1223 # Anything put into ns now would show up in %who. Think twice before
1224 # putting anything here, as we really want %who to show the user their
1224 # putting anything here, as we really want %who to show the user their
1225 # stuff, not our variables.
1225 # stuff, not our variables.
1226
1226
1227 # Finally, update the real user's namespace
1227 # Finally, update the real user's namespace
1228 self.user_ns.update(ns)
1228 self.user_ns.update(ns)
1229
1229
1230 @property
1230 @property
1231 def all_ns_refs(self):
1231 def all_ns_refs(self):
1232 """Get a list of references to all the namespace dictionaries in which
1232 """Get a list of references to all the namespace dictionaries in which
1233 IPython might store a user-created object.
1233 IPython might store a user-created object.
1234
1234
1235 Note that this does not include the displayhook, which also caches
1235 Note that this does not include the displayhook, which also caches
1236 objects from the output."""
1236 objects from the output."""
1237 return [self.user_ns, self.user_global_ns, self.user_ns_hidden] + \
1237 return [self.user_ns, self.user_global_ns, self.user_ns_hidden] + \
1238 [m.__dict__ for m in self._main_mod_cache.values()]
1238 [m.__dict__ for m in self._main_mod_cache.values()]
1239
1239
1240 def reset(self, new_session=True):
1240 def reset(self, new_session=True):
1241 """Clear all internal namespaces, and attempt to release references to
1241 """Clear all internal namespaces, and attempt to release references to
1242 user objects.
1242 user objects.
1243
1243
1244 If new_session is True, a new history session will be opened.
1244 If new_session is True, a new history session will be opened.
1245 """
1245 """
1246 # Clear histories
1246 # Clear histories
1247 self.history_manager.reset(new_session)
1247 self.history_manager.reset(new_session)
1248 # Reset counter used to index all histories
1248 # Reset counter used to index all histories
1249 if new_session:
1249 if new_session:
1250 self.execution_count = 1
1250 self.execution_count = 1
1251
1251
1252 # Reset last execution result
1252 # Reset last execution result
1253 self.last_execution_succeeded = True
1253 self.last_execution_succeeded = True
1254 self.last_execution_result = None
1254 self.last_execution_result = None
1255
1255
1256 # Flush cached output items
1256 # Flush cached output items
1257 if self.displayhook.do_full_cache:
1257 if self.displayhook.do_full_cache:
1258 self.displayhook.flush()
1258 self.displayhook.flush()
1259
1259
1260 # The main execution namespaces must be cleared very carefully,
1260 # The main execution namespaces must be cleared very carefully,
1261 # skipping the deletion of the builtin-related keys, because doing so
1261 # skipping the deletion of the builtin-related keys, because doing so
1262 # would cause errors in many object's __del__ methods.
1262 # would cause errors in many object's __del__ methods.
1263 if self.user_ns is not self.user_global_ns:
1263 if self.user_ns is not self.user_global_ns:
1264 self.user_ns.clear()
1264 self.user_ns.clear()
1265 ns = self.user_global_ns
1265 ns = self.user_global_ns
1266 drop_keys = set(ns.keys())
1266 drop_keys = set(ns.keys())
1267 drop_keys.discard('__builtin__')
1267 drop_keys.discard('__builtin__')
1268 drop_keys.discard('__builtins__')
1268 drop_keys.discard('__builtins__')
1269 drop_keys.discard('__name__')
1269 drop_keys.discard('__name__')
1270 for k in drop_keys:
1270 for k in drop_keys:
1271 del ns[k]
1271 del ns[k]
1272
1272
1273 self.user_ns_hidden.clear()
1273 self.user_ns_hidden.clear()
1274
1274
1275 # Restore the user namespaces to minimal usability
1275 # Restore the user namespaces to minimal usability
1276 self.init_user_ns()
1276 self.init_user_ns()
1277
1277
1278 # Restore the default and user aliases
1278 # Restore the default and user aliases
1279 self.alias_manager.clear_aliases()
1279 self.alias_manager.clear_aliases()
1280 self.alias_manager.init_aliases()
1280 self.alias_manager.init_aliases()
1281
1281
1282 # Flush the private list of module references kept for script
1282 # Flush the private list of module references kept for script
1283 # execution protection
1283 # execution protection
1284 self.clear_main_mod_cache()
1284 self.clear_main_mod_cache()
1285
1285
1286 def del_var(self, varname, by_name=False):
1286 def del_var(self, varname, by_name=False):
1287 """Delete a variable from the various namespaces, so that, as
1287 """Delete a variable from the various namespaces, so that, as
1288 far as possible, we're not keeping any hidden references to it.
1288 far as possible, we're not keeping any hidden references to it.
1289
1289
1290 Parameters
1290 Parameters
1291 ----------
1291 ----------
1292 varname : str
1292 varname : str
1293 The name of the variable to delete.
1293 The name of the variable to delete.
1294 by_name : bool
1294 by_name : bool
1295 If True, delete variables with the given name in each
1295 If True, delete variables with the given name in each
1296 namespace. If False (default), find the variable in the user
1296 namespace. If False (default), find the variable in the user
1297 namespace, and delete references to it.
1297 namespace, and delete references to it.
1298 """
1298 """
1299 if varname in ('__builtin__', '__builtins__'):
1299 if varname in ('__builtin__', '__builtins__'):
1300 raise ValueError("Refusing to delete %s" % varname)
1300 raise ValueError("Refusing to delete %s" % varname)
1301
1301
1302 ns_refs = self.all_ns_refs
1302 ns_refs = self.all_ns_refs
1303
1303
1304 if by_name: # Delete by name
1304 if by_name: # Delete by name
1305 for ns in ns_refs:
1305 for ns in ns_refs:
1306 try:
1306 try:
1307 del ns[varname]
1307 del ns[varname]
1308 except KeyError:
1308 except KeyError:
1309 pass
1309 pass
1310 else: # Delete by object
1310 else: # Delete by object
1311 try:
1311 try:
1312 obj = self.user_ns[varname]
1312 obj = self.user_ns[varname]
1313 except KeyError:
1313 except KeyError:
1314 raise NameError("name '%s' is not defined" % varname)
1314 raise NameError("name '%s' is not defined" % varname)
1315 # Also check in output history
1315 # Also check in output history
1316 ns_refs.append(self.history_manager.output_hist)
1316 ns_refs.append(self.history_manager.output_hist)
1317 for ns in ns_refs:
1317 for ns in ns_refs:
1318 to_delete = [n for n, o in ns.items() if o is obj]
1318 to_delete = [n for n, o in ns.items() if o is obj]
1319 for name in to_delete:
1319 for name in to_delete:
1320 del ns[name]
1320 del ns[name]
1321
1321
1322 # Ensure it is removed from the last execution result
1322 # Ensure it is removed from the last execution result
1323 if self.last_execution_result.result is obj:
1323 if self.last_execution_result.result is obj:
1324 self.last_execution_result = None
1324 self.last_execution_result = None
1325
1325
1326 # displayhook keeps extra references, but not in a dictionary
1326 # displayhook keeps extra references, but not in a dictionary
1327 for name in ('_', '__', '___'):
1327 for name in ('_', '__', '___'):
1328 if getattr(self.displayhook, name) is obj:
1328 if getattr(self.displayhook, name) is obj:
1329 setattr(self.displayhook, name, None)
1329 setattr(self.displayhook, name, None)
1330
1330
1331 def reset_selective(self, regex=None):
1331 def reset_selective(self, regex=None):
1332 """Clear selective variables from internal namespaces based on a
1332 """Clear selective variables from internal namespaces based on a
1333 specified regular expression.
1333 specified regular expression.
1334
1334
1335 Parameters
1335 Parameters
1336 ----------
1336 ----------
1337 regex : string or compiled pattern, optional
1337 regex : string or compiled pattern, optional
1338 A regular expression pattern that will be used in searching
1338 A regular expression pattern that will be used in searching
1339 variable names in the users namespaces.
1339 variable names in the users namespaces.
1340 """
1340 """
1341 if regex is not None:
1341 if regex is not None:
1342 try:
1342 try:
1343 m = re.compile(regex)
1343 m = re.compile(regex)
1344 except TypeError:
1344 except TypeError:
1345 raise TypeError('regex must be a string or compiled pattern')
1345 raise TypeError('regex must be a string or compiled pattern')
1346 # Search for keys in each namespace that match the given regex
1346 # Search for keys in each namespace that match the given regex
1347 # If a match is found, delete the key/value pair.
1347 # If a match is found, delete the key/value pair.
1348 for ns in self.all_ns_refs:
1348 for ns in self.all_ns_refs:
1349 for var in ns:
1349 for var in ns:
1350 if m.search(var):
1350 if m.search(var):
1351 del ns[var]
1351 del ns[var]
1352
1352
1353 def push(self, variables, interactive=True):
1353 def push(self, variables, interactive=True):
1354 """Inject a group of variables into the IPython user namespace.
1354 """Inject a group of variables into the IPython user namespace.
1355
1355
1356 Parameters
1356 Parameters
1357 ----------
1357 ----------
1358 variables : dict, str or list/tuple of str
1358 variables : dict, str or list/tuple of str
1359 The variables to inject into the user's namespace. If a dict, a
1359 The variables to inject into the user's namespace. If a dict, a
1360 simple update is done. If a str, the string is assumed to have
1360 simple update is done. If a str, the string is assumed to have
1361 variable names separated by spaces. A list/tuple of str can also
1361 variable names separated by spaces. A list/tuple of str can also
1362 be used to give the variable names. If just the variable names are
1362 be used to give the variable names. If just the variable names are
1363 give (list/tuple/str) then the variable values looked up in the
1363 give (list/tuple/str) then the variable values looked up in the
1364 callers frame.
1364 callers frame.
1365 interactive : bool
1365 interactive : bool
1366 If True (default), the variables will be listed with the ``who``
1366 If True (default), the variables will be listed with the ``who``
1367 magic.
1367 magic.
1368 """
1368 """
1369 vdict = None
1369 vdict = None
1370
1370
1371 # We need a dict of name/value pairs to do namespace updates.
1371 # We need a dict of name/value pairs to do namespace updates.
1372 if isinstance(variables, dict):
1372 if isinstance(variables, dict):
1373 vdict = variables
1373 vdict = variables
1374 elif isinstance(variables, (str, list, tuple)):
1374 elif isinstance(variables, (str, list, tuple)):
1375 if isinstance(variables, str):
1375 if isinstance(variables, str):
1376 vlist = variables.split()
1376 vlist = variables.split()
1377 else:
1377 else:
1378 vlist = variables
1378 vlist = variables
1379 vdict = {}
1379 vdict = {}
1380 cf = sys._getframe(1)
1380 cf = sys._getframe(1)
1381 for name in vlist:
1381 for name in vlist:
1382 try:
1382 try:
1383 vdict[name] = eval(name, cf.f_globals, cf.f_locals)
1383 vdict[name] = eval(name, cf.f_globals, cf.f_locals)
1384 except:
1384 except:
1385 print('Could not get variable %s from %s' %
1385 print('Could not get variable %s from %s' %
1386 (name,cf.f_code.co_name))
1386 (name,cf.f_code.co_name))
1387 else:
1387 else:
1388 raise ValueError('variables must be a dict/str/list/tuple')
1388 raise ValueError('variables must be a dict/str/list/tuple')
1389
1389
1390 # Propagate variables to user namespace
1390 # Propagate variables to user namespace
1391 self.user_ns.update(vdict)
1391 self.user_ns.update(vdict)
1392
1392
1393 # And configure interactive visibility
1393 # And configure interactive visibility
1394 user_ns_hidden = self.user_ns_hidden
1394 user_ns_hidden = self.user_ns_hidden
1395 if interactive:
1395 if interactive:
1396 for name in vdict:
1396 for name in vdict:
1397 user_ns_hidden.pop(name, None)
1397 user_ns_hidden.pop(name, None)
1398 else:
1398 else:
1399 user_ns_hidden.update(vdict)
1399 user_ns_hidden.update(vdict)
1400
1400
1401 def drop_by_id(self, variables):
1401 def drop_by_id(self, variables):
1402 """Remove a dict of variables from the user namespace, if they are the
1402 """Remove a dict of variables from the user namespace, if they are the
1403 same as the values in the dictionary.
1403 same as the values in the dictionary.
1404
1404
1405 This is intended for use by extensions: variables that they've added can
1405 This is intended for use by extensions: variables that they've added can
1406 be taken back out if they are unloaded, without removing any that the
1406 be taken back out if they are unloaded, without removing any that the
1407 user has overwritten.
1407 user has overwritten.
1408
1408
1409 Parameters
1409 Parameters
1410 ----------
1410 ----------
1411 variables : dict
1411 variables : dict
1412 A dictionary mapping object names (as strings) to the objects.
1412 A dictionary mapping object names (as strings) to the objects.
1413 """
1413 """
1414 for name, obj in variables.items():
1414 for name, obj in variables.items():
1415 if name in self.user_ns and self.user_ns[name] is obj:
1415 if name in self.user_ns and self.user_ns[name] is obj:
1416 del self.user_ns[name]
1416 del self.user_ns[name]
1417 self.user_ns_hidden.pop(name, None)
1417 self.user_ns_hidden.pop(name, None)
1418
1418
1419 #-------------------------------------------------------------------------
1419 #-------------------------------------------------------------------------
1420 # Things related to object introspection
1420 # Things related to object introspection
1421 #-------------------------------------------------------------------------
1421 #-------------------------------------------------------------------------
1422
1422
1423 def _ofind(self, oname, namespaces=None):
1423 def _ofind(self, oname, namespaces=None):
1424 """Find an object in the available namespaces.
1424 """Find an object in the available namespaces.
1425
1425
1426 self._ofind(oname) -> dict with keys: found,obj,ospace,ismagic
1426 self._ofind(oname) -> dict with keys: found,obj,ospace,ismagic
1427
1427
1428 Has special code to detect magic functions.
1428 Has special code to detect magic functions.
1429 """
1429 """
1430 oname = oname.strip()
1430 oname = oname.strip()
1431 if not oname.startswith(ESC_MAGIC) and \
1431 if not oname.startswith(ESC_MAGIC) and \
1432 not oname.startswith(ESC_MAGIC2) and \
1432 not oname.startswith(ESC_MAGIC2) and \
1433 not all(a.isidentifier() for a in oname.split(".")):
1433 not all(a.isidentifier() for a in oname.split(".")):
1434 return {'found': False}
1434 return {'found': False}
1435
1435
1436 if namespaces is None:
1436 if namespaces is None:
1437 # Namespaces to search in:
1437 # Namespaces to search in:
1438 # Put them in a list. The order is important so that we
1438 # Put them in a list. The order is important so that we
1439 # find things in the same order that Python finds them.
1439 # find things in the same order that Python finds them.
1440 namespaces = [ ('Interactive', self.user_ns),
1440 namespaces = [ ('Interactive', self.user_ns),
1441 ('Interactive (global)', self.user_global_ns),
1441 ('Interactive (global)', self.user_global_ns),
1442 ('Python builtin', builtin_mod.__dict__),
1442 ('Python builtin', builtin_mod.__dict__),
1443 ]
1443 ]
1444
1444
1445 ismagic = False
1445 ismagic = False
1446 isalias = False
1446 isalias = False
1447 found = False
1447 found = False
1448 ospace = None
1448 ospace = None
1449 parent = None
1449 parent = None
1450 obj = None
1450 obj = None
1451
1451
1452 # Look for the given name by splitting it in parts. If the head is
1452 # Look for the given name by splitting it in parts. If the head is
1453 # found, then we look for all the remaining parts as members, and only
1453 # found, then we look for all the remaining parts as members, and only
1454 # declare success if we can find them all.
1454 # declare success if we can find them all.
1455 oname_parts = oname.split('.')
1455 oname_parts = oname.split('.')
1456 oname_head, oname_rest = oname_parts[0],oname_parts[1:]
1456 oname_head, oname_rest = oname_parts[0],oname_parts[1:]
1457 for nsname,ns in namespaces:
1457 for nsname,ns in namespaces:
1458 try:
1458 try:
1459 obj = ns[oname_head]
1459 obj = ns[oname_head]
1460 except KeyError:
1460 except KeyError:
1461 continue
1461 continue
1462 else:
1462 else:
1463 for idx, part in enumerate(oname_rest):
1463 for idx, part in enumerate(oname_rest):
1464 try:
1464 try:
1465 parent = obj
1465 parent = obj
1466 # The last part is looked up in a special way to avoid
1466 # The last part is looked up in a special way to avoid
1467 # descriptor invocation as it may raise or have side
1467 # descriptor invocation as it may raise or have side
1468 # effects.
1468 # effects.
1469 if idx == len(oname_rest) - 1:
1469 if idx == len(oname_rest) - 1:
1470 obj = self._getattr_property(obj, part)
1470 obj = self._getattr_property(obj, part)
1471 else:
1471 else:
1472 obj = getattr(obj, part)
1472 obj = getattr(obj, part)
1473 except:
1473 except:
1474 # Blanket except b/c some badly implemented objects
1474 # Blanket except b/c some badly implemented objects
1475 # allow __getattr__ to raise exceptions other than
1475 # allow __getattr__ to raise exceptions other than
1476 # AttributeError, which then crashes IPython.
1476 # AttributeError, which then crashes IPython.
1477 break
1477 break
1478 else:
1478 else:
1479 # If we finish the for loop (no break), we got all members
1479 # If we finish the for loop (no break), we got all members
1480 found = True
1480 found = True
1481 ospace = nsname
1481 ospace = nsname
1482 break # namespace loop
1482 break # namespace loop
1483
1483
1484 # Try to see if it's magic
1484 # Try to see if it's magic
1485 if not found:
1485 if not found:
1486 obj = None
1486 obj = None
1487 if oname.startswith(ESC_MAGIC2):
1487 if oname.startswith(ESC_MAGIC2):
1488 oname = oname.lstrip(ESC_MAGIC2)
1488 oname = oname.lstrip(ESC_MAGIC2)
1489 obj = self.find_cell_magic(oname)
1489 obj = self.find_cell_magic(oname)
1490 elif oname.startswith(ESC_MAGIC):
1490 elif oname.startswith(ESC_MAGIC):
1491 oname = oname.lstrip(ESC_MAGIC)
1491 oname = oname.lstrip(ESC_MAGIC)
1492 obj = self.find_line_magic(oname)
1492 obj = self.find_line_magic(oname)
1493 else:
1493 else:
1494 # search without prefix, so run? will find %run?
1494 # search without prefix, so run? will find %run?
1495 obj = self.find_line_magic(oname)
1495 obj = self.find_line_magic(oname)
1496 if obj is None:
1496 if obj is None:
1497 obj = self.find_cell_magic(oname)
1497 obj = self.find_cell_magic(oname)
1498 if obj is not None:
1498 if obj is not None:
1499 found = True
1499 found = True
1500 ospace = 'IPython internal'
1500 ospace = 'IPython internal'
1501 ismagic = True
1501 ismagic = True
1502 isalias = isinstance(obj, Alias)
1502 isalias = isinstance(obj, Alias)
1503
1503
1504 # Last try: special-case some literals like '', [], {}, etc:
1504 # Last try: special-case some literals like '', [], {}, etc:
1505 if not found and oname_head in ["''",'""','[]','{}','()']:
1505 if not found and oname_head in ["''",'""','[]','{}','()']:
1506 obj = eval(oname_head)
1506 obj = eval(oname_head)
1507 found = True
1507 found = True
1508 ospace = 'Interactive'
1508 ospace = 'Interactive'
1509
1509
1510 return {
1510 return {
1511 'obj':obj,
1511 'obj':obj,
1512 'found':found,
1512 'found':found,
1513 'parent':parent,
1513 'parent':parent,
1514 'ismagic':ismagic,
1514 'ismagic':ismagic,
1515 'isalias':isalias,
1515 'isalias':isalias,
1516 'namespace':ospace
1516 'namespace':ospace
1517 }
1517 }
1518
1518
1519 @staticmethod
1519 @staticmethod
1520 def _getattr_property(obj, attrname):
1520 def _getattr_property(obj, attrname):
1521 """Property-aware getattr to use in object finding.
1521 """Property-aware getattr to use in object finding.
1522
1522
1523 If attrname represents a property, return it unevaluated (in case it has
1523 If attrname represents a property, return it unevaluated (in case it has
1524 side effects or raises an error.
1524 side effects or raises an error.
1525
1525
1526 """
1526 """
1527 if not isinstance(obj, type):
1527 if not isinstance(obj, type):
1528 try:
1528 try:
1529 # `getattr(type(obj), attrname)` is not guaranteed to return
1529 # `getattr(type(obj), attrname)` is not guaranteed to return
1530 # `obj`, but does so for property:
1530 # `obj`, but does so for property:
1531 #
1531 #
1532 # property.__get__(self, None, cls) -> self
1532 # property.__get__(self, None, cls) -> self
1533 #
1533 #
1534 # The universal alternative is to traverse the mro manually
1534 # The universal alternative is to traverse the mro manually
1535 # searching for attrname in class dicts.
1535 # searching for attrname in class dicts.
1536 attr = getattr(type(obj), attrname)
1536 attr = getattr(type(obj), attrname)
1537 except AttributeError:
1537 except AttributeError:
1538 pass
1538 pass
1539 else:
1539 else:
1540 # This relies on the fact that data descriptors (with both
1540 # This relies on the fact that data descriptors (with both
1541 # __get__ & __set__ magic methods) take precedence over
1541 # __get__ & __set__ magic methods) take precedence over
1542 # instance-level attributes:
1542 # instance-level attributes:
1543 #
1543 #
1544 # class A(object):
1544 # class A(object):
1545 # @property
1545 # @property
1546 # def foobar(self): return 123
1546 # def foobar(self): return 123
1547 # a = A()
1547 # a = A()
1548 # a.__dict__['foobar'] = 345
1548 # a.__dict__['foobar'] = 345
1549 # a.foobar # == 123
1549 # a.foobar # == 123
1550 #
1550 #
1551 # So, a property may be returned right away.
1551 # So, a property may be returned right away.
1552 if isinstance(attr, property):
1552 if isinstance(attr, property):
1553 return attr
1553 return attr
1554
1554
1555 # Nothing helped, fall back.
1555 # Nothing helped, fall back.
1556 return getattr(obj, attrname)
1556 return getattr(obj, attrname)
1557
1557
1558 def _object_find(self, oname, namespaces=None):
1558 def _object_find(self, oname, namespaces=None):
1559 """Find an object and return a struct with info about it."""
1559 """Find an object and return a struct with info about it."""
1560 return Struct(self._ofind(oname, namespaces))
1560 return Struct(self._ofind(oname, namespaces))
1561
1561
1562 def _inspect(self, meth, oname, namespaces=None, **kw):
1562 def _inspect(self, meth, oname, namespaces=None, **kw):
1563 """Generic interface to the inspector system.
1563 """Generic interface to the inspector system.
1564
1564
1565 This function is meant to be called by pdef, pdoc & friends.
1565 This function is meant to be called by pdef, pdoc & friends.
1566 """
1566 """
1567 info = self._object_find(oname, namespaces)
1567 info = self._object_find(oname, namespaces)
1568 docformat = sphinxify if self.sphinxify_docstring else None
1568 docformat = sphinxify if self.sphinxify_docstring else None
1569 if info.found:
1569 if info.found:
1570 pmethod = getattr(self.inspector, meth)
1570 pmethod = getattr(self.inspector, meth)
1571 # TODO: only apply format_screen to the plain/text repr of the mime
1571 # TODO: only apply format_screen to the plain/text repr of the mime
1572 # bundle.
1572 # bundle.
1573 formatter = format_screen if info.ismagic else docformat
1573 formatter = format_screen if info.ismagic else docformat
1574 if meth == 'pdoc':
1574 if meth == 'pdoc':
1575 pmethod(info.obj, oname, formatter)
1575 pmethod(info.obj, oname, formatter)
1576 elif meth == 'pinfo':
1576 elif meth == 'pinfo':
1577 pmethod(info.obj, oname, formatter, info,
1577 pmethod(info.obj, oname, formatter, info,
1578 enable_html_pager=self.enable_html_pager, **kw)
1578 enable_html_pager=self.enable_html_pager, **kw)
1579 else:
1579 else:
1580 pmethod(info.obj, oname)
1580 pmethod(info.obj, oname)
1581 else:
1581 else:
1582 print('Object `%s` not found.' % oname)
1582 print('Object `%s` not found.' % oname)
1583 return 'not found' # so callers can take other action
1583 return 'not found' # so callers can take other action
1584
1584
1585 def object_inspect(self, oname, detail_level=0):
1585 def object_inspect(self, oname, detail_level=0):
1586 """Get object info about oname"""
1586 """Get object info about oname"""
1587 with self.builtin_trap:
1587 with self.builtin_trap:
1588 info = self._object_find(oname)
1588 info = self._object_find(oname)
1589 if info.found:
1589 if info.found:
1590 return self.inspector.info(info.obj, oname, info=info,
1590 return self.inspector.info(info.obj, oname, info=info,
1591 detail_level=detail_level
1591 detail_level=detail_level
1592 )
1592 )
1593 else:
1593 else:
1594 return oinspect.object_info(name=oname, found=False)
1594 return oinspect.object_info(name=oname, found=False)
1595
1595
1596 def object_inspect_text(self, oname, detail_level=0):
1596 def object_inspect_text(self, oname, detail_level=0):
1597 """Get object info as formatted text"""
1597 """Get object info as formatted text"""
1598 return self.object_inspect_mime(oname, detail_level)['text/plain']
1598 return self.object_inspect_mime(oname, detail_level)['text/plain']
1599
1599
1600 def object_inspect_mime(self, oname, detail_level=0):
1600 def object_inspect_mime(self, oname, detail_level=0):
1601 """Get object info as a mimebundle of formatted representations.
1601 """Get object info as a mimebundle of formatted representations.
1602
1602
1603 A mimebundle is a dictionary, keyed by mime-type.
1603 A mimebundle is a dictionary, keyed by mime-type.
1604 It must always have the key `'text/plain'`.
1604 It must always have the key `'text/plain'`.
1605 """
1605 """
1606 with self.builtin_trap:
1606 with self.builtin_trap:
1607 info = self._object_find(oname)
1607 info = self._object_find(oname)
1608 if info.found:
1608 if info.found:
1609 return self.inspector._get_info(info.obj, oname, info=info,
1609 return self.inspector._get_info(info.obj, oname, info=info,
1610 detail_level=detail_level
1610 detail_level=detail_level
1611 )
1611 )
1612 else:
1612 else:
1613 raise KeyError(oname)
1613 raise KeyError(oname)
1614
1614
1615 #-------------------------------------------------------------------------
1615 #-------------------------------------------------------------------------
1616 # Things related to history management
1616 # Things related to history management
1617 #-------------------------------------------------------------------------
1617 #-------------------------------------------------------------------------
1618
1618
1619 def init_history(self):
1619 def init_history(self):
1620 """Sets up the command history, and starts regular autosaves."""
1620 """Sets up the command history, and starts regular autosaves."""
1621 self.history_manager = HistoryManager(shell=self, parent=self)
1621 self.history_manager = HistoryManager(shell=self, parent=self)
1622 self.configurables.append(self.history_manager)
1622 self.configurables.append(self.history_manager)
1623
1623
1624 #-------------------------------------------------------------------------
1624 #-------------------------------------------------------------------------
1625 # Things related to exception handling and tracebacks (not debugging)
1625 # Things related to exception handling and tracebacks (not debugging)
1626 #-------------------------------------------------------------------------
1626 #-------------------------------------------------------------------------
1627
1627
1628 debugger_cls = Pdb
1628 debugger_cls = Pdb
1629
1629
1630 def init_traceback_handlers(self, custom_exceptions):
1630 def init_traceback_handlers(self, custom_exceptions):
1631 # Syntax error handler.
1631 # Syntax error handler.
1632 self.SyntaxTB = ultratb.SyntaxTB(color_scheme='NoColor', parent=self)
1632 self.SyntaxTB = ultratb.SyntaxTB(color_scheme='NoColor', parent=self)
1633
1633
1634 # The interactive one is initialized with an offset, meaning we always
1634 # The interactive one is initialized with an offset, meaning we always
1635 # want to remove the topmost item in the traceback, which is our own
1635 # want to remove the topmost item in the traceback, which is our own
1636 # internal code. Valid modes: ['Plain','Context','Verbose']
1636 # internal code. Valid modes: ['Plain','Context','Verbose']
1637 self.InteractiveTB = ultratb.AutoFormattedTB(mode = 'Plain',
1637 self.InteractiveTB = ultratb.AutoFormattedTB(mode = 'Plain',
1638 color_scheme='NoColor',
1638 color_scheme='NoColor',
1639 tb_offset = 1,
1639 tb_offset = 1,
1640 check_cache=check_linecache_ipython,
1640 check_cache=check_linecache_ipython,
1641 debugger_cls=self.debugger_cls, parent=self)
1641 debugger_cls=self.debugger_cls, parent=self)
1642
1642
1643 # The instance will store a pointer to the system-wide exception hook,
1643 # The instance will store a pointer to the system-wide exception hook,
1644 # so that runtime code (such as magics) can access it. This is because
1644 # so that runtime code (such as magics) can access it. This is because
1645 # during the read-eval loop, it may get temporarily overwritten.
1645 # during the read-eval loop, it may get temporarily overwritten.
1646 self.sys_excepthook = sys.excepthook
1646 self.sys_excepthook = sys.excepthook
1647
1647
1648 # and add any custom exception handlers the user may have specified
1648 # and add any custom exception handlers the user may have specified
1649 self.set_custom_exc(*custom_exceptions)
1649 self.set_custom_exc(*custom_exceptions)
1650
1650
1651 # Set the exception mode
1651 # Set the exception mode
1652 self.InteractiveTB.set_mode(mode=self.xmode)
1652 self.InteractiveTB.set_mode(mode=self.xmode)
1653
1653
1654 def set_custom_exc(self, exc_tuple, handler):
1654 def set_custom_exc(self, exc_tuple, handler):
1655 """set_custom_exc(exc_tuple, handler)
1655 """set_custom_exc(exc_tuple, handler)
1656
1656
1657 Set a custom exception handler, which will be called if any of the
1657 Set a custom exception handler, which will be called if any of the
1658 exceptions in exc_tuple occur in the mainloop (specifically, in the
1658 exceptions in exc_tuple occur in the mainloop (specifically, in the
1659 run_code() method).
1659 run_code() method).
1660
1660
1661 Parameters
1661 Parameters
1662 ----------
1662 ----------
1663
1663
1664 exc_tuple : tuple of exception classes
1664 exc_tuple : tuple of exception classes
1665 A *tuple* of exception classes, for which to call the defined
1665 A *tuple* of exception classes, for which to call the defined
1666 handler. It is very important that you use a tuple, and NOT A
1666 handler. It is very important that you use a tuple, and NOT A
1667 LIST here, because of the way Python's except statement works. If
1667 LIST here, because of the way Python's except statement works. If
1668 you only want to trap a single exception, use a singleton tuple::
1668 you only want to trap a single exception, use a singleton tuple::
1669
1669
1670 exc_tuple == (MyCustomException,)
1670 exc_tuple == (MyCustomException,)
1671
1671
1672 handler : callable
1672 handler : callable
1673 handler must have the following signature::
1673 handler must have the following signature::
1674
1674
1675 def my_handler(self, etype, value, tb, tb_offset=None):
1675 def my_handler(self, etype, value, tb, tb_offset=None):
1676 ...
1676 ...
1677 return structured_traceback
1677 return structured_traceback
1678
1678
1679 Your handler must return a structured traceback (a list of strings),
1679 Your handler must return a structured traceback (a list of strings),
1680 or None.
1680 or None.
1681
1681
1682 This will be made into an instance method (via types.MethodType)
1682 This will be made into an instance method (via types.MethodType)
1683 of IPython itself, and it will be called if any of the exceptions
1683 of IPython itself, and it will be called if any of the exceptions
1684 listed in the exc_tuple are caught. If the handler is None, an
1684 listed in the exc_tuple are caught. If the handler is None, an
1685 internal basic one is used, which just prints basic info.
1685 internal basic one is used, which just prints basic info.
1686
1686
1687 To protect IPython from crashes, if your handler ever raises an
1687 To protect IPython from crashes, if your handler ever raises an
1688 exception or returns an invalid result, it will be immediately
1688 exception or returns an invalid result, it will be immediately
1689 disabled.
1689 disabled.
1690
1690
1691 WARNING: by putting in your own exception handler into IPython's main
1691 WARNING: by putting in your own exception handler into IPython's main
1692 execution loop, you run a very good chance of nasty crashes. This
1692 execution loop, you run a very good chance of nasty crashes. This
1693 facility should only be used if you really know what you are doing."""
1693 facility should only be used if you really know what you are doing."""
1694 if not isinstance(exc_tuple, tuple):
1694 if not isinstance(exc_tuple, tuple):
1695 raise TypeError("The custom exceptions must be given as a tuple.")
1695 raise TypeError("The custom exceptions must be given as a tuple.")
1696
1696
1697 def dummy_handler(self, etype, value, tb, tb_offset=None):
1697 def dummy_handler(self, etype, value, tb, tb_offset=None):
1698 print('*** Simple custom exception handler ***')
1698 print('*** Simple custom exception handler ***')
1699 print('Exception type :', etype)
1699 print('Exception type :', etype)
1700 print('Exception value:', value)
1700 print('Exception value:', value)
1701 print('Traceback :', tb)
1701 print('Traceback :', tb)
1702
1702
1703 def validate_stb(stb):
1703 def validate_stb(stb):
1704 """validate structured traceback return type
1704 """validate structured traceback return type
1705
1705
1706 return type of CustomTB *should* be a list of strings, but allow
1706 return type of CustomTB *should* be a list of strings, but allow
1707 single strings or None, which are harmless.
1707 single strings or None, which are harmless.
1708
1708
1709 This function will *always* return a list of strings,
1709 This function will *always* return a list of strings,
1710 and will raise a TypeError if stb is inappropriate.
1710 and will raise a TypeError if stb is inappropriate.
1711 """
1711 """
1712 msg = "CustomTB must return list of strings, not %r" % stb
1712 msg = "CustomTB must return list of strings, not %r" % stb
1713 if stb is None:
1713 if stb is None:
1714 return []
1714 return []
1715 elif isinstance(stb, str):
1715 elif isinstance(stb, str):
1716 return [stb]
1716 return [stb]
1717 elif not isinstance(stb, list):
1717 elif not isinstance(stb, list):
1718 raise TypeError(msg)
1718 raise TypeError(msg)
1719 # it's a list
1719 # it's a list
1720 for line in stb:
1720 for line in stb:
1721 # check every element
1721 # check every element
1722 if not isinstance(line, str):
1722 if not isinstance(line, str):
1723 raise TypeError(msg)
1723 raise TypeError(msg)
1724 return stb
1724 return stb
1725
1725
1726 if handler is None:
1726 if handler is None:
1727 wrapped = dummy_handler
1727 wrapped = dummy_handler
1728 else:
1728 else:
1729 def wrapped(self,etype,value,tb,tb_offset=None):
1729 def wrapped(self,etype,value,tb,tb_offset=None):
1730 """wrap CustomTB handler, to protect IPython from user code
1730 """wrap CustomTB handler, to protect IPython from user code
1731
1731
1732 This makes it harder (but not impossible) for custom exception
1732 This makes it harder (but not impossible) for custom exception
1733 handlers to crash IPython.
1733 handlers to crash IPython.
1734 """
1734 """
1735 try:
1735 try:
1736 stb = handler(self,etype,value,tb,tb_offset=tb_offset)
1736 stb = handler(self,etype,value,tb,tb_offset=tb_offset)
1737 return validate_stb(stb)
1737 return validate_stb(stb)
1738 except:
1738 except:
1739 # clear custom handler immediately
1739 # clear custom handler immediately
1740 self.set_custom_exc((), None)
1740 self.set_custom_exc((), None)
1741 print("Custom TB Handler failed, unregistering", file=sys.stderr)
1741 print("Custom TB Handler failed, unregistering", file=sys.stderr)
1742 # show the exception in handler first
1742 # show the exception in handler first
1743 stb = self.InteractiveTB.structured_traceback(*sys.exc_info())
1743 stb = self.InteractiveTB.structured_traceback(*sys.exc_info())
1744 print(self.InteractiveTB.stb2text(stb))
1744 print(self.InteractiveTB.stb2text(stb))
1745 print("The original exception:")
1745 print("The original exception:")
1746 stb = self.InteractiveTB.structured_traceback(
1746 stb = self.InteractiveTB.structured_traceback(
1747 (etype,value,tb), tb_offset=tb_offset
1747 (etype,value,tb), tb_offset=tb_offset
1748 )
1748 )
1749 return stb
1749 return stb
1750
1750
1751 self.CustomTB = types.MethodType(wrapped,self)
1751 self.CustomTB = types.MethodType(wrapped,self)
1752 self.custom_exceptions = exc_tuple
1752 self.custom_exceptions = exc_tuple
1753
1753
1754 def excepthook(self, etype, value, tb):
1754 def excepthook(self, etype, value, tb):
1755 """One more defense for GUI apps that call sys.excepthook.
1755 """One more defense for GUI apps that call sys.excepthook.
1756
1756
1757 GUI frameworks like wxPython trap exceptions and call
1757 GUI frameworks like wxPython trap exceptions and call
1758 sys.excepthook themselves. I guess this is a feature that
1758 sys.excepthook themselves. I guess this is a feature that
1759 enables them to keep running after exceptions that would
1759 enables them to keep running after exceptions that would
1760 otherwise kill their mainloop. This is a bother for IPython
1760 otherwise kill their mainloop. This is a bother for IPython
1761 which excepts to catch all of the program exceptions with a try:
1761 which excepts to catch all of the program exceptions with a try:
1762 except: statement.
1762 except: statement.
1763
1763
1764 Normally, IPython sets sys.excepthook to a CrashHandler instance, so if
1764 Normally, IPython sets sys.excepthook to a CrashHandler instance, so if
1765 any app directly invokes sys.excepthook, it will look to the user like
1765 any app directly invokes sys.excepthook, it will look to the user like
1766 IPython crashed. In order to work around this, we can disable the
1766 IPython crashed. In order to work around this, we can disable the
1767 CrashHandler and replace it with this excepthook instead, which prints a
1767 CrashHandler and replace it with this excepthook instead, which prints a
1768 regular traceback using our InteractiveTB. In this fashion, apps which
1768 regular traceback using our InteractiveTB. In this fashion, apps which
1769 call sys.excepthook will generate a regular-looking exception from
1769 call sys.excepthook will generate a regular-looking exception from
1770 IPython, and the CrashHandler will only be triggered by real IPython
1770 IPython, and the CrashHandler will only be triggered by real IPython
1771 crashes.
1771 crashes.
1772
1772
1773 This hook should be used sparingly, only in places which are not likely
1773 This hook should be used sparingly, only in places which are not likely
1774 to be true IPython errors.
1774 to be true IPython errors.
1775 """
1775 """
1776 self.showtraceback((etype, value, tb), tb_offset=0)
1776 self.showtraceback((etype, value, tb), tb_offset=0)
1777
1777
1778 def _get_exc_info(self, exc_tuple=None):
1778 def _get_exc_info(self, exc_tuple=None):
1779 """get exc_info from a given tuple, sys.exc_info() or sys.last_type etc.
1779 """get exc_info from a given tuple, sys.exc_info() or sys.last_type etc.
1780
1780
1781 Ensures sys.last_type,value,traceback hold the exc_info we found,
1781 Ensures sys.last_type,value,traceback hold the exc_info we found,
1782 from whichever source.
1782 from whichever source.
1783
1783
1784 raises ValueError if none of these contain any information
1784 raises ValueError if none of these contain any information
1785 """
1785 """
1786 if exc_tuple is None:
1786 if exc_tuple is None:
1787 etype, value, tb = sys.exc_info()
1787 etype, value, tb = sys.exc_info()
1788 else:
1788 else:
1789 etype, value, tb = exc_tuple
1789 etype, value, tb = exc_tuple
1790
1790
1791 if etype is None:
1791 if etype is None:
1792 if hasattr(sys, 'last_type'):
1792 if hasattr(sys, 'last_type'):
1793 etype, value, tb = sys.last_type, sys.last_value, \
1793 etype, value, tb = sys.last_type, sys.last_value, \
1794 sys.last_traceback
1794 sys.last_traceback
1795
1795
1796 if etype is None:
1796 if etype is None:
1797 raise ValueError("No exception to find")
1797 raise ValueError("No exception to find")
1798
1798
1799 # Now store the exception info in sys.last_type etc.
1799 # Now store the exception info in sys.last_type etc.
1800 # WARNING: these variables are somewhat deprecated and not
1800 # WARNING: these variables are somewhat deprecated and not
1801 # necessarily safe to use in a threaded environment, but tools
1801 # necessarily safe to use in a threaded environment, but tools
1802 # like pdb depend on their existence, so let's set them. If we
1802 # like pdb depend on their existence, so let's set them. If we
1803 # find problems in the field, we'll need to revisit their use.
1803 # find problems in the field, we'll need to revisit their use.
1804 sys.last_type = etype
1804 sys.last_type = etype
1805 sys.last_value = value
1805 sys.last_value = value
1806 sys.last_traceback = tb
1806 sys.last_traceback = tb
1807
1807
1808 return etype, value, tb
1808 return etype, value, tb
1809
1809
1810 def show_usage_error(self, exc):
1810 def show_usage_error(self, exc):
1811 """Show a short message for UsageErrors
1811 """Show a short message for UsageErrors
1812
1812
1813 These are special exceptions that shouldn't show a traceback.
1813 These are special exceptions that shouldn't show a traceback.
1814 """
1814 """
1815 print("UsageError: %s" % exc, file=sys.stderr)
1815 print("UsageError: %s" % exc, file=sys.stderr)
1816
1816
1817 def get_exception_only(self, exc_tuple=None):
1817 def get_exception_only(self, exc_tuple=None):
1818 """
1818 """
1819 Return as a string (ending with a newline) the exception that
1819 Return as a string (ending with a newline) the exception that
1820 just occurred, without any traceback.
1820 just occurred, without any traceback.
1821 """
1821 """
1822 etype, value, tb = self._get_exc_info(exc_tuple)
1822 etype, value, tb = self._get_exc_info(exc_tuple)
1823 msg = traceback.format_exception_only(etype, value)
1823 msg = traceback.format_exception_only(etype, value)
1824 return ''.join(msg)
1824 return ''.join(msg)
1825
1825
1826 def showtraceback(self, exc_tuple=None, filename=None, tb_offset=None,
1826 def showtraceback(self, exc_tuple=None, filename=None, tb_offset=None,
1827 exception_only=False, running_compiled_code=False):
1827 exception_only=False, running_compiled_code=False):
1828 """Display the exception that just occurred.
1828 """Display the exception that just occurred.
1829
1829
1830 If nothing is known about the exception, this is the method which
1830 If nothing is known about the exception, this is the method which
1831 should be used throughout the code for presenting user tracebacks,
1831 should be used throughout the code for presenting user tracebacks,
1832 rather than directly invoking the InteractiveTB object.
1832 rather than directly invoking the InteractiveTB object.
1833
1833
1834 A specific showsyntaxerror() also exists, but this method can take
1834 A specific showsyntaxerror() also exists, but this method can take
1835 care of calling it if needed, so unless you are explicitly catching a
1835 care of calling it if needed, so unless you are explicitly catching a
1836 SyntaxError exception, don't try to analyze the stack manually and
1836 SyntaxError exception, don't try to analyze the stack manually and
1837 simply call this method."""
1837 simply call this method."""
1838
1838
1839 try:
1839 try:
1840 try:
1840 try:
1841 etype, value, tb = self._get_exc_info(exc_tuple)
1841 etype, value, tb = self._get_exc_info(exc_tuple)
1842 except ValueError:
1842 except ValueError:
1843 print('No traceback available to show.', file=sys.stderr)
1843 print('No traceback available to show.', file=sys.stderr)
1844 return
1844 return
1845
1845
1846 if issubclass(etype, SyntaxError):
1846 if issubclass(etype, SyntaxError):
1847 # Though this won't be called by syntax errors in the input
1847 # Though this won't be called by syntax errors in the input
1848 # line, there may be SyntaxError cases with imported code.
1848 # line, there may be SyntaxError cases with imported code.
1849 self.showsyntaxerror(filename, running_compiled_code)
1849 self.showsyntaxerror(filename, running_compiled_code)
1850 elif etype is UsageError:
1850 elif etype is UsageError:
1851 self.show_usage_error(value)
1851 self.show_usage_error(value)
1852 else:
1852 else:
1853 if exception_only:
1853 if exception_only:
1854 stb = ['An exception has occurred, use %tb to see '
1854 stb = ['An exception has occurred, use %tb to see '
1855 'the full traceback.\n']
1855 'the full traceback.\n']
1856 stb.extend(self.InteractiveTB.get_exception_only(etype,
1856 stb.extend(self.InteractiveTB.get_exception_only(etype,
1857 value))
1857 value))
1858 else:
1858 else:
1859 try:
1859 try:
1860 # Exception classes can customise their traceback - we
1860 # Exception classes can customise their traceback - we
1861 # use this in IPython.parallel for exceptions occurring
1861 # use this in IPython.parallel for exceptions occurring
1862 # in the engines. This should return a list of strings.
1862 # in the engines. This should return a list of strings.
1863 stb = value._render_traceback_()
1863 stb = value._render_traceback_()
1864 except Exception:
1864 except Exception:
1865 stb = self.InteractiveTB.structured_traceback(etype,
1865 stb = self.InteractiveTB.structured_traceback(etype,
1866 value, tb, tb_offset=tb_offset)
1866 value, tb, tb_offset=tb_offset)
1867
1867
1868 self._showtraceback(etype, value, stb)
1868 self._showtraceback(etype, value, stb)
1869 if self.call_pdb:
1869 if self.call_pdb:
1870 # drop into debugger
1870 # drop into debugger
1871 self.debugger(force=True)
1871 self.debugger(force=True)
1872 return
1872 return
1873
1873
1874 # Actually show the traceback
1874 # Actually show the traceback
1875 self._showtraceback(etype, value, stb)
1875 self._showtraceback(etype, value, stb)
1876
1876
1877 except KeyboardInterrupt:
1877 except KeyboardInterrupt:
1878 print('\n' + self.get_exception_only(), file=sys.stderr)
1878 print('\n' + self.get_exception_only(), file=sys.stderr)
1879
1879
1880 def _showtraceback(self, etype, evalue, stb):
1880 def _showtraceback(self, etype, evalue, stb):
1881 """Actually show a traceback.
1881 """Actually show a traceback.
1882
1882
1883 Subclasses may override this method to put the traceback on a different
1883 Subclasses may override this method to put the traceback on a different
1884 place, like a side channel.
1884 place, like a side channel.
1885 """
1885 """
1886 print(self.InteractiveTB.stb2text(stb))
1886 print(self.InteractiveTB.stb2text(stb))
1887
1887
1888 def showsyntaxerror(self, filename=None, running_compiled_code=False):
1888 def showsyntaxerror(self, filename=None, running_compiled_code=False):
1889 """Display the syntax error that just occurred.
1889 """Display the syntax error that just occurred.
1890
1890
1891 This doesn't display a stack trace because there isn't one.
1891 This doesn't display a stack trace because there isn't one.
1892
1892
1893 If a filename is given, it is stuffed in the exception instead
1893 If a filename is given, it is stuffed in the exception instead
1894 of what was there before (because Python's parser always uses
1894 of what was there before (because Python's parser always uses
1895 "<string>" when reading from a string).
1895 "<string>" when reading from a string).
1896
1896
1897 If the syntax error occurred when running a compiled code (i.e. running_compile_code=True),
1897 If the syntax error occurred when running a compiled code (i.e. running_compile_code=True),
1898 longer stack trace will be displayed.
1898 longer stack trace will be displayed.
1899 """
1899 """
1900 etype, value, last_traceback = self._get_exc_info()
1900 etype, value, last_traceback = self._get_exc_info()
1901
1901
1902 if filename and issubclass(etype, SyntaxError):
1902 if filename and issubclass(etype, SyntaxError):
1903 try:
1903 try:
1904 value.filename = filename
1904 value.filename = filename
1905 except:
1905 except:
1906 # Not the format we expect; leave it alone
1906 # Not the format we expect; leave it alone
1907 pass
1907 pass
1908
1908
1909 # If the error occurred when executing compiled code, we should provide full stacktrace.
1909 # If the error occurred when executing compiled code, we should provide full stacktrace.
1910 elist = traceback.extract_tb(last_traceback) if running_compiled_code else []
1910 elist = traceback.extract_tb(last_traceback) if running_compiled_code else []
1911 stb = self.SyntaxTB.structured_traceback(etype, value, elist)
1911 stb = self.SyntaxTB.structured_traceback(etype, value, elist)
1912 self._showtraceback(etype, value, stb)
1912 self._showtraceback(etype, value, stb)
1913
1913
1914 # This is overridden in TerminalInteractiveShell to show a message about
1914 # This is overridden in TerminalInteractiveShell to show a message about
1915 # the %paste magic.
1915 # the %paste magic.
1916 def showindentationerror(self):
1916 def showindentationerror(self):
1917 """Called by _run_cell when there's an IndentationError in code entered
1917 """Called by _run_cell when there's an IndentationError in code entered
1918 at the prompt.
1918 at the prompt.
1919
1919
1920 This is overridden in TerminalInteractiveShell to show a message about
1920 This is overridden in TerminalInteractiveShell to show a message about
1921 the %paste magic."""
1921 the %paste magic."""
1922 self.showsyntaxerror()
1922 self.showsyntaxerror()
1923
1923
1924 #-------------------------------------------------------------------------
1924 #-------------------------------------------------------------------------
1925 # Things related to readline
1925 # Things related to readline
1926 #-------------------------------------------------------------------------
1926 #-------------------------------------------------------------------------
1927
1927
1928 def init_readline(self):
1928 def init_readline(self):
1929 """DEPRECATED
1929 """DEPRECATED
1930
1930
1931 Moved to terminal subclass, here only to simplify the init logic."""
1931 Moved to terminal subclass, here only to simplify the init logic."""
1932 # Set a number of methods that depend on readline to be no-op
1932 # Set a number of methods that depend on readline to be no-op
1933 warnings.warn('`init_readline` is no-op since IPython 5.0 and is Deprecated',
1933 warnings.warn('`init_readline` is no-op since IPython 5.0 and is Deprecated',
1934 DeprecationWarning, stacklevel=2)
1934 DeprecationWarning, stacklevel=2)
1935 self.set_custom_completer = no_op
1935 self.set_custom_completer = no_op
1936
1936
1937 @skip_doctest
1937 @skip_doctest
1938 def set_next_input(self, s, replace=False):
1938 def set_next_input(self, s, replace=False):
1939 """ Sets the 'default' input string for the next command line.
1939 """ Sets the 'default' input string for the next command line.
1940
1940
1941 Example::
1941 Example::
1942
1942
1943 In [1]: _ip.set_next_input("Hello Word")
1943 In [1]: _ip.set_next_input("Hello Word")
1944 In [2]: Hello Word_ # cursor is here
1944 In [2]: Hello Word_ # cursor is here
1945 """
1945 """
1946 self.rl_next_input = s
1946 self.rl_next_input = s
1947
1947
1948 def _indent_current_str(self):
1948 def _indent_current_str(self):
1949 """return the current level of indentation as a string"""
1949 """return the current level of indentation as a string"""
1950 return self.input_splitter.get_indent_spaces() * ' '
1950 return self.input_splitter.get_indent_spaces() * ' '
1951
1951
1952 #-------------------------------------------------------------------------
1952 #-------------------------------------------------------------------------
1953 # Things related to text completion
1953 # Things related to text completion
1954 #-------------------------------------------------------------------------
1954 #-------------------------------------------------------------------------
1955
1955
1956 def init_completer(self):
1956 def init_completer(self):
1957 """Initialize the completion machinery.
1957 """Initialize the completion machinery.
1958
1958
1959 This creates completion machinery that can be used by client code,
1959 This creates completion machinery that can be used by client code,
1960 either interactively in-process (typically triggered by the readline
1960 either interactively in-process (typically triggered by the readline
1961 library), programmatically (such as in test suites) or out-of-process
1961 library), programmatically (such as in test suites) or out-of-process
1962 (typically over the network by remote frontends).
1962 (typically over the network by remote frontends).
1963 """
1963 """
1964 from IPython.core.completer import IPCompleter
1964 from IPython.core.completer import IPCompleter
1965 from IPython.core.completerlib import (module_completer,
1965 from IPython.core.completerlib import (module_completer,
1966 magic_run_completer, cd_completer, reset_completer)
1966 magic_run_completer, cd_completer, reset_completer)
1967
1967
1968 self.Completer = IPCompleter(shell=self,
1968 self.Completer = IPCompleter(shell=self,
1969 namespace=self.user_ns,
1969 namespace=self.user_ns,
1970 global_namespace=self.user_global_ns,
1970 global_namespace=self.user_global_ns,
1971 parent=self,
1971 parent=self,
1972 )
1972 )
1973 self.configurables.append(self.Completer)
1973 self.configurables.append(self.Completer)
1974
1974
1975 # Add custom completers to the basic ones built into IPCompleter
1975 # Add custom completers to the basic ones built into IPCompleter
1976 sdisp = self.strdispatchers.get('complete_command', StrDispatch())
1976 sdisp = self.strdispatchers.get('complete_command', StrDispatch())
1977 self.strdispatchers['complete_command'] = sdisp
1977 self.strdispatchers['complete_command'] = sdisp
1978 self.Completer.custom_completers = sdisp
1978 self.Completer.custom_completers = sdisp
1979
1979
1980 self.set_hook('complete_command', module_completer, str_key = 'import')
1980 self.set_hook('complete_command', module_completer, str_key = 'import')
1981 self.set_hook('complete_command', module_completer, str_key = 'from')
1981 self.set_hook('complete_command', module_completer, str_key = 'from')
1982 self.set_hook('complete_command', module_completer, str_key = '%aimport')
1982 self.set_hook('complete_command', module_completer, str_key = '%aimport')
1983 self.set_hook('complete_command', magic_run_completer, str_key = '%run')
1983 self.set_hook('complete_command', magic_run_completer, str_key = '%run')
1984 self.set_hook('complete_command', cd_completer, str_key = '%cd')
1984 self.set_hook('complete_command', cd_completer, str_key = '%cd')
1985 self.set_hook('complete_command', reset_completer, str_key = '%reset')
1985 self.set_hook('complete_command', reset_completer, str_key = '%reset')
1986
1986
1987
1987
1988 @skip_doctest
1988 @skip_doctest
1989 def complete(self, text, line=None, cursor_pos=None):
1989 def complete(self, text, line=None, cursor_pos=None):
1990 """Return the completed text and a list of completions.
1990 """Return the completed text and a list of completions.
1991
1991
1992 Parameters
1992 Parameters
1993 ----------
1993 ----------
1994
1994
1995 text : string
1995 text : string
1996 A string of text to be completed on. It can be given as empty and
1996 A string of text to be completed on. It can be given as empty and
1997 instead a line/position pair are given. In this case, the
1997 instead a line/position pair are given. In this case, the
1998 completer itself will split the line like readline does.
1998 completer itself will split the line like readline does.
1999
1999
2000 line : string, optional
2000 line : string, optional
2001 The complete line that text is part of.
2001 The complete line that text is part of.
2002
2002
2003 cursor_pos : int, optional
2003 cursor_pos : int, optional
2004 The position of the cursor on the input line.
2004 The position of the cursor on the input line.
2005
2005
2006 Returns
2006 Returns
2007 -------
2007 -------
2008 text : string
2008 text : string
2009 The actual text that was completed.
2009 The actual text that was completed.
2010
2010
2011 matches : list
2011 matches : list
2012 A sorted list with all possible completions.
2012 A sorted list with all possible completions.
2013
2013
2014 The optional arguments allow the completion to take more context into
2014 The optional arguments allow the completion to take more context into
2015 account, and are part of the low-level completion API.
2015 account, and are part of the low-level completion API.
2016
2016
2017 This is a wrapper around the completion mechanism, similar to what
2017 This is a wrapper around the completion mechanism, similar to what
2018 readline does at the command line when the TAB key is hit. By
2018 readline does at the command line when the TAB key is hit. By
2019 exposing it as a method, it can be used by other non-readline
2019 exposing it as a method, it can be used by other non-readline
2020 environments (such as GUIs) for text completion.
2020 environments (such as GUIs) for text completion.
2021
2021
2022 Simple usage example:
2022 Simple usage example:
2023
2023
2024 In [1]: x = 'hello'
2024 In [1]: x = 'hello'
2025
2025
2026 In [2]: _ip.complete('x.l')
2026 In [2]: _ip.complete('x.l')
2027 Out[2]: ('x.l', ['x.ljust', 'x.lower', 'x.lstrip'])
2027 Out[2]: ('x.l', ['x.ljust', 'x.lower', 'x.lstrip'])
2028 """
2028 """
2029
2029
2030 # Inject names into __builtin__ so we can complete on the added names.
2030 # Inject names into __builtin__ so we can complete on the added names.
2031 with self.builtin_trap:
2031 with self.builtin_trap:
2032 return self.Completer.complete(text, line, cursor_pos)
2032 return self.Completer.complete(text, line, cursor_pos)
2033
2033
2034 def set_custom_completer(self, completer, pos=0):
2034 def set_custom_completer(self, completer, pos=0):
2035 """Adds a new custom completer function.
2035 """Adds a new custom completer function.
2036
2036
2037 The position argument (defaults to 0) is the index in the completers
2037 The position argument (defaults to 0) is the index in the completers
2038 list where you want the completer to be inserted."""
2038 list where you want the completer to be inserted."""
2039
2039
2040 newcomp = types.MethodType(completer,self.Completer)
2040 newcomp = types.MethodType(completer,self.Completer)
2041 self.Completer.matchers.insert(pos,newcomp)
2041 self.Completer.matchers.insert(pos,newcomp)
2042
2042
2043 def set_completer_frame(self, frame=None):
2043 def set_completer_frame(self, frame=None):
2044 """Set the frame of the completer."""
2044 """Set the frame of the completer."""
2045 if frame:
2045 if frame:
2046 self.Completer.namespace = frame.f_locals
2046 self.Completer.namespace = frame.f_locals
2047 self.Completer.global_namespace = frame.f_globals
2047 self.Completer.global_namespace = frame.f_globals
2048 else:
2048 else:
2049 self.Completer.namespace = self.user_ns
2049 self.Completer.namespace = self.user_ns
2050 self.Completer.global_namespace = self.user_global_ns
2050 self.Completer.global_namespace = self.user_global_ns
2051
2051
2052 #-------------------------------------------------------------------------
2052 #-------------------------------------------------------------------------
2053 # Things related to magics
2053 # Things related to magics
2054 #-------------------------------------------------------------------------
2054 #-------------------------------------------------------------------------
2055
2055
2056 def init_magics(self):
2056 def init_magics(self):
2057 from IPython.core import magics as m
2057 from IPython.core import magics as m
2058 self.magics_manager = magic.MagicsManager(shell=self,
2058 self.magics_manager = magic.MagicsManager(shell=self,
2059 parent=self,
2059 parent=self,
2060 user_magics=m.UserMagics(self))
2060 user_magics=m.UserMagics(self))
2061 self.configurables.append(self.magics_manager)
2061 self.configurables.append(self.magics_manager)
2062
2062
2063 # Expose as public API from the magics manager
2063 # Expose as public API from the magics manager
2064 self.register_magics = self.magics_manager.register
2064 self.register_magics = self.magics_manager.register
2065
2065
2066 self.register_magics(m.AutoMagics, m.BasicMagics, m.CodeMagics,
2066 self.register_magics(m.AutoMagics, m.BasicMagics, m.CodeMagics,
2067 m.ConfigMagics, m.DisplayMagics, m.ExecutionMagics,
2067 m.ConfigMagics, m.DisplayMagics, m.ExecutionMagics,
2068 m.ExtensionMagics, m.HistoryMagics, m.LoggingMagics,
2068 m.ExtensionMagics, m.HistoryMagics, m.LoggingMagics,
2069 m.NamespaceMagics, m.OSMagics, m.PylabMagics, m.ScriptMagics,
2069 m.NamespaceMagics, m.OSMagics, m.PylabMagics, m.ScriptMagics,
2070 )
2070 )
2071
2071
2072 # Register Magic Aliases
2072 # Register Magic Aliases
2073 mman = self.magics_manager
2073 mman = self.magics_manager
2074 # FIXME: magic aliases should be defined by the Magics classes
2074 # FIXME: magic aliases should be defined by the Magics classes
2075 # or in MagicsManager, not here
2075 # or in MagicsManager, not here
2076 mman.register_alias('ed', 'edit')
2076 mman.register_alias('ed', 'edit')
2077 mman.register_alias('hist', 'history')
2077 mman.register_alias('hist', 'history')
2078 mman.register_alias('rep', 'recall')
2078 mman.register_alias('rep', 'recall')
2079 mman.register_alias('SVG', 'svg', 'cell')
2079 mman.register_alias('SVG', 'svg', 'cell')
2080 mman.register_alias('HTML', 'html', 'cell')
2080 mman.register_alias('HTML', 'html', 'cell')
2081 mman.register_alias('file', 'writefile', 'cell')
2081 mman.register_alias('file', 'writefile', 'cell')
2082
2082
2083 # FIXME: Move the color initialization to the DisplayHook, which
2083 # FIXME: Move the color initialization to the DisplayHook, which
2084 # should be split into a prompt manager and displayhook. We probably
2084 # should be split into a prompt manager and displayhook. We probably
2085 # even need a centralize colors management object.
2085 # even need a centralize colors management object.
2086 self.run_line_magic('colors', self.colors)
2086 self.run_line_magic('colors', self.colors)
2087
2087
2088 # Defined here so that it's included in the documentation
2088 # Defined here so that it's included in the documentation
2089 @functools.wraps(magic.MagicsManager.register_function)
2089 @functools.wraps(magic.MagicsManager.register_function)
2090 def register_magic_function(self, func, magic_kind='line', magic_name=None):
2090 def register_magic_function(self, func, magic_kind='line', magic_name=None):
2091 self.magics_manager.register_function(func,
2091 self.magics_manager.register_function(func,
2092 magic_kind=magic_kind, magic_name=magic_name)
2092 magic_kind=magic_kind, magic_name=magic_name)
2093
2093
2094 def run_line_magic(self, magic_name, line, _stack_depth=1):
2094 def run_line_magic(self, magic_name, line, _stack_depth=1):
2095 """Execute the given line magic.
2095 """Execute the given line magic.
2096
2096
2097 Parameters
2097 Parameters
2098 ----------
2098 ----------
2099 magic_name : str
2099 magic_name : str
2100 Name of the desired magic function, without '%' prefix.
2100 Name of the desired magic function, without '%' prefix.
2101
2101
2102 line : str
2102 line : str
2103 The rest of the input line as a single string.
2103 The rest of the input line as a single string.
2104
2104
2105 _stack_depth : int
2105 _stack_depth : int
2106 If run_line_magic() is called from magic() then _stack_depth=2.
2106 If run_line_magic() is called from magic() then _stack_depth=2.
2107 This is added to ensure backward compatibility for use of 'get_ipython().magic()'
2107 This is added to ensure backward compatibility for use of 'get_ipython().magic()'
2108 """
2108 """
2109 fn = self.find_line_magic(magic_name)
2109 fn = self.find_line_magic(magic_name)
2110 if fn is None:
2110 if fn is None:
2111 cm = self.find_cell_magic(magic_name)
2111 cm = self.find_cell_magic(magic_name)
2112 etpl = "Line magic function `%%%s` not found%s."
2112 etpl = "Line magic function `%%%s` not found%s."
2113 extra = '' if cm is None else (' (But cell magic `%%%%%s` exists, '
2113 extra = '' if cm is None else (' (But cell magic `%%%%%s` exists, '
2114 'did you mean that instead?)' % magic_name )
2114 'did you mean that instead?)' % magic_name )
2115 raise UsageError(etpl % (magic_name, extra))
2115 raise UsageError(etpl % (magic_name, extra))
2116 else:
2116 else:
2117 # Note: this is the distance in the stack to the user's frame.
2117 # Note: this is the distance in the stack to the user's frame.
2118 # This will need to be updated if the internal calling logic gets
2118 # This will need to be updated if the internal calling logic gets
2119 # refactored, or else we'll be expanding the wrong variables.
2119 # refactored, or else we'll be expanding the wrong variables.
2120
2120
2121 # Determine stack_depth depending on where run_line_magic() has been called
2121 # Determine stack_depth depending on where run_line_magic() has been called
2122 stack_depth = _stack_depth
2122 stack_depth = _stack_depth
2123 magic_arg_s = self.var_expand(line, stack_depth)
2123 magic_arg_s = self.var_expand(line, stack_depth)
2124 # Put magic args in a list so we can call with f(*a) syntax
2124 # Put magic args in a list so we can call with f(*a) syntax
2125 args = [magic_arg_s]
2125 args = [magic_arg_s]
2126 kwargs = {}
2126 kwargs = {}
2127 # Grab local namespace if we need it:
2127 # Grab local namespace if we need it:
2128 if getattr(fn, "needs_local_scope", False):
2128 if getattr(fn, "needs_local_scope", False):
2129 kwargs['local_ns'] = sys._getframe(stack_depth).f_locals
2129 kwargs['local_ns'] = sys._getframe(stack_depth).f_locals
2130 with self.builtin_trap:
2130 with self.builtin_trap:
2131 result = fn(*args,**kwargs)
2131 result = fn(*args,**kwargs)
2132 return result
2132 return result
2133
2133
2134 def run_cell_magic(self, magic_name, line, cell):
2134 def run_cell_magic(self, magic_name, line, cell):
2135 """Execute the given cell magic.
2135 """Execute the given cell magic.
2136
2136
2137 Parameters
2137 Parameters
2138 ----------
2138 ----------
2139 magic_name : str
2139 magic_name : str
2140 Name of the desired magic function, without '%' prefix.
2140 Name of the desired magic function, without '%' prefix.
2141
2141
2142 line : str
2142 line : str
2143 The rest of the first input line as a single string.
2143 The rest of the first input line as a single string.
2144
2144
2145 cell : str
2145 cell : str
2146 The body of the cell as a (possibly multiline) string.
2146 The body of the cell as a (possibly multiline) string.
2147 """
2147 """
2148 fn = self.find_cell_magic(magic_name)
2148 fn = self.find_cell_magic(magic_name)
2149 if fn is None:
2149 if fn is None:
2150 lm = self.find_line_magic(magic_name)
2150 lm = self.find_line_magic(magic_name)
2151 etpl = "Cell magic `%%{0}` not found{1}."
2151 etpl = "Cell magic `%%{0}` not found{1}."
2152 extra = '' if lm is None else (' (But line magic `%{0}` exists, '
2152 extra = '' if lm is None else (' (But line magic `%{0}` exists, '
2153 'did you mean that instead?)'.format(magic_name))
2153 'did you mean that instead?)'.format(magic_name))
2154 raise UsageError(etpl.format(magic_name, extra))
2154 raise UsageError(etpl.format(magic_name, extra))
2155 elif cell == '':
2155 elif cell == '':
2156 message = '%%{0} is a cell magic, but the cell body is empty.'.format(magic_name)
2156 message = '%%{0} is a cell magic, but the cell body is empty.'.format(magic_name)
2157 if self.find_line_magic(magic_name) is not None:
2157 if self.find_line_magic(magic_name) is not None:
2158 message += ' Did you mean the line magic %{0} (single %)?'.format(magic_name)
2158 message += ' Did you mean the line magic %{0} (single %)?'.format(magic_name)
2159 raise UsageError(message)
2159 raise UsageError(message)
2160 else:
2160 else:
2161 # Note: this is the distance in the stack to the user's frame.
2161 # Note: this is the distance in the stack to the user's frame.
2162 # This will need to be updated if the internal calling logic gets
2162 # This will need to be updated if the internal calling logic gets
2163 # refactored, or else we'll be expanding the wrong variables.
2163 # refactored, or else we'll be expanding the wrong variables.
2164 stack_depth = 2
2164 stack_depth = 2
2165 magic_arg_s = self.var_expand(line, stack_depth)
2165 magic_arg_s = self.var_expand(line, stack_depth)
2166 with self.builtin_trap:
2166 with self.builtin_trap:
2167 result = fn(magic_arg_s, cell)
2167 result = fn(magic_arg_s, cell)
2168 return result
2168 return result
2169
2169
2170 def find_line_magic(self, magic_name):
2170 def find_line_magic(self, magic_name):
2171 """Find and return a line magic by name.
2171 """Find and return a line magic by name.
2172
2172
2173 Returns None if the magic isn't found."""
2173 Returns None if the magic isn't found."""
2174 return self.magics_manager.magics['line'].get(magic_name)
2174 return self.magics_manager.magics['line'].get(magic_name)
2175
2175
2176 def find_cell_magic(self, magic_name):
2176 def find_cell_magic(self, magic_name):
2177 """Find and return a cell magic by name.
2177 """Find and return a cell magic by name.
2178
2178
2179 Returns None if the magic isn't found."""
2179 Returns None if the magic isn't found."""
2180 return self.magics_manager.magics['cell'].get(magic_name)
2180 return self.magics_manager.magics['cell'].get(magic_name)
2181
2181
2182 def find_magic(self, magic_name, magic_kind='line'):
2182 def find_magic(self, magic_name, magic_kind='line'):
2183 """Find and return a magic of the given type by name.
2183 """Find and return a magic of the given type by name.
2184
2184
2185 Returns None if the magic isn't found."""
2185 Returns None if the magic isn't found."""
2186 return self.magics_manager.magics[magic_kind].get(magic_name)
2186 return self.magics_manager.magics[magic_kind].get(magic_name)
2187
2187
2188 def magic(self, arg_s):
2188 def magic(self, arg_s):
2189 """DEPRECATED. Use run_line_magic() instead.
2189 """DEPRECATED. Use run_line_magic() instead.
2190
2190
2191 Call a magic function by name.
2191 Call a magic function by name.
2192
2192
2193 Input: a string containing the name of the magic function to call and
2193 Input: a string containing the name of the magic function to call and
2194 any additional arguments to be passed to the magic.
2194 any additional arguments to be passed to the magic.
2195
2195
2196 magic('name -opt foo bar') is equivalent to typing at the ipython
2196 magic('name -opt foo bar') is equivalent to typing at the ipython
2197 prompt:
2197 prompt:
2198
2198
2199 In[1]: %name -opt foo bar
2199 In[1]: %name -opt foo bar
2200
2200
2201 To call a magic without arguments, simply use magic('name').
2201 To call a magic without arguments, simply use magic('name').
2202
2202
2203 This provides a proper Python function to call IPython's magics in any
2203 This provides a proper Python function to call IPython's magics in any
2204 valid Python code you can type at the interpreter, including loops and
2204 valid Python code you can type at the interpreter, including loops and
2205 compound statements.
2205 compound statements.
2206 """
2206 """
2207 # TODO: should we issue a loud deprecation warning here?
2207 # TODO: should we issue a loud deprecation warning here?
2208 magic_name, _, magic_arg_s = arg_s.partition(' ')
2208 magic_name, _, magic_arg_s = arg_s.partition(' ')
2209 magic_name = magic_name.lstrip(prefilter.ESC_MAGIC)
2209 magic_name = magic_name.lstrip(prefilter.ESC_MAGIC)
2210 return self.run_line_magic(magic_name, magic_arg_s, _stack_depth=2)
2210 return self.run_line_magic(magic_name, magic_arg_s, _stack_depth=2)
2211
2211
2212 #-------------------------------------------------------------------------
2212 #-------------------------------------------------------------------------
2213 # Things related to macros
2213 # Things related to macros
2214 #-------------------------------------------------------------------------
2214 #-------------------------------------------------------------------------
2215
2215
2216 def define_macro(self, name, themacro):
2216 def define_macro(self, name, themacro):
2217 """Define a new macro
2217 """Define a new macro
2218
2218
2219 Parameters
2219 Parameters
2220 ----------
2220 ----------
2221 name : str
2221 name : str
2222 The name of the macro.
2222 The name of the macro.
2223 themacro : str or Macro
2223 themacro : str or Macro
2224 The action to do upon invoking the macro. If a string, a new
2224 The action to do upon invoking the macro. If a string, a new
2225 Macro object is created by passing the string to it.
2225 Macro object is created by passing the string to it.
2226 """
2226 """
2227
2227
2228 from IPython.core import macro
2228 from IPython.core import macro
2229
2229
2230 if isinstance(themacro, str):
2230 if isinstance(themacro, str):
2231 themacro = macro.Macro(themacro)
2231 themacro = macro.Macro(themacro)
2232 if not isinstance(themacro, macro.Macro):
2232 if not isinstance(themacro, macro.Macro):
2233 raise ValueError('A macro must be a string or a Macro instance.')
2233 raise ValueError('A macro must be a string or a Macro instance.')
2234 self.user_ns[name] = themacro
2234 self.user_ns[name] = themacro
2235
2235
2236 #-------------------------------------------------------------------------
2236 #-------------------------------------------------------------------------
2237 # Things related to the running of system commands
2237 # Things related to the running of system commands
2238 #-------------------------------------------------------------------------
2238 #-------------------------------------------------------------------------
2239
2239
2240 def system_piped(self, cmd):
2240 def system_piped(self, cmd):
2241 """Call the given cmd in a subprocess, piping stdout/err
2241 """Call the given cmd in a subprocess, piping stdout/err
2242
2242
2243 Parameters
2243 Parameters
2244 ----------
2244 ----------
2245 cmd : str
2245 cmd : str
2246 Command to execute (can not end in '&', as background processes are
2246 Command to execute (can not end in '&', as background processes are
2247 not supported. Should not be a command that expects input
2247 not supported. Should not be a command that expects input
2248 other than simple text.
2248 other than simple text.
2249 """
2249 """
2250 if cmd.rstrip().endswith('&'):
2250 if cmd.rstrip().endswith('&'):
2251 # this is *far* from a rigorous test
2251 # this is *far* from a rigorous test
2252 # We do not support backgrounding processes because we either use
2252 # We do not support backgrounding processes because we either use
2253 # pexpect or pipes to read from. Users can always just call
2253 # pexpect or pipes to read from. Users can always just call
2254 # os.system() or use ip.system=ip.system_raw
2254 # os.system() or use ip.system=ip.system_raw
2255 # if they really want a background process.
2255 # if they really want a background process.
2256 raise OSError("Background processes not supported.")
2256 raise OSError("Background processes not supported.")
2257
2257
2258 # we explicitly do NOT return the subprocess status code, because
2258 # we explicitly do NOT return the subprocess status code, because
2259 # a non-None value would trigger :func:`sys.displayhook` calls.
2259 # a non-None value would trigger :func:`sys.displayhook` calls.
2260 # Instead, we store the exit_code in user_ns.
2260 # Instead, we store the exit_code in user_ns.
2261 self.user_ns['_exit_code'] = system(self.var_expand(cmd, depth=1))
2261 self.user_ns['_exit_code'] = system(self.var_expand(cmd, depth=1))
2262
2262
2263 def system_raw(self, cmd):
2263 def system_raw(self, cmd):
2264 """Call the given cmd in a subprocess using os.system on Windows or
2264 """Call the given cmd in a subprocess using os.system on Windows or
2265 subprocess.call using the system shell on other platforms.
2265 subprocess.call using the system shell on other platforms.
2266
2266
2267 Parameters
2267 Parameters
2268 ----------
2268 ----------
2269 cmd : str
2269 cmd : str
2270 Command to execute.
2270 Command to execute.
2271 """
2271 """
2272 cmd = self.var_expand(cmd, depth=1)
2272 cmd = self.var_expand(cmd, depth=1)
2273 # protect os.system from UNC paths on Windows, which it can't handle:
2273 # protect os.system from UNC paths on Windows, which it can't handle:
2274 if sys.platform == 'win32':
2274 if sys.platform == 'win32':
2275 from IPython.utils._process_win32 import AvoidUNCPath
2275 from IPython.utils._process_win32 import AvoidUNCPath
2276 with AvoidUNCPath() as path:
2276 with AvoidUNCPath() as path:
2277 if path is not None:
2277 if path is not None:
2278 cmd = '"pushd %s &&"%s' % (path, cmd)
2278 cmd = '"pushd %s &&"%s' % (path, cmd)
2279 try:
2279 try:
2280 ec = os.system(cmd)
2280 ec = os.system(cmd)
2281 except KeyboardInterrupt:
2281 except KeyboardInterrupt:
2282 print('\n' + self.get_exception_only(), file=sys.stderr)
2282 print('\n' + self.get_exception_only(), file=sys.stderr)
2283 ec = -2
2283 ec = -2
2284 else:
2284 else:
2285 # For posix the result of the subprocess.call() below is an exit
2285 # For posix the result of the subprocess.call() below is an exit
2286 # code, which by convention is zero for success, positive for
2286 # code, which by convention is zero for success, positive for
2287 # program failure. Exit codes above 128 are reserved for signals,
2287 # program failure. Exit codes above 128 are reserved for signals,
2288 # and the formula for converting a signal to an exit code is usually
2288 # and the formula for converting a signal to an exit code is usually
2289 # signal_number+128. To more easily differentiate between exit
2289 # signal_number+128. To more easily differentiate between exit
2290 # codes and signals, ipython uses negative numbers. For instance
2290 # codes and signals, ipython uses negative numbers. For instance
2291 # since control-c is signal 2 but exit code 130, ipython's
2291 # since control-c is signal 2 but exit code 130, ipython's
2292 # _exit_code variable will read -2. Note that some shells like
2292 # _exit_code variable will read -2. Note that some shells like
2293 # csh and fish don't follow sh/bash conventions for exit codes.
2293 # csh and fish don't follow sh/bash conventions for exit codes.
2294 executable = os.environ.get('SHELL', None)
2294 executable = os.environ.get('SHELL', None)
2295 try:
2295 try:
2296 # Use env shell instead of default /bin/sh
2296 # Use env shell instead of default /bin/sh
2297 ec = subprocess.call(cmd, shell=True, executable=executable)
2297 ec = subprocess.call(cmd, shell=True, executable=executable)
2298 except KeyboardInterrupt:
2298 except KeyboardInterrupt:
2299 # intercept control-C; a long traceback is not useful here
2299 # intercept control-C; a long traceback is not useful here
2300 print('\n' + self.get_exception_only(), file=sys.stderr)
2300 print('\n' + self.get_exception_only(), file=sys.stderr)
2301 ec = 130
2301 ec = 130
2302 if ec > 128:
2302 if ec > 128:
2303 ec = -(ec - 128)
2303 ec = -(ec - 128)
2304
2304
2305 # We explicitly do NOT return the subprocess status code, because
2305 # We explicitly do NOT return the subprocess status code, because
2306 # a non-None value would trigger :func:`sys.displayhook` calls.
2306 # a non-None value would trigger :func:`sys.displayhook` calls.
2307 # Instead, we store the exit_code in user_ns. Note the semantics
2307 # Instead, we store the exit_code in user_ns. Note the semantics
2308 # of _exit_code: for control-c, _exit_code == -signal.SIGNIT,
2308 # of _exit_code: for control-c, _exit_code == -signal.SIGNIT,
2309 # but raising SystemExit(_exit_code) will give status 254!
2309 # but raising SystemExit(_exit_code) will give status 254!
2310 self.user_ns['_exit_code'] = ec
2310 self.user_ns['_exit_code'] = ec
2311
2311
2312 # use piped system by default, because it is better behaved
2312 # use piped system by default, because it is better behaved
2313 system = system_piped
2313 system = system_piped
2314
2314
2315 def getoutput(self, cmd, split=True, depth=0):
2315 def getoutput(self, cmd, split=True, depth=0):
2316 """Get output (possibly including stderr) from a subprocess.
2316 """Get output (possibly including stderr) from a subprocess.
2317
2317
2318 Parameters
2318 Parameters
2319 ----------
2319 ----------
2320 cmd : str
2320 cmd : str
2321 Command to execute (can not end in '&', as background processes are
2321 Command to execute (can not end in '&', as background processes are
2322 not supported.
2322 not supported.
2323 split : bool, optional
2323 split : bool, optional
2324 If True, split the output into an IPython SList. Otherwise, an
2324 If True, split the output into an IPython SList. Otherwise, an
2325 IPython LSString is returned. These are objects similar to normal
2325 IPython LSString is returned. These are objects similar to normal
2326 lists and strings, with a few convenience attributes for easier
2326 lists and strings, with a few convenience attributes for easier
2327 manipulation of line-based output. You can use '?' on them for
2327 manipulation of line-based output. You can use '?' on them for
2328 details.
2328 details.
2329 depth : int, optional
2329 depth : int, optional
2330 How many frames above the caller are the local variables which should
2330 How many frames above the caller are the local variables which should
2331 be expanded in the command string? The default (0) assumes that the
2331 be expanded in the command string? The default (0) assumes that the
2332 expansion variables are in the stack frame calling this function.
2332 expansion variables are in the stack frame calling this function.
2333 """
2333 """
2334 if cmd.rstrip().endswith('&'):
2334 if cmd.rstrip().endswith('&'):
2335 # this is *far* from a rigorous test
2335 # this is *far* from a rigorous test
2336 raise OSError("Background processes not supported.")
2336 raise OSError("Background processes not supported.")
2337 out = getoutput(self.var_expand(cmd, depth=depth+1))
2337 out = getoutput(self.var_expand(cmd, depth=depth+1))
2338 if split:
2338 if split:
2339 out = SList(out.splitlines())
2339 out = SList(out.splitlines())
2340 else:
2340 else:
2341 out = LSString(out)
2341 out = LSString(out)
2342 return out
2342 return out
2343
2343
2344 #-------------------------------------------------------------------------
2344 #-------------------------------------------------------------------------
2345 # Things related to aliases
2345 # Things related to aliases
2346 #-------------------------------------------------------------------------
2346 #-------------------------------------------------------------------------
2347
2347
2348 def init_alias(self):
2348 def init_alias(self):
2349 self.alias_manager = AliasManager(shell=self, parent=self)
2349 self.alias_manager = AliasManager(shell=self, parent=self)
2350 self.configurables.append(self.alias_manager)
2350 self.configurables.append(self.alias_manager)
2351
2351
2352 #-------------------------------------------------------------------------
2352 #-------------------------------------------------------------------------
2353 # Things related to extensions
2353 # Things related to extensions
2354 #-------------------------------------------------------------------------
2354 #-------------------------------------------------------------------------
2355
2355
2356 def init_extension_manager(self):
2356 def init_extension_manager(self):
2357 self.extension_manager = ExtensionManager(shell=self, parent=self)
2357 self.extension_manager = ExtensionManager(shell=self, parent=self)
2358 self.configurables.append(self.extension_manager)
2358 self.configurables.append(self.extension_manager)
2359
2359
2360 #-------------------------------------------------------------------------
2360 #-------------------------------------------------------------------------
2361 # Things related to payloads
2361 # Things related to payloads
2362 #-------------------------------------------------------------------------
2362 #-------------------------------------------------------------------------
2363
2363
2364 def init_payload(self):
2364 def init_payload(self):
2365 self.payload_manager = PayloadManager(parent=self)
2365 self.payload_manager = PayloadManager(parent=self)
2366 self.configurables.append(self.payload_manager)
2366 self.configurables.append(self.payload_manager)
2367
2367
2368 #-------------------------------------------------------------------------
2368 #-------------------------------------------------------------------------
2369 # Things related to the prefilter
2369 # Things related to the prefilter
2370 #-------------------------------------------------------------------------
2370 #-------------------------------------------------------------------------
2371
2371
2372 def init_prefilter(self):
2372 def init_prefilter(self):
2373 self.prefilter_manager = PrefilterManager(shell=self, parent=self)
2373 self.prefilter_manager = PrefilterManager(shell=self, parent=self)
2374 self.configurables.append(self.prefilter_manager)
2374 self.configurables.append(self.prefilter_manager)
2375 # Ultimately this will be refactored in the new interpreter code, but
2375 # Ultimately this will be refactored in the new interpreter code, but
2376 # for now, we should expose the main prefilter method (there's legacy
2376 # for now, we should expose the main prefilter method (there's legacy
2377 # code out there that may rely on this).
2377 # code out there that may rely on this).
2378 self.prefilter = self.prefilter_manager.prefilter_lines
2378 self.prefilter = self.prefilter_manager.prefilter_lines
2379
2379
2380 def auto_rewrite_input(self, cmd):
2380 def auto_rewrite_input(self, cmd):
2381 """Print to the screen the rewritten form of the user's command.
2381 """Print to the screen the rewritten form of the user's command.
2382
2382
2383 This shows visual feedback by rewriting input lines that cause
2383 This shows visual feedback by rewriting input lines that cause
2384 automatic calling to kick in, like::
2384 automatic calling to kick in, like::
2385
2385
2386 /f x
2386 /f x
2387
2387
2388 into::
2388 into::
2389
2389
2390 ------> f(x)
2390 ------> f(x)
2391
2391
2392 after the user's input prompt. This helps the user understand that the
2392 after the user's input prompt. This helps the user understand that the
2393 input line was transformed automatically by IPython.
2393 input line was transformed automatically by IPython.
2394 """
2394 """
2395 if not self.show_rewritten_input:
2395 if not self.show_rewritten_input:
2396 return
2396 return
2397
2397
2398 # This is overridden in TerminalInteractiveShell to use fancy prompts
2398 # This is overridden in TerminalInteractiveShell to use fancy prompts
2399 print("------> " + cmd)
2399 print("------> " + cmd)
2400
2400
2401 #-------------------------------------------------------------------------
2401 #-------------------------------------------------------------------------
2402 # Things related to extracting values/expressions from kernel and user_ns
2402 # Things related to extracting values/expressions from kernel and user_ns
2403 #-------------------------------------------------------------------------
2403 #-------------------------------------------------------------------------
2404
2404
2405 def _user_obj_error(self):
2405 def _user_obj_error(self):
2406 """return simple exception dict
2406 """return simple exception dict
2407
2407
2408 for use in user_expressions
2408 for use in user_expressions
2409 """
2409 """
2410
2410
2411 etype, evalue, tb = self._get_exc_info()
2411 etype, evalue, tb = self._get_exc_info()
2412 stb = self.InteractiveTB.get_exception_only(etype, evalue)
2412 stb = self.InteractiveTB.get_exception_only(etype, evalue)
2413
2413
2414 exc_info = {
2414 exc_info = {
2415 u'status' : 'error',
2415 u'status' : 'error',
2416 u'traceback' : stb,
2416 u'traceback' : stb,
2417 u'ename' : etype.__name__,
2417 u'ename' : etype.__name__,
2418 u'evalue' : py3compat.safe_unicode(evalue),
2418 u'evalue' : py3compat.safe_unicode(evalue),
2419 }
2419 }
2420
2420
2421 return exc_info
2421 return exc_info
2422
2422
2423 def _format_user_obj(self, obj):
2423 def _format_user_obj(self, obj):
2424 """format a user object to display dict
2424 """format a user object to display dict
2425
2425
2426 for use in user_expressions
2426 for use in user_expressions
2427 """
2427 """
2428
2428
2429 data, md = self.display_formatter.format(obj)
2429 data, md = self.display_formatter.format(obj)
2430 value = {
2430 value = {
2431 'status' : 'ok',
2431 'status' : 'ok',
2432 'data' : data,
2432 'data' : data,
2433 'metadata' : md,
2433 'metadata' : md,
2434 }
2434 }
2435 return value
2435 return value
2436
2436
2437 def user_expressions(self, expressions):
2437 def user_expressions(self, expressions):
2438 """Evaluate a dict of expressions in the user's namespace.
2438 """Evaluate a dict of expressions in the user's namespace.
2439
2439
2440 Parameters
2440 Parameters
2441 ----------
2441 ----------
2442 expressions : dict
2442 expressions : dict
2443 A dict with string keys and string values. The expression values
2443 A dict with string keys and string values. The expression values
2444 should be valid Python expressions, each of which will be evaluated
2444 should be valid Python expressions, each of which will be evaluated
2445 in the user namespace.
2445 in the user namespace.
2446
2446
2447 Returns
2447 Returns
2448 -------
2448 -------
2449 A dict, keyed like the input expressions dict, with the rich mime-typed
2449 A dict, keyed like the input expressions dict, with the rich mime-typed
2450 display_data of each value.
2450 display_data of each value.
2451 """
2451 """
2452 out = {}
2452 out = {}
2453 user_ns = self.user_ns
2453 user_ns = self.user_ns
2454 global_ns = self.user_global_ns
2454 global_ns = self.user_global_ns
2455
2455
2456 for key, expr in expressions.items():
2456 for key, expr in expressions.items():
2457 try:
2457 try:
2458 value = self._format_user_obj(eval(expr, global_ns, user_ns))
2458 value = self._format_user_obj(eval(expr, global_ns, user_ns))
2459 except:
2459 except:
2460 value = self._user_obj_error()
2460 value = self._user_obj_error()
2461 out[key] = value
2461 out[key] = value
2462 return out
2462 return out
2463
2463
2464 #-------------------------------------------------------------------------
2464 #-------------------------------------------------------------------------
2465 # Things related to the running of code
2465 # Things related to the running of code
2466 #-------------------------------------------------------------------------
2466 #-------------------------------------------------------------------------
2467
2467
2468 def ex(self, cmd):
2468 def ex(self, cmd):
2469 """Execute a normal python statement in user namespace."""
2469 """Execute a normal python statement in user namespace."""
2470 with self.builtin_trap:
2470 with self.builtin_trap:
2471 exec(cmd, self.user_global_ns, self.user_ns)
2471 exec(cmd, self.user_global_ns, self.user_ns)
2472
2472
2473 def ev(self, expr):
2473 def ev(self, expr):
2474 """Evaluate python expression expr in user namespace.
2474 """Evaluate python expression expr in user namespace.
2475
2475
2476 Returns the result of evaluation
2476 Returns the result of evaluation
2477 """
2477 """
2478 with self.builtin_trap:
2478 with self.builtin_trap:
2479 return eval(expr, self.user_global_ns, self.user_ns)
2479 return eval(expr, self.user_global_ns, self.user_ns)
2480
2480
2481 def safe_execfile(self, fname, *where, exit_ignore=False, raise_exceptions=False, shell_futures=False):
2481 def safe_execfile(self, fname, *where, exit_ignore=False, raise_exceptions=False, shell_futures=False):
2482 """A safe version of the builtin execfile().
2482 """A safe version of the builtin execfile().
2483
2483
2484 This version will never throw an exception, but instead print
2484 This version will never throw an exception, but instead print
2485 helpful error messages to the screen. This only works on pure
2485 helpful error messages to the screen. This only works on pure
2486 Python files with the .py extension.
2486 Python files with the .py extension.
2487
2487
2488 Parameters
2488 Parameters
2489 ----------
2489 ----------
2490 fname : string
2490 fname : string
2491 The name of the file to be executed.
2491 The name of the file to be executed.
2492 where : tuple
2492 where : tuple
2493 One or two namespaces, passed to execfile() as (globals,locals).
2493 One or two namespaces, passed to execfile() as (globals,locals).
2494 If only one is given, it is passed as both.
2494 If only one is given, it is passed as both.
2495 exit_ignore : bool (False)
2495 exit_ignore : bool (False)
2496 If True, then silence SystemExit for non-zero status (it is always
2496 If True, then silence SystemExit for non-zero status (it is always
2497 silenced for zero status, as it is so common).
2497 silenced for zero status, as it is so common).
2498 raise_exceptions : bool (False)
2498 raise_exceptions : bool (False)
2499 If True raise exceptions everywhere. Meant for testing.
2499 If True raise exceptions everywhere. Meant for testing.
2500 shell_futures : bool (False)
2500 shell_futures : bool (False)
2501 If True, the code will share future statements with the interactive
2501 If True, the code will share future statements with the interactive
2502 shell. It will both be affected by previous __future__ imports, and
2502 shell. It will both be affected by previous __future__ imports, and
2503 any __future__ imports in the code will affect the shell. If False,
2503 any __future__ imports in the code will affect the shell. If False,
2504 __future__ imports are not shared in either direction.
2504 __future__ imports are not shared in either direction.
2505
2505
2506 """
2506 """
2507 fname = os.path.abspath(os.path.expanduser(fname))
2507 fname = os.path.abspath(os.path.expanduser(fname))
2508
2508
2509 # Make sure we can open the file
2509 # Make sure we can open the file
2510 try:
2510 try:
2511 with open(fname):
2511 with open(fname):
2512 pass
2512 pass
2513 except:
2513 except:
2514 warn('Could not open file <%s> for safe execution.' % fname)
2514 warn('Could not open file <%s> for safe execution.' % fname)
2515 return
2515 return
2516
2516
2517 # Find things also in current directory. This is needed to mimic the
2517 # Find things also in current directory. This is needed to mimic the
2518 # behavior of running a script from the system command line, where
2518 # behavior of running a script from the system command line, where
2519 # Python inserts the script's directory into sys.path
2519 # Python inserts the script's directory into sys.path
2520 dname = os.path.dirname(fname)
2520 dname = os.path.dirname(fname)
2521
2521
2522 with prepended_to_syspath(dname), self.builtin_trap:
2522 with prepended_to_syspath(dname), self.builtin_trap:
2523 try:
2523 try:
2524 glob, loc = (where + (None, ))[:2]
2524 glob, loc = (where + (None, ))[:2]
2525 py3compat.execfile(
2525 py3compat.execfile(
2526 fname, glob, loc,
2526 fname, glob, loc,
2527 self.compile if shell_futures else None)
2527 self.compile if shell_futures else None)
2528 except SystemExit as status:
2528 except SystemExit as status:
2529 # If the call was made with 0 or None exit status (sys.exit(0)
2529 # If the call was made with 0 or None exit status (sys.exit(0)
2530 # or sys.exit() ), don't bother showing a traceback, as both of
2530 # or sys.exit() ), don't bother showing a traceback, as both of
2531 # these are considered normal by the OS:
2531 # these are considered normal by the OS:
2532 # > python -c'import sys;sys.exit(0)'; echo $?
2532 # > python -c'import sys;sys.exit(0)'; echo $?
2533 # 0
2533 # 0
2534 # > python -c'import sys;sys.exit()'; echo $?
2534 # > python -c'import sys;sys.exit()'; echo $?
2535 # 0
2535 # 0
2536 # For other exit status, we show the exception unless
2536 # For other exit status, we show the exception unless
2537 # explicitly silenced, but only in short form.
2537 # explicitly silenced, but only in short form.
2538 if status.code:
2538 if status.code:
2539 if raise_exceptions:
2539 if raise_exceptions:
2540 raise
2540 raise
2541 if not exit_ignore:
2541 if not exit_ignore:
2542 self.showtraceback(exception_only=True)
2542 self.showtraceback(exception_only=True)
2543 except:
2543 except:
2544 if raise_exceptions:
2544 if raise_exceptions:
2545 raise
2545 raise
2546 # tb offset is 2 because we wrap execfile
2546 # tb offset is 2 because we wrap execfile
2547 self.showtraceback(tb_offset=2)
2547 self.showtraceback(tb_offset=2)
2548
2548
2549 def safe_execfile_ipy(self, fname, shell_futures=False, raise_exceptions=False):
2549 def safe_execfile_ipy(self, fname, shell_futures=False, raise_exceptions=False):
2550 """Like safe_execfile, but for .ipy or .ipynb files with IPython syntax.
2550 """Like safe_execfile, but for .ipy or .ipynb files with IPython syntax.
2551
2551
2552 Parameters
2552 Parameters
2553 ----------
2553 ----------
2554 fname : str
2554 fname : str
2555 The name of the file to execute. The filename must have a
2555 The name of the file to execute. The filename must have a
2556 .ipy or .ipynb extension.
2556 .ipy or .ipynb extension.
2557 shell_futures : bool (False)
2557 shell_futures : bool (False)
2558 If True, the code will share future statements with the interactive
2558 If True, the code will share future statements with the interactive
2559 shell. It will both be affected by previous __future__ imports, and
2559 shell. It will both be affected by previous __future__ imports, and
2560 any __future__ imports in the code will affect the shell. If False,
2560 any __future__ imports in the code will affect the shell. If False,
2561 __future__ imports are not shared in either direction.
2561 __future__ imports are not shared in either direction.
2562 raise_exceptions : bool (False)
2562 raise_exceptions : bool (False)
2563 If True raise exceptions everywhere. Meant for testing.
2563 If True raise exceptions everywhere. Meant for testing.
2564 """
2564 """
2565 fname = os.path.abspath(os.path.expanduser(fname))
2565 fname = os.path.abspath(os.path.expanduser(fname))
2566
2566
2567 # Make sure we can open the file
2567 # Make sure we can open the file
2568 try:
2568 try:
2569 with open(fname):
2569 with open(fname):
2570 pass
2570 pass
2571 except:
2571 except:
2572 warn('Could not open file <%s> for safe execution.' % fname)
2572 warn('Could not open file <%s> for safe execution.' % fname)
2573 return
2573 return
2574
2574
2575 # Find things also in current directory. This is needed to mimic the
2575 # Find things also in current directory. This is needed to mimic the
2576 # behavior of running a script from the system command line, where
2576 # behavior of running a script from the system command line, where
2577 # Python inserts the script's directory into sys.path
2577 # Python inserts the script's directory into sys.path
2578 dname = os.path.dirname(fname)
2578 dname = os.path.dirname(fname)
2579
2579
2580 def get_cells():
2580 def get_cells():
2581 """generator for sequence of code blocks to run"""
2581 """generator for sequence of code blocks to run"""
2582 if fname.endswith('.ipynb'):
2582 if fname.endswith('.ipynb'):
2583 from nbformat import read
2583 from nbformat import read
2584 nb = read(fname, as_version=4)
2584 nb = read(fname, as_version=4)
2585 if not nb.cells:
2585 if not nb.cells:
2586 return
2586 return
2587 for cell in nb.cells:
2587 for cell in nb.cells:
2588 if cell.cell_type == 'code':
2588 if cell.cell_type == 'code':
2589 yield cell.source
2589 yield cell.source
2590 else:
2590 else:
2591 with open(fname) as f:
2591 with open(fname) as f:
2592 yield f.read()
2592 yield f.read()
2593
2593
2594 with prepended_to_syspath(dname):
2594 with prepended_to_syspath(dname):
2595 try:
2595 try:
2596 for cell in get_cells():
2596 for cell in get_cells():
2597 result = self.run_cell(cell, silent=True, shell_futures=shell_futures)
2597 result = self.run_cell(cell, silent=True, shell_futures=shell_futures)
2598 if raise_exceptions:
2598 if raise_exceptions:
2599 result.raise_error()
2599 result.raise_error()
2600 elif not result.success:
2600 elif not result.success:
2601 break
2601 break
2602 except:
2602 except:
2603 if raise_exceptions:
2603 if raise_exceptions:
2604 raise
2604 raise
2605 self.showtraceback()
2605 self.showtraceback()
2606 warn('Unknown failure executing file: <%s>' % fname)
2606 warn('Unknown failure executing file: <%s>' % fname)
2607
2607
2608 def safe_run_module(self, mod_name, where):
2608 def safe_run_module(self, mod_name, where):
2609 """A safe version of runpy.run_module().
2609 """A safe version of runpy.run_module().
2610
2610
2611 This version will never throw an exception, but instead print
2611 This version will never throw an exception, but instead print
2612 helpful error messages to the screen.
2612 helpful error messages to the screen.
2613
2613
2614 `SystemExit` exceptions with status code 0 or None are ignored.
2614 `SystemExit` exceptions with status code 0 or None are ignored.
2615
2615
2616 Parameters
2616 Parameters
2617 ----------
2617 ----------
2618 mod_name : string
2618 mod_name : string
2619 The name of the module to be executed.
2619 The name of the module to be executed.
2620 where : dict
2620 where : dict
2621 The globals namespace.
2621 The globals namespace.
2622 """
2622 """
2623 try:
2623 try:
2624 try:
2624 try:
2625 where.update(
2625 where.update(
2626 runpy.run_module(str(mod_name), run_name="__main__",
2626 runpy.run_module(str(mod_name), run_name="__main__",
2627 alter_sys=True)
2627 alter_sys=True)
2628 )
2628 )
2629 except SystemExit as status:
2629 except SystemExit as status:
2630 if status.code:
2630 if status.code:
2631 raise
2631 raise
2632 except:
2632 except:
2633 self.showtraceback()
2633 self.showtraceback()
2634 warn('Unknown failure executing module: <%s>' % mod_name)
2634 warn('Unknown failure executing module: <%s>' % mod_name)
2635
2635
2636 def run_cell(self, raw_cell, store_history=False, silent=False, shell_futures=True):
2636 def run_cell(self, raw_cell, store_history=False, silent=False, shell_futures=True):
2637 """Run a complete IPython cell.
2637 """Run a complete IPython cell.
2638
2638
2639 Parameters
2639 Parameters
2640 ----------
2640 ----------
2641 raw_cell : str
2641 raw_cell : str
2642 The code (including IPython code such as %magic functions) to run.
2642 The code (including IPython code such as %magic functions) to run.
2643 store_history : bool
2643 store_history : bool
2644 If True, the raw and translated cell will be stored in IPython's
2644 If True, the raw and translated cell will be stored in IPython's
2645 history. For user code calling back into IPython's machinery, this
2645 history. For user code calling back into IPython's machinery, this
2646 should be set to False.
2646 should be set to False.
2647 silent : bool
2647 silent : bool
2648 If True, avoid side-effects, such as implicit displayhooks and
2648 If True, avoid side-effects, such as implicit displayhooks and
2649 and logging. silent=True forces store_history=False.
2649 and logging. silent=True forces store_history=False.
2650 shell_futures : bool
2650 shell_futures : bool
2651 If True, the code will share future statements with the interactive
2651 If True, the code will share future statements with the interactive
2652 shell. It will both be affected by previous __future__ imports, and
2652 shell. It will both be affected by previous __future__ imports, and
2653 any __future__ imports in the code will affect the shell. If False,
2653 any __future__ imports in the code will affect the shell. If False,
2654 __future__ imports are not shared in either direction.
2654 __future__ imports are not shared in either direction.
2655
2655
2656 Returns
2656 Returns
2657 -------
2657 -------
2658 result : :class:`ExecutionResult`
2658 result : :class:`ExecutionResult`
2659 """
2659 """
2660 try:
2660 try:
2661 result = self._run_cell(
2661 result = self._run_cell(
2662 raw_cell, store_history, silent, shell_futures)
2662 raw_cell, store_history, silent, shell_futures)
2663 finally:
2663 finally:
2664 self.events.trigger('post_execute')
2664 self.events.trigger('post_execute')
2665 if not silent:
2665 if not silent:
2666 self.events.trigger('post_run_cell', result)
2666 self.events.trigger('post_run_cell', result)
2667 return result
2667 return result
2668
2668
2669 def _run_cell(self, raw_cell, store_history, silent, shell_futures):
2669 def _run_cell(self, raw_cell, store_history, silent, shell_futures):
2670 """Internal method to run a complete IPython cell.
2670 """Internal method to run a complete IPython cell.
2671
2671
2672 Parameters
2672 Parameters
2673 ----------
2673 ----------
2674 raw_cell : str
2674 raw_cell : str
2675 store_history : bool
2675 store_history : bool
2676 silent : bool
2676 silent : bool
2677 shell_futures : bool
2677 shell_futures : bool
2678
2678
2679 Returns
2679 Returns
2680 -------
2680 -------
2681 result : :class:`ExecutionResult`
2681 result : :class:`ExecutionResult`
2682 """
2682 """
2683 info = ExecutionInfo(
2683 info = ExecutionInfo(
2684 raw_cell, store_history, silent, shell_futures)
2684 raw_cell, store_history, silent, shell_futures)
2685 result = ExecutionResult(info)
2685 result = ExecutionResult(info)
2686
2686
2687 if (not raw_cell) or raw_cell.isspace():
2687 if (not raw_cell) or raw_cell.isspace():
2688 self.last_execution_succeeded = True
2688 self.last_execution_succeeded = True
2689 self.last_execution_result = result
2689 self.last_execution_result = result
2690 return result
2690 return result
2691
2691
2692 if silent:
2692 if silent:
2693 store_history = False
2693 store_history = False
2694
2694
2695 if store_history:
2695 if store_history:
2696 result.execution_count = self.execution_count
2696 result.execution_count = self.execution_count
2697
2697
2698 def error_before_exec(value):
2698 def error_before_exec(value):
2699 result.error_before_exec = value
2699 result.error_before_exec = value
2700 self.last_execution_succeeded = False
2700 self.last_execution_succeeded = False
2701 self.last_execution_result = result
2701 self.last_execution_result = result
2702 return result
2702 return result
2703
2703
2704 self.events.trigger('pre_execute')
2704 self.events.trigger('pre_execute')
2705 if not silent:
2705 if not silent:
2706 self.events.trigger('pre_run_cell', info)
2706 self.events.trigger('pre_run_cell', info)
2707
2707
2708 # If any of our input transformation (input_transformer_manager or
2708 # If any of our input transformation (input_transformer_manager or
2709 # prefilter_manager) raises an exception, we store it in this variable
2709 # prefilter_manager) raises an exception, we store it in this variable
2710 # so that we can display the error after logging the input and storing
2710 # so that we can display the error after logging the input and storing
2711 # it in the history.
2711 # it in the history.
2712 preprocessing_exc_tuple = None
2712 preprocessing_exc_tuple = None
2713 try:
2713 try:
2714 # Static input transformations
2714 # Static input transformations
2715 cell = self.input_transformer_manager.transform_cell(raw_cell)
2715 cell = self.input_transformer_manager.transform_cell(raw_cell)
2716 except SyntaxError:
2716 except SyntaxError:
2717 preprocessing_exc_tuple = sys.exc_info()
2717 preprocessing_exc_tuple = sys.exc_info()
2718 cell = raw_cell # cell has to exist so it can be stored/logged
2718 cell = raw_cell # cell has to exist so it can be stored/logged
2719 else:
2719 else:
2720 if len(cell.splitlines()) == 1:
2720 if len(cell.splitlines()) == 1:
2721 # Dynamic transformations - only applied for single line commands
2721 # Dynamic transformations - only applied for single line commands
2722 with self.builtin_trap:
2722 with self.builtin_trap:
2723 try:
2723 try:
2724 # use prefilter_lines to handle trailing newlines
2724 # use prefilter_lines to handle trailing newlines
2725 # restore trailing newline for ast.parse
2725 # restore trailing newline for ast.parse
2726 cell = self.prefilter_manager.prefilter_lines(cell) + '\n'
2726 cell = self.prefilter_manager.prefilter_lines(cell) + '\n'
2727 except Exception:
2727 except Exception:
2728 # don't allow prefilter errors to crash IPython
2728 # don't allow prefilter errors to crash IPython
2729 preprocessing_exc_tuple = sys.exc_info()
2729 preprocessing_exc_tuple = sys.exc_info()
2730
2730
2731 # Store raw and processed history
2731 # Store raw and processed history
2732 if store_history:
2732 if store_history:
2733 self.history_manager.store_inputs(self.execution_count,
2733 self.history_manager.store_inputs(self.execution_count,
2734 cell, raw_cell)
2734 cell, raw_cell)
2735 if not silent:
2735 if not silent:
2736 self.logger.log(cell, raw_cell)
2736 self.logger.log(cell, raw_cell)
2737
2737
2738 # Display the exception if input processing failed.
2738 # Display the exception if input processing failed.
2739 if preprocessing_exc_tuple is not None:
2739 if preprocessing_exc_tuple is not None:
2740 self.showtraceback(preprocessing_exc_tuple)
2740 self.showtraceback(preprocessing_exc_tuple)
2741 if store_history:
2741 if store_history:
2742 self.execution_count += 1
2742 self.execution_count += 1
2743 return error_before_exec(preprocessing_exc_tuple[2])
2743 return error_before_exec(preprocessing_exc_tuple[2])
2744
2744
2745 # Our own compiler remembers the __future__ environment. If we want to
2745 # Our own compiler remembers the __future__ environment. If we want to
2746 # run code with a separate __future__ environment, use the default
2746 # run code with a separate __future__ environment, use the default
2747 # compiler
2747 # compiler
2748 compiler = self.compile if shell_futures else CachingCompiler()
2748 compiler = self.compile if shell_futures else CachingCompiler()
2749
2749
2750 with self.builtin_trap:
2750 with self.builtin_trap:
2751 cell_name = self.compile.cache(cell, self.execution_count)
2751 cell_name = self.compile.cache(cell, self.execution_count)
2752
2752
2753 with self.display_trap:
2753 with self.display_trap:
2754 # Compile to bytecode
2754 # Compile to bytecode
2755 ast_error = None
2755 try:
2756 try:
2756 code_ast = compiler.ast_parse(cell, filename=cell_name)
2757 code_ast = compiler.ast_parse(cell, filename=cell_name)
2757 except self.custom_exceptions as e:
2758 except self.custom_exceptions as e:
2758 etype, value, tb = sys.exc_info()
2759 etype, value, tb = sys.exc_info()
2759 self.CustomTB(etype, value, tb)
2760 self.CustomTB(etype, value, tb)
2760 return error_before_exec(e)
2761 ast_error = e
2761 except IndentationError as e:
2762 except IndentationError as e:
2762 self.showindentationerror()
2763 self.showindentationerror()
2763 if store_history:
2764 ast_error = e
2764 self.execution_count += 1
2765 return error_before_exec(e)
2766 except (OverflowError, SyntaxError, ValueError, TypeError,
2765 except (OverflowError, SyntaxError, ValueError, TypeError,
2767 MemoryError) as e:
2766 MemoryError) as e:
2768 self.showsyntaxerror()
2767 self.showsyntaxerror()
2768 ast_error = e
2769
2770 if ast_error is not None:
2769 if store_history:
2771 if store_history:
2770 self.execution_count += 1
2772 self.execution_count += 1
2771 return error_before_exec(e)
2773 return error_before_exec(ast_error)
2772
2774
2773 # Apply AST transformations
2775 # Apply AST transformations
2774 try:
2776 try:
2775 code_ast = self.transform_ast(code_ast)
2777 code_ast = self.transform_ast(code_ast)
2776 except InputRejected as e:
2778 except InputRejected as e:
2777 self.showtraceback()
2779 self.showtraceback()
2778 if store_history:
2780 if store_history:
2779 self.execution_count += 1
2781 self.execution_count += 1
2780 return error_before_exec(e)
2782 return error_before_exec(e)
2781
2783
2782 # Give the displayhook a reference to our ExecutionResult so it
2784 # Give the displayhook a reference to our ExecutionResult so it
2783 # can fill in the output value.
2785 # can fill in the output value.
2784 self.displayhook.exec_result = result
2786 self.displayhook.exec_result = result
2785
2787
2786 # Execute the user code
2788 # Execute the user code
2787 interactivity = 'none' if silent else self.ast_node_interactivity
2789 interactivity = 'none' if silent else self.ast_node_interactivity
2788 has_raised = self.run_ast_nodes(code_ast.body, cell_name,
2790 has_raised = self.run_ast_nodes(code_ast.body, cell_name,
2789 interactivity=interactivity, compiler=compiler, result=result)
2791 interactivity=interactivity, compiler=compiler, result=result)
2790
2792
2791 self.last_execution_succeeded = not has_raised
2793 self.last_execution_succeeded = not has_raised
2792 self.last_execution_result = result
2794 self.last_execution_result = result
2793
2795
2794 # Reset this so later displayed values do not modify the
2796 # Reset this so later displayed values do not modify the
2795 # ExecutionResult
2797 # ExecutionResult
2796 self.displayhook.exec_result = None
2798 self.displayhook.exec_result = None
2797
2799
2798 if store_history:
2800 if store_history:
2799 # Write output to the database. Does nothing unless
2801 # Write output to the database. Does nothing unless
2800 # history output logging is enabled.
2802 # history output logging is enabled.
2801 self.history_manager.store_output(self.execution_count)
2803 self.history_manager.store_output(self.execution_count)
2802 # Each cell is a *single* input, regardless of how many lines it has
2804 # Each cell is a *single* input, regardless of how many lines it has
2803 self.execution_count += 1
2805 self.execution_count += 1
2804
2806
2805 return result
2807 return result
2806
2808
2807 def transform_ast(self, node):
2809 def transform_ast(self, node):
2808 """Apply the AST transformations from self.ast_transformers
2810 """Apply the AST transformations from self.ast_transformers
2809
2811
2810 Parameters
2812 Parameters
2811 ----------
2813 ----------
2812 node : ast.Node
2814 node : ast.Node
2813 The root node to be transformed. Typically called with the ast.Module
2815 The root node to be transformed. Typically called with the ast.Module
2814 produced by parsing user input.
2816 produced by parsing user input.
2815
2817
2816 Returns
2818 Returns
2817 -------
2819 -------
2818 An ast.Node corresponding to the node it was called with. Note that it
2820 An ast.Node corresponding to the node it was called with. Note that it
2819 may also modify the passed object, so don't rely on references to the
2821 may also modify the passed object, so don't rely on references to the
2820 original AST.
2822 original AST.
2821 """
2823 """
2822 for transformer in self.ast_transformers:
2824 for transformer in self.ast_transformers:
2823 try:
2825 try:
2824 node = transformer.visit(node)
2826 node = transformer.visit(node)
2825 except InputRejected:
2827 except InputRejected:
2826 # User-supplied AST transformers can reject an input by raising
2828 # User-supplied AST transformers can reject an input by raising
2827 # an InputRejected. Short-circuit in this case so that we
2829 # an InputRejected. Short-circuit in this case so that we
2828 # don't unregister the transform.
2830 # don't unregister the transform.
2829 raise
2831 raise
2830 except Exception:
2832 except Exception:
2831 warn("AST transformer %r threw an error. It will be unregistered." % transformer)
2833 warn("AST transformer %r threw an error. It will be unregistered." % transformer)
2832 self.ast_transformers.remove(transformer)
2834 self.ast_transformers.remove(transformer)
2833
2835
2834 if self.ast_transformers:
2836 if self.ast_transformers:
2835 ast.fix_missing_locations(node)
2837 ast.fix_missing_locations(node)
2836 return node
2838 return node
2837
2839
2838
2840
2839 def run_ast_nodes(self, nodelist:ListType[AST], cell_name:str, interactivity='last_expr',
2841 def run_ast_nodes(self, nodelist:ListType[AST], cell_name:str, interactivity='last_expr',
2840 compiler=compile, result=None):
2842 compiler=compile, result=None):
2841 """Run a sequence of AST nodes. The execution mode depends on the
2843 """Run a sequence of AST nodes. The execution mode depends on the
2842 interactivity parameter.
2844 interactivity parameter.
2843
2845
2844 Parameters
2846 Parameters
2845 ----------
2847 ----------
2846 nodelist : list
2848 nodelist : list
2847 A sequence of AST nodes to run.
2849 A sequence of AST nodes to run.
2848 cell_name : str
2850 cell_name : str
2849 Will be passed to the compiler as the filename of the cell. Typically
2851 Will be passed to the compiler as the filename of the cell. Typically
2850 the value returned by ip.compile.cache(cell).
2852 the value returned by ip.compile.cache(cell).
2851 interactivity : str
2853 interactivity : str
2852 'all', 'last', 'last_expr' , 'last_expr_or_assign' or 'none',
2854 'all', 'last', 'last_expr' , 'last_expr_or_assign' or 'none',
2853 specifying which nodes should be run interactively (displaying output
2855 specifying which nodes should be run interactively (displaying output
2854 from expressions). 'last_expr' will run the last node interactively
2856 from expressions). 'last_expr' will run the last node interactively
2855 only if it is an expression (i.e. expressions in loops or other blocks
2857 only if it is an expression (i.e. expressions in loops or other blocks
2856 are not displayed) 'last_expr_or_assign' will run the last expression
2858 are not displayed) 'last_expr_or_assign' will run the last expression
2857 or the last assignment. Other values for this parameter will raise a
2859 or the last assignment. Other values for this parameter will raise a
2858 ValueError.
2860 ValueError.
2859 compiler : callable
2861 compiler : callable
2860 A function with the same interface as the built-in compile(), to turn
2862 A function with the same interface as the built-in compile(), to turn
2861 the AST nodes into code objects. Default is the built-in compile().
2863 the AST nodes into code objects. Default is the built-in compile().
2862 result : ExecutionResult, optional
2864 result : ExecutionResult, optional
2863 An object to store exceptions that occur during execution.
2865 An object to store exceptions that occur during execution.
2864
2866
2865 Returns
2867 Returns
2866 -------
2868 -------
2867 True if an exception occurred while running code, False if it finished
2869 True if an exception occurred while running code, False if it finished
2868 running.
2870 running.
2869 """
2871 """
2870 if not nodelist:
2872 if not nodelist:
2871 return
2873 return
2872
2874
2873 if interactivity == 'last_expr_or_assign':
2875 if interactivity == 'last_expr_or_assign':
2874 if isinstance(nodelist[-1], _assign_nodes):
2876 if isinstance(nodelist[-1], _assign_nodes):
2875 asg = nodelist[-1]
2877 asg = nodelist[-1]
2876 if isinstance(asg, ast.Assign) and len(asg.targets) == 1:
2878 if isinstance(asg, ast.Assign) and len(asg.targets) == 1:
2877 target = asg.targets[0]
2879 target = asg.targets[0]
2878 elif isinstance(asg, _single_targets_nodes):
2880 elif isinstance(asg, _single_targets_nodes):
2879 target = asg.target
2881 target = asg.target
2880 else:
2882 else:
2881 target = None
2883 target = None
2882 if isinstance(target, ast.Name):
2884 if isinstance(target, ast.Name):
2883 nnode = ast.Expr(ast.Name(target.id, ast.Load()))
2885 nnode = ast.Expr(ast.Name(target.id, ast.Load()))
2884 ast.fix_missing_locations(nnode)
2886 ast.fix_missing_locations(nnode)
2885 nodelist.append(nnode)
2887 nodelist.append(nnode)
2886 interactivity = 'last_expr'
2888 interactivity = 'last_expr'
2887
2889
2888 if interactivity == 'last_expr':
2890 if interactivity == 'last_expr':
2889 if isinstance(nodelist[-1], ast.Expr):
2891 if isinstance(nodelist[-1], ast.Expr):
2890 interactivity = "last"
2892 interactivity = "last"
2891 else:
2893 else:
2892 interactivity = "none"
2894 interactivity = "none"
2893
2895
2894 if interactivity == 'none':
2896 if interactivity == 'none':
2895 to_run_exec, to_run_interactive = nodelist, []
2897 to_run_exec, to_run_interactive = nodelist, []
2896 elif interactivity == 'last':
2898 elif interactivity == 'last':
2897 to_run_exec, to_run_interactive = nodelist[:-1], nodelist[-1:]
2899 to_run_exec, to_run_interactive = nodelist[:-1], nodelist[-1:]
2898 elif interactivity == 'all':
2900 elif interactivity == 'all':
2899 to_run_exec, to_run_interactive = [], nodelist
2901 to_run_exec, to_run_interactive = [], nodelist
2900 else:
2902 else:
2901 raise ValueError("Interactivity was %r" % interactivity)
2903 raise ValueError("Interactivity was %r" % interactivity)
2902
2904
2903 try:
2905 try:
2904 for i, node in enumerate(to_run_exec):
2906 for i, node in enumerate(to_run_exec):
2905 mod = ast.Module([node])
2907 mod = ast.Module([node])
2906 code = compiler(mod, cell_name, "exec")
2908 code = compiler(mod, cell_name, "exec")
2907 if self.run_code(code, result):
2909 if self.run_code(code, result):
2908 return True
2910 return True
2909
2911
2910 for i, node in enumerate(to_run_interactive):
2912 for i, node in enumerate(to_run_interactive):
2911 mod = ast.Interactive([node])
2913 mod = ast.Interactive([node])
2912 code = compiler(mod, cell_name, "single")
2914 code = compiler(mod, cell_name, "single")
2913 if self.run_code(code, result):
2915 if self.run_code(code, result):
2914 return True
2916 return True
2915
2917
2916 # Flush softspace
2918 # Flush softspace
2917 if softspace(sys.stdout, 0):
2919 if softspace(sys.stdout, 0):
2918 print()
2920 print()
2919
2921
2920 except:
2922 except:
2921 # It's possible to have exceptions raised here, typically by
2923 # It's possible to have exceptions raised here, typically by
2922 # compilation of odd code (such as a naked 'return' outside a
2924 # compilation of odd code (such as a naked 'return' outside a
2923 # function) that did parse but isn't valid. Typically the exception
2925 # function) that did parse but isn't valid. Typically the exception
2924 # is a SyntaxError, but it's safest just to catch anything and show
2926 # is a SyntaxError, but it's safest just to catch anything and show
2925 # the user a traceback.
2927 # the user a traceback.
2926
2928
2927 # We do only one try/except outside the loop to minimize the impact
2929 # We do only one try/except outside the loop to minimize the impact
2928 # on runtime, and also because if any node in the node list is
2930 # on runtime, and also because if any node in the node list is
2929 # broken, we should stop execution completely.
2931 # broken, we should stop execution completely.
2930 if result:
2932 if result:
2931 result.error_before_exec = sys.exc_info()[1]
2933 result.error_before_exec = sys.exc_info()[1]
2932 self.showtraceback()
2934 self.showtraceback()
2933 return True
2935 return True
2934
2936
2935 return False
2937 return False
2936
2938
2937 def run_code(self, code_obj, result=None):
2939 def run_code(self, code_obj, result=None):
2938 """Execute a code object.
2940 """Execute a code object.
2939
2941
2940 When an exception occurs, self.showtraceback() is called to display a
2942 When an exception occurs, self.showtraceback() is called to display a
2941 traceback.
2943 traceback.
2942
2944
2943 Parameters
2945 Parameters
2944 ----------
2946 ----------
2945 code_obj : code object
2947 code_obj : code object
2946 A compiled code object, to be executed
2948 A compiled code object, to be executed
2947 result : ExecutionResult, optional
2949 result : ExecutionResult, optional
2948 An object to store exceptions that occur during execution.
2950 An object to store exceptions that occur during execution.
2949
2951
2950 Returns
2952 Returns
2951 -------
2953 -------
2952 False : successful execution.
2954 False : successful execution.
2953 True : an error occurred.
2955 True : an error occurred.
2954 """
2956 """
2955 # Set our own excepthook in case the user code tries to call it
2957 # Set our own excepthook in case the user code tries to call it
2956 # directly, so that the IPython crash handler doesn't get triggered
2958 # directly, so that the IPython crash handler doesn't get triggered
2957 old_excepthook, sys.excepthook = sys.excepthook, self.excepthook
2959 old_excepthook, sys.excepthook = sys.excepthook, self.excepthook
2958
2960
2959 # we save the original sys.excepthook in the instance, in case config
2961 # we save the original sys.excepthook in the instance, in case config
2960 # code (such as magics) needs access to it.
2962 # code (such as magics) needs access to it.
2961 self.sys_excepthook = old_excepthook
2963 self.sys_excepthook = old_excepthook
2962 outflag = True # happens in more places, so it's easier as default
2964 outflag = True # happens in more places, so it's easier as default
2963 try:
2965 try:
2964 try:
2966 try:
2965 self.hooks.pre_run_code_hook()
2967 self.hooks.pre_run_code_hook()
2966 #rprint('Running code', repr(code_obj)) # dbg
2968 #rprint('Running code', repr(code_obj)) # dbg
2967 exec(code_obj, self.user_global_ns, self.user_ns)
2969 exec(code_obj, self.user_global_ns, self.user_ns)
2968 finally:
2970 finally:
2969 # Reset our crash handler in place
2971 # Reset our crash handler in place
2970 sys.excepthook = old_excepthook
2972 sys.excepthook = old_excepthook
2971 except SystemExit as e:
2973 except SystemExit as e:
2972 if result is not None:
2974 if result is not None:
2973 result.error_in_exec = e
2975 result.error_in_exec = e
2974 self.showtraceback(exception_only=True)
2976 self.showtraceback(exception_only=True)
2975 warn("To exit: use 'exit', 'quit', or Ctrl-D.", stacklevel=1)
2977 warn("To exit: use 'exit', 'quit', or Ctrl-D.", stacklevel=1)
2976 except self.custom_exceptions:
2978 except self.custom_exceptions:
2977 etype, value, tb = sys.exc_info()
2979 etype, value, tb = sys.exc_info()
2978 if result is not None:
2980 if result is not None:
2979 result.error_in_exec = value
2981 result.error_in_exec = value
2980 self.CustomTB(etype, value, tb)
2982 self.CustomTB(etype, value, tb)
2981 except:
2983 except:
2982 if result is not None:
2984 if result is not None:
2983 result.error_in_exec = sys.exc_info()[1]
2985 result.error_in_exec = sys.exc_info()[1]
2984 self.showtraceback(running_compiled_code=True)
2986 self.showtraceback(running_compiled_code=True)
2985 else:
2987 else:
2986 outflag = False
2988 outflag = False
2987 return outflag
2989 return outflag
2988
2990
2989 # For backwards compatibility
2991 # For backwards compatibility
2990 runcode = run_code
2992 runcode = run_code
2991
2993
2992 #-------------------------------------------------------------------------
2994 #-------------------------------------------------------------------------
2993 # Things related to GUI support and pylab
2995 # Things related to GUI support and pylab
2994 #-------------------------------------------------------------------------
2996 #-------------------------------------------------------------------------
2995
2997
2996 active_eventloop = None
2998 active_eventloop = None
2997
2999
2998 def enable_gui(self, gui=None):
3000 def enable_gui(self, gui=None):
2999 raise NotImplementedError('Implement enable_gui in a subclass')
3001 raise NotImplementedError('Implement enable_gui in a subclass')
3000
3002
3001 def enable_matplotlib(self, gui=None):
3003 def enable_matplotlib(self, gui=None):
3002 """Enable interactive matplotlib and inline figure support.
3004 """Enable interactive matplotlib and inline figure support.
3003
3005
3004 This takes the following steps:
3006 This takes the following steps:
3005
3007
3006 1. select the appropriate eventloop and matplotlib backend
3008 1. select the appropriate eventloop and matplotlib backend
3007 2. set up matplotlib for interactive use with that backend
3009 2. set up matplotlib for interactive use with that backend
3008 3. configure formatters for inline figure display
3010 3. configure formatters for inline figure display
3009 4. enable the selected gui eventloop
3011 4. enable the selected gui eventloop
3010
3012
3011 Parameters
3013 Parameters
3012 ----------
3014 ----------
3013 gui : optional, string
3015 gui : optional, string
3014 If given, dictates the choice of matplotlib GUI backend to use
3016 If given, dictates the choice of matplotlib GUI backend to use
3015 (should be one of IPython's supported backends, 'qt', 'osx', 'tk',
3017 (should be one of IPython's supported backends, 'qt', 'osx', 'tk',
3016 'gtk', 'wx' or 'inline'), otherwise we use the default chosen by
3018 'gtk', 'wx' or 'inline'), otherwise we use the default chosen by
3017 matplotlib (as dictated by the matplotlib build-time options plus the
3019 matplotlib (as dictated by the matplotlib build-time options plus the
3018 user's matplotlibrc configuration file). Note that not all backends
3020 user's matplotlibrc configuration file). Note that not all backends
3019 make sense in all contexts, for example a terminal ipython can't
3021 make sense in all contexts, for example a terminal ipython can't
3020 display figures inline.
3022 display figures inline.
3021 """
3023 """
3022 from IPython.core import pylabtools as pt
3024 from IPython.core import pylabtools as pt
3023 gui, backend = pt.find_gui_and_backend(gui, self.pylab_gui_select)
3025 gui, backend = pt.find_gui_and_backend(gui, self.pylab_gui_select)
3024
3026
3025 if gui != 'inline':
3027 if gui != 'inline':
3026 # If we have our first gui selection, store it
3028 # If we have our first gui selection, store it
3027 if self.pylab_gui_select is None:
3029 if self.pylab_gui_select is None:
3028 self.pylab_gui_select = gui
3030 self.pylab_gui_select = gui
3029 # Otherwise if they are different
3031 # Otherwise if they are different
3030 elif gui != self.pylab_gui_select:
3032 elif gui != self.pylab_gui_select:
3031 print('Warning: Cannot change to a different GUI toolkit: %s.'
3033 print('Warning: Cannot change to a different GUI toolkit: %s.'
3032 ' Using %s instead.' % (gui, self.pylab_gui_select))
3034 ' Using %s instead.' % (gui, self.pylab_gui_select))
3033 gui, backend = pt.find_gui_and_backend(self.pylab_gui_select)
3035 gui, backend = pt.find_gui_and_backend(self.pylab_gui_select)
3034
3036
3035 pt.activate_matplotlib(backend)
3037 pt.activate_matplotlib(backend)
3036 pt.configure_inline_support(self, backend)
3038 pt.configure_inline_support(self, backend)
3037
3039
3038 # Now we must activate the gui pylab wants to use, and fix %run to take
3040 # Now we must activate the gui pylab wants to use, and fix %run to take
3039 # plot updates into account
3041 # plot updates into account
3040 self.enable_gui(gui)
3042 self.enable_gui(gui)
3041 self.magics_manager.registry['ExecutionMagics'].default_runner = \
3043 self.magics_manager.registry['ExecutionMagics'].default_runner = \
3042 pt.mpl_runner(self.safe_execfile)
3044 pt.mpl_runner(self.safe_execfile)
3043
3045
3044 return gui, backend
3046 return gui, backend
3045
3047
3046 def enable_pylab(self, gui=None, import_all=True, welcome_message=False):
3048 def enable_pylab(self, gui=None, import_all=True, welcome_message=False):
3047 """Activate pylab support at runtime.
3049 """Activate pylab support at runtime.
3048
3050
3049 This turns on support for matplotlib, preloads into the interactive
3051 This turns on support for matplotlib, preloads into the interactive
3050 namespace all of numpy and pylab, and configures IPython to correctly
3052 namespace all of numpy and pylab, and configures IPython to correctly
3051 interact with the GUI event loop. The GUI backend to be used can be
3053 interact with the GUI event loop. The GUI backend to be used can be
3052 optionally selected with the optional ``gui`` argument.
3054 optionally selected with the optional ``gui`` argument.
3053
3055
3054 This method only adds preloading the namespace to InteractiveShell.enable_matplotlib.
3056 This method only adds preloading the namespace to InteractiveShell.enable_matplotlib.
3055
3057
3056 Parameters
3058 Parameters
3057 ----------
3059 ----------
3058 gui : optional, string
3060 gui : optional, string
3059 If given, dictates the choice of matplotlib GUI backend to use
3061 If given, dictates the choice of matplotlib GUI backend to use
3060 (should be one of IPython's supported backends, 'qt', 'osx', 'tk',
3062 (should be one of IPython's supported backends, 'qt', 'osx', 'tk',
3061 'gtk', 'wx' or 'inline'), otherwise we use the default chosen by
3063 'gtk', 'wx' or 'inline'), otherwise we use the default chosen by
3062 matplotlib (as dictated by the matplotlib build-time options plus the
3064 matplotlib (as dictated by the matplotlib build-time options plus the
3063 user's matplotlibrc configuration file). Note that not all backends
3065 user's matplotlibrc configuration file). Note that not all backends
3064 make sense in all contexts, for example a terminal ipython can't
3066 make sense in all contexts, for example a terminal ipython can't
3065 display figures inline.
3067 display figures inline.
3066 import_all : optional, bool, default: True
3068 import_all : optional, bool, default: True
3067 Whether to do `from numpy import *` and `from pylab import *`
3069 Whether to do `from numpy import *` and `from pylab import *`
3068 in addition to module imports.
3070 in addition to module imports.
3069 welcome_message : deprecated
3071 welcome_message : deprecated
3070 This argument is ignored, no welcome message will be displayed.
3072 This argument is ignored, no welcome message will be displayed.
3071 """
3073 """
3072 from IPython.core.pylabtools import import_pylab
3074 from IPython.core.pylabtools import import_pylab
3073
3075
3074 gui, backend = self.enable_matplotlib(gui)
3076 gui, backend = self.enable_matplotlib(gui)
3075
3077
3076 # We want to prevent the loading of pylab to pollute the user's
3078 # We want to prevent the loading of pylab to pollute the user's
3077 # namespace as shown by the %who* magics, so we execute the activation
3079 # namespace as shown by the %who* magics, so we execute the activation
3078 # code in an empty namespace, and we update *both* user_ns and
3080 # code in an empty namespace, and we update *both* user_ns and
3079 # user_ns_hidden with this information.
3081 # user_ns_hidden with this information.
3080 ns = {}
3082 ns = {}
3081 import_pylab(ns, import_all)
3083 import_pylab(ns, import_all)
3082 # warn about clobbered names
3084 # warn about clobbered names
3083 ignored = {"__builtins__"}
3085 ignored = {"__builtins__"}
3084 both = set(ns).intersection(self.user_ns).difference(ignored)
3086 both = set(ns).intersection(self.user_ns).difference(ignored)
3085 clobbered = [ name for name in both if self.user_ns[name] is not ns[name] ]
3087 clobbered = [ name for name in both if self.user_ns[name] is not ns[name] ]
3086 self.user_ns.update(ns)
3088 self.user_ns.update(ns)
3087 self.user_ns_hidden.update(ns)
3089 self.user_ns_hidden.update(ns)
3088 return gui, backend, clobbered
3090 return gui, backend, clobbered
3089
3091
3090 #-------------------------------------------------------------------------
3092 #-------------------------------------------------------------------------
3091 # Utilities
3093 # Utilities
3092 #-------------------------------------------------------------------------
3094 #-------------------------------------------------------------------------
3093
3095
3094 def var_expand(self, cmd, depth=0, formatter=DollarFormatter()):
3096 def var_expand(self, cmd, depth=0, formatter=DollarFormatter()):
3095 """Expand python variables in a string.
3097 """Expand python variables in a string.
3096
3098
3097 The depth argument indicates how many frames above the caller should
3099 The depth argument indicates how many frames above the caller should
3098 be walked to look for the local namespace where to expand variables.
3100 be walked to look for the local namespace where to expand variables.
3099
3101
3100 The global namespace for expansion is always the user's interactive
3102 The global namespace for expansion is always the user's interactive
3101 namespace.
3103 namespace.
3102 """
3104 """
3103 ns = self.user_ns.copy()
3105 ns = self.user_ns.copy()
3104 try:
3106 try:
3105 frame = sys._getframe(depth+1)
3107 frame = sys._getframe(depth+1)
3106 except ValueError:
3108 except ValueError:
3107 # This is thrown if there aren't that many frames on the stack,
3109 # This is thrown if there aren't that many frames on the stack,
3108 # e.g. if a script called run_line_magic() directly.
3110 # e.g. if a script called run_line_magic() directly.
3109 pass
3111 pass
3110 else:
3112 else:
3111 ns.update(frame.f_locals)
3113 ns.update(frame.f_locals)
3112
3114
3113 try:
3115 try:
3114 # We have to use .vformat() here, because 'self' is a valid and common
3116 # We have to use .vformat() here, because 'self' is a valid and common
3115 # name, and expanding **ns for .format() would make it collide with
3117 # name, and expanding **ns for .format() would make it collide with
3116 # the 'self' argument of the method.
3118 # the 'self' argument of the method.
3117 cmd = formatter.vformat(cmd, args=[], kwargs=ns)
3119 cmd = formatter.vformat(cmd, args=[], kwargs=ns)
3118 except Exception:
3120 except Exception:
3119 # if formatter couldn't format, just let it go untransformed
3121 # if formatter couldn't format, just let it go untransformed
3120 pass
3122 pass
3121 return cmd
3123 return cmd
3122
3124
3123 def mktempfile(self, data=None, prefix='ipython_edit_'):
3125 def mktempfile(self, data=None, prefix='ipython_edit_'):
3124 """Make a new tempfile and return its filename.
3126 """Make a new tempfile and return its filename.
3125
3127
3126 This makes a call to tempfile.mkstemp (created in a tempfile.mkdtemp),
3128 This makes a call to tempfile.mkstemp (created in a tempfile.mkdtemp),
3127 but it registers the created filename internally so ipython cleans it up
3129 but it registers the created filename internally so ipython cleans it up
3128 at exit time.
3130 at exit time.
3129
3131
3130 Optional inputs:
3132 Optional inputs:
3131
3133
3132 - data(None): if data is given, it gets written out to the temp file
3134 - data(None): if data is given, it gets written out to the temp file
3133 immediately, and the file is closed again."""
3135 immediately, and the file is closed again."""
3134
3136
3135 dirname = tempfile.mkdtemp(prefix=prefix)
3137 dirname = tempfile.mkdtemp(prefix=prefix)
3136 self.tempdirs.append(dirname)
3138 self.tempdirs.append(dirname)
3137
3139
3138 handle, filename = tempfile.mkstemp('.py', prefix, dir=dirname)
3140 handle, filename = tempfile.mkstemp('.py', prefix, dir=dirname)
3139 os.close(handle) # On Windows, there can only be one open handle on a file
3141 os.close(handle) # On Windows, there can only be one open handle on a file
3140 self.tempfiles.append(filename)
3142 self.tempfiles.append(filename)
3141
3143
3142 if data:
3144 if data:
3143 tmp_file = open(filename,'w')
3145 tmp_file = open(filename,'w')
3144 tmp_file.write(data)
3146 tmp_file.write(data)
3145 tmp_file.close()
3147 tmp_file.close()
3146 return filename
3148 return filename
3147
3149
3148 @undoc
3150 @undoc
3149 def write(self,data):
3151 def write(self,data):
3150 """DEPRECATED: Write a string to the default output"""
3152 """DEPRECATED: Write a string to the default output"""
3151 warn('InteractiveShell.write() is deprecated, use sys.stdout instead',
3153 warn('InteractiveShell.write() is deprecated, use sys.stdout instead',
3152 DeprecationWarning, stacklevel=2)
3154 DeprecationWarning, stacklevel=2)
3153 sys.stdout.write(data)
3155 sys.stdout.write(data)
3154
3156
3155 @undoc
3157 @undoc
3156 def write_err(self,data):
3158 def write_err(self,data):
3157 """DEPRECATED: Write a string to the default error output"""
3159 """DEPRECATED: Write a string to the default error output"""
3158 warn('InteractiveShell.write_err() is deprecated, use sys.stderr instead',
3160 warn('InteractiveShell.write_err() is deprecated, use sys.stderr instead',
3159 DeprecationWarning, stacklevel=2)
3161 DeprecationWarning, stacklevel=2)
3160 sys.stderr.write(data)
3162 sys.stderr.write(data)
3161
3163
3162 def ask_yes_no(self, prompt, default=None, interrupt=None):
3164 def ask_yes_no(self, prompt, default=None, interrupt=None):
3163 if self.quiet:
3165 if self.quiet:
3164 return True
3166 return True
3165 return ask_yes_no(prompt,default,interrupt)
3167 return ask_yes_no(prompt,default,interrupt)
3166
3168
3167 def show_usage(self):
3169 def show_usage(self):
3168 """Show a usage message"""
3170 """Show a usage message"""
3169 page.page(IPython.core.usage.interactive_usage)
3171 page.page(IPython.core.usage.interactive_usage)
3170
3172
3171 def extract_input_lines(self, range_str, raw=False):
3173 def extract_input_lines(self, range_str, raw=False):
3172 """Return as a string a set of input history slices.
3174 """Return as a string a set of input history slices.
3173
3175
3174 Parameters
3176 Parameters
3175 ----------
3177 ----------
3176 range_str : string
3178 range_str : string
3177 The set of slices is given as a string, like "~5/6-~4/2 4:8 9",
3179 The set of slices is given as a string, like "~5/6-~4/2 4:8 9",
3178 since this function is for use by magic functions which get their
3180 since this function is for use by magic functions which get their
3179 arguments as strings. The number before the / is the session
3181 arguments as strings. The number before the / is the session
3180 number: ~n goes n back from the current session.
3182 number: ~n goes n back from the current session.
3181
3183
3182 raw : bool, optional
3184 raw : bool, optional
3183 By default, the processed input is used. If this is true, the raw
3185 By default, the processed input is used. If this is true, the raw
3184 input history is used instead.
3186 input history is used instead.
3185
3187
3186 Notes
3188 Notes
3187 -----
3189 -----
3188
3190
3189 Slices can be described with two notations:
3191 Slices can be described with two notations:
3190
3192
3191 * ``N:M`` -> standard python form, means including items N...(M-1).
3193 * ``N:M`` -> standard python form, means including items N...(M-1).
3192 * ``N-M`` -> include items N..M (closed endpoint).
3194 * ``N-M`` -> include items N..M (closed endpoint).
3193 """
3195 """
3194 lines = self.history_manager.get_range_by_str(range_str, raw=raw)
3196 lines = self.history_manager.get_range_by_str(range_str, raw=raw)
3195 return "\n".join(x for _, _, x in lines)
3197 return "\n".join(x for _, _, x in lines)
3196
3198
3197 def find_user_code(self, target, raw=True, py_only=False, skip_encoding_cookie=True, search_ns=False):
3199 def find_user_code(self, target, raw=True, py_only=False, skip_encoding_cookie=True, search_ns=False):
3198 """Get a code string from history, file, url, or a string or macro.
3200 """Get a code string from history, file, url, or a string or macro.
3199
3201
3200 This is mainly used by magic functions.
3202 This is mainly used by magic functions.
3201
3203
3202 Parameters
3204 Parameters
3203 ----------
3205 ----------
3204
3206
3205 target : str
3207 target : str
3206
3208
3207 A string specifying code to retrieve. This will be tried respectively
3209 A string specifying code to retrieve. This will be tried respectively
3208 as: ranges of input history (see %history for syntax), url,
3210 as: ranges of input history (see %history for syntax), url,
3209 corresponding .py file, filename, or an expression evaluating to a
3211 corresponding .py file, filename, or an expression evaluating to a
3210 string or Macro in the user namespace.
3212 string or Macro in the user namespace.
3211
3213
3212 raw : bool
3214 raw : bool
3213 If true (default), retrieve raw history. Has no effect on the other
3215 If true (default), retrieve raw history. Has no effect on the other
3214 retrieval mechanisms.
3216 retrieval mechanisms.
3215
3217
3216 py_only : bool (default False)
3218 py_only : bool (default False)
3217 Only try to fetch python code, do not try alternative methods to decode file
3219 Only try to fetch python code, do not try alternative methods to decode file
3218 if unicode fails.
3220 if unicode fails.
3219
3221
3220 Returns
3222 Returns
3221 -------
3223 -------
3222 A string of code.
3224 A string of code.
3223
3225
3224 ValueError is raised if nothing is found, and TypeError if it evaluates
3226 ValueError is raised if nothing is found, and TypeError if it evaluates
3225 to an object of another type. In each case, .args[0] is a printable
3227 to an object of another type. In each case, .args[0] is a printable
3226 message.
3228 message.
3227 """
3229 """
3228 code = self.extract_input_lines(target, raw=raw) # Grab history
3230 code = self.extract_input_lines(target, raw=raw) # Grab history
3229 if code:
3231 if code:
3230 return code
3232 return code
3231 try:
3233 try:
3232 if target.startswith(('http://', 'https://')):
3234 if target.startswith(('http://', 'https://')):
3233 return openpy.read_py_url(target, skip_encoding_cookie=skip_encoding_cookie)
3235 return openpy.read_py_url(target, skip_encoding_cookie=skip_encoding_cookie)
3234 except UnicodeDecodeError:
3236 except UnicodeDecodeError:
3235 if not py_only :
3237 if not py_only :
3236 # Deferred import
3238 # Deferred import
3237 from urllib.request import urlopen
3239 from urllib.request import urlopen
3238 response = urlopen(target)
3240 response = urlopen(target)
3239 return response.read().decode('latin1')
3241 return response.read().decode('latin1')
3240 raise ValueError(("'%s' seem to be unreadable.") % target)
3242 raise ValueError(("'%s' seem to be unreadable.") % target)
3241
3243
3242 potential_target = [target]
3244 potential_target = [target]
3243 try :
3245 try :
3244 potential_target.insert(0,get_py_filename(target))
3246 potential_target.insert(0,get_py_filename(target))
3245 except IOError:
3247 except IOError:
3246 pass
3248 pass
3247
3249
3248 for tgt in potential_target :
3250 for tgt in potential_target :
3249 if os.path.isfile(tgt): # Read file
3251 if os.path.isfile(tgt): # Read file
3250 try :
3252 try :
3251 return openpy.read_py_file(tgt, skip_encoding_cookie=skip_encoding_cookie)
3253 return openpy.read_py_file(tgt, skip_encoding_cookie=skip_encoding_cookie)
3252 except UnicodeDecodeError :
3254 except UnicodeDecodeError :
3253 if not py_only :
3255 if not py_only :
3254 with io_open(tgt,'r', encoding='latin1') as f :
3256 with io_open(tgt,'r', encoding='latin1') as f :
3255 return f.read()
3257 return f.read()
3256 raise ValueError(("'%s' seem to be unreadable.") % target)
3258 raise ValueError(("'%s' seem to be unreadable.") % target)
3257 elif os.path.isdir(os.path.expanduser(tgt)):
3259 elif os.path.isdir(os.path.expanduser(tgt)):
3258 raise ValueError("'%s' is a directory, not a regular file." % target)
3260 raise ValueError("'%s' is a directory, not a regular file." % target)
3259
3261
3260 if search_ns:
3262 if search_ns:
3261 # Inspect namespace to load object source
3263 # Inspect namespace to load object source
3262 object_info = self.object_inspect(target, detail_level=1)
3264 object_info = self.object_inspect(target, detail_level=1)
3263 if object_info['found'] and object_info['source']:
3265 if object_info['found'] and object_info['source']:
3264 return object_info['source']
3266 return object_info['source']
3265
3267
3266 try: # User namespace
3268 try: # User namespace
3267 codeobj = eval(target, self.user_ns)
3269 codeobj = eval(target, self.user_ns)
3268 except Exception:
3270 except Exception:
3269 raise ValueError(("'%s' was not found in history, as a file, url, "
3271 raise ValueError(("'%s' was not found in history, as a file, url, "
3270 "nor in the user namespace.") % target)
3272 "nor in the user namespace.") % target)
3271
3273
3272 if isinstance(codeobj, str):
3274 if isinstance(codeobj, str):
3273 return codeobj
3275 return codeobj
3274 elif isinstance(codeobj, Macro):
3276 elif isinstance(codeobj, Macro):
3275 return codeobj.value
3277 return codeobj.value
3276
3278
3277 raise TypeError("%s is neither a string nor a macro." % target,
3279 raise TypeError("%s is neither a string nor a macro." % target,
3278 codeobj)
3280 codeobj)
3279
3281
3280 #-------------------------------------------------------------------------
3282 #-------------------------------------------------------------------------
3281 # Things related to IPython exiting
3283 # Things related to IPython exiting
3282 #-------------------------------------------------------------------------
3284 #-------------------------------------------------------------------------
3283 def atexit_operations(self):
3285 def atexit_operations(self):
3284 """This will be executed at the time of exit.
3286 """This will be executed at the time of exit.
3285
3287
3286 Cleanup operations and saving of persistent data that is done
3288 Cleanup operations and saving of persistent data that is done
3287 unconditionally by IPython should be performed here.
3289 unconditionally by IPython should be performed here.
3288
3290
3289 For things that may depend on startup flags or platform specifics (such
3291 For things that may depend on startup flags or platform specifics (such
3290 as having readline or not), register a separate atexit function in the
3292 as having readline or not), register a separate atexit function in the
3291 code that has the appropriate information, rather than trying to
3293 code that has the appropriate information, rather than trying to
3292 clutter
3294 clutter
3293 """
3295 """
3294 # Close the history session (this stores the end time and line count)
3296 # Close the history session (this stores the end time and line count)
3295 # this must be *before* the tempfile cleanup, in case of temporary
3297 # this must be *before* the tempfile cleanup, in case of temporary
3296 # history db
3298 # history db
3297 self.history_manager.end_session()
3299 self.history_manager.end_session()
3298
3300
3299 # Cleanup all tempfiles and folders left around
3301 # Cleanup all tempfiles and folders left around
3300 for tfile in self.tempfiles:
3302 for tfile in self.tempfiles:
3301 try:
3303 try:
3302 os.unlink(tfile)
3304 os.unlink(tfile)
3303 except OSError:
3305 except OSError:
3304 pass
3306 pass
3305
3307
3306 for tdir in self.tempdirs:
3308 for tdir in self.tempdirs:
3307 try:
3309 try:
3308 os.rmdir(tdir)
3310 os.rmdir(tdir)
3309 except OSError:
3311 except OSError:
3310 pass
3312 pass
3311
3313
3312 # Clear all user namespaces to release all references cleanly.
3314 # Clear all user namespaces to release all references cleanly.
3313 self.reset(new_session=False)
3315 self.reset(new_session=False)
3314
3316
3315 # Run user hooks
3317 # Run user hooks
3316 self.hooks.shutdown_hook()
3318 self.hooks.shutdown_hook()
3317
3319
3318 def cleanup(self):
3320 def cleanup(self):
3319 self.restore_sys_module_state()
3321 self.restore_sys_module_state()
3320
3322
3321
3323
3322 # Overridden in terminal subclass to change prompts
3324 # Overridden in terminal subclass to change prompts
3323 def switch_doctest_mode(self, mode):
3325 def switch_doctest_mode(self, mode):
3324 pass
3326 pass
3325
3327
3326
3328
3327 class InteractiveShellABC(metaclass=abc.ABCMeta):
3329 class InteractiveShellABC(metaclass=abc.ABCMeta):
3328 """An abstract base class for InteractiveShell."""
3330 """An abstract base class for InteractiveShell."""
3329
3331
3330 InteractiveShellABC.register(InteractiveShell)
3332 InteractiveShellABC.register(InteractiveShell)
@@ -1,924 +1,933 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2 """Tests for the key interactiveshell module.
2 """Tests for the key interactiveshell module.
3
3
4 Historically the main classes in interactiveshell have been under-tested. This
4 Historically the main classes in interactiveshell have been under-tested. This
5 module should grow as many single-method tests as possible to trap many of the
5 module should grow as many single-method tests as possible to trap many of the
6 recurring bugs we seem to encounter with high-level interaction.
6 recurring bugs we seem to encounter with high-level interaction.
7 """
7 """
8
8
9 # Copyright (c) IPython Development Team.
9 # Copyright (c) IPython Development Team.
10 # Distributed under the terms of the Modified BSD License.
10 # Distributed under the terms of the Modified BSD License.
11
11
12 import ast
12 import ast
13 import os
13 import os
14 import signal
14 import signal
15 import shutil
15 import shutil
16 import sys
16 import sys
17 import tempfile
17 import tempfile
18 import unittest
18 import unittest
19 from unittest import mock
19 from unittest import mock
20
20
21 from os.path import join
21 from os.path import join
22
22
23 import nose.tools as nt
23 import nose.tools as nt
24
24
25 from IPython.core.error import InputRejected
25 from IPython.core.error import InputRejected
26 from IPython.core.inputtransformer import InputTransformer
26 from IPython.core.inputtransformer import InputTransformer
27 from IPython.testing.decorators import (
27 from IPython.testing.decorators import (
28 skipif, skip_win32, onlyif_unicode_paths, onlyif_cmds_exist,
28 skipif, skip_win32, onlyif_unicode_paths, onlyif_cmds_exist,
29 )
29 )
30 from IPython.testing import tools as tt
30 from IPython.testing import tools as tt
31 from IPython.utils.process import find_cmd
31 from IPython.utils.process import find_cmd
32
32
33 #-----------------------------------------------------------------------------
33 #-----------------------------------------------------------------------------
34 # Globals
34 # Globals
35 #-----------------------------------------------------------------------------
35 #-----------------------------------------------------------------------------
36 # This is used by every single test, no point repeating it ad nauseam
36 # This is used by every single test, no point repeating it ad nauseam
37 ip = get_ipython()
37 ip = get_ipython()
38
38
39 #-----------------------------------------------------------------------------
39 #-----------------------------------------------------------------------------
40 # Tests
40 # Tests
41 #-----------------------------------------------------------------------------
41 #-----------------------------------------------------------------------------
42
42
43 class DerivedInterrupt(KeyboardInterrupt):
43 class DerivedInterrupt(KeyboardInterrupt):
44 pass
44 pass
45
45
46 class InteractiveShellTestCase(unittest.TestCase):
46 class InteractiveShellTestCase(unittest.TestCase):
47 def test_naked_string_cells(self):
47 def test_naked_string_cells(self):
48 """Test that cells with only naked strings are fully executed"""
48 """Test that cells with only naked strings are fully executed"""
49 # First, single-line inputs
49 # First, single-line inputs
50 ip.run_cell('"a"\n')
50 ip.run_cell('"a"\n')
51 self.assertEqual(ip.user_ns['_'], 'a')
51 self.assertEqual(ip.user_ns['_'], 'a')
52 # And also multi-line cells
52 # And also multi-line cells
53 ip.run_cell('"""a\nb"""\n')
53 ip.run_cell('"""a\nb"""\n')
54 self.assertEqual(ip.user_ns['_'], 'a\nb')
54 self.assertEqual(ip.user_ns['_'], 'a\nb')
55
55
56 def test_run_empty_cell(self):
56 def test_run_empty_cell(self):
57 """Just make sure we don't get a horrible error with a blank
57 """Just make sure we don't get a horrible error with a blank
58 cell of input. Yes, I did overlook that."""
58 cell of input. Yes, I did overlook that."""
59 old_xc = ip.execution_count
59 old_xc = ip.execution_count
60 res = ip.run_cell('')
60 res = ip.run_cell('')
61 self.assertEqual(ip.execution_count, old_xc)
61 self.assertEqual(ip.execution_count, old_xc)
62 self.assertEqual(res.execution_count, None)
62 self.assertEqual(res.execution_count, None)
63
63
64 def test_run_cell_multiline(self):
64 def test_run_cell_multiline(self):
65 """Multi-block, multi-line cells must execute correctly.
65 """Multi-block, multi-line cells must execute correctly.
66 """
66 """
67 src = '\n'.join(["x=1",
67 src = '\n'.join(["x=1",
68 "y=2",
68 "y=2",
69 "if 1:",
69 "if 1:",
70 " x += 1",
70 " x += 1",
71 " y += 1",])
71 " y += 1",])
72 res = ip.run_cell(src)
72 res = ip.run_cell(src)
73 self.assertEqual(ip.user_ns['x'], 2)
73 self.assertEqual(ip.user_ns['x'], 2)
74 self.assertEqual(ip.user_ns['y'], 3)
74 self.assertEqual(ip.user_ns['y'], 3)
75 self.assertEqual(res.success, True)
75 self.assertEqual(res.success, True)
76 self.assertEqual(res.result, None)
76 self.assertEqual(res.result, None)
77
77
78 def test_multiline_string_cells(self):
78 def test_multiline_string_cells(self):
79 "Code sprinkled with multiline strings should execute (GH-306)"
79 "Code sprinkled with multiline strings should execute (GH-306)"
80 ip.run_cell('tmp=0')
80 ip.run_cell('tmp=0')
81 self.assertEqual(ip.user_ns['tmp'], 0)
81 self.assertEqual(ip.user_ns['tmp'], 0)
82 res = ip.run_cell('tmp=1;"""a\nb"""\n')
82 res = ip.run_cell('tmp=1;"""a\nb"""\n')
83 self.assertEqual(ip.user_ns['tmp'], 1)
83 self.assertEqual(ip.user_ns['tmp'], 1)
84 self.assertEqual(res.success, True)
84 self.assertEqual(res.success, True)
85 self.assertEqual(res.result, "a\nb")
85 self.assertEqual(res.result, "a\nb")
86
86
87 def test_dont_cache_with_semicolon(self):
87 def test_dont_cache_with_semicolon(self):
88 "Ending a line with semicolon should not cache the returned object (GH-307)"
88 "Ending a line with semicolon should not cache the returned object (GH-307)"
89 oldlen = len(ip.user_ns['Out'])
89 oldlen = len(ip.user_ns['Out'])
90 for cell in ['1;', '1;1;']:
90 for cell in ['1;', '1;1;']:
91 res = ip.run_cell(cell, store_history=True)
91 res = ip.run_cell(cell, store_history=True)
92 newlen = len(ip.user_ns['Out'])
92 newlen = len(ip.user_ns['Out'])
93 self.assertEqual(oldlen, newlen)
93 self.assertEqual(oldlen, newlen)
94 self.assertIsNone(res.result)
94 self.assertIsNone(res.result)
95 i = 0
95 i = 0
96 #also test the default caching behavior
96 #also test the default caching behavior
97 for cell in ['1', '1;1']:
97 for cell in ['1', '1;1']:
98 ip.run_cell(cell, store_history=True)
98 ip.run_cell(cell, store_history=True)
99 newlen = len(ip.user_ns['Out'])
99 newlen = len(ip.user_ns['Out'])
100 i += 1
100 i += 1
101 self.assertEqual(oldlen+i, newlen)
101 self.assertEqual(oldlen+i, newlen)
102
102
103 def test_syntax_error(self):
103 def test_syntax_error(self):
104 res = ip.run_cell("raise = 3")
104 res = ip.run_cell("raise = 3")
105 self.assertIsInstance(res.error_before_exec, SyntaxError)
105 self.assertIsInstance(res.error_before_exec, SyntaxError)
106
106
107 def test_In_variable(self):
107 def test_In_variable(self):
108 "Verify that In variable grows with user input (GH-284)"
108 "Verify that In variable grows with user input (GH-284)"
109 oldlen = len(ip.user_ns['In'])
109 oldlen = len(ip.user_ns['In'])
110 ip.run_cell('1;', store_history=True)
110 ip.run_cell('1;', store_history=True)
111 newlen = len(ip.user_ns['In'])
111 newlen = len(ip.user_ns['In'])
112 self.assertEqual(oldlen+1, newlen)
112 self.assertEqual(oldlen+1, newlen)
113 self.assertEqual(ip.user_ns['In'][-1],'1;')
113 self.assertEqual(ip.user_ns['In'][-1],'1;')
114
114
115 def test_magic_names_in_string(self):
115 def test_magic_names_in_string(self):
116 ip.run_cell('a = """\n%exit\n"""')
116 ip.run_cell('a = """\n%exit\n"""')
117 self.assertEqual(ip.user_ns['a'], '\n%exit\n')
117 self.assertEqual(ip.user_ns['a'], '\n%exit\n')
118
118
119 def test_trailing_newline(self):
119 def test_trailing_newline(self):
120 """test that running !(command) does not raise a SyntaxError"""
120 """test that running !(command) does not raise a SyntaxError"""
121 ip.run_cell('!(true)\n', False)
121 ip.run_cell('!(true)\n', False)
122 ip.run_cell('!(true)\n\n\n', False)
122 ip.run_cell('!(true)\n\n\n', False)
123
123
124 def test_gh_597(self):
124 def test_gh_597(self):
125 """Pretty-printing lists of objects with non-ascii reprs may cause
125 """Pretty-printing lists of objects with non-ascii reprs may cause
126 problems."""
126 problems."""
127 class Spam(object):
127 class Spam(object):
128 def __repr__(self):
128 def __repr__(self):
129 return "\xe9"*50
129 return "\xe9"*50
130 import IPython.core.formatters
130 import IPython.core.formatters
131 f = IPython.core.formatters.PlainTextFormatter()
131 f = IPython.core.formatters.PlainTextFormatter()
132 f([Spam(),Spam()])
132 f([Spam(),Spam()])
133
133
134
134
135 def test_future_flags(self):
135 def test_future_flags(self):
136 """Check that future flags are used for parsing code (gh-777)"""
136 """Check that future flags are used for parsing code (gh-777)"""
137 ip.run_cell('from __future__ import barry_as_FLUFL')
137 ip.run_cell('from __future__ import barry_as_FLUFL')
138 try:
138 try:
139 ip.run_cell('prfunc_return_val = 1 <> 2')
139 ip.run_cell('prfunc_return_val = 1 <> 2')
140 assert 'prfunc_return_val' in ip.user_ns
140 assert 'prfunc_return_val' in ip.user_ns
141 finally:
141 finally:
142 # Reset compiler flags so we don't mess up other tests.
142 # Reset compiler flags so we don't mess up other tests.
143 ip.compile.reset_compiler_flags()
143 ip.compile.reset_compiler_flags()
144
144
145 def test_can_pickle(self):
145 def test_can_pickle(self):
146 "Can we pickle objects defined interactively (GH-29)"
146 "Can we pickle objects defined interactively (GH-29)"
147 ip = get_ipython()
147 ip = get_ipython()
148 ip.reset()
148 ip.reset()
149 ip.run_cell(("class Mylist(list):\n"
149 ip.run_cell(("class Mylist(list):\n"
150 " def __init__(self,x=[]):\n"
150 " def __init__(self,x=[]):\n"
151 " list.__init__(self,x)"))
151 " list.__init__(self,x)"))
152 ip.run_cell("w=Mylist([1,2,3])")
152 ip.run_cell("w=Mylist([1,2,3])")
153
153
154 from pickle import dumps
154 from pickle import dumps
155
155
156 # We need to swap in our main module - this is only necessary
156 # We need to swap in our main module - this is only necessary
157 # inside the test framework, because IPython puts the interactive module
157 # inside the test framework, because IPython puts the interactive module
158 # in place (but the test framework undoes this).
158 # in place (but the test framework undoes this).
159 _main = sys.modules['__main__']
159 _main = sys.modules['__main__']
160 sys.modules['__main__'] = ip.user_module
160 sys.modules['__main__'] = ip.user_module
161 try:
161 try:
162 res = dumps(ip.user_ns["w"])
162 res = dumps(ip.user_ns["w"])
163 finally:
163 finally:
164 sys.modules['__main__'] = _main
164 sys.modules['__main__'] = _main
165 self.assertTrue(isinstance(res, bytes))
165 self.assertTrue(isinstance(res, bytes))
166
166
167 def test_global_ns(self):
167 def test_global_ns(self):
168 "Code in functions must be able to access variables outside them."
168 "Code in functions must be able to access variables outside them."
169 ip = get_ipython()
169 ip = get_ipython()
170 ip.run_cell("a = 10")
170 ip.run_cell("a = 10")
171 ip.run_cell(("def f(x):\n"
171 ip.run_cell(("def f(x):\n"
172 " return x + a"))
172 " return x + a"))
173 ip.run_cell("b = f(12)")
173 ip.run_cell("b = f(12)")
174 self.assertEqual(ip.user_ns["b"], 22)
174 self.assertEqual(ip.user_ns["b"], 22)
175
175
176 def test_bad_custom_tb(self):
176 def test_bad_custom_tb(self):
177 """Check that InteractiveShell is protected from bad custom exception handlers"""
177 """Check that InteractiveShell is protected from bad custom exception handlers"""
178 ip.set_custom_exc((IOError,), lambda etype,value,tb: 1/0)
178 ip.set_custom_exc((IOError,), lambda etype,value,tb: 1/0)
179 self.assertEqual(ip.custom_exceptions, (IOError,))
179 self.assertEqual(ip.custom_exceptions, (IOError,))
180 with tt.AssertPrints("Custom TB Handler failed", channel='stderr'):
180 with tt.AssertPrints("Custom TB Handler failed", channel='stderr'):
181 ip.run_cell(u'raise IOError("foo")')
181 ip.run_cell(u'raise IOError("foo")')
182 self.assertEqual(ip.custom_exceptions, ())
182 self.assertEqual(ip.custom_exceptions, ())
183
183
184 def test_bad_custom_tb_return(self):
184 def test_bad_custom_tb_return(self):
185 """Check that InteractiveShell is protected from bad return types in custom exception handlers"""
185 """Check that InteractiveShell is protected from bad return types in custom exception handlers"""
186 ip.set_custom_exc((NameError,),lambda etype,value,tb, tb_offset=None: 1)
186 ip.set_custom_exc((NameError,),lambda etype,value,tb, tb_offset=None: 1)
187 self.assertEqual(ip.custom_exceptions, (NameError,))
187 self.assertEqual(ip.custom_exceptions, (NameError,))
188 with tt.AssertPrints("Custom TB Handler failed", channel='stderr'):
188 with tt.AssertPrints("Custom TB Handler failed", channel='stderr'):
189 ip.run_cell(u'a=abracadabra')
189 ip.run_cell(u'a=abracadabra')
190 self.assertEqual(ip.custom_exceptions, ())
190 self.assertEqual(ip.custom_exceptions, ())
191
191
192 def test_drop_by_id(self):
192 def test_drop_by_id(self):
193 myvars = {"a":object(), "b":object(), "c": object()}
193 myvars = {"a":object(), "b":object(), "c": object()}
194 ip.push(myvars, interactive=False)
194 ip.push(myvars, interactive=False)
195 for name in myvars:
195 for name in myvars:
196 assert name in ip.user_ns, name
196 assert name in ip.user_ns, name
197 assert name in ip.user_ns_hidden, name
197 assert name in ip.user_ns_hidden, name
198 ip.user_ns['b'] = 12
198 ip.user_ns['b'] = 12
199 ip.drop_by_id(myvars)
199 ip.drop_by_id(myvars)
200 for name in ["a", "c"]:
200 for name in ["a", "c"]:
201 assert name not in ip.user_ns, name
201 assert name not in ip.user_ns, name
202 assert name not in ip.user_ns_hidden, name
202 assert name not in ip.user_ns_hidden, name
203 assert ip.user_ns['b'] == 12
203 assert ip.user_ns['b'] == 12
204 ip.reset()
204 ip.reset()
205
205
206 def test_var_expand(self):
206 def test_var_expand(self):
207 ip.user_ns['f'] = u'Ca\xf1o'
207 ip.user_ns['f'] = u'Ca\xf1o'
208 self.assertEqual(ip.var_expand(u'echo $f'), u'echo Ca\xf1o')
208 self.assertEqual(ip.var_expand(u'echo $f'), u'echo Ca\xf1o')
209 self.assertEqual(ip.var_expand(u'echo {f}'), u'echo Ca\xf1o')
209 self.assertEqual(ip.var_expand(u'echo {f}'), u'echo Ca\xf1o')
210 self.assertEqual(ip.var_expand(u'echo {f[:-1]}'), u'echo Ca\xf1')
210 self.assertEqual(ip.var_expand(u'echo {f[:-1]}'), u'echo Ca\xf1')
211 self.assertEqual(ip.var_expand(u'echo {1*2}'), u'echo 2')
211 self.assertEqual(ip.var_expand(u'echo {1*2}'), u'echo 2')
212
212
213 self.assertEqual(ip.var_expand(u"grep x | awk '{print $1}'"), u"grep x | awk '{print $1}'")
213 self.assertEqual(ip.var_expand(u"grep x | awk '{print $1}'"), u"grep x | awk '{print $1}'")
214
214
215 ip.user_ns['f'] = b'Ca\xc3\xb1o'
215 ip.user_ns['f'] = b'Ca\xc3\xb1o'
216 # This should not raise any exception:
216 # This should not raise any exception:
217 ip.var_expand(u'echo $f')
217 ip.var_expand(u'echo $f')
218
218
219 def test_var_expand_local(self):
219 def test_var_expand_local(self):
220 """Test local variable expansion in !system and %magic calls"""
220 """Test local variable expansion in !system and %magic calls"""
221 # !system
221 # !system
222 ip.run_cell('def test():\n'
222 ip.run_cell('def test():\n'
223 ' lvar = "ttt"\n'
223 ' lvar = "ttt"\n'
224 ' ret = !echo {lvar}\n'
224 ' ret = !echo {lvar}\n'
225 ' return ret[0]\n')
225 ' return ret[0]\n')
226 res = ip.user_ns['test']()
226 res = ip.user_ns['test']()
227 nt.assert_in('ttt', res)
227 nt.assert_in('ttt', res)
228
228
229 # %magic
229 # %magic
230 ip.run_cell('def makemacro():\n'
230 ip.run_cell('def makemacro():\n'
231 ' macroname = "macro_var_expand_locals"\n'
231 ' macroname = "macro_var_expand_locals"\n'
232 ' %macro {macroname} codestr\n')
232 ' %macro {macroname} codestr\n')
233 ip.user_ns['codestr'] = "str(12)"
233 ip.user_ns['codestr'] = "str(12)"
234 ip.run_cell('makemacro()')
234 ip.run_cell('makemacro()')
235 nt.assert_in('macro_var_expand_locals', ip.user_ns)
235 nt.assert_in('macro_var_expand_locals', ip.user_ns)
236
236
237 def test_var_expand_self(self):
237 def test_var_expand_self(self):
238 """Test variable expansion with the name 'self', which was failing.
238 """Test variable expansion with the name 'self', which was failing.
239
239
240 See https://github.com/ipython/ipython/issues/1878#issuecomment-7698218
240 See https://github.com/ipython/ipython/issues/1878#issuecomment-7698218
241 """
241 """
242 ip.run_cell('class cTest:\n'
242 ip.run_cell('class cTest:\n'
243 ' classvar="see me"\n'
243 ' classvar="see me"\n'
244 ' def test(self):\n'
244 ' def test(self):\n'
245 ' res = !echo Variable: {self.classvar}\n'
245 ' res = !echo Variable: {self.classvar}\n'
246 ' return res[0]\n')
246 ' return res[0]\n')
247 nt.assert_in('see me', ip.user_ns['cTest']().test())
247 nt.assert_in('see me', ip.user_ns['cTest']().test())
248
248
249 def test_bad_var_expand(self):
249 def test_bad_var_expand(self):
250 """var_expand on invalid formats shouldn't raise"""
250 """var_expand on invalid formats shouldn't raise"""
251 # SyntaxError
251 # SyntaxError
252 self.assertEqual(ip.var_expand(u"{'a':5}"), u"{'a':5}")
252 self.assertEqual(ip.var_expand(u"{'a':5}"), u"{'a':5}")
253 # NameError
253 # NameError
254 self.assertEqual(ip.var_expand(u"{asdf}"), u"{asdf}")
254 self.assertEqual(ip.var_expand(u"{asdf}"), u"{asdf}")
255 # ZeroDivisionError
255 # ZeroDivisionError
256 self.assertEqual(ip.var_expand(u"{1/0}"), u"{1/0}")
256 self.assertEqual(ip.var_expand(u"{1/0}"), u"{1/0}")
257
257
258 def test_silent_postexec(self):
258 def test_silent_postexec(self):
259 """run_cell(silent=True) doesn't invoke pre/post_run_cell callbacks"""
259 """run_cell(silent=True) doesn't invoke pre/post_run_cell callbacks"""
260 pre_explicit = mock.Mock()
260 pre_explicit = mock.Mock()
261 pre_always = mock.Mock()
261 pre_always = mock.Mock()
262 post_explicit = mock.Mock()
262 post_explicit = mock.Mock()
263 post_always = mock.Mock()
263 post_always = mock.Mock()
264 all_mocks = [pre_explicit, pre_always, post_explicit, post_always]
264 all_mocks = [pre_explicit, pre_always, post_explicit, post_always]
265
265
266 ip.events.register('pre_run_cell', pre_explicit)
266 ip.events.register('pre_run_cell', pre_explicit)
267 ip.events.register('pre_execute', pre_always)
267 ip.events.register('pre_execute', pre_always)
268 ip.events.register('post_run_cell', post_explicit)
268 ip.events.register('post_run_cell', post_explicit)
269 ip.events.register('post_execute', post_always)
269 ip.events.register('post_execute', post_always)
270
270
271 try:
271 try:
272 ip.run_cell("1", silent=True)
272 ip.run_cell("1", silent=True)
273 assert pre_always.called
273 assert pre_always.called
274 assert not pre_explicit.called
274 assert not pre_explicit.called
275 assert post_always.called
275 assert post_always.called
276 assert not post_explicit.called
276 assert not post_explicit.called
277 # double-check that non-silent exec did what we expected
277 # double-check that non-silent exec did what we expected
278 # silent to avoid
278 # silent to avoid
279 ip.run_cell("1")
279 ip.run_cell("1")
280 assert pre_explicit.called
280 assert pre_explicit.called
281 assert post_explicit.called
281 assert post_explicit.called
282 info, = pre_explicit.call_args[0]
282 info, = pre_explicit.call_args[0]
283 result, = post_explicit.call_args[0]
283 result, = post_explicit.call_args[0]
284 self.assertEqual(info, result.info)
284 self.assertEqual(info, result.info)
285 # check that post hooks are always called
285 # check that post hooks are always called
286 [m.reset_mock() for m in all_mocks]
286 [m.reset_mock() for m in all_mocks]
287 ip.run_cell("syntax error")
287 ip.run_cell("syntax error")
288 assert pre_always.called
288 assert pre_always.called
289 assert pre_explicit.called
289 assert pre_explicit.called
290 assert post_always.called
290 assert post_always.called
291 assert post_explicit.called
291 assert post_explicit.called
292 info, = pre_explicit.call_args[0]
292 info, = pre_explicit.call_args[0]
293 result, = post_explicit.call_args[0]
293 result, = post_explicit.call_args[0]
294 self.assertEqual(info, result.info)
294 self.assertEqual(info, result.info)
295 finally:
295 finally:
296 # remove post-exec
296 # remove post-exec
297 ip.events.unregister('pre_run_cell', pre_explicit)
297 ip.events.unregister('pre_run_cell', pre_explicit)
298 ip.events.unregister('pre_execute', pre_always)
298 ip.events.unregister('pre_execute', pre_always)
299 ip.events.unregister('post_run_cell', post_explicit)
299 ip.events.unregister('post_run_cell', post_explicit)
300 ip.events.unregister('post_execute', post_always)
300 ip.events.unregister('post_execute', post_always)
301
301
302 def test_silent_noadvance(self):
302 def test_silent_noadvance(self):
303 """run_cell(silent=True) doesn't advance execution_count"""
303 """run_cell(silent=True) doesn't advance execution_count"""
304 ec = ip.execution_count
304 ec = ip.execution_count
305 # silent should force store_history=False
305 # silent should force store_history=False
306 ip.run_cell("1", store_history=True, silent=True)
306 ip.run_cell("1", store_history=True, silent=True)
307
307
308 self.assertEqual(ec, ip.execution_count)
308 self.assertEqual(ec, ip.execution_count)
309 # double-check that non-silent exec did what we expected
309 # double-check that non-silent exec did what we expected
310 # silent to avoid
310 # silent to avoid
311 ip.run_cell("1", store_history=True)
311 ip.run_cell("1", store_history=True)
312 self.assertEqual(ec+1, ip.execution_count)
312 self.assertEqual(ec+1, ip.execution_count)
313
313
314 def test_silent_nodisplayhook(self):
314 def test_silent_nodisplayhook(self):
315 """run_cell(silent=True) doesn't trigger displayhook"""
315 """run_cell(silent=True) doesn't trigger displayhook"""
316 d = dict(called=False)
316 d = dict(called=False)
317
317
318 trap = ip.display_trap
318 trap = ip.display_trap
319 save_hook = trap.hook
319 save_hook = trap.hook
320
320
321 def failing_hook(*args, **kwargs):
321 def failing_hook(*args, **kwargs):
322 d['called'] = True
322 d['called'] = True
323
323
324 try:
324 try:
325 trap.hook = failing_hook
325 trap.hook = failing_hook
326 res = ip.run_cell("1", silent=True)
326 res = ip.run_cell("1", silent=True)
327 self.assertFalse(d['called'])
327 self.assertFalse(d['called'])
328 self.assertIsNone(res.result)
328 self.assertIsNone(res.result)
329 # double-check that non-silent exec did what we expected
329 # double-check that non-silent exec did what we expected
330 # silent to avoid
330 # silent to avoid
331 ip.run_cell("1")
331 ip.run_cell("1")
332 self.assertTrue(d['called'])
332 self.assertTrue(d['called'])
333 finally:
333 finally:
334 trap.hook = save_hook
334 trap.hook = save_hook
335
335
336 def test_ofind_line_magic(self):
336 def test_ofind_line_magic(self):
337 from IPython.core.magic import register_line_magic
337 from IPython.core.magic import register_line_magic
338
338
339 @register_line_magic
339 @register_line_magic
340 def lmagic(line):
340 def lmagic(line):
341 "A line magic"
341 "A line magic"
342
342
343 # Get info on line magic
343 # Get info on line magic
344 lfind = ip._ofind('lmagic')
344 lfind = ip._ofind('lmagic')
345 info = dict(found=True, isalias=False, ismagic=True,
345 info = dict(found=True, isalias=False, ismagic=True,
346 namespace = 'IPython internal', obj= lmagic.__wrapped__,
346 namespace = 'IPython internal', obj= lmagic.__wrapped__,
347 parent = None)
347 parent = None)
348 nt.assert_equal(lfind, info)
348 nt.assert_equal(lfind, info)
349
349
350 def test_ofind_cell_magic(self):
350 def test_ofind_cell_magic(self):
351 from IPython.core.magic import register_cell_magic
351 from IPython.core.magic import register_cell_magic
352
352
353 @register_cell_magic
353 @register_cell_magic
354 def cmagic(line, cell):
354 def cmagic(line, cell):
355 "A cell magic"
355 "A cell magic"
356
356
357 # Get info on cell magic
357 # Get info on cell magic
358 find = ip._ofind('cmagic')
358 find = ip._ofind('cmagic')
359 info = dict(found=True, isalias=False, ismagic=True,
359 info = dict(found=True, isalias=False, ismagic=True,
360 namespace = 'IPython internal', obj= cmagic.__wrapped__,
360 namespace = 'IPython internal', obj= cmagic.__wrapped__,
361 parent = None)
361 parent = None)
362 nt.assert_equal(find, info)
362 nt.assert_equal(find, info)
363
363
364 def test_ofind_property_with_error(self):
364 def test_ofind_property_with_error(self):
365 class A(object):
365 class A(object):
366 @property
366 @property
367 def foo(self):
367 def foo(self):
368 raise NotImplementedError()
368 raise NotImplementedError()
369 a = A()
369 a = A()
370
370
371 found = ip._ofind('a.foo', [('locals', locals())])
371 found = ip._ofind('a.foo', [('locals', locals())])
372 info = dict(found=True, isalias=False, ismagic=False,
372 info = dict(found=True, isalias=False, ismagic=False,
373 namespace='locals', obj=A.foo, parent=a)
373 namespace='locals', obj=A.foo, parent=a)
374 nt.assert_equal(found, info)
374 nt.assert_equal(found, info)
375
375
376 def test_ofind_multiple_attribute_lookups(self):
376 def test_ofind_multiple_attribute_lookups(self):
377 class A(object):
377 class A(object):
378 @property
378 @property
379 def foo(self):
379 def foo(self):
380 raise NotImplementedError()
380 raise NotImplementedError()
381
381
382 a = A()
382 a = A()
383 a.a = A()
383 a.a = A()
384 a.a.a = A()
384 a.a.a = A()
385
385
386 found = ip._ofind('a.a.a.foo', [('locals', locals())])
386 found = ip._ofind('a.a.a.foo', [('locals', locals())])
387 info = dict(found=True, isalias=False, ismagic=False,
387 info = dict(found=True, isalias=False, ismagic=False,
388 namespace='locals', obj=A.foo, parent=a.a.a)
388 namespace='locals', obj=A.foo, parent=a.a.a)
389 nt.assert_equal(found, info)
389 nt.assert_equal(found, info)
390
390
391 def test_ofind_slotted_attributes(self):
391 def test_ofind_slotted_attributes(self):
392 class A(object):
392 class A(object):
393 __slots__ = ['foo']
393 __slots__ = ['foo']
394 def __init__(self):
394 def __init__(self):
395 self.foo = 'bar'
395 self.foo = 'bar'
396
396
397 a = A()
397 a = A()
398 found = ip._ofind('a.foo', [('locals', locals())])
398 found = ip._ofind('a.foo', [('locals', locals())])
399 info = dict(found=True, isalias=False, ismagic=False,
399 info = dict(found=True, isalias=False, ismagic=False,
400 namespace='locals', obj=a.foo, parent=a)
400 namespace='locals', obj=a.foo, parent=a)
401 nt.assert_equal(found, info)
401 nt.assert_equal(found, info)
402
402
403 found = ip._ofind('a.bar', [('locals', locals())])
403 found = ip._ofind('a.bar', [('locals', locals())])
404 info = dict(found=False, isalias=False, ismagic=False,
404 info = dict(found=False, isalias=False, ismagic=False,
405 namespace=None, obj=None, parent=a)
405 namespace=None, obj=None, parent=a)
406 nt.assert_equal(found, info)
406 nt.assert_equal(found, info)
407
407
408 def test_ofind_prefers_property_to_instance_level_attribute(self):
408 def test_ofind_prefers_property_to_instance_level_attribute(self):
409 class A(object):
409 class A(object):
410 @property
410 @property
411 def foo(self):
411 def foo(self):
412 return 'bar'
412 return 'bar'
413 a = A()
413 a = A()
414 a.__dict__['foo'] = 'baz'
414 a.__dict__['foo'] = 'baz'
415 nt.assert_equal(a.foo, 'bar')
415 nt.assert_equal(a.foo, 'bar')
416 found = ip._ofind('a.foo', [('locals', locals())])
416 found = ip._ofind('a.foo', [('locals', locals())])
417 nt.assert_is(found['obj'], A.foo)
417 nt.assert_is(found['obj'], A.foo)
418
418
419 def test_custom_syntaxerror_exception(self):
419 def test_custom_syntaxerror_exception(self):
420 called = []
420 called = []
421 def my_handler(shell, etype, value, tb, tb_offset=None):
421 def my_handler(shell, etype, value, tb, tb_offset=None):
422 called.append(etype)
422 called.append(etype)
423 shell.showtraceback((etype, value, tb), tb_offset=tb_offset)
423 shell.showtraceback((etype, value, tb), tb_offset=tb_offset)
424
424
425 ip.set_custom_exc((SyntaxError,), my_handler)
425 ip.set_custom_exc((SyntaxError,), my_handler)
426 try:
426 try:
427 ip.run_cell("1f")
427 ip.run_cell("1f")
428 # Check that this was called, and only once.
428 # Check that this was called, and only once.
429 self.assertEqual(called, [SyntaxError])
429 self.assertEqual(called, [SyntaxError])
430 finally:
430 finally:
431 # Reset the custom exception hook
431 # Reset the custom exception hook
432 ip.set_custom_exc((), None)
432 ip.set_custom_exc((), None)
433
433
434 def test_custom_exception(self):
434 def test_custom_exception(self):
435 called = []
435 called = []
436 def my_handler(shell, etype, value, tb, tb_offset=None):
436 def my_handler(shell, etype, value, tb, tb_offset=None):
437 called.append(etype)
437 called.append(etype)
438 shell.showtraceback((etype, value, tb), tb_offset=tb_offset)
438 shell.showtraceback((etype, value, tb), tb_offset=tb_offset)
439
439
440 ip.set_custom_exc((ValueError,), my_handler)
440 ip.set_custom_exc((ValueError,), my_handler)
441 try:
441 try:
442 res = ip.run_cell("raise ValueError('test')")
442 res = ip.run_cell("raise ValueError('test')")
443 # Check that this was called, and only once.
443 # Check that this was called, and only once.
444 self.assertEqual(called, [ValueError])
444 self.assertEqual(called, [ValueError])
445 # Check that the error is on the result object
445 # Check that the error is on the result object
446 self.assertIsInstance(res.error_in_exec, ValueError)
446 self.assertIsInstance(res.error_in_exec, ValueError)
447 finally:
447 finally:
448 # Reset the custom exception hook
448 # Reset the custom exception hook
449 ip.set_custom_exc((), None)
449 ip.set_custom_exc((), None)
450
450
451 def test_mktempfile(self):
451 def test_mktempfile(self):
452 filename = ip.mktempfile()
452 filename = ip.mktempfile()
453 # Check that we can open the file again on Windows
453 # Check that we can open the file again on Windows
454 with open(filename, 'w') as f:
454 with open(filename, 'w') as f:
455 f.write('abc')
455 f.write('abc')
456
456
457 filename = ip.mktempfile(data='blah')
457 filename = ip.mktempfile(data='blah')
458 with open(filename, 'r') as f:
458 with open(filename, 'r') as f:
459 self.assertEqual(f.read(), 'blah')
459 self.assertEqual(f.read(), 'blah')
460
460
461 def test_new_main_mod(self):
461 def test_new_main_mod(self):
462 # Smoketest to check that this accepts a unicode module name
462 # Smoketest to check that this accepts a unicode module name
463 name = u'jiefmw'
463 name = u'jiefmw'
464 mod = ip.new_main_mod(u'%s.py' % name, name)
464 mod = ip.new_main_mod(u'%s.py' % name, name)
465 self.assertEqual(mod.__name__, name)
465 self.assertEqual(mod.__name__, name)
466
466
467 def test_get_exception_only(self):
467 def test_get_exception_only(self):
468 try:
468 try:
469 raise KeyboardInterrupt
469 raise KeyboardInterrupt
470 except KeyboardInterrupt:
470 except KeyboardInterrupt:
471 msg = ip.get_exception_only()
471 msg = ip.get_exception_only()
472 self.assertEqual(msg, 'KeyboardInterrupt\n')
472 self.assertEqual(msg, 'KeyboardInterrupt\n')
473
473
474 try:
474 try:
475 raise DerivedInterrupt("foo")
475 raise DerivedInterrupt("foo")
476 except KeyboardInterrupt:
476 except KeyboardInterrupt:
477 msg = ip.get_exception_only()
477 msg = ip.get_exception_only()
478 self.assertEqual(msg, 'IPython.core.tests.test_interactiveshell.DerivedInterrupt: foo\n')
478 self.assertEqual(msg, 'IPython.core.tests.test_interactiveshell.DerivedInterrupt: foo\n')
479
479
480 def test_inspect_text(self):
480 def test_inspect_text(self):
481 ip.run_cell('a = 5')
481 ip.run_cell('a = 5')
482 text = ip.object_inspect_text('a')
482 text = ip.object_inspect_text('a')
483 self.assertIsInstance(text, str)
483 self.assertIsInstance(text, str)
484
484
485 def test_last_execution_result(self):
485 def test_last_execution_result(self):
486 """ Check that last execution result gets set correctly (GH-10702) """
486 """ Check that last execution result gets set correctly (GH-10702) """
487 result = ip.run_cell('a = 5; a')
487 result = ip.run_cell('a = 5; a')
488 self.assertTrue(ip.last_execution_succeeded)
488 self.assertTrue(ip.last_execution_succeeded)
489 self.assertEqual(ip.last_execution_result.result, 5)
489 self.assertEqual(ip.last_execution_result.result, 5)
490
490
491 result = ip.run_cell('a = x_invalid_id_x')
491 result = ip.run_cell('a = x_invalid_id_x')
492 self.assertFalse(ip.last_execution_succeeded)
492 self.assertFalse(ip.last_execution_succeeded)
493 self.assertFalse(ip.last_execution_result.success)
493 self.assertFalse(ip.last_execution_result.success)
494 self.assertIsInstance(ip.last_execution_result.error_in_exec, NameError)
494 self.assertIsInstance(ip.last_execution_result.error_in_exec, NameError)
495
495
496
496
497 class TestSafeExecfileNonAsciiPath(unittest.TestCase):
497 class TestSafeExecfileNonAsciiPath(unittest.TestCase):
498
498
499 @onlyif_unicode_paths
499 @onlyif_unicode_paths
500 def setUp(self):
500 def setUp(self):
501 self.BASETESTDIR = tempfile.mkdtemp()
501 self.BASETESTDIR = tempfile.mkdtemp()
502 self.TESTDIR = join(self.BASETESTDIR, u"Γ₯Àâ")
502 self.TESTDIR = join(self.BASETESTDIR, u"Γ₯Àâ")
503 os.mkdir(self.TESTDIR)
503 os.mkdir(self.TESTDIR)
504 with open(join(self.TESTDIR, u"Γ₯Àâtestscript.py"), "w") as sfile:
504 with open(join(self.TESTDIR, u"Γ₯Àâtestscript.py"), "w") as sfile:
505 sfile.write("pass\n")
505 sfile.write("pass\n")
506 self.oldpath = os.getcwd()
506 self.oldpath = os.getcwd()
507 os.chdir(self.TESTDIR)
507 os.chdir(self.TESTDIR)
508 self.fname = u"Γ₯Àâtestscript.py"
508 self.fname = u"Γ₯Àâtestscript.py"
509
509
510 def tearDown(self):
510 def tearDown(self):
511 os.chdir(self.oldpath)
511 os.chdir(self.oldpath)
512 shutil.rmtree(self.BASETESTDIR)
512 shutil.rmtree(self.BASETESTDIR)
513
513
514 @onlyif_unicode_paths
514 @onlyif_unicode_paths
515 def test_1(self):
515 def test_1(self):
516 """Test safe_execfile with non-ascii path
516 """Test safe_execfile with non-ascii path
517 """
517 """
518 ip.safe_execfile(self.fname, {}, raise_exceptions=True)
518 ip.safe_execfile(self.fname, {}, raise_exceptions=True)
519
519
520 class ExitCodeChecks(tt.TempFileMixin):
520 class ExitCodeChecks(tt.TempFileMixin):
521 def test_exit_code_ok(self):
521 def test_exit_code_ok(self):
522 self.system('exit 0')
522 self.system('exit 0')
523 self.assertEqual(ip.user_ns['_exit_code'], 0)
523 self.assertEqual(ip.user_ns['_exit_code'], 0)
524
524
525 def test_exit_code_error(self):
525 def test_exit_code_error(self):
526 self.system('exit 1')
526 self.system('exit 1')
527 self.assertEqual(ip.user_ns['_exit_code'], 1)
527 self.assertEqual(ip.user_ns['_exit_code'], 1)
528
528
529 @skipif(not hasattr(signal, 'SIGALRM'))
529 @skipif(not hasattr(signal, 'SIGALRM'))
530 def test_exit_code_signal(self):
530 def test_exit_code_signal(self):
531 self.mktmp("import signal, time\n"
531 self.mktmp("import signal, time\n"
532 "signal.setitimer(signal.ITIMER_REAL, 0.1)\n"
532 "signal.setitimer(signal.ITIMER_REAL, 0.1)\n"
533 "time.sleep(1)\n")
533 "time.sleep(1)\n")
534 self.system("%s %s" % (sys.executable, self.fname))
534 self.system("%s %s" % (sys.executable, self.fname))
535 self.assertEqual(ip.user_ns['_exit_code'], -signal.SIGALRM)
535 self.assertEqual(ip.user_ns['_exit_code'], -signal.SIGALRM)
536
536
537 @onlyif_cmds_exist("csh")
537 @onlyif_cmds_exist("csh")
538 def test_exit_code_signal_csh(self):
538 def test_exit_code_signal_csh(self):
539 SHELL = os.environ.get('SHELL', None)
539 SHELL = os.environ.get('SHELL', None)
540 os.environ['SHELL'] = find_cmd("csh")
540 os.environ['SHELL'] = find_cmd("csh")
541 try:
541 try:
542 self.test_exit_code_signal()
542 self.test_exit_code_signal()
543 finally:
543 finally:
544 if SHELL is not None:
544 if SHELL is not None:
545 os.environ['SHELL'] = SHELL
545 os.environ['SHELL'] = SHELL
546 else:
546 else:
547 del os.environ['SHELL']
547 del os.environ['SHELL']
548
548
549 class TestSystemRaw(unittest.TestCase, ExitCodeChecks):
549 class TestSystemRaw(unittest.TestCase, ExitCodeChecks):
550 system = ip.system_raw
550 system = ip.system_raw
551
551
552 @onlyif_unicode_paths
552 @onlyif_unicode_paths
553 def test_1(self):
553 def test_1(self):
554 """Test system_raw with non-ascii cmd
554 """Test system_raw with non-ascii cmd
555 """
555 """
556 cmd = u'''python -c "'Γ₯Àâ'" '''
556 cmd = u'''python -c "'Γ₯Àâ'" '''
557 ip.system_raw(cmd)
557 ip.system_raw(cmd)
558
558
559 @mock.patch('subprocess.call', side_effect=KeyboardInterrupt)
559 @mock.patch('subprocess.call', side_effect=KeyboardInterrupt)
560 @mock.patch('os.system', side_effect=KeyboardInterrupt)
560 @mock.patch('os.system', side_effect=KeyboardInterrupt)
561 def test_control_c(self, *mocks):
561 def test_control_c(self, *mocks):
562 try:
562 try:
563 self.system("sleep 1 # wont happen")
563 self.system("sleep 1 # wont happen")
564 except KeyboardInterrupt:
564 except KeyboardInterrupt:
565 self.fail("system call should intercept "
565 self.fail("system call should intercept "
566 "keyboard interrupt from subprocess.call")
566 "keyboard interrupt from subprocess.call")
567 self.assertEqual(ip.user_ns['_exit_code'], -signal.SIGINT)
567 self.assertEqual(ip.user_ns['_exit_code'], -signal.SIGINT)
568
568
569 # TODO: Exit codes are currently ignored on Windows.
569 # TODO: Exit codes are currently ignored on Windows.
570 class TestSystemPipedExitCode(unittest.TestCase, ExitCodeChecks):
570 class TestSystemPipedExitCode(unittest.TestCase, ExitCodeChecks):
571 system = ip.system_piped
571 system = ip.system_piped
572
572
573 @skip_win32
573 @skip_win32
574 def test_exit_code_ok(self):
574 def test_exit_code_ok(self):
575 ExitCodeChecks.test_exit_code_ok(self)
575 ExitCodeChecks.test_exit_code_ok(self)
576
576
577 @skip_win32
577 @skip_win32
578 def test_exit_code_error(self):
578 def test_exit_code_error(self):
579 ExitCodeChecks.test_exit_code_error(self)
579 ExitCodeChecks.test_exit_code_error(self)
580
580
581 @skip_win32
581 @skip_win32
582 def test_exit_code_signal(self):
582 def test_exit_code_signal(self):
583 ExitCodeChecks.test_exit_code_signal(self)
583 ExitCodeChecks.test_exit_code_signal(self)
584
584
585 class TestModules(unittest.TestCase, tt.TempFileMixin):
585 class TestModules(unittest.TestCase, tt.TempFileMixin):
586 def test_extraneous_loads(self):
586 def test_extraneous_loads(self):
587 """Test we're not loading modules on startup that we shouldn't.
587 """Test we're not loading modules on startup that we shouldn't.
588 """
588 """
589 self.mktmp("import sys\n"
589 self.mktmp("import sys\n"
590 "print('numpy' in sys.modules)\n"
590 "print('numpy' in sys.modules)\n"
591 "print('ipyparallel' in sys.modules)\n"
591 "print('ipyparallel' in sys.modules)\n"
592 "print('ipykernel' in sys.modules)\n"
592 "print('ipykernel' in sys.modules)\n"
593 )
593 )
594 out = "False\nFalse\nFalse\n"
594 out = "False\nFalse\nFalse\n"
595 tt.ipexec_validate(self.fname, out)
595 tt.ipexec_validate(self.fname, out)
596
596
597 class Negator(ast.NodeTransformer):
597 class Negator(ast.NodeTransformer):
598 """Negates all number literals in an AST."""
598 """Negates all number literals in an AST."""
599 def visit_Num(self, node):
599 def visit_Num(self, node):
600 node.n = -node.n
600 node.n = -node.n
601 return node
601 return node
602
602
603 class TestAstTransform(unittest.TestCase):
603 class TestAstTransform(unittest.TestCase):
604 def setUp(self):
604 def setUp(self):
605 self.negator = Negator()
605 self.negator = Negator()
606 ip.ast_transformers.append(self.negator)
606 ip.ast_transformers.append(self.negator)
607
607
608 def tearDown(self):
608 def tearDown(self):
609 ip.ast_transformers.remove(self.negator)
609 ip.ast_transformers.remove(self.negator)
610
610
611 def test_run_cell(self):
611 def test_run_cell(self):
612 with tt.AssertPrints('-34'):
612 with tt.AssertPrints('-34'):
613 ip.run_cell('print (12 + 22)')
613 ip.run_cell('print (12 + 22)')
614
614
615 # A named reference to a number shouldn't be transformed.
615 # A named reference to a number shouldn't be transformed.
616 ip.user_ns['n'] = 55
616 ip.user_ns['n'] = 55
617 with tt.AssertNotPrints('-55'):
617 with tt.AssertNotPrints('-55'):
618 ip.run_cell('print (n)')
618 ip.run_cell('print (n)')
619
619
620 def test_timeit(self):
620 def test_timeit(self):
621 called = set()
621 called = set()
622 def f(x):
622 def f(x):
623 called.add(x)
623 called.add(x)
624 ip.push({'f':f})
624 ip.push({'f':f})
625
625
626 with tt.AssertPrints("std. dev. of"):
626 with tt.AssertPrints("std. dev. of"):
627 ip.run_line_magic("timeit", "-n1 f(1)")
627 ip.run_line_magic("timeit", "-n1 f(1)")
628 self.assertEqual(called, {-1})
628 self.assertEqual(called, {-1})
629 called.clear()
629 called.clear()
630
630
631 with tt.AssertPrints("std. dev. of"):
631 with tt.AssertPrints("std. dev. of"):
632 ip.run_cell_magic("timeit", "-n1 f(2)", "f(3)")
632 ip.run_cell_magic("timeit", "-n1 f(2)", "f(3)")
633 self.assertEqual(called, {-2, -3})
633 self.assertEqual(called, {-2, -3})
634
634
635 def test_time(self):
635 def test_time(self):
636 called = []
636 called = []
637 def f(x):
637 def f(x):
638 called.append(x)
638 called.append(x)
639 ip.push({'f':f})
639 ip.push({'f':f})
640
640
641 # Test with an expression
641 # Test with an expression
642 with tt.AssertPrints("Wall time: "):
642 with tt.AssertPrints("Wall time: "):
643 ip.run_line_magic("time", "f(5+9)")
643 ip.run_line_magic("time", "f(5+9)")
644 self.assertEqual(called, [-14])
644 self.assertEqual(called, [-14])
645 called[:] = []
645 called[:] = []
646
646
647 # Test with a statement (different code path)
647 # Test with a statement (different code path)
648 with tt.AssertPrints("Wall time: "):
648 with tt.AssertPrints("Wall time: "):
649 ip.run_line_magic("time", "a = f(-3 + -2)")
649 ip.run_line_magic("time", "a = f(-3 + -2)")
650 self.assertEqual(called, [5])
650 self.assertEqual(called, [5])
651
651
652 def test_macro(self):
652 def test_macro(self):
653 ip.push({'a':10})
653 ip.push({'a':10})
654 # The AST transformation makes this do a+=-1
654 # The AST transformation makes this do a+=-1
655 ip.define_macro("amacro", "a+=1\nprint(a)")
655 ip.define_macro("amacro", "a+=1\nprint(a)")
656
656
657 with tt.AssertPrints("9"):
657 with tt.AssertPrints("9"):
658 ip.run_cell("amacro")
658 ip.run_cell("amacro")
659 with tt.AssertPrints("8"):
659 with tt.AssertPrints("8"):
660 ip.run_cell("amacro")
660 ip.run_cell("amacro")
661
661
662 class IntegerWrapper(ast.NodeTransformer):
662 class IntegerWrapper(ast.NodeTransformer):
663 """Wraps all integers in a call to Integer()"""
663 """Wraps all integers in a call to Integer()"""
664 def visit_Num(self, node):
664 def visit_Num(self, node):
665 if isinstance(node.n, int):
665 if isinstance(node.n, int):
666 return ast.Call(func=ast.Name(id='Integer', ctx=ast.Load()),
666 return ast.Call(func=ast.Name(id='Integer', ctx=ast.Load()),
667 args=[node], keywords=[])
667 args=[node], keywords=[])
668 return node
668 return node
669
669
670 class TestAstTransform2(unittest.TestCase):
670 class TestAstTransform2(unittest.TestCase):
671 def setUp(self):
671 def setUp(self):
672 self.intwrapper = IntegerWrapper()
672 self.intwrapper = IntegerWrapper()
673 ip.ast_transformers.append(self.intwrapper)
673 ip.ast_transformers.append(self.intwrapper)
674
674
675 self.calls = []
675 self.calls = []
676 def Integer(*args):
676 def Integer(*args):
677 self.calls.append(args)
677 self.calls.append(args)
678 return args
678 return args
679 ip.push({"Integer": Integer})
679 ip.push({"Integer": Integer})
680
680
681 def tearDown(self):
681 def tearDown(self):
682 ip.ast_transformers.remove(self.intwrapper)
682 ip.ast_transformers.remove(self.intwrapper)
683 del ip.user_ns['Integer']
683 del ip.user_ns['Integer']
684
684
685 def test_run_cell(self):
685 def test_run_cell(self):
686 ip.run_cell("n = 2")
686 ip.run_cell("n = 2")
687 self.assertEqual(self.calls, [(2,)])
687 self.assertEqual(self.calls, [(2,)])
688
688
689 # This shouldn't throw an error
689 # This shouldn't throw an error
690 ip.run_cell("o = 2.0")
690 ip.run_cell("o = 2.0")
691 self.assertEqual(ip.user_ns['o'], 2.0)
691 self.assertEqual(ip.user_ns['o'], 2.0)
692
692
693 def test_timeit(self):
693 def test_timeit(self):
694 called = set()
694 called = set()
695 def f(x):
695 def f(x):
696 called.add(x)
696 called.add(x)
697 ip.push({'f':f})
697 ip.push({'f':f})
698
698
699 with tt.AssertPrints("std. dev. of"):
699 with tt.AssertPrints("std. dev. of"):
700 ip.run_line_magic("timeit", "-n1 f(1)")
700 ip.run_line_magic("timeit", "-n1 f(1)")
701 self.assertEqual(called, {(1,)})
701 self.assertEqual(called, {(1,)})
702 called.clear()
702 called.clear()
703
703
704 with tt.AssertPrints("std. dev. of"):
704 with tt.AssertPrints("std. dev. of"):
705 ip.run_cell_magic("timeit", "-n1 f(2)", "f(3)")
705 ip.run_cell_magic("timeit", "-n1 f(2)", "f(3)")
706 self.assertEqual(called, {(2,), (3,)})
706 self.assertEqual(called, {(2,), (3,)})
707
707
708 class ErrorTransformer(ast.NodeTransformer):
708 class ErrorTransformer(ast.NodeTransformer):
709 """Throws an error when it sees a number."""
709 """Throws an error when it sees a number."""
710 def visit_Num(self, node):
710 def visit_Num(self, node):
711 raise ValueError("test")
711 raise ValueError("test")
712
712
713 class TestAstTransformError(unittest.TestCase):
713 class TestAstTransformError(unittest.TestCase):
714 def test_unregistering(self):
714 def test_unregistering(self):
715 err_transformer = ErrorTransformer()
715 err_transformer = ErrorTransformer()
716 ip.ast_transformers.append(err_transformer)
716 ip.ast_transformers.append(err_transformer)
717
717
718 with tt.AssertPrints("unregister", channel='stderr'):
718 with tt.AssertPrints("unregister", channel='stderr'):
719 ip.run_cell("1 + 2")
719 ip.run_cell("1 + 2")
720
720
721 # This should have been removed.
721 # This should have been removed.
722 nt.assert_not_in(err_transformer, ip.ast_transformers)
722 nt.assert_not_in(err_transformer, ip.ast_transformers)
723
723
724
724
725 class StringRejector(ast.NodeTransformer):
725 class StringRejector(ast.NodeTransformer):
726 """Throws an InputRejected when it sees a string literal.
726 """Throws an InputRejected when it sees a string literal.
727
727
728 Used to verify that NodeTransformers can signal that a piece of code should
728 Used to verify that NodeTransformers can signal that a piece of code should
729 not be executed by throwing an InputRejected.
729 not be executed by throwing an InputRejected.
730 """
730 """
731
731
732 def visit_Str(self, node):
732 def visit_Str(self, node):
733 raise InputRejected("test")
733 raise InputRejected("test")
734
734
735
735
736 class TestAstTransformInputRejection(unittest.TestCase):
736 class TestAstTransformInputRejection(unittest.TestCase):
737
737
738 def setUp(self):
738 def setUp(self):
739 self.transformer = StringRejector()
739 self.transformer = StringRejector()
740 ip.ast_transformers.append(self.transformer)
740 ip.ast_transformers.append(self.transformer)
741
741
742 def tearDown(self):
742 def tearDown(self):
743 ip.ast_transformers.remove(self.transformer)
743 ip.ast_transformers.remove(self.transformer)
744
744
745 def test_input_rejection(self):
745 def test_input_rejection(self):
746 """Check that NodeTransformers can reject input."""
746 """Check that NodeTransformers can reject input."""
747
747
748 expect_exception_tb = tt.AssertPrints("InputRejected: test")
748 expect_exception_tb = tt.AssertPrints("InputRejected: test")
749 expect_no_cell_output = tt.AssertNotPrints("'unsafe'", suppress=False)
749 expect_no_cell_output = tt.AssertNotPrints("'unsafe'", suppress=False)
750
750
751 # Run the same check twice to verify that the transformer is not
751 # Run the same check twice to verify that the transformer is not
752 # disabled after raising.
752 # disabled after raising.
753 with expect_exception_tb, expect_no_cell_output:
753 with expect_exception_tb, expect_no_cell_output:
754 ip.run_cell("'unsafe'")
754 ip.run_cell("'unsafe'")
755
755
756 with expect_exception_tb, expect_no_cell_output:
756 with expect_exception_tb, expect_no_cell_output:
757 res = ip.run_cell("'unsafe'")
757 res = ip.run_cell("'unsafe'")
758
758
759 self.assertIsInstance(res.error_before_exec, InputRejected)
759 self.assertIsInstance(res.error_before_exec, InputRejected)
760
760
761 def test__IPYTHON__():
761 def test__IPYTHON__():
762 # This shouldn't raise a NameError, that's all
762 # This shouldn't raise a NameError, that's all
763 __IPYTHON__
763 __IPYTHON__
764
764
765
765
766 class DummyRepr(object):
766 class DummyRepr(object):
767 def __repr__(self):
767 def __repr__(self):
768 return "DummyRepr"
768 return "DummyRepr"
769
769
770 def _repr_html_(self):
770 def _repr_html_(self):
771 return "<b>dummy</b>"
771 return "<b>dummy</b>"
772
772
773 def _repr_javascript_(self):
773 def _repr_javascript_(self):
774 return "console.log('hi');", {'key': 'value'}
774 return "console.log('hi');", {'key': 'value'}
775
775
776
776
777 def test_user_variables():
777 def test_user_variables():
778 # enable all formatters
778 # enable all formatters
779 ip.display_formatter.active_types = ip.display_formatter.format_types
779 ip.display_formatter.active_types = ip.display_formatter.format_types
780
780
781 ip.user_ns['dummy'] = d = DummyRepr()
781 ip.user_ns['dummy'] = d = DummyRepr()
782 keys = {'dummy', 'doesnotexist'}
782 keys = {'dummy', 'doesnotexist'}
783 r = ip.user_expressions({ key:key for key in keys})
783 r = ip.user_expressions({ key:key for key in keys})
784
784
785 nt.assert_equal(keys, set(r.keys()))
785 nt.assert_equal(keys, set(r.keys()))
786 dummy = r['dummy']
786 dummy = r['dummy']
787 nt.assert_equal({'status', 'data', 'metadata'}, set(dummy.keys()))
787 nt.assert_equal({'status', 'data', 'metadata'}, set(dummy.keys()))
788 nt.assert_equal(dummy['status'], 'ok')
788 nt.assert_equal(dummy['status'], 'ok')
789 data = dummy['data']
789 data = dummy['data']
790 metadata = dummy['metadata']
790 metadata = dummy['metadata']
791 nt.assert_equal(data.get('text/html'), d._repr_html_())
791 nt.assert_equal(data.get('text/html'), d._repr_html_())
792 js, jsmd = d._repr_javascript_()
792 js, jsmd = d._repr_javascript_()
793 nt.assert_equal(data.get('application/javascript'), js)
793 nt.assert_equal(data.get('application/javascript'), js)
794 nt.assert_equal(metadata.get('application/javascript'), jsmd)
794 nt.assert_equal(metadata.get('application/javascript'), jsmd)
795
795
796 dne = r['doesnotexist']
796 dne = r['doesnotexist']
797 nt.assert_equal(dne['status'], 'error')
797 nt.assert_equal(dne['status'], 'error')
798 nt.assert_equal(dne['ename'], 'NameError')
798 nt.assert_equal(dne['ename'], 'NameError')
799
799
800 # back to text only
800 # back to text only
801 ip.display_formatter.active_types = ['text/plain']
801 ip.display_formatter.active_types = ['text/plain']
802
802
803 def test_user_expression():
803 def test_user_expression():
804 # enable all formatters
804 # enable all formatters
805 ip.display_formatter.active_types = ip.display_formatter.format_types
805 ip.display_formatter.active_types = ip.display_formatter.format_types
806 query = {
806 query = {
807 'a' : '1 + 2',
807 'a' : '1 + 2',
808 'b' : '1/0',
808 'b' : '1/0',
809 }
809 }
810 r = ip.user_expressions(query)
810 r = ip.user_expressions(query)
811 import pprint
811 import pprint
812 pprint.pprint(r)
812 pprint.pprint(r)
813 nt.assert_equal(set(r.keys()), set(query.keys()))
813 nt.assert_equal(set(r.keys()), set(query.keys()))
814 a = r['a']
814 a = r['a']
815 nt.assert_equal({'status', 'data', 'metadata'}, set(a.keys()))
815 nt.assert_equal({'status', 'data', 'metadata'}, set(a.keys()))
816 nt.assert_equal(a['status'], 'ok')
816 nt.assert_equal(a['status'], 'ok')
817 data = a['data']
817 data = a['data']
818 metadata = a['metadata']
818 metadata = a['metadata']
819 nt.assert_equal(data.get('text/plain'), '3')
819 nt.assert_equal(data.get('text/plain'), '3')
820
820
821 b = r['b']
821 b = r['b']
822 nt.assert_equal(b['status'], 'error')
822 nt.assert_equal(b['status'], 'error')
823 nt.assert_equal(b['ename'], 'ZeroDivisionError')
823 nt.assert_equal(b['ename'], 'ZeroDivisionError')
824
824
825 # back to text only
825 # back to text only
826 ip.display_formatter.active_types = ['text/plain']
826 ip.display_formatter.active_types = ['text/plain']
827
827
828
828
829
829
830
830
831
831
832 class TestSyntaxErrorTransformer(unittest.TestCase):
832 class TestSyntaxErrorTransformer(unittest.TestCase):
833 """Check that SyntaxError raised by an input transformer is handled by run_cell()"""
833 """Check that SyntaxError raised by an input transformer is handled by run_cell()"""
834
834
835 class SyntaxErrorTransformer(InputTransformer):
835 class SyntaxErrorTransformer(InputTransformer):
836
836
837 def push(self, line):
837 def push(self, line):
838 pos = line.find('syntaxerror')
838 pos = line.find('syntaxerror')
839 if pos >= 0:
839 if pos >= 0:
840 e = SyntaxError('input contains "syntaxerror"')
840 e = SyntaxError('input contains "syntaxerror"')
841 e.text = line
841 e.text = line
842 e.offset = pos + 1
842 e.offset = pos + 1
843 raise e
843 raise e
844 return line
844 return line
845
845
846 def reset(self):
846 def reset(self):
847 pass
847 pass
848
848
849 def setUp(self):
849 def setUp(self):
850 self.transformer = TestSyntaxErrorTransformer.SyntaxErrorTransformer()
850 self.transformer = TestSyntaxErrorTransformer.SyntaxErrorTransformer()
851 ip.input_splitter.python_line_transforms.append(self.transformer)
851 ip.input_splitter.python_line_transforms.append(self.transformer)
852 ip.input_transformer_manager.python_line_transforms.append(self.transformer)
852 ip.input_transformer_manager.python_line_transforms.append(self.transformer)
853
853
854 def tearDown(self):
854 def tearDown(self):
855 ip.input_splitter.python_line_transforms.remove(self.transformer)
855 ip.input_splitter.python_line_transforms.remove(self.transformer)
856 ip.input_transformer_manager.python_line_transforms.remove(self.transformer)
856 ip.input_transformer_manager.python_line_transforms.remove(self.transformer)
857
857
858 def test_syntaxerror_input_transformer(self):
858 def test_syntaxerror_input_transformer(self):
859 with tt.AssertPrints('1234'):
859 with tt.AssertPrints('1234'):
860 ip.run_cell('1234')
860 ip.run_cell('1234')
861 with tt.AssertPrints('SyntaxError: invalid syntax'):
861 with tt.AssertPrints('SyntaxError: invalid syntax'):
862 ip.run_cell('1 2 3') # plain python syntax error
862 ip.run_cell('1 2 3') # plain python syntax error
863 with tt.AssertPrints('SyntaxError: input contains "syntaxerror"'):
863 with tt.AssertPrints('SyntaxError: input contains "syntaxerror"'):
864 ip.run_cell('2345 # syntaxerror') # input transformer syntax error
864 ip.run_cell('2345 # syntaxerror') # input transformer syntax error
865 with tt.AssertPrints('3456'):
865 with tt.AssertPrints('3456'):
866 ip.run_cell('3456')
866 ip.run_cell('3456')
867
867
868
868
869
869
870 def test_warning_suppression():
870 def test_warning_suppression():
871 ip.run_cell("import warnings")
871 ip.run_cell("import warnings")
872 try:
872 try:
873 with tt.AssertPrints("UserWarning: asdf", channel="stderr"):
873 with tt.AssertPrints("UserWarning: asdf", channel="stderr"):
874 ip.run_cell("warnings.warn('asdf')")
874 ip.run_cell("warnings.warn('asdf')")
875 # Here's the real test -- if we run that again, we should get the
875 # Here's the real test -- if we run that again, we should get the
876 # warning again. Traditionally, each warning was only issued once per
876 # warning again. Traditionally, each warning was only issued once per
877 # IPython session (approximately), even if the user typed in new and
877 # IPython session (approximately), even if the user typed in new and
878 # different code that should have also triggered the warning, leading
878 # different code that should have also triggered the warning, leading
879 # to much confusion.
879 # to much confusion.
880 with tt.AssertPrints("UserWarning: asdf", channel="stderr"):
880 with tt.AssertPrints("UserWarning: asdf", channel="stderr"):
881 ip.run_cell("warnings.warn('asdf')")
881 ip.run_cell("warnings.warn('asdf')")
882 finally:
882 finally:
883 ip.run_cell("del warnings")
883 ip.run_cell("del warnings")
884
884
885
885
886 def test_deprecation_warning():
886 def test_deprecation_warning():
887 ip.run_cell("""
887 ip.run_cell("""
888 import warnings
888 import warnings
889 def wrn():
889 def wrn():
890 warnings.warn(
890 warnings.warn(
891 "I AM A WARNING",
891 "I AM A WARNING",
892 DeprecationWarning
892 DeprecationWarning
893 )
893 )
894 """)
894 """)
895 try:
895 try:
896 with tt.AssertPrints("I AM A WARNING", channel="stderr"):
896 with tt.AssertPrints("I AM A WARNING", channel="stderr"):
897 ip.run_cell("wrn()")
897 ip.run_cell("wrn()")
898 finally:
898 finally:
899 ip.run_cell("del warnings")
899 ip.run_cell("del warnings")
900 ip.run_cell("del wrn")
900 ip.run_cell("del wrn")
901
901
902
902
903 class TestImportNoDeprecate(tt.TempFileMixin):
903 class TestImportNoDeprecate(tt.TempFileMixin):
904
904
905 def setup(self):
905 def setup(self):
906 """Make a valid python temp file."""
906 """Make a valid python temp file."""
907 self.mktmp("""
907 self.mktmp("""
908 import warnings
908 import warnings
909 def wrn():
909 def wrn():
910 warnings.warn(
910 warnings.warn(
911 "I AM A WARNING",
911 "I AM A WARNING",
912 DeprecationWarning
912 DeprecationWarning
913 )
913 )
914 """)
914 """)
915
915
916 def test_no_dep(self):
916 def test_no_dep(self):
917 """
917 """
918 No deprecation warning should be raised from imported functions
918 No deprecation warning should be raised from imported functions
919 """
919 """
920 ip.run_cell("from {} import wrn".format(self.fname))
920 ip.run_cell("from {} import wrn".format(self.fname))
921
921
922 with tt.AssertNotPrints("I AM A WARNING"):
922 with tt.AssertNotPrints("I AM A WARNING"):
923 ip.run_cell("wrn()")
923 ip.run_cell("wrn()")
924 ip.run_cell("del wrn")
924 ip.run_cell("del wrn")
925
926
927 def test_custom_exc_count():
928 hook = mock.Mock(return_value=None)
929 ip.set_custom_exc((SyntaxError,), hook)
930 before = ip.execution_count
931 ip.run_cell("def foo()", store_history=True)
932 nt.assert_equal(hook.call_count, 1)
933 nt.assert_equal(ip.execution_count, before + 1)
General Comments 0
You need to be logged in to leave comments. Login now