inputsplitter.py
865 lines
| 30.7 KiB
| text/x-python
|
PythonLexer
Fernando Perez
|
r2628 | """Analysis of text input into executable blocks. | ||
Fernando Perez
|
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
|
r2628 | |||
Fernando Perez
|
r2782 | A companion, :class:`IPythonInputSplitter`, provides the same functionality but | ||
with full support for the extended IPython syntax (magics, system calls, etc). | ||||
Fernando Perez
|
r2663 | For more details, see the class docstring below. | ||
Fernando Perez
|
r2780 | |||
Fernando Perez
|
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
|
r2782 | ToDo | ||
---- | ||||
Fernando Perez
|
r2828 | - Should we make push() actually raise an exception once push_accepts_more() | ||
returns False? | ||||
Fernando Perez
|
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
|
r2780 | Authors | ||
Fernando Perez
|
r2782 | ------- | ||
Fernando Perez
|
r2780 | |||
* Fernando Perez | ||||
* Brian Granger | ||||
Fernando Perez
|
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
|
r3004 | from __future__ import print_function | ||
Fernando Perez
|
r2628 | |||
#----------------------------------------------------------------------------- | ||||
# Imports | ||||
#----------------------------------------------------------------------------- | ||||
# stdlib | ||||
Thomas Kluyver
|
r3454 | import ast | ||
Fernando Perez
|
r2628 | import codeop | ||
import re | ||||
import sys | ||||
Fernando Perez
|
r2719 | # IPython modules | ||
from IPython.utils.text import make_quoted_expr | ||||
Fernando Perez
|
r3004 | |||
Fernando Perez
|
r2628 | #----------------------------------------------------------------------------- | ||
Fernando Perez
|
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
|
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
|
r2780 | |||
#----------------------------------------------------------------------------- | ||||
Fernando Perez
|
r2628 | # Utilities | ||
#----------------------------------------------------------------------------- | ||||
Fernando Perez
|
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
|
r2633 | |||
Fernando Perez
|
r2628 | # compiled regexps for autoindent management | ||
David Warde-Farley
|
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
|
r2628 | ini_spaces_re = re.compile(r'^([ \t\r\f\v]+)') | ||
Fernando Perez
|
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
|
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
|
r2663 | |||
Returns | ||||
------- | ||||
n : int | ||||
Fernando Perez
|
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
|
r2718 | """Return the default standard input encoding. | ||
If sys.stdin has no encoding, 'ascii' is returned.""" | ||||
epatters
|
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
|
r2628 | |||
#----------------------------------------------------------------------------- | ||||
Fernando Perez
|
r2780 | # Classes and functions for normal Python syntax handling | ||
Fernando Perez
|
r2628 | #----------------------------------------------------------------------------- | ||
Fernando Perez
|
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
|
r2628 | indent_spaces = 0 | ||
Fernando Perez
|
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
|
r2628 | encoding = '' | ||
Fernando Perez
|
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
|
r2628 | source = '' | ||
Fernando Perez
|
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
|
r2628 | code = None | ||
Fernando Perez
|
r2634 | # Input mode | ||
Fernando Perez
|
r2862 | input_mode = 'line' | ||
Fernando Perez
|
r2634 | |||
Fernando Perez
|
r2633 | # Private attributes | ||
Fernando Perez
|
r2663 | # List with lines of input accumulated so far | ||
Fernando Perez
|
r2633 | _buffer = None | ||
Fernando Perez
|
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
|
r2628 | |||
Fernando Perez
|
r2634 | def __init__(self, input_mode=None): | ||
Fernando Perez
|
r2663 | """Create a new InputSplitter instance. | ||
Fernando Perez
|
r2634 | |||
Parameters | ||||
---------- | ||||
input_mode : str | ||||
Fernando Perez
|
r3004 | One of ['line', 'cell']; default is 'line'. | ||
Fernando Perez
|
r2634 | |||
Fernando Perez
|
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
|
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
|
r2634 | """ | ||
Fernando Perez
|
r2633 | self._buffer = [] | ||
Fernando Perez
|
r2663 | self._compile = codeop.CommandCompiler() | ||
Fernando Perez
|
r2628 | self.encoding = get_input_encoding() | ||
Fernando Perez
|
r2663 | self.input_mode = InputSplitter.input_mode if input_mode is None \ | ||
Fernando Perez
|
r2634 | else input_mode | ||
Fernando Perez
|
r2628 | |||
def reset(self): | ||||
"""Reset the input buffer and associated state.""" | ||||
self.indent_spaces = 0 | ||||
Fernando Perez
|
r2633 | self._buffer[:] = [] | ||
Fernando Perez
|
r2628 | self.source = '' | ||
Fernando Perez
|
r2633 | self.code = None | ||
Fernando Perez
|
r2663 | self._is_complete = False | ||
self._full_dedent = False | ||||
Fernando Perez
|
r2628 | |||
Fernando Perez
|
r2636 | def source_reset(self): | ||
"""Return the input source and perform a full reset. | ||||
Fernando Perez
|
r2628 | """ | ||
out = self.source | ||||
Fernando Perez
|
r2636 | self.reset() | ||
Fernando Perez
|
r2628 | return out | ||
def push(self, lines): | ||||
Robert Kern
|
r3293 | """Push one or more lines of input. | ||
Fernando Perez
|
r2628 | |||
This stores the given lines and returns a status code indicating | ||||
whether the code forms a complete Python block or not. | ||||
Fernando Perez
|
r2663 | Any exceptions generated in compilation are swallowed, but if an | ||
exception was produced, the method returns True. | ||||
Fernando Perez
|
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
|
r2663 | this value is also stored as a private attribute (_is_complete), so it | ||
can be queried at any time. | ||||
Fernando Perez
|
r2628 | """ | ||
Fernando Perez
|
r3004 | if self.input_mode == 'cell': | ||
Fernando Perez
|
r2634 | self.reset() | ||
Fernando Perez
|
r2633 | self._store(lines) | ||
Fernando Perez
|
r2628 | source = self.source | ||
Fernando Perez
|
r2663 | # Before calling _compile(), reset the code object to None so that if an | ||
Fernando Perez
|
r2628 | # exception is raised in compilation, we don't mislead by having | ||
# inconsistent code/source attributes. | ||||
Fernando Perez
|
r2663 | self.code, self._is_complete = None, None | ||
Fernando Perez
|
r2645 | |||
Fernando Perez
|
r3013 | # Honor termination lines properly | ||
if source.rstrip().endswith('\\'): | ||||
return False | ||||
Fernando Perez
|
r2645 | self._update_indent(lines) | ||
Fernando Perez
|
r2635 | try: | ||
Thomas Kluyver
|
r3748 | self.code = self._compile(source, symbol="exec") | ||
Fernando Perez
|
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
|
r2645 | except (SyntaxError, OverflowError, ValueError, TypeError, | ||
MemoryError): | ||||
Fernando Perez
|
r2663 | self._is_complete = True | ||
Fernando Perez
|
r2635 | else: | ||
# Compilation didn't produce any exceptions (though it may not have | ||||
# given a complete code object) | ||||
Fernando Perez
|
r2663 | self._is_complete = self.code is not None | ||
Fernando Perez
|
r2635 | |||
Fernando Perez
|
r2663 | return self._is_complete | ||
Fernando Perez
|
r2628 | |||
Fernando Perez
|
r2663 | def push_accepts_more(self): | ||
"""Return whether a block of interactive input can accept more input. | ||||
Fernando Perez
|
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
|
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
|
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
|
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
|
r2628 | """ | ||
Fernando Perez
|
r3004 | |||
# With incomplete input, unconditionally accept more | ||||
Fernando Perez
|
r2663 | if not self._is_complete: | ||
Fernando Perez
|
r2628 | return True | ||
Fernando Perez
|
r3004 | # If we already have complete input and we're flush left, the answer | ||
Thomas Kluyver
|
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
|
r2663 | if self.indent_spaces==0: | ||
Fernando Perez
|
r3004 | if self.input_mode=='line': | ||
Thomas Kluyver
|
r3461 | if not self._full_dedent: | ||
return False | ||||
Fernando Perez
|
r3004 | else: | ||
Thomas Kluyver
|
r3526 | try: | ||
Thomas Kluyver
|
r3528 | code_ast = ast.parse(u''.join(self._buffer)) | ||
Thomas Kluyver
|
r3526 | except Exception: | ||
Fernando Perez
|
r3004 | return False | ||
Thomas Kluyver
|
r3526 | else: | ||
Thomas Kluyver
|
r3528 | if len(code_ast.body) == 1: | ||
Thomas Kluyver
|
r3526 | return False | ||
Fernando Perez
|
r3004 | |||
# When input is complete, then termination is marked by an extra blank | ||||
# line at the end. | ||||
Fernando Perez
|
r2663 | last_line = self.source.splitlines()[-1] | ||
return bool(last_line and not last_line.isspace()) | ||||
Fernando Perez
|
r2628 | |||
Fernando Perez
|
r2633 | #------------------------------------------------------------------------ | ||
# Private interface | ||||
#------------------------------------------------------------------------ | ||||
Fernando Perez
|
r2628 | |||
Fernando Perez
|
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
|
r2663 | full_dedent = self._full_dedent | ||
Fernando Perez
|
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
|
r2633 | |||
Fernando Perez
|
r2645 | return indent_spaces, full_dedent | ||
Fernando Perez
|
r3085 | |||
Fernando Perez
|
r2645 | def _update_indent(self, lines): | ||
for line in remove_comments(lines).splitlines(): | ||||
Fernando Perez
|
r2633 | if line and not line.isspace(): | ||
Fernando Perez
|
r2663 | self.indent_spaces, self._full_dedent = self._find_indent(line) | ||
Fernando Perez
|
r2628 | |||
Fernando Perez
|
r3080 | def _store(self, lines, buffer=None, store='source'): | ||
Fernando Perez
|
r2633 | """Store one or more lines of input. | ||
If input lines are not newline-terminated, a newline is automatically | ||||
appended.""" | ||||
Thomas Kluyver
|
r3448 | |||
Fernando Perez
|
r3080 | if buffer is None: | ||
buffer = self._buffer | ||||
Fernando Perez
|
r2633 | if lines.endswith('\n'): | ||
Fernando Perez
|
r3080 | buffer.append(lines) | ||
Fernando Perez
|
r2633 | else: | ||
Fernando Perez
|
r3080 | buffer.append(lines+'\n') | ||
setattr(self, store, self._set_source(buffer)) | ||||
Fernando Perez
|
r2645 | |||
Fernando Perez
|
r3080 | def _set_source(self, buffer): | ||
Thomas Kluyver
|
r3455 | return u''.join(buffer) | ||
Fernando Perez
|
r2719 | |||
#----------------------------------------------------------------------------- | ||||
Fernando Perez
|
r2780 | # Functions and classes for IPython-specific syntactic support | ||
Fernando Perez
|
r2719 | #----------------------------------------------------------------------------- | ||
Fernando Perez
|
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
|
r3068 | \s*(%?[\w\.\*]*) # function/method, possibly with leading % | ||
Fernando Perez
|
r2879 | # to correctly treat things like '?%magic' | ||
Fernando Perez
|
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
|
r2879 | >>> split_user_input('?%hist') | ||
('', '?', '%hist', '') | ||||
Fernando Perez
|
r3068 | >>> split_user_input('?x*') | ||
('', '?', 'x*', '') | ||||
Fernando Perez
|
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
|
r2782 | fpart, rest = line.split(None, 1) | ||
Fernando Perez
|
r2780 | except ValueError: | ||
# print "split failed for line '%s'" % line | ||||
fpart, rest = line,'' | ||||
Fernando Perez
|
r2782 | lspace = re.match('^(\s*)(.*)', line).groups()[0] | ||
Fernando Perez
|
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
|
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
|
r3002 | expr = make_quoted_expr(cmd) | ||
new_line = '%s = get_ipython().getoutput(%s)' % (lhs, expr) | ||||
Fernando Perez
|
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
|
r2780 | _classic_prompt_re = re.compile(r'^([ \t]*>>> |^[ \t]*\.\.\. )') | ||
Fernando Perez
|
r2719 | |||
def transform_classic_prompt(line): | ||||
"""Handle inputs that start with '>>> ' syntax.""" | ||||
Fernando Perez
|
r2780 | if not line or line.isspace(): | ||
return line | ||||
Fernando Perez
|
r2719 | m = _classic_prompt_re.match(line) | ||
if m: | ||||
return line[len(m.group(0)):] | ||||
else: | ||||
return line | ||||
Fernando Perez
|
r2780 | _ipy_prompt_re = re.compile(r'^([ \t]*In \[\d+\]: |^[ \t]*\ \ \ \.\.\.+: )') | ||
Fernando Perez
|
r2719 | |||
def transform_ipy_prompt(line): | ||||
"""Handle inputs that start classic IPython prompt syntax.""" | ||||
Fernando Perez
|
r2780 | if not line or line.isspace(): | ||
return line | ||||
Fernando Perez
|
r2861 | #print 'LINE: %r' % line # dbg | ||
Fernando Perez
|
r2719 | m = _ipy_prompt_re.match(line) | ||
if m: | ||||
Fernando Perez
|
r2861 | #print 'MATCH! %r -> %r' % (line, line[len(m.group(0)):]) # dbg | ||
Fernando Perez
|
r2719 | return line[len(m.group(0)):] | ||
else: | ||||
return line | ||||
Fernando Perez
|
r2782 | class EscapedTransformer(object): | ||
"""Class to transform lines that are explicitly escaped out.""" | ||||
Fernando Perez
|
r2780 | |||
Fernando Perez
|
r2782 | def __init__(self): | ||
Fernando Perez
|
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
|
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
|
r2828 | def _tr_system(line_info): | ||
Fernando Perez
|
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
|
r2828 | def _tr_system2(line_info): | ||
Fernando Perez
|
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
|
r2828 | def _tr_help(line_info): | ||
Fernando Perez
|
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
|
r3068 | l_ori = line_info | ||
Fernando Perez
|
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
|
r3068 | # From here on, simply choose which level of detail to get, and | ||
# special-case the psearch syntax | ||||
Fernando Perez
|
r3098 | pinfo = 'pinfo' # default | ||
Fernando Perez
|
r3068 | if '*' in line_info.line: | ||
pinfo = 'psearch' | ||||
Fernando Perez
|
r2782 | elif line_info.esc == '??': | ||
pinfo = 'pinfo2' | ||||
Thomas Kluyver
|
r3471 | tpl = '%sget_ipython().magic(u"%s %s")' | ||
Fernando Perez
|
r2782 | return tpl % (line_info.lspace, pinfo, | ||
' '.join([line_info.fpart, line_info.rest]).strip()) | ||||
@staticmethod | ||||
Fernando Perez
|
r2828 | def _tr_magic(line_info): | ||
Fernando Perez
|
r2782 | "Translate lines escaped with: %" | ||
tpl = '%sget_ipython().magic(%s)' | ||||
cmd = make_quoted_expr(' '.join([line_info.fpart, | ||||
Fernando Perez
|
r2821 | line_info.rest]).strip()) | ||
Fernando Perez
|
r2782 | return tpl % (line_info.lspace, cmd) | ||
@staticmethod | ||||
Fernando Perez
|
r2828 | def _tr_quote(line_info): | ||
Fernando Perez
|
r2782 | "Translate lines escaped with: ," | ||
return '%s%s("%s")' % (line_info.lspace, line_info.fpart, | ||||
'", "'.join(line_info.rest.split()) ) | ||||
@staticmethod | ||||
Fernando Perez
|
r2828 | def _tr_quote2(line_info): | ||
Fernando Perez
|
r2782 | "Translate lines escaped with: ;" | ||
return '%s%s("%s")' % (line_info.lspace, line_info.fpart, | ||||
line_info.rest) | ||||
@staticmethod | ||||
Fernando Perez
|
r2828 | def _tr_paren(line_info): | ||
Fernando Perez
|
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
|
r2828 | This calls the above _tr_* static methods for the actual line | ||
Fernando Perez
|
r2782 | translations.""" | ||
# Empty lines just get returned unmodified | ||||
if not line or line.isspace(): | ||||
return line | ||||
Fernando Perez
|
r2780 | |||
Fernando Perez
|
r2782 | # Get line endpoints, where the escapes can be | ||
line_info = LineInfo(line) | ||||
Fernando Perez
|
r2780 | |||
Fernando Perez
|
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
|
r2828 | return self._tr_help(line_info) | ||
Fernando Perez
|
r2782 | else: | ||
# If we don't recognize the escape, don't modify the line | ||||
return line | ||||
Fernando Perez
|
r2780 | |||
Fernando Perez
|
r2782 | return self.tr[line_info.esc](line_info) | ||
Fernando Perez
|
r2780 | |||
Fernando Perez
|
r2828 | |||
Fernando Perez
|
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
|
r2780 | |||
Fernando Perez
|
r2719 | |||
class IPythonInputSplitter(InputSplitter): | ||||
"""An input splitter that recognizes all of IPython's special syntax.""" | ||||
Fernando Perez
|
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
|
r2719 | def push(self, lines): | ||
"""Push one or more lines of IPython input. | ||||
""" | ||||
Fernando Perez
|
r2782 | if not lines: | ||
return super(IPythonInputSplitter, self).push(lines) | ||||
Fernando Perez
|
r3126 | # We must ensure all input is pure unicode | ||
if type(lines)==str: | ||||
lines = lines.decode(self.encoding) | ||||
Fernando Perez
|
r2782 | lines_list = lines.splitlines() | ||
transforms = [transform_escaped, transform_assign_system, | ||||
transform_assign_magic, transform_ipy_prompt, | ||||
transform_classic_prompt] | ||||
# Transform logic | ||||
# | ||||
Fernando Perez
|
r2780 | # We only apply the line transformers to the input if we have either no | ||
Fernando Perez
|
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
|
r2780 | # transformation of escapes inside multiline expressions like | ||
# triple-quoted strings or parenthesized expressions. | ||||
Fernando Perez
|
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
|
r2862 | # If we were in 'block' mode, since we're going to pump the parent | ||
Fernando Perez
|
r2861 | # class by hand line by line, we need to temporarily switch out to | ||
Fernando Perez
|
r2862 | # 'line' mode, do a single manual reset and then feed the lines one | ||
Fernando Perez
|
r2861 | # by one. Note that this only matters if the input has more than one | ||
# line. | ||||
changed_input_mode = False | ||||
Fernando Perez
|
r3080 | |||
if self.input_mode == 'cell': | ||||
Fernando Perez
|
r2861 | self.reset() | ||
changed_input_mode = True | ||||
Fernando Perez
|
r3004 | saved_input_mode = 'cell' | ||
Fernando Perez
|
r2862 | self.input_mode = 'line' | ||
Fernando Perez
|
r2780 | |||
Fernando Perez
|
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
|
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
|
r2782 | return out | ||