blob: d4f4613f480276bec4789b6859f2a2b64777732c [file] [log] [blame]
Jeremy Hyltonfa974a92000-03-06 18:50:48 +00001"""Parse tree transformation module.
2
3Transforms Python source code into an abstract syntax tree (AST)
4defined in the ast module.
5
6The simplest ways to invoke this module are via parse and parseFile.
7parse(buf) -> AST
8parseFile(path) -> AST
9"""
10
Jeremy Hylton9c048f92000-10-13 21:58:13 +000011# Original version written by Greg Stein (gstein@lyra.org)
12# and Bill Tutt (rassilon@lima.mudlib.org)
13# February 1997.
Jeremy Hyltonf968e852000-02-04 00:25:23 +000014#
Jeremy Hylton9c048f92000-10-13 21:58:13 +000015# Modifications and improvements for Python 2.0 by Jeremy Hylton and
16# Mark Hammond
Jeremy Hylton566d9342004-09-07 15:28:01 +000017#
18# Some fixes to try to have correct line number on almost all nodes
19# (except Module, Discard and Stmt) added by Sylvain Thenault
20#
Jeremy Hylton9c048f92000-10-13 21:58:13 +000021# Portions of this file are:
22# Copyright (C) 1997-1998 Greg Stein. All Rights Reserved.
23#
24# This module is provided under a BSD-ish license. See
25# http://www.opensource.org/licenses/bsd-license.html
26# and replace OWNER, ORGANIZATION, and YEAR as appropriate.
27
Jeremy Hylton566d9342004-09-07 15:28:01 +000028from compiler.ast import *
Jeremy Hyltonf968e852000-02-04 00:25:23 +000029import parser
30import symbol
31import token
Jeremy Hyltonf968e852000-02-04 00:25:23 +000032
Jeremy Hylton566d9342004-09-07 15:28:01 +000033class WalkerError(StandardError):
34 pass
Jeremy Hyltonf968e852000-02-04 00:25:23 +000035
Neal Norwitzb902f4e2006-04-03 04:45:34 +000036from compiler.consts import CO_VARARGS, CO_VARKEYWORDS
37from compiler.consts import OP_ASSIGN, OP_DELETE, OP_APPLY
Jeremy Hyltonf968e852000-02-04 00:25:23 +000038
Jeremy Hyltonfa974a92000-03-06 18:50:48 +000039def parseFile(path):
Jeremy Hylton566d9342004-09-07 15:28:01 +000040 f = open(path, "U")
Jeremy Hyltonaccb62b2002-12-31 18:17:44 +000041 # XXX The parser API tolerates files without a trailing newline,
42 # but not strings without a trailing newline. Always add an extra
43 # newline to the file contents, since we're going through the string
44 # version of the API.
45 src = f.read() + "\n"
Jeremy Hyltonfa974a92000-03-06 18:50:48 +000046 f.close()
47 return parse(src)
48
Jeremy Hylton9dca3642001-09-17 21:02:51 +000049def parse(buf, mode="exec"):
50 if mode == "exec" or mode == "single":
51 return Transformer().parsesuite(buf)
52 elif mode == "eval":
53 return Transformer().parseexpr(buf)
54 else:
55 raise ValueError("compile() arg 3 must be"
56 " 'exec' or 'eval' or 'single'")
Jeremy Hyltonfa974a92000-03-06 18:50:48 +000057
Jeremy Hyltonf968e852000-02-04 00:25:23 +000058def asList(nodes):
Jeremy Hylton572bdce2000-09-20 02:47:28 +000059 l = []
60 for item in nodes:
61 if hasattr(item, "asList"):
62 l.append(item.asList())
63 else:
64 if type(item) is type( (None, None) ):
65 l.append(tuple(asList(item)))
66 elif type(item) is type( [] ):
67 l.append(asList(item))
68 else:
69 l.append(item)
70 return l
Jeremy Hyltonf968e852000-02-04 00:25:23 +000071
Jeremy Hylton566d9342004-09-07 15:28:01 +000072def extractLineNo(ast):
73 if not isinstance(ast[1], tuple):
74 # get a terminal node
75 return ast[2]
76 for child in ast[1:]:
77 if isinstance(child, tuple):
78 lineno = extractLineNo(child)
79 if lineno is not None:
80 return lineno
Tim Peters0e9980f2004-09-12 03:49:31 +000081
Jeremy Hyltonf968e852000-02-04 00:25:23 +000082def Node(*args):
Jeremy Hylton572bdce2000-09-20 02:47:28 +000083 kind = args[0]
Georg Brandl2d2fe572008-12-15 08:58:59 +000084 if kind in nodes:
Jeremy Hylton572bdce2000-09-20 02:47:28 +000085 try:
Raymond Hettingerff41c482003-04-06 09:01:11 +000086 return nodes[kind](*args[1:])
Jeremy Hylton572bdce2000-09-20 02:47:28 +000087 except TypeError:
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +000088 print nodes[kind], len(args), args
Jeremy Hylton572bdce2000-09-20 02:47:28 +000089 raise
90 else:
Georg Brandlf0af0e72005-07-02 18:37:41 +000091 raise WalkerError, "Can't find appropriate Node type: %s" % str(args)
Jeremy Hylton572bdce2000-09-20 02:47:28 +000092 #return apply(ast.Node, args)
Jeremy Hyltonf968e852000-02-04 00:25:23 +000093
94class Transformer:
Jeremy Hylton572bdce2000-09-20 02:47:28 +000095 """Utility object for transforming Python parse trees.
Jeremy Hyltonf968e852000-02-04 00:25:23 +000096
Jeremy Hylton572bdce2000-09-20 02:47:28 +000097 Exposes the following methods:
98 tree = transform(ast_tree)
99 tree = parsesuite(text)
100 tree = parseexpr(text)
101 tree = parsefile(fileob | filename)
102 """
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000103
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000104 def __init__(self):
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000105 self._dispatch = {}
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000106 for value, name in symbol.sym_name.items():
107 if hasattr(self, name):
108 self._dispatch[value] = getattr(self, name)
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000109 self._dispatch[token.NEWLINE] = self.com_NEWLINE
110 self._atom_dispatch = {token.LPAR: self.atom_lpar,
111 token.LSQB: self.atom_lsqb,
112 token.LBRACE: self.atom_lbrace,
113 token.BACKQUOTE: self.atom_backquote,
114 token.NUMBER: self.atom_number,
115 token.STRING: self.atom_string,
116 token.NAME: self.atom_name,
117 }
Jeremy Hyltonaccb62b2002-12-31 18:17:44 +0000118 self.encoding = None
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000119
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000120 def transform(self, tree):
121 """Transform an AST into a modified parse tree."""
Jeremy Hylton566d9342004-09-07 15:28:01 +0000122 if not (isinstance(tree, tuple) or isinstance(tree, list)):
Georg Brandl2d2fe572008-12-15 08:58:59 +0000123 tree = parser.st2tuple(tree, line_info=1)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000124 return self.compile_node(tree)
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000125
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000126 def parsesuite(self, text):
127 """Return a modified parse tree for the given suite text."""
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000128 return self.transform(parser.suite(text))
Tim Peters0e9980f2004-09-12 03:49:31 +0000129
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000130 def parseexpr(self, text):
131 """Return a modified parse tree for the given expression text."""
132 return self.transform(parser.expr(text))
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000133
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000134 def parsefile(self, file):
135 """Return a modified parse tree for the contents of the given file."""
136 if type(file) == type(''):
137 file = open(file)
138 return self.parsesuite(file.read())
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000139
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000140 # --------------------------------------------------------------
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000141 #
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000142 # PRIVATE METHODS
143 #
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000144
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000145 def compile_node(self, node):
146 ### emit a line-number node?
147 n = node[0]
Jeremy Hyltonaccb62b2002-12-31 18:17:44 +0000148
149 if n == symbol.encoding_decl:
150 self.encoding = node[2]
151 node = node[1]
152 n = node[0]
Tim Peters669454e2003-03-07 17:30:48 +0000153
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000154 if n == symbol.single_input:
155 return self.single_input(node[1:])
156 if n == symbol.file_input:
157 return self.file_input(node[1:])
158 if n == symbol.eval_input:
159 return self.eval_input(node[1:])
160 if n == symbol.lambdef:
161 return self.lambdef(node[1:])
162 if n == symbol.funcdef:
163 return self.funcdef(node[1:])
164 if n == symbol.classdef:
165 return self.classdef(node[1:])
166
Georg Brandlf0af0e72005-07-02 18:37:41 +0000167 raise WalkerError, ('unexpected node type', n)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000168
169 def single_input(self, node):
170 ### do we want to do anything about being "interactive" ?
171
172 # NEWLINE | simple_stmt | compound_stmt NEWLINE
173 n = node[0][0]
174 if n != token.NEWLINE:
175 return self.com_stmt(node[0])
176
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000177 return Pass()
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000178
179 def file_input(self, nodelist):
180 doc = self.get_docstring(nodelist, symbol.file_input)
Jeremy Hyltonec927342001-08-18 00:04:31 +0000181 if doc is not None:
182 i = 1
183 else:
184 i = 0
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000185 stmts = []
Jeremy Hyltonec927342001-08-18 00:04:31 +0000186 for node in nodelist[i:]:
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000187 if node[0] != token.ENDMARKER and node[0] != token.NEWLINE:
188 self.com_append_stmt(stmts, node)
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000189 return Module(doc, Stmt(stmts))
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000190
191 def eval_input(self, nodelist):
192 # from the built-in function input()
193 ### is this sufficient?
Barry Warsaw52acb492001-12-21 20:04:22 +0000194 return Expression(self.com_node(nodelist[0]))
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000195
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000196 def decorator_name(self, nodelist):
197 listlen = len(nodelist)
198 assert listlen >= 1 and listlen % 2 == 1
199
200 item = self.atom_name(nodelist)
201 i = 1
202 while i < listlen:
203 assert nodelist[i][0] == token.DOT
204 assert nodelist[i + 1][0] == token.NAME
205 item = Getattr(item, nodelist[i + 1][1])
206 i += 2
207
208 return item
Tim Peters6db15d72004-08-04 02:36:18 +0000209
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000210 def decorator(self, nodelist):
211 # '@' dotted_name [ '(' [arglist] ')' ]
Michael W. Hudson0ccff072004-08-17 17:29:16 +0000212 assert len(nodelist) in (3, 5, 6)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000213 assert nodelist[0][0] == token.AT
Michael W. Hudson0ccff072004-08-17 17:29:16 +0000214 assert nodelist[-1][0] == token.NEWLINE
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000215
216 assert nodelist[1][0] == symbol.dotted_name
217 funcname = self.decorator_name(nodelist[1][1:])
218
Michael W. Hudson0ccff072004-08-17 17:29:16 +0000219 if len(nodelist) > 3:
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000220 assert nodelist[2][0] == token.LPAR
221 expr = self.com_call_function(funcname, nodelist[3])
222 else:
223 expr = funcname
Tim Peters6db15d72004-08-04 02:36:18 +0000224
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000225 return expr
Tim Peters6db15d72004-08-04 02:36:18 +0000226
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000227 def decorators(self, nodelist):
228 # decorators: decorator ([NEWLINE] decorator)* NEWLINE
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000229 items = []
Michael W. Hudson0ccff072004-08-17 17:29:16 +0000230 for dec_nodelist in nodelist:
231 assert dec_nodelist[0] == symbol.decorator
232 items.append(self.decorator(dec_nodelist[1:]))
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000233 return Decorators(items)
Tim Peters6db15d72004-08-04 02:36:18 +0000234
Facundo Batista4219da42008-02-25 18:06:00 +0000235 def decorated(self, nodelist):
236 assert nodelist[0][0] == symbol.decorators
237 if nodelist[1][0] == symbol.funcdef:
238 n = [nodelist[0]] + list(nodelist[1][1:])
239 return self.funcdef(n)
240 elif nodelist[1][0] == symbol.classdef:
241 decorators = self.decorators(nodelist[0][1:])
242 cls = self.classdef(nodelist[1][1:])
243 cls.decorators = decorators
244 return cls
245 raise WalkerError()
246
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000247 def funcdef(self, nodelist):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000248 # -6 -5 -4 -3 -2 -1
249 # funcdef: [decorators] 'def' NAME parameters ':' suite
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000250 # parameters: '(' [varargslist] ')'
251
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000252 if len(nodelist) == 6:
253 assert nodelist[0][0] == symbol.decorators
254 decorators = self.decorators(nodelist[0][1:])
255 else:
256 assert len(nodelist) == 5
257 decorators = None
Tim Peters6db15d72004-08-04 02:36:18 +0000258
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000259 lineno = nodelist[-4][2]
260 name = nodelist[-4][1]
261 args = nodelist[-3][2]
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000262
263 if args[0] == symbol.varargslist:
264 names, defaults, flags = self.com_arglist(args[1:])
265 else:
266 names = defaults = ()
267 flags = 0
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000268 doc = self.get_docstring(nodelist[-1])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000269
270 # code for function
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000271 code = self.com_node(nodelist[-1])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000272
Jeremy Hyltona59ac0a2000-11-06 03:33:52 +0000273 if doc is not None:
274 assert isinstance(code, Stmt)
275 assert isinstance(code.nodes[0], Discard)
276 del code.nodes[0]
Jeremy Hylton566d9342004-09-07 15:28:01 +0000277 return Function(decorators, name, names, defaults, flags, doc, code,
278 lineno=lineno)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000279
280 def lambdef(self, nodelist):
281 # lambdef: 'lambda' [varargslist] ':' test
282 if nodelist[2][0] == symbol.varargslist:
283 names, defaults, flags = self.com_arglist(nodelist[2][1:])
284 else:
285 names = defaults = ()
286 flags = 0
287
288 # code for lambda
289 code = self.com_node(nodelist[-1])
290
Jeremy Hylton566d9342004-09-07 15:28:01 +0000291 return Lambda(names, defaults, flags, code, lineno=nodelist[1][2])
Martin v. Löwis49a13022006-02-27 16:03:39 +0000292 old_lambdef = lambdef
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000293
294 def classdef(self, nodelist):
Brett Cannonf4189912005-04-09 02:30:16 +0000295 # classdef: 'class' NAME ['(' [testlist] ')'] ':' suite
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000296
297 name = nodelist[1][1]
298 doc = self.get_docstring(nodelist[-1])
299 if nodelist[2][0] == token.COLON:
300 bases = []
Brett Cannonf4189912005-04-09 02:30:16 +0000301 elif nodelist[3][0] == token.RPAR:
302 bases = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000303 else:
304 bases = self.com_bases(nodelist[3])
305
306 # code for class
307 code = self.com_node(nodelist[-1])
308
Jeremy Hyltonda8db8c2001-08-29 17:19:02 +0000309 if doc is not None:
310 assert isinstance(code, Stmt)
311 assert isinstance(code.nodes[0], Discard)
312 del code.nodes[0]
313
Jeremy Hylton566d9342004-09-07 15:28:01 +0000314 return Class(name, bases, doc, code, lineno=nodelist[1][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000315
316 def stmt(self, nodelist):
317 return self.com_stmt(nodelist[0])
318
319 small_stmt = stmt
320 flow_stmt = stmt
321 compound_stmt = stmt
322
323 def simple_stmt(self, nodelist):
324 # small_stmt (';' small_stmt)* [';'] NEWLINE
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000325 stmts = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000326 for i in range(0, len(nodelist), 2):
327 self.com_append_stmt(stmts, nodelist[i])
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000328 return Stmt(stmts)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000329
330 def parameters(self, nodelist):
Georg Brandlf0af0e72005-07-02 18:37:41 +0000331 raise WalkerError
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000332
333 def varargslist(self, nodelist):
Georg Brandlf0af0e72005-07-02 18:37:41 +0000334 raise WalkerError
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000335
336 def fpdef(self, nodelist):
Georg Brandlf0af0e72005-07-02 18:37:41 +0000337 raise WalkerError
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000338
339 def fplist(self, nodelist):
Georg Brandlf0af0e72005-07-02 18:37:41 +0000340 raise WalkerError
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000341
342 def dotted_name(self, nodelist):
Georg Brandlf0af0e72005-07-02 18:37:41 +0000343 raise WalkerError
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000344
345 def comp_op(self, nodelist):
Georg Brandlf0af0e72005-07-02 18:37:41 +0000346 raise WalkerError
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000347
348 def trailer(self, nodelist):
Georg Brandlf0af0e72005-07-02 18:37:41 +0000349 raise WalkerError
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000350
351 def sliceop(self, nodelist):
Georg Brandlf0af0e72005-07-02 18:37:41 +0000352 raise WalkerError
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000353
354 def argument(self, nodelist):
Georg Brandlf0af0e72005-07-02 18:37:41 +0000355 raise WalkerError
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000356
357 # --------------------------------------------------------------
358 #
359 # STATEMENT NODES (invoked by com_node())
360 #
361
362 def expr_stmt(self, nodelist):
Jeremy Hylton9c048f92000-10-13 21:58:13 +0000363 # augassign testlist | testlist ('=' testlist)*
Jeremy Hylton7e30c9b2001-08-27 21:02:51 +0000364 en = nodelist[-1]
365 exprNode = self.lookup_node(en)(en[1:])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000366 if len(nodelist) == 1:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000367 return Discard(exprNode, lineno=exprNode.lineno)
Jeremy Hylton9c048f92000-10-13 21:58:13 +0000368 if nodelist[1][0] == token.EQUAL:
Martin v. Löwis51e234a2002-02-28 17:48:48 +0000369 nodesl = []
Jeremy Hylton9c048f92000-10-13 21:58:13 +0000370 for i in range(0, len(nodelist) - 2, 2):
Martin v. Löwis51e234a2002-02-28 17:48:48 +0000371 nodesl.append(self.com_assign(nodelist[i], OP_ASSIGN))
Jeremy Hylton566d9342004-09-07 15:28:01 +0000372 return Assign(nodesl, exprNode, lineno=nodelist[1][2])
Jeremy Hylton9c048f92000-10-13 21:58:13 +0000373 else:
374 lval = self.com_augassign(nodelist[0])
375 op = self.com_augassign_op(nodelist[1])
Jeremy Hylton566d9342004-09-07 15:28:01 +0000376 return AugAssign(lval, op[1], exprNode, lineno=op[2])
377 raise WalkerError, "can't get here"
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000378
379 def print_stmt(self, nodelist):
Jeremy Hylton9c048f92000-10-13 21:58:13 +0000380 # print ([ test (',' test)* [','] ] | '>>' test [ (',' test)+ [','] ])
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000381 items = []
Jeremy Hylton9c048f92000-10-13 21:58:13 +0000382 if len(nodelist) == 1:
383 start = 1
384 dest = None
385 elif nodelist[1][0] == token.RIGHTSHIFT:
386 assert len(nodelist) == 3 \
387 or nodelist[3][0] == token.COMMA
388 dest = self.com_node(nodelist[2])
389 start = 4
390 else:
391 dest = None
392 start = 1
393 for i in range(start, len(nodelist), 2):
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000394 items.append(self.com_node(nodelist[i]))
395 if nodelist[-1][0] == token.COMMA:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000396 return Print(items, dest, lineno=nodelist[0][2])
397 return Printnl(items, dest, lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000398
399 def del_stmt(self, nodelist):
400 return self.com_assign(nodelist[1], OP_DELETE)
401
402 def pass_stmt(self, nodelist):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000403 return Pass(lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000404
405 def break_stmt(self, nodelist):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000406 return Break(lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000407
408 def continue_stmt(self, nodelist):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000409 return Continue(lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000410
411 def return_stmt(self, nodelist):
412 # return: [testlist]
413 if len(nodelist) < 2:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000414 return Return(Const(None), lineno=nodelist[0][2])
415 return Return(self.com_node(nodelist[1]), lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000416
Jeremy Hyltonec927342001-08-18 00:04:31 +0000417 def yield_stmt(self, nodelist):
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000418 expr = self.com_node(nodelist[0])
419 return Discard(expr, lineno=expr.lineno)
420
421 def yield_expr(self, nodelist):
Guido van Rossum5bde08d2006-03-02 04:24:01 +0000422 if len(nodelist) > 1:
423 value = self.com_node(nodelist[1])
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000424 else:
425 value = Const(None)
Guido van Rossum5bde08d2006-03-02 04:24:01 +0000426 return Yield(value, lineno=nodelist[0][2])
Jeremy Hyltonec927342001-08-18 00:04:31 +0000427
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000428 def raise_stmt(self, nodelist):
429 # raise: [test [',' test [',' test]]]
430 if len(nodelist) > 5:
431 expr3 = self.com_node(nodelist[5])
432 else:
433 expr3 = None
434 if len(nodelist) > 3:
435 expr2 = self.com_node(nodelist[3])
436 else:
437 expr2 = None
438 if len(nodelist) > 1:
439 expr1 = self.com_node(nodelist[1])
440 else:
441 expr1 = None
Jeremy Hylton566d9342004-09-07 15:28:01 +0000442 return Raise(expr1, expr2, expr3, lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000443
444 def import_stmt(self, nodelist):
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000445 # import_stmt: import_name | import_from
446 assert len(nodelist) == 1
447 return self.com_node(nodelist[0])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000448
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000449 def import_name(self, nodelist):
450 # import_name: 'import' dotted_as_names
Tim Peters0e9980f2004-09-12 03:49:31 +0000451 return Import(self.com_dotted_as_names(nodelist[1]),
Jeremy Hylton566d9342004-09-07 15:28:01 +0000452 lineno=nodelist[0][2])
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000453
454 def import_from(self, nodelist):
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000455 # import_from: 'from' ('.'* dotted_name | '.') 'import' ('*' |
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000456 # '(' import_as_names ')' | import_as_names)
457 assert nodelist[0][1] == 'from'
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000458 idx = 1
459 while nodelist[idx][1] == '.':
460 idx += 1
461 level = idx - 1
462 if nodelist[idx][0] == symbol.dotted_name:
463 fromname = self.com_dotted_name(nodelist[idx])
464 idx += 1
465 else:
466 fromname = ""
467 assert nodelist[idx][1] == 'import'
468 if nodelist[idx + 1][0] == token.STAR:
469 return From(fromname, [('*', None)], level,
Michael W. Hudson05522ad2004-11-08 12:17:34 +0000470 lineno=nodelist[0][2])
Jeremy Hylton9c048f92000-10-13 21:58:13 +0000471 else:
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000472 node = nodelist[idx + 1 + (nodelist[idx + 1][0] == token.LPAR)]
473 return From(fromname, self.com_import_as_names(node), level,
Jeremy Hylton566d9342004-09-07 15:28:01 +0000474 lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000475
476 def global_stmt(self, nodelist):
477 # global: NAME (',' NAME)*
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000478 names = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000479 for i in range(1, len(nodelist), 2):
480 names.append(nodelist[i][1])
Jeremy Hylton566d9342004-09-07 15:28:01 +0000481 return Global(names, lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000482
483 def exec_stmt(self, nodelist):
484 # exec_stmt: 'exec' expr ['in' expr [',' expr]]
485 expr1 = self.com_node(nodelist[1])
486 if len(nodelist) >= 4:
487 expr2 = self.com_node(nodelist[3])
488 if len(nodelist) >= 6:
489 expr3 = self.com_node(nodelist[5])
490 else:
491 expr3 = None
492 else:
493 expr2 = expr3 = None
494
Jeremy Hylton566d9342004-09-07 15:28:01 +0000495 return Exec(expr1, expr2, expr3, lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000496
497 def assert_stmt(self, nodelist):
498 # 'assert': test, [',' test]
499 expr1 = self.com_node(nodelist[1])
500 if (len(nodelist) == 4):
501 expr2 = self.com_node(nodelist[3])
502 else:
Jeremy Hyltona59ac0a2000-11-06 03:33:52 +0000503 expr2 = None
Jeremy Hylton566d9342004-09-07 15:28:01 +0000504 return Assert(expr1, expr2, lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000505
506 def if_stmt(self, nodelist):
507 # if: test ':' suite ('elif' test ':' suite)* ['else' ':' suite]
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000508 tests = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000509 for i in range(0, len(nodelist) - 3, 4):
510 testNode = self.com_node(nodelist[i + 1])
511 suiteNode = self.com_node(nodelist[i + 3])
512 tests.append((testNode, suiteNode))
513
514 if len(nodelist) % 4 == 3:
515 elseNode = self.com_node(nodelist[-1])
516## elseNode.lineno = nodelist[-1][1][2]
517 else:
518 elseNode = None
Jeremy Hylton566d9342004-09-07 15:28:01 +0000519 return If(tests, elseNode, lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000520
521 def while_stmt(self, nodelist):
522 # 'while' test ':' suite ['else' ':' suite]
523
524 testNode = self.com_node(nodelist[1])
525 bodyNode = self.com_node(nodelist[3])
526
527 if len(nodelist) > 4:
528 elseNode = self.com_node(nodelist[6])
529 else:
530 elseNode = None
531
Jeremy Hylton566d9342004-09-07 15:28:01 +0000532 return While(testNode, bodyNode, elseNode, lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000533
534 def for_stmt(self, nodelist):
535 # 'for' exprlist 'in' exprlist ':' suite ['else' ':' suite]
536
537 assignNode = self.com_assign(nodelist[1], OP_ASSIGN)
538 listNode = self.com_node(nodelist[3])
539 bodyNode = self.com_node(nodelist[5])
540
541 if len(nodelist) > 8:
542 elseNode = self.com_node(nodelist[8])
543 else:
544 elseNode = None
545
Tim Peters0e9980f2004-09-12 03:49:31 +0000546 return For(assignNode, listNode, bodyNode, elseNode,
Jeremy Hylton566d9342004-09-07 15:28:01 +0000547 lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000548
549 def try_stmt(self, nodelist):
Georg Brandlad29e632006-06-21 17:45:17 +0000550 return self.com_try_except_finally(nodelist)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000551
Guido van Rossum7ad94f02006-02-28 00:32:16 +0000552 def with_stmt(self, nodelist):
553 return self.com_with(nodelist)
554
555 def with_var(self, nodelist):
556 return self.com_with_var(nodelist)
557
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000558 def suite(self, nodelist):
559 # simple_stmt | NEWLINE INDENT NEWLINE* (stmt NEWLINE*)+ DEDENT
560 if len(nodelist) == 1:
561 return self.com_stmt(nodelist[0])
562
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000563 stmts = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000564 for node in nodelist:
565 if node[0] == symbol.stmt:
566 self.com_append_stmt(stmts, node)
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000567 return Stmt(stmts)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000568
569 # --------------------------------------------------------------
570 #
571 # EXPRESSION NODES (invoked by com_node())
572 #
573
574 def testlist(self, nodelist):
575 # testlist: expr (',' expr)* [',']
Jeremy Hyltond1142612001-10-17 13:32:52 +0000576 # testlist_safe: test [(',' test)+ [',']]
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000577 # exprlist: expr (',' expr)* [',']
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000578 return self.com_binary(Tuple, nodelist)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000579
Jeremy Hyltond1142612001-10-17 13:32:52 +0000580 testlist_safe = testlist # XXX
Jeremy Hyltonaccb62b2002-12-31 18:17:44 +0000581 testlist1 = testlist
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000582 exprlist = testlist
583
Alexandre Vassalottib6465472010-01-11 22:36:12 +0000584 def testlist_comp(self, nodelist):
585 # test ( comp_for | (',' test)* [','] )
586 assert nodelist[0][0] == symbol.test
587 if len(nodelist) == 2 and nodelist[1][0] == symbol.comp_for:
Raymond Hettinger354433a2004-05-19 08:20:33 +0000588 test = self.com_node(nodelist[0])
589 return self.com_generator_expression(test, nodelist[1])
590 return self.testlist(nodelist)
591
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000592 def test(self, nodelist):
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000593 # or_test ['if' or_test 'else' test] | lambdef
594 if len(nodelist) == 1 and nodelist[0][0] == symbol.lambdef:
595 return self.lambdef(nodelist[0])
596 then = self.com_node(nodelist[0])
597 if len(nodelist) > 1:
598 assert len(nodelist) == 5
599 assert nodelist[1][1] == 'if'
600 assert nodelist[3][1] == 'else'
601 test = self.com_node(nodelist[2])
602 else_ = self.com_node(nodelist[4])
603 return IfExp(test, then, else_, lineno=nodelist[1][2])
604 return then
605
606 def or_test(self, nodelist):
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000607 # and_test ('or' and_test)* | lambdef
608 if len(nodelist) == 1 and nodelist[0][0] == symbol.lambdef:
609 return self.lambdef(nodelist[0])
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000610 return self.com_binary(Or, nodelist)
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000611 old_test = or_test
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000612
613 def and_test(self, nodelist):
614 # not_test ('and' not_test)*
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000615 return self.com_binary(And, nodelist)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000616
617 def not_test(self, nodelist):
618 # 'not' not_test | comparison
619 result = self.com_node(nodelist[-1])
620 if len(nodelist) == 2:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000621 return Not(result, lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000622 return result
623
624 def comparison(self, nodelist):
625 # comparison: expr (comp_op expr)*
626 node = self.com_node(nodelist[0])
627 if len(nodelist) == 1:
628 return node
629
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000630 results = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000631 for i in range(2, len(nodelist), 2):
632 nl = nodelist[i-1]
633
634 # comp_op: '<' | '>' | '=' | '>=' | '<=' | '<>' | '!=' | '=='
635 # | 'in' | 'not' 'in' | 'is' | 'is' 'not'
636 n = nl[1]
637 if n[0] == token.NAME:
638 type = n[1]
639 if len(nl) == 3:
640 if type == 'not':
641 type = 'not in'
642 else:
643 type = 'is not'
644 else:
645 type = _cmp_types[n[0]]
646
647 lineno = nl[1][2]
648 results.append((type, self.com_node(nodelist[i])))
649
650 # we need a special "compare" node so that we can distinguish
651 # 3 < x < 5 from (3 < x) < 5
652 # the two have very different semantics and results (note that the
653 # latter form is always true)
654
Jeremy Hylton566d9342004-09-07 15:28:01 +0000655 return Compare(node, results, lineno=lineno)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000656
657 def expr(self, nodelist):
658 # xor_expr ('|' xor_expr)*
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000659 return self.com_binary(Bitor, nodelist)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000660
661 def xor_expr(self, nodelist):
662 # xor_expr ('^' xor_expr)*
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000663 return self.com_binary(Bitxor, nodelist)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000664
665 def and_expr(self, nodelist):
666 # xor_expr ('&' xor_expr)*
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000667 return self.com_binary(Bitand, nodelist)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000668
669 def shift_expr(self, nodelist):
670 # shift_expr ('<<'|'>>' shift_expr)*
671 node = self.com_node(nodelist[0])
672 for i in range(2, len(nodelist), 2):
673 right = self.com_node(nodelist[i])
674 if nodelist[i-1][0] == token.LEFTSHIFT:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000675 node = LeftShift([node, right], lineno=nodelist[1][2])
Jeremy Hylton7abf5202001-08-29 18:12:30 +0000676 elif nodelist[i-1][0] == token.RIGHTSHIFT:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000677 node = RightShift([node, right], lineno=nodelist[1][2])
Jeremy Hylton7abf5202001-08-29 18:12:30 +0000678 else:
679 raise ValueError, "unexpected token: %s" % nodelist[i-1][0]
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000680 return node
681
682 def arith_expr(self, nodelist):
683 node = self.com_node(nodelist[0])
684 for i in range(2, len(nodelist), 2):
685 right = self.com_node(nodelist[i])
686 if nodelist[i-1][0] == token.PLUS:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000687 node = Add([node, right], lineno=nodelist[1][2])
Jeremy Hylton7abf5202001-08-29 18:12:30 +0000688 elif nodelist[i-1][0] == token.MINUS:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000689 node = Sub([node, right], lineno=nodelist[1][2])
Jeremy Hylton7abf5202001-08-29 18:12:30 +0000690 else:
691 raise ValueError, "unexpected token: %s" % nodelist[i-1][0]
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000692 return node
693
694 def term(self, nodelist):
695 node = self.com_node(nodelist[0])
696 for i in range(2, len(nodelist), 2):
697 right = self.com_node(nodelist[i])
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000698 t = nodelist[i-1][0]
699 if t == token.STAR:
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000700 node = Mul([node, right])
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000701 elif t == token.SLASH:
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000702 node = Div([node, right])
Jeremy Hylton7abf5202001-08-29 18:12:30 +0000703 elif t == token.PERCENT:
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000704 node = Mod([node, right])
Jeremy Hylton7abf5202001-08-29 18:12:30 +0000705 elif t == token.DOUBLESLASH:
706 node = FloorDiv([node, right])
707 else:
708 raise ValueError, "unexpected token: %s" % t
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000709 node.lineno = nodelist[1][2]
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000710 return node
711
712 def factor(self, nodelist):
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000713 elt = nodelist[0]
714 t = elt[0]
Jeremy Hylton566d9342004-09-07 15:28:01 +0000715 node = self.lookup_node(nodelist[-1])(nodelist[-1][1:])
Jeremy Hyltonaccb62b2002-12-31 18:17:44 +0000716 # need to handle (unary op)constant here...
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000717 if t == token.PLUS:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000718 return UnaryAdd(node, lineno=elt[2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000719 elif t == token.MINUS:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000720 return UnarySub(node, lineno=elt[2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000721 elif t == token.TILDE:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000722 node = Invert(node, lineno=elt[2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000723 return node
724
725 def power(self, nodelist):
726 # power: atom trailer* ('**' factor)*
727 node = self.com_node(nodelist[0])
728 for i in range(1, len(nodelist)):
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000729 elt = nodelist[i]
730 if elt[0] == token.DOUBLESTAR:
Jeremy Hylton566d9342004-09-07 15:28:01 +0000731 return Power([node, self.com_node(nodelist[i+1])],
732 lineno=elt[2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000733
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000734 node = self.com_apply_trailer(node, elt)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000735
736 return node
737
738 def atom(self, nodelist):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000739 return self._atom_dispatch[nodelist[0][0]](nodelist)
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000740
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000741 def atom_lpar(self, nodelist):
742 if nodelist[1][0] == token.RPAR:
Georg Brandlc3f49ca2006-06-22 14:46:17 +0000743 return Tuple((), lineno=nodelist[0][2])
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000744 return self.com_node(nodelist[1])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000745
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000746 def atom_lsqb(self, nodelist):
747 if nodelist[1][0] == token.RSQB:
Georg Brandlc3f49ca2006-06-22 14:46:17 +0000748 return List((), lineno=nodelist[0][2])
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000749 return self.com_list_constructor(nodelist[1])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000750
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000751 def atom_lbrace(self, nodelist):
752 if nodelist[1][0] == token.RBRACE:
Georg Brandlc3f49ca2006-06-22 14:46:17 +0000753 return Dict((), lineno=nodelist[0][2])
Alexandre Vassalottiee936a22010-01-09 23:35:54 +0000754 return self.com_dictorsetmaker(nodelist[1])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000755
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000756 def atom_backquote(self, nodelist):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000757 return Backquote(self.com_node(nodelist[1]))
Tim Peterse0c446b2001-10-18 21:57:37 +0000758
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000759 def atom_number(self, nodelist):
760 ### need to verify this matches compile.c
761 k = eval(nodelist[0][1])
Jeremy Hylton566d9342004-09-07 15:28:01 +0000762 return Const(k, lineno=nodelist[0][2])
Tim Peterse0c446b2001-10-18 21:57:37 +0000763
Jeremy Hyltonaccb62b2002-12-31 18:17:44 +0000764 def decode_literal(self, lit):
765 if self.encoding:
766 # this is particularly fragile & a bit of a
767 # hack... changes in compile.c:parsestr and
768 # tokenizer.c must be reflected here.
769 if self.encoding not in ['utf-8', 'iso-8859-1']:
770 lit = unicode(lit, 'utf-8').encode(self.encoding)
771 return eval("# coding: %s\n%s" % (self.encoding, lit))
772 else:
773 return eval(lit)
774
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000775 def atom_string(self, nodelist):
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000776 k = ''
777 for node in nodelist:
Jeremy Hyltonaccb62b2002-12-31 18:17:44 +0000778 k += self.decode_literal(node[1])
Jeremy Hylton566d9342004-09-07 15:28:01 +0000779 return Const(k, lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000780
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000781 def atom_name(self, nodelist):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000782 return Name(nodelist[0][1], lineno=nodelist[0][2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000783
784 # --------------------------------------------------------------
785 #
786 # INTERNAL PARSING UTILITIES
787 #
788
Jeremy Hylton7e30c9b2001-08-27 21:02:51 +0000789 # The use of com_node() introduces a lot of extra stack frames,
790 # enough to cause a stack overflow compiling test.test_parser with
791 # the standard interpreter recursionlimit. The com_node() is a
792 # convenience function that hides the dispatch details, but comes
793 # at a very high cost. It is more efficient to dispatch directly
794 # in the callers. In these cases, use lookup_node() and call the
795 # dispatched node directly.
796
797 def lookup_node(self, node):
798 return self._dispatch[node[0]]
799
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000800 def com_node(self, node):
801 # Note: compile.c has handling in com_node for del_stmt, pass_stmt,
802 # break_stmt, stmt, small_stmt, flow_stmt, simple_stmt,
803 # and compound_stmt.
804 # We'll just dispatch them.
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000805 return self._dispatch[node[0]](node[1:])
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000806
Jeremy Hylton42a08302001-04-09 04:27:12 +0000807 def com_NEWLINE(self, *args):
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000808 # A ';' at the end of a line can make a NEWLINE token appear
809 # here, Render it harmless. (genc discards ('discard',
810 # ('const', xxxx)) Nodes)
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +0000811 return Discard(Const(None))
Tim Peterse0c446b2001-10-18 21:57:37 +0000812
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000813 def com_arglist(self, nodelist):
814 # varargslist:
Jeremy Hylton8d22b042002-04-19 22:56:37 +0000815 # (fpdef ['=' test] ',')* ('*' NAME [',' '**' NAME] | '**' NAME)
816 # | fpdef ['=' test] (',' fpdef ['=' test])* [',']
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000817 # fpdef: NAME | '(' fplist ')'
818 # fplist: fpdef (',' fpdef)* [',']
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000819 names = []
820 defaults = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000821 flags = 0
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000822
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000823 i = 0
824 while i < len(nodelist):
825 node = nodelist[i]
826 if node[0] == token.STAR or node[0] == token.DOUBLESTAR:
827 if node[0] == token.STAR:
828 node = nodelist[i+1]
829 if node[0] == token.NAME:
830 names.append(node[1])
831 flags = flags | CO_VARARGS
832 i = i + 3
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000833
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000834 if i < len(nodelist):
Jeremy Hylton8d22b042002-04-19 22:56:37 +0000835 # should be DOUBLESTAR
Jeremy Hylton7abf5202001-08-29 18:12:30 +0000836 t = nodelist[i][0]
837 if t == token.DOUBLESTAR:
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000838 node = nodelist[i+1]
Jeremy Hylton7abf5202001-08-29 18:12:30 +0000839 else:
840 raise ValueError, "unexpected token: %s" % t
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000841 names.append(node[1])
842 flags = flags | CO_VARKEYWORDS
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000843
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000844 break
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000845
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000846 # fpdef: NAME | '(' fplist ')'
847 names.append(self.com_fpdef(node))
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000848
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000849 i = i + 1
Georg Brandl1bb62302006-05-03 18:18:32 +0000850 if i < len(nodelist) and nodelist[i][0] == token.EQUAL:
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000851 defaults.append(self.com_node(nodelist[i + 1]))
852 i = i + 2
853 elif len(defaults):
Georg Brandl1bb62302006-05-03 18:18:32 +0000854 # we have already seen an argument with default, but here
855 # came one without
856 raise SyntaxError, "non-default argument follows default argument"
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000857
Georg Brandl1bb62302006-05-03 18:18:32 +0000858 # skip the comma
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000859 i = i + 1
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000860
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000861 return names, defaults, flags
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000862
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000863 def com_fpdef(self, node):
864 # fpdef: NAME | '(' fplist ')'
865 if node[1][0] == token.LPAR:
866 return self.com_fplist(node[2])
867 return node[1][1]
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000868
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000869 def com_fplist(self, node):
Tim Peterse0c446b2001-10-18 21:57:37 +0000870 # fplist: fpdef (',' fpdef)* [',']
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000871 if len(node) == 2:
872 return self.com_fpdef(node[1])
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000873 list = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000874 for i in range(1, len(node), 2):
875 list.append(self.com_fpdef(node[i]))
876 return tuple(list)
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000877
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000878 def com_dotted_name(self, node):
879 # String together the dotted names and return the string
880 name = ""
881 for n in node:
882 if type(n) == type(()) and n[0] == 1:
883 name = name + n[1] + '.'
884 return name[:-1]
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000885
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000886 def com_dotted_as_name(self, node):
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000887 assert node[0] == symbol.dotted_as_name
888 node = node[1:]
889 dot = self.com_dotted_name(node[0][1:])
890 if len(node) == 1:
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000891 return dot, None
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000892 assert node[1][1] == 'as'
893 assert node[2][0] == token.NAME
894 return dot, node[2][1]
895
896 def com_dotted_as_names(self, node):
897 assert node[0] == symbol.dotted_as_names
898 node = node[1:]
899 names = [self.com_dotted_as_name(node[0])]
900 for i in range(2, len(node), 2):
901 names.append(self.com_dotted_as_name(node[i]))
902 return names
Jeremy Hylton20516082000-09-01 20:33:26 +0000903
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000904 def com_import_as_name(self, node):
Jeremy Hylton42a08302001-04-09 04:27:12 +0000905 assert node[0] == symbol.import_as_name
906 node = node[1:]
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000907 assert node[0][0] == token.NAME
Jeremy Hylton42a08302001-04-09 04:27:12 +0000908 if len(node) == 1:
Jeremy Hylton42a08302001-04-09 04:27:12 +0000909 return node[0][1], None
Jeremy Hylton42a08302001-04-09 04:27:12 +0000910 assert node[1][1] == 'as', node
911 assert node[2][0] == token.NAME
912 return node[0][1], node[2][1]
Jeremy Hylton20516082000-09-01 20:33:26 +0000913
Anthony Baxter1a4ddae2004-08-31 10:07:13 +0000914 def com_import_as_names(self, node):
915 assert node[0] == symbol.import_as_names
916 node = node[1:]
917 names = [self.com_import_as_name(node[0])]
918 for i in range(2, len(node), 2):
919 names.append(self.com_import_as_name(node[i]))
920 return names
921
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000922 def com_bases(self, node):
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000923 bases = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000924 for i in range(1, len(node), 2):
925 bases.append(self.com_node(node[i]))
926 return bases
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000927
Georg Brandlad29e632006-06-21 17:45:17 +0000928 def com_try_except_finally(self, nodelist):
929 # ('try' ':' suite
930 # ((except_clause ':' suite)+ ['else' ':' suite] ['finally' ':' suite]
931 # | 'finally' ':' suite))
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000932
Georg Brandlad29e632006-06-21 17:45:17 +0000933 if nodelist[3][0] == token.NAME:
934 # first clause is a finally clause: only try-finally
935 return TryFinally(self.com_node(nodelist[2]),
936 self.com_node(nodelist[5]),
937 lineno=nodelist[0][2])
Tim Peters3249d002006-06-27 11:52:49 +0000938
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000939 #tryexcept: [TryNode, [except_clauses], elseNode)]
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000940 clauses = []
941 elseNode = None
Georg Brandlad29e632006-06-21 17:45:17 +0000942 finallyNode = None
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000943 for i in range(3, len(nodelist), 3):
944 node = nodelist[i]
945 if node[0] == symbol.except_clause:
Collin Winter62903052007-05-18 23:11:24 +0000946 # except_clause: 'except' [expr [(',' | 'as') expr]] */
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000947 if len(node) > 2:
948 expr1 = self.com_node(node[2])
949 if len(node) > 4:
950 expr2 = self.com_assign(node[4], OP_ASSIGN)
951 else:
952 expr2 = None
953 else:
954 expr1 = expr2 = None
955 clauses.append((expr1, expr2, self.com_node(nodelist[i+2])))
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000956
Jeremy Hylton572bdce2000-09-20 02:47:28 +0000957 if node[0] == token.NAME:
Georg Brandlad29e632006-06-21 17:45:17 +0000958 if node[1] == 'else':
959 elseNode = self.com_node(nodelist[i+2])
960 elif node[1] == 'finally':
961 finallyNode = self.com_node(nodelist[i+2])
962 try_except = TryExcept(self.com_node(nodelist[2]), clauses, elseNode,
963 lineno=nodelist[0][2])
964 if finallyNode:
965 return TryFinally(try_except, finallyNode, lineno=nodelist[0][2])
966 else:
967 return try_except
Jeremy Hyltonf968e852000-02-04 00:25:23 +0000968
Guido van Rossum7ad94f02006-02-28 00:32:16 +0000969 def com_with(self, nodelist):
Georg Brandl944f6842009-05-25 21:02:56 +0000970 # with_stmt: 'with' with_item (',' with_item)* ':' suite
Guido van Rossum7ad94f02006-02-28 00:32:16 +0000971 body = self.com_node(nodelist[-1])
Georg Brandl944f6842009-05-25 21:02:56 +0000972 for i in range(len(nodelist) - 3, 0, -2):
973 ret = self.com_with_item(nodelist[i], body, nodelist[0][2])
974 if i == 1:
975 return ret
976 body = ret
Guido van Rossum7ad94f02006-02-28 00:32:16 +0000977
Georg Brandl944f6842009-05-25 21:02:56 +0000978 def com_with_item(self, nodelist, body, lineno):
979 # with_item: test ['as' expr]
980 if len(nodelist) == 4:
981 var = self.com_assign(nodelist[3], OP_ASSIGN)
982 else:
983 var = None
984 expr = self.com_node(nodelist[1])
985 return With(expr, var, body, lineno=lineno)
Guido van Rossum7ad94f02006-02-28 00:32:16 +0000986
Jeremy Hylton9c048f92000-10-13 21:58:13 +0000987 def com_augassign_op(self, node):
988 assert node[0] == symbol.augassign
989 return node[1]
990
991 def com_augassign(self, node):
992 """Return node suitable for lvalue of augmented assignment
993
994 Names, slices, and attributes are the only allowable nodes.
995 """
996 l = self.com_node(node)
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000997 if l.__class__ in (Name, Slice, Subscript, Getattr):
Jeremy Hylton9c048f92000-10-13 21:58:13 +0000998 return l
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +0000999 raise SyntaxError, "can't assign to %s" % l.__class__.__name__
Jeremy Hylton9c048f92000-10-13 21:58:13 +00001000
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001001 def com_assign(self, node, assigning):
1002 # return a node suitable for use as an "lvalue"
1003 # loop to avoid trivial recursion
1004 while 1:
1005 t = node[0]
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001006 if t in (symbol.exprlist, symbol.testlist, symbol.testlist_safe, symbol.testlist_comp):
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001007 if len(node) > 2:
1008 return self.com_assign_tuple(node, assigning)
1009 node = node[1]
1010 elif t in _assign_types:
1011 if len(node) > 2:
1012 raise SyntaxError, "can't assign to operator"
1013 node = node[1]
1014 elif t == symbol.power:
1015 if node[1][0] != symbol.atom:
1016 raise SyntaxError, "can't assign to operator"
1017 if len(node) > 2:
1018 primary = self.com_node(node[1])
1019 for i in range(2, len(node)-1):
1020 ch = node[i]
1021 if ch[0] == token.DOUBLESTAR:
1022 raise SyntaxError, "can't assign to operator"
1023 primary = self.com_apply_trailer(primary, ch)
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001024 return self.com_assign_trailer(primary, node[-1],
Tim Peterse0c446b2001-10-18 21:57:37 +00001025 assigning)
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001026 node = node[1]
1027 elif t == symbol.atom:
1028 t = node[1][0]
1029 if t == token.LPAR:
1030 node = node[2]
1031 if node[0] == token.RPAR:
1032 raise SyntaxError, "can't assign to ()"
1033 elif t == token.LSQB:
1034 node = node[2]
1035 if node[0] == token.RSQB:
1036 raise SyntaxError, "can't assign to []"
1037 return self.com_assign_list(node, assigning)
1038 elif t == token.NAME:
1039 return self.com_assign_name(node[1], assigning)
1040 else:
1041 raise SyntaxError, "can't assign to literal"
1042 else:
Martin v. Löwis16c7f712006-02-27 16:11:03 +00001043 raise SyntaxError, "bad assignment (%s)" % t
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001044
1045 def com_assign_tuple(self, node, assigning):
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001046 assigns = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001047 for i in range(1, len(node), 2):
1048 assigns.append(self.com_assign(node[i], assigning))
Jeremy Hylton566d9342004-09-07 15:28:01 +00001049 return AssTuple(assigns, lineno=extractLineNo(node))
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001050
1051 def com_assign_list(self, node, assigning):
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001052 assigns = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001053 for i in range(1, len(node), 2):
Jeremy Hylton2e4cc7e2001-09-17 19:33:48 +00001054 if i + 1 < len(node):
1055 if node[i + 1][0] == symbol.list_for:
1056 raise SyntaxError, "can't assign to list comprehension"
1057 assert node[i + 1][0] == token.COMMA, node[i + 1]
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001058 assigns.append(self.com_assign(node[i], assigning))
Jeremy Hylton566d9342004-09-07 15:28:01 +00001059 return AssList(assigns, lineno=extractLineNo(node))
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001060
1061 def com_assign_name(self, node, assigning):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001062 return AssName(node[1], assigning, lineno=node[2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001063
1064 def com_assign_trailer(self, primary, node, assigning):
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001065 t = node[1][0]
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001066 if t == token.DOT:
1067 return self.com_assign_attr(primary, node[2], assigning)
1068 if t == token.LSQB:
1069 return self.com_subscriptlist(primary, node[2], assigning)
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +00001070 if t == token.LPAR:
1071 raise SyntaxError, "can't assign to function call"
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001072 raise SyntaxError, "unknown trailer type: %s" % t
1073
1074 def com_assign_attr(self, primary, node, assigning):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001075 return AssAttr(primary, node[1], assigning, lineno=node[-1])
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001076
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +00001077 def com_binary(self, constructor, nodelist):
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001078 "Compile 'NODE (OP NODE)*' into (type, [ node1, ..., nodeN ])."
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001079 l = len(nodelist)
1080 if l == 1:
Jeremy Hylton7e30c9b2001-08-27 21:02:51 +00001081 n = nodelist[0]
1082 return self.lookup_node(n)(n[1:])
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001083 items = []
1084 for i in range(0, l, 2):
Jeremy Hylton7e30c9b2001-08-27 21:02:51 +00001085 n = nodelist[i]
1086 items.append(self.lookup_node(n)(n[1:]))
Jeremy Hylton566d9342004-09-07 15:28:01 +00001087 return constructor(items, lineno=extractLineNo(nodelist))
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001088
1089 def com_stmt(self, node):
Jeremy Hylton7e30c9b2001-08-27 21:02:51 +00001090 result = self.lookup_node(node)(node[1:])
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001091 assert result is not None
1092 if isinstance(result, Stmt):
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001093 return result
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001094 return Stmt([result])
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001095
1096 def com_append_stmt(self, stmts, node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001097 result = self.lookup_node(node)(node[1:])
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001098 assert result is not None
1099 if isinstance(result, Stmt):
1100 stmts.extend(result.nodes)
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001101 else:
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001102 stmts.append(result)
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001103
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001104 def com_list_constructor(self, nodelist):
1105 # listmaker: test ( list_for | (',' test)* [','] )
1106 values = []
1107 for i in range(1, len(nodelist)):
1108 if nodelist[i][0] == symbol.list_for:
1109 assert len(nodelist[i:]) == 1
1110 return self.com_list_comprehension(values[0],
1111 nodelist[i])
1112 elif nodelist[i][0] == token.COMMA:
1113 continue
1114 values.append(self.com_node(nodelist[i]))
1115 return List(values, lineno=values[0].lineno)
Jeremy Hylton9c048f92000-10-13 21:58:13 +00001116
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001117 def com_list_comprehension(self, expr, node):
1118 return self.com_comprehension(expr, None, node, 'list')
Jeremy Hylton42a08302001-04-09 04:27:12 +00001119
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001120 def com_comprehension(self, expr1, expr2, node, type):
1121 # list_iter: list_for | list_if
1122 # list_for: 'for' exprlist 'in' testlist [list_iter]
1123 # list_if: 'if' test [list_iter]
Tim Peterse0c446b2001-10-18 21:57:37 +00001124
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001125 # XXX should raise SyntaxError for assignment
1126 # XXX(avassalotti) Set and dict comprehensions should have generator
1127 # semantics. In other words, they shouldn't leak
1128 # variables outside of the comprehension's scope.
1129
1130 lineno = node[1][2]
1131 fors = []
1132 while node:
1133 t = node[1][1]
1134 if t == 'for':
1135 assignNode = self.com_assign(node[2], OP_ASSIGN)
1136 compNode = self.com_node(node[4])
1137 newfor = ListCompFor(assignNode, compNode, [])
1138 newfor.lineno = node[1][2]
1139 fors.append(newfor)
1140 if len(node) == 5:
1141 node = None
1142 elif type == 'list':
1143 node = self.com_list_iter(node[5])
Jeremy Hylton9c048f92000-10-13 21:58:13 +00001144 else:
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001145 node = self.com_comp_iter(node[5])
1146 elif t == 'if':
1147 test = self.com_node(node[2])
1148 newif = ListCompIf(test, lineno=node[1][2])
1149 newfor.ifs.append(newif)
1150 if len(node) == 3:
1151 node = None
1152 elif type == 'list':
1153 node = self.com_list_iter(node[3])
Raymond Hettinger354433a2004-05-19 08:20:33 +00001154 else:
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001155 node = self.com_comp_iter(node[3])
1156 else:
1157 raise SyntaxError, \
1158 ("unexpected comprehension element: %s %d"
1159 % (node, lineno))
1160 if type == 'list':
1161 return ListComp(expr1, fors, lineno=lineno)
1162 elif type == 'set':
1163 return SetComp(expr1, fors, lineno=lineno)
1164 elif type == 'dict':
1165 return DictComp(expr1, expr2, fors, lineno=lineno)
1166 else:
1167 raise ValueError("unexpected comprehension type: " + repr(type))
Raymond Hettinger354433a2004-05-19 08:20:33 +00001168
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001169 def com_list_iter(self, node):
1170 assert node[0] == symbol.list_iter
1171 return node[1]
1172
1173 def com_comp_iter(self, node):
1174 assert node[0] == symbol.comp_iter
1175 return node[1]
1176
1177 def com_generator_expression(self, expr, node):
1178 # comp_iter: comp_for | comp_if
1179 # comp_for: 'for' exprlist 'in' test [comp_iter]
1180 # comp_if: 'if' test [comp_iter]
1181
1182 lineno = node[1][2]
1183 fors = []
1184 while node:
1185 t = node[1][1]
1186 if t == 'for':
1187 assignNode = self.com_assign(node[2], OP_ASSIGN)
1188 genNode = self.com_node(node[4])
1189 newfor = GenExprFor(assignNode, genNode, [],
1190 lineno=node[1][2])
1191 fors.append(newfor)
1192 if (len(node)) == 5:
1193 node = None
1194 else:
1195 node = self.com_comp_iter(node[5])
1196 elif t == 'if':
1197 test = self.com_node(node[2])
1198 newif = GenExprIf(test, lineno=node[1][2])
1199 newfor.ifs.append(newif)
1200 if len(node) == 3:
1201 node = None
1202 else:
1203 node = self.com_comp_iter(node[3])
1204 else:
1205 raise SyntaxError, \
1206 ("unexpected generator expression element: %s %d"
1207 % (node, lineno))
1208 fors[0].is_outmost = True
1209 return GenExpr(GenExprInner(expr, fors), lineno=lineno)
Raymond Hettinger354433a2004-05-19 08:20:33 +00001210
Alexandre Vassalottiee936a22010-01-09 23:35:54 +00001211 def com_dictorsetmaker(self, nodelist):
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001212 # dictorsetmaker: ( (test ':' test (comp_for | (',' test ':' test)* [','])) |
1213 # (test (comp_for | (',' test)* [','])) )
Alexandre Vassalottiee936a22010-01-09 23:35:54 +00001214 assert nodelist[0] == symbol.dictorsetmaker
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001215 nodelist = nodelist[1:]
1216 if len(nodelist) == 1 or nodelist[1][0] == token.COMMA:
1217 # set literal
Alexandre Vassalottiee936a22010-01-09 23:35:54 +00001218 items = []
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001219 for i in range(0, len(nodelist), 2):
Alexandre Vassalottiee936a22010-01-09 23:35:54 +00001220 items.append(self.com_node(nodelist[i]))
1221 return Set(items, lineno=items[0].lineno)
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001222 elif nodelist[1][0] == symbol.comp_for:
1223 # set comprehension
1224 expr = self.com_node(nodelist[0])
1225 return self.com_comprehension(expr, None, nodelist[1], 'set')
1226 elif len(nodelist) > 3 and nodelist[3][0] == symbol.comp_for:
1227 # dict comprehension
1228 assert nodelist[1][0] == token.COLON
1229 key = self.com_node(nodelist[0])
1230 value = self.com_node(nodelist[2])
1231 return self.com_comprehension(key, value, nodelist[3], 'dict')
Alexandre Vassalottiee936a22010-01-09 23:35:54 +00001232 else:
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001233 # dict literal
Alexandre Vassalottiee936a22010-01-09 23:35:54 +00001234 items = []
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001235 for i in range(0, len(nodelist), 4):
Alexandre Vassalottiee936a22010-01-09 23:35:54 +00001236 items.append((self.com_node(nodelist[i]),
1237 self.com_node(nodelist[i+2])))
1238 return Dict(items, lineno=items[0][0].lineno)
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001239
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001240 def com_apply_trailer(self, primaryNode, nodelist):
1241 t = nodelist[1][0]
1242 if t == token.LPAR:
1243 return self.com_call_function(primaryNode, nodelist[2])
1244 if t == token.DOT:
1245 return self.com_select_member(primaryNode, nodelist[2])
1246 if t == token.LSQB:
1247 return self.com_subscriptlist(primaryNode, nodelist[2], OP_APPLY)
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001248
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001249 raise SyntaxError, 'unknown node type: %s' % t
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001250
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001251 def com_select_member(self, primaryNode, nodelist):
1252 if nodelist[0] != token.NAME:
1253 raise SyntaxError, "member must be a name"
Jeremy Hylton566d9342004-09-07 15:28:01 +00001254 return Getattr(primaryNode, nodelist[1], lineno=nodelist[2])
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001255
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001256 def com_call_function(self, primaryNode, nodelist):
1257 if nodelist[0] == token.RPAR:
Jeremy Hylton566d9342004-09-07 15:28:01 +00001258 return CallFunc(primaryNode, [], lineno=extractLineNo(nodelist))
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001259 args = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001260 kw = 0
Amaury Forgeot d'Arc67f24f12008-08-20 00:08:47 +00001261 star_node = dstar_node = None
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001262 len_nodelist = len(nodelist)
Amaury Forgeot d'Arc67f24f12008-08-20 00:08:47 +00001263 i = 1
1264 while i < len_nodelist:
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001265 node = nodelist[i]
Amaury Forgeot d'Arc67f24f12008-08-20 00:08:47 +00001266
1267 if node[0]==token.STAR:
1268 if star_node is not None:
1269 raise SyntaxError, 'already have the varargs indentifier'
1270 star_node = self.com_node(nodelist[i+1])
1271 i = i + 3
1272 continue
1273 elif node[0]==token.DOUBLESTAR:
1274 if dstar_node is not None:
1275 raise SyntaxError, 'already have the kwargs indentifier'
1276 dstar_node = self.com_node(nodelist[i+1])
1277 i = i + 3
1278 continue
1279
1280 # positional or named parameters
1281 kw, result = self.com_argument(node, kw, star_node)
Michael W. Hudsonc054a8b2004-10-11 15:35:53 +00001282
1283 if len_nodelist != 2 and isinstance(result, GenExpr) \
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001284 and len(node) == 3 and node[2][0] == symbol.comp_for:
Michael W. Hudsonc054a8b2004-10-11 15:35:53 +00001285 # allow f(x for x in y), but reject f(x for x in y, 1)
1286 # should use f((x for x in y), 1) instead of f(x for x in y, 1)
Raymond Hettinger354433a2004-05-19 08:20:33 +00001287 raise SyntaxError, 'generator expression needs parenthesis'
Michael W. Hudsonc054a8b2004-10-11 15:35:53 +00001288
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001289 args.append(result)
Amaury Forgeot d'Arc67f24f12008-08-20 00:08:47 +00001290 i = i + 2
1291
Jeremy Hylton566d9342004-09-07 15:28:01 +00001292 return CallFunc(primaryNode, args, star_node, dstar_node,
1293 lineno=extractLineNo(nodelist))
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001294
Amaury Forgeot d'Arc67f24f12008-08-20 00:08:47 +00001295 def com_argument(self, nodelist, kw, star_node):
Alexandre Vassalottib6465472010-01-11 22:36:12 +00001296 if len(nodelist) == 3 and nodelist[2][0] == symbol.comp_for:
Raymond Hettinger354433a2004-05-19 08:20:33 +00001297 test = self.com_node(nodelist[1])
1298 return 0, self.com_generator_expression(test, nodelist[2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001299 if len(nodelist) == 2:
1300 if kw:
1301 raise SyntaxError, "non-keyword arg after keyword arg"
Amaury Forgeot d'Arc67f24f12008-08-20 00:08:47 +00001302 if star_node:
1303 raise SyntaxError, "only named arguments may follow *expression"
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001304 return 0, self.com_node(nodelist[1])
1305 result = self.com_node(nodelist[3])
1306 n = nodelist[1]
1307 while len(n) == 2 and n[0] != token.NAME:
1308 n = n[1]
1309 if n[0] != token.NAME:
1310 raise SyntaxError, "keyword can't be an expression (%s)"%n[0]
Jeremy Hylton566d9342004-09-07 15:28:01 +00001311 node = Keyword(n[1], result, lineno=n[2])
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001312 return 1, node
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001313
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001314 def com_subscriptlist(self, primary, nodelist, assigning):
1315 # slicing: simple_slicing | extended_slicing
1316 # simple_slicing: primary "[" short_slice "]"
Tim Peterse0c446b2001-10-18 21:57:37 +00001317 # extended_slicing: primary "[" slice_list "]"
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001318 # slice_list: slice_item ("," slice_item)* [","]
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001319
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001320 # backwards compat slice for '[i:j]'
1321 if len(nodelist) == 2:
1322 sub = nodelist[1]
1323 if (sub[1][0] == token.COLON or \
1324 (len(sub) > 2 and sub[2][0] == token.COLON)) and \
1325 sub[-1][0] != symbol.sliceop:
1326 return self.com_slice(primary, sub, assigning)
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001327
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001328 subscripts = []
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001329 for i in range(1, len(nodelist), 2):
1330 subscripts.append(self.com_subscript(nodelist[i]))
Tim Peters0e9980f2004-09-12 03:49:31 +00001331 return Subscript(primary, assigning, subscripts,
Jeremy Hylton566d9342004-09-07 15:28:01 +00001332 lineno=extractLineNo(nodelist))
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001333
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001334 def com_subscript(self, node):
1335 # slice_item: expression | proper_slice | ellipsis
1336 ch = node[1]
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +00001337 t = ch[0]
1338 if t == token.DOT and node[2][0] == token.DOT:
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001339 return Ellipsis()
Jeremy Hylton4ebf3be2000-11-06 16:03:52 +00001340 if t == token.COLON or len(node) > 2:
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001341 return self.com_sliceobj(node)
1342 return self.com_node(ch)
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001343
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001344 def com_sliceobj(self, node):
1345 # proper_slice: short_slice | long_slice
1346 # short_slice: [lower_bound] ":" [upper_bound]
1347 # long_slice: short_slice ":" [stride]
1348 # lower_bound: expression
1349 # upper_bound: expression
1350 # stride: expression
1351 #
1352 # Note: a stride may be further slicing...
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001353
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001354 items = []
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001355
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001356 if node[1][0] == token.COLON:
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001357 items.append(Const(None))
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001358 i = 2
1359 else:
1360 items.append(self.com_node(node[1]))
1361 # i == 2 is a COLON
1362 i = 3
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001363
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001364 if i < len(node) and node[i][0] == symbol.test:
1365 items.append(self.com_node(node[i]))
1366 i = i + 1
1367 else:
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001368 items.append(Const(None))
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001369
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001370 # a short_slice has been built. look for long_slice now by looking
1371 # for strides...
1372 for j in range(i, len(node)):
1373 ch = node[j]
1374 if len(ch) == 2:
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001375 items.append(Const(None))
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001376 else:
1377 items.append(self.com_node(ch[2]))
Jeremy Hylton566d9342004-09-07 15:28:01 +00001378 return Sliceobj(items, lineno=extractLineNo(node))
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001379
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001380 def com_slice(self, primary, node, assigning):
1381 # short_slice: [lower_bound] ":" [upper_bound]
1382 lower = upper = None
1383 if len(node) == 3:
1384 if node[1][0] == token.COLON:
1385 upper = self.com_node(node[2])
1386 else:
1387 lower = self.com_node(node[1])
1388 elif len(node) == 4:
1389 lower = self.com_node(node[1])
1390 upper = self.com_node(node[3])
Jeremy Hylton566d9342004-09-07 15:28:01 +00001391 return Slice(primary, assigning, lower, upper,
1392 lineno=extractLineNo(node))
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001393
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001394 def get_docstring(self, node, n=None):
1395 if n is None:
1396 n = node[0]
1397 node = node[1:]
1398 if n == symbol.suite:
1399 if len(node) == 1:
1400 return self.get_docstring(node[0])
1401 for sub in node:
1402 if sub[0] == symbol.stmt:
1403 return self.get_docstring(sub)
1404 return None
1405 if n == symbol.file_input:
1406 for sub in node:
1407 if sub[0] == symbol.stmt:
1408 return self.get_docstring(sub)
1409 return None
1410 if n == symbol.atom:
1411 if node[0][0] == token.STRING:
1412 s = ''
1413 for t in node:
1414 s = s + eval(t[1])
1415 return s
1416 return None
Jeremy Hylton7cff7fe2000-10-25 18:10:32 +00001417 if n == symbol.stmt or n == symbol.simple_stmt \
1418 or n == symbol.small_stmt:
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001419 return self.get_docstring(node[0])
1420 if n in _doc_nodes and len(node) == 1:
1421 return self.get_docstring(node[0])
1422 return None
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001423
1424
1425_doc_nodes = [
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001426 symbol.expr_stmt,
1427 symbol.testlist,
Jeremy Hyltond1142612001-10-17 13:32:52 +00001428 symbol.testlist_safe,
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001429 symbol.test,
Georg Brandledd9b0d2006-07-29 09:33:26 +00001430 symbol.or_test,
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001431 symbol.and_test,
1432 symbol.not_test,
1433 symbol.comparison,
1434 symbol.expr,
1435 symbol.xor_expr,
1436 symbol.and_expr,
1437 symbol.shift_expr,
1438 symbol.arith_expr,
1439 symbol.term,
1440 symbol.factor,
1441 symbol.power,
1442 ]
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001443
1444# comp_op: '<' | '>' | '=' | '>=' | '<=' | '<>' | '!=' | '=='
1445# | 'in' | 'not' 'in' | 'is' | 'is' 'not'
1446_cmp_types = {
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001447 token.LESS : '<',
1448 token.GREATER : '>',
1449 token.EQEQUAL : '==',
1450 token.EQUAL : '==',
1451 token.LESSEQUAL : '<=',
1452 token.GREATEREQUAL : '>=',
1453 token.NOTEQUAL : '!=',
1454 }
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001455
1456_legal_node_types = [
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001457 symbol.funcdef,
1458 symbol.classdef,
1459 symbol.stmt,
1460 symbol.small_stmt,
1461 symbol.flow_stmt,
1462 symbol.simple_stmt,
1463 symbol.compound_stmt,
1464 symbol.expr_stmt,
1465 symbol.print_stmt,
1466 symbol.del_stmt,
1467 symbol.pass_stmt,
1468 symbol.break_stmt,
1469 symbol.continue_stmt,
1470 symbol.return_stmt,
1471 symbol.raise_stmt,
1472 symbol.import_stmt,
1473 symbol.global_stmt,
1474 symbol.exec_stmt,
1475 symbol.assert_stmt,
1476 symbol.if_stmt,
1477 symbol.while_stmt,
1478 symbol.for_stmt,
1479 symbol.try_stmt,
Guido van Rossum7ad94f02006-02-28 00:32:16 +00001480 symbol.with_stmt,
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001481 symbol.suite,
1482 symbol.testlist,
Jeremy Hyltond1142612001-10-17 13:32:52 +00001483 symbol.testlist_safe,
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001484 symbol.test,
1485 symbol.and_test,
1486 symbol.not_test,
1487 symbol.comparison,
1488 symbol.exprlist,
1489 symbol.expr,
1490 symbol.xor_expr,
1491 symbol.and_expr,
1492 symbol.shift_expr,
1493 symbol.arith_expr,
1494 symbol.term,
1495 symbol.factor,
1496 symbol.power,
1497 symbol.atom,
1498 ]
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001499
Jeremy Hylton7e30c9b2001-08-27 21:02:51 +00001500if hasattr(symbol, 'yield_stmt'):
1501 _legal_node_types.append(symbol.yield_stmt)
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00001502if hasattr(symbol, 'yield_expr'):
1503 _legal_node_types.append(symbol.yield_expr)
Jeremy Hylton7e30c9b2001-08-27 21:02:51 +00001504
Jeremy Hyltonf968e852000-02-04 00:25:23 +00001505_assign_types = [
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001506 symbol.test,
Martin v. Löwis59977a62006-02-27 19:57:01 +00001507 symbol.or_test,
Jeremy Hylton572bdce2000-09-20 02:47:28 +00001508 symbol.and_test,
1509 symbol.not_test,
1510 symbol.comparison,
1511 symbol.expr,
1512 symbol.xor_expr,
1513 symbol.and_expr,
1514 symbol.shift_expr,
1515 symbol.arith_expr,
1516 symbol.term,
1517 symbol.factor,
1518 ]
Jeremy Hylton9c048f92000-10-13 21:58:13 +00001519
Jeremy Hylton9c048f92000-10-13 21:58:13 +00001520_names = {}
1521for k, v in symbol.sym_name.items():
1522 _names[k] = v
1523for k, v in token.tok_name.items():
1524 _names[k] = v
1525
1526def debug_tree(tree):
1527 l = []
1528 for elt in tree:
Neal Norwitz212a5752005-11-25 03:19:29 +00001529 if isinstance(elt, int):
Jeremy Hylton9c048f92000-10-13 21:58:13 +00001530 l.append(_names.get(elt, elt))
Neal Norwitz212a5752005-11-25 03:19:29 +00001531 elif isinstance(elt, str):
Jeremy Hylton9c048f92000-10-13 21:58:13 +00001532 l.append(elt)
1533 else:
1534 l.append(debug_tree(elt))
1535 return l