cpython/Doc/lib/libtokenize.tex

\section{\module{tokenize} ---
         Tokenizer for Python source}

\declaremodule{standard}{tokenize}
\modulesynopsis{Lexical scanner for Python source code.}
\moduleauthor{Ka Ping Yee}{}
\sectionauthor{Fred L. Drake, Jr.}{fdrake@acm.org}


The \module{tokenize} module provides a lexical scanner for Python
source code, implemented in Python.  The scanner in this module
returns comments as tokens as well, making it useful for implementing
``pretty-printers,'' including colorizers for on-screen displays.

The primary entry point is a generator:

\begin{funcdesc}{generate_tokens}{readline}
  The \function{generate_tokens()} generator requires one argment,
  \var{readline}, which must be a callable object which
  provides the same interface as the \method{readline()} method of
  built-in file objects (see section~\ref{bltin-file-objects}).  Each
  call to the function should return one line of input as a string.

  The generator produces 5-tuples with these members:
  the token type;
  the token string;
  a 2-tuple \code{(\var{srow}, \var{scol})} of ints specifying the
  row and column where the token begins in the source;
  a 2-tuple \code{(\var{erow}, \var{ecol})} of ints specifying the
  row and column where the token ends in the source;
  and the line on which the token was found.
  The line passed is the \emph{logical} line;
  continuation lines are included.
  \versionadded{2.2}
\end{funcdesc}

An older entry point is retained for backward compatibility:

\begin{funcdesc}{tokenize}{readline\optional{, tokeneater}}
  The \function{tokenize()} function accepts two parameters: one
  representing the input stream, and one providing an output mechanism
  for \function{tokenize()}.

  The first parameter, \var{readline}, must be a callable object which
  provides the same interface as the \method{readline()} method of
  built-in file objects (see section~\ref{bltin-file-objects}).  Each
  call to the function should return one line of input as a string.
  Alternately, \var{readline} may be a callable object that signals
  completion by raising \exception{StopIteration}.
  \versionchanged[Added \exception{StopIteration} support]{2.5}

  The second parameter, \var{tokeneater}, must also be a callable
  object.  It is called once for each token, with five arguments,
  corresponding to the tuples generated by \function{generate_tokens()}.
\end{funcdesc}


All constants from the \refmodule{token} module are also exported from
\module{tokenize}, as are two additional token type values that might be
passed to the \var{tokeneater} function by \function{tokenize()}:

\begin{datadesc}{COMMENT}
  Token value used to indicate a comment.
\end{datadesc}
\begin{datadesc}{NL}
  Token value used to indicate a non-terminating newline.  The NEWLINE
  token indicates the end of a logical line of Python code; NL tokens
  are generated when a logical line of code is continued over multiple
  physical lines.
\end{datadesc}

Another function is provided to reverse the tokenization process.
This is useful for creating tools that tokenize a script, modify
the token stream, and write back the modified script.

\begin{funcdesc}{untokenize}{iterable}
  Converts tokens back into Python source code.  The \var{iterable}
  must return sequences with at least two elements, the token type and
  the token string.  Any additional sequence elements are ignored.

  The reconstructed script is returned as a single string.  The
  result is guaranteed to tokenize back to match the input so that
  the conversion is lossless and round-trips are assured.  The
  guarantee applies only to the token type and token string as
  the spacing between tokens (column positions) may change.
  \versionadded{2.5}
\end{funcdesc}

Example of a script re-writer that transforms float literals into
Decimal objects:
\begin{verbatim}
def decistmt(s):
    """Substitute Decimals for floats in a string of statements.

    >>> from decimal import Decimal
    >>> s = 'print +21.3e-5*-.1234/81.7'
    >>> decistmt(s)
    "print +Decimal ('21.3e-5')*-Decimal ('.1234')/Decimal ('81.7')"

    >>> exec(s)
    -3.21716034272e-007
    >>> exec(decistmt(s))
    -3.217160342717258261933904529E-7

    """
    result = []
    g = generate_tokens(StringIO(s).readline)   # tokenize the string
    for toknum, tokval, _, _, _  in g:
        if toknum == NUMBER and '.' in tokval:  # replace NUMBER tokens
            result.extend([
                (NAME, 'Decimal'),
                (OP, '('),
                (STRING, repr(tokval)),
                (OP, ')')
            ])
        else:
            result.append((toknum, tokval))
    return untokenize(result)
\end{verbatim}