blob: cbea16d3a69e61fb27b2c5c7adf92051e58bc537 [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
Fred Drake3a0351c1998-04-04 07:23:21 +000012\section{Built-in Module \module{parser}}
Fred Drakeb91e9341998-07-23 17:59:49 +000013\declaremodule{builtin}{parser}
14
15\modulesynopsis{Retrieve and submit parse trees from and to the runtime support
16environment.}
17
Fred Drake88223901998-02-09 20:52:48 +000018\index{parsing!Python source code}
Guido van Rossum4b73a061995-10-11 17:30:04 +000019
Fred Drake88223901998-02-09 20:52:48 +000020The \module{parser} module provides an interface to Python's internal
Guido van Rossum4b73a061995-10-11 17:30:04 +000021parser and byte-code compiler. The primary purpose for this interface
22is to allow Python code to edit the parse tree of a Python expression
Fred Drake4b7d5a41996-09-11 21:57:40 +000023and create executable code from this. This is better than trying
24to parse and modify an arbitrary Python code fragment as a string
25because parsing is performed in a manner identical to the code
26forming the application. It is also faster.
Guido van Rossum4b73a061995-10-11 17:30:04 +000027
Fred Drake5bd7fcc1998-04-03 05:31:45 +000028The \module{parser} module was written and documented by Fred
29L. Drake, Jr. (\email{fdrake@acm.org}).%
30\index{Drake, Fred L., Jr.}
31
Guido van Rossum4b73a061995-10-11 17:30:04 +000032There are a few things to note about this module which are important
33to making use of the data structures created. This is not a tutorial
Fred Drake4b7d5a41996-09-11 21:57:40 +000034on editing the parse trees for Python code, but some examples of using
Fred Drake88223901998-02-09 20:52:48 +000035the \module{parser} module are presented.
Guido van Rossum4b73a061995-10-11 17:30:04 +000036
37Most importantly, a good understanding of the Python grammar processed
38by the internal parser is required. For full information on the
Fred Drake88223901998-02-09 20:52:48 +000039language syntax, refer to the \emph{Python Language Reference}. The
40parser itself is created from a grammar specification defined in the file
Fred Drake4b7d5a41996-09-11 21:57:40 +000041\file{Grammar/Grammar} in the standard Python distribution. The parse
Fred Drakecc444e31998-03-08 06:47:24 +000042trees stored in the AST objects created by this module are the
Guido van Rossum4b73a061995-10-11 17:30:04 +000043actual output from the internal parser when created by the
Fred Drake88223901998-02-09 20:52:48 +000044\function{expr()} or \function{suite()} functions, described below. The AST
45objects created by \function{sequence2ast()} faithfully simulate those
Guido van Rossum47478871996-08-21 14:32:37 +000046structures. Be aware that the values of the sequences which are
47considered ``correct'' will vary from one version of Python to another
48as the formal grammar for the language is revised. However,
49transporting code from one Python version to another as source text
50will always allow correct parse trees to be created in the target
51version, with the only restriction being that migrating to an older
52version of the interpreter will not support more recent language
53constructs. The parse trees are not typically compatible from one
54version to another, whereas source code has always been
55forward-compatible.
Guido van Rossum4b73a061995-10-11 17:30:04 +000056
Fred Drake88223901998-02-09 20:52:48 +000057Each element of the sequences returned by \function{ast2list()} or
58\function{ast2tuple()} has a simple form. Sequences representing
Guido van Rossum47478871996-08-21 14:32:37 +000059non-terminal elements in the grammar always have a length greater than
60one. The first element is an integer which identifies a production in
61the grammar. These integers are given symbolic names in the C header
Fred Drake4b7d5a41996-09-11 21:57:40 +000062file \file{Include/graminit.h} and the Python module
Fred Drake88223901998-02-09 20:52:48 +000063\module{symbol}. Each additional element of the sequence represents
Guido van Rossum47478871996-08-21 14:32:37 +000064a component of the production as recognized in the input string: these
65are always sequences which have the same form as the parent. An
66important aspect of this structure which should be noted is that
67keywords used to identify the parent node type, such as the keyword
Fred Drake88223901998-02-09 20:52:48 +000068\keyword{if} in an \constant{if_stmt}, are included in the node tree without
69any special treatment. For example, the \keyword{if} keyword is
Guido van Rossum4b73a061995-10-11 17:30:04 +000070represented by the tuple \code{(1, 'if')}, where \code{1} is the
Fred Drake4b7d5a41996-09-11 21:57:40 +000071numeric value associated with all \code{NAME} tokens, including
Guido van Rossum47478871996-08-21 14:32:37 +000072variable and function names defined by the user. In an alternate form
73returned when line number information is requested, the same token
74might be represented as \code{(1, 'if', 12)}, where the \code{12}
75represents the line number at which the terminal symbol was found.
Guido van Rossum4b73a061995-10-11 17:30:04 +000076
77Terminal elements are represented in much the same way, but without
78any child elements and the addition of the source text which was
Fred Drake88223901998-02-09 20:52:48 +000079identified. The example of the \keyword{if} keyword above is
Guido van Rossum4b73a061995-10-11 17:30:04 +000080representative. The various types of terminal symbols are defined in
Fred Drake4b7d5a41996-09-11 21:57:40 +000081the C header file \file{Include/token.h} and the Python module
Fred Drake88223901998-02-09 20:52:48 +000082\module{token}.
Guido van Rossum4b73a061995-10-11 17:30:04 +000083
Fred Drake4b7d5a41996-09-11 21:57:40 +000084The AST objects are not required to support the functionality of this
85module, but are provided for three purposes: to allow an application
86to amortize the cost of processing complex parse trees, to provide a
87parse tree representation which conserves memory space when compared
88to the Python list or tuple representation, and to ease the creation
89of additional modules in C which manipulate parse trees. A simple
90``wrapper'' class may be created in Python to hide the use of AST
Fred Drake88223901998-02-09 20:52:48 +000091objects.
Guido van Rossum4b73a061995-10-11 17:30:04 +000092
Fred Drake88223901998-02-09 20:52:48 +000093The \module{parser} module defines functions for a few distinct
Fred Drake4b7d5a41996-09-11 21:57:40 +000094purposes. The most important purposes are to create AST objects and
95to convert AST objects to other representations such as parse trees
96and compiled code objects, but there are also functions which serve to
97query the type of parse tree represented by an AST object.
Guido van Rossum4b73a061995-10-11 17:30:04 +000098
Fred Drake4b7d5a41996-09-11 21:57:40 +000099
100\subsection{Creating AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000101\label{Creating ASTs}
Fred Drake4b7d5a41996-09-11 21:57:40 +0000102
103AST objects may be created from source code or from a parse tree.
104When creating an AST object from source, different functions are used
105to create the \code{'eval'} and \code{'exec'} forms.
106
107\begin{funcdesc}{expr}{string}
Fred Drake88223901998-02-09 20:52:48 +0000108The \function{expr()} function parses the parameter \code{\var{string}}
109as if it were an input to \samp{compile(\var{string}, 'eval')}. If
Fred Drake4b7d5a41996-09-11 21:57:40 +0000110the parse succeeds, an AST object is created to hold the internal
111parse tree representation, otherwise an appropriate exception is
112thrown.
113\end{funcdesc}
114
115\begin{funcdesc}{suite}{string}
Fred Drake88223901998-02-09 20:52:48 +0000116The \function{suite()} function parses the parameter \code{\var{string}}
117as if it were an input to \samp{compile(\var{string}, 'exec')}. If
Fred Drake4b7d5a41996-09-11 21:57:40 +0000118the parse succeeds, an AST object is created to hold the internal
119parse tree representation, otherwise an appropriate exception is
120thrown.
121\end{funcdesc}
122
123\begin{funcdesc}{sequence2ast}{sequence}
124This function accepts a parse tree represented as a sequence and
125builds an internal representation if possible. If it can validate
126that the tree conforms to the Python grammar and all nodes are valid
127node types in the host version of Python, an AST object is created
128from the internal representation and returned to the called. If there
129is a problem creating the internal representation, or if the tree
Fred Drake88223901998-02-09 20:52:48 +0000130cannot be validated, a \exception{ParserError} exception is thrown. An AST
Fred Drake4b7d5a41996-09-11 21:57:40 +0000131object created this way should not be assumed to compile correctly;
132normal exceptions thrown by compilation may still be initiated when
Fred Drake88223901998-02-09 20:52:48 +0000133the AST object is passed to \function{compileast()}. This may indicate
134problems not related to syntax (such as a \exception{MemoryError}
Fred Drake4b7d5a41996-09-11 21:57:40 +0000135exception), but may also be due to constructs such as the result of
136parsing \code{del f(0)}, which escapes the Python parser but is
137checked by the bytecode compiler.
138
139Sequences representing terminal tokens may be represented as either
140two-element lists of the form \code{(1, 'name')} or as three-element
141lists of the form \code{(1, 'name', 56)}. If the third element is
142present, it is assumed to be a valid line number. The line number
143may be specified for any subset of the terminal symbols in the input
144tree.
145\end{funcdesc}
146
147\begin{funcdesc}{tuple2ast}{sequence}
Fred Drake88223901998-02-09 20:52:48 +0000148This is the same function as \function{sequence2ast()}. This entry point
Fred Drake4b7d5a41996-09-11 21:57:40 +0000149is maintained for backward compatibility.
150\end{funcdesc}
151
152
153\subsection{Converting AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000154\label{Converting ASTs}
Fred Drake4b7d5a41996-09-11 21:57:40 +0000155
156AST objects, regardless of the input used to create them, may be
157converted to parse trees represented as list- or tuple- trees, or may
158be compiled into executable code objects. Parse trees may be
159extracted with or without line numbering information.
160
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000161\begin{funcdesc}{ast2list}{ast\optional{, line_info}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000162This function accepts an AST object from the caller in
Guido van Rossum47478871996-08-21 14:32:37 +0000163\code{\var{ast}} and returns a Python list representing the
164equivelent parse tree. The resulting list representation can be used
Fred Drake4b7d5a41996-09-11 21:57:40 +0000165for inspection or the creation of a new parse tree in list form. This
166function does not fail so long as memory is available to build the
167list representation. If the parse tree will only be used for
Fred Drake88223901998-02-09 20:52:48 +0000168inspection, \function{ast2tuple()} should be used instead to reduce memory
Fred Drake4b7d5a41996-09-11 21:57:40 +0000169consumption and fragmentation. When the list representation is
170required, this function is significantly faster than retrieving a
171tuple representation and converting that to nested lists.
Guido van Rossum47478871996-08-21 14:32:37 +0000172
Fred Drake4b7d5a41996-09-11 21:57:40 +0000173If \code{\var{line_info}} is true, line number information will be
174included for all terminal tokens as a third element of the list
Fred Drake9abe64a1996-12-05 22:28:43 +0000175representing the token. Note that the line number provided specifies
Fred Drake88223901998-02-09 20:52:48 +0000176the line on which the token \emph{ends}. This information is
Fred Drake9abe64a1996-12-05 22:28:43 +0000177omitted if the flag is false or omitted.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000178\end{funcdesc}
179
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000180\begin{funcdesc}{ast2tuple}{ast\optional{, line_info}}
Guido van Rossum47478871996-08-21 14:32:37 +0000181This function accepts an AST object from the caller in
182\code{\var{ast}} and returns a Python tuple representing the
183equivelent parse tree. Other than returning a tuple instead of a
Fred Drake88223901998-02-09 20:52:48 +0000184list, this function is identical to \function{ast2list()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000185
Fred Drake4b7d5a41996-09-11 21:57:40 +0000186If \code{\var{line_info}} is true, line number information will be
187included for all terminal tokens as a third element of the list
188representing the token. This information is omitted if the flag is
189false or omitted.
Guido van Rossum47478871996-08-21 14:32:37 +0000190\end{funcdesc}
191
Fred Drakecce10901998-03-17 06:33:25 +0000192\begin{funcdesc}{compileast}{ast\optional{, filename\code{ = '<ast>'}}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000193The Python byte compiler can be invoked on an AST object to produce
194code objects which can be used as part of an \code{exec} statement or
Fred Drake88223901998-02-09 20:52:48 +0000195a call to the built-in \function{eval()}\bifuncindex{eval} function.
196This function provides the interface to the compiler, passing the
197internal parse tree from \code{\var{ast}} to the parser, using the
198source file name specified by the \code{\var{filename}} parameter.
199The default value supplied for \code{\var{filename}} indicates that
200the source was an AST object.
Guido van Rossum47478871996-08-21 14:32:37 +0000201
202Compiling an AST object may result in exceptions related to
Fred Drake88223901998-02-09 20:52:48 +0000203compilation; an example would be a \exception{SyntaxError} caused by the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000204parse tree for \code{del f(0)}: this statement is considered legal
Guido van Rossum47478871996-08-21 14:32:37 +0000205within the formal grammar for Python but is not a legal language
Fred Drake88223901998-02-09 20:52:48 +0000206construct. The \exception{SyntaxError} raised for this condition is
Guido van Rossum47478871996-08-21 14:32:37 +0000207actually generated by the Python byte-compiler normally, which is why
Fred Drake88223901998-02-09 20:52:48 +0000208it can be raised at this point by the \module{parser} module. Most
Guido van Rossum47478871996-08-21 14:32:37 +0000209causes of compilation failure can be diagnosed programmatically by
210inspection of the parse tree.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000211\end{funcdesc}
212
213
Fred Drake4b7d5a41996-09-11 21:57:40 +0000214\subsection{Queries on AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000215\label{Querying ASTs}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000216
Fred Drake4b7d5a41996-09-11 21:57:40 +0000217Two functions are provided which allow an application to determine if
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000218an AST was created as an expression or a suite. Neither of these
Fred Drake4b7d5a41996-09-11 21:57:40 +0000219functions can be used to determine if an AST was created from source
Fred Drake88223901998-02-09 20:52:48 +0000220code via \function{expr()} or \function{suite()} or from a parse tree
221via \function{sequence2ast()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000222
223\begin{funcdesc}{isexpr}{ast}
224When \code{\var{ast}} represents an \code{'eval'} form, this function
Fred Drake88223901998-02-09 20:52:48 +0000225returns true, otherwise it returns false. This is useful, since code
226objects normally cannot be queried for this information using existing
227built-in functions. Note that the code objects created by
228\function{compileast()} cannot be queried like this either, and are
229identical to those created by the built-in
230\function{compile()}\bifuncindex{compile} function.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000231\end{funcdesc}
232
233
234\begin{funcdesc}{issuite}{ast}
Fred Drake88223901998-02-09 20:52:48 +0000235This function mirrors \function{isexpr()} in that it reports whether an
Fred Drake4b7d5a41996-09-11 21:57:40 +0000236AST object represents an \code{'exec'} form, commonly known as a
237``suite.'' It is not safe to assume that this function is equivelent
Fred Drake88223901998-02-09 20:52:48 +0000238to \samp{not isexpr(\var{ast})}, as additional syntactic fragments may
Fred Drake4b7d5a41996-09-11 21:57:40 +0000239be supported in the future.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000240\end{funcdesc}
241
242
Guido van Rossum4b73a061995-10-11 17:30:04 +0000243\subsection{Exceptions and Error Handling}
Fred Draked67e12e1998-02-20 05:49:37 +0000244\label{AST Errors}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000245
246The parser module defines a single exception, but may also pass other
247built-in exceptions from other portions of the Python runtime
248environment. See each function for information about the exceptions
249it can raise.
250
251\begin{excdesc}{ParserError}
252Exception raised when a failure occurs within the parser module. This
253is generally produced for validation failures rather than the built in
Fred Drake88223901998-02-09 20:52:48 +0000254\exception{SyntaxError} thrown during normal parsing.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000255The exception argument is either a string describing the reason of the
Guido van Rossum47478871996-08-21 14:32:37 +0000256failure or a tuple containing a sequence causing the failure from a parse
Fred Drake88223901998-02-09 20:52:48 +0000257tree passed to \function{sequence2ast()} and an explanatory string. Calls to
258\function{sequence2ast()} need to be able to handle either type of exception,
Guido van Rossum4b73a061995-10-11 17:30:04 +0000259while calls to other functions in the module will only need to be
260aware of the simple string values.
261\end{excdesc}
262
Fred Drake88223901998-02-09 20:52:48 +0000263Note that the functions \function{compileast()}, \function{expr()}, and
264\function{suite()} may throw exceptions which are normally thrown by the
Guido van Rossum4b73a061995-10-11 17:30:04 +0000265parsing and compilation process. These include the built in
Fred Drake88223901998-02-09 20:52:48 +0000266exceptions \exception{MemoryError}, \exception{OverflowError},
267\exception{SyntaxError}, and \exception{SystemError}. In these cases, these
Guido van Rossum4b73a061995-10-11 17:30:04 +0000268exceptions carry all the meaning normally associated with them. Refer
269to the descriptions of each function for detailed information.
270
Guido van Rossum4b73a061995-10-11 17:30:04 +0000271
Guido van Rossum47478871996-08-21 14:32:37 +0000272\subsection{AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000273\label{AST Objects}
Guido van Rossum47478871996-08-21 14:32:37 +0000274
Fred Drakecc444e31998-03-08 06:47:24 +0000275AST objects returned by \function{expr()}, \function{suite()} and
Fred Drake88223901998-02-09 20:52:48 +0000276\function{sequence2ast()} have no methods of their own.
Fred Drakecc444e31998-03-08 06:47:24 +0000277
Fred Drakeaf370ea1998-04-05 20:23:02 +0000278Ordered and equality comparisons are supported between AST objects.
Fred Drakec4f1ca11998-04-13 16:27:27 +0000279Pickling of AST objects (using the \module{pickle} module) is also
280supported.
Fred Drakeaf370ea1998-04-05 20:23:02 +0000281
Fred Drakecc444e31998-03-08 06:47:24 +0000282\begin{datadesc}{ASTType}
283The type of the objects returned by \function{expr()},
284\function{suite()} and \function{sequence2ast()}.
Fred Drakecc444e31998-03-08 06:47:24 +0000285\end{datadesc}
Guido van Rossum47478871996-08-21 14:32:37 +0000286
287
Fred Drake916d8f81998-04-13 18:46:16 +0000288AST objects have the following methods:
289
290
291\begin{methoddesc}[AST]{compile}{\optional{filename}}
292Same as \code{compileast(\var{ast}, \var{filename})}.
293\end{methoddesc}
294
295\begin{methoddesc}[AST]{isexpr}{}
296Same as \code{isexpr(\var{ast})}.
297\end{methoddesc}
298
299\begin{methoddesc}[AST]{issuite}{}
300Same as \code{issuite(\var{ast})}.
301\end{methoddesc}
302
303\begin{methoddesc}[AST]{tolist}{\optional{line_info}}
304Same as \code{ast2list(\var{ast}, \var{line_info})}.
305\end{methoddesc}
306
307\begin{methoddesc}[AST]{totuple}{\optional{line_info}}
308Same as \code{ast2tuple(\var{ast}, \var{line_info})}.
309\end{methoddesc}
310
311
Guido van Rossum8206fb91996-08-26 00:33:29 +0000312\subsection{Examples}
Fred Drake4b3f0311996-12-13 22:04:31 +0000313\nodename{AST Examples}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000314
Guido van Rossum47478871996-08-21 14:32:37 +0000315The parser modules allows operations to be performed on the parse tree
316of Python source code before the bytecode is generated, and provides
Fred Drake4b7d5a41996-09-11 21:57:40 +0000317for inspection of the parse tree for information gathering purposes.
318Two examples are presented. The simple example demonstrates emulation
Fred Drake88223901998-02-09 20:52:48 +0000319of the \function{compile()}\bifuncindex{compile} built-in function and
320the complex example shows the use of a parse tree for information
321discovery.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000322
Fred Drakeaf370ea1998-04-05 20:23:02 +0000323\subsubsection{Emulation of \function{compile()}}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000324
325While many useful operations may take place between parsing and
Guido van Rossum47478871996-08-21 14:32:37 +0000326bytecode generation, the simplest operation is to do nothing. For
Fred Drake88223901998-02-09 20:52:48 +0000327this purpose, using the \module{parser} module to produce an
Guido van Rossum47478871996-08-21 14:32:37 +0000328intermediate data structure is equivelent to the code
329
Fred Drake19479911998-02-13 06:58:54 +0000330\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000331>>> code = compile('a + 5', 'eval')
332>>> a = 5
333>>> eval(code)
33410
Fred Drake19479911998-02-13 06:58:54 +0000335\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000336
Fred Drake88223901998-02-09 20:52:48 +0000337The equivelent operation using the \module{parser} module is somewhat
Guido van Rossum47478871996-08-21 14:32:37 +0000338longer, and allows the intermediate internal parse tree to be retained
339as an AST object:
Guido van Rossum4b73a061995-10-11 17:30:04 +0000340
Fred Drake19479911998-02-13 06:58:54 +0000341\begin{verbatim}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000342>>> import parser
343>>> ast = parser.expr('a + 5')
344>>> code = parser.compileast(ast)
345>>> a = 5
346>>> eval(code)
34710
Fred Drake19479911998-02-13 06:58:54 +0000348\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000349
Guido van Rossum8206fb91996-08-26 00:33:29 +0000350An application which needs both AST and code objects can package this
351code into readily available functions:
352
Fred Drake19479911998-02-13 06:58:54 +0000353\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000354import parser
355
356def load_suite(source_string):
357 ast = parser.suite(source_string)
358 code = parser.compileast(ast)
359 return ast, code
360
361def load_expression(source_string):
362 ast = parser.expr(source_string)
363 code = parser.compileast(ast)
364 return ast, code
Fred Drake19479911998-02-13 06:58:54 +0000365\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000366
Guido van Rossum8206fb91996-08-26 00:33:29 +0000367\subsubsection{Information Discovery}
368
Fred Drake4b7d5a41996-09-11 21:57:40 +0000369Some applications benefit from direct access to the parse tree. The
370remainder of this section demonstrates how the parse tree provides
371access to module documentation defined in docstrings without requiring
372that the code being examined be loaded into a running interpreter via
Fred Drake88223901998-02-09 20:52:48 +0000373\keyword{import}. This can be very useful for performing analyses of
Fred Drake4b7d5a41996-09-11 21:57:40 +0000374untrusted code.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000375
Guido van Rossum47478871996-08-21 14:32:37 +0000376Generally, the example will demonstrate how the parse tree may be
377traversed to distill interesting information. Two functions and a set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000378of classes are developed which provide programmatic access to high
Guido van Rossum47478871996-08-21 14:32:37 +0000379level function and class definitions provided by a module. The
380classes extract information from the parse tree and provide access to
381the information at a useful semantic level, one function provides a
382simple low-level pattern matching capability, and the other function
383defines a high-level interface to the classes by handling file
384operations on behalf of the caller. All source files mentioned here
385which are not part of the Python installation are located in the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000386\file{Demo/parser/} directory of the distribution.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000387
Guido van Rossum8206fb91996-08-26 00:33:29 +0000388The dynamic nature of Python allows the programmer a great deal of
389flexibility, but most modules need only a limited measure of this when
390defining classes, functions, and methods. In this example, the only
391definitions that will be considered are those which are defined in the
Fred Drake88223901998-02-09 20:52:48 +0000392top level of their context, e.g., a function defined by a \keyword{def}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000393statement at column zero of a module, but not a function defined
Fred Drake4b7d5a41996-09-11 21:57:40 +0000394within a branch of an \code{if} ... \code{else} construct, though
Guido van Rossum8206fb91996-08-26 00:33:29 +0000395there are some good reasons for doing so in some situations. Nesting
396of definitions will be handled by the code developed in the example.
397
Guido van Rossum47478871996-08-21 14:32:37 +0000398To construct the upper-level extraction methods, we need to know what
399the parse tree structure looks like and how much of it we actually
Fred Drake4b7d5a41996-09-11 21:57:40 +0000400need to be concerned about. Python uses a moderately deep parse tree
Guido van Rossum47478871996-08-21 14:32:37 +0000401so there are a large number of intermediate nodes. It is important to
402read and understand the formal grammar used by Python. This is
403specified in the file \file{Grammar/Grammar} in the distribution.
404Consider the simplest case of interest when searching for docstrings:
Guido van Rossum8206fb91996-08-26 00:33:29 +0000405a module consisting of a docstring and nothing else. (See file
406\file{docstring.py}.)
Guido van Rossum4b73a061995-10-11 17:30:04 +0000407
Fred Drake19479911998-02-13 06:58:54 +0000408\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000409"""Some documentation.
410"""
Fred Drake19479911998-02-13 06:58:54 +0000411\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000412
Guido van Rossum47478871996-08-21 14:32:37 +0000413Using the interpreter to take a look at the parse tree, we find a
414bewildering mass of numbers and parentheses, with the documentation
Fred Drake4b7d5a41996-09-11 21:57:40 +0000415buried deep in nested tuples.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000416
Fred Drake19479911998-02-13 06:58:54 +0000417\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000418>>> import parser
419>>> import pprint
420>>> ast = parser.suite(open('docstring.py').read())
421>>> tup = parser.ast2tuple(ast)
422>>> pprint.pprint(tup)
423(257,
424 (264,
425 (265,
426 (266,
427 (267,
428 (307,
429 (287,
430 (288,
431 (289,
432 (290,
433 (292,
434 (293,
435 (294,
436 (295,
437 (296,
438 (297,
439 (298,
440 (299,
441 (300, (3, '"""Some documentation.\012"""'))))))))))))))))),
442 (4, ''))),
443 (4, ''),
444 (0, ''))
Fred Drake19479911998-02-13 06:58:54 +0000445\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000446
Guido van Rossum47478871996-08-21 14:32:37 +0000447The numbers at the first element of each node in the tree are the node
448types; they map directly to terminal and non-terminal symbols in the
449grammar. Unfortunately, they are represented as integers in the
450internal representation, and the Python structures generated do not
Fred Drake88223901998-02-09 20:52:48 +0000451change that. However, the \module{symbol} and \module{token} modules
Guido van Rossum47478871996-08-21 14:32:37 +0000452provide symbolic names for the node types and dictionaries which map
453from the integers to the symbolic names for the node types.
454
455In the output presented above, the outermost tuple contains four
456elements: the integer \code{257} and three additional tuples. Node
Fred Drake88223901998-02-09 20:52:48 +0000457type \code{257} has the symbolic name \constant{file_input}. Each of
Guido van Rossum47478871996-08-21 14:32:37 +0000458these inner tuples contains an integer as the first element; these
459integers, \code{264}, \code{4}, and \code{0}, represent the node types
Fred Drake88223901998-02-09 20:52:48 +0000460\constant{stmt}, \constant{NEWLINE}, and \constant{ENDMARKER},
461respectively.
Guido van Rossum47478871996-08-21 14:32:37 +0000462Note that these values may change depending on the version of Python
463you are using; consult \file{symbol.py} and \file{token.py} for
464details of the mapping. It should be fairly clear that the outermost
465node is related primarily to the input source rather than the contents
Fred Drake88223901998-02-09 20:52:48 +0000466of the file, and may be disregarded for the moment. The \constant{stmt}
Guido van Rossum47478871996-08-21 14:32:37 +0000467node is much more interesting. In particular, all docstrings are
468found in subtrees which are formed exactly as this node is formed,
469with the only difference being the string itself. The association
470between the docstring in a similar tree and the defined entity (class,
471function, or module) which it describes is given by the position of
472the docstring subtree within the tree defining the described
473structure.
474
475By replacing the actual docstring with something to signify a variable
Fred Drake4b7d5a41996-09-11 21:57:40 +0000476component of the tree, we allow a simple pattern matching approach to
477check any given subtree for equivelence to the general pattern for
478docstrings. Since the example demonstrates information extraction, we
479can safely require that the tree be in tuple form rather than list
480form, allowing a simple variable representation to be
481\code{['variable_name']}. A simple recursive function can implement
Guido van Rossum47478871996-08-21 14:32:37 +0000482the pattern matching, returning a boolean and a dictionary of variable
Guido van Rossum8206fb91996-08-26 00:33:29 +0000483name to value mappings. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000484
Fred Drake19479911998-02-13 06:58:54 +0000485\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000486from types import ListType, TupleType
487
488def match(pattern, data, vars=None):
489 if vars is None:
490 vars = {}
491 if type(pattern) is ListType:
492 vars[pattern[0]] = data
493 return 1, vars
494 if type(pattern) is not TupleType:
495 return (pattern == data), vars
496 if len(data) != len(pattern):
497 return 0, vars
498 for pattern, data in map(None, pattern, data):
499 same, vars = match(pattern, data, vars)
500 if not same:
501 break
502 return same, vars
Fred Drake19479911998-02-13 06:58:54 +0000503\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000504
Fred Drake4b7d5a41996-09-11 21:57:40 +0000505Using this simple representation for syntactic variables and the symbolic
Guido van Rossum8206fb91996-08-26 00:33:29 +0000506node types, the pattern for the candidate docstring subtrees becomes
507fairly readable. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000508
Fred Drake19479911998-02-13 06:58:54 +0000509\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000510import symbol
511import token
512
513DOCSTRING_STMT_PATTERN = (
514 symbol.stmt,
515 (symbol.simple_stmt,
516 (symbol.small_stmt,
517 (symbol.expr_stmt,
518 (symbol.testlist,
519 (symbol.test,
520 (symbol.and_test,
521 (symbol.not_test,
522 (symbol.comparison,
523 (symbol.expr,
524 (symbol.xor_expr,
525 (symbol.and_expr,
526 (symbol.shift_expr,
527 (symbol.arith_expr,
528 (symbol.term,
529 (symbol.factor,
530 (symbol.power,
531 (symbol.atom,
532 (token.STRING, ['docstring'])
533 )))))))))))))))),
534 (token.NEWLINE, '')
535 ))
Fred Drake19479911998-02-13 06:58:54 +0000536\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000537
Fred Drake88223901998-02-09 20:52:48 +0000538Using the \function{match()} function with this pattern, extracting the
Guido van Rossum47478871996-08-21 14:32:37 +0000539module docstring from the parse tree created previously is easy:
540
Fred Drake19479911998-02-13 06:58:54 +0000541\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000542>>> found, vars = match(DOCSTRING_STMT_PATTERN, tup[1])
543>>> found
5441
545>>> vars
546{'docstring': '"""Some documentation.\012"""'}
Fred Drake19479911998-02-13 06:58:54 +0000547\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000548
Guido van Rossum47478871996-08-21 14:32:37 +0000549Once specific data can be extracted from a location where it is
550expected, the question of where information can be expected
551needs to be answered. When dealing with docstrings, the answer is
Fred Drake88223901998-02-09 20:52:48 +0000552fairly simple: the docstring is the first \constant{stmt} node in a code
553block (\constant{file_input} or \constant{suite} node types). A module
554consists of a single \constant{file_input} node, and class and function
555definitions each contain exactly one \constant{suite} node. Classes and
Guido van Rossum47478871996-08-21 14:32:37 +0000556functions are readily identified as subtrees of code block nodes which
557start with \code{(stmt, (compound_stmt, (classdef, ...} or
558\code{(stmt, (compound_stmt, (funcdef, ...}. Note that these subtrees
Fred Drake88223901998-02-09 20:52:48 +0000559cannot be matched by \function{match()} since it does not support multiple
Guido van Rossum47478871996-08-21 14:32:37 +0000560sibling nodes to match without regard to number. A more elaborate
561matching function could be used to overcome this limitation, but this
562is sufficient for the example.
563
Guido van Rossum8206fb91996-08-26 00:33:29 +0000564Given the ability to determine whether a statement might be a
565docstring and extract the actual string from the statement, some work
566needs to be performed to walk the parse tree for an entire module and
567extract information about the names defined in each context of the
568module and associate any docstrings with the names. The code to
569perform this work is not complicated, but bears some explanation.
570
571The public interface to the classes is straightforward and should
572probably be somewhat more flexible. Each ``major'' block of the
573module is described by an object providing several methods for inquiry
574and a constructor which accepts at least the subtree of the complete
Fred Drakeb0df5671998-02-18 15:59:13 +0000575parse tree which it represents. The \class{ModuleInfo} constructor
576accepts an optional \var{name} parameter since it cannot
Guido van Rossum8206fb91996-08-26 00:33:29 +0000577otherwise determine the name of the module.
578
Fred Drake88223901998-02-09 20:52:48 +0000579The public classes include \class{ClassInfo}, \class{FunctionInfo},
580and \class{ModuleInfo}. All objects provide the
581methods \method{get_name()}, \method{get_docstring()},
582\method{get_class_names()}, and \method{get_class_info()}. The
583\class{ClassInfo} objects support \method{get_method_names()} and
584\method{get_method_info()} while the other classes provide
585\method{get_function_names()} and \method{get_function_info()}.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000586
587Within each of the forms of code block that the public classes
588represent, most of the required information is in the same form and is
Fred Drake4b7d5a41996-09-11 21:57:40 +0000589accessed in the same way, with classes having the distinction that
Guido van Rossum8206fb91996-08-26 00:33:29 +0000590functions defined at the top level are referred to as ``methods.''
591Since the difference in nomenclature reflects a real semantic
Fred Drake4b7d5a41996-09-11 21:57:40 +0000592distinction from functions defined outside of a class, the
593implementation needs to maintain the distinction.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000594Hence, most of the functionality of the public classes can be
Fred Drake88223901998-02-09 20:52:48 +0000595implemented in a common base class, \class{SuiteInfoBase}, with the
Guido van Rossum8206fb91996-08-26 00:33:29 +0000596accessors for function and method information provided elsewhere.
597Note that there is only one class which represents function and method
Fred Drake88223901998-02-09 20:52:48 +0000598information; this parallels the use of the \keyword{def} statement to
Fred Drake4b7d5a41996-09-11 21:57:40 +0000599define both types of elements.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000600
Fred Drake88223901998-02-09 20:52:48 +0000601Most of the accessor functions are declared in \class{SuiteInfoBase}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000602and do not need to be overriden by subclasses. More importantly, the
603extraction of most information from a parse tree is handled through a
Fred Drake88223901998-02-09 20:52:48 +0000604method called by the \class{SuiteInfoBase} constructor. The example
Guido van Rossum8206fb91996-08-26 00:33:29 +0000605code for most of the classes is clear when read alongside the formal
606grammar, but the method which recursively creates new information
607objects requires further examination. Here is the relevant part of
Fred Drake88223901998-02-09 20:52:48 +0000608the \class{SuiteInfoBase} definition from \file{example.py}:
Guido van Rossum8206fb91996-08-26 00:33:29 +0000609
Fred Drake19479911998-02-13 06:58:54 +0000610\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000611class SuiteInfoBase:
612 _docstring = ''
613 _name = ''
614
615 def __init__(self, tree = None):
616 self._class_info = {}
617 self._function_info = {}
618 if tree:
619 self._extract_info(tree)
620
621 def _extract_info(self, tree):
622 # extract docstring
623 if len(tree) == 2:
624 found, vars = match(DOCSTRING_STMT_PATTERN[1], tree[1])
625 else:
626 found, vars = match(DOCSTRING_STMT_PATTERN, tree[3])
627 if found:
628 self._docstring = eval(vars['docstring'])
629 # discover inner definitions
630 for node in tree[1:]:
631 found, vars = match(COMPOUND_STMT_PATTERN, node)
632 if found:
633 cstmt = vars['compound']
634 if cstmt[0] == symbol.funcdef:
635 name = cstmt[2][1]
636 self._function_info[name] = FunctionInfo(cstmt)
637 elif cstmt[0] == symbol.classdef:
638 name = cstmt[2][1]
639 self._class_info[name] = ClassInfo(cstmt)
Fred Drake19479911998-02-13 06:58:54 +0000640\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000641
Guido van Rossum8206fb91996-08-26 00:33:29 +0000642After initializing some internal state, the constructor calls the
Fred Drake88223901998-02-09 20:52:48 +0000643\method{_extract_info()} method. This method performs the bulk of the
Guido van Rossum8206fb91996-08-26 00:33:29 +0000644information extraction which takes place in the entire example. The
645extraction has two distinct phases: the location of the docstring for
646the parse tree passed in, and the discovery of additional definitions
647within the code block represented by the parse tree.
648
Fred Drake88223901998-02-09 20:52:48 +0000649The initial \keyword{if} test determines whether the nested suite is of
Guido van Rossum8206fb91996-08-26 00:33:29 +0000650the ``short form'' or the ``long form.'' The short form is used when
651the code block is on the same line as the definition of the code
652block, as in
653
Fred Drakebbe60681998-01-09 22:24:14 +0000654\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000655def square(x): "Square an argument."; return x ** 2
Fred Drakebbe60681998-01-09 22:24:14 +0000656\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000657
Guido van Rossum8206fb91996-08-26 00:33:29 +0000658while the long form uses an indented block and allows nested
659definitions:
660
Fred Drake19479911998-02-13 06:58:54 +0000661\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000662def make_power(exp):
663 "Make a function that raises an argument to the exponent `exp'."
664 def raiser(x, y=exp):
665 return x ** y
666 return raiser
Fred Drake19479911998-02-13 06:58:54 +0000667\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000668
Guido van Rossum8206fb91996-08-26 00:33:29 +0000669When the short form is used, the code block may contain a docstring as
Fred Drake88223901998-02-09 20:52:48 +0000670the first, and possibly only, \constant{small_stmt} element. The
Guido van Rossum8206fb91996-08-26 00:33:29 +0000671extraction of such a docstring is slightly different and requires only
672a portion of the complete pattern used in the more common case. As
Fred Drake4b7d5a41996-09-11 21:57:40 +0000673implemented, the docstring will only be found if there is only
Fred Drake88223901998-02-09 20:52:48 +0000674one \constant{small_stmt} node in the \constant{simple_stmt} node.
675Since most functions and methods which use the short form do not
676provide a docstring, this may be considered sufficient. The
677extraction of the docstring proceeds using the \function{match()} function
678as described above, and the value of the docstring is stored as an
679attribute of the \class{SuiteInfoBase} object.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000680
Fred Drake4b7d5a41996-09-11 21:57:40 +0000681After docstring extraction, a simple definition discovery
Fred Drake88223901998-02-09 20:52:48 +0000682algorithm operates on the \constant{stmt} nodes of the
683\constant{suite} node. The special case of the short form is not
684tested; since there are no \constant{stmt} nodes in the short form,
685the algorithm will silently skip the single \constant{simple_stmt}
686node and correctly not discover any nested definitions.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000687
Fred Drake4b7d5a41996-09-11 21:57:40 +0000688Each statement in the code block is categorized as
689a class definition, function or method definition, or
Guido van Rossum8206fb91996-08-26 00:33:29 +0000690something else. For the definition statements, the name of the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000691element defined is extracted and a representation object
Guido van Rossum8206fb91996-08-26 00:33:29 +0000692appropriate to the definition is created with the defining subtree
693passed as an argument to the constructor. The repesentation objects
694are stored in instance variables and may be retrieved by name using
695the appropriate accessor methods.
696
697The public classes provide any accessors required which are more
Fred Drake88223901998-02-09 20:52:48 +0000698specific than those provided by the \class{SuiteInfoBase} class, but
Guido van Rossum8206fb91996-08-26 00:33:29 +0000699the real extraction algorithm remains common to all forms of code
700blocks. A high-level function can be used to extract the complete set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000701of information from a source file. (See file \file{example.py}.)
Guido van Rossum8206fb91996-08-26 00:33:29 +0000702
Fred Drake19479911998-02-13 06:58:54 +0000703\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000704def get_docs(fileName):
705 source = open(fileName).read()
706 import os
707 basename = os.path.basename(os.path.splitext(fileName)[0])
708 import parser
709 ast = parser.suite(source)
710 tup = parser.ast2tuple(ast)
711 return ModuleInfo(tup, basename)
Fred Drake19479911998-02-13 06:58:54 +0000712\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000713
Guido van Rossum8206fb91996-08-26 00:33:29 +0000714This provides an easy-to-use interface to the documentation of a
715module. If information is required which is not extracted by the code
716of this example, the code may be extended at clearly defined points to
717provide additional capabilities.
Guido van Rossum47478871996-08-21 14:32:37 +0000718
Fred Drakebbe60681998-01-09 22:24:14 +0000719\begin{seealso}
720
Fred Drake45c634e1998-04-09 15:44:58 +0000721\seemodule{symbol}{useful constants representing internal nodes of the
722parse tree}
Fred Drakebbe60681998-01-09 22:24:14 +0000723
Fred Drake45c634e1998-04-09 15:44:58 +0000724\seemodule{token}{useful constants representing leaf nodes of the
725parse tree and functions for testing node values}
Fred Drakebbe60681998-01-09 22:24:14 +0000726
727\end{seealso}