blob: 47b5bd454383c307cba7c085f506d2a731873b2e [file] [log] [blame]
Guido van Rossum4b73a061995-10-11 17:30:04 +00001% libparser.tex
2%
Guido van Rossum4b73a061995-10-11 17:30:04 +00003% Copyright 1995 Virginia Polytechnic Institute and State University
4% and Fred L. Drake, Jr. This copyright notice must be distributed on
5% all copies, but this document otherwise may be distributed as part
6% of the Python distribution. No fee may be charged for this document
7% in any representation, either on paper or electronically. This
8% restriction does not affect other elements in a distributed package
9% in any way.
10%
11
12\section{Built-in Module \sectcode{parser}}
Fred Drakebbe60681998-01-09 22:24:14 +000013\label{module-parser}
Guido van Rossum4b73a061995-10-11 17:30:04 +000014\bimodindex{parser}
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
25There are a few things to note about this module which are important
26to making use of the data structures created. This is not a tutorial
Fred Drake4b7d5a41996-09-11 21:57:40 +000027on editing the parse trees for Python code, but some examples of using
Fred Drake88223901998-02-09 20:52:48 +000028the \module{parser} module are presented.
Guido van Rossum4b73a061995-10-11 17:30:04 +000029
30Most importantly, a good understanding of the Python grammar processed
31by the internal parser is required. For full information on the
Fred Drake88223901998-02-09 20:52:48 +000032language syntax, refer to the \emph{Python Language Reference}. The
33parser itself is created from a grammar specification defined in the file
Fred Drake4b7d5a41996-09-11 21:57:40 +000034\file{Grammar/Grammar} in the standard Python distribution. The parse
Guido van Rossum4b73a061995-10-11 17:30:04 +000035trees stored in the ``AST objects'' created by this module are the
36actual output from the internal parser when created by the
Fred Drake88223901998-02-09 20:52:48 +000037\function{expr()} or \function{suite()} functions, described below. The AST
38objects created by \function{sequence2ast()} faithfully simulate those
Guido van Rossum47478871996-08-21 14:32:37 +000039structures. Be aware that the values of the sequences which are
40considered ``correct'' will vary from one version of Python to another
41as the formal grammar for the language is revised. However,
42transporting code from one Python version to another as source text
43will always allow correct parse trees to be created in the target
44version, with the only restriction being that migrating to an older
45version of the interpreter will not support more recent language
46constructs. The parse trees are not typically compatible from one
47version to another, whereas source code has always been
48forward-compatible.
Guido van Rossum4b73a061995-10-11 17:30:04 +000049
Fred Drake88223901998-02-09 20:52:48 +000050Each element of the sequences returned by \function{ast2list()} or
51\function{ast2tuple()} has a simple form. Sequences representing
Guido van Rossum47478871996-08-21 14:32:37 +000052non-terminal elements in the grammar always have a length greater than
53one. The first element is an integer which identifies a production in
54the grammar. These integers are given symbolic names in the C header
Fred Drake4b7d5a41996-09-11 21:57:40 +000055file \file{Include/graminit.h} and the Python module
Fred Drake88223901998-02-09 20:52:48 +000056\module{symbol}. Each additional element of the sequence represents
Guido van Rossum47478871996-08-21 14:32:37 +000057a component of the production as recognized in the input string: these
58are always sequences which have the same form as the parent. An
59important aspect of this structure which should be noted is that
60keywords used to identify the parent node type, such as the keyword
Fred Drake88223901998-02-09 20:52:48 +000061\keyword{if} in an \constant{if_stmt}, are included in the node tree without
62any special treatment. For example, the \keyword{if} keyword is
Guido van Rossum4b73a061995-10-11 17:30:04 +000063represented by the tuple \code{(1, 'if')}, where \code{1} is the
Fred Drake4b7d5a41996-09-11 21:57:40 +000064numeric value associated with all \code{NAME} tokens, including
Guido van Rossum47478871996-08-21 14:32:37 +000065variable and function names defined by the user. In an alternate form
66returned when line number information is requested, the same token
67might be represented as \code{(1, 'if', 12)}, where the \code{12}
68represents the line number at which the terminal symbol was found.
Guido van Rossum4b73a061995-10-11 17:30:04 +000069
70Terminal elements are represented in much the same way, but without
71any child elements and the addition of the source text which was
Fred Drake88223901998-02-09 20:52:48 +000072identified. The example of the \keyword{if} keyword above is
Guido van Rossum4b73a061995-10-11 17:30:04 +000073representative. The various types of terminal symbols are defined in
Fred Drake4b7d5a41996-09-11 21:57:40 +000074the C header file \file{Include/token.h} and the Python module
Fred Drake88223901998-02-09 20:52:48 +000075\module{token}.
Guido van Rossum4b73a061995-10-11 17:30:04 +000076
Fred Drake4b7d5a41996-09-11 21:57:40 +000077The AST objects are not required to support the functionality of this
78module, but are provided for three purposes: to allow an application
79to amortize the cost of processing complex parse trees, to provide a
80parse tree representation which conserves memory space when compared
81to the Python list or tuple representation, and to ease the creation
82of additional modules in C which manipulate parse trees. A simple
83``wrapper'' class may be created in Python to hide the use of AST
Fred Drake88223901998-02-09 20:52:48 +000084objects.
Guido van Rossum4b73a061995-10-11 17:30:04 +000085
Fred Drake88223901998-02-09 20:52:48 +000086The \module{parser} module defines functions for a few distinct
Fred Drake4b7d5a41996-09-11 21:57:40 +000087purposes. The most important purposes are to create AST objects and
88to convert AST objects to other representations such as parse trees
89and compiled code objects, but there are also functions which serve to
90query the type of parse tree represented by an AST object.
Guido van Rossum4b73a061995-10-11 17:30:04 +000091
Fred Drake19479911998-02-13 06:58:54 +000092\setindexsubitem{(in module parser)}
Guido van Rossum4b73a061995-10-11 17:30:04 +000093
Fred Drake4b7d5a41996-09-11 21:57:40 +000094
95\subsection{Creating AST Objects}
96
97AST objects may be created from source code or from a parse tree.
98When creating an AST object from source, different functions are used
99to create the \code{'eval'} and \code{'exec'} forms.
100
101\begin{funcdesc}{expr}{string}
Fred Drake88223901998-02-09 20:52:48 +0000102The \function{expr()} function parses the parameter \code{\var{string}}
103as if it were an input to \samp{compile(\var{string}, 'eval')}. If
Fred Drake4b7d5a41996-09-11 21:57:40 +0000104the parse succeeds, an AST object is created to hold the internal
105parse tree representation, otherwise an appropriate exception is
106thrown.
107\end{funcdesc}
108
109\begin{funcdesc}{suite}{string}
Fred Drake88223901998-02-09 20:52:48 +0000110The \function{suite()} function parses the parameter \code{\var{string}}
111as if it were an input to \samp{compile(\var{string}, 'exec')}. If
Fred Drake4b7d5a41996-09-11 21:57:40 +0000112the parse succeeds, an AST object is created to hold the internal
113parse tree representation, otherwise an appropriate exception is
114thrown.
115\end{funcdesc}
116
117\begin{funcdesc}{sequence2ast}{sequence}
118This function accepts a parse tree represented as a sequence and
119builds an internal representation if possible. If it can validate
120that the tree conforms to the Python grammar and all nodes are valid
121node types in the host version of Python, an AST object is created
122from the internal representation and returned to the called. If there
123is a problem creating the internal representation, or if the tree
Fred Drake88223901998-02-09 20:52:48 +0000124cannot be validated, a \exception{ParserError} exception is thrown. An AST
Fred Drake4b7d5a41996-09-11 21:57:40 +0000125object created this way should not be assumed to compile correctly;
126normal exceptions thrown by compilation may still be initiated when
Fred Drake88223901998-02-09 20:52:48 +0000127the AST object is passed to \function{compileast()}. This may indicate
128problems not related to syntax (such as a \exception{MemoryError}
Fred Drake4b7d5a41996-09-11 21:57:40 +0000129exception), but may also be due to constructs such as the result of
130parsing \code{del f(0)}, which escapes the Python parser but is
131checked by the bytecode compiler.
132
133Sequences representing terminal tokens may be represented as either
134two-element lists of the form \code{(1, 'name')} or as three-element
135lists of the form \code{(1, 'name', 56)}. If the third element is
136present, it is assumed to be a valid line number. The line number
137may be specified for any subset of the terminal symbols in the input
138tree.
139\end{funcdesc}
140
141\begin{funcdesc}{tuple2ast}{sequence}
Fred Drake88223901998-02-09 20:52:48 +0000142This is the same function as \function{sequence2ast()}. This entry point
Fred Drake4b7d5a41996-09-11 21:57:40 +0000143is maintained for backward compatibility.
144\end{funcdesc}
145
146
147\subsection{Converting AST Objects}
148
149AST objects, regardless of the input used to create them, may be
150converted to parse trees represented as list- or tuple- trees, or may
151be compiled into executable code objects. Parse trees may be
152extracted with or without line numbering information.
153
154\begin{funcdesc}{ast2list}{ast\optional{\, line_info\code{ = 0}}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000155This function accepts an AST object from the caller in
Guido van Rossum47478871996-08-21 14:32:37 +0000156\code{\var{ast}} and returns a Python list representing the
157equivelent parse tree. The resulting list representation can be used
Fred Drake4b7d5a41996-09-11 21:57:40 +0000158for inspection or the creation of a new parse tree in list form. This
159function does not fail so long as memory is available to build the
160list representation. If the parse tree will only be used for
Fred Drake88223901998-02-09 20:52:48 +0000161inspection, \function{ast2tuple()} should be used instead to reduce memory
Fred Drake4b7d5a41996-09-11 21:57:40 +0000162consumption and fragmentation. When the list representation is
163required, this function is significantly faster than retrieving a
164tuple representation and converting that to nested lists.
Guido van Rossum47478871996-08-21 14:32:37 +0000165
Fred Drake4b7d5a41996-09-11 21:57:40 +0000166If \code{\var{line_info}} is true, line number information will be
167included for all terminal tokens as a third element of the list
Fred Drake9abe64a1996-12-05 22:28:43 +0000168representing the token. Note that the line number provided specifies
Fred Drake88223901998-02-09 20:52:48 +0000169the line on which the token \emph{ends}. This information is
Fred Drake9abe64a1996-12-05 22:28:43 +0000170omitted if the flag is false or omitted.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000171\end{funcdesc}
172
Fred Drake4b7d5a41996-09-11 21:57:40 +0000173\begin{funcdesc}{ast2tuple}{ast\optional{\, line_info\code{ = 0}}}
Guido van Rossum47478871996-08-21 14:32:37 +0000174This function accepts an AST object from the caller in
175\code{\var{ast}} and returns a Python tuple representing the
176equivelent parse tree. Other than returning a tuple instead of a
Fred Drake88223901998-02-09 20:52:48 +0000177list, this function is identical to \function{ast2list()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000178
Fred Drake4b7d5a41996-09-11 21:57:40 +0000179If \code{\var{line_info}} is true, line number information will be
180included for all terminal tokens as a third element of the list
181representing the token. This information is omitted if the flag is
182false or omitted.
Guido van Rossum47478871996-08-21 14:32:37 +0000183\end{funcdesc}
184
185\begin{funcdesc}{compileast}{ast\optional{\, filename\code{ = '<ast>'}}}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000186The Python byte compiler can be invoked on an AST object to produce
187code objects which can be used as part of an \code{exec} statement or
Fred Drake88223901998-02-09 20:52:48 +0000188a call to the built-in \function{eval()}\bifuncindex{eval} function.
189This function provides the interface to the compiler, passing the
190internal parse tree from \code{\var{ast}} to the parser, using the
191source file name specified by the \code{\var{filename}} parameter.
192The default value supplied for \code{\var{filename}} indicates that
193the source was an AST object.
Guido van Rossum47478871996-08-21 14:32:37 +0000194
195Compiling an AST object may result in exceptions related to
Fred Drake88223901998-02-09 20:52:48 +0000196compilation; an example would be a \exception{SyntaxError} caused by the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000197parse tree for \code{del f(0)}: this statement is considered legal
Guido van Rossum47478871996-08-21 14:32:37 +0000198within the formal grammar for Python but is not a legal language
Fred Drake88223901998-02-09 20:52:48 +0000199construct. The \exception{SyntaxError} raised for this condition is
Guido van Rossum47478871996-08-21 14:32:37 +0000200actually generated by the Python byte-compiler normally, which is why
Fred Drake88223901998-02-09 20:52:48 +0000201it can be raised at this point by the \module{parser} module. Most
Guido van Rossum47478871996-08-21 14:32:37 +0000202causes of compilation failure can be diagnosed programmatically by
203inspection of the parse tree.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000204\end{funcdesc}
205
206
Fred Drake4b7d5a41996-09-11 21:57:40 +0000207\subsection{Queries on AST Objects}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000208
Fred Drake4b7d5a41996-09-11 21:57:40 +0000209Two functions are provided which allow an application to determine if
210an AST was create as an expression or a suite. Neither of these
211functions can be used to determine if an AST was created from source
Fred Drake88223901998-02-09 20:52:48 +0000212code via \function{expr()} or \function{suite()} or from a parse tree
213via \function{sequence2ast()}.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000214
215\begin{funcdesc}{isexpr}{ast}
216When \code{\var{ast}} represents an \code{'eval'} form, this function
Fred Drake88223901998-02-09 20:52:48 +0000217returns true, otherwise it returns false. This is useful, since code
218objects normally cannot be queried for this information using existing
219built-in functions. Note that the code objects created by
220\function{compileast()} cannot be queried like this either, and are
221identical to those created by the built-in
222\function{compile()}\bifuncindex{compile} function.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000223\end{funcdesc}
224
225
226\begin{funcdesc}{issuite}{ast}
Fred Drake88223901998-02-09 20:52:48 +0000227This function mirrors \function{isexpr()} in that it reports whether an
Fred Drake4b7d5a41996-09-11 21:57:40 +0000228AST object represents an \code{'exec'} form, commonly known as a
229``suite.'' It is not safe to assume that this function is equivelent
Fred Drake88223901998-02-09 20:52:48 +0000230to \samp{not isexpr(\var{ast})}, as additional syntactic fragments may
Fred Drake4b7d5a41996-09-11 21:57:40 +0000231be supported in the future.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000232\end{funcdesc}
233
234
Guido van Rossum4b73a061995-10-11 17:30:04 +0000235\subsection{Exceptions and Error Handling}
236
237The parser module defines a single exception, but may also pass other
238built-in exceptions from other portions of the Python runtime
239environment. See each function for information about the exceptions
240it can raise.
241
242\begin{excdesc}{ParserError}
243Exception raised when a failure occurs within the parser module. This
244is generally produced for validation failures rather than the built in
Fred Drake88223901998-02-09 20:52:48 +0000245\exception{SyntaxError} thrown during normal parsing.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000246The exception argument is either a string describing the reason of the
Guido van Rossum47478871996-08-21 14:32:37 +0000247failure or a tuple containing a sequence causing the failure from a parse
Fred Drake88223901998-02-09 20:52:48 +0000248tree passed to \function{sequence2ast()} and an explanatory string. Calls to
249\function{sequence2ast()} need to be able to handle either type of exception,
Guido van Rossum4b73a061995-10-11 17:30:04 +0000250while calls to other functions in the module will only need to be
251aware of the simple string values.
252\end{excdesc}
253
Fred Drake88223901998-02-09 20:52:48 +0000254Note that the functions \function{compileast()}, \function{expr()}, and
255\function{suite()} may throw exceptions which are normally thrown by the
Guido van Rossum4b73a061995-10-11 17:30:04 +0000256parsing and compilation process. These include the built in
Fred Drake88223901998-02-09 20:52:48 +0000257exceptions \exception{MemoryError}, \exception{OverflowError},
258\exception{SyntaxError}, and \exception{SystemError}. In these cases, these
Guido van Rossum4b73a061995-10-11 17:30:04 +0000259exceptions carry all the meaning normally associated with them. Refer
260to the descriptions of each function for detailed information.
261
Guido van Rossum4b73a061995-10-11 17:30:04 +0000262
Guido van Rossum47478871996-08-21 14:32:37 +0000263\subsection{AST Objects}
264
Fred Drake88223901998-02-09 20:52:48 +0000265AST objects returned by \function{expr()}, \function{suite()}, and
266\function{sequence2ast()} have no methods of their own.
Guido van Rossum47478871996-08-21 14:32:37 +0000267Some of the functions defined which accept an AST object as their
Fred Drake4b7d5a41996-09-11 21:57:40 +0000268first argument may change to object methods in the future. The type
269of these objects is available as \code{ASTType} in the module.
Guido van Rossum47478871996-08-21 14:32:37 +0000270
271Ordered and equality comparisons are supported between AST objects.
272
273
Guido van Rossum8206fb91996-08-26 00:33:29 +0000274\subsection{Examples}
Fred Drake4b3f0311996-12-13 22:04:31 +0000275\nodename{AST Examples}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000276
Guido van Rossum47478871996-08-21 14:32:37 +0000277The parser modules allows operations to be performed on the parse tree
278of Python source code before the bytecode is generated, and provides
Fred Drake4b7d5a41996-09-11 21:57:40 +0000279for inspection of the parse tree for information gathering purposes.
280Two examples are presented. The simple example demonstrates emulation
Fred Drake88223901998-02-09 20:52:48 +0000281of the \function{compile()}\bifuncindex{compile} built-in function and
282the complex example shows the use of a parse tree for information
283discovery.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000284
Fred Drake4b7d5a41996-09-11 21:57:40 +0000285\subsubsection{Emulation of \sectcode{compile()}}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000286
287While many useful operations may take place between parsing and
Guido van Rossum47478871996-08-21 14:32:37 +0000288bytecode generation, the simplest operation is to do nothing. For
Fred Drake88223901998-02-09 20:52:48 +0000289this purpose, using the \module{parser} module to produce an
Guido van Rossum47478871996-08-21 14:32:37 +0000290intermediate data structure is equivelent to the code
291
Fred Drake19479911998-02-13 06:58:54 +0000292\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000293>>> code = compile('a + 5', 'eval')
294>>> a = 5
295>>> eval(code)
29610
Fred Drake19479911998-02-13 06:58:54 +0000297\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000298%
Fred Drake88223901998-02-09 20:52:48 +0000299The equivelent operation using the \module{parser} module is somewhat
Guido van Rossum47478871996-08-21 14:32:37 +0000300longer, and allows the intermediate internal parse tree to be retained
301as an AST object:
Guido van Rossum4b73a061995-10-11 17:30:04 +0000302
Fred Drake19479911998-02-13 06:58:54 +0000303\begin{verbatim}
Guido van Rossum4b73a061995-10-11 17:30:04 +0000304>>> import parser
305>>> ast = parser.expr('a + 5')
306>>> code = parser.compileast(ast)
307>>> a = 5
308>>> eval(code)
30910
Fred Drake19479911998-02-13 06:58:54 +0000310\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000311%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000312An application which needs both AST and code objects can package this
313code into readily available functions:
314
Fred Drake19479911998-02-13 06:58:54 +0000315\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000316import parser
317
318def load_suite(source_string):
319 ast = parser.suite(source_string)
320 code = parser.compileast(ast)
321 return ast, code
322
323def load_expression(source_string):
324 ast = parser.expr(source_string)
325 code = parser.compileast(ast)
326 return ast, code
Fred Drake19479911998-02-13 06:58:54 +0000327\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000328%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000329\subsubsection{Information Discovery}
330
Fred Drake4b7d5a41996-09-11 21:57:40 +0000331Some applications benefit from direct access to the parse tree. The
332remainder of this section demonstrates how the parse tree provides
333access to module documentation defined in docstrings without requiring
334that the code being examined be loaded into a running interpreter via
Fred Drake88223901998-02-09 20:52:48 +0000335\keyword{import}. This can be very useful for performing analyses of
Fred Drake4b7d5a41996-09-11 21:57:40 +0000336untrusted code.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000337
Guido van Rossum47478871996-08-21 14:32:37 +0000338Generally, the example will demonstrate how the parse tree may be
339traversed to distill interesting information. Two functions and a set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000340of classes are developed which provide programmatic access to high
Guido van Rossum47478871996-08-21 14:32:37 +0000341level function and class definitions provided by a module. The
342classes extract information from the parse tree and provide access to
343the information at a useful semantic level, one function provides a
344simple low-level pattern matching capability, and the other function
345defines a high-level interface to the classes by handling file
346operations on behalf of the caller. All source files mentioned here
347which are not part of the Python installation are located in the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000348\file{Demo/parser/} directory of the distribution.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000349
Guido van Rossum8206fb91996-08-26 00:33:29 +0000350The dynamic nature of Python allows the programmer a great deal of
351flexibility, but most modules need only a limited measure of this when
352defining classes, functions, and methods. In this example, the only
353definitions that will be considered are those which are defined in the
Fred Drake88223901998-02-09 20:52:48 +0000354top level of their context, e.g., a function defined by a \keyword{def}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000355statement at column zero of a module, but not a function defined
Fred Drake4b7d5a41996-09-11 21:57:40 +0000356within a branch of an \code{if} ... \code{else} construct, though
Guido van Rossum8206fb91996-08-26 00:33:29 +0000357there are some good reasons for doing so in some situations. Nesting
358of definitions will be handled by the code developed in the example.
359
Guido van Rossum47478871996-08-21 14:32:37 +0000360To construct the upper-level extraction methods, we need to know what
361the parse tree structure looks like and how much of it we actually
Fred Drake4b7d5a41996-09-11 21:57:40 +0000362need to be concerned about. Python uses a moderately deep parse tree
Guido van Rossum47478871996-08-21 14:32:37 +0000363so there are a large number of intermediate nodes. It is important to
364read and understand the formal grammar used by Python. This is
365specified in the file \file{Grammar/Grammar} in the distribution.
366Consider the simplest case of interest when searching for docstrings:
Guido van Rossum8206fb91996-08-26 00:33:29 +0000367a module consisting of a docstring and nothing else. (See file
368\file{docstring.py}.)
Guido van Rossum4b73a061995-10-11 17:30:04 +0000369
Fred Drake19479911998-02-13 06:58:54 +0000370\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000371"""Some documentation.
372"""
Fred Drake19479911998-02-13 06:58:54 +0000373\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000374%
Guido van Rossum47478871996-08-21 14:32:37 +0000375Using the interpreter to take a look at the parse tree, we find a
376bewildering mass of numbers and parentheses, with the documentation
Fred Drake4b7d5a41996-09-11 21:57:40 +0000377buried deep in nested tuples.
Guido van Rossum4b73a061995-10-11 17:30:04 +0000378
Fred Drake19479911998-02-13 06:58:54 +0000379\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000380>>> import parser
381>>> import pprint
382>>> ast = parser.suite(open('docstring.py').read())
383>>> tup = parser.ast2tuple(ast)
384>>> pprint.pprint(tup)
385(257,
386 (264,
387 (265,
388 (266,
389 (267,
390 (307,
391 (287,
392 (288,
393 (289,
394 (290,
395 (292,
396 (293,
397 (294,
398 (295,
399 (296,
400 (297,
401 (298,
402 (299,
403 (300, (3, '"""Some documentation.\012"""'))))))))))))))))),
404 (4, ''))),
405 (4, ''),
406 (0, ''))
Fred Drake19479911998-02-13 06:58:54 +0000407\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000408%
Guido van Rossum47478871996-08-21 14:32:37 +0000409The numbers at the first element of each node in the tree are the node
410types; they map directly to terminal and non-terminal symbols in the
411grammar. Unfortunately, they are represented as integers in the
412internal representation, and the Python structures generated do not
Fred Drake88223901998-02-09 20:52:48 +0000413change that. However, the \module{symbol} and \module{token} modules
Guido van Rossum47478871996-08-21 14:32:37 +0000414provide symbolic names for the node types and dictionaries which map
415from the integers to the symbolic names for the node types.
416
417In the output presented above, the outermost tuple contains four
418elements: the integer \code{257} and three additional tuples. Node
Fred Drake88223901998-02-09 20:52:48 +0000419type \code{257} has the symbolic name \constant{file_input}. Each of
Guido van Rossum47478871996-08-21 14:32:37 +0000420these inner tuples contains an integer as the first element; these
421integers, \code{264}, \code{4}, and \code{0}, represent the node types
Fred Drake88223901998-02-09 20:52:48 +0000422\constant{stmt}, \constant{NEWLINE}, and \constant{ENDMARKER},
423respectively.
Guido van Rossum47478871996-08-21 14:32:37 +0000424Note that these values may change depending on the version of Python
425you are using; consult \file{symbol.py} and \file{token.py} for
426details of the mapping. It should be fairly clear that the outermost
427node is related primarily to the input source rather than the contents
Fred Drake88223901998-02-09 20:52:48 +0000428of the file, and may be disregarded for the moment. The \constant{stmt}
Guido van Rossum47478871996-08-21 14:32:37 +0000429node is much more interesting. In particular, all docstrings are
430found in subtrees which are formed exactly as this node is formed,
431with the only difference being the string itself. The association
432between the docstring in a similar tree and the defined entity (class,
433function, or module) which it describes is given by the position of
434the docstring subtree within the tree defining the described
435structure.
436
437By replacing the actual docstring with something to signify a variable
Fred Drake4b7d5a41996-09-11 21:57:40 +0000438component of the tree, we allow a simple pattern matching approach to
439check any given subtree for equivelence to the general pattern for
440docstrings. Since the example demonstrates information extraction, we
441can safely require that the tree be in tuple form rather than list
442form, allowing a simple variable representation to be
443\code{['variable_name']}. A simple recursive function can implement
Guido van Rossum47478871996-08-21 14:32:37 +0000444the pattern matching, returning a boolean and a dictionary of variable
Guido van Rossum8206fb91996-08-26 00:33:29 +0000445name to value mappings. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000446
Fred Drake19479911998-02-13 06:58:54 +0000447\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000448from types import ListType, TupleType
449
450def match(pattern, data, vars=None):
451 if vars is None:
452 vars = {}
453 if type(pattern) is ListType:
454 vars[pattern[0]] = data
455 return 1, vars
456 if type(pattern) is not TupleType:
457 return (pattern == data), vars
458 if len(data) != len(pattern):
459 return 0, vars
460 for pattern, data in map(None, pattern, data):
461 same, vars = match(pattern, data, vars)
462 if not same:
463 break
464 return same, vars
Fred Drake19479911998-02-13 06:58:54 +0000465\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000466%
Fred Drake4b7d5a41996-09-11 21:57:40 +0000467Using this simple representation for syntactic variables and the symbolic
Guido van Rossum8206fb91996-08-26 00:33:29 +0000468node types, the pattern for the candidate docstring subtrees becomes
469fairly readable. (See file \file{example.py}.)
Guido van Rossum47478871996-08-21 14:32:37 +0000470
Fred Drake19479911998-02-13 06:58:54 +0000471\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000472import symbol
473import token
474
475DOCSTRING_STMT_PATTERN = (
476 symbol.stmt,
477 (symbol.simple_stmt,
478 (symbol.small_stmt,
479 (symbol.expr_stmt,
480 (symbol.testlist,
481 (symbol.test,
482 (symbol.and_test,
483 (symbol.not_test,
484 (symbol.comparison,
485 (symbol.expr,
486 (symbol.xor_expr,
487 (symbol.and_expr,
488 (symbol.shift_expr,
489 (symbol.arith_expr,
490 (symbol.term,
491 (symbol.factor,
492 (symbol.power,
493 (symbol.atom,
494 (token.STRING, ['docstring'])
495 )))))))))))))))),
496 (token.NEWLINE, '')
497 ))
Fred Drake19479911998-02-13 06:58:54 +0000498\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000499%
Fred Drake88223901998-02-09 20:52:48 +0000500Using the \function{match()} function with this pattern, extracting the
Guido van Rossum47478871996-08-21 14:32:37 +0000501module docstring from the parse tree created previously is easy:
502
Fred Drake19479911998-02-13 06:58:54 +0000503\begin{verbatim}
Guido van Rossum47478871996-08-21 14:32:37 +0000504>>> found, vars = match(DOCSTRING_STMT_PATTERN, tup[1])
505>>> found
5061
507>>> vars
508{'docstring': '"""Some documentation.\012"""'}
Fred Drake19479911998-02-13 06:58:54 +0000509\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000510%
Guido van Rossum47478871996-08-21 14:32:37 +0000511Once specific data can be extracted from a location where it is
512expected, the question of where information can be expected
513needs to be answered. When dealing with docstrings, the answer is
Fred Drake88223901998-02-09 20:52:48 +0000514fairly simple: the docstring is the first \constant{stmt} node in a code
515block (\constant{file_input} or \constant{suite} node types). A module
516consists of a single \constant{file_input} node, and class and function
517definitions each contain exactly one \constant{suite} node. Classes and
Guido van Rossum47478871996-08-21 14:32:37 +0000518functions are readily identified as subtrees of code block nodes which
519start with \code{(stmt, (compound_stmt, (classdef, ...} or
520\code{(stmt, (compound_stmt, (funcdef, ...}. Note that these subtrees
Fred Drake88223901998-02-09 20:52:48 +0000521cannot be matched by \function{match()} since it does not support multiple
Guido van Rossum47478871996-08-21 14:32:37 +0000522sibling nodes to match without regard to number. A more elaborate
523matching function could be used to overcome this limitation, but this
524is sufficient for the example.
525
Guido van Rossum8206fb91996-08-26 00:33:29 +0000526Given the ability to determine whether a statement might be a
527docstring and extract the actual string from the statement, some work
528needs to be performed to walk the parse tree for an entire module and
529extract information about the names defined in each context of the
530module and associate any docstrings with the names. The code to
531perform this work is not complicated, but bears some explanation.
532
533The public interface to the classes is straightforward and should
534probably be somewhat more flexible. Each ``major'' block of the
535module is described by an object providing several methods for inquiry
536and a constructor which accepts at least the subtree of the complete
537parse tree which it represents. The \code{ModuleInfo} constructor
538accepts an optional \code{\var{name}} parameter since it cannot
539otherwise determine the name of the module.
540
Fred Drake88223901998-02-09 20:52:48 +0000541The public classes include \class{ClassInfo}, \class{FunctionInfo},
542and \class{ModuleInfo}. All objects provide the
543methods \method{get_name()}, \method{get_docstring()},
544\method{get_class_names()}, and \method{get_class_info()}. The
545\class{ClassInfo} objects support \method{get_method_names()} and
546\method{get_method_info()} while the other classes provide
547\method{get_function_names()} and \method{get_function_info()}.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000548
549Within each of the forms of code block that the public classes
550represent, most of the required information is in the same form and is
Fred Drake4b7d5a41996-09-11 21:57:40 +0000551accessed in the same way, with classes having the distinction that
Guido van Rossum8206fb91996-08-26 00:33:29 +0000552functions defined at the top level are referred to as ``methods.''
553Since the difference in nomenclature reflects a real semantic
Fred Drake4b7d5a41996-09-11 21:57:40 +0000554distinction from functions defined outside of a class, the
555implementation needs to maintain the distinction.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000556Hence, most of the functionality of the public classes can be
Fred Drake88223901998-02-09 20:52:48 +0000557implemented in a common base class, \class{SuiteInfoBase}, with the
Guido van Rossum8206fb91996-08-26 00:33:29 +0000558accessors for function and method information provided elsewhere.
559Note that there is only one class which represents function and method
Fred Drake88223901998-02-09 20:52:48 +0000560information; this parallels the use of the \keyword{def} statement to
Fred Drake4b7d5a41996-09-11 21:57:40 +0000561define both types of elements.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000562
Fred Drake88223901998-02-09 20:52:48 +0000563Most of the accessor functions are declared in \class{SuiteInfoBase}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000564and do not need to be overriden by subclasses. More importantly, the
565extraction of most information from a parse tree is handled through a
Fred Drake88223901998-02-09 20:52:48 +0000566method called by the \class{SuiteInfoBase} constructor. The example
Guido van Rossum8206fb91996-08-26 00:33:29 +0000567code for most of the classes is clear when read alongside the formal
568grammar, but the method which recursively creates new information
569objects requires further examination. Here is the relevant part of
Fred Drake88223901998-02-09 20:52:48 +0000570the \class{SuiteInfoBase} definition from \file{example.py}:
Guido van Rossum8206fb91996-08-26 00:33:29 +0000571
Fred Drake19479911998-02-13 06:58:54 +0000572\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000573class SuiteInfoBase:
574 _docstring = ''
575 _name = ''
576
577 def __init__(self, tree = None):
578 self._class_info = {}
579 self._function_info = {}
580 if tree:
581 self._extract_info(tree)
582
583 def _extract_info(self, tree):
584 # extract docstring
585 if len(tree) == 2:
586 found, vars = match(DOCSTRING_STMT_PATTERN[1], tree[1])
587 else:
588 found, vars = match(DOCSTRING_STMT_PATTERN, tree[3])
589 if found:
590 self._docstring = eval(vars['docstring'])
591 # discover inner definitions
592 for node in tree[1:]:
593 found, vars = match(COMPOUND_STMT_PATTERN, node)
594 if found:
595 cstmt = vars['compound']
596 if cstmt[0] == symbol.funcdef:
597 name = cstmt[2][1]
598 self._function_info[name] = FunctionInfo(cstmt)
599 elif cstmt[0] == symbol.classdef:
600 name = cstmt[2][1]
601 self._class_info[name] = ClassInfo(cstmt)
Fred Drake19479911998-02-13 06:58:54 +0000602\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000603%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000604After initializing some internal state, the constructor calls the
Fred Drake88223901998-02-09 20:52:48 +0000605\method{_extract_info()} method. This method performs the bulk of the
Guido van Rossum8206fb91996-08-26 00:33:29 +0000606information extraction which takes place in the entire example. The
607extraction has two distinct phases: the location of the docstring for
608the parse tree passed in, and the discovery of additional definitions
609within the code block represented by the parse tree.
610
Fred Drake88223901998-02-09 20:52:48 +0000611The initial \keyword{if} test determines whether the nested suite is of
Guido van Rossum8206fb91996-08-26 00:33:29 +0000612the ``short form'' or the ``long form.'' The short form is used when
613the code block is on the same line as the definition of the code
614block, as in
615
Fred Drakebbe60681998-01-09 22:24:14 +0000616\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000617def square(x): "Square an argument."; return x ** 2
Fred Drakebbe60681998-01-09 22:24:14 +0000618\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000619%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000620while the long form uses an indented block and allows nested
621definitions:
622
Fred Drake19479911998-02-13 06:58:54 +0000623\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000624def make_power(exp):
625 "Make a function that raises an argument to the exponent `exp'."
626 def raiser(x, y=exp):
627 return x ** y
628 return raiser
Fred Drake19479911998-02-13 06:58:54 +0000629\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000630%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000631When the short form is used, the code block may contain a docstring as
Fred Drake88223901998-02-09 20:52:48 +0000632the first, and possibly only, \constant{small_stmt} element. The
Guido van Rossum8206fb91996-08-26 00:33:29 +0000633extraction of such a docstring is slightly different and requires only
634a portion of the complete pattern used in the more common case. As
Fred Drake4b7d5a41996-09-11 21:57:40 +0000635implemented, the docstring will only be found if there is only
Fred Drake88223901998-02-09 20:52:48 +0000636one \constant{small_stmt} node in the \constant{simple_stmt} node.
637Since most functions and methods which use the short form do not
638provide a docstring, this may be considered sufficient. The
639extraction of the docstring proceeds using the \function{match()} function
640as described above, and the value of the docstring is stored as an
641attribute of the \class{SuiteInfoBase} object.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000642
Fred Drake4b7d5a41996-09-11 21:57:40 +0000643After docstring extraction, a simple definition discovery
Fred Drake88223901998-02-09 20:52:48 +0000644algorithm operates on the \constant{stmt} nodes of the
645\constant{suite} node. The special case of the short form is not
646tested; since there are no \constant{stmt} nodes in the short form,
647the algorithm will silently skip the single \constant{simple_stmt}
648node and correctly not discover any nested definitions.
Guido van Rossum8206fb91996-08-26 00:33:29 +0000649
Fred Drake4b7d5a41996-09-11 21:57:40 +0000650Each statement in the code block is categorized as
651a class definition, function or method definition, or
Guido van Rossum8206fb91996-08-26 00:33:29 +0000652something else. For the definition statements, the name of the
Fred Drake4b7d5a41996-09-11 21:57:40 +0000653element defined is extracted and a representation object
Guido van Rossum8206fb91996-08-26 00:33:29 +0000654appropriate to the definition is created with the defining subtree
655passed as an argument to the constructor. The repesentation objects
656are stored in instance variables and may be retrieved by name using
657the appropriate accessor methods.
658
659The public classes provide any accessors required which are more
Fred Drake88223901998-02-09 20:52:48 +0000660specific than those provided by the \class{SuiteInfoBase} class, but
Guido van Rossum8206fb91996-08-26 00:33:29 +0000661the real extraction algorithm remains common to all forms of code
662blocks. A high-level function can be used to extract the complete set
Fred Drake4b7d5a41996-09-11 21:57:40 +0000663of information from a source file. (See file \file{example.py}.)
Guido van Rossum8206fb91996-08-26 00:33:29 +0000664
Fred Drake19479911998-02-13 06:58:54 +0000665\begin{verbatim}
Guido van Rossum8206fb91996-08-26 00:33:29 +0000666def get_docs(fileName):
667 source = open(fileName).read()
668 import os
669 basename = os.path.basename(os.path.splitext(fileName)[0])
670 import parser
671 ast = parser.suite(source)
672 tup = parser.ast2tuple(ast)
673 return ModuleInfo(tup, basename)
Fred Drake19479911998-02-13 06:58:54 +0000674\end{verbatim}
Guido van Rossume47da0a1997-07-17 16:34:52 +0000675%
Guido van Rossum8206fb91996-08-26 00:33:29 +0000676This provides an easy-to-use interface to the documentation of a
677module. If information is required which is not extracted by the code
678of this example, the code may be extended at clearly defined points to
679provide additional capabilities.
Guido van Rossum47478871996-08-21 14:32:37 +0000680
Fred Drakebbe60681998-01-09 22:24:14 +0000681\begin{seealso}
682
683\seemodule{symbol}%
684 {useful constants representing internal nodes of the parse tree}
685
686\seemodule{token}%
687 {useful constants representing leaf nodes of the parse tree and
688functions for testing node values}
689
690\end{seealso}
691
Guido van Rossum47478871996-08-21 14:32:37 +0000692
Fred Drakee061a511997-10-06 21:40:20 +0000693\section{Standard Module \sectcode{symbol}}
Fred Drakebbe60681998-01-09 22:24:14 +0000694\label{module-symbol}
Fred Drakee061a511997-10-06 21:40:20 +0000695\stmodindex{symbol}
696
697This module provides constants which represent the numeric values of
698internal nodes of the parse tree. Unlike most Python constants, these
699use lower-case names. Refer to the file \file{Grammar/Grammar} in the
700Python distribution for the defintions of the names in the context of
701the language grammar. The specific numeric values which the names map
702to may change between Python versions.
703
704This module also provides one additional data object:
705
Fred Drake19479911998-02-13 06:58:54 +0000706\setindexsubitem{(in module symbol)}
Fred Drakee624e0f1997-11-25 04:04:00 +0000707
708
Fred Drakee061a511997-10-06 21:40:20 +0000709\begin{datadesc}{sym_name}
710Dictionary mapping the numeric values of the constants defined in this
711module back to name strings, allowing more human-readable
712representation of parse trees to be generated.
713\end{datadesc}
714
Fred Drakebbe60681998-01-09 22:24:14 +0000715\begin{seealso}
716\seemodule{parser}{second example uses this module}
717\end{seealso}
718
Fred Drakee061a511997-10-06 21:40:20 +0000719
720\section{Standard Module \sectcode{token}}
Fred Drakebbe60681998-01-09 22:24:14 +0000721\label{module-token}
Fred Drakee061a511997-10-06 21:40:20 +0000722\stmodindex{token}
723
724This module provides constants which represent the numeric values of
725leaf nodes of the parse tree (terminal tokens). Refer to the file
726\file{Grammar/Grammar} in the Python distribution for the defintions
727of the names in the context of the language grammar. The specific
728numeric values which the names map to may change between Python
729versions.
730
731This module also provides one data object and some functions. The
732functions mirror definitions in the Python C header files.
733
Fred Drake19479911998-02-13 06:58:54 +0000734\setindexsubitem{(in module token)}
Fred Drakee624e0f1997-11-25 04:04:00 +0000735
736
Fred Drakee061a511997-10-06 21:40:20 +0000737\begin{datadesc}{tok_name}
738Dictionary mapping the numeric values of the constants defined in this
739module back to name strings, allowing more human-readable
740representation of parse trees to be generated.
741\end{datadesc}
742
743\begin{funcdesc}{ISTERMINAL}{x}
744Return true for terminal token values.
745\end{funcdesc}
746
747\begin{funcdesc}{ISNONTERMINAL}{x}
748Return true for non-terminal token values.
749\end{funcdesc}
750
751\begin{funcdesc}{ISEOF}{x}
752Return true if \var{x} is the marker indicating the end of input.
753\end{funcdesc}
754
Fred Drakebbe60681998-01-09 22:24:14 +0000755\begin{seealso}
756\seemodule{parser}{second example uses this module}
757\end{seealso}