libparser.tex 30.3 KB
Newer Older
1 2 3
\section{\module{parser} ---
         Access Python parse trees}

4 5 6 7 8 9 10
% Copyright 1995 Virginia Polytechnic Institute and State University
% and Fred L. Drake, Jr.  This copyright notice must be distributed on
% all copies, but this document otherwise may be distributed as part
% of the Python distribution.  No fee may be charged for this document
% in any representation, either on paper or electronically.  This
% restriction does not affect other elements in a distributed package
% in any way.
11

12
\declaremodule{builtin}{parser}
13
\modulesynopsis{Access parse trees for Python source code.}
Fred Drake's avatar
Fred Drake committed
14 15
\moduleauthor{Fred L. Drake, Jr.}{fdrake@acm.org}
\sectionauthor{Fred L. Drake, Jr.}{fdrake@acm.org}
16 17


18
\index{parsing!Python source code}
19

20
The \module{parser} module provides an interface to Python's internal
21 22
parser and byte-code compiler.  The primary purpose for this interface
is to allow Python code to edit the parse tree of a Python expression
23 24 25 26
and create executable code from this.  This is better than trying
to parse and modify an arbitrary Python code fragment as a string
because parsing is performed in a manner identical to the code
forming the application.  It is also faster.
27 28 29

There are a few things to note about this module which are important
to making use of the data structures created.  This is not a tutorial
30
on editing the parse trees for Python code, but some examples of using
31
the \module{parser} module are presented.
32 33 34

Most importantly, a good understanding of the Python grammar processed
by the internal parser is required.  For full information on the
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
language syntax, refer to the \citetitle[../ref/ref.html]{Python
Language Reference}.  The parser itself is created from a grammar
specification defined in the file \file{Grammar/Grammar} in the
standard Python distribution.  The parse trees stored in the AST
objects created by this module are the actual output from the internal
parser when created by the \function{expr()} or \function{suite()}
functions, described below.  The AST objects created by
\function{sequence2ast()} faithfully simulate those structures.  Be
aware that the values of the sequences which are considered
``correct'' will vary from one version of Python to another as the
formal grammar for the language is revised.  However, transporting
code from one Python version to another as source text will always
allow correct parse trees to be created in the target version, with
the only restriction being that migrating to an older version of the
interpreter will not support more recent language constructs.  The
parse trees are not typically compatible from one version to another,
whereas source code has always been forward-compatible.
Guido van Rossum's avatar
Guido van Rossum committed
52

53 54
Each element of the sequences returned by \function{ast2list()} or
\function{ast2tuple()} has a simple form.  Sequences representing
Guido van Rossum's avatar
Guido van Rossum committed
55 56 57
non-terminal elements in the grammar always have a length greater than
one.  The first element is an integer which identifies a production in
the grammar.  These integers are given symbolic names in the C header
58
file \file{Include/graminit.h} and the Python module
59
\refmodule{symbol}.  Each additional element of the sequence represents
Guido van Rossum's avatar
Guido van Rossum committed
60 61 62 63
a component of the production as recognized in the input string: these
are always sequences which have the same form as the parent.  An
important aspect of this structure which should be noted is that
keywords used to identify the parent node type, such as the keyword
64 65
\keyword{if} in an \constant{if_stmt}, are included in the node tree without
any special treatment.  For example, the \keyword{if} keyword is
66
represented by the tuple \code{(1, 'if')}, where \code{1} is the
Fred Drake's avatar
Fred Drake committed
67
numeric value associated with all \constant{NAME} tokens, including
Guido van Rossum's avatar
Guido van Rossum committed
68 69 70 71
variable and function names defined by the user.  In an alternate form
returned when line number information is requested, the same token
might be represented as \code{(1, 'if', 12)}, where the \code{12}
represents the line number at which the terminal symbol was found.
72 73 74

Terminal elements are represented in much the same way, but without
any child elements and the addition of the source text which was
75
identified.  The example of the \keyword{if} keyword above is
76
representative.  The various types of terminal symbols are defined in
77
the C header file \file{Include/token.h} and the Python module
78
\refmodule{token}.
79 80 81 82 83 84 85 86

