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