blob: 49a92e51f8bb24c525374e10beae22a76649cfa6 [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 Drake295da241998-08-10 19:42:37 +000012\section{\module{parser} ---
Fred Drake9f033801999-02-19 22:56:08 +000013 Access parse trees for Python code}
14
Fred Drakeb91e9341998-07-23 17:59:49 +000015\declaremodule{builtin}{parser}
Fred Drake9f033801999-02-19 22:56:08 +000016\modulesynopsis{Access parse trees for Python source code.}
Fred Drake295da241998-08-10 19:42:37 +000017\moduleauthor{Fred L. Drake, Jr.}{fdrake@acm.org}
18\sectionauthor{Fred L. Drake, Jr.}{fdrake@acm.org}
Fred Drakeb91e9341998-07-23 17:59:49 +000019
Fred Drakeb91e9341998-07-23 17:59:49 +000020
Fred Drake88223901998-02-09 20:52:48 +000021\index{parsing!Python source code}
Guido van Rossum4b73a061995-10-11 17:30:04 +000022
Fred Drake88223901998-02-09 20:52:48 +000023The \module{parser} module provides an interface to Python's internal
Guido van Rossum4b73a061995-10-11 17:30:04 +000024parser and byte-code compiler. The primary purpose for this interface
25is to allow Python code to edit the parse tree of a Python expression
Fred Drake4b7d5a41996-09-11 21:57:40 +000026and create executable code from this. This is better than trying
27to parse and modify an arbitrary Python code fragment as a string
28because parsing is performed in a manner identical to the code
29forming the application. It is also faster.
Guido van Rossum4b73a061995-10-11 17:30:04 +000030
31There are a few things to note about this module which are important
32to making use of the data structures created. This is not a tutorial
Fred Drake4b7d5a41996-09-11 21:57:40 +000033on editing the parse trees for Python code, but some examples of using
Fred Drake88223901998-02-09 20:52:48 +000034the \module{parser} module are presented.
Guido van Rossum4b73a061995-10-11 17:30:04 +000035
36Most importantly, a good understanding of the Python grammar processed
37by the internal parser is required. For full information on the
Fred Drake88223901998-02-09 20:52:48 +000038language syntax, refer to the \emph{Python Language Reference}. The
39parser itself is created from a grammar specification defined in the file
Fred Drake4b7d5a41996-09-11 21:57:40 +000040\file{Grammar/Grammar} in the standard Python distribution. The parse
Fred Drakecc444e31998-03-08 06:47:24 +000041trees stored in the AST objects created by this module are the
Guido van Rossum4b73a061995-10-11 17:30:04 +000042actual output from the internal parser when created by the
Fred Drake88223901998-02-09 20:52:48 +000043\function{expr()} or \function{suite()} functions, described below. The AST
44objects created by \function{sequence2ast()} faithfully simulate those
Guido van Rossum47478871996-08-21 14:32:37 +000045structures. Be aware that the values of the sequences which are
46considered ``correct'' will vary from one version of Python to another
47as the formal grammar for the language is revised. However,
48transporting code from one Python version to another as source text
49will always allow correct parse trees to be created in the target
50version, with the only restriction being that migrating to an older
51version of the interpreter will not support more recent language
52constructs. The parse trees are not typically compatible from one
53version to another, whereas source code has always been
54forward-compatible.
Guido van Rossum4b73a061995-10-11 17:30:04 +000055
Fred Drake88223901998-02-09 20:52:48 +000056Each element of the sequences returned by \function{ast2list()} or
57\function{ast2tuple()} has a simple form. Sequences representing
Guido van Rossum47478871996-08-21 14:32:37 +000058non-terminal elements in the grammar always have a length greater than
59one. The first element is an integer which identifies a production in
60the grammar. These integers are given symbolic names in the C header
Fred Drake4b7d5a41996-09-11 21:57:40 +000061file \file{Include/graminit.h} and the Python module
Fred Drake88223901998-02-09 20:52:48 +000062\module{symbol}. Each additional element of the sequence represents
Guido van Rossum47478871996-08-21 14:32:37 +000063a component of the production as recognized in the input string: these
64are always sequences which have the same form as the parent. An
65important aspect of this structure which should be noted is that
66keywords used to identify the parent node type, such as the keyword
Fred Drake88223901998-02-09 20:52:48 +000067\keyword{if} in an \constant{if_stmt}, are included in the node tree without
68any special treatment. For example, the \keyword{if} keyword is
Guido van Rossum4b73a061995-10-11 17:30:04 +000069represented by the tuple \code{(1, 'if')}, where \code{1} is the
Fred Drake4b7d5a41996-09-11 21:57:40 +000070numeric value associated with all \code{NAME} tokens, including
Guido van Rossum47478871996-08-21 14:32:37 +000071variable and function names defined by the user. In an alternate form
72returned when line number information is requested, the same token
73might be represented as \code{(1, 'if', 12)}, where the \code{12}
74represents the line number at which the terminal symbol was found.
Guido van Rossum4b73a061995-10-11 17:30:04 +000075
76Terminal elements are represented in much the same way, but without
77any child elements and the addition of the source text which was
Fred Drake88223901998-02-09 20:52:48 +000078identified. The example of the \keyword{if} keyword above is
Guido van Rossum4b73a061995-10-11 17:30:04 +000079representative. The various types of terminal symbols are defined in
Fred Drake4b7d5a41996-09-11 21:57:40 +000080the C header file \file{Include/token.h} and the Python module
Fred Drake88223901998-02-09 20:52:48 +000081\module{token}.
Guido van Rossum4b73a061995-10-11 17:30:04 +000082
Fred Drake4b7d5a41996-09-11 21:57:40 +000083The AST objects are not required to support the functionality of this
84module, but are provided for three purposes: to allow an application
85to amortize the cost of processing complex parse trees, to provide a
86parse tree representation which conserves memory space when compared
87to the Python list or tuple representation, and to ease the creation
88of additional modules in C which manipulate parse trees. A simple
89``wrapper'' class may be created in Python to hide the use of AST
Fred Drake88223901998-02-09 20:52:48 +000090objects.
Guido van Rossum4b73a061995-10-11 17:30:04 +000091
Fred Drake88223901998-02-09 20:52:48 +000092The \module{parser} module defines functions for a few distinct
Fred Drake4b7d5a41996-09-11 21:57:40 +000093purposes. The most important purposes are to create AST objects and
94to convert AST objects to other representations such as parse trees
95and compiled code objects, but there are also functions which serve to
96query the type of parse tree represented by an AST object.
Guido van Rossum4b73a061995-10-11 17:30:04 +000097
Fred Drake4b7d5a41996-09-11 21:57:40 +000098
99\subsection{Creating AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000100\label{Creating ASTs}
Fred Drake4b7d5a41996-09-11 21:57:40 +0000101
102AST objects may be created from source code or from a parse tree.
103When creating an AST object from source, different functions are used
104to create the \code{'eval'} and \code{'exec'} forms.
105
106\begin{funcdesc}{expr}{string}
Fred Drake88223901998-02-09 20:52:48 +0000107The \function{expr()} function parses the parameter \code{\var{string}}
108as if it were an input to \samp{compile(\var{string}, 'eval')}. If
Fred Drake4b7d5a41996-09-11 21:57:40 +0000109the parse succeeds, an AST object is created to hold the internal
110parse tree representation, otherwise an appropriate exception is
111thrown.
112\end{funcdesc}
113
114\begin{funcdesc}{suite}{string}
Fred Drake88223901998-02-09 20:52:48 +0000115The \function{suite()} function parses the parameter \code{\var{string}}
116as if it were an input to \samp{compile(\var{string}, 'exec')}. If
Fred Drake4b7d5a41996-09-11 21:57:40 +0000117the parse succeeds, an AST object is created to hold the internal
118parse tree representation, otherwise an appropriate exception is
119thrown.
120\end{funcdesc}
121
122\begin{funcdesc}{sequence2ast}{sequence}
123This function accepts a parse tree represented as a sequence and
124builds an internal representation if possible. If it can validate
125that the tree conforms to the Python grammar and all nodes are valid
126node types in the host version of Python, an AST object is created
127from the internal representation and returned to the called. If there
128is a problem creating the internal representation, or if the tree
Fred Drake88223901998-02-09 20:52:48 +0000129cannot be validated, a \exception{ParserError} exception is thrown. An AST
Fred Drake4b7d5a41996-09-11 21:57:40 +0000130object created this way should not be assumed to compile correctly;
131normal exceptions thrown by compilation may still be initiated when
Fred Drake88223901998-02-09 20:52:48 +0000132the AST object is passed to \function{compileast()}. This may indicate
133problems not related to syntax (such as a \exception{MemoryError}
Fred Drake4b7d5a41996-09-11 21:57:40 +0000134exception), but may also be due to constructs such as the result of
135parsing \code{del f(0)}, which escapes the Python parser but is
136checked by the bytecode compiler.
137
138Sequences representing terminal tokens may be represented as either
139two-element lists of the form \code{(1, 'name')} or as three-element
140lists of the form \code{(1, 'name', 56)}. If the third element is
141present, it is assumed to be a valid line number. The line number
142may be specified for any subset of the terminal symbols in the input
143tree.
144\end{funcdesc}
145
146\begin{funcdesc}{tuple2ast}{sequence}
Fred Drake88223901998-02-09 20:52:48 +0000147This is the same function as \function{sequence2ast()}. This entry point
Fred Drake4b7d5a41996-09-11 21:57:40 +0000148is maintained for backward compatibility.
149\end{funcdesc}
150
151
152\subsection{Converting AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000153\label{Converting ASTs}
Fred Drake4b7d5a41996-09-11 21:57:40 +0000154
155AST objects, regardless of the input used to create them, may be
156converted to parse trees represented as list- or tuple- trees, or may
157be compiled into executable code objects. Parse trees may be
158extracted with or without line numbering information.
159
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000160\begin{funcdesc}{ast2list}{ast\optional{, line_info}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000161This function accepts an AST object from the caller in
Guido van Rossum47478871996-08-21 14:32:37 +0000162\code{\var{ast}} and returns a Python list representing the
163equivelent parse tree. The resulting list representation can be used
Fred Drake4b7d5a41996-09-11 21:57:40 +0000164for inspection or the creation of a new parse tree in list form. This
165function does not fail so long as memory is available to build the
166list representation. If the parse tree will only be used for
Fred Drake88223901998-02-09 20:52:48 +0000167inspection, \function{ast2tuple()} should be used instead to reduce memory
Fred Drake4b7d5a41996-09-11 21:57:40 +0000168consumption and fragmentation. When the list representation is
169required, this function is significantly faster than retrieving a
170tuple representation and converting that to nested lists.
Guido van Rossum47478871996-08-21 14:32:37 +0000171
Fred Drake4b7d5a41996-09-11 21:57:40 +0000172If \code{\var{line_info}} is true, line number information will be
173included for all terminal tokens as a third element of the list
Fred Drake9abe64a1996-12-05 22:28:43 +0000174representing the token. Note that the line number provided specifies
Fred Drake88223901998-02-09 20:52:48 +0000175the line on which the token \emph{ends}. This information is
Fred Drake9abe64a1996-12-05 22:28:43 +0000176omitted if the flag is false or omitted.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000177\end{funcdesc}
178
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000179\begin{funcdesc}{ast2tuple}{ast\optional{, line_info}}
Guido van Rossum47478871996-08-21 14:32:37 +0000180This function accepts an AST object from the caller in
181\code{\var{ast}} and returns a Python tuple representing the
182equivelent parse tree. Other than returning a tuple instead of a
Fred Drake88223901998-02-09 20:52:48 +0000183list, this function is identical to \function{ast2list()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000184
Fred Drake4b7d5a41996-09-11 21:57:40 +0000185If \code{\var{line_info}} is true, line number information will be
186included for all terminal tokens as a third element of the list
187representing the token. This information is omitted if the flag is
188false or omitted.
Guido van Rossum47478871996-08-21 14:32:37 +0000189\end{funcdesc}
190
Fred Drakecce10901998-03-17 06:33:25 +0000191\begin{funcdesc}{compileast}{ast\optional{, filename\code{ = '<ast>'}}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000192The Python byte compiler can be invoked on an AST object to produce
193code objects which can be used as part of an \code{exec} statement or
Fred Drake88223901998-02-09 20:52:48 +0000194a call to the built-in \function{eval()}\bifuncindex{eval} function.
195This function provides the interface to the compiler, passing the
196internal parse tree from \code{\var{ast}} to the parser, using the
197source file name specified by the \code{\var{filename}} parameter.
198The default value supplied for \code{\var{filename}} indicates that
199the source was an AST object.
Guido van Rossum47478871996-08-21 14:32:37 +0000200
201Compiling an AST object may result in exceptions related to
Fred Drake88223901998-02-09 20:52:48 +0000202compilation; an example would be a \exception{SyntaxError} caused by the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000203parse tree for \code{del f(0)}: this statement is considered legal
Guido van Rossum47478871996-08-21 14:32:37 +0000204within the formal grammar for Python but is not a legal language
Fred Drake88223901998-02-09 20:52:48 +0000205construct. The \exception{SyntaxError} raised for this condition is
Guido van Rossum47478871996-08-21 14:32:37 +0000206actually generated by the Python byte-compiler normally, which is why
Fred Drake88223901998-02-09 20:52:48 +0000207it can be raised at this point by the \module{parser} module. Most
Guido van Rossum47478871996-08-21 14:32:37 +0000208causes of compilation failure can be diagnosed programmatically by
209inspection of the parse tree.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000210\end{funcdesc}
211
212
Fred Drake4b7d5a41996-09-11 21:57:40 +0000213\subsection{Queries on AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000214\label{Querying ASTs}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000215
Fred Drake4b7d5a41996-09-11 21:57:40 +0000216Two functions are provided which allow an application to determine if
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000217an AST was created as an expression or a suite. Neither of these
Fred Drake4b7d5a41996-09-11 21:57:40 +0000218functions can be used to determine if an AST was created from source
Fred Drake88223901998-02-09 20:52:48 +0000219code via \function{expr()} or \function{suite()} or from a parse tree
220via \function{sequence2ast()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000221
222\begin{funcdesc}{isexpr}{ast}
223When \code{\var{ast}} represents an \code{'eval'} form, this function
Fred Drake88223901998-02-09 20:52:48 +0000224returns true, otherwise it returns false. This is useful, since code
225objects normally cannot be queried for this information using existing
226built-in functions. Note that the code objects created by
227\function{compileast()} cannot be queried like this either, and are
228identical to those created by the built-in
229\function{compile()}\bifuncindex{compile} function.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000230\end{funcdesc}
231
232
233\begin{funcdesc}{issuite}{ast}
Fred Drake88223901998-02-09 20:52:48 +0000234This function mirrors \function{isexpr()} in that it reports whether an
Fred Drake4b7d5a41996-09-11 21:57:40 +0000235AST object represents an \code{'exec'} form, commonly known as a
236``suite.'' It is not safe to assume that this function is equivelent
Fred Drake88223901998-02-09 20:52:48 +0000237to \samp{not isexpr(\var{ast})}, as additional syntactic fragments may
Fred Drake4b7d5a41996-09-11 21:57:40 +0000238be supported in the future.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000239\end{funcdesc}
240
241
Guido van Rossum4b73a061995-10-11 17:30:04 +0000242\subsection{Exceptions and Error Handling}
Fred Draked67e12e1998-02-20 05:49:37 +0000243\label{AST Errors}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000244
245The parser module defines a single exception, but may also pass other
246built-in exceptions from other portions of the Python runtime
247environment. See each function for information about the exceptions
248it can raise.
249
250\begin{excdesc}{ParserError}
251Exception raised when a failure occurs within the parser module. This
252is generally produced for validation failures rather than the built in
Fred Drake88223901998-02-09 20:52:48 +0000253\exception{SyntaxError} thrown during normal parsing.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000254The exception argument is either a string describing the reason of the
Guido van Rossum47478871996-08-21 14:32:37 +0000255failure or a tuple containing a sequence causing the failure from a parse
Fred Drake88223901998-02-09 20:52:48 +0000256tree passed to \function{sequence2ast()} and an explanatory string. Calls to
257\function{sequence2ast()} need to be able to handle either type of exception,
Guido van Rossum4b73a061995-10-11 17:30:04 +0000258while calls to other functions in the module will only need to be
259aware of the simple string values.
260\end{excdesc}
261
Fred Drake88223901998-02-09 20:52:48 +0000262Note that the functions \function{compileast()}, \function{expr()}, and
263\function{suite()} may throw exceptions which are normally thrown by the
Guido van Rossum4b73a061995-10-11 17:30:04 +0000264parsing and compilation process. These include the built in
Fred Drake88223901998-02-09 20:52:48 +0000265exceptions \exception{MemoryError}, \exception{OverflowError},
266\exception{SyntaxError}, and \exception{SystemError}. In these cases, these
Guido van Rossum4b73a061995-10-11 17:30:04 +0000267exceptions carry all the meaning normally associated with them. Refer
268to the descriptions of each function for detailed information.
269
Guido van Rossum4b73a061995-10-11 17:30:04 +0000270
Guido van Rossum47478871996-08-21 14:32:37 +0000271\subsection{AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000272\label{AST Objects}
Guido van Rossum47478871996-08-21 14:32:37 +0000273
Fred Drakecc444e31998-03-08 06:47:24 +0000274AST objects returned by \function{expr()}, \function{suite()} and
Fred Drake88223901998-02-09 20:52:48 +0000275\function{sequence2ast()} have no methods of their own.
Fred Drakecc444e31998-03-08 06:47:24 +0000276
Fred Drakeaf370ea1998-04-05 20:23:02 +0000277Ordered and equality comparisons are supported between AST objects.
Fred Drakec4f1ca11998-04-13 16:27:27 +0000278Pickling of AST objects (using the \module{pickle} module) is also
279supported.
Fred Drakeaf370ea1998-04-05 20:23:02 +0000280
Fred Drakecc444e31998-03-08 06:47:24 +0000281\begin{datadesc}{ASTType}
282The type of the objects returned by \function{expr()},
283\function{suite()} and \function{sequence2ast()}.
Fred Drakecc444e31998-03-08 06:47:24 +0000284\end{datadesc}
Guido van Rossum47478871996-08-21 14:32:37 +0000285
286
Fred Drake916d8f81998-04-13 18:46:16 +0000287AST objects have the following methods:
288
289
290\begin{methoddesc}[AST]{compile}{\optional{filename}}
291Same as \code{compileast(\var{ast}, \var{filename})}.
292\end{methoddesc}
293
294\begin{methoddesc}[AST]{isexpr}{}
295Same as \code{isexpr(\var{ast})}.
296\end{methoddesc}
297
298\begin{methoddesc}[AST]{issuite}{}
299Same as \code{issuite(\var{ast})}.
300\end{methoddesc}
301
302\begin{methoddesc}[AST]{tolist}{\optional{line_info}}
303Same as \code{ast2list(\var{ast}, \var{line_info})}.
304\end{methoddesc}
305
306\begin{methoddesc}[AST]{totuple}{\optional{line_info}}
307Same as \code{ast2tuple(\var{ast}, \var{line_info})}.
308\end{methoddesc}
309
310
Guido van Rossum8206fb91996-08-26 00:33:29 +0000311\subsection{Examples}
Fred Drake4b3f0311996-12-13 22:04:31 +0000312\nodename{AST Examples}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000313
Guido van Rossum47478871996-08-21 14:32:37 +0000314The parser modules allows operations to be performed on the parse tree
315of Python source code before the bytecode is generated, and provides
Fred Drake4b7d5a41996-09-11 21:57:40 +0000316for inspection of the parse tree for information gathering purposes.
317Two examples are presented. The simple example demonstrates emulation
Fred Drake88223901998-02-09 20:52:48 +0000318of the \function{compile()}\bifuncindex{compile} built-in function and
319the complex example shows the use of a parse tree for information
320discovery.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000321
Fred Drakeaf370ea1998-04-05 20:23:02 +0000322\subsubsection{Emulation of \function{compile()}}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000323
324While many useful operations may take place between parsing and
Guido van Rossum47478871996-08-21 14:32:37 +0000325bytecode generation, the simplest operation is to do nothing. For
Fred Drake88223901998-02-09 20:52:48 +0000326this purpose, using the \module{parser} module to produce an
Guido van Rossum47478871996-08-21 14:32:37 +0000327intermediate data structure is equivelent to the code
328
Fred Drake19479911998-02-13 06:58:54 +0000329\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000330>>> code = compile('a + 5', 'eval')
331>>> a = 5
332>>> eval(code)
33310
Fred Drake19479911998-02-13 06:58:54 +0000334\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000335
Fred Drake88223901998-02-09 20:52:48 +0000336The equivelent operation using the \module{parser} module is somewhat
Guido van Rossum47478871996-08-21 14:32:37 +0000337longer, and allows the intermediate internal parse tree to be retained
338as an AST object:
Guido van Rossum4b73a061995-10-11 17:30:04 +0000339
Fred Drake19479911998-02-13 06:58:54 +0000340\begin{verbatim}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000341>>> import parser
342>>> ast = parser.expr('a + 5')
343>>> code = parser.compileast(ast)
344>>> a = 5
345>>> eval(code)
34610
Fred Drake19479911998-02-13 06:58:54 +0000347\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000348
Guido van Rossum8206fb91996-08-26 00:33:29 +0000349An application which needs both AST and code objects can package this
350code into readily available functions:
351
Fred Drake19479911998-02-13 06:58:54 +0000352\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000353import parser
354
355def load_suite(source_string):
356 ast = parser.suite(source_string)
357 code = parser.compileast(ast)
358 return ast, code
359
360def load_expression(source_string):
361 ast = parser.expr(source_string)
362 code = parser.compileast(ast)
363 return ast, code
Fred Drake19479911998-02-13 06:58:54 +0000364\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000365
Guido van Rossum8206fb91996-08-26 00:33:29 +0000366\subsubsection{Information Discovery}
367
Fred Drake4b7d5a41996-09-11 21:57:40 +0000368Some applications benefit from direct access to the parse tree. The
369remainder of this section demonstrates how the parse tree provides
370access to module documentation defined in docstrings without requiring
371that the code being examined be loaded into a running interpreter via
Fred Drake88223901998-02-09 20:52:48 +0000372\keyword{import}. This can be very useful for performing analyses of
Fred Drake4b7d5a41996-09-11 21:57:40 +0000373untrusted code.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000374
Guido van Rossum47478871996-08-21 14:32:37 +0000375Generally, the example will demonstrate how the parse tree may be
376traversed to distill interesting information. Two functions and a set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000377of classes are developed which provide programmatic access to high
Guido van Rossum47478871996-08-21 14:32:37 +0000378level function and class definitions provided by a module. The
379classes extract information from the parse tree and provide access to
380the information at a useful semantic level, one function provides a
381simple low-level pattern matching capability, and the other function
382defines a high-level interface to the classes by handling file
383operations on behalf of the caller. All source files mentioned here
384which are not part of the Python installation are located in the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000385\file{Demo/parser/} directory of the distribution.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000386
Guido van Rossum8206fb91996-08-26 00:33:29 +0000387The dynamic nature of Python allows the programmer a great deal of
388flexibility, but most modules need only a limited measure of this when
389defining classes, functions, and methods. In this example, the only
390definitions that will be considered are those which are defined in the
Fred Drake88223901998-02-09 20:52:48 +0000391top level of their context, e.g., a function defined by a \keyword{def}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000392statement at column zero of a module, but not a function defined
Fred Drake4b7d5a41996-09-11 21:57:40 +0000393within a branch of an \code{if} ... \code{else} construct, though
Guido van Rossum8206fb91996-08-26 00:33:29 +0000394there are some good reasons for doing so in some situations. Nesting
395of definitions will be handled by the code developed in the example.
396
Guido van Rossum47478871996-08-21 14:32:37 +0000397To construct the upper-level extraction methods, we need to know what
398the parse tree structure looks like and how much of it we actually
Fred Drake4b7d5a41996-09-11 21:57:40 +0000399need to be concerned about. Python uses a moderately deep parse tree
Guido van Rossum47478871996-08-21 14:32:37 +0000400so there are a large number of intermediate nodes. It is important to
401read and understand the formal grammar used by Python. This is
402specified in the file \file{Grammar/Grammar} in the distribution.
403Consider the simplest case of interest when searching for docstrings:
Guido van Rossum8206fb91996-08-26 00:33:29 +0000404a module consisting of a docstring and nothing else. (See file
405\file{docstring.py}.)
Guido van Rossum4b73a061995-10-11 17:30:04 +0000406
Fred Drake19479911998-02-13 06:58:54 +0000407\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000408"""Some documentation.
409"""
Fred Drake19479911998-02-13 06:58:54 +0000410\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000411
Guido van Rossum47478871996-08-21 14:32:37 +0000412Using the interpreter to take a look at the parse tree, we find a
413bewildering mass of numbers and parentheses, with the documentation
Fred Drake4b7d5a41996-09-11 21:57:40 +0000414buried deep in nested tuples.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000415
Fred Drake19479911998-02-13 06:58:54 +0000416\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000417>>> import parser
418>>> import pprint
419>>> ast = parser.suite(open('docstring.py').read())
420>>> tup = parser.ast2tuple(ast)
421>>> pprint.pprint(tup)
422(257,
423 (264,
424 (265,
425 (266,
426 (267,
427 (307,
428 (287,
429 (288,
430 (289,
431 (290,
432 (292,
433 (293,
434 (294,
435 (295,
436 (296,
437 (297,
438 (298,
439 (299,
440 (300, (3, '"""Some documentation.\012"""'))))))))))))))))),
441 (4, ''))),
442 (4, ''),
443 (0, ''))
Fred Drake19479911998-02-13 06:58:54 +0000444\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000445
Guido van Rossum47478871996-08-21 14:32:37 +0000446The numbers at the first element of each node in the tree are the node
447types; they map directly to terminal and non-terminal symbols in the
448grammar. Unfortunately, they are represented as integers in the
449internal representation, and the Python structures generated do not
Fred Drake88223901998-02-09 20:52:48 +0000450change that. However, the \module{symbol} and \module{token} modules
Guido van Rossum47478871996-08-21 14:32:37 +0000451provide symbolic names for the node types and dictionaries which map
452from the integers to the symbolic names for the node types.
453
454In the output presented above, the outermost tuple contains four
455elements: the integer \code{257} and three additional tuples. Node
Fred Drake88223901998-02-09 20:52:48 +0000456type \code{257} has the symbolic name \constant{file_input}. Each of
Guido van Rossum47478871996-08-21 14:32:37 +0000457these inner tuples contains an integer as the first element; these
458integers, \code{264}, \code{4}, and \code{0}, represent the node types
Fred Drake88223901998-02-09 20:52:48 +0000459\constant{stmt}, \constant{NEWLINE}, and \constant{ENDMARKER},
460respectively.
Guido van Rossum47478871996-08-21 14:32:37 +0000461Note that these values may change depending on the version of Python
462you are using; consult \file{symbol.py} and \file{token.py} for
463details of the mapping. It should be fairly clear that the outermost
464node is related primarily to the input source rather than the contents
Fred Drake88223901998-02-09 20:52:48 +0000465of the file, and may be disregarded for the moment. The \constant{stmt}
Guido van Rossum47478871996-08-21 14:32:37 +0000466node is much more interesting. In particular, all docstrings are
467found in subtrees which are formed exactly as this node is formed,
468with the only difference being the string itself. The association
469between the docstring in a similar tree and the defined entity (class,
470function, or module) which it describes is given by the position of
471the docstring subtree within the tree defining the described
472structure.
473
474By replacing the actual docstring with something to signify a variable
Fred Drake4b7d5a41996-09-11 21:57:40 +0000475component of the tree, we allow a simple pattern matching approach to
476check any given subtree for equivelence to the general pattern for
477docstrings. Since the example demonstrates information extraction, we
478can safely require that the tree be in tuple form rather than list
479form, allowing a simple variable representation to be
480\code{['variable_name']}. A simple recursive function can implement
Guido van Rossum47478871996-08-21 14:32:37 +0000481the pattern matching, returning a boolean and a dictionary of variable
Guido van Rossum8206fb91996-08-26 00:33:29 +0000482name to value mappings. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000483
Fred Drake19479911998-02-13 06:58:54 +0000484\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000485from types import ListType, TupleType
486
487def match(pattern, data, vars=None):
488 if vars is None:
489 vars = {}
490 if type(pattern) is ListType:
491 vars[pattern[0]] = data
492 return 1, vars
493 if type(pattern) is not TupleType:
494 return (pattern == data), vars
495 if len(data) != len(pattern):
496 return 0, vars
497 for pattern, data in map(None, pattern, data):
498 same, vars = match(pattern, data, vars)
499 if not same:
500 break
501 return same, vars
Fred Drake19479911998-02-13 06:58:54 +0000502\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000503
Fred Drake4b7d5a41996-09-11 21:57:40 +0000504Using this simple representation for syntactic variables and the symbolic
Guido van Rossum8206fb91996-08-26 00:33:29 +0000505node types, the pattern for the candidate docstring subtrees becomes
506fairly readable. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000507
Fred Drake19479911998-02-13 06:58:54 +0000508\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000509import symbol
510import token
511
512DOCSTRING_STMT_PATTERN = (
513 symbol.stmt,
514 (symbol.simple_stmt,
515 (symbol.small_stmt,
516 (symbol.expr_stmt,
517 (symbol.testlist,
518 (symbol.test,
519 (symbol.and_test,
520 (symbol.not_test,
521 (symbol.comparison,
522 (symbol.expr,
523 (symbol.xor_expr,
524 (symbol.and_expr,
525 (symbol.shift_expr,
526 (symbol.arith_expr,
527 (symbol.term,
528 (symbol.factor,
529 (symbol.power,
530 (symbol.atom,
531 (token.STRING, ['docstring'])
532 )))))))))))))))),
533 (token.NEWLINE, '')
534 ))
Fred Drake19479911998-02-13 06:58:54 +0000535\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000536
Fred Drake88223901998-02-09 20:52:48 +0000537Using the \function{match()} function with this pattern, extracting the
Guido van Rossum47478871996-08-21 14:32:37 +0000538module docstring from the parse tree created previously is easy:
539
Fred Drake19479911998-02-13 06:58:54 +0000540\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000541>>> found, vars = match(DOCSTRING_STMT_PATTERN, tup[1])
542>>> found
5431
544>>> vars
545{'docstring': '"""Some documentation.\012"""'}
Fred Drake19479911998-02-13 06:58:54 +0000546\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000547
Guido van Rossum47478871996-08-21 14:32:37 +0000548Once specific data can be extracted from a location where it is
549expected, the question of where information can be expected
550needs to be answered. When dealing with docstrings, the answer is
Fred Drake88223901998-02-09 20:52:48 +0000551fairly simple: the docstring is the first \constant{stmt} node in a code
552block (\constant{file_input} or \constant{suite} node types). A module
553consists of a single \constant{file_input} node, and class and function
554definitions each contain exactly one \constant{suite} node. Classes and
Guido van Rossum47478871996-08-21 14:32:37 +0000555functions are readily identified as subtrees of code block nodes which
556start with \code{(stmt, (compound_stmt, (classdef, ...} or
557\code{(stmt, (compound_stmt, (funcdef, ...}. Note that these subtrees
Fred Drake88223901998-02-09 20:52:48 +0000558cannot be matched by \function{match()} since it does not support multiple
Guido van Rossum47478871996-08-21 14:32:37 +0000559sibling nodes to match without regard to number. A more elaborate
560matching function could be used to overcome this limitation, but this
561is sufficient for the example.
562
Guido van Rossum8206fb91996-08-26 00:33:29 +0000563Given the ability to determine whether a statement might be a
564docstring and extract the actual string from the statement, some work
565needs to be performed to walk the parse tree for an entire module and
566extract information about the names defined in each context of the
567module and associate any docstrings with the names. The code to
568perform this work is not complicated, but bears some explanation.
569
570The public interface to the classes is straightforward and should
571probably be somewhat more flexible. Each ``major'' block of the
572module is described by an object providing several methods for inquiry
573and a constructor which accepts at least the subtree of the complete
Fred Drakeb0df5671998-02-18 15:59:13 +0000574parse tree which it represents. The \class{ModuleInfo} constructor
575accepts an optional \var{name} parameter since it cannot
Guido van Rossum8206fb91996-08-26 00:33:29 +0000576otherwise determine the name of the module.
577
Fred Drake88223901998-02-09 20:52:48 +0000578The public classes include \class{ClassInfo}, \class{FunctionInfo},
579and \class{ModuleInfo}. All objects provide the
580methods \method{get_name()}, \method{get_docstring()},
581\method{get_class_names()}, and \method{get_class_info()}. The
582\class{ClassInfo} objects support \method{get_method_names()} and
583\method{get_method_info()} while the other classes provide
584\method{get_function_names()} and \method{get_function_info()}.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000585
586Within each of the forms of code block that the public classes
587represent, most of the required information is in the same form and is
Fred Drake4b7d5a41996-09-11 21:57:40 +0000588accessed in the same way, with classes having the distinction that
Guido van Rossum8206fb91996-08-26 00:33:29 +0000589functions defined at the top level are referred to as ``methods.''
590Since the difference in nomenclature reflects a real semantic
Fred Drake4b7d5a41996-09-11 21:57:40 +0000591distinction from functions defined outside of a class, the
592implementation needs to maintain the distinction.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000593Hence, most of the functionality of the public classes can be
Fred Drake88223901998-02-09 20:52:48 +0000594implemented in a common base class, \class{SuiteInfoBase}, with the
Guido van Rossum8206fb91996-08-26 00:33:29 +0000595accessors for function and method information provided elsewhere.
596Note that there is only one class which represents function and method
Fred Drake88223901998-02-09 20:52:48 +0000597information; this parallels the use of the \keyword{def} statement to
Fred Drake4b7d5a41996-09-11 21:57:40 +0000598define both types of elements.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000599
Fred Drake88223901998-02-09 20:52:48 +0000600Most of the accessor functions are declared in \class{SuiteInfoBase}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000601and do not need to be overriden by subclasses. More importantly, the
602extraction of most information from a parse tree is handled through a
Fred Drake88223901998-02-09 20:52:48 +0000603method called by the \class{SuiteInfoBase} constructor. The example
Guido van Rossum8206fb91996-08-26 00:33:29 +0000604code for most of the classes is clear when read alongside the formal
605grammar, but the method which recursively creates new information
606objects requires further examination. Here is the relevant part of
Fred Drake88223901998-02-09 20:52:48 +0000607the \class{SuiteInfoBase} definition from \file{example.py}:
Guido van Rossum8206fb91996-08-26 00:33:29 +0000608
Fred Drake19479911998-02-13 06:58:54 +0000609\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000610class SuiteInfoBase:
611 _docstring = ''
612 _name = ''
613
614 def __init__(self, tree = None):
615 self._class_info = {}
616 self._function_info = {}
617 if tree:
618 self._extract_info(tree)
619
620 def _extract_info(self, tree):
621 # extract docstring
622 if len(tree) == 2:
623 found, vars = match(DOCSTRING_STMT_PATTERN[1], tree[1])
624 else:
625 found, vars = match(DOCSTRING_STMT_PATTERN, tree[3])
626 if found:
627 self._docstring = eval(vars['docstring'])
628 # discover inner definitions
629 for node in tree[1:]:
630 found, vars = match(COMPOUND_STMT_PATTERN, node)
631 if found:
632 cstmt = vars['compound']
633 if cstmt[0] == symbol.funcdef:
634 name = cstmt[2][1]
635 self._function_info[name] = FunctionInfo(cstmt)
636 elif cstmt[0] == symbol.classdef:
637 name = cstmt[2][1]
638 self._class_info[name] = ClassInfo(cstmt)
Fred Drake19479911998-02-13 06:58:54 +0000639\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000640
Guido van Rossum8206fb91996-08-26 00:33:29 +0000641After initializing some internal state, the constructor calls the
Fred Drake88223901998-02-09 20:52:48 +0000642\method{_extract_info()} method. This method performs the bulk of the
Guido van Rossum8206fb91996-08-26 00:33:29 +0000643information extraction which takes place in the entire example. The
644extraction has two distinct phases: the location of the docstring for
645the parse tree passed in, and the discovery of additional definitions
646within the code block represented by the parse tree.
647
Fred Drake88223901998-02-09 20:52:48 +0000648The initial \keyword{if} test determines whether the nested suite is of
Guido van Rossum8206fb91996-08-26 00:33:29 +0000649the ``short form'' or the ``long form.'' The short form is used when
650the code block is on the same line as the definition of the code
651block, as in
652
Fred Drakebbe60681998-01-09 22:24:14 +0000653\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000654def square(x): "Square an argument."; return x ** 2
Fred Drakebbe60681998-01-09 22:24:14 +0000655\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000656
Guido van Rossum8206fb91996-08-26 00:33:29 +0000657while the long form uses an indented block and allows nested
658definitions:
659
Fred Drake19479911998-02-13 06:58:54 +0000660\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000661def make_power(exp):
662 "Make a function that raises an argument to the exponent `exp'."
663 def raiser(x, y=exp):
664 return x ** y
665 return raiser
Fred Drake19479911998-02-13 06:58:54 +0000666\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000667
Guido van Rossum8206fb91996-08-26 00:33:29 +0000668When the short form is used, the code block may contain a docstring as
Fred Drake88223901998-02-09 20:52:48 +0000669the first, and possibly only, \constant{small_stmt} element. The
Guido van Rossum8206fb91996-08-26 00:33:29 +0000670extraction of such a docstring is slightly different and requires only
671a portion of the complete pattern used in the more common case. As
Fred Drake4b7d5a41996-09-11 21:57:40 +0000672implemented, the docstring will only be found if there is only
Fred Drake88223901998-02-09 20:52:48 +0000673one \constant{small_stmt} node in the \constant{simple_stmt} node.
674Since most functions and methods which use the short form do not
675provide a docstring, this may be considered sufficient. The
676extraction of the docstring proceeds using the \function{match()} function
677as described above, and the value of the docstring is stored as an
678attribute of the \class{SuiteInfoBase} object.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000679
Fred Drake4b7d5a41996-09-11 21:57:40 +0000680After docstring extraction, a simple definition discovery
Fred Drake88223901998-02-09 20:52:48 +0000681algorithm operates on the \constant{stmt} nodes of the
682\constant{suite} node. The special case of the short form is not
683tested; since there are no \constant{stmt} nodes in the short form,
684the algorithm will silently skip the single \constant{simple_stmt}
685node and correctly not discover any nested definitions.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000686
Fred Drake4b7d5a41996-09-11 21:57:40 +0000687Each statement in the code block is categorized as
688a class definition, function or method definition, or
Guido van Rossum8206fb91996-08-26 00:33:29 +0000689something else. For the definition statements, the name of the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000690element defined is extracted and a representation object
Guido van Rossum8206fb91996-08-26 00:33:29 +0000691appropriate to the definition is created with the defining subtree
692passed as an argument to the constructor. The repesentation objects
693are stored in instance variables and may be retrieved by name using
694the appropriate accessor methods.
695
696The public classes provide any accessors required which are more
Fred Drake88223901998-02-09 20:52:48 +0000697specific than those provided by the \class{SuiteInfoBase} class, but
Guido van Rossum8206fb91996-08-26 00:33:29 +0000698the real extraction algorithm remains common to all forms of code
699blocks. A high-level function can be used to extract the complete set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000700of information from a source file. (See file \file{example.py}.)
Guido van Rossum8206fb91996-08-26 00:33:29 +0000701
Fred Drake19479911998-02-13 06:58:54 +0000702\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000703def get_docs(fileName):
704 source = open(fileName).read()
705 import os
706 basename = os.path.basename(os.path.splitext(fileName)[0])
707 import parser
708 ast = parser.suite(source)
709 tup = parser.ast2tuple(ast)
710 return ModuleInfo(tup, basename)
Fred Drake19479911998-02-13 06:58:54 +0000711\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000712
Guido van Rossum8206fb91996-08-26 00:33:29 +0000713This provides an easy-to-use interface to the documentation of a
714module. If information is required which is not extracted by the code
715of this example, the code may be extended at clearly defined points to
716provide additional capabilities.
Guido van Rossum47478871996-08-21 14:32:37 +0000717
Fred Drakebbe60681998-01-09 22:24:14 +0000718\begin{seealso}
719
Fred Drake45c634e1998-04-09 15:44:58 +0000720\seemodule{symbol}{useful constants representing internal nodes of the
721parse tree}
Fred Drakebbe60681998-01-09 22:24:14 +0000722
Fred Drake45c634e1998-04-09 15:44:58 +0000723\seemodule{token}{useful constants representing leaf nodes of the
724parse tree and functions for testing node values}
Fred Drakebbe60681998-01-09 22:24:14 +0000725
726\end{seealso}