The AST objects are not required to support the functionality of this
module, but are provided for three purposes: to allow an application
to amortize the cost of processing complex parse trees, to provide a
parse tree representation which conserves memory space when compared
to the Python list or tuple representation, and to ease the creation
of additional modules in C which manipulate parse trees.  A simple
``wrapper'' class may be created in Python to hide the use of AST
87
objects.
88

89
The \module{parser} module defines functions for a few distinct
90 91 92 93
purposes.  The most important purposes are to create AST objects and
to convert AST objects to other representations such as parse trees
and compiled code objects, but there are also functions which serve to
query the type of parse tree represented by an AST object.
94 95


96 97 98 99 100 101 102 103
\begin{seealso}
  \seemodule{symbol}{Useful constants representing internal nodes of
                     the parse tree.}
  \seemodule{token}{Useful constants representing leaf nodes of the
                    parse tree and functions for testing node values.}
\end{seealso}


Fred Drake's avatar
Fred Drake committed
104
\subsection{Creating AST Objects \label{Creating ASTs}}
105 106 107 108 109

AST objects may be created from source code or from a parse tree.
When creating an AST object from source, different functions are used
to create the \code{'eval'} and \code{'exec'} forms.

110 111
\begin{funcdesc}{expr}{source}
The \function{expr()} function parses the parameter \var{source}
112 113 114 115
as if it were an input to \samp{compile(\var{source}, 'file.py',
'eval')}.  If the parse succeeds, an AST object is created to hold the
internal parse tree representation, otherwise an appropriate exception
is thrown.
116 117
\end{funcdesc}

118 119
\begin{funcdesc}{suite}{source}
The \function{suite()} function parses the parameter \var{source}
120 121 122 123
as if it were an input to \samp{compile(\var{source}, 'file.py',
'exec')}.  If the parse succeeds, an AST object is created to hold the
internal parse tree representation, otherwise an appropriate exception
is thrown.
124 125 126 127 128 129 130 131 132
\end{funcdesc}

\begin{funcdesc}{sequence2ast}{sequence}
This function accepts a parse tree represented as a sequence and
builds an internal representation if possible.  If it can validate
that the tree conforms to the Python grammar and all nodes are valid
node types in the host version of Python, an AST object is created
from the internal representation and returned to the called.  If there
is a problem creating the internal representation, or if the tree
133
cannot be validated, a \exception{ParserError} exception is thrown.  An AST
134 135
object created this way should not be assumed to compile correctly;
normal exceptions thrown by compilation may still be initiated when
136 137
the AST object is passed to \function{compileast()}.  This may indicate
problems not related to syntax (such as a \exception{MemoryError}
138 139 140 141 142 143 144 145 146 147 148 149 150
exception), but may also be due to constructs such as the result of
parsing \code{del f(0)}, which escapes the Python parser but is
checked by the bytecode compiler.

Sequences representing terminal tokens may be represented as either
two-element lists of the form \code{(1, 'name')} or as three-element
lists of the form \code{(1, 'name', 56)}.  If the third element is
present, it is assumed to be a valid line number.  The line number
may be specified for any subset of the terminal symbols in the input
tree.
\end{funcdesc}

\begin{funcdesc}{tuple2ast}{sequence}
151
This is the same function as \function{sequence2ast()}.  This entry point
152 153
is maintained for backward compatibility.
\end{funcdesc}
154 155


Fred Drake's avatar
Fred Drake committed
156
\subsection{Converting AST Objects \label{Converting ASTs}}
157 158 159 160 161 162

AST objects, regardless of the input used to create them, may be
converted to parse trees represented as list- or tuple- trees, or may
be compiled into executable code objects.  Parse trees may be
extracted with or without line numbering information.

Fred Drake's avatar
Fred Drake committed
163
\begin{funcdesc}{ast2list}{ast\optional{, line_info}}
164
This function accepts an AST object from the caller in
Fred Drake's avatar
Fred Drake committed
165
\var{ast} and returns a Python list representing the
166
equivalent parse tree.  The resulting list representation can be used
167 168 169
for inspection or the creation of a new parse tree in list form.  This
function does not fail so long as memory is available to build the
list representation.  If the parse tree will only be used for
170
inspection, \function{ast2tuple()} should be used instead to reduce memory
171 172 173 174
consumption and fragmentation.  When the list representation is
required, this function is significantly faster than retrieving a
tuple representation and converting that to nested lists.

