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