blob: 55e6c6b3091c98901c8eeac6651459672d816e2c [file] [log] [blame]
Guido van Rossum4b73a061995-10-11 17:30:04 +00001% libparser.tex
2%
Guido van Rossum4b73a061995-10-11 17:30:04 +00003% Copyright 1995 Virginia Polytechnic Institute and State University
4% and Fred L. Drake, Jr. This copyright notice must be distributed on
5% all copies, but this document otherwise may be distributed as part
6% of the Python distribution. No fee may be charged for this document
7% in any representation, either on paper or electronically. This
8% restriction does not affect other elements in a distributed package
9% in any way.
10%
11
12\section{Built-in Module \sectcode{parser}}
Fred Drakebbe60681998-01-09 22:24:14 +000013\label{module-parser}
Guido van Rossum4b73a061995-10-11 17:30:04 +000014\bimodindex{parser}
15
Guido van Rossum4b73a061995-10-11 17:30:04 +000016The \code{parser} module provides an interface to Python's internal
17parser and byte-code compiler. The primary purpose for this interface
18is to allow Python code to edit the parse tree of a Python expression
Fred Drake4b7d5a41996-09-11 21:57:40 +000019and create executable code from this. This is better than trying
20to parse and modify an arbitrary Python code fragment as a string
21because parsing is performed in a manner identical to the code
22forming the application. It is also faster.
Guido van Rossum4b73a061995-10-11 17:30:04 +000023
24There are a few things to note about this module which are important
25to making use of the data structures created. This is not a tutorial
Fred Drake4b7d5a41996-09-11 21:57:40 +000026on editing the parse trees for Python code, but some examples of using
27the \code{parser} module are presented.
Guido van Rossum4b73a061995-10-11 17:30:04 +000028
29Most importantly, a good understanding of the Python grammar processed
30by the internal parser is required. For full information on the
31language syntax, refer to the Language Reference. The parser itself
32is created from a grammar specification defined in the file
Fred Drake4b7d5a41996-09-11 21:57:40 +000033\file{Grammar/Grammar} in the standard Python distribution. The parse
Guido van Rossum4b73a061995-10-11 17:30:04 +000034trees stored in the ``AST objects'' created by this module are the
35actual output from the internal parser when created by the
36\code{expr()} or \code{suite()} functions, described below. The AST
Guido van Rossum47478871996-08-21 14:32:37 +000037objects created by \code{sequence2ast()} faithfully simulate those
38structures. Be aware that the values of the sequences which are
39considered ``correct'' will vary from one version of Python to another
40as the formal grammar for the language is revised. However,
41transporting code from one Python version to another as source text
42will always allow correct parse trees to be created in the target
43version, with the only restriction being that migrating to an older
44version of the interpreter will not support more recent language
45constructs. The parse trees are not typically compatible from one
46version to another, whereas source code has always been
47forward-compatible.
Guido van Rossum4b73a061995-10-11 17:30:04 +000048
Guido van Rossum47478871996-08-21 14:32:37 +000049Each element of the sequences returned by \code{ast2list} or
50\code{ast2tuple()} has a simple form. Sequences representing
51non-terminal elements in the grammar always have a length greater than
52one. The first element is an integer which identifies a production in
53the grammar. These integers are given symbolic names in the C header
Fred Drake4b7d5a41996-09-11 21:57:40 +000054file \file{Include/graminit.h} and the Python module
Fred Drakee061a511997-10-06 21:40:20 +000055\code{symbol}. Each additional element of the sequence represents
Guido van Rossum47478871996-08-21 14:32:37 +000056a component of the production as recognized in the input string: these
57are always sequences which have the same form as the parent. An
58important aspect of this structure which should be noted is that
59keywords used to identify the parent node type, such as the keyword
Fred Drake4b7d5a41996-09-11 21:57:40 +000060\code{if} in an \code{if_stmt}, are included in the node tree without
Guido van Rossum47478871996-08-21 14:32:37 +000061any special treatment. For example, the \code{if} keyword is
Guido van Rossum4b73a061995-10-11 17:30:04 +000062represented by the tuple \code{(1, 'if')}, where \code{1} is the
Fred Drake4b7d5a41996-09-11 21:57:40 +000063numeric value associated with all \code{NAME} tokens, including
Guido van Rossum47478871996-08-21 14:32:37 +000064variable and function names defined by the user. In an alternate form
65returned when line number information is requested, the same token
66might be represented as \code{(1, 'if', 12)}, where the \code{12}
67represents the line number at which the terminal symbol was found.
Guido van Rossum4b73a061995-10-11 17:30:04 +000068
69Terminal elements are represented in much the same way, but without
70any child elements and the addition of the source text which was
71identified. The example of the \code{if} keyword above is
72representative. The various types of terminal symbols are defined in
Fred Drake4b7d5a41996-09-11 21:57:40 +000073the C header file \file{Include/token.h} and the Python module
Fred Drakee061a511997-10-06 21:40:20 +000074\code{token}.
Guido van Rossum4b73a061995-10-11 17:30:04 +000075
Fred Drake4b7d5a41996-09-11 21:57:40 +000076The AST objects are not required to support the functionality of this
77module, but are provided for three purposes: to allow an application
78to amortize the cost of processing complex parse trees, to provide a
79parse tree representation which conserves memory space when compared
80to the Python list or tuple representation, and to ease the creation
81of additional modules in C which manipulate parse trees. A simple
82``wrapper'' class may be created in Python to hide the use of AST
83objects; the \code{AST} library module provides a variety of such
84classes.
Guido van Rossum4b73a061995-10-11 17:30:04 +000085
Fred Drake4b7d5a41996-09-11 21:57:40 +000086The \code{parser} module defines functions for a few distinct
87purposes. The most important purposes are to create AST objects and
88to convert AST objects to other representations such as parse trees
89and compiled code objects, but there are also functions which serve to
90query the type of parse tree represented by an AST object.
Guido van Rossum4b73a061995-10-11 17:30:04 +000091
Guido van Rossum4b73a061995-10-11 17:30:04 +000092\renewcommand{\indexsubitem}{(in module parser)}
93
Fred Drake4b7d5a41996-09-11 21:57:40 +000094
95\subsection{Creating AST Objects}
96
97AST objects may be created from source code or from a parse tree.
98When creating an AST object from source, different functions are used
99to create the \code{'eval'} and \code{'exec'} forms.
100
101\begin{funcdesc}{expr}{string}
102The \code{expr()} function parses the parameter \code{\var{string}}
103as if it were an input to \code{compile(\var{string}, 'eval')}. If
104the parse succeeds, an AST object is created to hold the internal
105parse tree representation, otherwise an appropriate exception is
106thrown.
107\end{funcdesc}
108
109\begin{funcdesc}{suite}{string}
110The \code{suite()} function parses the parameter \code{\var{string}}
111as if it were an input to \code{compile(\var{string}, 'exec')}. If
112the parse succeeds, an AST object is created to hold the internal
113parse tree representation, otherwise an appropriate exception is
114thrown.
115\end{funcdesc}
116
117\begin{funcdesc}{sequence2ast}{sequence}
118This function accepts a parse tree represented as a sequence and
119builds an internal representation if possible. If it can validate
120that the tree conforms to the Python grammar and all nodes are valid
121node types in the host version of Python, an AST object is created
122from the internal representation and returned to the called. If there
123is a problem creating the internal representation, or if the tree
124cannot be validated, a \code{ParserError} exception is thrown. An AST
125object created this way should not be assumed to compile correctly;
126normal exceptions thrown by compilation may still be initiated when
127the AST object is passed to \code{compileast()}. This may indicate
128problems not related to syntax (such as a \code{MemoryError}
129exception), but may also be due to constructs such as the result of
130parsing \code{del f(0)}, which escapes the Python parser but is
131checked by the bytecode compiler.
132
133Sequences representing terminal tokens may be represented as either
134two-element lists of the form \code{(1, 'name')} or as three-element
135lists of the form \code{(1, 'name', 56)}. If the third element is
136present, it is assumed to be a valid line number. The line number
137may be specified for any subset of the terminal symbols in the input
138tree.
139\end{funcdesc}
140
141\begin{funcdesc}{tuple2ast}{sequence}
142This is the same function as \code{sequence2ast()}. This entry point
143is maintained for backward compatibility.
144\end{funcdesc}
145
146
147\subsection{Converting AST Objects}
148
149AST objects, regardless of the input used to create them, may be
150converted to parse trees represented as list- or tuple- trees, or may
151be compiled into executable code objects. Parse trees may be
152extracted with or without line numbering information.
153
154\begin{funcdesc}{ast2list}{ast\optional{\, line_info\code{ = 0}}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000155This function accepts an AST object from the caller in
Guido van Rossum47478871996-08-21 14:32:37 +0000156\code{\var{ast}} and returns a Python list representing the
157equivelent parse tree. The resulting list representation can be used
Fred Drake4b7d5a41996-09-11 21:57:40 +0000158for inspection or the creation of a new parse tree in list form. This
159function does not fail so long as memory is available to build the
160list representation. If the parse tree will only be used for
Guido van Rossum47478871996-08-21 14:32:37 +0000161inspection, \code{ast2tuple()} should be used instead to reduce memory
Fred Drake4b7d5a41996-09-11 21:57:40 +0000162consumption and fragmentation. When the list representation is
163required, this function is significantly faster than retrieving a
164tuple representation and converting that to nested lists.
Guido van Rossum47478871996-08-21 14:32:37 +0000165
Fred Drake4b7d5a41996-09-11 21:57:40 +0000166If \code{\var{line_info}} is true, line number information will be
167included for all terminal tokens as a third element of the list
Fred Drake9abe64a1996-12-05 22:28:43 +0000168representing the token. Note that the line number provided specifies
169the line on which the token \emph{ends\/}. This information is
170omitted if the flag is false or omitted.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000171\end{funcdesc}
172
Fred Drake4b7d5a41996-09-11 21:57:40 +0000173\begin{funcdesc}{ast2tuple}{ast\optional{\, line_info\code{ = 0}}}
Guido van Rossum47478871996-08-21 14:32:37 +0000174This function accepts an AST object from the caller in
175\code{\var{ast}} and returns a Python tuple representing the
176equivelent parse tree. Other than returning a tuple instead of a
177list, this function is identical to \code{ast2list()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000178
Fred Drake4b7d5a41996-09-11 21:57:40 +0000179If \code{\var{line_info}} is true, line number information will be
180included for all terminal tokens as a third element of the list
181representing the token. This information is omitted if the flag is
182false or omitted.
Guido van Rossum47478871996-08-21 14:32:37 +0000183\end{funcdesc}
184
185\begin{funcdesc}{compileast}{ast\optional{\, filename\code{ = '<ast>'}}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000186The Python byte compiler can be invoked on an AST object to produce
187code objects which can be used as part of an \code{exec} statement or
188a call to the built-in \code{eval()} function. This function provides
189the interface to the compiler, passing the internal parse tree from
190\code{\var{ast}} to the parser, using the source file name specified
191by the \code{\var{filename}} parameter. The default value supplied
192for \code{\var{filename}} indicates that the source was an AST object.
Guido van Rossum47478871996-08-21 14:32:37 +0000193
194Compiling an AST object may result in exceptions related to
195compilation; an example would be a \code{SyntaxError} caused by the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000196parse tree for \code{del f(0)}: this statement is considered legal
Guido van Rossum47478871996-08-21 14:32:37 +0000197within the formal grammar for Python but is not a legal language
198construct. The \code{SyntaxError} raised for this condition is
199actually generated by the Python byte-compiler normally, which is why
200it can be raised at this point by the \code{parser} module. Most
201causes of compilation failure can be diagnosed programmatically by
202inspection of the parse tree.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000203\end{funcdesc}
204
205
Fred Drake4b7d5a41996-09-11 21:57:40 +0000206\subsection{Queries on AST Objects}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000207
Fred Drake4b7d5a41996-09-11 21:57:40 +0000208Two functions are provided which allow an application to determine if
209an AST was create as an expression or a suite. Neither of these
210functions can be used to determine if an AST was created from source
211code via \code{expr()} or \code{suite()} or from a parse tree via
212\code{sequence2ast()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000213
214\begin{funcdesc}{isexpr}{ast}
215When \code{\var{ast}} represents an \code{'eval'} form, this function
216returns a true value (\code{1}), otherwise it returns false
217(\code{0}). This is useful, since code objects normally cannot be
218queried for this information using existing built-in functions. Note
219that the code objects created by \code{compileast()} cannot be queried
220like this either, and are identical to those created by the built-in
221\code{compile()} function.
222\end{funcdesc}
223
224
225\begin{funcdesc}{issuite}{ast}
226This function mirrors \code{isexpr()} in that it reports whether an
Fred Drake4b7d5a41996-09-11 21:57:40 +0000227AST object represents an \code{'exec'} form, commonly known as a
228``suite.'' It is not safe to assume that this function is equivelent
229to \code{not isexpr(\var{ast})}, as additional syntactic fragments may
230be supported in the future.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000231\end{funcdesc}
232
233
Guido van Rossum4b73a061995-10-11 17:30:04 +0000234\subsection{Exceptions and Error Handling}
235
236The parser module defines a single exception, but may also pass other
237built-in exceptions from other portions of the Python runtime
238environment. See each function for information about the exceptions
239it can raise.
240
241\begin{excdesc}{ParserError}
242Exception raised when a failure occurs within the parser module. This
243is generally produced for validation failures rather than the built in
244\code{SyntaxError} thrown during normal parsing.
245The exception argument is either a string describing the reason of the
Guido van Rossum47478871996-08-21 14:32:37 +0000246failure or a tuple containing a sequence causing the failure from a parse
247tree passed to \code{sequence2ast()} and an explanatory string. Calls to
248\code{sequence2ast()} need to be able to handle either type of exception,
Guido van Rossum4b73a061995-10-11 17:30:04 +0000249while calls to other functions in the module will only need to be
250aware of the simple string values.
251\end{excdesc}
252
253Note that the functions \code{compileast()}, \code{expr()}, and
254\code{suite()} may throw exceptions which are normally thrown by the
255parsing and compilation process. These include the built in
256exceptions \code{MemoryError}, \code{OverflowError},
257\code{SyntaxError}, and \code{SystemError}. In these cases, these
258exceptions carry all the meaning normally associated with them. Refer
259to the descriptions of each function for detailed information.
260
Guido van Rossum4b73a061995-10-11 17:30:04 +0000261
Guido van Rossum47478871996-08-21 14:32:37 +0000262\subsection{AST Objects}
263
Fred Drake4b7d5a41996-09-11 21:57:40 +0000264AST objects returned by \code{expr()}, \code{suite()}, and
265\code{sequence2ast()} have no methods of their own.
Guido van Rossum47478871996-08-21 14:32:37 +0000266Some of the functions defined which accept an AST object as their
Fred Drake4b7d5a41996-09-11 21:57:40 +0000267first argument may change to object methods in the future. The type
268of these objects is available as \code{ASTType} in the module.
Guido van Rossum47478871996-08-21 14:32:37 +0000269
270Ordered and equality comparisons are supported between AST objects.
271
272
Guido van Rossum8206fb91996-08-26 00:33:29 +0000273\subsection{Examples}
Fred Drake4b3f0311996-12-13 22:04:31 +0000274\nodename{AST Examples}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000275
Guido van Rossum47478871996-08-21 14:32:37 +0000276The parser modules allows operations to be performed on the parse tree
277of Python source code before the bytecode is generated, and provides
Fred Drake4b7d5a41996-09-11 21:57:40 +0000278for inspection of the parse tree for information gathering purposes.
279Two examples are presented. The simple example demonstrates emulation
280of the \code{compile()} built-in function and the complex example
281shows the use of a parse tree for information discovery.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000282
Fred Drake4b7d5a41996-09-11 21:57:40 +0000283\subsubsection{Emulation of \sectcode{compile()}}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000284
285While many useful operations may take place between parsing and
Guido van Rossum47478871996-08-21 14:32:37 +0000286bytecode generation, the simplest operation is to do nothing. For
287this purpose, using the \code{parser} module to produce an
288intermediate data structure is equivelent to the code
289
Guido van Rossume47da0a1997-07-17 16:34:52 +0000290\bcode\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000291>>> code = compile('a + 5', 'eval')
292>>> a = 5
293>>> eval(code)
29410
Guido van Rossume47da0a1997-07-17 16:34:52 +0000295\end{verbatim}\ecode
296%
Guido van Rossum47478871996-08-21 14:32:37 +0000297The equivelent operation using the \code{parser} module is somewhat
298longer, and allows the intermediate internal parse tree to be retained
299as an AST object:
Guido van Rossum4b73a061995-10-11 17:30:04 +0000300
Guido van Rossume47da0a1997-07-17 16:34:52 +0000301\bcode\begin{verbatim}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000302>>> import parser
303>>> ast = parser.expr('a + 5')
304>>> code = parser.compileast(ast)
305>>> a = 5
306>>> eval(code)
30710
Guido van Rossume47da0a1997-07-17 16:34:52 +0000308\end{verbatim}\ecode
309%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000310An application which needs both AST and code objects can package this
311code into readily available functions:
312
Guido van Rossume47da0a1997-07-17 16:34:52 +0000313\bcode\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000314import parser
315
316def load_suite(source_string):
317 ast = parser.suite(source_string)
318 code = parser.compileast(ast)
319 return ast, code
320
321def load_expression(source_string):
322 ast = parser.expr(source_string)
323 code = parser.compileast(ast)
324 return ast, code
Guido van Rossume47da0a1997-07-17 16:34:52 +0000325\end{verbatim}\ecode
326%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000327\subsubsection{Information Discovery}
328
Fred Drake4b7d5a41996-09-11 21:57:40 +0000329Some applications benefit from direct access to the parse tree. The
330remainder of this section demonstrates how the parse tree provides
331access to module documentation defined in docstrings without requiring
332that the code being examined be loaded into a running interpreter via
333\code{import}. This can be very useful for performing analyses of
334untrusted code.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000335
Guido van Rossum47478871996-08-21 14:32:37 +0000336Generally, the example will demonstrate how the parse tree may be
337traversed to distill interesting information. Two functions and a set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000338of classes are developed which provide programmatic access to high
Guido van Rossum47478871996-08-21 14:32:37 +0000339level function and class definitions provided by a module. The
340classes extract information from the parse tree and provide access to
341the information at a useful semantic level, one function provides a
342simple low-level pattern matching capability, and the other function
343defines a high-level interface to the classes by handling file
344operations on behalf of the caller. All source files mentioned here
345which are not part of the Python installation are located in the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000346\file{Demo/parser/} directory of the distribution.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000347
Guido van Rossum8206fb91996-08-26 00:33:29 +0000348The dynamic nature of Python allows the programmer a great deal of
349flexibility, but most modules need only a limited measure of this when
350defining classes, functions, and methods. In this example, the only
351definitions that will be considered are those which are defined in the
352top level of their context, e.g., a function defined by a \code{def}
353statement at column zero of a module, but not a function defined
Fred Drake4b7d5a41996-09-11 21:57:40 +0000354within a branch of an \code{if} ... \code{else} construct, though
Guido van Rossum8206fb91996-08-26 00:33:29 +0000355there are some good reasons for doing so in some situations. Nesting
356of definitions will be handled by the code developed in the example.
357
Guido van Rossum47478871996-08-21 14:32:37 +0000358To construct the upper-level extraction methods, we need to know what
359the parse tree structure looks like and how much of it we actually
Fred Drake4b7d5a41996-09-11 21:57:40 +0000360need to be concerned about. Python uses a moderately deep parse tree
Guido van Rossum47478871996-08-21 14:32:37 +0000361so there are a large number of intermediate nodes. It is important to
362read and understand the formal grammar used by Python. This is
363specified in the file \file{Grammar/Grammar} in the distribution.
364Consider the simplest case of interest when searching for docstrings:
Guido van Rossum8206fb91996-08-26 00:33:29 +0000365a module consisting of a docstring and nothing else. (See file
366\file{docstring.py}.)
Guido van Rossum4b73a061995-10-11 17:30:04 +0000367
Guido van Rossume47da0a1997-07-17 16:34:52 +0000368\bcode\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000369"""Some documentation.
370"""
Guido van Rossume47da0a1997-07-17 16:34:52 +0000371\end{verbatim}\ecode
372%
Guido van Rossum47478871996-08-21 14:32:37 +0000373Using the interpreter to take a look at the parse tree, we find a
374bewildering mass of numbers and parentheses, with the documentation
Fred Drake4b7d5a41996-09-11 21:57:40 +0000375buried deep in nested tuples.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000376
Guido van Rossume47da0a1997-07-17 16:34:52 +0000377\bcode\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000378>>> import parser
379>>> import pprint
380>>> ast = parser.suite(open('docstring.py').read())
381>>> tup = parser.ast2tuple(ast)
382>>> pprint.pprint(tup)
383(257,
384 (264,
385 (265,
386 (266,
387 (267,
388 (307,
389 (287,
390 (288,
391 (289,
392 (290,
393 (292,
394 (293,
395 (294,
396 (295,
397 (296,
398 (297,
399 (298,
400 (299,
401 (300, (3, '"""Some documentation.\012"""'))))))))))))))))),
402 (4, ''))),
403 (4, ''),
404 (0, ''))
Guido van Rossume47da0a1997-07-17 16:34:52 +0000405\end{verbatim}\ecode
406%
Guido van Rossum47478871996-08-21 14:32:37 +0000407The numbers at the first element of each node in the tree are the node
408types; they map directly to terminal and non-terminal symbols in the
409grammar. Unfortunately, they are represented as integers in the
410internal representation, and the Python structures generated do not
411change that. However, the \code{symbol} and \code{token} modules
412provide symbolic names for the node types and dictionaries which map
413from the integers to the symbolic names for the node types.
414
415In the output presented above, the outermost tuple contains four
416elements: the integer \code{257} and three additional tuples. Node
417type \code{257} has the symbolic name \code{file_input}. Each of
418these inner tuples contains an integer as the first element; these
419integers, \code{264}, \code{4}, and \code{0}, represent the node types
420\code{stmt}, \code{NEWLINE}, and \code{ENDMARKER}, respectively.
421Note that these values may change depending on the version of Python
422you are using; consult \file{symbol.py} and \file{token.py} for
423details of the mapping. It should be fairly clear that the outermost
424node is related primarily to the input source rather than the contents
425of the file, and may be disregarded for the moment. The \code{stmt}
426node is much more interesting. In particular, all docstrings are
427found in subtrees which are formed exactly as this node is formed,
428with the only difference being the string itself. The association
429between the docstring in a similar tree and the defined entity (class,
430function, or module) which it describes is given by the position of
431the docstring subtree within the tree defining the described
432structure.
433
434By replacing the actual docstring with something to signify a variable
Fred Drake4b7d5a41996-09-11 21:57:40 +0000435component of the tree, we allow a simple pattern matching approach to
436check any given subtree for equivelence to the general pattern for
437docstrings. Since the example demonstrates information extraction, we
438can safely require that the tree be in tuple form rather than list
439form, allowing a simple variable representation to be
440\code{['variable_name']}. A simple recursive function can implement
Guido van Rossum47478871996-08-21 14:32:37 +0000441the pattern matching, returning a boolean and a dictionary of variable
Guido van Rossum8206fb91996-08-26 00:33:29 +0000442name to value mappings. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000443
Guido van Rossume47da0a1997-07-17 16:34:52 +0000444\bcode\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000445from types import ListType, TupleType
446
447def match(pattern, data, vars=None):
448 if vars is None:
449 vars = {}
450 if type(pattern) is ListType:
451 vars[pattern[0]] = data
452 return 1, vars
453 if type(pattern) is not TupleType:
454 return (pattern == data), vars
455 if len(data) != len(pattern):
456 return 0, vars
457 for pattern, data in map(None, pattern, data):
458 same, vars = match(pattern, data, vars)
459 if not same:
460 break
461 return same, vars
Guido van Rossume47da0a1997-07-17 16:34:52 +0000462\end{verbatim}\ecode
463%
Fred Drake4b7d5a41996-09-11 21:57:40 +0000464Using this simple representation for syntactic variables and the symbolic
Guido van Rossum8206fb91996-08-26 00:33:29 +0000465node types, the pattern for the candidate docstring subtrees becomes
466fairly readable. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000467
Guido van Rossume47da0a1997-07-17 16:34:52 +0000468\bcode\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000469import symbol
470import token
471
472DOCSTRING_STMT_PATTERN = (
473 symbol.stmt,
474 (symbol.simple_stmt,
475 (symbol.small_stmt,
476 (symbol.expr_stmt,
477 (symbol.testlist,
478 (symbol.test,
479 (symbol.and_test,
480 (symbol.not_test,
481 (symbol.comparison,
482 (symbol.expr,
483 (symbol.xor_expr,
484 (symbol.and_expr,
485 (symbol.shift_expr,
486 (symbol.arith_expr,
487 (symbol.term,
488 (symbol.factor,
489 (symbol.power,
490 (symbol.atom,
491 (token.STRING, ['docstring'])
492 )))))))))))))))),
493 (token.NEWLINE, '')
494 ))
Guido van Rossume47da0a1997-07-17 16:34:52 +0000495\end{verbatim}\ecode
496%
Guido van Rossum47478871996-08-21 14:32:37 +0000497Using the \code{match()} function with this pattern, extracting the
498module docstring from the parse tree created previously is easy:
499
Guido van Rossume47da0a1997-07-17 16:34:52 +0000500\bcode\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000501>>> found, vars = match(DOCSTRING_STMT_PATTERN, tup[1])
502>>> found
5031
504>>> vars
505{'docstring': '"""Some documentation.\012"""'}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000506\end{verbatim}\ecode
507%
Guido van Rossum47478871996-08-21 14:32:37 +0000508Once specific data can be extracted from a location where it is
509expected, the question of where information can be expected
510needs to be answered. When dealing with docstrings, the answer is
511fairly simple: the docstring is the first \code{stmt} node in a code
512block (\code{file_input} or \code{suite} node types). A module
513consists of a single \code{file_input} node, and class and function
514definitions each contain exactly one \code{suite} node. Classes and
515functions are readily identified as subtrees of code block nodes which
516start with \code{(stmt, (compound_stmt, (classdef, ...} or
517\code{(stmt, (compound_stmt, (funcdef, ...}. Note that these subtrees
518cannot be matched by \code{match()} since it does not support multiple
519sibling nodes to match without regard to number. A more elaborate
520matching function could be used to overcome this limitation, but this
521is sufficient for the example.
522
Guido van Rossum8206fb91996-08-26 00:33:29 +0000523Given the ability to determine whether a statement might be a
524docstring and extract the actual string from the statement, some work
525needs to be performed to walk the parse tree for an entire module and
526extract information about the names defined in each context of the
527module and associate any docstrings with the names. The code to
528perform this work is not complicated, but bears some explanation.
529
530The public interface to the classes is straightforward and should
531probably be somewhat more flexible. Each ``major'' block of the
532module is described by an object providing several methods for inquiry
533and a constructor which accepts at least the subtree of the complete
534parse tree which it represents. The \code{ModuleInfo} constructor
535accepts an optional \code{\var{name}} parameter since it cannot
536otherwise determine the name of the module.
537
538The public classes include \code{ClassInfo}, \code{FunctionInfo},
539and \code{ModuleInfo}. All objects provide the
540methods \code{get_name()}, \code{get_docstring()},
541\code{get_class_names()}, and \code{get_class_info()}. The
542\code{ClassInfo} objects support \code{get_method_names()} and
543\code{get_method_info()} while the other classes provide
544\code{get_function_names()} and \code{get_function_info()}.
545
546Within each of the forms of code block that the public classes
547represent, most of the required information is in the same form and is
Fred Drake4b7d5a41996-09-11 21:57:40 +0000548accessed in the same way, with classes having the distinction that
Guido van Rossum8206fb91996-08-26 00:33:29 +0000549functions defined at the top level are referred to as ``methods.''
550Since the difference in nomenclature reflects a real semantic
Fred Drake4b7d5a41996-09-11 21:57:40 +0000551distinction from functions defined outside of a class, the
552implementation needs to maintain the distinction.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000553Hence, most of the functionality of the public classes can be
554implemented in a common base class, \code{SuiteInfoBase}, with the
555accessors for function and method information provided elsewhere.
556Note that there is only one class which represents function and method
Fred Drake43d287a1997-01-22 14:25:21 +0000557information; this parallels the use of the \code{def} statement to
Fred Drake4b7d5a41996-09-11 21:57:40 +0000558define both types of elements.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000559
560Most of the accessor functions are declared in \code{SuiteInfoBase}
561and do not need to be overriden by subclasses. More importantly, the
562extraction of most information from a parse tree is handled through a
563method called by the \code{SuiteInfoBase} constructor. The example
564code for most of the classes is clear when read alongside the formal
565grammar, but the method which recursively creates new information
566objects requires further examination. Here is the relevant part of
567the \code{SuiteInfoBase} definition from \file{example.py}:
568
Guido van Rossume47da0a1997-07-17 16:34:52 +0000569\bcode\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000570class SuiteInfoBase:
571 _docstring = ''
572 _name = ''
573
574 def __init__(self, tree = None):
575 self._class_info = {}
576 self._function_info = {}
577 if tree:
578 self._extract_info(tree)
579
580 def _extract_info(self, tree):
581 # extract docstring
582 if len(tree) == 2:
583 found, vars = match(DOCSTRING_STMT_PATTERN[1], tree[1])
584 else:
585 found, vars = match(DOCSTRING_STMT_PATTERN, tree[3])
586 if found:
587 self._docstring = eval(vars['docstring'])
588 # discover inner definitions
589 for node in tree[1:]:
590 found, vars = match(COMPOUND_STMT_PATTERN, node)
591 if found:
592 cstmt = vars['compound']
593 if cstmt[0] == symbol.funcdef:
594 name = cstmt[2][1]
595 self._function_info[name] = FunctionInfo(cstmt)
596 elif cstmt[0] == symbol.classdef:
597 name = cstmt[2][1]
598 self._class_info[name] = ClassInfo(cstmt)
Guido van Rossume47da0a1997-07-17 16:34:52 +0000599\end{verbatim}\ecode
600%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000601After initializing some internal state, the constructor calls the
602\code{_extract_info()} method. This method performs the bulk of the
603information extraction which takes place in the entire example. The
604extraction has two distinct phases: the location of the docstring for
605the parse tree passed in, and the discovery of additional definitions
606within the code block represented by the parse tree.
607
608The initial \code{if} test determines whether the nested suite is of
609the ``short form'' or the ``long form.'' The short form is used when
610the code block is on the same line as the definition of the code
611block, as in
612
Fred Drakebbe60681998-01-09 22:24:14 +0000613\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000614def square(x): "Square an argument."; return x ** 2
Fred Drakebbe60681998-01-09 22:24:14 +0000615\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000616%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000617while the long form uses an indented block and allows nested
618definitions:
619
Guido van Rossume47da0a1997-07-17 16:34:52 +0000620\bcode\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000621def make_power(exp):
622 "Make a function that raises an argument to the exponent `exp'."
623 def raiser(x, y=exp):
624 return x ** y
625 return raiser
Guido van Rossume47da0a1997-07-17 16:34:52 +0000626\end{verbatim}\ecode
627%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000628When the short form is used, the code block may contain a docstring as
629the first, and possibly only, \code{small_stmt} element. The
630extraction of such a docstring is slightly different and requires only
631a portion of the complete pattern used in the more common case. As
Fred Drake4b7d5a41996-09-11 21:57:40 +0000632implemented, the docstring will only be found if there is only
Guido van Rossum8206fb91996-08-26 00:33:29 +0000633one \code{small_stmt} node in the \code{simple_stmt} node. Since most
Fred Drake4b7d5a41996-09-11 21:57:40 +0000634functions and methods which use the short form do not provide a
Guido van Rossum8206fb91996-08-26 00:33:29 +0000635docstring, this may be considered sufficient. The extraction of the
636docstring proceeds using the \code{match()} function as described
637above, and the value of the docstring is stored as an attribute of the
638\code{SuiteInfoBase} object.
639
Fred Drake4b7d5a41996-09-11 21:57:40 +0000640After docstring extraction, a simple definition discovery
641algorithm operates on the \code{stmt} nodes of the \code{suite} node. The
Guido van Rossum8206fb91996-08-26 00:33:29 +0000642special case of the short form is not tested; since there are no
643\code{stmt} nodes in the short form, the algorithm will silently skip
644the single \code{simple_stmt} node and correctly not discover any
645nested definitions.
646
Fred Drake4b7d5a41996-09-11 21:57:40 +0000647Each statement in the code block is categorized as
648a class definition, function or method definition, or
Guido van Rossum8206fb91996-08-26 00:33:29 +0000649something else. For the definition statements, the name of the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000650element defined is extracted and a representation object
Guido van Rossum8206fb91996-08-26 00:33:29 +0000651appropriate to the definition is created with the defining subtree
652passed as an argument to the constructor. The repesentation objects
653are stored in instance variables and may be retrieved by name using
654the appropriate accessor methods.
655
656The public classes provide any accessors required which are more
657specific than those provided by the \code{SuiteInfoBase} class, but
658the real extraction algorithm remains common to all forms of code
659blocks. A high-level function can be used to extract the complete set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000660of information from a source file. (See file \file{example.py}.)
Guido van Rossum8206fb91996-08-26 00:33:29 +0000661
Guido van Rossume47da0a1997-07-17 16:34:52 +0000662\bcode\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000663def get_docs(fileName):
664 source = open(fileName).read()
665 import os
666 basename = os.path.basename(os.path.splitext(fileName)[0])
667 import parser
668 ast = parser.suite(source)
669 tup = parser.ast2tuple(ast)
670 return ModuleInfo(tup, basename)
Guido van Rossume47da0a1997-07-17 16:34:52 +0000671\end{verbatim}\ecode
672%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000673This provides an easy-to-use interface to the documentation of a
674module. If information is required which is not extracted by the code
675of this example, the code may be extended at clearly defined points to
676provide additional capabilities.
Guido van Rossum47478871996-08-21 14:32:37 +0000677
Fred Drakebbe60681998-01-09 22:24:14 +0000678\begin{seealso}
679
680\seemodule{symbol}%
681 {useful constants representing internal nodes of the parse tree}
682
683\seemodule{token}%
684 {useful constants representing leaf nodes of the parse tree and
685functions for testing node values}
686
687\end{seealso}
688
Guido van Rossum47478871996-08-21 14:32:37 +0000689
Fred Drakee061a511997-10-06 21:40:20 +0000690\section{Standard Module \sectcode{symbol}}
Fred Drakebbe60681998-01-09 22:24:14 +0000691\label{module-symbol}
Fred Drakee061a511997-10-06 21:40:20 +0000692\stmodindex{symbol}
693
694This module provides constants which represent the numeric values of
695internal nodes of the parse tree. Unlike most Python constants, these
696use lower-case names. Refer to the file \file{Grammar/Grammar} in the
697Python distribution for the defintions of the names in the context of
698the language grammar. The specific numeric values which the names map
699to may change between Python versions.
700
701This module also provides one additional data object:
702
Fred Drakee624e0f1997-11-25 04:04:00 +0000703\renewcommand{\indexsubitem}{(in module symbol)}
704
705
Fred Drakee061a511997-10-06 21:40:20 +0000706\begin{datadesc}{sym_name}
707Dictionary mapping the numeric values of the constants defined in this
708module back to name strings, allowing more human-readable
709representation of parse trees to be generated.
710\end{datadesc}
711
Fred Drakebbe60681998-01-09 22:24:14 +0000712\begin{seealso}
713\seemodule{parser}{second example uses this module}
714\end{seealso}
715
Fred Drakee061a511997-10-06 21:40:20 +0000716
717\section{Standard Module \sectcode{token}}
Fred Drakebbe60681998-01-09 22:24:14 +0000718\label{module-token}
Fred Drakee061a511997-10-06 21:40:20 +0000719\stmodindex{token}
720
721This module provides constants which represent the numeric values of
722leaf nodes of the parse tree (terminal tokens). Refer to the file
723\file{Grammar/Grammar} in the Python distribution for the defintions
724of the names in the context of the language grammar. The specific
725numeric values which the names map to may change between Python
726versions.
727
728This module also provides one data object and some functions. The
729functions mirror definitions in the Python C header files.
730
Fred Drakee624e0f1997-11-25 04:04:00 +0000731\renewcommand{\indexsubitem}{(in module token)}
732
733
Fred Drakee061a511997-10-06 21:40:20 +0000734\begin{datadesc}{tok_name}
735Dictionary mapping the numeric values of the constants defined in this
736module back to name strings, allowing more human-readable
737representation of parse trees to be generated.
738\end{datadesc}
739
740\begin{funcdesc}{ISTERMINAL}{x}
741Return true for terminal token values.
742\end{funcdesc}
743
744\begin{funcdesc}{ISNONTERMINAL}{x}
745Return true for non-terminal token values.
746\end{funcdesc}
747
748\begin{funcdesc}{ISEOF}{x}
749Return true if \var{x} is the marker indicating the end of input.
750\end{funcdesc}
751
Fred Drakebbe60681998-01-09 22:24:14 +0000752\begin{seealso}
753\seemodule{parser}{second example uses this module}
754\end{seealso}