Fred Drake's avatar
Fred Drake committed
175
If \var{line_info} is true, line number information will be
176
included for all terminal tokens as a third element of the list
177
representing the token.  Note that the line number provided specifies
178
the line on which the token \emph{ends}.  This information is
179
omitted if the flag is false or omitted.
180 181
\end{funcdesc}

Fred Drake's avatar
Fred Drake committed
182
\begin{funcdesc}{ast2tuple}{ast\optional{, line_info}}
Guido van Rossum's avatar
Guido van Rossum committed
183
This function accepts an AST object from the caller in
Fred Drake's avatar
Fred Drake committed
184
\var{ast} and returns a Python tuple representing the
185
equivalent parse tree.  Other than returning a tuple instead of a
186
list, this function is identical to \function{ast2list()}.
Guido van Rossum's avatar
Guido van Rossum committed
187

Fred Drake's avatar
Fred Drake committed
188
If \var{line_info} is true, line number information will be
189 190 191
included for all terminal tokens as a third element of the list
representing the token.  This information is omitted if the flag is
false or omitted.
Guido van Rossum's avatar
Guido van Rossum committed
192
\end{funcdesc}
193

194
\begin{funcdesc}{compileast}{ast\optional{, filename\code{ = '<ast>'}}}
195
The Python byte compiler can be invoked on an AST object to produce
Fred Drake's avatar
Fred Drake committed
196
code objects which can be used as part of an \keyword{exec} statement or
197 198
a call to the built-in \function{eval()}\bifuncindex{eval} function.
This function provides the interface to the compiler, passing the
Fred Drake's avatar
Fred Drake committed
199 200 201
internal parse tree from \var{ast} to the parser, using the
source file name specified by the \var{filename} parameter.
The default value supplied for \var{filename} indicates that
202
the source was an AST object.
Guido van Rossum's avatar
Guido van Rossum committed
203 204

Compiling an AST object may result in exceptions related to
205
compilation; an example would be a \exception{SyntaxError} caused by the
206
parse tree for \code{del f(0)}: this statement is considered legal
Guido van Rossum's avatar
Guido van Rossum committed
207
within the formal grammar for Python but is not a legal language
208
construct.  The \exception{SyntaxError} raised for this condition is
Guido van Rossum's avatar
Guido van Rossum committed
209
actually generated by the Python byte-compiler normally, which is why
210
it can be raised at this point by the \module{parser} module.  Most
Guido van Rossum's avatar
Guido van Rossum committed
211 212
causes of compilation failure can be diagnosed programmatically by
inspection of the parse tree.
213 214 215
\end{funcdesc}


Fred Drake's avatar
Fred Drake committed
216
\subsection{Queries on AST Objects \label{Querying ASTs}}
217

218
Two functions are provided which allow an application to determine if
Fred Drake's avatar
Fred Drake committed
219
an AST was created as an expression or a suite.  Neither of these
220
functions can be used to determine if an AST was created from source
221 222
code via \function{expr()} or \function{suite()} or from a parse tree
via \function{sequence2ast()}.
223 224

\begin{funcdesc}{isexpr}{ast}
Fred Drake's avatar
Fred Drake committed
225
When \var{ast} represents an \code{'eval'} form, this function
226 227 228 229 230 231
returns true, otherwise it returns false.  This is useful, since code
objects normally cannot be queried for this information using existing
built-in functions.  Note that the code objects created by
\function{compileast()} cannot be queried like this either, and are
identical to those created by the built-in
\function{compile()}\bifuncindex{compile} function.
232 233 234 235
\end{funcdesc}


\begin{funcdesc}{issuite}{ast}
236
This function mirrors \function{isexpr()} in that it reports whether an
237
AST object represents an \code{'exec'} form, commonly known as a
238
``suite.''  It is not safe to assume that this function is equivalent
239
to \samp{not isexpr(\var{ast})}, as additional syntactic fragments may
240
be supported in the future.
241 242 243
\end{funcdesc}


