##// END OF EJS Templates
Restore Fernando's fix for the __del__ methods on exit bug.
Thomas Kluyver -
Show More
@@ -1,2524 +1,2539 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-2010 The IPython Development Team
7 # Copyright (C) 2008-2010 The IPython Development Team
8 #
8 #
9 # Distributed under the terms of the BSD License. The full license is in
9 # Distributed under the terms of the BSD License. The full license is in
10 # the file COPYING, distributed as part of this software.
10 # the file COPYING, distributed as part of this software.
11 #-----------------------------------------------------------------------------
11 #-----------------------------------------------------------------------------
12
12
13 #-----------------------------------------------------------------------------
13 #-----------------------------------------------------------------------------
14 # Imports
14 # Imports
15 #-----------------------------------------------------------------------------
15 #-----------------------------------------------------------------------------
16
16
17 from __future__ import with_statement
17 from __future__ import with_statement
18 from __future__ import absolute_import
18 from __future__ import absolute_import
19
19
20 import __builtin__
20 import __builtin__
21 import __future__
21 import __future__
22 import abc
22 import abc
23 import atexit
23 import atexit
24 import codeop
24 import codeop
25 import exceptions
25 import exceptions
26 import new
26 import new
27 import os
27 import os
28 import re
28 import re
29 import string
29 import string
30 import sys
30 import sys
31 import tempfile
31 import tempfile
32 from contextlib import nested
32 from contextlib import nested
33
33
34 from IPython.config.configurable import Configurable
34 from IPython.config.configurable import Configurable
35 from IPython.core import debugger, oinspect
35 from IPython.core import debugger, oinspect
36 from IPython.core import history as ipcorehist
36 from IPython.core import history as ipcorehist
37 from IPython.core import page
37 from IPython.core import page
38 from IPython.core import prefilter
38 from IPython.core import prefilter
39 from IPython.core import shadowns
39 from IPython.core import shadowns
40 from IPython.core import ultratb
40 from IPython.core import ultratb
41 from IPython.core.alias import AliasManager
41 from IPython.core.alias import AliasManager
42 from IPython.core.builtin_trap import BuiltinTrap
42 from IPython.core.builtin_trap import BuiltinTrap
43 from IPython.core.display_trap import DisplayTrap
43 from IPython.core.display_trap import DisplayTrap
44 from IPython.core.displayhook import DisplayHook
44 from IPython.core.displayhook import DisplayHook
45 from IPython.core.error import TryNext, UsageError
45 from IPython.core.error import TryNext, UsageError
46 from IPython.core.extensions import ExtensionManager
46 from IPython.core.extensions import ExtensionManager
47 from IPython.core.fakemodule import FakeModule, init_fakemod_dict
47 from IPython.core.fakemodule import FakeModule, init_fakemod_dict
48 from IPython.core.history import HistoryManager
48 from IPython.core.history import HistoryManager
49 from IPython.core.inputlist import InputList
49 from IPython.core.inputlist import InputList
50 from IPython.core.inputsplitter import IPythonInputSplitter
50 from IPython.core.inputsplitter import IPythonInputSplitter
51 from IPython.core.logger import Logger
51 from IPython.core.logger import Logger
52 from IPython.core.magic import Magic
52 from IPython.core.magic import Magic
53 from IPython.core.payload import PayloadManager
53 from IPython.core.payload import PayloadManager
54 from IPython.core.plugin import PluginManager
54 from IPython.core.plugin import PluginManager
55 from IPython.core.prefilter import PrefilterManager, ESC_MAGIC
55 from IPython.core.prefilter import PrefilterManager, ESC_MAGIC
56 from IPython.external.Itpl import ItplNS
56 from IPython.external.Itpl import ItplNS
57 from IPython.utils import PyColorize
57 from IPython.utils import PyColorize
58 from IPython.utils import io
58 from IPython.utils import io
59 from IPython.utils import pickleshare
59 from IPython.utils import pickleshare
60 from IPython.utils.doctestreload import doctest_reload
60 from IPython.utils.doctestreload import doctest_reload
61 from IPython.utils.io import ask_yes_no, rprint
61 from IPython.utils.io import ask_yes_no, rprint
62 from IPython.utils.ipstruct import Struct
62 from IPython.utils.ipstruct import Struct
63 from IPython.utils.path import get_home_dir, get_ipython_dir, HomeDirError
63 from IPython.utils.path import get_home_dir, get_ipython_dir, HomeDirError
64 from IPython.utils.process import system, getoutput
64 from IPython.utils.process import system, getoutput
65 from IPython.utils.strdispatch import StrDispatch
65 from IPython.utils.strdispatch import StrDispatch
66 from IPython.utils.syspathcontext import prepended_to_syspath
66 from IPython.utils.syspathcontext import prepended_to_syspath
67 from IPython.utils.text import num_ini_spaces, format_screen, LSString, SList
67 from IPython.utils.text import num_ini_spaces, format_screen, LSString, SList
68 from IPython.utils.traitlets import (Int, Str, CBool, CaselessStrEnum, Enum,
68 from IPython.utils.traitlets import (Int, Str, CBool, CaselessStrEnum, Enum,
69 List, Unicode, Instance, Type)
69 List, Unicode, Instance, Type)
70 from IPython.utils.warn import warn, error, fatal
70 from IPython.utils.warn import warn, error, fatal
71 import IPython.core.hooks
71 import IPython.core.hooks
72
72
73 #-----------------------------------------------------------------------------
73 #-----------------------------------------------------------------------------
74 # Globals
74 # Globals
75 #-----------------------------------------------------------------------------
75 #-----------------------------------------------------------------------------
76
76
77 # compiled regexps for autoindent management
77 # compiled regexps for autoindent management
78 dedent_re = re.compile(r'^\s+raise|^\s+return|^\s+pass')
78 dedent_re = re.compile(r'^\s+raise|^\s+return|^\s+pass')
79
79
80 #-----------------------------------------------------------------------------
80 #-----------------------------------------------------------------------------
81 # Utilities
81 # Utilities
82 #-----------------------------------------------------------------------------
82 #-----------------------------------------------------------------------------
83
83
84 # store the builtin raw_input globally, and use this always, in case user code
84 # store the builtin raw_input globally, and use this always, in case user code
85 # overwrites it (like wx.py.PyShell does)
85 # overwrites it (like wx.py.PyShell does)
86 raw_input_original = raw_input
86 raw_input_original = raw_input
87
87
88 def softspace(file, newvalue):
88 def softspace(file, newvalue):
89 """Copied from code.py, to remove the dependency"""
89 """Copied from code.py, to remove the dependency"""
90
90
91 oldvalue = 0
91 oldvalue = 0
92 try:
92 try:
93 oldvalue = file.softspace
93 oldvalue = file.softspace
94 except AttributeError:
94 except AttributeError:
95 pass
95 pass
96 try:
96 try:
97 file.softspace = newvalue
97 file.softspace = newvalue
98 except (AttributeError, TypeError):
98 except (AttributeError, TypeError):
99 # "attribute-less object" or "read-only attributes"
99 # "attribute-less object" or "read-only attributes"
100 pass
100 pass
101 return oldvalue
101 return oldvalue
102
102
103
103
104 def no_op(*a, **kw): pass
104 def no_op(*a, **kw): pass
105
105
106 class SpaceInInput(exceptions.Exception): pass
106 class SpaceInInput(exceptions.Exception): pass
107
107
108 class Bunch: pass
108 class Bunch: pass
109
109
110
110
111 def get_default_colors():
111 def get_default_colors():
112 if sys.platform=='darwin':
112 if sys.platform=='darwin':
113 return "LightBG"
113 return "LightBG"
114 elif os.name=='nt':
114 elif os.name=='nt':
115 return 'Linux'
115 return 'Linux'
116 else:
116 else:
117 return 'Linux'
117 return 'Linux'
118
118
119
119
120 class SeparateStr(Str):
120 class SeparateStr(Str):
121 """A Str subclass to validate separate_in, separate_out, etc.
121 """A Str subclass to validate separate_in, separate_out, etc.
122
122
123 This is a Str based trait that converts '0'->'' and '\\n'->'\n'.
123 This is a Str based trait that converts '0'->'' and '\\n'->'\n'.
124 """
124 """
125
125
126 def validate(self, obj, value):
126 def validate(self, obj, value):
127 if value == '0': value = ''
127 if value == '0': value = ''
128 value = value.replace('\\n','\n')
128 value = value.replace('\\n','\n')
129 return super(SeparateStr, self).validate(obj, value)
129 return super(SeparateStr, self).validate(obj, value)
130
130
131 class MultipleInstanceError(Exception):
131 class MultipleInstanceError(Exception):
132 pass
132 pass
133
133
134
134
135 #-----------------------------------------------------------------------------
135 #-----------------------------------------------------------------------------
136 # Main IPython class
136 # Main IPython class
137 #-----------------------------------------------------------------------------
137 #-----------------------------------------------------------------------------
138
138
139
139
140 class InteractiveShell(Configurable, Magic):
140 class InteractiveShell(Configurable, Magic):
141 """An enhanced, interactive shell for Python."""
141 """An enhanced, interactive shell for Python."""
142
142
143 _instance = None
143 _instance = None
144 autocall = Enum((0,1,2), default_value=1, config=True)
144 autocall = Enum((0,1,2), default_value=1, config=True)
145 # TODO: remove all autoindent logic and put into frontends.
145 # TODO: remove all autoindent logic and put into frontends.
146 # We can't do this yet because even runlines uses the autoindent.
146 # We can't do this yet because even runlines uses the autoindent.
147 autoindent = CBool(True, config=True)
147 autoindent = CBool(True, config=True)
148 automagic = CBool(True, config=True)
148 automagic = CBool(True, config=True)
149 cache_size = Int(1000, config=True)
149 cache_size = Int(1000, config=True)
150 color_info = CBool(True, config=True)
150 color_info = CBool(True, config=True)
151 colors = CaselessStrEnum(('NoColor','LightBG','Linux'),
151 colors = CaselessStrEnum(('NoColor','LightBG','Linux'),
152 default_value=get_default_colors(), config=True)
152 default_value=get_default_colors(), config=True)
153 debug = CBool(False, config=True)
153 debug = CBool(False, config=True)
154 deep_reload = CBool(False, config=True)
154 deep_reload = CBool(False, config=True)
155 displayhook_class = Type(DisplayHook)
155 displayhook_class = Type(DisplayHook)
156 exit_now = CBool(False)
156 exit_now = CBool(False)
157 # Monotonically increasing execution counter
157 # Monotonically increasing execution counter
158 execution_count = Int(1)
158 execution_count = Int(1)
159 filename = Str("<ipython console>")
159 filename = Str("<ipython console>")
160 ipython_dir= Unicode('', config=True) # Set to get_ipython_dir() in __init__
160 ipython_dir= Unicode('', config=True) # Set to get_ipython_dir() in __init__
161
161
162 # Input splitter, to split entire cells of input into either individual
162 # Input splitter, to split entire cells of input into either individual
163 # interactive statements or whole blocks.
163 # interactive statements or whole blocks.
164 input_splitter = Instance('IPython.core.inputsplitter.IPythonInputSplitter',
164 input_splitter = Instance('IPython.core.inputsplitter.IPythonInputSplitter',
165 (), {})
165 (), {})
166 logstart = CBool(False, config=True)
166 logstart = CBool(False, config=True)
167 logfile = Str('', config=True)
167 logfile = Str('', config=True)
168 logappend = Str('', config=True)
168 logappend = Str('', config=True)
169 object_info_string_level = Enum((0,1,2), default_value=0,
169 object_info_string_level = Enum((0,1,2), default_value=0,
170 config=True)
170 config=True)
171 pdb = CBool(False, config=True)
171 pdb = CBool(False, config=True)
172
172
173 pprint = CBool(True, config=True)
173 pprint = CBool(True, config=True)
174 profile = Str('', config=True)
174 profile = Str('', config=True)
175 prompt_in1 = Str('In [\\#]: ', config=True)
175 prompt_in1 = Str('In [\\#]: ', config=True)
176 prompt_in2 = Str(' .\\D.: ', config=True)
176 prompt_in2 = Str(' .\\D.: ', config=True)
177 prompt_out = Str('Out[\\#]: ', config=True)
177 prompt_out = Str('Out[\\#]: ', config=True)
178 prompts_pad_left = CBool(True, config=True)
178 prompts_pad_left = CBool(True, config=True)
179 quiet = CBool(False, config=True)
179 quiet = CBool(False, config=True)
180
180
181 # The readline stuff will eventually be moved to the terminal subclass
181 # The readline stuff will eventually be moved to the terminal subclass
182 # but for now, we can't do that as readline is welded in everywhere.
182 # but for now, we can't do that as readline is welded in everywhere.
183 readline_use = CBool(True, config=True)
183 readline_use = CBool(True, config=True)
184 readline_merge_completions = CBool(True, config=True)
184 readline_merge_completions = CBool(True, config=True)
185 readline_omit__names = Enum((0,1,2), default_value=0, config=True)
185 readline_omit__names = Enum((0,1,2), default_value=0, config=True)
186 readline_remove_delims = Str('-/~', config=True)
186 readline_remove_delims = Str('-/~', config=True)
187 readline_parse_and_bind = List([
187 readline_parse_and_bind = List([
188 'tab: complete',
188 'tab: complete',
189 '"\C-l": clear-screen',
189 '"\C-l": clear-screen',
190 'set show-all-if-ambiguous on',
190 'set show-all-if-ambiguous on',
191 '"\C-o": tab-insert',
191 '"\C-o": tab-insert',
192 '"\M-i": " "',
192 '"\M-i": " "',
193 '"\M-o": "\d\d\d\d"',
193 '"\M-o": "\d\d\d\d"',
194 '"\M-I": "\d\d\d\d"',
194 '"\M-I": "\d\d\d\d"',
195 '"\C-r": reverse-search-history',
195 '"\C-r": reverse-search-history',
196 '"\C-s": forward-search-history',
196 '"\C-s": forward-search-history',
197 '"\C-p": history-search-backward',
197 '"\C-p": history-search-backward',
198 '"\C-n": history-search-forward',
198 '"\C-n": history-search-forward',
199 '"\e[A": history-search-backward',
199 '"\e[A": history-search-backward',
200 '"\e[B": history-search-forward',
200 '"\e[B": history-search-forward',
201 '"\C-k": kill-line',
201 '"\C-k": kill-line',
202 '"\C-u": unix-line-discard',
202 '"\C-u": unix-line-discard',
203 ], allow_none=False, config=True)
203 ], allow_none=False, config=True)
204
204
205 # TODO: this part of prompt management should be moved to the frontends.
205 # TODO: this part of prompt management should be moved to the frontends.
206 # Use custom TraitTypes that convert '0'->'' and '\\n'->'\n'
206 # Use custom TraitTypes that convert '0'->'' and '\\n'->'\n'
207 separate_in = SeparateStr('\n', config=True)
207 separate_in = SeparateStr('\n', config=True)
208 separate_out = SeparateStr('', config=True)
208 separate_out = SeparateStr('', config=True)
209 separate_out2 = SeparateStr('', config=True)
209 separate_out2 = SeparateStr('', config=True)
210 wildcards_case_sensitive = CBool(True, config=True)
210 wildcards_case_sensitive = CBool(True, config=True)
211 xmode = CaselessStrEnum(('Context','Plain', 'Verbose'),
211 xmode = CaselessStrEnum(('Context','Plain', 'Verbose'),
212 default_value='Context', config=True)
212 default_value='Context', config=True)
213
213
214 # Subcomponents of InteractiveShell
214 # Subcomponents of InteractiveShell
215 alias_manager = Instance('IPython.core.alias.AliasManager')
215 alias_manager = Instance('IPython.core.alias.AliasManager')
216 prefilter_manager = Instance('IPython.core.prefilter.PrefilterManager')
216 prefilter_manager = Instance('IPython.core.prefilter.PrefilterManager')
217 builtin_trap = Instance('IPython.core.builtin_trap.BuiltinTrap')
217 builtin_trap = Instance('IPython.core.builtin_trap.BuiltinTrap')
218 display_trap = Instance('IPython.core.display_trap.DisplayTrap')
218 display_trap = Instance('IPython.core.display_trap.DisplayTrap')
219 extension_manager = Instance('IPython.core.extensions.ExtensionManager')
219 extension_manager = Instance('IPython.core.extensions.ExtensionManager')
220 plugin_manager = Instance('IPython.core.plugin.PluginManager')
220 plugin_manager = Instance('IPython.core.plugin.PluginManager')
221 payload_manager = Instance('IPython.core.payload.PayloadManager')
221 payload_manager = Instance('IPython.core.payload.PayloadManager')
222 history_manager = Instance('IPython.core.history.HistoryManager')
222 history_manager = Instance('IPython.core.history.HistoryManager')
223
223
224 # Private interface
224 # Private interface
225 _post_execute = set()
225 _post_execute = set()
226
226
227 def __init__(self, config=None, ipython_dir=None,
227 def __init__(self, config=None, ipython_dir=None,
228 user_ns=None, user_global_ns=None,
228 user_ns=None, user_global_ns=None,
229 custom_exceptions=((), None)):
229 custom_exceptions=((), None)):
230
230
231 # This is where traits with a config_key argument are updated
231 # This is where traits with a config_key argument are updated
232 # from the values on config.
232 # from the values on config.
233 super(InteractiveShell, self).__init__(config=config)
233 super(InteractiveShell, self).__init__(config=config)
234
234
235 # These are relatively independent and stateless
235 # These are relatively independent and stateless
236 self.init_ipython_dir(ipython_dir)
236 self.init_ipython_dir(ipython_dir)
237 self.init_instance_attrs()
237 self.init_instance_attrs()
238 self.init_environment()
238 self.init_environment()
239
239
240 # Create namespaces (user_ns, user_global_ns, etc.)
240 # Create namespaces (user_ns, user_global_ns, etc.)
241 self.init_create_namespaces(user_ns, user_global_ns)
241 self.init_create_namespaces(user_ns, user_global_ns)
242 # This has to be done after init_create_namespaces because it uses
242 # This has to be done after init_create_namespaces because it uses
243 # something in self.user_ns, but before init_sys_modules, which
243 # something in self.user_ns, but before init_sys_modules, which
244 # is the first thing to modify sys.
244 # is the first thing to modify sys.
245 # TODO: When we override sys.stdout and sys.stderr before this class
245 # TODO: When we override sys.stdout and sys.stderr before this class
246 # is created, we are saving the overridden ones here. Not sure if this
246 # is created, we are saving the overridden ones here. Not sure if this
247 # is what we want to do.
247 # is what we want to do.
248 self.save_sys_module_state()
248 self.save_sys_module_state()
249 self.init_sys_modules()
249 self.init_sys_modules()
250
250
251 self.init_history()
251 self.init_history()
252 self.init_encoding()
252 self.init_encoding()
253 self.init_prefilter()
253 self.init_prefilter()
254
254
255 Magic.__init__(self, self)
255 Magic.__init__(self, self)
256
256
257 self.init_syntax_highlighting()
257 self.init_syntax_highlighting()
258 self.init_hooks()
258 self.init_hooks()
259 self.init_pushd_popd_magic()
259 self.init_pushd_popd_magic()
260 # self.init_traceback_handlers use to be here, but we moved it below
260 # self.init_traceback_handlers use to be here, but we moved it below
261 # because it and init_io have to come after init_readline.
261 # because it and init_io have to come after init_readline.
262 self.init_user_ns()
262 self.init_user_ns()
263 self.init_logger()
263 self.init_logger()
264 self.init_alias()
264 self.init_alias()
265 self.init_builtins()
265 self.init_builtins()
266
266
267 # pre_config_initialization
267 # pre_config_initialization
268
268
269 # The next section should contain everything that was in ipmaker.
269 # The next section should contain everything that was in ipmaker.
270 self.init_logstart()
270 self.init_logstart()
271
271
272 # The following was in post_config_initialization
272 # The following was in post_config_initialization
273 self.init_inspector()
273 self.init_inspector()
274 # init_readline() must come before init_io(), because init_io uses
274 # init_readline() must come before init_io(), because init_io uses
275 # readline related things.
275 # readline related things.
276 self.init_readline()
276 self.init_readline()
277 # init_completer must come after init_readline, because it needs to
277 # init_completer must come after init_readline, because it needs to
278 # know whether readline is present or not system-wide to configure the
278 # know whether readline is present or not system-wide to configure the
279 # completers, since the completion machinery can now operate
279 # completers, since the completion machinery can now operate
280 # independently of readline (e.g. over the network)
280 # independently of readline (e.g. over the network)
281 self.init_completer()
281 self.init_completer()
282 # TODO: init_io() needs to happen before init_traceback handlers
282 # TODO: init_io() needs to happen before init_traceback handlers
283 # because the traceback handlers hardcode the stdout/stderr streams.
283 # because the traceback handlers hardcode the stdout/stderr streams.
284 # This logic in in debugger.Pdb and should eventually be changed.
284 # This logic in in debugger.Pdb and should eventually be changed.
285 self.init_io()
285 self.init_io()
286 self.init_traceback_handlers(custom_exceptions)
286 self.init_traceback_handlers(custom_exceptions)
287 self.init_prompts()
287 self.init_prompts()
288 self.init_displayhook()
288 self.init_displayhook()
289 self.init_reload_doctest()
289 self.init_reload_doctest()
290 self.init_magics()
290 self.init_magics()
291 self.init_pdb()
291 self.init_pdb()
292 self.init_extension_manager()
292 self.init_extension_manager()
293 self.init_plugin_manager()
293 self.init_plugin_manager()
294 self.init_payload()
294 self.init_payload()
295 self.hooks.late_startup_hook()
295 self.hooks.late_startup_hook()
296 atexit.register(self.atexit_operations)
296 atexit.register(self.atexit_operations)
297
297
298 @classmethod
298 @classmethod
299 def instance(cls, *args, **kwargs):
299 def instance(cls, *args, **kwargs):
300 """Returns a global InteractiveShell instance."""
300 """Returns a global InteractiveShell instance."""
301 if cls._instance is None:
301 if cls._instance is None:
302 inst = cls(*args, **kwargs)
302 inst = cls(*args, **kwargs)
303 # Now make sure that the instance will also be returned by
303 # Now make sure that the instance will also be returned by
304 # the subclasses instance attribute.
304 # the subclasses instance attribute.
305 for subclass in cls.mro():
305 for subclass in cls.mro():
306 if issubclass(cls, subclass) and \
306 if issubclass(cls, subclass) and \
307 issubclass(subclass, InteractiveShell):
307 issubclass(subclass, InteractiveShell):
308 subclass._instance = inst
308 subclass._instance = inst
309 else:
309 else:
310 break
310 break
311 if isinstance(cls._instance, cls):
311 if isinstance(cls._instance, cls):
312 return cls._instance
312 return cls._instance
313 else:
313 else:
314 raise MultipleInstanceError(
314 raise MultipleInstanceError(
315 'Multiple incompatible subclass instances of '
315 'Multiple incompatible subclass instances of '
316 'InteractiveShell are being created.'
316 'InteractiveShell are being created.'
317 )
317 )
318
318
319 @classmethod
319 @classmethod
320 def initialized(cls):
320 def initialized(cls):
321 return hasattr(cls, "_instance")
321 return hasattr(cls, "_instance")
322
322
323 def get_ipython(self):
323 def get_ipython(self):
324 """Return the currently running IPython instance."""
324 """Return the currently running IPython instance."""
325 return self
325 return self
326
326
327 #-------------------------------------------------------------------------
327 #-------------------------------------------------------------------------
328 # Trait changed handlers
328 # Trait changed handlers
329 #-------------------------------------------------------------------------
329 #-------------------------------------------------------------------------
330
330
331 def _ipython_dir_changed(self, name, new):
331 def _ipython_dir_changed(self, name, new):
332 if not os.path.isdir(new):
332 if not os.path.isdir(new):
333 os.makedirs(new, mode = 0777)
333 os.makedirs(new, mode = 0777)
334
334
335 def set_autoindent(self,value=None):
335 def set_autoindent(self,value=None):
336 """Set the autoindent flag, checking for readline support.
336 """Set the autoindent flag, checking for readline support.
337
337
338 If called with no arguments, it acts as a toggle."""
338 If called with no arguments, it acts as a toggle."""
339
339
340 if not self.has_readline:
340 if not self.has_readline:
341 if os.name == 'posix':
341 if os.name == 'posix':
342 warn("The auto-indent feature requires the readline library")
342 warn("The auto-indent feature requires the readline library")
343 self.autoindent = 0
343 self.autoindent = 0
344 return
344 return
345 if value is None:
345 if value is None:
346 self.autoindent = not self.autoindent
346 self.autoindent = not self.autoindent
347 else:
347 else:
348 self.autoindent = value
348 self.autoindent = value
349
349
350 #-------------------------------------------------------------------------
350 #-------------------------------------------------------------------------
351 # init_* methods called by __init__
351 # init_* methods called by __init__
352 #-------------------------------------------------------------------------
352 #-------------------------------------------------------------------------
353
353
354 def init_ipython_dir(self, ipython_dir):
354 def init_ipython_dir(self, ipython_dir):
355 if ipython_dir is not None:
355 if ipython_dir is not None:
356 self.ipython_dir = ipython_dir
356 self.ipython_dir = ipython_dir
357 self.config.Global.ipython_dir = self.ipython_dir
357 self.config.Global.ipython_dir = self.ipython_dir
358 return
358 return
359
359
360 if hasattr(self.config.Global, 'ipython_dir'):
360 if hasattr(self.config.Global, 'ipython_dir'):
361 self.ipython_dir = self.config.Global.ipython_dir
361 self.ipython_dir = self.config.Global.ipython_dir
362 else:
362 else:
363 self.ipython_dir = get_ipython_dir()
363 self.ipython_dir = get_ipython_dir()
364
364
365 # All children can just read this
365 # All children can just read this
366 self.config.Global.ipython_dir = self.ipython_dir
366 self.config.Global.ipython_dir = self.ipython_dir
367
367
368 def init_instance_attrs(self):
368 def init_instance_attrs(self):
369 self.more = False
369 self.more = False
370
370
371 # command compiler
371 # command compiler
372 self.compile = codeop.CommandCompiler()
372 self.compile = codeop.CommandCompiler()
373
373
374 # User input buffers
374 # User input buffers
375 # NOTE: these variables are slated for full removal, once we are 100%
375 # NOTE: these variables are slated for full removal, once we are 100%
376 # sure that the new execution logic is solid. We will delte runlines,
376 # sure that the new execution logic is solid. We will delte runlines,
377 # push_line and these buffers, as all input will be managed by the
377 # push_line and these buffers, as all input will be managed by the
378 # frontends via an inputsplitter instance.
378 # frontends via an inputsplitter instance.
379 self.buffer = []
379 self.buffer = []
380 self.buffer_raw = []
380 self.buffer_raw = []
381
381
382 # Make an empty namespace, which extension writers can rely on both
382 # Make an empty namespace, which extension writers can rely on both
383 # existing and NEVER being used by ipython itself. This gives them a
383 # existing and NEVER being used by ipython itself. This gives them a
384 # convenient location for storing additional information and state
384 # convenient location for storing additional information and state
385 # their extensions may require, without fear of collisions with other
385 # their extensions may require, without fear of collisions with other
386 # ipython names that may develop later.
386 # ipython names that may develop later.
387 self.meta = Struct()
387 self.meta = Struct()
388
388
389 # Object variable to store code object waiting execution. This is
389 # Object variable to store code object waiting execution. This is
390 # used mainly by the multithreaded shells, but it can come in handy in
390 # used mainly by the multithreaded shells, but it can come in handy in
391 # other situations. No need to use a Queue here, since it's a single
391 # other situations. No need to use a Queue here, since it's a single
392 # item which gets cleared once run.
392 # item which gets cleared once run.
393 self.code_to_run = None
393 self.code_to_run = None
394
394
395 # Temporary files used for various purposes. Deleted at exit.
395 # Temporary files used for various purposes. Deleted at exit.
396 self.tempfiles = []
396 self.tempfiles = []
397
397
398 # Keep track of readline usage (later set by init_readline)
398 # Keep track of readline usage (later set by init_readline)
399 self.has_readline = False
399 self.has_readline = False
400
400
401 # keep track of where we started running (mainly for crash post-mortem)
401 # keep track of where we started running (mainly for crash post-mortem)
402 # This is not being used anywhere currently.
402 # This is not being used anywhere currently.
403 self.starting_dir = os.getcwd()
403 self.starting_dir = os.getcwd()
404
404
405 # Indentation management
405 # Indentation management
406 self.indent_current_nsp = 0
406 self.indent_current_nsp = 0
407
407
408 def init_environment(self):
408 def init_environment(self):
409 """Any changes we need to make to the user's environment."""
409 """Any changes we need to make to the user's environment."""
410 pass
410 pass
411
411
412 def init_encoding(self):
412 def init_encoding(self):
413 # Get system encoding at startup time. Certain terminals (like Emacs
413 # Get system encoding at startup time. Certain terminals (like Emacs
414 # under Win32 have it set to None, and we need to have a known valid
414 # under Win32 have it set to None, and we need to have a known valid
415 # encoding to use in the raw_input() method
415 # encoding to use in the raw_input() method
416 try:
416 try:
417 self.stdin_encoding = sys.stdin.encoding or 'ascii'
417 self.stdin_encoding = sys.stdin.encoding or 'ascii'
418 except AttributeError:
418 except AttributeError:
419 self.stdin_encoding = 'ascii'
419 self.stdin_encoding = 'ascii'
420
420
421 def init_syntax_highlighting(self):
421 def init_syntax_highlighting(self):
422 # Python source parser/formatter for syntax highlighting
422 # Python source parser/formatter for syntax highlighting
423 pyformat = PyColorize.Parser().format
423 pyformat = PyColorize.Parser().format
424 self.pycolorize = lambda src: pyformat(src,'str',self.colors)
424 self.pycolorize = lambda src: pyformat(src,'str',self.colors)
425
425
426 def init_pushd_popd_magic(self):
426 def init_pushd_popd_magic(self):
427 # for pushd/popd management
427 # for pushd/popd management
428 try:
428 try:
429 self.home_dir = get_home_dir()
429 self.home_dir = get_home_dir()
430 except HomeDirError, msg:
430 except HomeDirError, msg:
431 fatal(msg)
431 fatal(msg)
432
432
433 self.dir_stack = []
433 self.dir_stack = []
434
434
435 def init_logger(self):
435 def init_logger(self):
436 self.logger = Logger(self.home_dir, logfname='ipython_log.py',
436 self.logger = Logger(self.home_dir, logfname='ipython_log.py',
437 logmode='rotate')
437 logmode='rotate')
438
438
439 def init_logstart(self):
439 def init_logstart(self):
440 """Initialize logging in case it was requested at the command line.
440 """Initialize logging in case it was requested at the command line.
441 """
441 """
442 if self.logappend:
442 if self.logappend:
443 self.magic_logstart(self.logappend + ' append')
443 self.magic_logstart(self.logappend + ' append')
444 elif self.logfile:
444 elif self.logfile:
445 self.magic_logstart(self.logfile)
445 self.magic_logstart(self.logfile)
446 elif self.logstart:
446 elif self.logstart:
447 self.magic_logstart()
447 self.magic_logstart()
448
448
449 def init_builtins(self):
449 def init_builtins(self):
450 self.builtin_trap = BuiltinTrap(shell=self)
450 self.builtin_trap = BuiltinTrap(shell=self)
451
451
452 def init_inspector(self):
452 def init_inspector(self):
453 # Object inspector
453 # Object inspector
454 self.inspector = oinspect.Inspector(oinspect.InspectColors,
454 self.inspector = oinspect.Inspector(oinspect.InspectColors,
455 PyColorize.ANSICodeColors,
455 PyColorize.ANSICodeColors,
456 'NoColor',
456 'NoColor',
457 self.object_info_string_level)
457 self.object_info_string_level)
458
458
459 def init_io(self):
459 def init_io(self):
460 # This will just use sys.stdout and sys.stderr. If you want to
460 # This will just use sys.stdout and sys.stderr. If you want to
461 # override sys.stdout and sys.stderr themselves, you need to do that
461 # override sys.stdout and sys.stderr themselves, you need to do that
462 # *before* instantiating this class, because Term holds onto
462 # *before* instantiating this class, because Term holds onto
463 # references to the underlying streams.
463 # references to the underlying streams.
464 if sys.platform == 'win32' and self.has_readline:
464 if sys.platform == 'win32' and self.has_readline:
465 Term = io.IOTerm(cout=self.readline._outputfile,
465 Term = io.IOTerm(cout=self.readline._outputfile,
466 cerr=self.readline._outputfile)
466 cerr=self.readline._outputfile)
467 else:
467 else:
468 Term = io.IOTerm()
468 Term = io.IOTerm()
469 io.Term = Term
469 io.Term = Term
470
470
471 def init_prompts(self):
471 def init_prompts(self):
472 # TODO: This is a pass for now because the prompts are managed inside
472 # TODO: This is a pass for now because the prompts are managed inside
473 # the DisplayHook. Once there is a separate prompt manager, this
473 # the DisplayHook. Once there is a separate prompt manager, this
474 # will initialize that object and all prompt related information.
474 # will initialize that object and all prompt related information.
475 pass
475 pass
476
476
477 def init_displayhook(self):
477 def init_displayhook(self):
478 # Initialize displayhook, set in/out prompts and printing system
478 # Initialize displayhook, set in/out prompts and printing system
479 self.displayhook = self.displayhook_class(
479 self.displayhook = self.displayhook_class(
480 shell=self,
480 shell=self,
481 cache_size=self.cache_size,
481 cache_size=self.cache_size,
482 input_sep = self.separate_in,
482 input_sep = self.separate_in,
483 output_sep = self.separate_out,
483 output_sep = self.separate_out,
484 output_sep2 = self.separate_out2,
484 output_sep2 = self.separate_out2,
485 ps1 = self.prompt_in1,
485 ps1 = self.prompt_in1,
486 ps2 = self.prompt_in2,
486 ps2 = self.prompt_in2,
487 ps_out = self.prompt_out,
487 ps_out = self.prompt_out,
488 pad_left = self.prompts_pad_left
488 pad_left = self.prompts_pad_left
489 )
489 )
490 # This is a context manager that installs/revmoes the displayhook at
490 # This is a context manager that installs/revmoes the displayhook at
491 # the appropriate time.
491 # the appropriate time.
492 self.display_trap = DisplayTrap(hook=self.displayhook)
492 self.display_trap = DisplayTrap(hook=self.displayhook)
493
493
494 def init_reload_doctest(self):
494 def init_reload_doctest(self):
495 # Do a proper resetting of doctest, including the necessary displayhook
495 # Do a proper resetting of doctest, including the necessary displayhook
496 # monkeypatching
496 # monkeypatching
497 try:
497 try:
498 doctest_reload()
498 doctest_reload()
499 except ImportError:
499 except ImportError:
500 warn("doctest module does not exist.")
500 warn("doctest module does not exist.")
501
501
502 #-------------------------------------------------------------------------
502 #-------------------------------------------------------------------------
503 # Things related to injections into the sys module
503 # Things related to injections into the sys module
504 #-------------------------------------------------------------------------
504 #-------------------------------------------------------------------------
505
505
506 def save_sys_module_state(self):
506 def save_sys_module_state(self):
507 """Save the state of hooks in the sys module.
507 """Save the state of hooks in the sys module.
508
508
509 This has to be called after self.user_ns is created.
509 This has to be called after self.user_ns is created.
510 """
510 """
511 self._orig_sys_module_state = {}
511 self._orig_sys_module_state = {}
512 self._orig_sys_module_state['stdin'] = sys.stdin
512 self._orig_sys_module_state['stdin'] = sys.stdin
513 self._orig_sys_module_state['stdout'] = sys.stdout
513 self._orig_sys_module_state['stdout'] = sys.stdout
514 self._orig_sys_module_state['stderr'] = sys.stderr
514 self._orig_sys_module_state['stderr'] = sys.stderr
515 self._orig_sys_module_state['excepthook'] = sys.excepthook
515 self._orig_sys_module_state['excepthook'] = sys.excepthook
516 try:
516 try:
517 self._orig_sys_modules_main_name = self.user_ns['__name__']
517 self._orig_sys_modules_main_name = self.user_ns['__name__']
518 except KeyError:
518 except KeyError:
519 pass
519 pass
520
520
521 def restore_sys_module_state(self):
521 def restore_sys_module_state(self):
522 """Restore the state of the sys module."""
522 """Restore the state of the sys module."""
523 try:
523 try:
524 for k, v in self._orig_sys_module_state.items():
524 for k, v in self._orig_sys_module_state.items():
525 setattr(sys, k, v)
525 setattr(sys, k, v)
526 except AttributeError:
526 except AttributeError:
527 pass
527 pass
528 # Reset what what done in self.init_sys_modules
528 # Reset what what done in self.init_sys_modules
529 try:
529 try:
530 sys.modules[self.user_ns['__name__']] = self._orig_sys_modules_main_name
530 sys.modules[self.user_ns['__name__']] = self._orig_sys_modules_main_name
531 except (AttributeError, KeyError):
531 except (AttributeError, KeyError):
532 pass
532 pass
533
533
534 #-------------------------------------------------------------------------
534 #-------------------------------------------------------------------------
535 # Things related to hooks
535 # Things related to hooks
536 #-------------------------------------------------------------------------
536 #-------------------------------------------------------------------------
537
537
538 def init_hooks(self):
538 def init_hooks(self):
539 # hooks holds pointers used for user-side customizations
539 # hooks holds pointers used for user-side customizations
540 self.hooks = Struct()
540 self.hooks = Struct()
541
541
542 self.strdispatchers = {}
542 self.strdispatchers = {}
543
543
544 # Set all default hooks, defined in the IPython.hooks module.
544 # Set all default hooks, defined in the IPython.hooks module.
545 hooks = IPython.core.hooks
545 hooks = IPython.core.hooks
546 for hook_name in hooks.__all__:
546 for hook_name in hooks.__all__:
547 # default hooks have priority 100, i.e. low; user hooks should have
547 # default hooks have priority 100, i.e. low; user hooks should have
548 # 0-100 priority
548 # 0-100 priority
549 self.set_hook(hook_name,getattr(hooks,hook_name), 100)
549 self.set_hook(hook_name,getattr(hooks,hook_name), 100)
550
550
551 def set_hook(self,name,hook, priority = 50, str_key = None, re_key = None):
551 def set_hook(self,name,hook, priority = 50, str_key = None, re_key = None):
552 """set_hook(name,hook) -> sets an internal IPython hook.
552 """set_hook(name,hook) -> sets an internal IPython hook.
553
553
554 IPython exposes some of its internal API as user-modifiable hooks. By
554 IPython exposes some of its internal API as user-modifiable hooks. By
555 adding your function to one of these hooks, you can modify IPython's
555 adding your function to one of these hooks, you can modify IPython's
556 behavior to call at runtime your own routines."""
556 behavior to call at runtime your own routines."""
557
557
558 # At some point in the future, this should validate the hook before it
558 # At some point in the future, this should validate the hook before it
559 # accepts it. Probably at least check that the hook takes the number
559 # accepts it. Probably at least check that the hook takes the number
560 # of args it's supposed to.
560 # of args it's supposed to.
561
561
562 f = new.instancemethod(hook,self,self.__class__)
562 f = new.instancemethod(hook,self,self.__class__)
563
563
564 # check if the hook is for strdispatcher first
564 # check if the hook is for strdispatcher first
565 if str_key is not None:
565 if str_key is not None:
566 sdp = self.strdispatchers.get(name, StrDispatch())
566 sdp = self.strdispatchers.get(name, StrDispatch())
567 sdp.add_s(str_key, f, priority )
567 sdp.add_s(str_key, f, priority )
568 self.strdispatchers[name] = sdp
568 self.strdispatchers[name] = sdp
569 return
569 return
570 if re_key is not None:
570 if re_key is not None:
571 sdp = self.strdispatchers.get(name, StrDispatch())
571 sdp = self.strdispatchers.get(name, StrDispatch())
572 sdp.add_re(re.compile(re_key), f, priority )
572 sdp.add_re(re.compile(re_key), f, priority )
573 self.strdispatchers[name] = sdp
573 self.strdispatchers[name] = sdp
574 return
574 return
575
575
576 dp = getattr(self.hooks, name, None)
576 dp = getattr(self.hooks, name, None)
577 if name not in IPython.core.hooks.__all__:
577 if name not in IPython.core.hooks.__all__:
578 print "Warning! Hook '%s' is not one of %s" % \
578 print "Warning! Hook '%s' is not one of %s" % \
579 (name, IPython.core.hooks.__all__ )
579 (name, IPython.core.hooks.__all__ )
580 if not dp:
580 if not dp:
581 dp = IPython.core.hooks.CommandChainDispatcher()
581 dp = IPython.core.hooks.CommandChainDispatcher()
582
582
583 try:
583 try:
584 dp.add(f,priority)
584 dp.add(f,priority)
585 except AttributeError:
585 except AttributeError:
586 # it was not commandchain, plain old func - replace
586 # it was not commandchain, plain old func - replace
587 dp = f
587 dp = f
588
588
589 setattr(self.hooks,name, dp)
589 setattr(self.hooks,name, dp)
590
590
591 def register_post_execute(self, func):
591 def register_post_execute(self, func):
592 """Register a function for calling after code execution.
592 """Register a function for calling after code execution.
593 """
593 """
594 if not callable(func):
594 if not callable(func):
595 raise ValueError('argument %s must be callable' % func)
595 raise ValueError('argument %s must be callable' % func)
596 self._post_execute.add(func)
596 self._post_execute.add(func)
597
597
598 #-------------------------------------------------------------------------
598 #-------------------------------------------------------------------------
599 # Things related to the "main" module
599 # Things related to the "main" module
600 #-------------------------------------------------------------------------
600 #-------------------------------------------------------------------------
601
601
602 def new_main_mod(self,ns=None):
602 def new_main_mod(self,ns=None):
603 """Return a new 'main' module object for user code execution.
603 """Return a new 'main' module object for user code execution.
604 """
604 """
605 main_mod = self._user_main_module
605 main_mod = self._user_main_module
606 init_fakemod_dict(main_mod,ns)
606 init_fakemod_dict(main_mod,ns)
607 return main_mod
607 return main_mod
608
608
609 def cache_main_mod(self,ns,fname):
609 def cache_main_mod(self,ns,fname):
610 """Cache a main module's namespace.
610 """Cache a main module's namespace.
611
611
612 When scripts are executed via %run, we must keep a reference to the
612 When scripts are executed via %run, we must keep a reference to the
613 namespace of their __main__ module (a FakeModule instance) around so
613 namespace of their __main__ module (a FakeModule instance) around so
614 that Python doesn't clear it, rendering objects defined therein
614 that Python doesn't clear it, rendering objects defined therein
615 useless.
615 useless.
616
616
617 This method keeps said reference in a private dict, keyed by the
617 This method keeps said reference in a private dict, keyed by the
618 absolute path of the module object (which corresponds to the script
618 absolute path of the module object (which corresponds to the script
619 path). This way, for multiple executions of the same script we only
619 path). This way, for multiple executions of the same script we only
620 keep one copy of the namespace (the last one), thus preventing memory
620 keep one copy of the namespace (the last one), thus preventing memory
621 leaks from old references while allowing the objects from the last
621 leaks from old references while allowing the objects from the last
622 execution to be accessible.
622 execution to be accessible.
623
623
624 Note: we can not allow the actual FakeModule instances to be deleted,
624 Note: we can not allow the actual FakeModule instances to be deleted,
625 because of how Python tears down modules (it hard-sets all their
625 because of how Python tears down modules (it hard-sets all their
626 references to None without regard for reference counts). This method
626 references to None without regard for reference counts). This method
627 must therefore make a *copy* of the given namespace, to allow the
627 must therefore make a *copy* of the given namespace, to allow the
628 original module's __dict__ to be cleared and reused.
628 original module's __dict__ to be cleared and reused.
629
629
630
630
631 Parameters
631 Parameters
632 ----------
632 ----------
633 ns : a namespace (a dict, typically)
633 ns : a namespace (a dict, typically)
634
634
635 fname : str
635 fname : str
636 Filename associated with the namespace.
636 Filename associated with the namespace.
637
637
638 Examples
638 Examples
639 --------
639 --------
640
640
641 In [10]: import IPython
641 In [10]: import IPython
642
642
643 In [11]: _ip.cache_main_mod(IPython.__dict__,IPython.__file__)
643 In [11]: _ip.cache_main_mod(IPython.__dict__,IPython.__file__)
644
644
645 In [12]: IPython.__file__ in _ip._main_ns_cache
645 In [12]: IPython.__file__ in _ip._main_ns_cache
646 Out[12]: True
646 Out[12]: True
647 """
647 """
648 self._main_ns_cache[os.path.abspath(fname)] = ns.copy()
648 self._main_ns_cache[os.path.abspath(fname)] = ns.copy()
649
649
650 def clear_main_mod_cache(self):
650 def clear_main_mod_cache(self):
651 """Clear the cache of main modules.
651 """Clear the cache of main modules.
652
652
653 Mainly for use by utilities like %reset.
653 Mainly for use by utilities like %reset.
654
654
655 Examples
655 Examples
656 --------
656 --------
657
657
658 In [15]: import IPython
658 In [15]: import IPython
659
659
660 In [16]: _ip.cache_main_mod(IPython.__dict__,IPython.__file__)
660 In [16]: _ip.cache_main_mod(IPython.__dict__,IPython.__file__)
661
661
662 In [17]: len(_ip._main_ns_cache) > 0
662 In [17]: len(_ip._main_ns_cache) > 0
663 Out[17]: True
663 Out[17]: True
664
664
665 In [18]: _ip.clear_main_mod_cache()
665 In [18]: _ip.clear_main_mod_cache()
666
666
667 In [19]: len(_ip._main_ns_cache) == 0
667 In [19]: len(_ip._main_ns_cache) == 0
668 Out[19]: True
668 Out[19]: True
669 """
669 """
670 self._main_ns_cache.clear()
670 self._main_ns_cache.clear()
671
671
672 #-------------------------------------------------------------------------
672 #-------------------------------------------------------------------------
673 # Things related to debugging
673 # Things related to debugging
674 #-------------------------------------------------------------------------
674 #-------------------------------------------------------------------------
675
675
676 def init_pdb(self):
676 def init_pdb(self):
677 # Set calling of pdb on exceptions
677 # Set calling of pdb on exceptions
678 # self.call_pdb is a property
678 # self.call_pdb is a property
679 self.call_pdb = self.pdb
679 self.call_pdb = self.pdb
680
680
681 def _get_call_pdb(self):
681 def _get_call_pdb(self):
682 return self._call_pdb
682 return self._call_pdb
683
683
684 def _set_call_pdb(self,val):
684 def _set_call_pdb(self,val):
685
685
686 if val not in (0,1,False,True):
686 if val not in (0,1,False,True):
687 raise ValueError,'new call_pdb value must be boolean'
687 raise ValueError,'new call_pdb value must be boolean'
688
688
689 # store value in instance
689 # store value in instance
690 self._call_pdb = val
690 self._call_pdb = val
691
691
692 # notify the actual exception handlers
692 # notify the actual exception handlers
693 self.InteractiveTB.call_pdb = val
693 self.InteractiveTB.call_pdb = val
694
694
695 call_pdb = property(_get_call_pdb,_set_call_pdb,None,
695 call_pdb = property(_get_call_pdb,_set_call_pdb,None,
696 'Control auto-activation of pdb at exceptions')
696 'Control auto-activation of pdb at exceptions')
697
697
698 def debugger(self,force=False):
698 def debugger(self,force=False):
699 """Call the pydb/pdb debugger.
699 """Call the pydb/pdb debugger.
700
700
701 Keywords:
701 Keywords:
702
702
703 - force(False): by default, this routine checks the instance call_pdb
703 - force(False): by default, this routine checks the instance call_pdb
704 flag and does not actually invoke the debugger if the flag is false.
704 flag and does not actually invoke the debugger if the flag is false.
705 The 'force' option forces the debugger to activate even if the flag
705 The 'force' option forces the debugger to activate even if the flag
706 is false.
706 is false.
707 """
707 """
708
708
709 if not (force or self.call_pdb):
709 if not (force or self.call_pdb):
710 return
710 return
711
711
712 if not hasattr(sys,'last_traceback'):
712 if not hasattr(sys,'last_traceback'):
713 error('No traceback has been produced, nothing to debug.')
713 error('No traceback has been produced, nothing to debug.')
714 return
714 return
715
715
716 # use pydb if available
716 # use pydb if available
717 if debugger.has_pydb:
717 if debugger.has_pydb:
718 from pydb import pm
718 from pydb import pm
719 else:
719 else:
720 # fallback to our internal debugger
720 # fallback to our internal debugger
721 pm = lambda : self.InteractiveTB.debugger(force=True)
721 pm = lambda : self.InteractiveTB.debugger(force=True)
722 self.history_saving_wrapper(pm)()
722 self.history_saving_wrapper(pm)()
723
723
724 #-------------------------------------------------------------------------
724 #-------------------------------------------------------------------------
725 # Things related to IPython's various namespaces
725 # Things related to IPython's various namespaces
726 #-------------------------------------------------------------------------
726 #-------------------------------------------------------------------------
727
727
728 def init_create_namespaces(self, user_ns=None, user_global_ns=None):
728 def init_create_namespaces(self, user_ns=None, user_global_ns=None):
729 # Create the namespace where the user will operate. user_ns is
729 # Create the namespace where the user will operate. user_ns is
730 # normally the only one used, and it is passed to the exec calls as
730 # normally the only one used, and it is passed to the exec calls as
731 # the locals argument. But we do carry a user_global_ns namespace
731 # the locals argument. But we do carry a user_global_ns namespace
732 # given as the exec 'globals' argument, This is useful in embedding
732 # given as the exec 'globals' argument, This is useful in embedding
733 # situations where the ipython shell opens in a context where the
733 # situations where the ipython shell opens in a context where the
734 # distinction between locals and globals is meaningful. For
734 # distinction between locals and globals is meaningful. For
735 # non-embedded contexts, it is just the same object as the user_ns dict.
735 # non-embedded contexts, it is just the same object as the user_ns dict.
736
736
737 # FIXME. For some strange reason, __builtins__ is showing up at user
737 # FIXME. For some strange reason, __builtins__ is showing up at user
738 # level as a dict instead of a module. This is a manual fix, but I
738 # level as a dict instead of a module. This is a manual fix, but I
739 # should really track down where the problem is coming from. Alex
739 # should really track down where the problem is coming from. Alex
740 # Schmolck reported this problem first.
740 # Schmolck reported this problem first.
741
741
742 # A useful post by Alex Martelli on this topic:
742 # A useful post by Alex Martelli on this topic:
743 # Re: inconsistent value from __builtins__
743 # Re: inconsistent value from __builtins__
744 # Von: Alex Martelli <aleaxit@yahoo.com>
744 # Von: Alex Martelli <aleaxit@yahoo.com>
745 # Datum: Freitag 01 Oktober 2004 04:45:34 nachmittags/abends
745 # Datum: Freitag 01 Oktober 2004 04:45:34 nachmittags/abends
746 # Gruppen: comp.lang.python
746 # Gruppen: comp.lang.python
747
747
748 # Michael Hohn <hohn@hooknose.lbl.gov> wrote:
748 # Michael Hohn <hohn@hooknose.lbl.gov> wrote:
749 # > >>> print type(builtin_check.get_global_binding('__builtins__'))
749 # > >>> print type(builtin_check.get_global_binding('__builtins__'))
750 # > <type 'dict'>
750 # > <type 'dict'>
751 # > >>> print type(__builtins__)
751 # > >>> print type(__builtins__)
752 # > <type 'module'>
752 # > <type 'module'>
753 # > Is this difference in return value intentional?
753 # > Is this difference in return value intentional?
754
754
755 # Well, it's documented that '__builtins__' can be either a dictionary
755 # Well, it's documented that '__builtins__' can be either a dictionary
756 # or a module, and it's been that way for a long time. Whether it's
756 # or a module, and it's been that way for a long time. Whether it's
757 # intentional (or sensible), I don't know. In any case, the idea is
757 # intentional (or sensible), I don't know. In any case, the idea is
758 # that if you need to access the built-in namespace directly, you
758 # that if you need to access the built-in namespace directly, you
759 # should start with "import __builtin__" (note, no 's') which will
759 # should start with "import __builtin__" (note, no 's') which will
760 # definitely give you a module. Yeah, it's somewhat confusing:-(.
760 # definitely give you a module. Yeah, it's somewhat confusing:-(.
761
761
762 # These routines return properly built dicts as needed by the rest of
762 # These routines return properly built dicts as needed by the rest of
763 # the code, and can also be used by extension writers to generate
763 # the code, and can also be used by extension writers to generate
764 # properly initialized namespaces.
764 # properly initialized namespaces.
765 user_ns, user_global_ns = self.make_user_namespaces(user_ns,
765 user_ns, user_global_ns = self.make_user_namespaces(user_ns,
766 user_global_ns)
766 user_global_ns)
767
767
768 # Assign namespaces
768 # Assign namespaces
769 # This is the namespace where all normal user variables live
769 # This is the namespace where all normal user variables live
770 self.user_ns = user_ns
770 self.user_ns = user_ns
771 self.user_global_ns = user_global_ns
771 self.user_global_ns = user_global_ns
772
772
773 # An auxiliary namespace that checks what parts of the user_ns were
773 # An auxiliary namespace that checks what parts of the user_ns were
774 # loaded at startup, so we can list later only variables defined in
774 # loaded at startup, so we can list later only variables defined in
775 # actual interactive use. Since it is always a subset of user_ns, it
775 # actual interactive use. Since it is always a subset of user_ns, it
776 # doesn't need to be separately tracked in the ns_table.
776 # doesn't need to be separately tracked in the ns_table.
777 self.user_ns_hidden = {}
777 self.user_ns_hidden = {}
778
778
779 # A namespace to keep track of internal data structures to prevent
779 # A namespace to keep track of internal data structures to prevent
780 # them from cluttering user-visible stuff. Will be updated later
780 # them from cluttering user-visible stuff. Will be updated later
781 self.internal_ns = {}
781 self.internal_ns = {}
782
782
783 # Now that FakeModule produces a real module, we've run into a nasty
783 # Now that FakeModule produces a real module, we've run into a nasty
784 # problem: after script execution (via %run), the module where the user
784 # problem: after script execution (via %run), the module where the user
785 # code ran is deleted. Now that this object is a true module (needed
785 # code ran is deleted. Now that this object is a true module (needed
786 # so docetst and other tools work correctly), the Python module
786 # so docetst and other tools work correctly), the Python module
787 # teardown mechanism runs over it, and sets to None every variable
787 # teardown mechanism runs over it, and sets to None every variable
788 # present in that module. Top-level references to objects from the
788 # present in that module. Top-level references to objects from the
789 # script survive, because the user_ns is updated with them. However,
789 # script survive, because the user_ns is updated with them. However,
790 # calling functions defined in the script that use other things from
790 # calling functions defined in the script that use other things from
791 # the script will fail, because the function's closure had references
791 # the script will fail, because the function's closure had references
792 # to the original objects, which are now all None. So we must protect
792 # to the original objects, which are now all None. So we must protect
793 # these modules from deletion by keeping a cache.
793 # these modules from deletion by keeping a cache.
794 #
794 #
795 # To avoid keeping stale modules around (we only need the one from the
795 # To avoid keeping stale modules around (we only need the one from the
796 # last run), we use a dict keyed with the full path to the script, so
796 # last run), we use a dict keyed with the full path to the script, so
797 # only the last version of the module is held in the cache. Note,
797 # only the last version of the module is held in the cache. Note,
798 # however, that we must cache the module *namespace contents* (their
798 # however, that we must cache the module *namespace contents* (their
799 # __dict__). Because if we try to cache the actual modules, old ones
799 # __dict__). Because if we try to cache the actual modules, old ones
800 # (uncached) could be destroyed while still holding references (such as
800 # (uncached) could be destroyed while still holding references (such as
801 # those held by GUI objects that tend to be long-lived)>
801 # those held by GUI objects that tend to be long-lived)>
802 #
802 #
803 # The %reset command will flush this cache. See the cache_main_mod()
803 # The %reset command will flush this cache. See the cache_main_mod()
804 # and clear_main_mod_cache() methods for details on use.
804 # and clear_main_mod_cache() methods for details on use.
805
805
806 # This is the cache used for 'main' namespaces
806 # This is the cache used for 'main' namespaces
807 self._main_ns_cache = {}
807 self._main_ns_cache = {}
808 # And this is the single instance of FakeModule whose __dict__ we keep
808 # And this is the single instance of FakeModule whose __dict__ we keep
809 # copying and clearing for reuse on each %run
809 # copying and clearing for reuse on each %run
810 self._user_main_module = FakeModule()
810 self._user_main_module = FakeModule()
811
811
812 # A table holding all the namespaces IPython deals with, so that
812 # A table holding all the namespaces IPython deals with, so that
813 # introspection facilities can search easily.
813 # introspection facilities can search easily.
814 self.ns_table = {'user':user_ns,
814 self.ns_table = {'user':user_ns,
815 'user_global':user_global_ns,
815 'user_global':user_global_ns,
816 'internal':self.internal_ns,
816 'internal':self.internal_ns,
817 'builtin':__builtin__.__dict__
817 'builtin':__builtin__.__dict__
818 }
818 }
819
819
820 # Similarly, track all namespaces where references can be held and that
820 # Similarly, track all namespaces where references can be held and that
821 # we can safely clear (so it can NOT include builtin). This one can be
821 # we can safely clear (so it can NOT include builtin). This one can be
822 # a simple list.
822 # a simple list. Note that the main execution namespaces, user_ns and
823 self.ns_refs_table = [ user_ns, user_global_ns, self.user_ns_hidden,
823 # user_global_ns, can NOT be listed here, as clearing them blindly
824 # causes errors in object __del__ methods. Instead, the reset() method
825 # clears them manually and carefully.
826 self.ns_refs_table = [ self.user_ns_hidden,
824 self.internal_ns, self._main_ns_cache ]
827 self.internal_ns, self._main_ns_cache ]
825
828
826 def make_user_namespaces(self, user_ns=None, user_global_ns=None):
829 def make_user_namespaces(self, user_ns=None, user_global_ns=None):
827 """Return a valid local and global user interactive namespaces.
830 """Return a valid local and global user interactive namespaces.
828
831
829 This builds a dict with the minimal information needed to operate as a
832 This builds a dict with the minimal information needed to operate as a
830 valid IPython user namespace, which you can pass to the various
833 valid IPython user namespace, which you can pass to the various
831 embedding classes in ipython. The default implementation returns the
834 embedding classes in ipython. The default implementation returns the
832 same dict for both the locals and the globals to allow functions to
835 same dict for both the locals and the globals to allow functions to
833 refer to variables in the namespace. Customized implementations can
836 refer to variables in the namespace. Customized implementations can
834 return different dicts. The locals dictionary can actually be anything
837 return different dicts. The locals dictionary can actually be anything
835 following the basic mapping protocol of a dict, but the globals dict
838 following the basic mapping protocol of a dict, but the globals dict
836 must be a true dict, not even a subclass. It is recommended that any
839 must be a true dict, not even a subclass. It is recommended that any
837 custom object for the locals namespace synchronize with the globals
840 custom object for the locals namespace synchronize with the globals
838 dict somehow.
841 dict somehow.
839
842
840 Raises TypeError if the provided globals namespace is not a true dict.
843 Raises TypeError if the provided globals namespace is not a true dict.
841
844
842 Parameters
845 Parameters
843 ----------
846 ----------
844 user_ns : dict-like, optional
847 user_ns : dict-like, optional
845 The current user namespace. The items in this namespace should
848 The current user namespace. The items in this namespace should
846 be included in the output. If None, an appropriate blank
849 be included in the output. If None, an appropriate blank
847 namespace should be created.
850 namespace should be created.
848 user_global_ns : dict, optional
851 user_global_ns : dict, optional
849 The current user global namespace. The items in this namespace
852 The current user global namespace. The items in this namespace
850 should be included in the output. If None, an appropriate
853 should be included in the output. If None, an appropriate
851 blank namespace should be created.
854 blank namespace should be created.
852
855
853 Returns
856 Returns
854 -------
857 -------
855 A pair of dictionary-like object to be used as the local namespace
858 A pair of dictionary-like object to be used as the local namespace
856 of the interpreter and a dict to be used as the global namespace.
859 of the interpreter and a dict to be used as the global namespace.
857 """
860 """
858
861
859
862
860 # We must ensure that __builtin__ (without the final 's') is always
863 # We must ensure that __builtin__ (without the final 's') is always
861 # available and pointing to the __builtin__ *module*. For more details:
864 # available and pointing to the __builtin__ *module*. For more details:
862 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
865 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
863
866
864 if user_ns is None:
867 if user_ns is None:
865 # Set __name__ to __main__ to better match the behavior of the
868 # Set __name__ to __main__ to better match the behavior of the
866 # normal interpreter.
869 # normal interpreter.
867 user_ns = {'__name__' :'__main__',
870 user_ns = {'__name__' :'__main__',
868 '__builtin__' : __builtin__,
871 '__builtin__' : __builtin__,
869 '__builtins__' : __builtin__,
872 '__builtins__' : __builtin__,
870 }
873 }
871 else:
874 else:
872 user_ns.setdefault('__name__','__main__')
875 user_ns.setdefault('__name__','__main__')
873 user_ns.setdefault('__builtin__',__builtin__)
876 user_ns.setdefault('__builtin__',__builtin__)
874 user_ns.setdefault('__builtins__',__builtin__)
877 user_ns.setdefault('__builtins__',__builtin__)
875
878
876 if user_global_ns is None:
879 if user_global_ns is None:
877 user_global_ns = user_ns
880 user_global_ns = user_ns
878 if type(user_global_ns) is not dict:
881 if type(user_global_ns) is not dict:
879 raise TypeError("user_global_ns must be a true dict; got %r"
882 raise TypeError("user_global_ns must be a true dict; got %r"
880 % type(user_global_ns))
883 % type(user_global_ns))
881
884
882 return user_ns, user_global_ns
885 return user_ns, user_global_ns
883
886
884 def init_sys_modules(self):
887 def init_sys_modules(self):
885 # We need to insert into sys.modules something that looks like a
888 # We need to insert into sys.modules something that looks like a
886 # module but which accesses the IPython namespace, for shelve and
889 # module but which accesses the IPython namespace, for shelve and
887 # pickle to work interactively. Normally they rely on getting
890 # pickle to work interactively. Normally they rely on getting
888 # everything out of __main__, but for embedding purposes each IPython
891 # everything out of __main__, but for embedding purposes each IPython
889 # instance has its own private namespace, so we can't go shoving
892 # instance has its own private namespace, so we can't go shoving
890 # everything into __main__.
893 # everything into __main__.
891
894
892 # note, however, that we should only do this for non-embedded
895 # note, however, that we should only do this for non-embedded
893 # ipythons, which really mimic the __main__.__dict__ with their own
896 # ipythons, which really mimic the __main__.__dict__ with their own
894 # namespace. Embedded instances, on the other hand, should not do
897 # namespace. Embedded instances, on the other hand, should not do
895 # this because they need to manage the user local/global namespaces
898 # this because they need to manage the user local/global namespaces
896 # only, but they live within a 'normal' __main__ (meaning, they
899 # only, but they live within a 'normal' __main__ (meaning, they
897 # shouldn't overtake the execution environment of the script they're
900 # shouldn't overtake the execution environment of the script they're
898 # embedded in).
901 # embedded in).
899
902
900 # This is overridden in the InteractiveShellEmbed subclass to a no-op.
903 # This is overridden in the InteractiveShellEmbed subclass to a no-op.
901
904
902 try:
905 try:
903 main_name = self.user_ns['__name__']
906 main_name = self.user_ns['__name__']
904 except KeyError:
907 except KeyError:
905 raise KeyError('user_ns dictionary MUST have a "__name__" key')
908 raise KeyError('user_ns dictionary MUST have a "__name__" key')
906 else:
909 else:
907 sys.modules[main_name] = FakeModule(self.user_ns)
910 sys.modules[main_name] = FakeModule(self.user_ns)
908
911
909 def init_user_ns(self):
912 def init_user_ns(self):
910 """Initialize all user-visible namespaces to their minimum defaults.
913 """Initialize all user-visible namespaces to their minimum defaults.
911
914
912 Certain history lists are also initialized here, as they effectively
915 Certain history lists are also initialized here, as they effectively
913 act as user namespaces.
916 act as user namespaces.
914
917
915 Notes
918 Notes
916 -----
919 -----
917 All data structures here are only filled in, they are NOT reset by this
920 All data structures here are only filled in, they are NOT reset by this
918 method. If they were not empty before, data will simply be added to
921 method. If they were not empty before, data will simply be added to
919 therm.
922 therm.
920 """
923 """
921 # This function works in two parts: first we put a few things in
924 # This function works in two parts: first we put a few things in
922 # user_ns, and we sync that contents into user_ns_hidden so that these
925 # user_ns, and we sync that contents into user_ns_hidden so that these
923 # initial variables aren't shown by %who. After the sync, we add the
926 # initial variables aren't shown by %who. After the sync, we add the
924 # rest of what we *do* want the user to see with %who even on a new
927 # rest of what we *do* want the user to see with %who even on a new
925 # session (probably nothing, so theye really only see their own stuff)
928 # session (probably nothing, so theye really only see their own stuff)
926
929
927 # The user dict must *always* have a __builtin__ reference to the
930 # The user dict must *always* have a __builtin__ reference to the
928 # Python standard __builtin__ namespace, which must be imported.
931 # Python standard __builtin__ namespace, which must be imported.
929 # This is so that certain operations in prompt evaluation can be
932 # This is so that certain operations in prompt evaluation can be
930 # reliably executed with builtins. Note that we can NOT use
933 # reliably executed with builtins. Note that we can NOT use
931 # __builtins__ (note the 's'), because that can either be a dict or a
934 # __builtins__ (note the 's'), because that can either be a dict or a
932 # module, and can even mutate at runtime, depending on the context
935 # module, and can even mutate at runtime, depending on the context
933 # (Python makes no guarantees on it). In contrast, __builtin__ is
936 # (Python makes no guarantees on it). In contrast, __builtin__ is
934 # always a module object, though it must be explicitly imported.
937 # always a module object, though it must be explicitly imported.
935
938
936 # For more details:
939 # For more details:
937 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
940 # http://mail.python.org/pipermail/python-dev/2001-April/014068.html
938 ns = dict(__builtin__ = __builtin__)
941 ns = dict(__builtin__ = __builtin__)
939
942
940 # Put 'help' in the user namespace
943 # Put 'help' in the user namespace
941 try:
944 try:
942 from site import _Helper
945 from site import _Helper
943 ns['help'] = _Helper()
946 ns['help'] = _Helper()
944 except ImportError:
947 except ImportError:
945 warn('help() not available - check site.py')
948 warn('help() not available - check site.py')
946
949
947 # make global variables for user access to the histories
950 # make global variables for user access to the histories
948 ns['_ih'] = self.input_hist
951 ns['_ih'] = self.input_hist
949 ns['_oh'] = self.output_hist
952 ns['_oh'] = self.output_hist
950 ns['_dh'] = self.dir_hist
953 ns['_dh'] = self.dir_hist
951
954
952 ns['_sh'] = shadowns
955 ns['_sh'] = shadowns
953
956
954 # user aliases to input and output histories. These shouldn't show up
957 # user aliases to input and output histories. These shouldn't show up
955 # in %who, as they can have very large reprs.
958 # in %who, as they can have very large reprs.
956 ns['In'] = self.input_hist
959 ns['In'] = self.input_hist
957 ns['Out'] = self.output_hist
960 ns['Out'] = self.output_hist
958
961
959 # Store myself as the public api!!!
962 # Store myself as the public api!!!
960 ns['get_ipython'] = self.get_ipython
963 ns['get_ipython'] = self.get_ipython
961
964
962 # Sync what we've added so far to user_ns_hidden so these aren't seen
965 # Sync what we've added so far to user_ns_hidden so these aren't seen
963 # by %who
966 # by %who
964 self.user_ns_hidden.update(ns)
967 self.user_ns_hidden.update(ns)
965
968
966 # Anything put into ns now would show up in %who. Think twice before
969 # Anything put into ns now would show up in %who. Think twice before
967 # putting anything here, as we really want %who to show the user their
970 # putting anything here, as we really want %who to show the user their
968 # stuff, not our variables.
971 # stuff, not our variables.
969
972
970 # Finally, update the real user's namespace
973 # Finally, update the real user's namespace
971 self.user_ns.update(ns)
974 self.user_ns.update(ns)
972
975
973 def reset(self):
976 def reset(self):
974 """Clear all internal namespaces.
977 """Clear all internal namespaces.
975
978
976 Note that this is much more aggressive than %reset, since it clears
979 Note that this is much more aggressive than %reset, since it clears
977 fully all namespaces, as well as all input/output lists.
980 fully all namespaces, as well as all input/output lists.
978 """
981 """
979 # Clear histories
982 # Clear histories
980 self.history_manager.reset()
983 self.history_manager.reset()
981
984
982 # Reset counter used to index all histories
985 # Reset counter used to index all histories
983 self.execution_count = 0
986 self.execution_count = 0
984
987
985 # Restore the user namespaces to minimal usability
988 # Restore the user namespaces to minimal usability
986 for ns in self.ns_refs_table:
989 for ns in self.ns_refs_table:
987 ns.clear()
990 ns.clear()
991
992 # The main execution namespaces must be cleared very carefully,
993 # skipping the deletion of the builtin-related keys, because doing so
994 # would cause errors in many object's __del__ methods.
995 for ns in [self.user_ns, self.user_global_ns]:
996 drop_keys = set(ns.keys())
997 drop_keys.discard('__builtin__')
998 drop_keys.discard('__builtins__')
999 for k in drop_keys:
1000 del ns[k]
1001
1002 # Restore the user namespaces to minimal usability
988 self.init_user_ns()
1003 self.init_user_ns()
989
1004
990 # Restore the default and user aliases
1005 # Restore the default and user aliases
991 self.alias_manager.clear_aliases()
1006 self.alias_manager.clear_aliases()
992 self.alias_manager.init_aliases()
1007 self.alias_manager.init_aliases()
993
1008
994 def reset_selective(self, regex=None):
1009 def reset_selective(self, regex=None):
995 """Clear selective variables from internal namespaces based on a
1010 """Clear selective variables from internal namespaces based on a
996 specified regular expression.
1011 specified regular expression.
997
1012
998 Parameters
1013 Parameters
999 ----------
1014 ----------
1000 regex : string or compiled pattern, optional
1015 regex : string or compiled pattern, optional
1001 A regular expression pattern that will be used in searching
1016 A regular expression pattern that will be used in searching
1002 variable names in the users namespaces.
1017 variable names in the users namespaces.
1003 """
1018 """
1004 if regex is not None:
1019 if regex is not None:
1005 try:
1020 try:
1006 m = re.compile(regex)
1021 m = re.compile(regex)
1007 except TypeError:
1022 except TypeError:
1008 raise TypeError('regex must be a string or compiled pattern')
1023 raise TypeError('regex must be a string or compiled pattern')
1009 # Search for keys in each namespace that match the given regex
1024 # Search for keys in each namespace that match the given regex
1010 # If a match is found, delete the key/value pair.
1025 # If a match is found, delete the key/value pair.
1011 for ns in self.ns_refs_table:
1026 for ns in self.ns_refs_table:
1012 for var in ns:
1027 for var in ns:
1013 if m.search(var):
1028 if m.search(var):
1014 del ns[var]
1029 del ns[var]
1015
1030
1016 def push(self, variables, interactive=True):
1031 def push(self, variables, interactive=True):
1017 """Inject a group of variables into the IPython user namespace.
1032 """Inject a group of variables into the IPython user namespace.
1018
1033
1019 Parameters
1034 Parameters
1020 ----------
1035 ----------
1021 variables : dict, str or list/tuple of str
1036 variables : dict, str or list/tuple of str
1022 The variables to inject into the user's namespace. If a dict, a
1037 The variables to inject into the user's namespace. If a dict, a
1023 simple update is done. If a str, the string is assumed to have
1038 simple update is done. If a str, the string is assumed to have
1024 variable names separated by spaces. A list/tuple of str can also
1039 variable names separated by spaces. A list/tuple of str can also
1025 be used to give the variable names. If just the variable names are
1040 be used to give the variable names. If just the variable names are
1026 give (list/tuple/str) then the variable values looked up in the
1041 give (list/tuple/str) then the variable values looked up in the
1027 callers frame.
1042 callers frame.
1028 interactive : bool
1043 interactive : bool
1029 If True (default), the variables will be listed with the ``who``
1044 If True (default), the variables will be listed with the ``who``
1030 magic.
1045 magic.
1031 """
1046 """
1032 vdict = None
1047 vdict = None
1033
1048
1034 # We need a dict of name/value pairs to do namespace updates.
1049 # We need a dict of name/value pairs to do namespace updates.
1035 if isinstance(variables, dict):
1050 if isinstance(variables, dict):
1036 vdict = variables
1051 vdict = variables
1037 elif isinstance(variables, (basestring, list, tuple)):
1052 elif isinstance(variables, (basestring, list, tuple)):
1038 if isinstance(variables, basestring):
1053 if isinstance(variables, basestring):
1039 vlist = variables.split()
1054 vlist = variables.split()
1040 else:
1055 else:
1041 vlist = variables
1056 vlist = variables
1042 vdict = {}
1057 vdict = {}
1043 cf = sys._getframe(1)
1058 cf = sys._getframe(1)
1044 for name in vlist:
1059 for name in vlist:
1045 try:
1060 try:
1046 vdict[name] = eval(name, cf.f_globals, cf.f_locals)
1061 vdict[name] = eval(name, cf.f_globals, cf.f_locals)
1047 except:
1062 except:
1048 print ('Could not get variable %s from %s' %
1063 print ('Could not get variable %s from %s' %
1049 (name,cf.f_code.co_name))
1064 (name,cf.f_code.co_name))
1050 else:
1065 else:
1051 raise ValueError('variables must be a dict/str/list/tuple')
1066 raise ValueError('variables must be a dict/str/list/tuple')
1052
1067
1053 # Propagate variables to user namespace
1068 # Propagate variables to user namespace
1054 self.user_ns.update(vdict)
1069 self.user_ns.update(vdict)
1055
1070
1056 # And configure interactive visibility
1071 # And configure interactive visibility
1057 config_ns = self.user_ns_hidden
1072 config_ns = self.user_ns_hidden
1058 if interactive:
1073 if interactive:
1059 for name, val in vdict.iteritems():
1074 for name, val in vdict.iteritems():
1060 config_ns.pop(name, None)
1075 config_ns.pop(name, None)
1061 else:
1076 else:
1062 for name,val in vdict.iteritems():
1077 for name,val in vdict.iteritems():
1063 config_ns[name] = val
1078 config_ns[name] = val
1064
1079
1065 #-------------------------------------------------------------------------
1080 #-------------------------------------------------------------------------
1066 # Things related to object introspection
1081 # Things related to object introspection
1067 #-------------------------------------------------------------------------
1082 #-------------------------------------------------------------------------
1068
1083
1069 def _ofind(self, oname, namespaces=None):
1084 def _ofind(self, oname, namespaces=None):
1070 """Find an object in the available namespaces.
1085 """Find an object in the available namespaces.
1071
1086
1072 self._ofind(oname) -> dict with keys: found,obj,ospace,ismagic
1087 self._ofind(oname) -> dict with keys: found,obj,ospace,ismagic
1073
1088
1074 Has special code to detect magic functions.
1089 Has special code to detect magic functions.
1075 """
1090 """
1076 #oname = oname.strip()
1091 #oname = oname.strip()
1077 #print '1- oname: <%r>' % oname # dbg
1092 #print '1- oname: <%r>' % oname # dbg
1078 try:
1093 try:
1079 oname = oname.strip().encode('ascii')
1094 oname = oname.strip().encode('ascii')
1080 #print '2- oname: <%r>' % oname # dbg
1095 #print '2- oname: <%r>' % oname # dbg
1081 except UnicodeEncodeError:
1096 except UnicodeEncodeError:
1082 print 'Python identifiers can only contain ascii characters.'
1097 print 'Python identifiers can only contain ascii characters.'
1083 return dict(found=False)
1098 return dict(found=False)
1084
1099
1085 alias_ns = None
1100 alias_ns = None
1086 if namespaces is None:
1101 if namespaces is None:
1087 # Namespaces to search in:
1102 # Namespaces to search in:
1088 # Put them in a list. The order is important so that we
1103 # Put them in a list. The order is important so that we
1089 # find things in the same order that Python finds them.
1104 # find things in the same order that Python finds them.
1090 namespaces = [ ('Interactive', self.user_ns),
1105 namespaces = [ ('Interactive', self.user_ns),
1091 ('IPython internal', self.internal_ns),
1106 ('IPython internal', self.internal_ns),
1092 ('Python builtin', __builtin__.__dict__),
1107 ('Python builtin', __builtin__.__dict__),
1093 ('Alias', self.alias_manager.alias_table),
1108 ('Alias', self.alias_manager.alias_table),
1094 ]
1109 ]
1095 alias_ns = self.alias_manager.alias_table
1110 alias_ns = self.alias_manager.alias_table
1096
1111
1097 # initialize results to 'null'
1112 # initialize results to 'null'
1098 found = False; obj = None; ospace = None; ds = None;
1113 found = False; obj = None; ospace = None; ds = None;
1099 ismagic = False; isalias = False; parent = None
1114 ismagic = False; isalias = False; parent = None
1100
1115
1101 # We need to special-case 'print', which as of python2.6 registers as a
1116 # We need to special-case 'print', which as of python2.6 registers as a
1102 # function but should only be treated as one if print_function was
1117 # function but should only be treated as one if print_function was
1103 # loaded with a future import. In this case, just bail.
1118 # loaded with a future import. In this case, just bail.
1104 if (oname == 'print' and not (self.compile.compiler.flags &
1119 if (oname == 'print' and not (self.compile.compiler.flags &
1105 __future__.CO_FUTURE_PRINT_FUNCTION)):
1120 __future__.CO_FUTURE_PRINT_FUNCTION)):
1106 return {'found':found, 'obj':obj, 'namespace':ospace,
1121 return {'found':found, 'obj':obj, 'namespace':ospace,
1107 'ismagic':ismagic, 'isalias':isalias, 'parent':parent}
1122 'ismagic':ismagic, 'isalias':isalias, 'parent':parent}
1108
1123
1109 # Look for the given name by splitting it in parts. If the head is
1124 # Look for the given name by splitting it in parts. If the head is
1110 # found, then we look for all the remaining parts as members, and only
1125 # found, then we look for all the remaining parts as members, and only
1111 # declare success if we can find them all.
1126 # declare success if we can find them all.
1112 oname_parts = oname.split('.')
1127 oname_parts = oname.split('.')
1113 oname_head, oname_rest = oname_parts[0],oname_parts[1:]
1128 oname_head, oname_rest = oname_parts[0],oname_parts[1:]
1114 for nsname,ns in namespaces:
1129 for nsname,ns in namespaces:
1115 try:
1130 try:
1116 obj = ns[oname_head]
1131 obj = ns[oname_head]
1117 except KeyError:
1132 except KeyError:
1118 continue
1133 continue
1119 else:
1134 else:
1120 #print 'oname_rest:', oname_rest # dbg
1135 #print 'oname_rest:', oname_rest # dbg
1121 for part in oname_rest:
1136 for part in oname_rest:
1122 try:
1137 try:
1123 parent = obj
1138 parent = obj
1124 obj = getattr(obj,part)
1139 obj = getattr(obj,part)
1125 except:
1140 except:
1126 # Blanket except b/c some badly implemented objects
1141 # Blanket except b/c some badly implemented objects
1127 # allow __getattr__ to raise exceptions other than
1142 # allow __getattr__ to raise exceptions other than
1128 # AttributeError, which then crashes IPython.
1143 # AttributeError, which then crashes IPython.
1129 break
1144 break
1130 else:
1145 else:
1131 # If we finish the for loop (no break), we got all members
1146 # If we finish the for loop (no break), we got all members
1132 found = True
1147 found = True
1133 ospace = nsname
1148 ospace = nsname
1134 if ns == alias_ns:
1149 if ns == alias_ns:
1135 isalias = True
1150 isalias = True
1136 break # namespace loop
1151 break # namespace loop
1137
1152
1138 # Try to see if it's magic
1153 # Try to see if it's magic
1139 if not found:
1154 if not found:
1140 if oname.startswith(ESC_MAGIC):
1155 if oname.startswith(ESC_MAGIC):
1141 oname = oname[1:]
1156 oname = oname[1:]
1142 obj = getattr(self,'magic_'+oname,None)
1157 obj = getattr(self,'magic_'+oname,None)
1143 if obj is not None:
1158 if obj is not None:
1144 found = True
1159 found = True
1145 ospace = 'IPython internal'
1160 ospace = 'IPython internal'
1146 ismagic = True
1161 ismagic = True
1147
1162
1148 # Last try: special-case some literals like '', [], {}, etc:
1163 # Last try: special-case some literals like '', [], {}, etc:
1149 if not found and oname_head in ["''",'""','[]','{}','()']:
1164 if not found and oname_head in ["''",'""','[]','{}','()']:
1150 obj = eval(oname_head)
1165 obj = eval(oname_head)
1151 found = True
1166 found = True
1152 ospace = 'Interactive'
1167 ospace = 'Interactive'
1153
1168
1154 return {'found':found, 'obj':obj, 'namespace':ospace,
1169 return {'found':found, 'obj':obj, 'namespace':ospace,
1155 'ismagic':ismagic, 'isalias':isalias, 'parent':parent}
1170 'ismagic':ismagic, 'isalias':isalias, 'parent':parent}
1156
1171
1157 def _ofind_property(self, oname, info):
1172 def _ofind_property(self, oname, info):
1158 """Second part of object finding, to look for property details."""
1173 """Second part of object finding, to look for property details."""
1159 if info.found:
1174 if info.found:
1160 # Get the docstring of the class property if it exists.
1175 # Get the docstring of the class property if it exists.
1161 path = oname.split('.')
1176 path = oname.split('.')
1162 root = '.'.join(path[:-1])
1177 root = '.'.join(path[:-1])
1163 if info.parent is not None:
1178 if info.parent is not None:
1164 try:
1179 try:
1165 target = getattr(info.parent, '__class__')
1180 target = getattr(info.parent, '__class__')
1166 # The object belongs to a class instance.
1181 # The object belongs to a class instance.
1167 try:
1182 try:
1168 target = getattr(target, path[-1])
1183 target = getattr(target, path[-1])
1169 # The class defines the object.
1184 # The class defines the object.
1170 if isinstance(target, property):
1185 if isinstance(target, property):
1171 oname = root + '.__class__.' + path[-1]
1186 oname = root + '.__class__.' + path[-1]
1172 info = Struct(self._ofind(oname))
1187 info = Struct(self._ofind(oname))
1173 except AttributeError: pass
1188 except AttributeError: pass
1174 except AttributeError: pass
1189 except AttributeError: pass
1175
1190
1176 # We return either the new info or the unmodified input if the object
1191 # We return either the new info or the unmodified input if the object
1177 # hadn't been found
1192 # hadn't been found
1178 return info
1193 return info
1179
1194
1180 def _object_find(self, oname, namespaces=None):
1195 def _object_find(self, oname, namespaces=None):
1181 """Find an object and return a struct with info about it."""
1196 """Find an object and return a struct with info about it."""
1182 inf = Struct(self._ofind(oname, namespaces))
1197 inf = Struct(self._ofind(oname, namespaces))
1183 return Struct(self._ofind_property(oname, inf))
1198 return Struct(self._ofind_property(oname, inf))
1184
1199
1185 def _inspect(self, meth, oname, namespaces=None, **kw):
1200 def _inspect(self, meth, oname, namespaces=None, **kw):
1186 """Generic interface to the inspector system.
1201 """Generic interface to the inspector system.
1187
1202
1188 This function is meant to be called by pdef, pdoc & friends."""
1203 This function is meant to be called by pdef, pdoc & friends."""
1189 info = self._object_find(oname)
1204 info = self._object_find(oname)
1190 if info.found:
1205 if info.found:
1191 pmethod = getattr(self.inspector, meth)
1206 pmethod = getattr(self.inspector, meth)
1192 formatter = format_screen if info.ismagic else None
1207 formatter = format_screen if info.ismagic else None
1193 if meth == 'pdoc':
1208 if meth == 'pdoc':
1194 pmethod(info.obj, oname, formatter)
1209 pmethod(info.obj, oname, formatter)
1195 elif meth == 'pinfo':
1210 elif meth == 'pinfo':
1196 pmethod(info.obj, oname, formatter, info, **kw)
1211 pmethod(info.obj, oname, formatter, info, **kw)
1197 else:
1212 else:
1198 pmethod(info.obj, oname)
1213 pmethod(info.obj, oname)
1199 else:
1214 else:
1200 print 'Object `%s` not found.' % oname
1215 print 'Object `%s` not found.' % oname
1201 return 'not found' # so callers can take other action
1216 return 'not found' # so callers can take other action
1202
1217
1203 def object_inspect(self, oname):
1218 def object_inspect(self, oname):
1204 info = self._object_find(oname)
1219 info = self._object_find(oname)
1205 if info.found:
1220 if info.found:
1206 return self.inspector.info(info.obj, oname, info=info)
1221 return self.inspector.info(info.obj, oname, info=info)
1207 else:
1222 else:
1208 return oinspect.object_info(name=oname, found=False)
1223 return oinspect.object_info(name=oname, found=False)
1209
1224
1210 #-------------------------------------------------------------------------
1225 #-------------------------------------------------------------------------
1211 # Things related to history management
1226 # Things related to history management
1212 #-------------------------------------------------------------------------
1227 #-------------------------------------------------------------------------
1213
1228
1214 def init_history(self):
1229 def init_history(self):
1215 self.history_manager = HistoryManager(shell=self)
1230 self.history_manager = HistoryManager(shell=self)
1216
1231
1217 def save_hist(self):
1232 def save_hist(self):
1218 """Save input history to a file (via readline library)."""
1233 """Save input history to a file (via readline library)."""
1219 self.history_manager.save_hist()
1234 self.history_manager.save_hist()
1220
1235
1221 # For backwards compatibility
1236 # For backwards compatibility
1222 savehist = save_hist
1237 savehist = save_hist
1223
1238
1224 def reload_hist(self):
1239 def reload_hist(self):
1225 """Reload the input history from disk file."""
1240 """Reload the input history from disk file."""
1226 self.history_manager.reload_hist()
1241 self.history_manager.reload_hist()
1227
1242
1228 # For backwards compatibility
1243 # For backwards compatibility
1229 reloadhist = reload_hist
1244 reloadhist = reload_hist
1230
1245
1231 def history_saving_wrapper(self, func):
1246 def history_saving_wrapper(self, func):
1232 """ Wrap func for readline history saving
1247 """ Wrap func for readline history saving
1233
1248
1234 Convert func into callable that saves & restores
1249 Convert func into callable that saves & restores
1235 history around the call """
1250 history around the call """
1236
1251
1237 if self.has_readline:
1252 if self.has_readline:
1238 from IPython.utils import rlineimpl as readline
1253 from IPython.utils import rlineimpl as readline
1239 else:
1254 else:
1240 return func
1255 return func
1241
1256
1242 def wrapper():
1257 def wrapper():
1243 self.save_hist()
1258 self.save_hist()
1244 try:
1259 try:
1245 func()
1260 func()
1246 finally:
1261 finally:
1247 readline.read_history_file(self.histfile)
1262 readline.read_history_file(self.histfile)
1248 return wrapper
1263 return wrapper
1249
1264
1250 #-------------------------------------------------------------------------
1265 #-------------------------------------------------------------------------
1251 # Things related to exception handling and tracebacks (not debugging)
1266 # Things related to exception handling and tracebacks (not debugging)
1252 #-------------------------------------------------------------------------
1267 #-------------------------------------------------------------------------
1253
1268
1254 def init_traceback_handlers(self, custom_exceptions):
1269 def init_traceback_handlers(self, custom_exceptions):
1255 # Syntax error handler.
1270 # Syntax error handler.
1256 self.SyntaxTB = ultratb.SyntaxTB(color_scheme='NoColor')
1271 self.SyntaxTB = ultratb.SyntaxTB(color_scheme='NoColor')
1257
1272
1258 # The interactive one is initialized with an offset, meaning we always
1273 # The interactive one is initialized with an offset, meaning we always
1259 # want to remove the topmost item in the traceback, which is our own
1274 # want to remove the topmost item in the traceback, which is our own
1260 # internal code. Valid modes: ['Plain','Context','Verbose']
1275 # internal code. Valid modes: ['Plain','Context','Verbose']
1261 self.InteractiveTB = ultratb.AutoFormattedTB(mode = 'Plain',
1276 self.InteractiveTB = ultratb.AutoFormattedTB(mode = 'Plain',
1262 color_scheme='NoColor',
1277 color_scheme='NoColor',
1263 tb_offset = 1)
1278 tb_offset = 1)
1264
1279
1265 # The instance will store a pointer to the system-wide exception hook,
1280 # The instance will store a pointer to the system-wide exception hook,
1266 # so that runtime code (such as magics) can access it. This is because
1281 # so that runtime code (such as magics) can access it. This is because
1267 # during the read-eval loop, it may get temporarily overwritten.
1282 # during the read-eval loop, it may get temporarily overwritten.
1268 self.sys_excepthook = sys.excepthook
1283 self.sys_excepthook = sys.excepthook
1269
1284
1270 # and add any custom exception handlers the user may have specified
1285 # and add any custom exception handlers the user may have specified
1271 self.set_custom_exc(*custom_exceptions)
1286 self.set_custom_exc(*custom_exceptions)
1272
1287
1273 # Set the exception mode
1288 # Set the exception mode
1274 self.InteractiveTB.set_mode(mode=self.xmode)
1289 self.InteractiveTB.set_mode(mode=self.xmode)
1275
1290
1276 def set_custom_exc(self, exc_tuple, handler):
1291 def set_custom_exc(self, exc_tuple, handler):
1277 """set_custom_exc(exc_tuple,handler)
1292 """set_custom_exc(exc_tuple,handler)
1278
1293
1279 Set a custom exception handler, which will be called if any of the
1294 Set a custom exception handler, which will be called if any of the
1280 exceptions in exc_tuple occur in the mainloop (specifically, in the
1295 exceptions in exc_tuple occur in the mainloop (specifically, in the
1281 run_code() method.
1296 run_code() method.
1282
1297
1283 Inputs:
1298 Inputs:
1284
1299
1285 - exc_tuple: a *tuple* of valid exceptions to call the defined
1300 - exc_tuple: a *tuple* of valid exceptions to call the defined
1286 handler for. It is very important that you use a tuple, and NOT A
1301 handler for. It is very important that you use a tuple, and NOT A
1287 LIST here, because of the way Python's except statement works. If
1302 LIST here, because of the way Python's except statement works. If
1288 you only want to trap a single exception, use a singleton tuple:
1303 you only want to trap a single exception, use a singleton tuple:
1289
1304
1290 exc_tuple == (MyCustomException,)
1305 exc_tuple == (MyCustomException,)
1291
1306
1292 - handler: this must be defined as a function with the following
1307 - handler: this must be defined as a function with the following
1293 basic interface::
1308 basic interface::
1294
1309
1295 def my_handler(self, etype, value, tb, tb_offset=None)
1310 def my_handler(self, etype, value, tb, tb_offset=None)
1296 ...
1311 ...
1297 # The return value must be
1312 # The return value must be
1298 return structured_traceback
1313 return structured_traceback
1299
1314
1300 This will be made into an instance method (via new.instancemethod)
1315 This will be made into an instance method (via new.instancemethod)
1301 of IPython itself, and it will be called if any of the exceptions
1316 of IPython itself, and it will be called if any of the exceptions
1302 listed in the exc_tuple are caught. If the handler is None, an
1317 listed in the exc_tuple are caught. If the handler is None, an
1303 internal basic one is used, which just prints basic info.
1318 internal basic one is used, which just prints basic info.
1304
1319
1305 WARNING: by putting in your own exception handler into IPython's main
1320 WARNING: by putting in your own exception handler into IPython's main
1306 execution loop, you run a very good chance of nasty crashes. This
1321 execution loop, you run a very good chance of nasty crashes. This
1307 facility should only be used if you really know what you are doing."""
1322 facility should only be used if you really know what you are doing."""
1308
1323
1309 assert type(exc_tuple)==type(()) , \
1324 assert type(exc_tuple)==type(()) , \
1310 "The custom exceptions must be given AS A TUPLE."
1325 "The custom exceptions must be given AS A TUPLE."
1311
1326
1312 def dummy_handler(self,etype,value,tb):
1327 def dummy_handler(self,etype,value,tb):
1313 print '*** Simple custom exception handler ***'
1328 print '*** Simple custom exception handler ***'
1314 print 'Exception type :',etype
1329 print 'Exception type :',etype
1315 print 'Exception value:',value
1330 print 'Exception value:',value
1316 print 'Traceback :',tb
1331 print 'Traceback :',tb
1317 print 'Source code :','\n'.join(self.buffer)
1332 print 'Source code :','\n'.join(self.buffer)
1318
1333
1319 if handler is None: handler = dummy_handler
1334 if handler is None: handler = dummy_handler
1320
1335
1321 self.CustomTB = new.instancemethod(handler,self,self.__class__)
1336 self.CustomTB = new.instancemethod(handler,self,self.__class__)
1322 self.custom_exceptions = exc_tuple
1337 self.custom_exceptions = exc_tuple
1323
1338
1324 def excepthook(self, etype, value, tb):
1339 def excepthook(self, etype, value, tb):
1325 """One more defense for GUI apps that call sys.excepthook.
1340 """One more defense for GUI apps that call sys.excepthook.
1326
1341
1327 GUI frameworks like wxPython trap exceptions and call
1342 GUI frameworks like wxPython trap exceptions and call
1328 sys.excepthook themselves. I guess this is a feature that
1343 sys.excepthook themselves. I guess this is a feature that
1329 enables them to keep running after exceptions that would
1344 enables them to keep running after exceptions that would
1330 otherwise kill their mainloop. This is a bother for IPython
1345 otherwise kill their mainloop. This is a bother for IPython
1331 which excepts to catch all of the program exceptions with a try:
1346 which excepts to catch all of the program exceptions with a try:
1332 except: statement.
1347 except: statement.
1333
1348
1334 Normally, IPython sets sys.excepthook to a CrashHandler instance, so if
1349 Normally, IPython sets sys.excepthook to a CrashHandler instance, so if
1335 any app directly invokes sys.excepthook, it will look to the user like
1350 any app directly invokes sys.excepthook, it will look to the user like
1336 IPython crashed. In order to work around this, we can disable the
1351 IPython crashed. In order to work around this, we can disable the
1337 CrashHandler and replace it with this excepthook instead, which prints a
1352 CrashHandler and replace it with this excepthook instead, which prints a
1338 regular traceback using our InteractiveTB. In this fashion, apps which
1353 regular traceback using our InteractiveTB. In this fashion, apps which
1339 call sys.excepthook will generate a regular-looking exception from
1354 call sys.excepthook will generate a regular-looking exception from
1340 IPython, and the CrashHandler will only be triggered by real IPython
1355 IPython, and the CrashHandler will only be triggered by real IPython
1341 crashes.
1356 crashes.
1342
1357
1343 This hook should be used sparingly, only in places which are not likely
1358 This hook should be used sparingly, only in places which are not likely
1344 to be true IPython errors.
1359 to be true IPython errors.
1345 """
1360 """
1346 self.showtraceback((etype,value,tb),tb_offset=0)
1361 self.showtraceback((etype,value,tb),tb_offset=0)
1347
1362
1348 def showtraceback(self,exc_tuple = None,filename=None,tb_offset=None,
1363 def showtraceback(self,exc_tuple = None,filename=None,tb_offset=None,
1349 exception_only=False):
1364 exception_only=False):
1350 """Display the exception that just occurred.
1365 """Display the exception that just occurred.
1351
1366
1352 If nothing is known about the exception, this is the method which
1367 If nothing is known about the exception, this is the method which
1353 should be used throughout the code for presenting user tracebacks,
1368 should be used throughout the code for presenting user tracebacks,
1354 rather than directly invoking the InteractiveTB object.
1369 rather than directly invoking the InteractiveTB object.
1355
1370
1356 A specific showsyntaxerror() also exists, but this method can take
1371 A specific showsyntaxerror() also exists, but this method can take
1357 care of calling it if needed, so unless you are explicitly catching a
1372 care of calling it if needed, so unless you are explicitly catching a
1358 SyntaxError exception, don't try to analyze the stack manually and
1373 SyntaxError exception, don't try to analyze the stack manually and
1359 simply call this method."""
1374 simply call this method."""
1360
1375
1361 try:
1376 try:
1362 if exc_tuple is None:
1377 if exc_tuple is None:
1363 etype, value, tb = sys.exc_info()
1378 etype, value, tb = sys.exc_info()
1364 else:
1379 else:
1365 etype, value, tb = exc_tuple
1380 etype, value, tb = exc_tuple
1366
1381
1367 if etype is None:
1382 if etype is None:
1368 if hasattr(sys, 'last_type'):
1383 if hasattr(sys, 'last_type'):
1369 etype, value, tb = sys.last_type, sys.last_value, \
1384 etype, value, tb = sys.last_type, sys.last_value, \
1370 sys.last_traceback
1385 sys.last_traceback
1371 else:
1386 else:
1372 self.write_err('No traceback available to show.\n')
1387 self.write_err('No traceback available to show.\n')
1373 return
1388 return
1374
1389
1375 if etype is SyntaxError:
1390 if etype is SyntaxError:
1376 # Though this won't be called by syntax errors in the input
1391 # Though this won't be called by syntax errors in the input
1377 # line, there may be SyntaxError cases whith imported code.
1392 # line, there may be SyntaxError cases whith imported code.
1378 self.showsyntaxerror(filename)
1393 self.showsyntaxerror(filename)
1379 elif etype is UsageError:
1394 elif etype is UsageError:
1380 print "UsageError:", value
1395 print "UsageError:", value
1381 else:
1396 else:
1382 # WARNING: these variables are somewhat deprecated and not
1397 # WARNING: these variables are somewhat deprecated and not
1383 # necessarily safe to use in a threaded environment, but tools
1398 # necessarily safe to use in a threaded environment, but tools
1384 # like pdb depend on their existence, so let's set them. If we
1399 # like pdb depend on their existence, so let's set them. If we
1385 # find problems in the field, we'll need to revisit their use.
1400 # find problems in the field, we'll need to revisit their use.
1386 sys.last_type = etype
1401 sys.last_type = etype
1387 sys.last_value = value
1402 sys.last_value = value
1388 sys.last_traceback = tb
1403 sys.last_traceback = tb
1389
1404
1390 if etype in self.custom_exceptions:
1405 if etype in self.custom_exceptions:
1391 # FIXME: Old custom traceback objects may just return a
1406 # FIXME: Old custom traceback objects may just return a
1392 # string, in that case we just put it into a list
1407 # string, in that case we just put it into a list
1393 stb = self.CustomTB(etype, value, tb, tb_offset)
1408 stb = self.CustomTB(etype, value, tb, tb_offset)
1394 if isinstance(ctb, basestring):
1409 if isinstance(ctb, basestring):
1395 stb = [stb]
1410 stb = [stb]
1396 else:
1411 else:
1397 if exception_only:
1412 if exception_only:
1398 stb = ['An exception has occurred, use %tb to see '
1413 stb = ['An exception has occurred, use %tb to see '
1399 'the full traceback.\n']
1414 'the full traceback.\n']
1400 stb.extend(self.InteractiveTB.get_exception_only(etype,
1415 stb.extend(self.InteractiveTB.get_exception_only(etype,
1401 value))
1416 value))
1402 else:
1417 else:
1403 stb = self.InteractiveTB.structured_traceback(etype,
1418 stb = self.InteractiveTB.structured_traceback(etype,
1404 value, tb, tb_offset=tb_offset)
1419 value, tb, tb_offset=tb_offset)
1405 # FIXME: the pdb calling should be done by us, not by
1420 # FIXME: the pdb calling should be done by us, not by
1406 # the code computing the traceback.
1421 # the code computing the traceback.
1407 if self.InteractiveTB.call_pdb:
1422 if self.InteractiveTB.call_pdb:
1408 # pdb mucks up readline, fix it back
1423 # pdb mucks up readline, fix it back
1409 self.set_readline_completer()
1424 self.set_readline_completer()
1410
1425
1411 # Actually show the traceback
1426 # Actually show the traceback
1412 self._showtraceback(etype, value, stb)
1427 self._showtraceback(etype, value, stb)
1413
1428
1414 except KeyboardInterrupt:
1429 except KeyboardInterrupt:
1415 self.write_err("\nKeyboardInterrupt\n")
1430 self.write_err("\nKeyboardInterrupt\n")
1416
1431
1417 def _showtraceback(self, etype, evalue, stb):
1432 def _showtraceback(self, etype, evalue, stb):
1418 """Actually show a traceback.
1433 """Actually show a traceback.
1419
1434
1420 Subclasses may override this method to put the traceback on a different
1435 Subclasses may override this method to put the traceback on a different
1421 place, like a side channel.
1436 place, like a side channel.
1422 """
1437 """
1423 print >> io.Term.cout, self.InteractiveTB.stb2text(stb)
1438 print >> io.Term.cout, self.InteractiveTB.stb2text(stb)
1424
1439
1425 def showsyntaxerror(self, filename=None):
1440 def showsyntaxerror(self, filename=None):
1426 """Display the syntax error that just occurred.
1441 """Display the syntax error that just occurred.
1427
1442
1428 This doesn't display a stack trace because there isn't one.
1443 This doesn't display a stack trace because there isn't one.
1429
1444
1430 If a filename is given, it is stuffed in the exception instead
1445 If a filename is given, it is stuffed in the exception instead
1431 of what was there before (because Python's parser always uses
1446 of what was there before (because Python's parser always uses
1432 "<string>" when reading from a string).
1447 "<string>" when reading from a string).
1433 """
1448 """
1434 etype, value, last_traceback = sys.exc_info()
1449 etype, value, last_traceback = sys.exc_info()
1435
1450
1436 # See note about these variables in showtraceback() above
1451 # See note about these variables in showtraceback() above
1437 sys.last_type = etype
1452 sys.last_type = etype
1438 sys.last_value = value
1453 sys.last_value = value
1439 sys.last_traceback = last_traceback
1454 sys.last_traceback = last_traceback
1440
1455
1441 if filename and etype is SyntaxError:
1456 if filename and etype is SyntaxError:
1442 # Work hard to stuff the correct filename in the exception
1457 # Work hard to stuff the correct filename in the exception
1443 try:
1458 try:
1444 msg, (dummy_filename, lineno, offset, line) = value
1459 msg, (dummy_filename, lineno, offset, line) = value
1445 except:
1460 except:
1446 # Not the format we expect; leave it alone
1461 # Not the format we expect; leave it alone
1447 pass
1462 pass
1448 else:
1463 else:
1449 # Stuff in the right filename
1464 # Stuff in the right filename
1450 try:
1465 try:
1451 # Assume SyntaxError is a class exception
1466 # Assume SyntaxError is a class exception
1452 value = SyntaxError(msg, (filename, lineno, offset, line))
1467 value = SyntaxError(msg, (filename, lineno, offset, line))
1453 except:
1468 except:
1454 # If that failed, assume SyntaxError is a string
1469 # If that failed, assume SyntaxError is a string
1455 value = msg, (filename, lineno, offset, line)
1470 value = msg, (filename, lineno, offset, line)
1456 stb = self.SyntaxTB.structured_traceback(etype, value, [])
1471 stb = self.SyntaxTB.structured_traceback(etype, value, [])
1457 self._showtraceback(etype, value, stb)
1472 self._showtraceback(etype, value, stb)
1458
1473
1459 #-------------------------------------------------------------------------
1474 #-------------------------------------------------------------------------
1460 # Things related to readline
1475 # Things related to readline
1461 #-------------------------------------------------------------------------
1476 #-------------------------------------------------------------------------
1462
1477
1463 def init_readline(self):
1478 def init_readline(self):
1464 """Command history completion/saving/reloading."""
1479 """Command history completion/saving/reloading."""
1465
1480
1466 if self.readline_use:
1481 if self.readline_use:
1467 import IPython.utils.rlineimpl as readline
1482 import IPython.utils.rlineimpl as readline
1468
1483
1469 self.rl_next_input = None
1484 self.rl_next_input = None
1470 self.rl_do_indent = False
1485 self.rl_do_indent = False
1471
1486
1472 if not self.readline_use or not readline.have_readline:
1487 if not self.readline_use or not readline.have_readline:
1473 self.has_readline = False
1488 self.has_readline = False
1474 self.readline = None
1489 self.readline = None
1475 # Set a number of methods that depend on readline to be no-op
1490 # Set a number of methods that depend on readline to be no-op
1476 self.save_hist = no_op
1491 self.save_hist = no_op
1477 self.reload_hist = no_op
1492 self.reload_hist = no_op
1478 self.set_readline_completer = no_op
1493 self.set_readline_completer = no_op
1479 self.set_custom_completer = no_op
1494 self.set_custom_completer = no_op
1480 self.set_completer_frame = no_op
1495 self.set_completer_frame = no_op
1481 warn('Readline services not available or not loaded.')
1496 warn('Readline services not available or not loaded.')
1482 else:
1497 else:
1483 self.has_readline = True
1498 self.has_readline = True
1484 self.readline = readline
1499 self.readline = readline
1485 sys.modules['readline'] = readline
1500 sys.modules['readline'] = readline
1486
1501
1487 # Platform-specific configuration
1502 # Platform-specific configuration
1488 if os.name == 'nt':
1503 if os.name == 'nt':
1489 # FIXME - check with Frederick to see if we can harmonize
1504 # FIXME - check with Frederick to see if we can harmonize
1490 # naming conventions with pyreadline to avoid this
1505 # naming conventions with pyreadline to avoid this
1491 # platform-dependent check
1506 # platform-dependent check
1492 self.readline_startup_hook = readline.set_pre_input_hook
1507 self.readline_startup_hook = readline.set_pre_input_hook
1493 else:
1508 else:
1494 self.readline_startup_hook = readline.set_startup_hook
1509 self.readline_startup_hook = readline.set_startup_hook
1495
1510
1496 # Load user's initrc file (readline config)
1511 # Load user's initrc file (readline config)
1497 # Or if libedit is used, load editrc.
1512 # Or if libedit is used, load editrc.
1498 inputrc_name = os.environ.get('INPUTRC')
1513 inputrc_name = os.environ.get('INPUTRC')
1499 if inputrc_name is None:
1514 if inputrc_name is None:
1500 home_dir = get_home_dir()
1515 home_dir = get_home_dir()
1501 if home_dir is not None:
1516 if home_dir is not None:
1502 inputrc_name = '.inputrc'
1517 inputrc_name = '.inputrc'
1503 if readline.uses_libedit:
1518 if readline.uses_libedit:
1504 inputrc_name = '.editrc'
1519 inputrc_name = '.editrc'
1505 inputrc_name = os.path.join(home_dir, inputrc_name)
1520 inputrc_name = os.path.join(home_dir, inputrc_name)
1506 if os.path.isfile(inputrc_name):
1521 if os.path.isfile(inputrc_name):
1507 try:
1522 try:
1508 readline.read_init_file(inputrc_name)
1523 readline.read_init_file(inputrc_name)
1509 except:
1524 except:
1510 warn('Problems reading readline initialization file <%s>'
1525 warn('Problems reading readline initialization file <%s>'
1511 % inputrc_name)
1526 % inputrc_name)
1512
1527
1513 # Configure readline according to user's prefs
1528 # Configure readline according to user's prefs
1514 # This is only done if GNU readline is being used. If libedit
1529 # This is only done if GNU readline is being used. If libedit
1515 # is being used (as on Leopard) the readline config is
1530 # is being used (as on Leopard) the readline config is
1516 # not run as the syntax for libedit is different.
1531 # not run as the syntax for libedit is different.
1517 if not readline.uses_libedit:
1532 if not readline.uses_libedit:
1518 for rlcommand in self.readline_parse_and_bind:
1533 for rlcommand in self.readline_parse_and_bind:
1519 #print "loading rl:",rlcommand # dbg
1534 #print "loading rl:",rlcommand # dbg
1520 readline.parse_and_bind(rlcommand)
1535 readline.parse_and_bind(rlcommand)
1521
1536
1522 # Remove some chars from the delimiters list. If we encounter
1537 # Remove some chars from the delimiters list. If we encounter
1523 # unicode chars, discard them.
1538 # unicode chars, discard them.
1524 delims = readline.get_completer_delims().encode("ascii", "ignore")
1539 delims = readline.get_completer_delims().encode("ascii", "ignore")
1525 delims = delims.translate(string._idmap,
1540 delims = delims.translate(string._idmap,
1526 self.readline_remove_delims)
1541 self.readline_remove_delims)
1527 delims = delims.replace(ESC_MAGIC, '')
1542 delims = delims.replace(ESC_MAGIC, '')
1528 readline.set_completer_delims(delims)
1543 readline.set_completer_delims(delims)
1529 # otherwise we end up with a monster history after a while:
1544 # otherwise we end up with a monster history after a while:
1530 readline.set_history_length(1000)
1545 readline.set_history_length(1000)
1531 try:
1546 try:
1532 #print '*** Reading readline history' # dbg
1547 #print '*** Reading readline history' # dbg
1533 readline.read_history_file(self.histfile)
1548 readline.read_history_file(self.histfile)
1534 except IOError:
1549 except IOError:
1535 pass # It doesn't exist yet.
1550 pass # It doesn't exist yet.
1536
1551
1537 # If we have readline, we want our history saved upon ipython
1552 # If we have readline, we want our history saved upon ipython
1538 # exiting.
1553 # exiting.
1539 atexit.register(self.save_hist)
1554 atexit.register(self.save_hist)
1540
1555
1541 # Configure auto-indent for all platforms
1556 # Configure auto-indent for all platforms
1542 self.set_autoindent(self.autoindent)
1557 self.set_autoindent(self.autoindent)
1543
1558
1544 def set_next_input(self, s):
1559 def set_next_input(self, s):
1545 """ Sets the 'default' input string for the next command line.
1560 """ Sets the 'default' input string for the next command line.
1546
1561
1547 Requires readline.
1562 Requires readline.
1548
1563
1549 Example:
1564 Example:
1550
1565
1551 [D:\ipython]|1> _ip.set_next_input("Hello Word")
1566 [D:\ipython]|1> _ip.set_next_input("Hello Word")
1552 [D:\ipython]|2> Hello Word_ # cursor is here
1567 [D:\ipython]|2> Hello Word_ # cursor is here
1553 """
1568 """
1554
1569
1555 self.rl_next_input = s
1570 self.rl_next_input = s
1556
1571
1557 # Maybe move this to the terminal subclass?
1572 # Maybe move this to the terminal subclass?
1558 def pre_readline(self):
1573 def pre_readline(self):
1559 """readline hook to be used at the start of each line.
1574 """readline hook to be used at the start of each line.
1560
1575
1561 Currently it handles auto-indent only."""
1576 Currently it handles auto-indent only."""
1562
1577
1563 if self.rl_do_indent:
1578 if self.rl_do_indent:
1564 self.readline.insert_text(self._indent_current_str())
1579 self.readline.insert_text(self._indent_current_str())
1565 if self.rl_next_input is not None:
1580 if self.rl_next_input is not None:
1566 self.readline.insert_text(self.rl_next_input)
1581 self.readline.insert_text(self.rl_next_input)
1567 self.rl_next_input = None
1582 self.rl_next_input = None
1568
1583
1569 def _indent_current_str(self):
1584 def _indent_current_str(self):
1570 """return the current level of indentation as a string"""
1585 """return the current level of indentation as a string"""
1571 return self.input_splitter.indent_spaces * ' '
1586 return self.input_splitter.indent_spaces * ' '
1572
1587
1573 #-------------------------------------------------------------------------
1588 #-------------------------------------------------------------------------
1574 # Things related to text completion
1589 # Things related to text completion
1575 #-------------------------------------------------------------------------
1590 #-------------------------------------------------------------------------
1576
1591
1577 def init_completer(self):
1592 def init_completer(self):
1578 """Initialize the completion machinery.
1593 """Initialize the completion machinery.
1579
1594
1580 This creates completion machinery that can be used by client code,
1595 This creates completion machinery that can be used by client code,
1581 either interactively in-process (typically triggered by the readline
1596 either interactively in-process (typically triggered by the readline
1582 library), programatically (such as in test suites) or out-of-prcess
1597 library), programatically (such as in test suites) or out-of-prcess
1583 (typically over the network by remote frontends).
1598 (typically over the network by remote frontends).
1584 """
1599 """
1585 from IPython.core.completer import IPCompleter
1600 from IPython.core.completer import IPCompleter
1586 from IPython.core.completerlib import (module_completer,
1601 from IPython.core.completerlib import (module_completer,
1587 magic_run_completer, cd_completer)
1602 magic_run_completer, cd_completer)
1588
1603
1589 self.Completer = IPCompleter(self,
1604 self.Completer = IPCompleter(self,
1590 self.user_ns,
1605 self.user_ns,
1591 self.user_global_ns,
1606 self.user_global_ns,
1592 self.readline_omit__names,
1607 self.readline_omit__names,
1593 self.alias_manager.alias_table,
1608 self.alias_manager.alias_table,
1594 self.has_readline)
1609 self.has_readline)
1595
1610
1596 # Add custom completers to the basic ones built into IPCompleter
1611 # Add custom completers to the basic ones built into IPCompleter
1597 sdisp = self.strdispatchers.get('complete_command', StrDispatch())
1612 sdisp = self.strdispatchers.get('complete_command', StrDispatch())
1598 self.strdispatchers['complete_command'] = sdisp
1613 self.strdispatchers['complete_command'] = sdisp
1599 self.Completer.custom_completers = sdisp
1614 self.Completer.custom_completers = sdisp
1600
1615
1601 self.set_hook('complete_command', module_completer, str_key = 'import')
1616 self.set_hook('complete_command', module_completer, str_key = 'import')
1602 self.set_hook('complete_command', module_completer, str_key = 'from')
1617 self.set_hook('complete_command', module_completer, str_key = 'from')
1603 self.set_hook('complete_command', magic_run_completer, str_key = '%run')
1618 self.set_hook('complete_command', magic_run_completer, str_key = '%run')
1604 self.set_hook('complete_command', cd_completer, str_key = '%cd')
1619 self.set_hook('complete_command', cd_completer, str_key = '%cd')
1605
1620
1606 # Only configure readline if we truly are using readline. IPython can
1621 # Only configure readline if we truly are using readline. IPython can
1607 # do tab-completion over the network, in GUIs, etc, where readline
1622 # do tab-completion over the network, in GUIs, etc, where readline
1608 # itself may be absent
1623 # itself may be absent
1609 if self.has_readline:
1624 if self.has_readline:
1610 self.set_readline_completer()
1625 self.set_readline_completer()
1611
1626
1612 def complete(self, text, line=None, cursor_pos=None):
1627 def complete(self, text, line=None, cursor_pos=None):
1613 """Return the completed text and a list of completions.
1628 """Return the completed text and a list of completions.
1614
1629
1615 Parameters
1630 Parameters
1616 ----------
1631 ----------
1617
1632
1618 text : string
1633 text : string
1619 A string of text to be completed on. It can be given as empty and
1634 A string of text to be completed on. It can be given as empty and
1620 instead a line/position pair are given. In this case, the
1635 instead a line/position pair are given. In this case, the
1621 completer itself will split the line like readline does.
1636 completer itself will split the line like readline does.
1622
1637
1623 line : string, optional
1638 line : string, optional
1624 The complete line that text is part of.
1639 The complete line that text is part of.
1625
1640
1626 cursor_pos : int, optional
1641 cursor_pos : int, optional
1627 The position of the cursor on the input line.
1642 The position of the cursor on the input line.
1628
1643
1629 Returns
1644 Returns
1630 -------
1645 -------
1631 text : string
1646 text : string
1632 The actual text that was completed.
1647 The actual text that was completed.
1633
1648
1634 matches : list
1649 matches : list
1635 A sorted list with all possible completions.
1650 A sorted list with all possible completions.
1636
1651
1637 The optional arguments allow the completion to take more context into
1652 The optional arguments allow the completion to take more context into
1638 account, and are part of the low-level completion API.
1653 account, and are part of the low-level completion API.
1639
1654
1640 This is a wrapper around the completion mechanism, similar to what
1655 This is a wrapper around the completion mechanism, similar to what
1641 readline does at the command line when the TAB key is hit. By
1656 readline does at the command line when the TAB key is hit. By
1642 exposing it as a method, it can be used by other non-readline
1657 exposing it as a method, it can be used by other non-readline
1643 environments (such as GUIs) for text completion.
1658 environments (such as GUIs) for text completion.
1644
1659
1645 Simple usage example:
1660 Simple usage example:
1646
1661
1647 In [1]: x = 'hello'
1662 In [1]: x = 'hello'
1648
1663
1649 In [2]: _ip.complete('x.l')
1664 In [2]: _ip.complete('x.l')
1650 Out[2]: ('x.l', ['x.ljust', 'x.lower', 'x.lstrip'])
1665 Out[2]: ('x.l', ['x.ljust', 'x.lower', 'x.lstrip'])
1651 """
1666 """
1652
1667
1653 # Inject names into __builtin__ so we can complete on the added names.
1668 # Inject names into __builtin__ so we can complete on the added names.
1654 with self.builtin_trap:
1669 with self.builtin_trap:
1655 return self.Completer.complete(text, line, cursor_pos)
1670 return self.Completer.complete(text, line, cursor_pos)
1656
1671
1657 def set_custom_completer(self, completer, pos=0):
1672 def set_custom_completer(self, completer, pos=0):
1658 """Adds a new custom completer function.
1673 """Adds a new custom completer function.
1659
1674
1660 The position argument (defaults to 0) is the index in the completers
1675 The position argument (defaults to 0) is the index in the completers
1661 list where you want the completer to be inserted."""
1676 list where you want the completer to be inserted."""
1662
1677
1663 newcomp = new.instancemethod(completer,self.Completer,
1678 newcomp = new.instancemethod(completer,self.Completer,
1664 self.Completer.__class__)
1679 self.Completer.__class__)
1665 self.Completer.matchers.insert(pos,newcomp)
1680 self.Completer.matchers.insert(pos,newcomp)
1666
1681
1667 def set_readline_completer(self):
1682 def set_readline_completer(self):
1668 """Reset readline's completer to be our own."""
1683 """Reset readline's completer to be our own."""
1669 self.readline.set_completer(self.Completer.rlcomplete)
1684 self.readline.set_completer(self.Completer.rlcomplete)
1670
1685
1671 def set_completer_frame(self, frame=None):
1686 def set_completer_frame(self, frame=None):
1672 """Set the frame of the completer."""
1687 """Set the frame of the completer."""
1673 if frame:
1688 if frame:
1674 self.Completer.namespace = frame.f_locals
1689 self.Completer.namespace = frame.f_locals
1675 self.Completer.global_namespace = frame.f_globals
1690 self.Completer.global_namespace = frame.f_globals
1676 else:
1691 else:
1677 self.Completer.namespace = self.user_ns
1692 self.Completer.namespace = self.user_ns
1678 self.Completer.global_namespace = self.user_global_ns
1693 self.Completer.global_namespace = self.user_global_ns
1679
1694
1680 #-------------------------------------------------------------------------
1695 #-------------------------------------------------------------------------
1681 # Things related to magics
1696 # Things related to magics
1682 #-------------------------------------------------------------------------
1697 #-------------------------------------------------------------------------
1683
1698
1684 def init_magics(self):
1699 def init_magics(self):
1685 # FIXME: Move the color initialization to the DisplayHook, which
1700 # FIXME: Move the color initialization to the DisplayHook, which
1686 # should be split into a prompt manager and displayhook. We probably
1701 # should be split into a prompt manager and displayhook. We probably
1687 # even need a centralize colors management object.
1702 # even need a centralize colors management object.
1688 self.magic_colors(self.colors)
1703 self.magic_colors(self.colors)
1689 # History was moved to a separate module
1704 # History was moved to a separate module
1690 from . import history
1705 from . import history
1691 history.init_ipython(self)
1706 history.init_ipython(self)
1692
1707
1693 def magic(self,arg_s):
1708 def magic(self,arg_s):
1694 """Call a magic function by name.
1709 """Call a magic function by name.
1695
1710
1696 Input: a string containing the name of the magic function to call and
1711 Input: a string containing the name of the magic function to call and
1697 any additional arguments to be passed to the magic.
1712 any additional arguments to be passed to the magic.
1698
1713
1699 magic('name -opt foo bar') is equivalent to typing at the ipython
1714 magic('name -opt foo bar') is equivalent to typing at the ipython
1700 prompt:
1715 prompt:
1701
1716
1702 In[1]: %name -opt foo bar
1717 In[1]: %name -opt foo bar
1703
1718
1704 To call a magic without arguments, simply use magic('name').
1719 To call a magic without arguments, simply use magic('name').
1705
1720
1706 This provides a proper Python function to call IPython's magics in any
1721 This provides a proper Python function to call IPython's magics in any
1707 valid Python code you can type at the interpreter, including loops and
1722 valid Python code you can type at the interpreter, including loops and
1708 compound statements.
1723 compound statements.
1709 """
1724 """
1710 args = arg_s.split(' ',1)
1725 args = arg_s.split(' ',1)
1711 magic_name = args[0]
1726 magic_name = args[0]
1712 magic_name = magic_name.lstrip(prefilter.ESC_MAGIC)
1727 magic_name = magic_name.lstrip(prefilter.ESC_MAGIC)
1713
1728
1714 try:
1729 try:
1715 magic_args = args[1]
1730 magic_args = args[1]
1716 except IndexError:
1731 except IndexError:
1717 magic_args = ''
1732 magic_args = ''
1718 fn = getattr(self,'magic_'+magic_name,None)
1733 fn = getattr(self,'magic_'+magic_name,None)
1719 if fn is None:
1734 if fn is None:
1720 error("Magic function `%s` not found." % magic_name)
1735 error("Magic function `%s` not found." % magic_name)
1721 else:
1736 else:
1722 magic_args = self.var_expand(magic_args,1)
1737 magic_args = self.var_expand(magic_args,1)
1723 with nested(self.builtin_trap,):
1738 with nested(self.builtin_trap,):
1724 result = fn(magic_args)
1739 result = fn(magic_args)
1725 return result
1740 return result
1726
1741
1727 def define_magic(self, magicname, func):
1742 def define_magic(self, magicname, func):
1728 """Expose own function as magic function for ipython
1743 """Expose own function as magic function for ipython
1729
1744
1730 def foo_impl(self,parameter_s=''):
1745 def foo_impl(self,parameter_s=''):
1731 'My very own magic!. (Use docstrings, IPython reads them).'
1746 'My very own magic!. (Use docstrings, IPython reads them).'
1732 print 'Magic function. Passed parameter is between < >:'
1747 print 'Magic function. Passed parameter is between < >:'
1733 print '<%s>' % parameter_s
1748 print '<%s>' % parameter_s
1734 print 'The self object is:',self
1749 print 'The self object is:',self
1735
1750
1736 self.define_magic('foo',foo_impl)
1751 self.define_magic('foo',foo_impl)
1737 """
1752 """
1738
1753
1739 import new
1754 import new
1740 im = new.instancemethod(func,self, self.__class__)
1755 im = new.instancemethod(func,self, self.__class__)
1741 old = getattr(self, "magic_" + magicname, None)
1756 old = getattr(self, "magic_" + magicname, None)
1742 setattr(self, "magic_" + magicname, im)
1757 setattr(self, "magic_" + magicname, im)
1743 return old
1758 return old
1744
1759
1745 #-------------------------------------------------------------------------
1760 #-------------------------------------------------------------------------
1746 # Things related to macros
1761 # Things related to macros
1747 #-------------------------------------------------------------------------
1762 #-------------------------------------------------------------------------
1748
1763
1749 def define_macro(self, name, themacro):
1764 def define_macro(self, name, themacro):
1750 """Define a new macro
1765 """Define a new macro
1751
1766
1752 Parameters
1767 Parameters
1753 ----------
1768 ----------
1754 name : str
1769 name : str
1755 The name of the macro.
1770 The name of the macro.
1756 themacro : str or Macro
1771 themacro : str or Macro
1757 The action to do upon invoking the macro. If a string, a new
1772 The action to do upon invoking the macro. If a string, a new
1758 Macro object is created by passing the string to it.
1773 Macro object is created by passing the string to it.
1759 """
1774 """
1760
1775
1761 from IPython.core import macro
1776 from IPython.core import macro
1762
1777
1763 if isinstance(themacro, basestring):
1778 if isinstance(themacro, basestring):
1764 themacro = macro.Macro(themacro)
1779 themacro = macro.Macro(themacro)
1765 if not isinstance(themacro, macro.Macro):
1780 if not isinstance(themacro, macro.Macro):
1766 raise ValueError('A macro must be a string or a Macro instance.')
1781 raise ValueError('A macro must be a string or a Macro instance.')
1767 self.user_ns[name] = themacro
1782 self.user_ns[name] = themacro
1768
1783
1769 #-------------------------------------------------------------------------
1784 #-------------------------------------------------------------------------
1770 # Things related to the running of system commands
1785 # Things related to the running of system commands
1771 #-------------------------------------------------------------------------
1786 #-------------------------------------------------------------------------
1772
1787
1773 def system(self, cmd):
1788 def system(self, cmd):
1774 """Call the given cmd in a subprocess.
1789 """Call the given cmd in a subprocess.
1775
1790
1776 Parameters
1791 Parameters
1777 ----------
1792 ----------
1778 cmd : str
1793 cmd : str
1779 Command to execute (can not end in '&', as bacground processes are
1794 Command to execute (can not end in '&', as bacground processes are
1780 not supported.
1795 not supported.
1781 """
1796 """
1782 # We do not support backgrounding processes because we either use
1797 # We do not support backgrounding processes because we either use
1783 # pexpect or pipes to read from. Users can always just call
1798 # pexpect or pipes to read from. Users can always just call
1784 # os.system() if they really want a background process.
1799 # os.system() if they really want a background process.
1785 if cmd.endswith('&'):
1800 if cmd.endswith('&'):
1786 raise OSError("Background processes not supported.")
1801 raise OSError("Background processes not supported.")
1787
1802
1788 return system(self.var_expand(cmd, depth=2))
1803 return system(self.var_expand(cmd, depth=2))
1789
1804
1790 def getoutput(self, cmd, split=True):
1805 def getoutput(self, cmd, split=True):
1791 """Get output (possibly including stderr) from a subprocess.
1806 """Get output (possibly including stderr) from a subprocess.
1792
1807
1793 Parameters
1808 Parameters
1794 ----------
1809 ----------
1795 cmd : str
1810 cmd : str
1796 Command to execute (can not end in '&', as background processes are
1811 Command to execute (can not end in '&', as background processes are
1797 not supported.
1812 not supported.
1798 split : bool, optional
1813 split : bool, optional
1799
1814
1800 If True, split the output into an IPython SList. Otherwise, an
1815 If True, split the output into an IPython SList. Otherwise, an
1801 IPython LSString is returned. These are objects similar to normal
1816 IPython LSString is returned. These are objects similar to normal
1802 lists and strings, with a few convenience attributes for easier
1817 lists and strings, with a few convenience attributes for easier
1803 manipulation of line-based output. You can use '?' on them for
1818 manipulation of line-based output. You can use '?' on them for
1804 details.
1819 details.
1805 """
1820 """
1806 if cmd.endswith('&'):
1821 if cmd.endswith('&'):
1807 raise OSError("Background processes not supported.")
1822 raise OSError("Background processes not supported.")
1808 out = getoutput(self.var_expand(cmd, depth=2))
1823 out = getoutput(self.var_expand(cmd, depth=2))
1809 if split:
1824 if split:
1810 out = SList(out.splitlines())
1825 out = SList(out.splitlines())
1811 else:
1826 else:
1812 out = LSString(out)
1827 out = LSString(out)
1813 return out
1828 return out
1814
1829
1815 #-------------------------------------------------------------------------
1830 #-------------------------------------------------------------------------
1816 # Things related to aliases
1831 # Things related to aliases
1817 #-------------------------------------------------------------------------
1832 #-------------------------------------------------------------------------
1818
1833
1819 def init_alias(self):
1834 def init_alias(self):
1820 self.alias_manager = AliasManager(shell=self, config=self.config)
1835 self.alias_manager = AliasManager(shell=self, config=self.config)
1821 self.ns_table['alias'] = self.alias_manager.alias_table,
1836 self.ns_table['alias'] = self.alias_manager.alias_table,
1822
1837
1823 #-------------------------------------------------------------------------
1838 #-------------------------------------------------------------------------
1824 # Things related to extensions and plugins
1839 # Things related to extensions and plugins
1825 #-------------------------------------------------------------------------
1840 #-------------------------------------------------------------------------
1826
1841
1827 def init_extension_manager(self):
1842 def init_extension_manager(self):
1828 self.extension_manager = ExtensionManager(shell=self, config=self.config)
1843 self.extension_manager = ExtensionManager(shell=self, config=self.config)
1829
1844
1830 def init_plugin_manager(self):
1845 def init_plugin_manager(self):
1831 self.plugin_manager = PluginManager(config=self.config)
1846 self.plugin_manager = PluginManager(config=self.config)
1832
1847
1833 #-------------------------------------------------------------------------
1848 #-------------------------------------------------------------------------
1834 # Things related to payloads
1849 # Things related to payloads
1835 #-------------------------------------------------------------------------
1850 #-------------------------------------------------------------------------
1836
1851
1837 def init_payload(self):
1852 def init_payload(self):
1838 self.payload_manager = PayloadManager(config=self.config)
1853 self.payload_manager = PayloadManager(config=self.config)
1839
1854
1840 #-------------------------------------------------------------------------
1855 #-------------------------------------------------------------------------
1841 # Things related to the prefilter
1856 # Things related to the prefilter
1842 #-------------------------------------------------------------------------
1857 #-------------------------------------------------------------------------
1843
1858
1844 def init_prefilter(self):
1859 def init_prefilter(self):
1845 self.prefilter_manager = PrefilterManager(shell=self, config=self.config)
1860 self.prefilter_manager = PrefilterManager(shell=self, config=self.config)
1846 # Ultimately this will be refactored in the new interpreter code, but
1861 # Ultimately this will be refactored in the new interpreter code, but
1847 # for now, we should expose the main prefilter method (there's legacy
1862 # for now, we should expose the main prefilter method (there's legacy
1848 # code out there that may rely on this).
1863 # code out there that may rely on this).
1849 self.prefilter = self.prefilter_manager.prefilter_lines
1864 self.prefilter = self.prefilter_manager.prefilter_lines
1850
1865
1851 def auto_rewrite_input(self, cmd):
1866 def auto_rewrite_input(self, cmd):
1852 """Print to the screen the rewritten form of the user's command.
1867 """Print to the screen the rewritten form of the user's command.
1853
1868
1854 This shows visual feedback by rewriting input lines that cause
1869 This shows visual feedback by rewriting input lines that cause
1855 automatic calling to kick in, like::
1870 automatic calling to kick in, like::
1856
1871
1857 /f x
1872 /f x
1858
1873
1859 into::
1874 into::
1860
1875
1861 ------> f(x)
1876 ------> f(x)
1862
1877
1863 after the user's input prompt. This helps the user understand that the
1878 after the user's input prompt. This helps the user understand that the
1864 input line was transformed automatically by IPython.
1879 input line was transformed automatically by IPython.
1865 """
1880 """
1866 rw = self.displayhook.prompt1.auto_rewrite() + cmd
1881 rw = self.displayhook.prompt1.auto_rewrite() + cmd
1867
1882
1868 try:
1883 try:
1869 # plain ascii works better w/ pyreadline, on some machines, so
1884 # plain ascii works better w/ pyreadline, on some machines, so
1870 # we use it and only print uncolored rewrite if we have unicode
1885 # we use it and only print uncolored rewrite if we have unicode
1871 rw = str(rw)
1886 rw = str(rw)
1872 print >> IPython.utils.io.Term.cout, rw
1887 print >> IPython.utils.io.Term.cout, rw
1873 except UnicodeEncodeError:
1888 except UnicodeEncodeError:
1874 print "------> " + cmd
1889 print "------> " + cmd
1875
1890
1876 #-------------------------------------------------------------------------
1891 #-------------------------------------------------------------------------
1877 # Things related to extracting values/expressions from kernel and user_ns
1892 # Things related to extracting values/expressions from kernel and user_ns
1878 #-------------------------------------------------------------------------
1893 #-------------------------------------------------------------------------
1879
1894
1880 def _simple_error(self):
1895 def _simple_error(self):
1881 etype, value = sys.exc_info()[:2]
1896 etype, value = sys.exc_info()[:2]
1882 return u'[ERROR] {e.__name__}: {v}'.format(e=etype, v=value)
1897 return u'[ERROR] {e.__name__}: {v}'.format(e=etype, v=value)
1883
1898
1884 def user_variables(self, names):
1899 def user_variables(self, names):
1885 """Get a list of variable names from the user's namespace.
1900 """Get a list of variable names from the user's namespace.
1886
1901
1887 Parameters
1902 Parameters
1888 ----------
1903 ----------
1889 names : list of strings
1904 names : list of strings
1890 A list of names of variables to be read from the user namespace.
1905 A list of names of variables to be read from the user namespace.
1891
1906
1892 Returns
1907 Returns
1893 -------
1908 -------
1894 A dict, keyed by the input names and with the repr() of each value.
1909 A dict, keyed by the input names and with the repr() of each value.
1895 """
1910 """
1896 out = {}
1911 out = {}
1897 user_ns = self.user_ns
1912 user_ns = self.user_ns
1898 for varname in names:
1913 for varname in names:
1899 try:
1914 try:
1900 value = repr(user_ns[varname])
1915 value = repr(user_ns[varname])
1901 except:
1916 except:
1902 value = self._simple_error()
1917 value = self._simple_error()
1903 out[varname] = value
1918 out[varname] = value
1904 return out
1919 return out
1905
1920
1906 def user_expressions(self, expressions):
1921 def user_expressions(self, expressions):
1907 """Evaluate a dict of expressions in the user's namespace.
1922 """Evaluate a dict of expressions in the user's namespace.
1908
1923
1909 Parameters
1924 Parameters
1910 ----------
1925 ----------
1911 expressions : dict
1926 expressions : dict
1912 A dict with string keys and string values. The expression values
1927 A dict with string keys and string values. The expression values
1913 should be valid Python expressions, each of which will be evaluated
1928 should be valid Python expressions, each of which will be evaluated
1914 in the user namespace.
1929 in the user namespace.
1915
1930
1916 Returns
1931 Returns
1917 -------
1932 -------
1918 A dict, keyed like the input expressions dict, with the repr() of each
1933 A dict, keyed like the input expressions dict, with the repr() of each
1919 value.
1934 value.
1920 """
1935 """
1921 out = {}
1936 out = {}
1922 user_ns = self.user_ns
1937 user_ns = self.user_ns
1923 global_ns = self.user_global_ns
1938 global_ns = self.user_global_ns
1924 for key, expr in expressions.iteritems():
1939 for key, expr in expressions.iteritems():
1925 try:
1940 try:
1926 value = repr(eval(expr, global_ns, user_ns))
1941 value = repr(eval(expr, global_ns, user_ns))
1927 except:
1942 except:
1928 value = self._simple_error()
1943 value = self._simple_error()
1929 out[key] = value
1944 out[key] = value
1930 return out
1945 return out
1931
1946
1932 #-------------------------------------------------------------------------
1947 #-------------------------------------------------------------------------
1933 # Things related to the running of code
1948 # Things related to the running of code
1934 #-------------------------------------------------------------------------
1949 #-------------------------------------------------------------------------
1935
1950
1936 def ex(self, cmd):
1951 def ex(self, cmd):
1937 """Execute a normal python statement in user namespace."""
1952 """Execute a normal python statement in user namespace."""
1938 with nested(self.builtin_trap,):
1953 with nested(self.builtin_trap,):
1939 exec cmd in self.user_global_ns, self.user_ns
1954 exec cmd in self.user_global_ns, self.user_ns
1940
1955
1941 def ev(self, expr):
1956 def ev(self, expr):
1942 """Evaluate python expression expr in user namespace.
1957 """Evaluate python expression expr in user namespace.
1943
1958
1944 Returns the result of evaluation
1959 Returns the result of evaluation
1945 """
1960 """
1946 with nested(self.builtin_trap,):
1961 with nested(self.builtin_trap,):
1947 return eval(expr, self.user_global_ns, self.user_ns)
1962 return eval(expr, self.user_global_ns, self.user_ns)
1948
1963
1949 def safe_execfile(self, fname, *where, **kw):
1964 def safe_execfile(self, fname, *where, **kw):
1950 """A safe version of the builtin execfile().
1965 """A safe version of the builtin execfile().
1951
1966
1952 This version will never throw an exception, but instead print
1967 This version will never throw an exception, but instead print
1953 helpful error messages to the screen. This only works on pure
1968 helpful error messages to the screen. This only works on pure
1954 Python files with the .py extension.
1969 Python files with the .py extension.
1955
1970
1956 Parameters
1971 Parameters
1957 ----------
1972 ----------
1958 fname : string
1973 fname : string
1959 The name of the file to be executed.
1974 The name of the file to be executed.
1960 where : tuple
1975 where : tuple
1961 One or two namespaces, passed to execfile() as (globals,locals).
1976 One or two namespaces, passed to execfile() as (globals,locals).
1962 If only one is given, it is passed as both.
1977 If only one is given, it is passed as both.
1963 exit_ignore : bool (False)
1978 exit_ignore : bool (False)
1964 If True, then silence SystemExit for non-zero status (it is always
1979 If True, then silence SystemExit for non-zero status (it is always
1965 silenced for zero status, as it is so common).
1980 silenced for zero status, as it is so common).
1966 """
1981 """
1967 kw.setdefault('exit_ignore', False)
1982 kw.setdefault('exit_ignore', False)
1968
1983
1969 fname = os.path.abspath(os.path.expanduser(fname))
1984 fname = os.path.abspath(os.path.expanduser(fname))
1970
1985
1971 # Make sure we have a .py file
1986 # Make sure we have a .py file
1972 if not fname.endswith('.py'):
1987 if not fname.endswith('.py'):
1973 warn('File must end with .py to be run using execfile: <%s>' % fname)
1988 warn('File must end with .py to be run using execfile: <%s>' % fname)
1974
1989
1975 # Make sure we can open the file
1990 # Make sure we can open the file
1976 try:
1991 try:
1977 with open(fname) as thefile:
1992 with open(fname) as thefile:
1978 pass
1993 pass
1979 except:
1994 except:
1980 warn('Could not open file <%s> for safe execution.' % fname)
1995 warn('Could not open file <%s> for safe execution.' % fname)
1981 return
1996 return
1982
1997
1983 # Find things also in current directory. This is needed to mimic the
1998 # Find things also in current directory. This is needed to mimic the
1984 # behavior of running a script from the system command line, where
1999 # behavior of running a script from the system command line, where
1985 # Python inserts the script's directory into sys.path
2000 # Python inserts the script's directory into sys.path
1986 dname = os.path.dirname(fname)
2001 dname = os.path.dirname(fname)
1987
2002
1988 with prepended_to_syspath(dname):
2003 with prepended_to_syspath(dname):
1989 try:
2004 try:
1990 execfile(fname,*where)
2005 execfile(fname,*where)
1991 except SystemExit, status:
2006 except SystemExit, status:
1992 # If the call was made with 0 or None exit status (sys.exit(0)
2007 # If the call was made with 0 or None exit status (sys.exit(0)
1993 # or sys.exit() ), don't bother showing a traceback, as both of
2008 # or sys.exit() ), don't bother showing a traceback, as both of
1994 # these are considered normal by the OS:
2009 # these are considered normal by the OS:
1995 # > python -c'import sys;sys.exit(0)'; echo $?
2010 # > python -c'import sys;sys.exit(0)'; echo $?
1996 # 0
2011 # 0
1997 # > python -c'import sys;sys.exit()'; echo $?
2012 # > python -c'import sys;sys.exit()'; echo $?
1998 # 0
2013 # 0
1999 # For other exit status, we show the exception unless
2014 # For other exit status, we show the exception unless
2000 # explicitly silenced, but only in short form.
2015 # explicitly silenced, but only in short form.
2001 if status.code not in (0, None) and not kw['exit_ignore']:
2016 if status.code not in (0, None) and not kw['exit_ignore']:
2002 self.showtraceback(exception_only=True)
2017 self.showtraceback(exception_only=True)
2003 except:
2018 except:
2004 self.showtraceback()
2019 self.showtraceback()
2005
2020
2006 def safe_execfile_ipy(self, fname):
2021 def safe_execfile_ipy(self, fname):
2007 """Like safe_execfile, but for .ipy files with IPython syntax.
2022 """Like safe_execfile, but for .ipy files with IPython syntax.
2008
2023
2009 Parameters
2024 Parameters
2010 ----------
2025 ----------
2011 fname : str
2026 fname : str
2012 The name of the file to execute. The filename must have a
2027 The name of the file to execute. The filename must have a
2013 .ipy extension.
2028 .ipy extension.
2014 """
2029 """
2015 fname = os.path.abspath(os.path.expanduser(fname))
2030 fname = os.path.abspath(os.path.expanduser(fname))
2016
2031
2017 # Make sure we have a .py file
2032 # Make sure we have a .py file
2018 if not fname.endswith('.ipy'):
2033 if not fname.endswith('.ipy'):
2019 warn('File must end with .py to be run using execfile: <%s>' % fname)
2034 warn('File must end with .py to be run using execfile: <%s>' % fname)
2020
2035
2021 # Make sure we can open the file
2036 # Make sure we can open the file
2022 try:
2037 try:
2023 with open(fname) as thefile:
2038 with open(fname) as thefile:
2024 pass
2039 pass
2025 except:
2040 except:
2026 warn('Could not open file <%s> for safe execution.' % fname)
2041 warn('Could not open file <%s> for safe execution.' % fname)
2027 return
2042 return
2028
2043
2029 # Find things also in current directory. This is needed to mimic the
2044 # Find things also in current directory. This is needed to mimic the
2030 # behavior of running a script from the system command line, where
2045 # behavior of running a script from the system command line, where
2031 # Python inserts the script's directory into sys.path
2046 # Python inserts the script's directory into sys.path
2032 dname = os.path.dirname(fname)
2047 dname = os.path.dirname(fname)
2033
2048
2034 with prepended_to_syspath(dname):
2049 with prepended_to_syspath(dname):
2035 try:
2050 try:
2036 with open(fname) as thefile:
2051 with open(fname) as thefile:
2037 # self.run_cell currently captures all exceptions
2052 # self.run_cell currently captures all exceptions
2038 # raised in user code. It would be nice if there were
2053 # raised in user code. It would be nice if there were
2039 # versions of runlines, execfile that did raise, so
2054 # versions of runlines, execfile that did raise, so
2040 # we could catch the errors.
2055 # we could catch the errors.
2041 self.run_cell(thefile.read())
2056 self.run_cell(thefile.read())
2042 except:
2057 except:
2043 self.showtraceback()
2058 self.showtraceback()
2044 warn('Unknown failure executing file: <%s>' % fname)
2059 warn('Unknown failure executing file: <%s>' % fname)
2045
2060
2046 def run_cell(self, cell):
2061 def run_cell(self, cell):
2047 """Run the contents of an entire multiline 'cell' of code.
2062 """Run the contents of an entire multiline 'cell' of code.
2048
2063
2049 The cell is split into separate blocks which can be executed
2064 The cell is split into separate blocks which can be executed
2050 individually. Then, based on how many blocks there are, they are
2065 individually. Then, based on how many blocks there are, they are
2051 executed as follows:
2066 executed as follows:
2052
2067
2053 - A single block: 'single' mode.
2068 - A single block: 'single' mode.
2054
2069
2055 If there's more than one block, it depends:
2070 If there's more than one block, it depends:
2056
2071
2057 - if the last one is no more than two lines long, run all but the last
2072 - if the last one is no more than two lines long, run all but the last
2058 in 'exec' mode and the very last one in 'single' mode. This makes it
2073 in 'exec' mode and the very last one in 'single' mode. This makes it
2059 easy to type simple expressions at the end to see computed values. -
2074 easy to type simple expressions at the end to see computed values. -
2060 otherwise (last one is also multiline), run all in 'exec' mode
2075 otherwise (last one is also multiline), run all in 'exec' mode
2061
2076
2062 When code is executed in 'single' mode, :func:`sys.displayhook` fires,
2077 When code is executed in 'single' mode, :func:`sys.displayhook` fires,
2063 results are displayed and output prompts are computed. In 'exec' mode,
2078 results are displayed and output prompts are computed. In 'exec' mode,
2064 no results are displayed unless :func:`print` is called explicitly;
2079 no results are displayed unless :func:`print` is called explicitly;
2065 this mode is more akin to running a script.
2080 this mode is more akin to running a script.
2066
2081
2067 Parameters
2082 Parameters
2068 ----------
2083 ----------
2069 cell : str
2084 cell : str
2070 A single or multiline string.
2085 A single or multiline string.
2071 """
2086 """
2072
2087
2073 # We need to break up the input into executable blocks that can be run
2088 # We need to break up the input into executable blocks that can be run
2074 # in 'single' mode, to provide comfortable user behavior.
2089 # in 'single' mode, to provide comfortable user behavior.
2075 blocks = self.input_splitter.split_blocks(cell)
2090 blocks = self.input_splitter.split_blocks(cell)
2076
2091
2077 if not blocks:
2092 if not blocks:
2078 return
2093 return
2079
2094
2080 # Store the 'ipython' version of the cell as well, since that's what
2095 # Store the 'ipython' version of the cell as well, since that's what
2081 # needs to go into the translated history and get executed (the
2096 # needs to go into the translated history and get executed (the
2082 # original cell may contain non-python syntax).
2097 # original cell may contain non-python syntax).
2083 ipy_cell = ''.join(blocks)
2098 ipy_cell = ''.join(blocks)
2084
2099
2085 # Store raw and processed history
2100 # Store raw and processed history
2086 self.history_manager.store_inputs(ipy_cell, cell)
2101 self.history_manager.store_inputs(ipy_cell, cell)
2087
2102
2088 self.logger.log(ipy_cell, cell)
2103 self.logger.log(ipy_cell, cell)
2089 # dbg code!!!
2104 # dbg code!!!
2090 if 0:
2105 if 0:
2091 def myapp(self, val): # dbg
2106 def myapp(self, val): # dbg
2092 import traceback as tb
2107 import traceback as tb
2093 stack = ''.join(tb.format_stack())
2108 stack = ''.join(tb.format_stack())
2094 print 'Value:', val
2109 print 'Value:', val
2095 print 'Stack:\n', stack
2110 print 'Stack:\n', stack
2096 list.append(self, val)
2111 list.append(self, val)
2097
2112
2098 import new
2113 import new
2099 self.input_hist.append = new.instancemethod(myapp, self.input_hist,
2114 self.input_hist.append = new.instancemethod(myapp, self.input_hist,
2100 list)
2115 list)
2101 # End dbg
2116 # End dbg
2102
2117
2103 # All user code execution must happen with our context managers active
2118 # All user code execution must happen with our context managers active
2104 with nested(self.builtin_trap, self.display_trap):
2119 with nested(self.builtin_trap, self.display_trap):
2105
2120
2106 # Single-block input should behave like an interactive prompt
2121 # Single-block input should behave like an interactive prompt
2107 if len(blocks) == 1:
2122 if len(blocks) == 1:
2108 # since we return here, we need to update the execution count
2123 # since we return here, we need to update the execution count
2109 out = self.run_one_block(blocks[0])
2124 out = self.run_one_block(blocks[0])
2110 self.execution_count += 1
2125 self.execution_count += 1
2111 return out
2126 return out
2112
2127
2113 # In multi-block input, if the last block is a simple (one-two
2128 # In multi-block input, if the last block is a simple (one-two
2114 # lines) expression, run it in single mode so it produces output.
2129 # lines) expression, run it in single mode so it produces output.
2115 # Otherwise just feed the whole thing to run_code. This seems like
2130 # Otherwise just feed the whole thing to run_code. This seems like
2116 # a reasonable usability design.
2131 # a reasonable usability design.
2117 last = blocks[-1]
2132 last = blocks[-1]
2118 last_nlines = len(last.splitlines())
2133 last_nlines = len(last.splitlines())
2119
2134
2120 # Note: below, whenever we call run_code, we must sync history
2135 # Note: below, whenever we call run_code, we must sync history
2121 # ourselves, because run_code is NOT meant to manage history at all.
2136 # ourselves, because run_code is NOT meant to manage history at all.
2122 if last_nlines < 2:
2137 if last_nlines < 2:
2123 # Here we consider the cell split between 'body' and 'last',
2138 # Here we consider the cell split between 'body' and 'last',
2124 # store all history and execute 'body', and if successful, then
2139 # store all history and execute 'body', and if successful, then
2125 # proceed to execute 'last'.
2140 # proceed to execute 'last'.
2126
2141
2127 # Get the main body to run as a cell
2142 # Get the main body to run as a cell
2128 ipy_body = ''.join(blocks[:-1])
2143 ipy_body = ''.join(blocks[:-1])
2129 retcode = self.run_code(ipy_body, post_execute=False)
2144 retcode = self.run_code(ipy_body, post_execute=False)
2130 if retcode==0:
2145 if retcode==0:
2131 # And the last expression via runlines so it produces output
2146 # And the last expression via runlines so it produces output
2132 self.run_one_block(last)
2147 self.run_one_block(last)
2133 else:
2148 else:
2134 # Run the whole cell as one entity, storing both raw and
2149 # Run the whole cell as one entity, storing both raw and
2135 # processed input in history
2150 # processed input in history
2136 self.run_code(ipy_cell)
2151 self.run_code(ipy_cell)
2137
2152
2138 # Each cell is a *single* input, regardless of how many lines it has
2153 # Each cell is a *single* input, regardless of how many lines it has
2139 self.execution_count += 1
2154 self.execution_count += 1
2140
2155
2141 def run_one_block(self, block):
2156 def run_one_block(self, block):
2142 """Run a single interactive block.
2157 """Run a single interactive block.
2143
2158
2144 If the block is single-line, dynamic transformations are applied to it
2159 If the block is single-line, dynamic transformations are applied to it
2145 (like automagics, autocall and alias recognition).
2160 (like automagics, autocall and alias recognition).
2146 """
2161 """
2147 if len(block.splitlines()) <= 1:
2162 if len(block.splitlines()) <= 1:
2148 out = self.run_single_line(block)
2163 out = self.run_single_line(block)
2149 else:
2164 else:
2150 out = self.run_code(block)
2165 out = self.run_code(block)
2151 return out
2166 return out
2152
2167
2153 def run_single_line(self, line):
2168 def run_single_line(self, line):
2154 """Run a single-line interactive statement.
2169 """Run a single-line interactive statement.
2155
2170
2156 This assumes the input has been transformed to IPython syntax by
2171 This assumes the input has been transformed to IPython syntax by
2157 applying all static transformations (those with an explicit prefix like
2172 applying all static transformations (those with an explicit prefix like
2158 % or !), but it will further try to apply the dynamic ones.
2173 % or !), but it will further try to apply the dynamic ones.
2159
2174
2160 It does not update history.
2175 It does not update history.
2161 """
2176 """
2162 tline = self.prefilter_manager.prefilter_line(line)
2177 tline = self.prefilter_manager.prefilter_line(line)
2163 return self.run_source(tline)
2178 return self.run_source(tline)
2164
2179
2165 # PENDING REMOVAL: this method is slated for deletion, once our new
2180 # PENDING REMOVAL: this method is slated for deletion, once our new
2166 # input logic has been 100% moved to frontends and is stable.
2181 # input logic has been 100% moved to frontends and is stable.
2167 def runlines(self, lines, clean=False):
2182 def runlines(self, lines, clean=False):
2168 """Run a string of one or more lines of source.
2183 """Run a string of one or more lines of source.
2169
2184
2170 This method is capable of running a string containing multiple source
2185 This method is capable of running a string containing multiple source
2171 lines, as if they had been entered at the IPython prompt. Since it
2186 lines, as if they had been entered at the IPython prompt. Since it
2172 exposes IPython's processing machinery, the given strings can contain
2187 exposes IPython's processing machinery, the given strings can contain
2173 magic calls (%magic), special shell access (!cmd), etc.
2188 magic calls (%magic), special shell access (!cmd), etc.
2174 """
2189 """
2175
2190
2176 if isinstance(lines, (list, tuple)):
2191 if isinstance(lines, (list, tuple)):
2177 lines = '\n'.join(lines)
2192 lines = '\n'.join(lines)
2178
2193
2179 if clean:
2194 if clean:
2180 lines = self._cleanup_ipy_script(lines)
2195 lines = self._cleanup_ipy_script(lines)
2181
2196
2182 # We must start with a clean buffer, in case this is run from an
2197 # We must start with a clean buffer, in case this is run from an
2183 # interactive IPython session (via a magic, for example).
2198 # interactive IPython session (via a magic, for example).
2184 self.reset_buffer()
2199 self.reset_buffer()
2185 lines = lines.splitlines()
2200 lines = lines.splitlines()
2186
2201
2187 # Since we will prefilter all lines, store the user's raw input too
2202 # Since we will prefilter all lines, store the user's raw input too
2188 # before we apply any transformations
2203 # before we apply any transformations
2189 self.buffer_raw[:] = [ l+'\n' for l in lines]
2204 self.buffer_raw[:] = [ l+'\n' for l in lines]
2190
2205
2191 more = False
2206 more = False
2192 prefilter_lines = self.prefilter_manager.prefilter_lines
2207 prefilter_lines = self.prefilter_manager.prefilter_lines
2193 with nested(self.builtin_trap, self.display_trap):
2208 with nested(self.builtin_trap, self.display_trap):
2194 for line in lines:
2209 for line in lines:
2195 # skip blank lines so we don't mess up the prompt counter, but
2210 # skip blank lines so we don't mess up the prompt counter, but
2196 # do NOT skip even a blank line if we are in a code block (more
2211 # do NOT skip even a blank line if we are in a code block (more
2197 # is true)
2212 # is true)
2198
2213
2199 if line or more:
2214 if line or more:
2200 more = self.push_line(prefilter_lines(line, more))
2215 more = self.push_line(prefilter_lines(line, more))
2201 # IPython's run_source returns None if there was an error
2216 # IPython's run_source returns None if there was an error
2202 # compiling the code. This allows us to stop processing
2217 # compiling the code. This allows us to stop processing
2203 # right away, so the user gets the error message at the
2218 # right away, so the user gets the error message at the
2204 # right place.
2219 # right place.
2205 if more is None:
2220 if more is None:
2206 break
2221 break
2207 # final newline in case the input didn't have it, so that the code
2222 # final newline in case the input didn't have it, so that the code
2208 # actually does get executed
2223 # actually does get executed
2209 if more:
2224 if more:
2210 self.push_line('\n')
2225 self.push_line('\n')
2211
2226
2212 def run_source(self, source, filename='<ipython console>', symbol='single'):
2227 def run_source(self, source, filename='<ipython console>', symbol='single'):
2213 """Compile and run some source in the interpreter.
2228 """Compile and run some source in the interpreter.
2214
2229
2215 Arguments are as for compile_command().
2230 Arguments are as for compile_command().
2216
2231
2217 One several things can happen:
2232 One several things can happen:
2218
2233
2219 1) The input is incorrect; compile_command() raised an
2234 1) The input is incorrect; compile_command() raised an
2220 exception (SyntaxError or OverflowError). A syntax traceback
2235 exception (SyntaxError or OverflowError). A syntax traceback
2221 will be printed by calling the showsyntaxerror() method.
2236 will be printed by calling the showsyntaxerror() method.
2222
2237
2223 2) The input is incomplete, and more input is required;
2238 2) The input is incomplete, and more input is required;
2224 compile_command() returned None. Nothing happens.
2239 compile_command() returned None. Nothing happens.
2225
2240
2226 3) The input is complete; compile_command() returned a code
2241 3) The input is complete; compile_command() returned a code
2227 object. The code is executed by calling self.run_code() (which
2242 object. The code is executed by calling self.run_code() (which
2228 also handles run-time exceptions, except for SystemExit).
2243 also handles run-time exceptions, except for SystemExit).
2229
2244
2230 The return value is:
2245 The return value is:
2231
2246
2232 - True in case 2
2247 - True in case 2
2233
2248
2234 - False in the other cases, unless an exception is raised, where
2249 - False in the other cases, unless an exception is raised, where
2235 None is returned instead. This can be used by external callers to
2250 None is returned instead. This can be used by external callers to
2236 know whether to continue feeding input or not.
2251 know whether to continue feeding input or not.
2237
2252
2238 The return value can be used to decide whether to use sys.ps1 or
2253 The return value can be used to decide whether to use sys.ps1 or
2239 sys.ps2 to prompt the next line."""
2254 sys.ps2 to prompt the next line."""
2240
2255
2241 # We need to ensure that the source is unicode from here on.
2256 # We need to ensure that the source is unicode from here on.
2242 if type(source)==str:
2257 if type(source)==str:
2243 usource = source.decode(self.stdin_encoding)
2258 usource = source.decode(self.stdin_encoding)
2244 else:
2259 else:
2245 usource = source
2260 usource = source
2246
2261
2247 if 0: # dbg
2262 if 0: # dbg
2248 print 'Source:', repr(source) # dbg
2263 print 'Source:', repr(source) # dbg
2249 print 'USource:', repr(usource) # dbg
2264 print 'USource:', repr(usource) # dbg
2250 print 'type:', type(source) # dbg
2265 print 'type:', type(source) # dbg
2251 print 'encoding', self.stdin_encoding # dbg
2266 print 'encoding', self.stdin_encoding # dbg
2252
2267
2253 try:
2268 try:
2254 code = self.compile(usource,filename,symbol)
2269 code = self.compile(usource,filename,symbol)
2255 except (OverflowError, SyntaxError, ValueError, TypeError, MemoryError):
2270 except (OverflowError, SyntaxError, ValueError, TypeError, MemoryError):
2256 # Case 1
2271 # Case 1
2257 self.showsyntaxerror(filename)
2272 self.showsyntaxerror(filename)
2258 return None
2273 return None
2259
2274
2260 if code is None:
2275 if code is None:
2261 # Case 2
2276 # Case 2
2262 return True
2277 return True
2263
2278
2264 # Case 3
2279 # Case 3
2265 # We store the code object so that threaded shells and
2280 # We store the code object so that threaded shells and
2266 # custom exception handlers can access all this info if needed.
2281 # custom exception handlers can access all this info if needed.
2267 # The source corresponding to this can be obtained from the
2282 # The source corresponding to this can be obtained from the
2268 # buffer attribute as '\n'.join(self.buffer).
2283 # buffer attribute as '\n'.join(self.buffer).
2269 self.code_to_run = code
2284 self.code_to_run = code
2270 # now actually execute the code object
2285 # now actually execute the code object
2271 if self.run_code(code) == 0:
2286 if self.run_code(code) == 0:
2272 return False
2287 return False
2273 else:
2288 else:
2274 return None
2289 return None
2275
2290
2276 # For backwards compatibility
2291 # For backwards compatibility
2277 runsource = run_source
2292 runsource = run_source
2278
2293
2279 def run_code(self, code_obj, post_execute=True):
2294 def run_code(self, code_obj, post_execute=True):
2280 """Execute a code object.
2295 """Execute a code object.
2281
2296
2282 When an exception occurs, self.showtraceback() is called to display a
2297 When an exception occurs, self.showtraceback() is called to display a
2283 traceback.
2298 traceback.
2284
2299
2285 Return value: a flag indicating whether the code to be run completed
2300 Return value: a flag indicating whether the code to be run completed
2286 successfully:
2301 successfully:
2287
2302
2288 - 0: successful execution.
2303 - 0: successful execution.
2289 - 1: an error occurred.
2304 - 1: an error occurred.
2290 """
2305 """
2291
2306
2292 # Set our own excepthook in case the user code tries to call it
2307 # Set our own excepthook in case the user code tries to call it
2293 # directly, so that the IPython crash handler doesn't get triggered
2308 # directly, so that the IPython crash handler doesn't get triggered
2294 old_excepthook,sys.excepthook = sys.excepthook, self.excepthook
2309 old_excepthook,sys.excepthook = sys.excepthook, self.excepthook
2295
2310
2296 # we save the original sys.excepthook in the instance, in case config
2311 # we save the original sys.excepthook in the instance, in case config
2297 # code (such as magics) needs access to it.
2312 # code (such as magics) needs access to it.
2298 self.sys_excepthook = old_excepthook
2313 self.sys_excepthook = old_excepthook
2299 outflag = 1 # happens in more places, so it's easier as default
2314 outflag = 1 # happens in more places, so it's easier as default
2300 try:
2315 try:
2301 try:
2316 try:
2302 self.hooks.pre_run_code_hook()
2317 self.hooks.pre_run_code_hook()
2303 #rprint('Running code') # dbg
2318 #rprint('Running code') # dbg
2304 exec code_obj in self.user_global_ns, self.user_ns
2319 exec code_obj in self.user_global_ns, self.user_ns
2305 finally:
2320 finally:
2306 # Reset our crash handler in place
2321 # Reset our crash handler in place
2307 sys.excepthook = old_excepthook
2322 sys.excepthook = old_excepthook
2308 except SystemExit:
2323 except SystemExit:
2309 self.reset_buffer()
2324 self.reset_buffer()
2310 self.showtraceback(exception_only=True)
2325 self.showtraceback(exception_only=True)
2311 warn("To exit: use any of 'exit', 'quit', %Exit or Ctrl-D.", level=1)
2326 warn("To exit: use any of 'exit', 'quit', %Exit or Ctrl-D.", level=1)
2312 except self.custom_exceptions:
2327 except self.custom_exceptions:
2313 etype,value,tb = sys.exc_info()
2328 etype,value,tb = sys.exc_info()
2314 self.CustomTB(etype,value,tb)
2329 self.CustomTB(etype,value,tb)
2315 except:
2330 except:
2316 self.showtraceback()
2331 self.showtraceback()
2317 else:
2332 else:
2318 outflag = 0
2333 outflag = 0
2319 if softspace(sys.stdout, 0):
2334 if softspace(sys.stdout, 0):
2320 print
2335 print
2321
2336
2322 # Execute any registered post-execution functions. Here, any errors
2337 # Execute any registered post-execution functions. Here, any errors
2323 # are reported only minimally and just on the terminal, because the
2338 # are reported only minimally and just on the terminal, because the
2324 # main exception channel may be occupied with a user traceback.
2339 # main exception channel may be occupied with a user traceback.
2325 # FIXME: we need to think this mechanism a little more carefully.
2340 # FIXME: we need to think this mechanism a little more carefully.
2326 if post_execute:
2341 if post_execute:
2327 for func in self._post_execute:
2342 for func in self._post_execute:
2328 try:
2343 try:
2329 func()
2344 func()
2330 except:
2345 except:
2331 head = '[ ERROR ] Evaluating post_execute function: %s' % \
2346 head = '[ ERROR ] Evaluating post_execute function: %s' % \
2332 func
2347 func
2333 print >> io.Term.cout, head
2348 print >> io.Term.cout, head
2334 print >> io.Term.cout, self._simple_error()
2349 print >> io.Term.cout, self._simple_error()
2335 print >> io.Term.cout, 'Removing from post_execute'
2350 print >> io.Term.cout, 'Removing from post_execute'
2336 self._post_execute.remove(func)
2351 self._post_execute.remove(func)
2337
2352
2338 # Flush out code object which has been run (and source)
2353 # Flush out code object which has been run (and source)
2339 self.code_to_run = None
2354 self.code_to_run = None
2340 return outflag
2355 return outflag
2341
2356
2342 # For backwards compatibility
2357 # For backwards compatibility
2343 runcode = run_code
2358 runcode = run_code
2344
2359
2345 # PENDING REMOVAL: this method is slated for deletion, once our new
2360 # PENDING REMOVAL: this method is slated for deletion, once our new
2346 # input logic has been 100% moved to frontends and is stable.
2361 # input logic has been 100% moved to frontends and is stable.
2347 def push_line(self, line):
2362 def push_line(self, line):
2348 """Push a line to the interpreter.
2363 """Push a line to the interpreter.
2349
2364
2350 The line should not have a trailing newline; it may have
2365 The line should not have a trailing newline; it may have
2351 internal newlines. The line is appended to a buffer and the
2366 internal newlines. The line is appended to a buffer and the
2352 interpreter's run_source() method is called with the
2367 interpreter's run_source() method is called with the
2353 concatenated contents of the buffer as source. If this
2368 concatenated contents of the buffer as source. If this
2354 indicates that the command was executed or invalid, the buffer
2369 indicates that the command was executed or invalid, the buffer
2355 is reset; otherwise, the command is incomplete, and the buffer
2370 is reset; otherwise, the command is incomplete, and the buffer
2356 is left as it was after the line was appended. The return
2371 is left as it was after the line was appended. The return
2357 value is 1 if more input is required, 0 if the line was dealt
2372 value is 1 if more input is required, 0 if the line was dealt
2358 with in some way (this is the same as run_source()).
2373 with in some way (this is the same as run_source()).
2359 """
2374 """
2360
2375
2361 # autoindent management should be done here, and not in the
2376 # autoindent management should be done here, and not in the
2362 # interactive loop, since that one is only seen by keyboard input. We
2377 # interactive loop, since that one is only seen by keyboard input. We
2363 # need this done correctly even for code run via runlines (which uses
2378 # need this done correctly even for code run via runlines (which uses
2364 # push).
2379 # push).
2365
2380
2366 #print 'push line: <%s>' % line # dbg
2381 #print 'push line: <%s>' % line # dbg
2367 self.buffer.append(line)
2382 self.buffer.append(line)
2368 full_source = '\n'.join(self.buffer)
2383 full_source = '\n'.join(self.buffer)
2369 more = self.run_source(full_source, self.filename)
2384 more = self.run_source(full_source, self.filename)
2370 if not more:
2385 if not more:
2371 self.history_manager.store_inputs('\n'.join(self.buffer_raw),
2386 self.history_manager.store_inputs('\n'.join(self.buffer_raw),
2372 full_source)
2387 full_source)
2373 self.reset_buffer()
2388 self.reset_buffer()
2374 self.execution_count += 1
2389 self.execution_count += 1
2375 return more
2390 return more
2376
2391
2377 def reset_buffer(self):
2392 def reset_buffer(self):
2378 """Reset the input buffer."""
2393 """Reset the input buffer."""
2379 self.buffer[:] = []
2394 self.buffer[:] = []
2380 self.buffer_raw[:] = []
2395 self.buffer_raw[:] = []
2381 self.input_splitter.reset()
2396 self.input_splitter.reset()
2382
2397
2383 # For backwards compatibility
2398 # For backwards compatibility
2384 resetbuffer = reset_buffer
2399 resetbuffer = reset_buffer
2385
2400
2386 def _is_secondary_block_start(self, s):
2401 def _is_secondary_block_start(self, s):
2387 if not s.endswith(':'):
2402 if not s.endswith(':'):
2388 return False
2403 return False
2389 if (s.startswith('elif') or
2404 if (s.startswith('elif') or
2390 s.startswith('else') or
2405 s.startswith('else') or
2391 s.startswith('except') or
2406 s.startswith('except') or
2392 s.startswith('finally')):
2407 s.startswith('finally')):
2393 return True
2408 return True
2394
2409
2395 def _cleanup_ipy_script(self, script):
2410 def _cleanup_ipy_script(self, script):
2396 """Make a script safe for self.runlines()
2411 """Make a script safe for self.runlines()
2397
2412
2398 Currently, IPython is lines based, with blocks being detected by
2413 Currently, IPython is lines based, with blocks being detected by
2399 empty lines. This is a problem for block based scripts that may
2414 empty lines. This is a problem for block based scripts that may
2400 not have empty lines after blocks. This script adds those empty
2415 not have empty lines after blocks. This script adds those empty
2401 lines to make scripts safe for running in the current line based
2416 lines to make scripts safe for running in the current line based
2402 IPython.
2417 IPython.
2403 """
2418 """
2404 res = []
2419 res = []
2405 lines = script.splitlines()
2420 lines = script.splitlines()
2406 level = 0
2421 level = 0
2407
2422
2408 for l in lines:
2423 for l in lines:
2409 lstripped = l.lstrip()
2424 lstripped = l.lstrip()
2410 stripped = l.strip()
2425 stripped = l.strip()
2411 if not stripped:
2426 if not stripped:
2412 continue
2427 continue
2413 newlevel = len(l) - len(lstripped)
2428 newlevel = len(l) - len(lstripped)
2414 if level > 0 and newlevel == 0 and \
2429 if level > 0 and newlevel == 0 and \
2415 not self._is_secondary_block_start(stripped):
2430 not self._is_secondary_block_start(stripped):
2416 # add empty line
2431 # add empty line
2417 res.append('')
2432 res.append('')
2418 res.append(l)
2433 res.append(l)
2419 level = newlevel
2434 level = newlevel
2420
2435
2421 return '\n'.join(res) + '\n'
2436 return '\n'.join(res) + '\n'
2422
2437
2423 #-------------------------------------------------------------------------
2438 #-------------------------------------------------------------------------
2424 # Things related to GUI support and pylab
2439 # Things related to GUI support and pylab
2425 #-------------------------------------------------------------------------
2440 #-------------------------------------------------------------------------
2426
2441
2427 def enable_pylab(self, gui=None):
2442 def enable_pylab(self, gui=None):
2428 raise NotImplementedError('Implement enable_pylab in a subclass')
2443 raise NotImplementedError('Implement enable_pylab in a subclass')
2429
2444
2430 #-------------------------------------------------------------------------
2445 #-------------------------------------------------------------------------
2431 # Utilities
2446 # Utilities
2432 #-------------------------------------------------------------------------
2447 #-------------------------------------------------------------------------
2433
2448
2434 def var_expand(self,cmd,depth=0):
2449 def var_expand(self,cmd,depth=0):
2435 """Expand python variables in a string.
2450 """Expand python variables in a string.
2436
2451
2437 The depth argument indicates how many frames above the caller should
2452 The depth argument indicates how many frames above the caller should
2438 be walked to look for the local namespace where to expand variables.
2453 be walked to look for the local namespace where to expand variables.
2439
2454
2440 The global namespace for expansion is always the user's interactive
2455 The global namespace for expansion is always the user's interactive
2441 namespace.
2456 namespace.
2442 """
2457 """
2443
2458
2444 return str(ItplNS(cmd,
2459 return str(ItplNS(cmd,
2445 self.user_ns, # globals
2460 self.user_ns, # globals
2446 # Skip our own frame in searching for locals:
2461 # Skip our own frame in searching for locals:
2447 sys._getframe(depth+1).f_locals # locals
2462 sys._getframe(depth+1).f_locals # locals
2448 ))
2463 ))
2449
2464
2450 def mktempfile(self,data=None):
2465 def mktempfile(self,data=None):
2451 """Make a new tempfile and return its filename.
2466 """Make a new tempfile and return its filename.
2452
2467
2453 This makes a call to tempfile.mktemp, but it registers the created
2468 This makes a call to tempfile.mktemp, but it registers the created
2454 filename internally so ipython cleans it up at exit time.
2469 filename internally so ipython cleans it up at exit time.
2455
2470
2456 Optional inputs:
2471 Optional inputs:
2457
2472
2458 - data(None): if data is given, it gets written out to the temp file
2473 - data(None): if data is given, it gets written out to the temp file
2459 immediately, and the file is closed again."""
2474 immediately, and the file is closed again."""
2460
2475
2461 filename = tempfile.mktemp('.py','ipython_edit_')
2476 filename = tempfile.mktemp('.py','ipython_edit_')
2462 self.tempfiles.append(filename)
2477 self.tempfiles.append(filename)
2463
2478
2464 if data:
2479 if data:
2465 tmp_file = open(filename,'w')
2480 tmp_file = open(filename,'w')
2466 tmp_file.write(data)
2481 tmp_file.write(data)
2467 tmp_file.close()
2482 tmp_file.close()
2468 return filename
2483 return filename
2469
2484
2470 # TODO: This should be removed when Term is refactored.
2485 # TODO: This should be removed when Term is refactored.
2471 def write(self,data):
2486 def write(self,data):
2472 """Write a string to the default output"""
2487 """Write a string to the default output"""
2473 io.Term.cout.write(data)
2488 io.Term.cout.write(data)
2474
2489
2475 # TODO: This should be removed when Term is refactored.
2490 # TODO: This should be removed when Term is refactored.
2476 def write_err(self,data):
2491 def write_err(self,data):
2477 """Write a string to the default error output"""
2492 """Write a string to the default error output"""
2478 io.Term.cerr.write(data)
2493 io.Term.cerr.write(data)
2479
2494
2480 def ask_yes_no(self,prompt,default=True):
2495 def ask_yes_no(self,prompt,default=True):
2481 if self.quiet:
2496 if self.quiet:
2482 return True
2497 return True
2483 return ask_yes_no(prompt,default)
2498 return ask_yes_no(prompt,default)
2484
2499
2485 def show_usage(self):
2500 def show_usage(self):
2486 """Show a usage message"""
2501 """Show a usage message"""
2487 page.page(IPython.core.usage.interactive_usage)
2502 page.page(IPython.core.usage.interactive_usage)
2488
2503
2489 #-------------------------------------------------------------------------
2504 #-------------------------------------------------------------------------
2490 # Things related to IPython exiting
2505 # Things related to IPython exiting
2491 #-------------------------------------------------------------------------
2506 #-------------------------------------------------------------------------
2492 def atexit_operations(self):
2507 def atexit_operations(self):
2493 """This will be executed at the time of exit.
2508 """This will be executed at the time of exit.
2494
2509
2495 Cleanup operations and saving of persistent data that is done
2510 Cleanup operations and saving of persistent data that is done
2496 unconditionally by IPython should be performed here.
2511 unconditionally by IPython should be performed here.
2497
2512
2498 For things that may depend on startup flags or platform specifics (such
2513 For things that may depend on startup flags or platform specifics (such
2499 as having readline or not), register a separate atexit function in the
2514 as having readline or not), register a separate atexit function in the
2500 code that has the appropriate information, rather than trying to
2515 code that has the appropriate information, rather than trying to
2501 clutter
2516 clutter
2502 """
2517 """
2503 # Cleanup all tempfiles left around
2518 # Cleanup all tempfiles left around
2504 for tfile in self.tempfiles:
2519 for tfile in self.tempfiles:
2505 try:
2520 try:
2506 os.unlink(tfile)
2521 os.unlink(tfile)
2507 except OSError:
2522 except OSError:
2508 pass
2523 pass
2509
2524
2510 # Clear all user namespaces to release all references cleanly.
2525 # Clear all user namespaces to release all references cleanly.
2511 self.reset()
2526 self.reset()
2512
2527
2513 # Run user hooks
2528 # Run user hooks
2514 self.hooks.shutdown_hook()
2529 self.hooks.shutdown_hook()
2515
2530
2516 def cleanup(self):
2531 def cleanup(self):
2517 self.restore_sys_module_state()
2532 self.restore_sys_module_state()
2518
2533
2519
2534
2520 class InteractiveShellABC(object):
2535 class InteractiveShellABC(object):
2521 """An abstract base class for InteractiveShell."""
2536 """An abstract base class for InteractiveShell."""
2522 __metaclass__ = abc.ABCMeta
2537 __metaclass__ = abc.ABCMeta
2523
2538
2524 InteractiveShellABC.register(InteractiveShell)
2539 InteractiveShellABC.register(InteractiveShell)
General Comments 0
You need to be logged in to leave comments. Login now