blob: 6759a9fc78ee7262b2f6ce2d8a27b5e2c5c10fbb [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 Drakebbe60681998-01-09 22:24:14 +000013\label{module-parser}
Guido van Rossum4b73a061995-10-11 17:30:04 +000014\bimodindex{parser}
Fred Drake88223901998-02-09 20:52:48 +000015\index{parsing!Python source code}
Guido van Rossum4b73a061995-10-11 17:30:04 +000016
Fred Drake88223901998-02-09 20:52:48 +000017The \module{parser} module provides an interface to Python's internal
Guido van Rossum4b73a061995-10-11 17:30:04 +000018parser and byte-code compiler. The primary purpose for this interface
19is to allow Python code to edit the parse tree of a Python expression
Fred Drake4b7d5a41996-09-11 21:57:40 +000020and create executable code from this. This is better than trying
21to parse and modify an arbitrary Python code fragment as a string
22because parsing is performed in a manner identical to the code
23forming the application. It is also faster.
Guido van Rossum4b73a061995-10-11 17:30:04 +000024
Fred Drake5bd7fcc1998-04-03 05:31:45 +000025The \module{parser} module was written and documented by Fred
26L. Drake, Jr. (\email{fdrake@acm.org}).%
27\index{Drake, Fred L., Jr.}
28
Guido van Rossum4b73a061995-10-11 17:30:04 +000029There are a few things to note about this module which are important
30to making use of the data structures created. This is not a tutorial
Fred Drake4b7d5a41996-09-11 21:57:40 +000031on editing the parse trees for Python code, but some examples of using
Fred Drake88223901998-02-09 20:52:48 +000032the \module{parser} module are presented.
Guido van Rossum4b73a061995-10-11 17:30:04 +000033
34Most importantly, a good understanding of the Python grammar processed
35by the internal parser is required. For full information on the
Fred Drake88223901998-02-09 20:52:48 +000036language syntax, refer to the \emph{Python Language Reference}. The
37parser itself is created from a grammar specification defined in the file
Fred Drake4b7d5a41996-09-11 21:57:40 +000038\file{Grammar/Grammar} in the standard Python distribution. The parse
Fred Drakecc444e31998-03-08 06:47:24 +000039trees stored in the AST objects created by this module are the
Guido van Rossum4b73a061995-10-11 17:30:04 +000040actual output from the internal parser when created by the
Fred Drake88223901998-02-09 20:52:48 +000041\function{expr()} or \function{suite()} functions, described below. The AST
42objects created by \function{sequence2ast()} faithfully simulate those
Guido van Rossum47478871996-08-21 14:32:37 +000043structures. Be aware that the values of the sequences which are
44considered ``correct'' will vary from one version of Python to another
45as the formal grammar for the language is revised. However,
46transporting code from one Python version to another as source text
47will always allow correct parse trees to be created in the target
48version, with the only restriction being that migrating to an older
49version of the interpreter will not support more recent language
50constructs. The parse trees are not typically compatible from one
51version to another, whereas source code has always been
52forward-compatible.
Guido van Rossum4b73a061995-10-11 17:30:04 +000053
Fred Drake88223901998-02-09 20:52:48 +000054Each element of the sequences returned by \function{ast2list()} or
55\function{ast2tuple()} has a simple form. Sequences representing
Guido van Rossum47478871996-08-21 14:32:37 +000056non-terminal elements in the grammar always have a length greater than
57one. The first element is an integer which identifies a production in
58the grammar. These integers are given symbolic names in the C header
Fred Drake4b7d5a41996-09-11 21:57:40 +000059file \file{Include/graminit.h} and the Python module
Fred Drake88223901998-02-09 20:52:48 +000060\module{symbol}. Each additional element of the sequence represents
Guido van Rossum47478871996-08-21 14:32:37 +000061a component of the production as recognized in the input string: these
62are always sequences which have the same form as the parent. An
63important aspect of this structure which should be noted is that
64keywords used to identify the parent node type, such as the keyword
Fred Drake88223901998-02-09 20:52:48 +000065\keyword{if} in an \constant{if_stmt}, are included in the node tree without
66any special treatment. For example, the \keyword{if} keyword is
Guido van Rossum4b73a061995-10-11 17:30:04 +000067represented by the tuple \code{(1, 'if')}, where \code{1} is the
Fred Drake4b7d5a41996-09-11 21:57:40 +000068numeric value associated with all \code{NAME} tokens, including
Guido van Rossum47478871996-08-21 14:32:37 +000069variable and function names defined by the user. In an alternate form
70returned when line number information is requested, the same token
71might be represented as \code{(1, 'if', 12)}, where the \code{12}
72represents the line number at which the terminal symbol was found.
Guido van Rossum4b73a061995-10-11 17:30:04 +000073
74Terminal elements are represented in much the same way, but without
75any child elements and the addition of the source text which was
Fred Drake88223901998-02-09 20:52:48 +000076identified. The example of the \keyword{if} keyword above is
Guido van Rossum4b73a061995-10-11 17:30:04 +000077representative. The various types of terminal symbols are defined in
Fred Drake4b7d5a41996-09-11 21:57:40 +000078the C header file \file{Include/token.h} and the Python module
Fred Drake88223901998-02-09 20:52:48 +000079\module{token}.
Guido van Rossum4b73a061995-10-11 17:30:04 +000080
Fred Drake4b7d5a41996-09-11 21:57:40 +000081The AST objects are not required to support the functionality of this
82module, but are provided for three purposes: to allow an application
83to amortize the cost of processing complex parse trees, to provide a
84parse tree representation which conserves memory space when compared
85to the Python list or tuple representation, and to ease the creation
86of additional modules in C which manipulate parse trees. A simple
87``wrapper'' class may be created in Python to hide the use of AST
Fred Drake88223901998-02-09 20:52:48 +000088objects.
Guido van Rossum4b73a061995-10-11 17:30:04 +000089
Fred Drake88223901998-02-09 20:52:48 +000090The \module{parser} module defines functions for a few distinct
Fred Drake4b7d5a41996-09-11 21:57:40 +000091purposes. The most important purposes are to create AST objects and
92to convert AST objects to other representations such as parse trees
93and compiled code objects, but there are also functions which serve to
94query the type of parse tree represented by an AST object.
Guido van Rossum4b73a061995-10-11 17:30:04 +000095
Fred Drake4b7d5a41996-09-11 21:57:40 +000096
97\subsection{Creating AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +000098\label{Creating ASTs}
Fred Drake4b7d5a41996-09-11 21:57:40 +000099
100AST objects may be created from source code or from a parse tree.
101When creating an AST object from source, different functions are used
102to create the \code{'eval'} and \code{'exec'} forms.
103
104\begin{funcdesc}{expr}{string}
Fred Drake88223901998-02-09 20:52:48 +0000105The \function{expr()} function parses the parameter \code{\var{string}}
106as if it were an input to \samp{compile(\var{string}, 'eval')}. If
Fred Drake4b7d5a41996-09-11 21:57:40 +0000107the parse succeeds, an AST object is created to hold the internal
108parse tree representation, otherwise an appropriate exception is
109thrown.
110\end{funcdesc}
111
112\begin{funcdesc}{suite}{string}
Fred Drake88223901998-02-09 20:52:48 +0000113The \function{suite()} function parses the parameter \code{\var{string}}
114as if it were an input to \samp{compile(\var{string}, 'exec')}. If
Fred Drake4b7d5a41996-09-11 21:57:40 +0000115the parse succeeds, an AST object is created to hold the internal
116parse tree representation, otherwise an appropriate exception is
117thrown.
118\end{funcdesc}
119
120\begin{funcdesc}{sequence2ast}{sequence}
121This function accepts a parse tree represented as a sequence and
122builds an internal representation if possible. If it can validate
123that the tree conforms to the Python grammar and all nodes are valid
124node types in the host version of Python, an AST object is created
125from the internal representation and returned to the called. If there
126is a problem creating the internal representation, or if the tree
Fred Drake88223901998-02-09 20:52:48 +0000127cannot be validated, a \exception{ParserError} exception is thrown. An AST
Fred Drake4b7d5a41996-09-11 21:57:40 +0000128object created this way should not be assumed to compile correctly;
129normal exceptions thrown by compilation may still be initiated when
Fred Drake88223901998-02-09 20:52:48 +0000130the AST object is passed to \function{compileast()}. This may indicate
131problems not related to syntax (such as a \exception{MemoryError}
Fred Drake4b7d5a41996-09-11 21:57:40 +0000132exception), but may also be due to constructs such as the result of
133parsing \code{del f(0)}, which escapes the Python parser but is
134checked by the bytecode compiler.
135
136Sequences representing terminal tokens may be represented as either
137two-element lists of the form \code{(1, 'name')} or as three-element
138lists of the form \code{(1, 'name', 56)}. If the third element is
139present, it is assumed to be a valid line number. The line number
140may be specified for any subset of the terminal symbols in the input
141tree.
142\end{funcdesc}
143
144\begin{funcdesc}{tuple2ast}{sequence}
Fred Drake88223901998-02-09 20:52:48 +0000145This is the same function as \function{sequence2ast()}. This entry point
Fred Drake4b7d5a41996-09-11 21:57:40 +0000146is maintained for backward compatibility.
147\end{funcdesc}
148
149
150\subsection{Converting AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000151\label{Converting ASTs}
Fred Drake4b7d5a41996-09-11 21:57:40 +0000152
153AST objects, regardless of the input used to create them, may be
154converted to parse trees represented as list- or tuple- trees, or may
155be compiled into executable code objects. Parse trees may be
156extracted with or without line numbering information.
157
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000158\begin{funcdesc}{ast2list}{ast\optional{, line_info}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000159This function accepts an AST object from the caller in
Guido van Rossum47478871996-08-21 14:32:37 +0000160\code{\var{ast}} and returns a Python list representing the
161equivelent parse tree. The resulting list representation can be used
Fred Drake4b7d5a41996-09-11 21:57:40 +0000162for inspection or the creation of a new parse tree in list form. This
163function does not fail so long as memory is available to build the
164list representation. If the parse tree will only be used for
Fred Drake88223901998-02-09 20:52:48 +0000165inspection, \function{ast2tuple()} should be used instead to reduce memory
Fred Drake4b7d5a41996-09-11 21:57:40 +0000166consumption and fragmentation. When the list representation is
167required, this function is significantly faster than retrieving a
168tuple representation and converting that to nested lists.
Guido van Rossum47478871996-08-21 14:32:37 +0000169
Fred Drake4b7d5a41996-09-11 21:57:40 +0000170If \code{\var{line_info}} is true, line number information will be
171included for all terminal tokens as a third element of the list
Fred Drake9abe64a1996-12-05 22:28:43 +0000172representing the token. Note that the line number provided specifies
Fred Drake88223901998-02-09 20:52:48 +0000173the line on which the token \emph{ends}. This information is
Fred Drake9abe64a1996-12-05 22:28:43 +0000174omitted if the flag is false or omitted.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000175\end{funcdesc}
176
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000177\begin{funcdesc}{ast2tuple}{ast\optional{, line_info}}
Guido van Rossum47478871996-08-21 14:32:37 +0000178This function accepts an AST object from the caller in
179\code{\var{ast}} and returns a Python tuple representing the
180equivelent parse tree. Other than returning a tuple instead of a
Fred Drake88223901998-02-09 20:52:48 +0000181list, this function is identical to \function{ast2list()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000182
Fred Drake4b7d5a41996-09-11 21:57:40 +0000183If \code{\var{line_info}} is true, line number information will be
184included for all terminal tokens as a third element of the list
185representing the token. This information is omitted if the flag is
186false or omitted.
Guido van Rossum47478871996-08-21 14:32:37 +0000187\end{funcdesc}
188
Fred Drakecce10901998-03-17 06:33:25 +0000189\begin{funcdesc}{compileast}{ast\optional{, filename\code{ = '<ast>'}}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000190The Python byte compiler can be invoked on an AST object to produce
191code objects which can be used as part of an \code{exec} statement or
Fred Drake88223901998-02-09 20:52:48 +0000192a call to the built-in \function{eval()}\bifuncindex{eval} function.
193This function provides the interface to the compiler, passing the
194internal parse tree from \code{\var{ast}} to the parser, using the
195source file name specified by the \code{\var{filename}} parameter.
196The default value supplied for \code{\var{filename}} indicates that
197the source was an AST object.
Guido van Rossum47478871996-08-21 14:32:37 +0000198
199Compiling an AST object may result in exceptions related to
Fred Drake88223901998-02-09 20:52:48 +0000200compilation; an example would be a \exception{SyntaxError} caused by the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000201parse tree for \code{del f(0)}: this statement is considered legal
Guido van Rossum47478871996-08-21 14:32:37 +0000202within the formal grammar for Python but is not a legal language
Fred Drake88223901998-02-09 20:52:48 +0000203construct. The \exception{SyntaxError} raised for this condition is
Guido van Rossum47478871996-08-21 14:32:37 +0000204actually generated by the Python byte-compiler normally, which is why
Fred Drake88223901998-02-09 20:52:48 +0000205it can be raised at this point by the \module{parser} module. Most
Guido van Rossum47478871996-08-21 14:32:37 +0000206causes of compilation failure can be diagnosed programmatically by
207inspection of the parse tree.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000208\end{funcdesc}
209
210
Fred Drake4b7d5a41996-09-11 21:57:40 +0000211\subsection{Queries on AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000212\label{Querying ASTs}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000213
Fred Drake4b7d5a41996-09-11 21:57:40 +0000214Two functions are provided which allow an application to determine if
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000215an AST was created as an expression or a suite. Neither of these
Fred Drake4b7d5a41996-09-11 21:57:40 +0000216functions can be used to determine if an AST was created from source
Fred Drake88223901998-02-09 20:52:48 +0000217code via \function{expr()} or \function{suite()} or from a parse tree
218via \function{sequence2ast()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000219
220\begin{funcdesc}{isexpr}{ast}
221When \code{\var{ast}} represents an \code{'eval'} form, this function
Fred Drake88223901998-02-09 20:52:48 +0000222returns true, otherwise it returns false. This is useful, since code
223objects normally cannot be queried for this information using existing
224built-in functions. Note that the code objects created by
225\function{compileast()} cannot be queried like this either, and are
226identical to those created by the built-in
227\function{compile()}\bifuncindex{compile} function.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000228\end{funcdesc}
229
230
231\begin{funcdesc}{issuite}{ast}
Fred Drake88223901998-02-09 20:52:48 +0000232This function mirrors \function{isexpr()} in that it reports whether an
Fred Drake4b7d5a41996-09-11 21:57:40 +0000233AST object represents an \code{'exec'} form, commonly known as a
234``suite.'' It is not safe to assume that this function is equivelent
Fred Drake88223901998-02-09 20:52:48 +0000235to \samp{not isexpr(\var{ast})}, as additional syntactic fragments may
Fred Drake4b7d5a41996-09-11 21:57:40 +0000236be supported in the future.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000237\end{funcdesc}
238
239
Guido van Rossum4b73a061995-10-11 17:30:04 +0000240\subsection{Exceptions and Error Handling}
Fred Draked67e12e1998-02-20 05:49:37 +0000241\label{AST Errors}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000242
243The parser module defines a single exception, but may also pass other
244built-in exceptions from other portions of the Python runtime
245environment. See each function for information about the exceptions
246it can raise.
247
248\begin{excdesc}{ParserError}
249Exception raised when a failure occurs within the parser module. This
250is generally produced for validation failures rather than the built in
Fred Drake88223901998-02-09 20:52:48 +0000251\exception{SyntaxError} thrown during normal parsing.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000252The exception argument is either a string describing the reason of the
Guido van Rossum47478871996-08-21 14:32:37 +0000253failure or a tuple containing a sequence causing the failure from a parse
Fred Drake88223901998-02-09 20:52:48 +0000254tree passed to \function{sequence2ast()} and an explanatory string. Calls to
255\function{sequence2ast()} need to be able to handle either type of exception,
Guido van Rossum4b73a061995-10-11 17:30:04 +0000256while calls to other functions in the module will only need to be
257aware of the simple string values.
258\end{excdesc}
259
Fred Drake88223901998-02-09 20:52:48 +0000260Note that the functions \function{compileast()}, \function{expr()}, and
261\function{suite()} may throw exceptions which are normally thrown by the
Guido van Rossum4b73a061995-10-11 17:30:04 +0000262parsing and compilation process. These include the built in
Fred Drake88223901998-02-09 20:52:48 +0000263exceptions \exception{MemoryError}, \exception{OverflowError},
264\exception{SyntaxError}, and \exception{SystemError}. In these cases, these
Guido van Rossum4b73a061995-10-11 17:30:04 +0000265exceptions carry all the meaning normally associated with them. Refer
266to the descriptions of each function for detailed information.
267
Guido van Rossum4b73a061995-10-11 17:30:04 +0000268
Guido van Rossum47478871996-08-21 14:32:37 +0000269\subsection{AST Objects}
Fred Draked67e12e1998-02-20 05:49:37 +0000270\label{AST Objects}
Guido van Rossum47478871996-08-21 14:32:37 +0000271
Fred Drakecc444e31998-03-08 06:47:24 +0000272AST objects returned by \function{expr()}, \function{suite()} and
Fred Drake88223901998-02-09 20:52:48 +0000273\function{sequence2ast()} have no methods of their own.
Fred Drakecc444e31998-03-08 06:47:24 +0000274
Fred Drakeaf370ea1998-04-05 20:23:02 +0000275Ordered and equality comparisons are supported between AST objects.
Fred Drakec4f1ca11998-04-13 16:27:27 +0000276Pickling of AST objects (using the \module{pickle} module) is also
277supported.
Fred Drakeaf370ea1998-04-05 20:23:02 +0000278
Fred Drakecc444e31998-03-08 06:47:24 +0000279\begin{datadesc}{ASTType}
280The type of the objects returned by \function{expr()},
281\function{suite()} and \function{sequence2ast()}.
Fred Drakecc444e31998-03-08 06:47:24 +0000282\end{datadesc}
Guido van Rossum47478871996-08-21 14:32:37 +0000283
284
Fred Drake916d8f81998-04-13 18:46:16 +0000285AST objects have the following methods:
286
287
288\begin{methoddesc}[AST]{compile}{\optional{filename}}
289Same as \code{compileast(\var{ast}, \var{filename})}.
290\end{methoddesc}
291
292\begin{methoddesc}[AST]{isexpr}{}
293Same as \code{isexpr(\var{ast})}.
294\end{methoddesc}
295
296\begin{methoddesc}[AST]{issuite}{}
297Same as \code{issuite(\var{ast})}.
298\end{methoddesc}
299
300\begin{methoddesc}[AST]{tolist}{\optional{line_info}}
301Same as \code{ast2list(\var{ast}, \var{line_info})}.
302\end{methoddesc}
303
304\begin{methoddesc}[AST]{totuple}{\optional{line_info}}
305Same as \code{ast2tuple(\var{ast}, \var{line_info})}.
306\end{methoddesc}
307
308
Guido van Rossum8206fb91996-08-26 00:33:29 +0000309\subsection{Examples}
Fred Drake4b3f0311996-12-13 22:04:31 +0000310\nodename{AST Examples}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000311
Guido van Rossum47478871996-08-21 14:32:37 +0000312The parser modules allows operations to be performed on the parse tree
313of Python source code before the bytecode is generated, and provides
Fred Drake4b7d5a41996-09-11 21:57:40 +0000314for inspection of the parse tree for information gathering purposes.
315Two examples are presented. The simple example demonstrates emulation
Fred Drake88223901998-02-09 20:52:48 +0000316of the \function{compile()}\bifuncindex{compile} built-in function and
317the complex example shows the use of a parse tree for information
318discovery.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000319
Fred Drakeaf370ea1998-04-05 20:23:02 +0000320\subsubsection{Emulation of \function{compile()}}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000321
322While many useful operations may take place between parsing and
Guido van Rossum47478871996-08-21 14:32:37 +0000323bytecode generation, the simplest operation is to do nothing. For
Fred Drake88223901998-02-09 20:52:48 +0000324this purpose, using the \module{parser} module to produce an
Guido van Rossum47478871996-08-21 14:32:37 +0000325intermediate data structure is equivelent to the code
326
Fred Drake19479911998-02-13 06:58:54 +0000327\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000328>>> code = compile('a + 5', 'eval')
329>>> a = 5
330>>> eval(code)
33110
Fred Drake19479911998-02-13 06:58:54 +0000332\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000333
Fred Drake88223901998-02-09 20:52:48 +0000334The equivelent operation using the \module{parser} module is somewhat
Guido van Rossum47478871996-08-21 14:32:37 +0000335longer, and allows the intermediate internal parse tree to be retained
336as an AST object:
Guido van Rossum4b73a061995-10-11 17:30:04 +0000337
Fred Drake19479911998-02-13 06:58:54 +0000338\begin{verbatim}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000339>>> import parser
340>>> ast = parser.expr('a + 5')
341>>> code = parser.compileast(ast)
342>>> a = 5
343>>> eval(code)
34410
Fred Drake19479911998-02-13 06:58:54 +0000345\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000346
Guido van Rossum8206fb91996-08-26 00:33:29 +0000347An application which needs both AST and code objects can package this
348code into readily available functions:
349
Fred Drake19479911998-02-13 06:58:54 +0000350\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000351import parser
352
353def load_suite(source_string):
354 ast = parser.suite(source_string)
355 code = parser.compileast(ast)
356 return ast, code
357
358def load_expression(source_string):
359 ast = parser.expr(source_string)
360 code = parser.compileast(ast)
361 return ast, code
Fred Drake19479911998-02-13 06:58:54 +0000362\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000363
Guido van Rossum8206fb91996-08-26 00:33:29 +0000364\subsubsection{Information Discovery}
365
Fred Drake4b7d5a41996-09-11 21:57:40 +0000366Some applications benefit from direct access to the parse tree. The
367remainder of this section demonstrates how the parse tree provides
368access to module documentation defined in docstrings without requiring
369that the code being examined be loaded into a running interpreter via
Fred Drake88223901998-02-09 20:52:48 +0000370\keyword{import}. This can be very useful for performing analyses of
Fred Drake4b7d5a41996-09-11 21:57:40 +0000371untrusted code.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000372
Guido van Rossum47478871996-08-21 14:32:37 +0000373Generally, the example will demonstrate how the parse tree may be
374traversed to distill interesting information. Two functions and a set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000375of classes are developed which provide programmatic access to high
Guido van Rossum47478871996-08-21 14:32:37 +0000376level function and class definitions provided by a module. The
377classes extract information from the parse tree and provide access to
378the information at a useful semantic level, one function provides a
379simple low-level pattern matching capability, and the other function
380defines a high-level interface to the classes by handling file
381operations on behalf of the caller. All source files mentioned here
382which are not part of the Python installation are located in the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000383\file{Demo/parser/} directory of the distribution.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000384
Guido van Rossum8206fb91996-08-26 00:33:29 +0000385The dynamic nature of Python allows the programmer a great deal of
386flexibility, but most modules need only a limited measure of this when
387defining classes, functions, and methods. In this example, the only
388definitions that will be considered are those which are defined in the
Fred Drake88223901998-02-09 20:52:48 +0000389top level of their context, e.g., a function defined by a \keyword{def}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000390statement at column zero of a module, but not a function defined
Fred Drake4b7d5a41996-09-11 21:57:40 +0000391within a branch of an \code{if} ... \code{else} construct, though
Guido van Rossum8206fb91996-08-26 00:33:29 +0000392there are some good reasons for doing so in some situations. Nesting
393of definitions will be handled by the code developed in the example.
394
Guido van Rossum47478871996-08-21 14:32:37 +0000395To construct the upper-level extraction methods, we need to know what
396the parse tree structure looks like and how much of it we actually
Fred Drake4b7d5a41996-09-11 21:57:40 +0000397need to be concerned about. Python uses a moderately deep parse tree
Guido van Rossum47478871996-08-21 14:32:37 +0000398so there are a large number of intermediate nodes. It is important to
399read and understand the formal grammar used by Python. This is
400specified in the file \file{Grammar/Grammar} in the distribution.
401Consider the simplest case of interest when searching for docstrings:
Guido van Rossum8206fb91996-08-26 00:33:29 +0000402a module consisting of a docstring and nothing else. (See file
403\file{docstring.py}.)
Guido van Rossum4b73a061995-10-11 17:30:04 +0000404
Fred Drake19479911998-02-13 06:58:54 +0000405\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000406"""Some documentation.
407"""
Fred Drake19479911998-02-13 06:58:54 +0000408\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000409
Guido van Rossum47478871996-08-21 14:32:37 +0000410Using the interpreter to take a look at the parse tree, we find a
411bewildering mass of numbers and parentheses, with the documentation
Fred Drake4b7d5a41996-09-11 21:57:40 +0000412buried deep in nested tuples.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000413
Fred Drake19479911998-02-13 06:58:54 +0000414\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000415>>> import parser
416>>> import pprint
417>>> ast = parser.suite(open('docstring.py').read())
418>>> tup = parser.ast2tuple(ast)
419>>> pprint.pprint(tup)
420(257,
421 (264,
422 (265,
423 (266,
424 (267,
425 (307,
426 (287,
427 (288,
428 (289,
429 (290,
430 (292,
431 (293,
432 (294,
433 (295,
434 (296,
435 (297,
436 (298,
437 (299,
438 (300, (3, '"""Some documentation.\012"""'))))))))))))))))),
439 (4, ''))),
440 (4, ''),
441 (0, ''))
Fred Drake19479911998-02-13 06:58:54 +0000442\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000443
Guido van Rossum47478871996-08-21 14:32:37 +0000444The numbers at the first element of each node in the tree are the node
445types; they map directly to terminal and non-terminal symbols in the
446grammar. Unfortunately, they are represented as integers in the
447internal representation, and the Python structures generated do not
Fred Drake88223901998-02-09 20:52:48 +0000448change that. However, the \module{symbol} and \module{token} modules
Guido van Rossum47478871996-08-21 14:32:37 +0000449provide symbolic names for the node types and dictionaries which map
450from the integers to the symbolic names for the node types.
451
452In the output presented above, the outermost tuple contains four
453elements: the integer \code{257} and three additional tuples. Node
Fred Drake88223901998-02-09 20:52:48 +0000454type \code{257} has the symbolic name \constant{file_input}. Each of
Guido van Rossum47478871996-08-21 14:32:37 +0000455these inner tuples contains an integer as the first element; these
456integers, \code{264}, \code{4}, and \code{0}, represent the node types
Fred Drake88223901998-02-09 20:52:48 +0000457\constant{stmt}, \constant{NEWLINE}, and \constant{ENDMARKER},
458respectively.
Guido van Rossum47478871996-08-21 14:32:37 +0000459Note that these values may change depending on the version of Python
460you are using; consult \file{symbol.py} and \file{token.py} for
461details of the mapping. It should be fairly clear that the outermost
462node is related primarily to the input source rather than the contents
Fred Drake88223901998-02-09 20:52:48 +0000463of the file, and may be disregarded for the moment. The \constant{stmt}
Guido van Rossum47478871996-08-21 14:32:37 +0000464node is much more interesting. In particular, all docstrings are
465found in subtrees which are formed exactly as this node is formed,
466with the only difference being the string itself. The association
467between the docstring in a similar tree and the defined entity (class,
468function, or module) which it describes is given by the position of
469the docstring subtree within the tree defining the described
470structure.
471
472By replacing the actual docstring with something to signify a variable
Fred Drake4b7d5a41996-09-11 21:57:40 +0000473component of the tree, we allow a simple pattern matching approach to
474check any given subtree for equivelence to the general pattern for
475docstrings. Since the example demonstrates information extraction, we
476can safely require that the tree be in tuple form rather than list
477form, allowing a simple variable representation to be
478\code{['variable_name']}. A simple recursive function can implement
Guido van Rossum47478871996-08-21 14:32:37 +0000479the pattern matching, returning a boolean and a dictionary of variable
Guido van Rossum8206fb91996-08-26 00:33:29 +0000480name to value mappings. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000481
Fred Drake19479911998-02-13 06:58:54 +0000482\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000483from types import ListType, TupleType
484
485def match(pattern, data, vars=None):
486 if vars is None:
487 vars = {}
488 if type(pattern) is ListType:
489 vars[pattern[0]] = data
490 return 1, vars
491 if type(pattern) is not TupleType:
492 return (pattern == data), vars
493 if len(data) != len(pattern):
494 return 0, vars
495 for pattern, data in map(None, pattern, data):
496 same, vars = match(pattern, data, vars)
497 if not same:
498 break
499 return same, vars
Fred Drake19479911998-02-13 06:58:54 +0000500\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000501
Fred Drake4b7d5a41996-09-11 21:57:40 +0000502Using this simple representation for syntactic variables and the symbolic
Guido van Rossum8206fb91996-08-26 00:33:29 +0000503node types, the pattern for the candidate docstring subtrees becomes
504fairly readable. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000505
Fred Drake19479911998-02-13 06:58:54 +0000506\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000507import symbol
508import token
509
510DOCSTRING_STMT_PATTERN = (
511 symbol.stmt,
512 (symbol.simple_stmt,
513 (symbol.small_stmt,
514 (symbol.expr_stmt,
515 (symbol.testlist,
516 (symbol.test,
517 (symbol.and_test,
518 (symbol.not_test,
519 (symbol.comparison,
520 (symbol.expr,
521 (symbol.xor_expr,
522 (symbol.and_expr,
523 (symbol.shift_expr,
524 (symbol.arith_expr,
525 (symbol.term,
526 (symbol.factor,
527 (symbol.power,
528 (symbol.atom,
529 (token.STRING, ['docstring'])
530 )))))))))))))))),
531 (token.NEWLINE, '')
532 ))
Fred Drake19479911998-02-13 06:58:54 +0000533\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000534
Fred Drake88223901998-02-09 20:52:48 +0000535Using the \function{match()} function with this pattern, extracting the
Guido van Rossum47478871996-08-21 14:32:37 +0000536module docstring from the parse tree created previously is easy:
537
Fred Drake19479911998-02-13 06:58:54 +0000538\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000539>>> found, vars = match(DOCSTRING_STMT_PATTERN, tup[1])
540>>> found
5411
542>>> vars
543{'docstring': '"""Some documentation.\012"""'}
Fred Drake19479911998-02-13 06:58:54 +0000544\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000545
Guido van Rossum47478871996-08-21 14:32:37 +0000546Once specific data can be extracted from a location where it is
547expected, the question of where information can be expected
548needs to be answered. When dealing with docstrings, the answer is
Fred Drake88223901998-02-09 20:52:48 +0000549fairly simple: the docstring is the first \constant{stmt} node in a code
550block (\constant{file_input} or \constant{suite} node types). A module
551consists of a single \constant{file_input} node, and class and function
552definitions each contain exactly one \constant{suite} node. Classes and
Guido van Rossum47478871996-08-21 14:32:37 +0000553functions are readily identified as subtrees of code block nodes which
554start with \code{(stmt, (compound_stmt, (classdef, ...} or
555\code{(stmt, (compound_stmt, (funcdef, ...}. Note that these subtrees
Fred Drake88223901998-02-09 20:52:48 +0000556cannot be matched by \function{match()} since it does not support multiple
Guido van Rossum47478871996-08-21 14:32:37 +0000557sibling nodes to match without regard to number. A more elaborate
558matching function could be used to overcome this limitation, but this
559is sufficient for the example.
560
Guido van Rossum8206fb91996-08-26 00:33:29 +0000561Given the ability to determine whether a statement might be a
562docstring and extract the actual string from the statement, some work
563needs to be performed to walk the parse tree for an entire module and
564extract information about the names defined in each context of the
565module and associate any docstrings with the names. The code to
566perform this work is not complicated, but bears some explanation.
567
568The public interface to the classes is straightforward and should
569probably be somewhat more flexible. Each ``major'' block of the
570module is described by an object providing several methods for inquiry
571and a constructor which accepts at least the subtree of the complete
Fred Drakeb0df5671998-02-18 15:59:13 +0000572parse tree which it represents. The \class{ModuleInfo} constructor
573accepts an optional \var{name} parameter since it cannot
Guido van Rossum8206fb91996-08-26 00:33:29 +0000574otherwise determine the name of the module.
575
Fred Drake88223901998-02-09 20:52:48 +0000576The public classes include \class{ClassInfo}, \class{FunctionInfo},
577and \class{ModuleInfo}. All objects provide the
578methods \method{get_name()}, \method{get_docstring()},
579\method{get_class_names()}, and \method{get_class_info()}. The
580\class{ClassInfo} objects support \method{get_method_names()} and
581\method{get_method_info()} while the other classes provide
582\method{get_function_names()} and \method{get_function_info()}.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000583
584Within each of the forms of code block that the public classes
585represent, most of the required information is in the same form and is
Fred Drake4b7d5a41996-09-11 21:57:40 +0000586accessed in the same way, with classes having the distinction that
Guido van Rossum8206fb91996-08-26 00:33:29 +0000587functions defined at the top level are referred to as ``methods.''
588Since the difference in nomenclature reflects a real semantic
Fred Drake4b7d5a41996-09-11 21:57:40 +0000589distinction from functions defined outside of a class, the
590implementation needs to maintain the distinction.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000591Hence, most of the functionality of the public classes can be
Fred Drake88223901998-02-09 20:52:48 +0000592implemented in a common base class, \class{SuiteInfoBase}, with the
Guido van Rossum8206fb91996-08-26 00:33:29 +0000593accessors for function and method information provided elsewhere.
594Note that there is only one class which represents function and method
Fred Drake88223901998-02-09 20:52:48 +0000595information; this parallels the use of the \keyword{def} statement to
Fred Drake4b7d5a41996-09-11 21:57:40 +0000596define both types of elements.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000597
Fred Drake88223901998-02-09 20:52:48 +0000598Most of the accessor functions are declared in \class{SuiteInfoBase}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000599and do not need to be overriden by subclasses. More importantly, the
600extraction of most information from a parse tree is handled through a
Fred Drake88223901998-02-09 20:52:48 +0000601method called by the \class{SuiteInfoBase} constructor. The example
Guido van Rossum8206fb91996-08-26 00:33:29 +0000602code for most of the classes is clear when read alongside the formal
603grammar, but the method which recursively creates new information
604objects requires further examination. Here is the relevant part of
Fred Drake88223901998-02-09 20:52:48 +0000605the \class{SuiteInfoBase} definition from \file{example.py}:
Guido van Rossum8206fb91996-08-26 00:33:29 +0000606
Fred Drake19479911998-02-13 06:58:54 +0000607\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000608class SuiteInfoBase:
609 _docstring = ''
610 _name = ''
611
612 def __init__(self, tree = None):
613 self._class_info = {}
614 self._function_info = {}
615 if tree:
616 self._extract_info(tree)
617
618 def _extract_info(self, tree):
619 # extract docstring
620 if len(tree) == 2:
621 found, vars = match(DOCSTRING_STMT_PATTERN[1], tree[1])
622 else:
623 found, vars = match(DOCSTRING_STMT_PATTERN, tree[3])
624 if found:
625 self._docstring = eval(vars['docstring'])
626 # discover inner definitions
627 for node in tree[1:]:
628 found, vars = match(COMPOUND_STMT_PATTERN, node)
629 if found:
630 cstmt = vars['compound']
631 if cstmt[0] == symbol.funcdef:
632 name = cstmt[2][1]
633 self._function_info[name] = FunctionInfo(cstmt)
634 elif cstmt[0] == symbol.classdef:
635 name = cstmt[2][1]
636 self._class_info[name] = ClassInfo(cstmt)
Fred Drake19479911998-02-13 06:58:54 +0000637\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000638
Guido van Rossum8206fb91996-08-26 00:33:29 +0000639After initializing some internal state, the constructor calls the
Fred Drake88223901998-02-09 20:52:48 +0000640\method{_extract_info()} method. This method performs the bulk of the
Guido van Rossum8206fb91996-08-26 00:33:29 +0000641information extraction which takes place in the entire example. The
642extraction has two distinct phases: the location of the docstring for
643the parse tree passed in, and the discovery of additional definitions
644within the code block represented by the parse tree.
645
Fred Drake88223901998-02-09 20:52:48 +0000646The initial \keyword{if} test determines whether the nested suite is of
Guido van Rossum8206fb91996-08-26 00:33:29 +0000647the ``short form'' or the ``long form.'' The short form is used when
648the code block is on the same line as the definition of the code
649block, as in
650
Fred Drakebbe60681998-01-09 22:24:14 +0000651\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000652def square(x): "Square an argument."; return x ** 2
Fred Drakebbe60681998-01-09 22:24:14 +0000653\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000654
Guido van Rossum8206fb91996-08-26 00:33:29 +0000655while the long form uses an indented block and allows nested
656definitions:
657
Fred Drake19479911998-02-13 06:58:54 +0000658\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000659def make_power(exp):
660 "Make a function that raises an argument to the exponent `exp'."
661 def raiser(x, y=exp):
662 return x ** y
663 return raiser
Fred Drake19479911998-02-13 06:58:54 +0000664\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000665
Guido van Rossum8206fb91996-08-26 00:33:29 +0000666When the short form is used, the code block may contain a docstring as
Fred Drake88223901998-02-09 20:52:48 +0000667the first, and possibly only, \constant{small_stmt} element. The
Guido van Rossum8206fb91996-08-26 00:33:29 +0000668extraction of such a docstring is slightly different and requires only
669a portion of the complete pattern used in the more common case. As
Fred Drake4b7d5a41996-09-11 21:57:40 +0000670implemented, the docstring will only be found if there is only
Fred Drake88223901998-02-09 20:52:48 +0000671one \constant{small_stmt} node in the \constant{simple_stmt} node.
672Since most functions and methods which use the short form do not
673provide a docstring, this may be considered sufficient. The
674extraction of the docstring proceeds using the \function{match()} function
675as described above, and the value of the docstring is stored as an
676attribute of the \class{SuiteInfoBase} object.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000677
Fred Drake4b7d5a41996-09-11 21:57:40 +0000678After docstring extraction, a simple definition discovery
Fred Drake88223901998-02-09 20:52:48 +0000679algorithm operates on the \constant{stmt} nodes of the
680\constant{suite} node. The special case of the short form is not
681tested; since there are no \constant{stmt} nodes in the short form,
682the algorithm will silently skip the single \constant{simple_stmt}
683node and correctly not discover any nested definitions.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000684
Fred Drake4b7d5a41996-09-11 21:57:40 +0000685Each statement in the code block is categorized as
686a class definition, function or method definition, or
Guido van Rossum8206fb91996-08-26 00:33:29 +0000687something else. For the definition statements, the name of the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000688element defined is extracted and a representation object
Guido van Rossum8206fb91996-08-26 00:33:29 +0000689appropriate to the definition is created with the defining subtree
690passed as an argument to the constructor. The repesentation objects
691are stored in instance variables and may be retrieved by name using
692the appropriate accessor methods.
693
694The public classes provide any accessors required which are more
Fred Drake88223901998-02-09 20:52:48 +0000695specific than those provided by the \class{SuiteInfoBase} class, but
Guido van Rossum8206fb91996-08-26 00:33:29 +0000696the real extraction algorithm remains common to all forms of code
697blocks. A high-level function can be used to extract the complete set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000698of information from a source file. (See file \file{example.py}.)
Guido van Rossum8206fb91996-08-26 00:33:29 +0000699
Fred Drake19479911998-02-13 06:58:54 +0000700\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000701def get_docs(fileName):
702 source = open(fileName).read()
703 import os
704 basename = os.path.basename(os.path.splitext(fileName)[0])
705 import parser
706 ast = parser.suite(source)
707 tup = parser.ast2tuple(ast)
708 return ModuleInfo(tup, basename)
Fred Drake19479911998-02-13 06:58:54 +0000709\end{verbatim}
Fred Drake5bd7fcc1998-04-03 05:31:45 +0000710
Guido van Rossum8206fb91996-08-26 00:33:29 +0000711This provides an easy-to-use interface to the documentation of a
712module. If information is required which is not extracted by the code
713of this example, the code may be extended at clearly defined points to
714provide additional capabilities.
Guido van Rossum47478871996-08-21 14:32:37 +0000715
Fred Drakebbe60681998-01-09 22:24:14 +0000716\begin{seealso}
717
Fred Drake45c634e1998-04-09 15:44:58 +0000718\seemodule{symbol}{useful constants representing internal nodes of the
719parse tree}
Fred Drakebbe60681998-01-09 22:24:14 +0000720
Fred Drake45c634e1998-04-09 15:44:58 +0000721\seemodule{token}{useful constants representing leaf nodes of the
722parse tree and functions for testing node values}
Fred Drakebbe60681998-01-09 22:24:14 +0000723
724\end{seealso}