Fred Drake's avatar
Fred Drake committed
244
\subsection{Exceptions and Error Handling \label{AST Errors}}
245 246 247 248 249 250 251 252 253

The parser module defines a single exception, but may also pass other
built-in exceptions from other portions of the Python runtime
environment.  See each function for information about the exceptions
it can raise.

\begin{excdesc}{ParserError}
Exception raised when a failure occurs within the parser module.  This
is generally produced for validation failures rather than the built in
254
\exception{SyntaxError} thrown during normal parsing.
255
The exception argument is either a string describing the reason of the
Guido van Rossum's avatar
Guido van Rossum committed
256
failure or a tuple containing a sequence causing the failure from a parse
257 258
tree passed to \function{sequence2ast()} and an explanatory string.  Calls to
\function{sequence2ast()} need to be able to handle either type of exception,
259 260 261 262
while calls to other functions in the module will only need to be
aware of the simple string values.
\end{excdesc}

263 264
Note that the functions \function{compileast()}, \function{expr()}, and
\function{suite()} may throw exceptions which are normally thrown by the
265
parsing and compilation process.  These include the built in
266 267
exceptions \exception{MemoryError}, \exception{OverflowError},
\exception{SyntaxError}, and \exception{SystemError}.  In these cases, these
268 269 270 271
exceptions carry all the meaning normally associated with them.  Refer
to the descriptions of each function for detailed information.


Fred Drake's avatar
Fred Drake committed
272
\subsection{AST Objects \label{AST Objects}}
Guido van Rossum's avatar
Guido van Rossum committed
273

274
Ordered and equality comparisons are supported between AST objects.
275
Pickling of AST objects (using the \refmodule{pickle} module) is also
276
supported.
277

Fred Drake's avatar
Fred Drake committed
278 279 280 281
\begin{datadesc}{ASTType}
The type of the objects returned by \function{expr()},
\function{suite()} and \function{sequence2ast()}.
\end{datadesc}
Guido van Rossum's avatar
Guido van Rossum committed
282 283


284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307
AST objects have the following methods:


\begin{methoddesc}[AST]{compile}{\optional{filename}}
Same as \code{compileast(\var{ast}, \var{filename})}.
\end{methoddesc}

\begin{methoddesc}[AST]{isexpr}{}
Same as \code{isexpr(\var{ast})}.
\end{methoddesc}

\begin{methoddesc}[AST]{issuite}{}
Same as \code{issuite(\var{ast})}.
\end{methoddesc}

\begin{methoddesc}[AST]{tolist}{\optional{line_info}}
Same as \code{ast2list(\var{ast}, \var{line_info})}.
\end{methoddesc}

\begin{methoddesc}[AST]{totuple}{\optional{line_info}}
Same as \code{ast2tuple(\var{ast}, \var{line_info})}.
\end{methoddesc}


Fred Drake's avatar
Fred Drake committed
308
\subsection{Examples \label{AST Examples}}
309

Guido van Rossum's avatar
Guido van Rossum committed
310 311
The parser modules allows operations to be performed on the parse tree
of Python source code before the bytecode is generated, and provides
312 313
for inspection of the parse tree for information gathering purposes.
Two examples are presented.  The simple example demonstrates emulation
314 315 316
of the \function{compile()}\bifuncindex{compile} built-in function and
the complex example shows the use of a parse tree for information
discovery.
317

318
\subsubsection{Emulation of \function{compile()}}
319 320

While many useful operations may take place between parsing and
Guido van Rossum's avatar
Guido van Rossum committed
321
bytecode generation, the simplest operation is to do nothing.  For
322
this purpose, using the \module{parser} module to produce an
323
intermediate data structure is equivalent to the code
Guido van Rossum's avatar
Guido van Rossum committed
324

325
\begin{verbatim}
326
>>> code = compile('a + 5', 'file.py', 'eval')
Guido van Rossum's avatar
Guido van Rossum committed
327 328 329
>>> a = 5
>>> eval(code)
10
330
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
331

