##// END OF EJS Templates
Add message when kernel catches KeyboardInterrupt.
Add message when kernel catches KeyboardInterrupt.

File last commit:

r3748:e55836ea
r3901:5cd1ae15
Show More
inputsplitter.py
865 lines | 30.7 KiB | text/x-python | PythonLexer
Fernando Perez
Created tool for handling interactive input blocks....
r2628 """Analysis of text input into executable blocks.
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 The main class in this module, :class:`InputSplitter`, is designed to break
input from either interactive, line-by-line environments or block-based ones,
into standalone blocks that can be executed by Python as 'single' statements
(thus triggering sys.displayhook).
Fernando Perez
Created tool for handling interactive input blocks....
r2628
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 A companion, :class:`IPythonInputSplitter`, provides the same functionality but
with full support for the extended IPython syntax (magics, system calls, etc).
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 For more details, see the class docstring below.
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 Syntax Transformations
----------------------
One of the main jobs of the code in this file is to apply all syntax
transformations that make up 'the IPython language', i.e. magics, shell
escapes, etc. All transformations should be implemented as *fully stateless*
entities, that simply take one line as their input and return a line.
Internally for implementation purposes they may be a normal function or a
callable object, but the only input they receive will be a single line and they
should only return a line, without holding any data-dependent state between
calls.
As an example, the EscapedTransformer is a class so we can more clearly group
together the functionality of dispatching to individual functions based on the
starting escape character, but the only method for public use is its call
method.
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 ToDo
----
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 - Should we make push() actually raise an exception once push_accepts_more()
returns False?
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 - Naming cleanups. The tr_* names aren't the most elegant, though now they are
at least just attributes of a class so not really very exposed.
- Think about the best way to support dynamic things: automagic, autocall,
macros, etc.
- Think of a better heuristic for the application of the transforms in
IPythonInputSplitter.push() than looking at the buffer ending in ':'. Idea:
track indentation change events (indent, dedent, nothing) and apply them only
if the indentation went up, but not otherwise.
- Think of the cleanest way for supporting user-specified transformations (the
user prefilters we had before).
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 Authors
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 -------
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
* Fernando Perez
* Brian Granger
Fernando Perez
Created tool for handling interactive input blocks....
r2628 """
#-----------------------------------------------------------------------------
# Copyright (C) 2010 The IPython Development Team
#
# Distributed under the terms of the BSD License. The full license is in
# the file COPYING, distributed as part of this software.
#-----------------------------------------------------------------------------
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 from __future__ import print_function
Fernando Perez
Created tool for handling interactive input blocks....
r2628
#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
# stdlib
Thomas Kluyver
Fix up so tests pass again. input_splitter now uses ast module instead of compiler, bringing it closer to the Python 3 implementation.
r3454 import ast
Fernando Perez
Created tool for handling interactive input blocks....
r2628 import codeop
import re
import sys
Fernando Perez
First pass of input syntax transformation support
r2719 # IPython modules
from IPython.utils.text import make_quoted_expr
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004
Fernando Perez
Created tool for handling interactive input blocks....
r2628 #-----------------------------------------------------------------------------
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 # Globals
#-----------------------------------------------------------------------------
# The escape sequences that define the syntax transformations IPython will
# apply to user input. These can NOT be just changed here: many regular
# expressions and other parts of the code may use their hardcoded values, and
# for all intents and purposes they constitute the 'IPython syntax', so they
# should be considered fixed.
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 ESC_SHELL = '!' # Send line to underlying system shell
ESC_SH_CAP = '!!' # Send line to system shell and capture output
ESC_HELP = '?' # Find information about object
ESC_HELP2 = '??' # Find extra-detailed information about object
ESC_MAGIC = '%' # Call magic function
ESC_QUOTE = ',' # Split args on whitespace, quote each as string and call
ESC_QUOTE2 = ';' # Quote all args as a single string, call
ESC_PAREN = '/' # Call first argument with rest of line as arguments
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
#-----------------------------------------------------------------------------
Fernando Perez
Created tool for handling interactive input blocks....
r2628 # Utilities
#-----------------------------------------------------------------------------
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 # FIXME: These are general-purpose utilities that later can be moved to the
# general ward. Kept here for now because we're being very strict about test
# coverage with this code, and this lets us ensure that we keep 100% coverage
# while developing.
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633
Fernando Perez
Created tool for handling interactive input blocks....
r2628 # compiled regexps for autoindent management
David Warde-Farley
Fix behaviour of dedent triggering. Closes gh-142.
r3704 dedent_re = re.compile('|'.join([
r'^\s+raise(\s.*)?$', # raise statement (+ space + other stuff, maybe)
r'^\s+raise\([^\)]*\).*$', # wacky raise with immediate open paren
r'^\s+return(\s.*)?$', # normal return (+ space + other stuff, maybe)
r'^\s+return\([^\)]*\).*$', # wacky return with immediate open paren
r'^\s+pass\s*$' # pass (optionally followed by trailing spaces)
]))
Fernando Perez
Created tool for handling interactive input blocks....
r2628 ini_spaces_re = re.compile(r'^([ \t\r\f\v]+)')
Fernando Perez
Fix bug where 'if 1:' was being added to comment-only code....
r2979 # regexp to match pure comment lines so we don't accidentally insert 'if 1:'
# before pure comments
comment_line_re = re.compile('^\s*\#')
Fernando Perez
Created tool for handling interactive input blocks....
r2628
def num_ini_spaces(s):
"""Return the number of initial spaces in a string.
Note that tabs are counted as a single space. For now, we do *not* support
mixing of tabs and spaces in the user's input.
Parameters
----------
s : string
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663
Returns
-------
n : int
Fernando Perez
Created tool for handling interactive input blocks....
r2628 """
ini_spaces = ini_spaces_re.match(s)
if ini_spaces:
return ini_spaces.end()
else:
return 0
def remove_comments(src):
"""Remove all comments from input source.
Note: comments are NOT recognized inside of strings!
Parameters
----------
src : string
A single or multiline input string.
Returns
-------
String with all Python comments removed.
"""
return re.sub('#.*', '', src)
def get_input_encoding():
Fernando Perez
Add test for missing input encoding. Back to 100% coverage.
r2718 """Return the default standard input encoding.
If sys.stdin has no encoding, 'ascii' is returned."""
epatters
Made blockbreakers' input encoding detection more robust to strange...
r2674 # There are strange environments for which sys.stdin.encoding is None. We
# ensure that a valid encoding is returned.
encoding = getattr(sys.stdin, 'encoding', None)
if encoding is None:
encoding = 'ascii'
return encoding
Fernando Perez
Created tool for handling interactive input blocks....
r2628
#-----------------------------------------------------------------------------
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 # Classes and functions for normal Python syntax handling
Fernando Perez
Created tool for handling interactive input blocks....
r2628 #-----------------------------------------------------------------------------
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 class InputSplitter(object):
"""An object that can split Python source input in executable blocks.
This object is designed to be used in one of two basic modes:
1. By feeding it python source line-by-line, using :meth:`push`. In this
mode, it will return on each push whether the currently pushed code
could be executed already. In addition, it provides a method called
:meth:`push_accepts_more` that can be used to query whether more input
can be pushed into a single interactive block.
2. By calling :meth:`split_blocks` with a single, multiline Python string,
that is then split into blocks each of which can be executed
interactively as a single statement.
This is a simple example of how an interactive terminal-based client can use
this tool::
isp = InputSplitter()
while isp.push_accepts_more():
indent = ' '*isp.indent_spaces
prompt = '>>> ' + indent
line = indent + raw_input(prompt)
isp.push(line)
print 'Input source was:\n', isp.source_reset(),
"""
# Number of spaces of indentation computed from input that has been pushed
# so far. This is the attributes callers should query to get the current
# indentation level, in order to provide auto-indent facilities.
Fernando Perez
Created tool for handling interactive input blocks....
r2628 indent_spaces = 0
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 # String, indicating the default input encoding. It is computed by default
# at initialization time via get_input_encoding(), but it can be reset by a
# client with specific knowledge of the encoding.
Fernando Perez
Created tool for handling interactive input blocks....
r2628 encoding = ''
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 # String where the current full source input is stored, properly encoded.
# Reading this attribute is the normal way of querying the currently pushed
# source code, that has been properly encoded.
Fernando Perez
Created tool for handling interactive input blocks....
r2628 source = ''
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 # Code object corresponding to the current source. It is automatically
# synced to the source, so it can be queried at any time to obtain the code
# object; it will be None if the source doesn't compile to valid Python.
Fernando Perez
Created tool for handling interactive input blocks....
r2628 code = None
Fernando Perez
Add support for append/replace mode after discussion with Evan....
r2634 # Input mode
Fernando Perez
Rename input modes of input splitter to 'line' and 'block'....
r2862 input_mode = 'line'
Fernando Perez
Add support for append/replace mode after discussion with Evan....
r2634
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 # Private attributes
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 # List with lines of input accumulated so far
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 _buffer = None
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 # Command compiler
_compile = None
# Mark when input has changed indentation all the way back to flush-left
_full_dedent = False
# Boolean indicating whether the current block is complete
_is_complete = None
Fernando Perez
Created tool for handling interactive input blocks....
r2628
Fernando Perez
Add support for append/replace mode after discussion with Evan....
r2634 def __init__(self, input_mode=None):
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 """Create a new InputSplitter instance.
Fernando Perez
Add support for append/replace mode after discussion with Evan....
r2634
Parameters
----------
input_mode : str
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 One of ['line', 'cell']; default is 'line'.
Fernando Perez
Add support for append/replace mode after discussion with Evan....
r2634
Fernando Perez
Rename input modes of input splitter to 'line' and 'block'....
r2862 The input_mode parameter controls how new inputs are used when fed via
the :meth:`push` method:
- 'line': meant for line-oriented clients, inputs are appended one at a
time to the internal buffer and the whole buffer is compiled.
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 - 'cell': meant for clients that can edit multi-line 'cells' of text at
a time. A cell can contain one or more blocks that can be compile in
'single' mode by Python. In this mode, each new input new input
completely replaces all prior inputs. Cell mode is thus equivalent
to prepending a full reset() to every push() call.
Fernando Perez
Add support for append/replace mode after discussion with Evan....
r2634 """
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 self._buffer = []
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 self._compile = codeop.CommandCompiler()
Fernando Perez
Created tool for handling interactive input blocks....
r2628 self.encoding = get_input_encoding()
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 self.input_mode = InputSplitter.input_mode if input_mode is None \
Fernando Perez
Add support for append/replace mode after discussion with Evan....
r2634 else input_mode
Fernando Perez
Created tool for handling interactive input blocks....
r2628
def reset(self):
"""Reset the input buffer and associated state."""
self.indent_spaces = 0
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 self._buffer[:] = []
Fernando Perez
Created tool for handling interactive input blocks....
r2628 self.source = ''
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 self.code = None
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 self._is_complete = False
self._full_dedent = False
Fernando Perez
Created tool for handling interactive input blocks....
r2628
Fernando Perez
Change get_source() api as per code review, to source_reset()....
r2636 def source_reset(self):
"""Return the input source and perform a full reset.
Fernando Perez
Created tool for handling interactive input blocks....
r2628 """
out = self.source
Fernando Perez
Change get_source() api as per code review, to source_reset()....
r2636 self.reset()
Fernando Perez
Created tool for handling interactive input blocks....
r2628 return out
def push(self, lines):
Robert Kern
Fix bug where bare strings would be silently ignored in input....
r3293 """Push one or more lines of input.
Fernando Perez
Created tool for handling interactive input blocks....
r2628
This stores the given lines and returns a status code indicating
whether the code forms a complete Python block or not.
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 Any exceptions generated in compilation are swallowed, but if an
exception was produced, the method returns True.
Fernando Perez
Created tool for handling interactive input blocks....
r2628
Parameters
----------
lines : string
One or more lines of Python input.
Returns
-------
is_complete : boolean
True if the current input source (the result of the current input
plus prior inputs) forms a complete Python execution block. Note that
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 this value is also stored as a private attribute (_is_complete), so it
can be queried at any time.
Fernando Perez
Created tool for handling interactive input blocks....
r2628 """
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 if self.input_mode == 'cell':
Fernando Perez
Add support for append/replace mode after discussion with Evan....
r2634 self.reset()
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 self._store(lines)
Fernando Perez
Created tool for handling interactive input blocks....
r2628 source = self.source
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 # Before calling _compile(), reset the code object to None so that if an
Fernando Perez
Created tool for handling interactive input blocks....
r2628 # exception is raised in compilation, we don't mislead by having
# inconsistent code/source attributes.
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 self.code, self._is_complete = None, None
Fernando Perez
Completed full block splitting for block-based frontends.
r2645
Fernando Perez
Fix bug with lines ending in continuation markers (\)....
r3013 # Honor termination lines properly
if source.rstrip().endswith('\\'):
return False
Fernando Perez
Completed full block splitting for block-based frontends.
r2645 self._update_indent(lines)
Fernando Perez
push() now swallows syntax errors and immediately produces a 'ready'...
r2635 try:
Thomas Kluyver
Add exec flag so inputsplitter handles multiline cells better.
r3748 self.code = self._compile(source, symbol="exec")
Fernando Perez
push() now swallows syntax errors and immediately produces a 'ready'...
r2635 # Invalid syntax can produce any of a number of different errors from
# inside the compiler, so we have to catch them all. Syntax errors
# immediately produce a 'ready' block, so the invalid Python can be
# sent to the kernel for evaluation with possible ipython
# special-syntax conversion.
Fernando Perez
Completed full block splitting for block-based frontends.
r2645 except (SyntaxError, OverflowError, ValueError, TypeError,
MemoryError):
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 self._is_complete = True
Fernando Perez
push() now swallows syntax errors and immediately produces a 'ready'...
r2635 else:
# Compilation didn't produce any exceptions (though it may not have
# given a complete code object)
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 self._is_complete = self.code is not None
Fernando Perez
push() now swallows syntax errors and immediately produces a 'ready'...
r2635
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 return self._is_complete
Fernando Perez
Created tool for handling interactive input blocks....
r2628
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 def push_accepts_more(self):
"""Return whether a block of interactive input can accept more input.
Fernando Perez
Created tool for handling interactive input blocks....
r2628
This method is meant to be used by line-oriented frontends, who need to
guess whether a block is complete or not based solely on prior and
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 current input lines. The InputSplitter considers it has a complete
interactive block and will not accept more input only when either a
SyntaxError is raised, or *all* of the following are true:
Fernando Perez
Created tool for handling interactive input blocks....
r2628
1. The input compiles to a complete statement.
2. The indentation level is flush-left (because if we are indented,
like inside a function definition or for loop, we need to keep
reading new input).
3. There is one extra line consisting only of whitespace.
Because of condition #3, this method should be used only by
*line-oriented* frontends, since it means that intermediate blank lines
are not allowed in function definitions (or any other indented block).
Block-oriented frontends that have a separate keyboard event to
indicate execution should use the :meth:`split_blocks` method instead.
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663
If the current input produces a syntax error, this method immediately
returns False but does *not* raise the syntax error exception, as
typically clients will want to send invalid syntax to an execution
backend which might convert the invalid syntax into valid Python via
one of the dynamic IPython mechanisms.
Fernando Perez
Created tool for handling interactive input blocks....
r2628 """
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004
# With incomplete input, unconditionally accept more
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 if not self._is_complete:
Fernando Perez
Created tool for handling interactive input blocks....
r2628 return True
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 # If we already have complete input and we're flush left, the answer
Thomas Kluyver
Require blank line to end input cell immediately after dedenting....
r3461 # depends. In line mode, if there hasn't been any indentation,
# that's it. If we've come back from some indentation, we need
# the blank final line to finish.
# In cell mode, we need to check how many blocks the input so far
# compiles into, because if there's already more than one full
# independent block of input, then the client has entered full
# 'cell' mode and is feeding lines that each is complete. In this
# case we should then keep accepting. The Qt terminal-like console
# does precisely this, to provide the convenience of terminal-like
# input of single expressions, but allowing the user (with a
# separate keystroke) to switch to 'cell' mode and type multiple
# expressions in one shot.
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 if self.indent_spaces==0:
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 if self.input_mode=='line':
Thomas Kluyver
Require blank line to end input cell immediately after dedenting....
r3461 if not self._full_dedent:
return False
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 else:
Thomas Kluyver
New ast_nodes method, and change push_accepts_more to use it.
r3526 try:
Thomas Kluyver
Simplify - call ast.parse inline, rather than via a separate method.
r3528 code_ast = ast.parse(u''.join(self._buffer))
Thomas Kluyver
New ast_nodes method, and change push_accepts_more to use it.
r3526 except Exception:
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 return False
Thomas Kluyver
New ast_nodes method, and change push_accepts_more to use it.
r3526 else:
Thomas Kluyver
Simplify - call ast.parse inline, rather than via a separate method.
r3528 if len(code_ast.body) == 1:
Thomas Kluyver
New ast_nodes method, and change push_accepts_more to use it.
r3526 return False
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004
# When input is complete, then termination is marked by an extra blank
# line at the end.
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 last_line = self.source.splitlines()[-1]
return bool(last_line and not last_line.isspace())
Fernando Perez
Created tool for handling interactive input blocks....
r2628
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 #------------------------------------------------------------------------
# Private interface
#------------------------------------------------------------------------
Fernando Perez
Created tool for handling interactive input blocks....
r2628
Fernando Perez
Completed full block splitting for block-based frontends.
r2645 def _find_indent(self, line):
"""Compute the new indentation level for a single line.
Parameters
----------
line : str
A single new line of non-whitespace, non-comment Python input.
Returns
-------
indent_spaces : int
New value for the indent level (it may be equal to self.indent_spaces
if indentation doesn't change.
full_dedent : boolean
Whether the new line causes a full flush-left dedent.
"""
indent_spaces = self.indent_spaces
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 full_dedent = self._full_dedent
Fernando Perez
Completed full block splitting for block-based frontends.
r2645
inisp = num_ini_spaces(line)
if inisp < indent_spaces:
indent_spaces = inisp
if indent_spaces <= 0:
#print 'Full dedent in text',self.source # dbg
full_dedent = True
if line[-1] == ':':
indent_spaces += 4
elif dedent_re.match(line):
indent_spaces -= 4
if indent_spaces <= 0:
full_dedent = True
# Safety
if indent_spaces < 0:
indent_spaces = 0
#print 'safety' # dbg
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633
Fernando Perez
Completed full block splitting for block-based frontends.
r2645 return indent_spaces, full_dedent
Fernando Perez
Continue refactoring input handling across clients....
r3085
Fernando Perez
Completed full block splitting for block-based frontends.
r2645 def _update_indent(self, lines):
for line in remove_comments(lines).splitlines():
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 if line and not line.isspace():
Fernando Perez
Renamed to inputsplitter, added more tests and examples....
r2663 self.indent_spaces, self._full_dedent = self._find_indent(line)
Fernando Perez
Created tool for handling interactive input blocks....
r2628
Fernando Perez
Add support for accessing raw data to inputsplitter....
r3080 def _store(self, lines, buffer=None, store='source'):
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 """Store one or more lines of input.
If input lines are not newline-terminated, a newline is automatically
appended."""
Thomas Kluyver
Fix for unicode in inputsplitter.
r3448
Fernando Perez
Add support for accessing raw data to inputsplitter....
r3080 if buffer is None:
buffer = self._buffer
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 if lines.endswith('\n'):
Fernando Perez
Add support for accessing raw data to inputsplitter....
r3080 buffer.append(lines)
Fernando Perez
Split blockbreaker tests into a separate file and clean up api....
r2633 else:
Fernando Perez
Add support for accessing raw data to inputsplitter....
r3080 buffer.append(lines+'\n')
setattr(self, store, self._set_source(buffer))
Fernando Perez
Completed full block splitting for block-based frontends.
r2645
Fernando Perez
Add support for accessing raw data to inputsplitter....
r3080 def _set_source(self, buffer):
Thomas Kluyver
Further fixes and tweaks for inputsplitter.
r3455 return u''.join(buffer)
Fernando Perez
First pass of input syntax transformation support
r2719
#-----------------------------------------------------------------------------
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 # Functions and classes for IPython-specific syntactic support
Fernando Perez
First pass of input syntax transformation support
r2719 #-----------------------------------------------------------------------------
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 # RegExp for splitting line contents into pre-char//first word-method//rest.
# For clarity, each group in on one line.
line_split = re.compile("""
^(\s*) # any leading space
([,;/%]|!!?|\?\??) # escape character or characters
Fernando Perez
Fix bug for x.*a*? syntax, added tests for it.
r3068 \s*(%?[\w\.\*]*) # function/method, possibly with leading %
Fernando Perez
Fix subtle regular expression bug to treat '%foo?' correctly.
r2879 # to correctly treat things like '?%magic'
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 (\s+.*$|$) # rest of line
""", re.VERBOSE)
def split_user_input(line):
"""Split user input into early whitespace, esc-char, function part and rest.
This is currently handles lines with '=' in them in a very inconsistent
manner.
Examples
========
>>> split_user_input('x=1')
('', '', 'x=1', '')
>>> split_user_input('?')
('', '?', '', '')
>>> split_user_input('??')
('', '??', '', '')
>>> split_user_input(' ?')
(' ', '?', '', '')
>>> split_user_input(' ??')
(' ', '??', '', '')
>>> split_user_input('??x')
('', '??', 'x', '')
>>> split_user_input('?x=1')
('', '', '?x=1', '')
>>> split_user_input('!ls')
('', '!', 'ls', '')
>>> split_user_input(' !ls')
(' ', '!', 'ls', '')
>>> split_user_input('!!ls')
('', '!!', 'ls', '')
>>> split_user_input(' !!ls')
(' ', '!!', 'ls', '')
>>> split_user_input(',ls')
('', ',', 'ls', '')
>>> split_user_input(';ls')
('', ';', 'ls', '')
>>> split_user_input(' ;ls')
(' ', ';', 'ls', '')
>>> split_user_input('f.g(x)')
('', '', 'f.g(x)', '')
>>> split_user_input('f.g (x)')
('', '', 'f.g', '(x)')
Fernando Perez
Fix subtle regular expression bug to treat '%foo?' correctly.
r2879 >>> split_user_input('?%hist')
('', '?', '%hist', '')
Fernando Perez
Fix bug for x.*a*? syntax, added tests for it.
r3068 >>> split_user_input('?x*')
('', '?', 'x*', '')
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 """
match = line_split.match(line)
if match:
lspace, esc, fpart, rest = match.groups()
else:
# print "match failed for line '%s'" % line
try:
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 fpart, rest = line.split(None, 1)
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 except ValueError:
# print "split failed for line '%s'" % line
fpart, rest = line,''
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 lspace = re.match('^(\s*)(.*)', line).groups()[0]
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 esc = ''
# fpart has to be a valid python identifier, so it better be only pure
# ascii, no unicode:
try:
fpart = fpart.encode('ascii')
except UnicodeEncodeError:
lspace = unicode(lspace)
rest = fpart + u' ' + rest
fpart = u''
#print 'line:<%s>' % line # dbg
#print 'esc <%s> fpart <%s> rest <%s>' % (esc,fpart.strip(),rest) # dbg
return lspace, esc, fpart.strip(), rest.lstrip()
# The escaped translators ALL receive a line where their own escape has been
# stripped. Only '?' is valid at the end of the line, all others can only be
# placed at the start.
class LineInfo(object):
"""A single line of input and associated info.
This is a utility class that mostly wraps the output of
:func:`split_user_input` into a convenient object to be passed around
during input transformations.
Includes the following as properties:
line
The original, raw line
lspace
Any early whitespace before actual text starts.
esc
The initial esc character (or characters, for double-char escapes like
'??' or '!!').
fpart
The 'function part', which is basically the maximal initial sequence
of valid python identifiers and the '.' character. This is what is
checked for alias and magic transformations, used for auto-calling,
etc.
rest
Everything else on the line.
"""
def __init__(self, line):
self.line = line
self.lspace, self.esc, self.fpart, self.rest = \
split_user_input(line)
def __str__(self):
return "LineInfo [%s|%s|%s|%s]" % (self.lspace, self.esc,
self.fpart, self.rest)
# Transformations of the special syntaxes that don't rely on an explicit escape
# character but instead on patterns on the input line
# The core transformations are implemented as standalone functions that can be
# tested and validated in isolation. Each of these uses a regexp, we
# pre-compile these and keep them close to each function definition for clarity
Fernando Perez
First pass of input syntax transformation support
r2719
_assign_system_re = re.compile(r'(?P<lhs>(\s*)([\w\.]+)((\s*,\s*[\w\.]+)*))'
r'\s*=\s*!\s*(?P<cmd>.*)')
def transform_assign_system(line):
"""Handle the `files = !ls` syntax."""
m = _assign_system_re.match(line)
if m is not None:
cmd = m.group('cmd')
lhs = m.group('lhs')
Fernando Perez
Fix bugs in x=!cmd; we can't use pexpect at all....
r3002 expr = make_quoted_expr(cmd)
new_line = '%s = get_ipython().getoutput(%s)' % (lhs, expr)
Fernando Perez
First pass of input syntax transformation support
r2719 return new_line
return line
_assign_magic_re = re.compile(r'(?P<lhs>(\s*)([\w\.]+)((\s*,\s*[\w\.]+)*))'
r'\s*=\s*%\s*(?P<cmd>.*)')
def transform_assign_magic(line):
"""Handle the `a = %who` syntax."""
m = _assign_magic_re.match(line)
if m is not None:
cmd = m.group('cmd')
lhs = m.group('lhs')
expr = make_quoted_expr(cmd)
new_line = '%s = get_ipython().magic(%s)' % (lhs, expr)
return new_line
return line
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 _classic_prompt_re = re.compile(r'^([ \t]*>>> |^[ \t]*\.\.\. )')
Fernando Perez
First pass of input syntax transformation support
r2719
def transform_classic_prompt(line):
"""Handle inputs that start with '>>> ' syntax."""
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 if not line or line.isspace():
return line
Fernando Perez
First pass of input syntax transformation support
r2719 m = _classic_prompt_re.match(line)
if m:
return line[len(m.group(0)):]
else:
return line
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 _ipy_prompt_re = re.compile(r'^([ \t]*In \[\d+\]: |^[ \t]*\ \ \ \.\.\.+: )')
Fernando Perez
First pass of input syntax transformation support
r2719
def transform_ipy_prompt(line):
"""Handle inputs that start classic IPython prompt syntax."""
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 if not line or line.isspace():
return line
Fernando Perez
Fix bug with IPythonInputSplitter in block input mode.
r2861 #print 'LINE: %r' % line # dbg
Fernando Perez
First pass of input syntax transformation support
r2719 m = _ipy_prompt_re.match(line)
if m:
Fernando Perez
Fix bug with IPythonInputSplitter in block input mode.
r2861 #print 'MATCH! %r -> %r' % (line, line[len(m.group(0)):]) # dbg
Fernando Perez
First pass of input syntax transformation support
r2719 return line[len(m.group(0)):]
else:
return line
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 class EscapedTransformer(object):
"""Class to transform lines that are explicitly escaped out."""
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 def __init__(self):
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 tr = { ESC_SHELL : self._tr_system,
ESC_SH_CAP : self._tr_system2,
ESC_HELP : self._tr_help,
ESC_HELP2 : self._tr_help,
ESC_MAGIC : self._tr_magic,
ESC_QUOTE : self._tr_quote,
ESC_QUOTE2 : self._tr_quote2,
ESC_PAREN : self._tr_paren }
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 self.tr = tr
# Support for syntax transformations that use explicit escapes typed by the
# user at the beginning of a line
@staticmethod
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 def _tr_system(line_info):
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 "Translate lines escaped with: !"
cmd = line_info.line.lstrip().lstrip(ESC_SHELL)
return '%sget_ipython().system(%s)' % (line_info.lspace,
make_quoted_expr(cmd))
@staticmethod
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 def _tr_system2(line_info):
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 "Translate lines escaped with: !!"
cmd = line_info.line.lstrip()[2:]
return '%sget_ipython().getoutput(%s)' % (line_info.lspace,
make_quoted_expr(cmd))
@staticmethod
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 def _tr_help(line_info):
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 "Translate lines escaped with: ?/??"
# A naked help line should just fire the intro help screen
if not line_info.line[1:]:
return 'get_ipython().show_usage()'
# There may be one or two '?' at the end, move them to the front so that
# the rest of the logic can assume escapes are at the start
Fernando Perez
Fix bug for x.*a*? syntax, added tests for it.
r3068 l_ori = line_info
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 line = line_info.line
if line.endswith('?'):
line = line[-1] + line[:-1]
if line.endswith('?'):
line = line[-1] + line[:-1]
line_info = LineInfo(line)
Fernando Perez
Fix bug for x.*a*? syntax, added tests for it.
r3068 # From here on, simply choose which level of detail to get, and
# special-case the psearch syntax
Fernando Perez
Fix crash with 'x?' when input line had unusual patterns.
r3098 pinfo = 'pinfo' # default
Fernando Perez
Fix bug for x.*a*? syntax, added tests for it.
r3068 if '*' in line_info.line:
pinfo = 'psearch'
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 elif line_info.esc == '??':
pinfo = 'pinfo2'
Thomas Kluyver
Modify tests to expect unicode. Now all tests pass again.
r3471 tpl = '%sget_ipython().magic(u"%s %s")'
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 return tpl % (line_info.lspace, pinfo,
' '.join([line_info.fpart, line_info.rest]).strip())
@staticmethod
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 def _tr_magic(line_info):
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 "Translate lines escaped with: %"
tpl = '%sget_ipython().magic(%s)'
cmd = make_quoted_expr(' '.join([line_info.fpart,
Fernando Perez
Fix failing tests for escaped magics and undo changes to text.py....
r2821 line_info.rest]).strip())
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 return tpl % (line_info.lspace, cmd)
@staticmethod
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 def _tr_quote(line_info):
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 "Translate lines escaped with: ,"
return '%s%s("%s")' % (line_info.lspace, line_info.fpart,
'", "'.join(line_info.rest.split()) )
@staticmethod
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 def _tr_quote2(line_info):
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 "Translate lines escaped with: ;"
return '%s%s("%s")' % (line_info.lspace, line_info.fpart,
line_info.rest)
@staticmethod
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 def _tr_paren(line_info):
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 "Translate lines escaped with: /"
return '%s%s(%s)' % (line_info.lspace, line_info.fpart,
", ".join(line_info.rest.split()))
def __call__(self, line):
"""Class to transform lines that are explicitly escaped out.
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 This calls the above _tr_* static methods for the actual line
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 translations."""
# Empty lines just get returned unmodified
if not line or line.isspace():
return line
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 # Get line endpoints, where the escapes can be
line_info = LineInfo(line)
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 # If the escape is not at the start, only '?' needs to be special-cased.
# All other escapes are only valid at the start
if not line_info.esc in self.tr:
if line.endswith(ESC_HELP):
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828 return self._tr_help(line_info)
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 else:
# If we don't recognize the escape, don't modify the line
return line
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 return self.tr[line_info.esc](line_info)
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
Fernando Perez
Final documentation changes from code review with Brian, ready to merge.
r2828
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 # A function-looking object to be used by the rest of the code. The purpose of
# the class in this case is to organize related functionality, more than to
# manage state.
transform_escaped = EscapedTransformer()
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
Fernando Perez
First pass of input syntax transformation support
r2719
class IPythonInputSplitter(InputSplitter):
"""An input splitter that recognizes all of IPython's special syntax."""
Fernando Perez
Add support for accessing raw data to inputsplitter....
r3080 # String with raw, untransformed input.
source_raw = ''
# Private attributes
# List with lines of raw input accumulated so far.
_buffer_raw = None
def __init__(self, input_mode=None):
InputSplitter.__init__(self, input_mode)
self._buffer_raw = []
def reset(self):
"""Reset the input buffer and associated state."""
InputSplitter.reset(self)
self._buffer_raw[:] = []
self.source_raw = ''
def source_raw_reset(self):
"""Return input and raw source and perform a full reset.
"""
out = self.source
out_r = self.source_raw
self.reset()
return out, out_r
Fernando Perez
First pass of input syntax transformation support
r2719 def push(self, lines):
"""Push one or more lines of IPython input.
"""
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 if not lines:
return super(IPythonInputSplitter, self).push(lines)
Fernando Perez
Stop-gap fix for crash with unicode input....
r3126 # We must ensure all input is pure unicode
if type(lines)==str:
lines = lines.decode(self.encoding)
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 lines_list = lines.splitlines()
transforms = [transform_escaped, transform_assign_system,
transform_assign_magic, transform_ipy_prompt,
transform_classic_prompt]
# Transform logic
#
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 # We only apply the line transformers to the input if we have either no
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 # input yet, or complete input, or if the last line of the buffer ends
# with ':' (opening an indented block). This prevents the accidental
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780 # transformation of escapes inside multiline expressions like
# triple-quoted strings or parenthesized expressions.
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 #
# The last heuristic, while ugly, ensures that the first line of an
# indented block is correctly transformed.
#
# FIXME: try to find a cleaner approach for this last bit.
Fernando Perez
Rename input modes of input splitter to 'line' and 'block'....
r2862 # If we were in 'block' mode, since we're going to pump the parent
Fernando Perez
Fix bug with IPythonInputSplitter in block input mode.
r2861 # class by hand line by line, we need to temporarily switch out to
Fernando Perez
Rename input modes of input splitter to 'line' and 'block'....
r2862 # 'line' mode, do a single manual reset and then feed the lines one
Fernando Perez
Fix bug with IPythonInputSplitter in block input mode.
r2861 # by one. Note that this only matters if the input has more than one
# line.
changed_input_mode = False
Fernando Perez
Add support for accessing raw data to inputsplitter....
r3080
if self.input_mode == 'cell':
Fernando Perez
Fix bug with IPythonInputSplitter in block input mode.
r2861 self.reset()
changed_input_mode = True
Fernando Perez
Implement support for 'cell' mode with Ctrl-Enter....
r3004 saved_input_mode = 'cell'
Fernando Perez
Rename input modes of input splitter to 'line' and 'block'....
r2862 self.input_mode = 'line'
Fernando Perez
Completed first pass of inputsplitter with IPython syntax....
r2780
Fernando Perez
Add support for accessing raw data to inputsplitter....
r3080 # Store raw source before applying any transformations to it. Note
# that this must be done *after* the reset() call that would otherwise
# flush the buffer.
self._store(lines, self._buffer_raw, 'source_raw')
Fernando Perez
Fix bug with IPythonInputSplitter in block input mode.
r2861 try:
push = super(IPythonInputSplitter, self).push
for line in lines_list:
if self._is_complete or not self._buffer or \
(self._buffer and self._buffer[-1].rstrip().endswith(':')):
for f in transforms:
line = f(line)
out = push(line)
finally:
if changed_input_mode:
self.input_mode = saved_input_mode
Fernando Perez
Final cleanups responding to Brian's code review....
r2782 return out