##// END OF EJS Templates
Major restructuring of magics, breaking them up into separate classes....
Fernando Perez -
Show More

The requested changes are too big and content was truncated. Show full diff

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