332
The equivalent operation using the \module{parser} module is somewhat
Guido van Rossum's avatar
Guido van Rossum committed
333 334
longer, and allows the intermediate internal parse tree to be retained
as an AST object:
335

336
\begin{verbatim}
337 338
>>> import parser
>>> ast = parser.expr('a + 5')
339
>>> code = ast.compile('file.py')
340 341 342
>>> a = 5
>>> eval(code)
10
343
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
344

345 346 347
An application which needs both AST and code objects can package this
code into readily available functions:

348
\begin{verbatim}
349 350 351 352
import parser

def load_suite(source_string):
    ast = parser.suite(source_string)
Fred Drake's avatar
Fred Drake committed
353
    return ast, ast.compile()
354 355 356

def load_expression(source_string):
    ast = parser.expr(source_string)
Fred Drake's avatar
Fred Drake committed
357
    return ast, ast.compile()
358
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
359

360 361
\subsubsection{Information Discovery}

362 363
Some applications benefit from direct access to the parse tree.  The
remainder of this section demonstrates how the parse tree provides
Fred Drake's avatar
Fred Drake committed
364 365 366 367 368
access to module documentation defined in
docstrings\index{string!documentation}\index{docstrings} without
requiring that the code being examined be loaded into a running
interpreter via \keyword{import}.  This can be very useful for
performing analyses of untrusted code.
Guido van Rossum's avatar
Guido van Rossum committed
369 370 371

Generally, the example will demonstrate how the parse tree may be
traversed to distill interesting information.  Two functions and a set
372
of classes are developed which provide programmatic access to high
Guido van Rossum's avatar
Guido van Rossum committed
373 374 375 376 377 378 379
level function and class definitions provided by a module.  The
classes extract information from the parse tree and provide access to
the information at a useful semantic level, one function provides a
simple low-level pattern matching capability, and the other function
defines a high-level interface to the classes by handling file
operations on behalf of the caller.  All source files mentioned here
which are not part of the Python installation are located in the
380
\file{Demo/parser/} directory of the distribution.
Guido van Rossum's avatar
Guido van Rossum committed
381

382 383 384 385
The dynamic nature of Python allows the programmer a great deal of
flexibility, but most modules need only a limited measure of this when
defining classes, functions, and methods.  In this example, the only
definitions that will be considered are those which are defined in the
386
top level of their context, e.g., a function defined by a \keyword{def}
387
statement at column zero of a module, but not a function defined
Fred Drake's avatar
Fred Drake committed
388
within a branch of an \keyword{if} ... \keyword{else} construct, though
389 390 391
there are some good reasons for doing so in some situations.  Nesting
of definitions will be handled by the code developed in the example.

Guido van Rossum's avatar
Guido van Rossum committed
392 393
To construct the upper-level extraction methods, we need to know what
the parse tree structure looks like and how much of it we actually
394
need to be concerned about.  Python uses a moderately deep parse tree
Guido van Rossum's avatar
Guido van Rossum committed
395 396 397 398
so there are a large number of intermediate nodes.  It is important to
read and understand the formal grammar used by Python.  This is
specified in the file \file{Grammar/Grammar} in the distribution.
Consider the simplest case of interest when searching for docstrings:
399 400
a module consisting of a docstring and nothing else.  (See file
\file{docstring.py}.)
401

402
\begin{verbatim}
Guido van Rossum's avatar
Guido van Rossum committed
403 404
"""Some documentation.
"""
405
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
406

Guido van Rossum's avatar
Guido van Rossum committed
407 408
Using the interpreter to take a look at the parse tree, we find a
bewildering mass of numbers and parentheses, with the documentation
409
buried deep in nested tuples.
410

411
\begin{verbatim}
Guido van Rossum's avatar
Guido van Rossum committed
412 413 414
>>> import parser
>>> import pprint
>>> ast = parser.suite(open('docstring.py').read())
Fred Drake's avatar
Fred Drake committed
415
>>> tup = ast.totuple()
Guido van Rossum's avatar
Guido van Rossum committed
416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434
>>> pprint.pprint(tup)
(257,
 (264,
  (265,
   (266,
    (267,
     (307,
      (287,
       (288,
        (289,
         (290,
          (292,
           (293,
            (294,
             (295,
              (296,
               (297,
                (298,
                 (299,
435
                  (300, (3, '"""Some documentation.\n"""'))))))))))))))))),
Guido van Rossum's avatar
Guido van Rossum committed
436 437 438
   (4, ''))),
 (4, ''),
 (0, ''))
439
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
440

Guido van Rossum's avatar
Guido van Rossum committed
441 442 443 444
The numbers at the first element of each node in the tree are the node
types; they map directly to terminal and non-terminal symbols in the
grammar.  Unfortunately, they are represented as integers in the
internal representation, and the Python structures generated do not
445
change that.  However, the \refmodule{symbol} and \refmodule{token} modules
Guido van Rossum's avatar
Guido van Rossum committed
446 447 448 449 450
provide symbolic names for the node types and dictionaries which map
from the integers to the symbolic names for the node types.

In the output presented above, the outermost tuple contains four
elements: the integer \code{257} and three additional tuples.  Node
451
type \code{257} has the symbolic name \constant{file_input}.  Each of
Guido van Rossum's avatar
Guido van Rossum committed
452 453
these inner tuples contains an integer as the first element; these
integers, \code{264}, \code{4}, and \code{0}, represent the node types
454 455
\constant{stmt}, \constant{NEWLINE}, and \constant{ENDMARKER},
respectively.
Guido van Rossum's avatar
Guido van Rossum committed
456 457 458 459
Note that these values may change depending on the version of Python
you are using; consult \file{symbol.py} and \file{token.py} for
details of the mapping.  It should be fairly clear that the outermost
node is related primarily to the input source rather than the contents
460
of the file, and may be disregarded for the moment.  The \constant{stmt}
Guido van Rossum's avatar
Guido van Rossum committed
461 462 463 464 465 466 467 468 469
node is much more interesting.  In particular, all docstrings are
found in subtrees which are formed exactly as this node is formed,
with the only difference being the string itself.  The association
between the docstring in a similar tree and the defined entity (class,
function, or module) which it describes is given by the position of
the docstring subtree within the tree defining the described
structure.

By replacing the actual docstring with something to signify a variable
470
component of the tree, we allow a simple pattern matching approach to
471
check any given subtree for equivalence to the general pattern for
472 473 474 475
docstrings.  Since the example demonstrates information extraction, we
can safely require that the tree be in tuple form rather than list
form, allowing a simple variable representation to be
\code{['variable_name']}.  A simple recursive function can implement
Guido van Rossum's avatar
Guido van Rossum committed
476
the pattern matching, returning a boolean and a dictionary of variable
477
name to value mappings.  (See file \file{example.py}.)
Guido van Rossum's avatar
Guido van Rossum committed
478

479
\begin{verbatim}
Guido van Rossum's avatar
Guido van Rossum committed
480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496
from types import ListType, TupleType

def match(pattern, data, vars=None):
    if vars is None:
        vars = {}
    if type(pattern) is ListType:
        vars[pattern[0]] = data
        return 1, vars
    if type(pattern) is not TupleType:
        return (pattern == data), vars
    if len(data) != len(pattern):
        return 0, vars
    for pattern, data in map(None, pattern, data):
        same, vars = match(pattern, data, vars)
        if not same:
            break
    return same, vars
497
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
498

499
Using this simple representation for syntactic variables and the symbolic
500 501
node types, the pattern for the candidate docstring subtrees becomes
fairly readable.  (See file \file{example.py}.)
502

503
\begin{verbatim}
504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529
import symbol
import token

DOCSTRING_STMT_PATTERN = (
    symbol.stmt,
    (symbol.simple_stmt,
     (symbol.small_stmt,
      (symbol.expr_stmt,
       (symbol.testlist,
        (symbol.test,
         (symbol.and_test,
          (symbol.not_test,
           (symbol.comparison,
            (symbol.expr,
             (symbol.xor_expr,
              (symbol.and_expr,
               (symbol.shift_expr,
                (symbol.arith_expr,
                 (symbol.term,
                  (symbol.factor,
                   (symbol.power,
                    (symbol.atom,
                     (token.STRING, ['docstring'])
                     )))))))))))))))),
     (token.NEWLINE, '')
     ))
530
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
531

532
Using the \function{match()} function with this pattern, extracting the
Guido van Rossum's avatar
Guido van Rossum committed
533 534
module docstring from the parse tree created previously is easy:

535
\begin{verbatim}
Guido van Rossum's avatar
Guido van Rossum committed
536 537 538 539
>>> found, vars = match(DOCSTRING_STMT_PATTERN, tup[1])
>>> found
1
>>> vars
540
{'docstring': '"""Some documentation.\n"""'}
541
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
542

Guido van Rossum's avatar
Guido van Rossum committed
543 544 545
Once specific data can be extracted from a location where it is
expected, the question of where information can be expected
needs to be answered.  When dealing with docstrings, the answer is
546 547 548 549
fairly simple: the docstring is the first \constant{stmt} node in a code
block (\constant{file_input} or \constant{suite} node types).  A module
consists of a single \constant{file_input} node, and class and function
definitions each contain exactly one \constant{suite} node.  Classes and
Guido van Rossum's avatar
Guido van Rossum committed
550 551 552
functions are readily identified as subtrees of code block nodes which
start with \code{(stmt, (compound_stmt, (classdef, ...} or
\code{(stmt, (compound_stmt, (funcdef, ...}.  Note that these subtrees
553
cannot be matched by \function{match()} since it does not support multiple
Guido van Rossum's avatar
Guido van Rossum committed
554 555 556 557
sibling nodes to match without regard to number.  A more elaborate
matching function could be used to overcome this limitation, but this
is sufficient for the example.

558 559 560 561 562 563 564 565 566 567 568
Given the ability to determine whether a statement might be a
docstring and extract the actual string from the statement, some work
needs to be performed to walk the parse tree for an entire module and
extract information about the names defined in each context of the
module and associate any docstrings with the names.  The code to
perform this work is not complicated, but bears some explanation.

The public interface to the classes is straightforward and should
probably be somewhat more flexible.  Each ``major'' block of the
module is described by an object providing several methods for inquiry
and a constructor which accepts at least the subtree of the complete
569 570
parse tree which it represents.  The \class{ModuleInfo} constructor
accepts an optional \var{name} parameter since it cannot
571 572
otherwise determine the name of the module.

573 574 575 576 577 578 579
The public classes include \class{ClassInfo}, \class{FunctionInfo},
and \class{ModuleInfo}.  All objects provide the
methods \method{get_name()}, \method{get_docstring()},
\method{get_class_names()}, and \method{get_class_info()}.  The
\class{ClassInfo} objects support \method{get_method_names()} and
\method{get_method_info()} while the other classes provide
\method{get_function_names()} and \method{get_function_info()}.
580 581 582

Within each of the forms of code block that the public classes
represent, most of the required information is in the same form and is
583
accessed in the same way, with classes having the distinction that
584 585
functions defined at the top level are referred to as ``methods.''
Since the difference in nomenclature reflects a real semantic
586 587
distinction from functions defined outside of a class, the
implementation needs to maintain the distinction.
588
Hence, most of the functionality of the public classes can be
589
implemented in a common base class, \class{SuiteInfoBase}, with the
590 591
accessors for function and method information provided elsewhere.
Note that there is only one class which represents function and method
592
information; this parallels the use of the \keyword{def} statement to
593
define both types of elements.
594

595
Most of the accessor functions are declared in \class{SuiteInfoBase}
596
and do not need to be overridden by subclasses.  More importantly, the
597
extraction of most information from a parse tree is handled through a
598
method called by the \class{SuiteInfoBase} constructor.  The example
599 600 601
code for most of the classes is clear when read alongside the formal
grammar, but the method which recursively creates new information
objects requires further examination.  Here is the relevant part of
602
the \class{SuiteInfoBase} definition from \file{example.py}:
603

604
\begin{verbatim}
605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633
class SuiteInfoBase:
    _docstring = ''
    _name = ''

    def __init__(self, tree = None):
        self._class_info = {}
        self._function_info = {}
        if tree:
            self._extract_info(tree)

    def _extract_info(self, tree):
        # extract docstring
        if len(tree) == 2:
            found, vars = match(DOCSTRING_STMT_PATTERN[1], tree[1])
        else:
            found, vars = match(DOCSTRING_STMT_PATTERN, tree[3])
        if found:
            self._docstring = eval(vars['docstring'])
        # discover inner definitions
        for node in tree[1:]:
            found, vars = match(COMPOUND_STMT_PATTERN, node)
            if found:
                cstmt = vars['compound']
                if cstmt[0] == symbol.funcdef:
                    name = cstmt[2][1]
                    self._function_info[name] = FunctionInfo(cstmt)
                elif cstmt[0] == symbol.classdef:
                    name = cstmt[2][1]
                    self._class_info[name] = ClassInfo(cstmt)
634
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
635

636
After initializing some internal state, the constructor calls the
637
\method{_extract_info()} method.  This method performs the bulk of the
638 639 640 641 642
information extraction which takes place in the entire example.  The
extraction has two distinct phases: the location of the docstring for
the parse tree passed in, and the discovery of additional definitions
within the code block represented by the parse tree.

643
The initial \keyword{if} test determines whether the nested suite is of
644 645 646 647
the ``short form'' or the ``long form.''  The short form is used when
the code block is on the same line as the definition of the code
block, as in

648
\begin{verbatim}
649
def square(x): "Square an argument."; return x ** 2
650
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
651

652 653 654
while the long form uses an indented block and allows nested
definitions:

655
\begin{verbatim}
656 657 658 659 660
def make_power(exp):
    "Make a function that raises an argument to the exponent `exp'."
    def raiser(x, y=exp):
        return x ** y
    return raiser
661
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
662

663
When the short form is used, the code block may contain a docstring as
664
the first, and possibly only, \constant{small_stmt} element.  The
665 666
extraction of such a docstring is slightly different and requires only
a portion of the complete pattern used in the more common case.  As
667
implemented, the docstring will only be found if there is only
668 669 670 671 672 673
one \constant{small_stmt} node in the \constant{simple_stmt} node.
Since most functions and methods which use the short form do not
provide a docstring, this may be considered sufficient.  The
extraction of the docstring proceeds using the \function{match()} function
as described above, and the value of the docstring is stored as an
attribute of the \class{SuiteInfoBase} object.
674

675
After docstring extraction, a simple definition discovery
676 677 678 679 680
algorithm operates on the \constant{stmt} nodes of the
\constant{suite} node.  The special case of the short form is not
tested; since there are no \constant{stmt} nodes in the short form,
the algorithm will silently skip the single \constant{simple_stmt}
node and correctly not discover any nested definitions.
681

682 683
Each statement in the code block is categorized as
a class definition, function or method definition, or
684
something else.  For the definition statements, the name of the
685
element defined is extracted and a representation object
686
appropriate to the definition is created with the defining subtree
687
passed as an argument to the constructor.  The representation objects
688 689 690 691
are stored in instance variables and may be retrieved by name using
the appropriate accessor methods.

The public classes provide any accessors required which are more
692
specific than those provided by the \class{SuiteInfoBase} class, but
693 694
the real extraction algorithm remains common to all forms of code
blocks.  A high-level function can be used to extract the complete set
695
of information from a source file.  (See file \file{example.py}.)
696

697
\begin{verbatim}
698 699 700
def get_docs(fileName):
    import os
    import parser
Fred Drake's avatar
Fred Drake committed
701 702 703

    source = open(fileName).read()
    basename = os.path.basename(os.path.splitext(fileName)[0])
704
    ast = parser.suite(source)
Fred Drake's avatar
Fred Drake committed
705
    return ModuleInfo(ast.totuple(), basename)
706
\end{verbatim}
Fred Drake's avatar
Fred Drake committed
707

708 709 710 711
This provides an easy-to-use interface to the documentation of a
module.  If information is required which is not extracted by the code
of this example, the code may be extended at clearly defined points to
provide additional capabilities.