blob: 4c3fc161d39f9faabfa6a390f0b281669f321096 [file] [log] [blame]
Jeremy Hylton628d2892000-10-25 18:02:59 +00001"""Python abstract syntax node definitions
2
Jeremy Hylton566d9342004-09-07 15:28:01 +00003This file is automatically generated by Tools/compiler/astgen.py
Jeremy Hylton628d2892000-10-25 18:02:59 +00004"""
Neal Norwitzb902f4e2006-04-03 04:45:34 +00005from compiler.consts import CO_VARARGS, CO_VARKEYWORDS
Jeremy Hyltonf968e852000-02-04 00:25:23 +00006
Neil Schemenauerf3694702005-06-02 05:55:20 +00007def flatten(seq):
Jeremy Hylton628d2892000-10-25 18:02:59 +00008 l = []
Neil Schemenauerf3694702005-06-02 05:55:20 +00009 for elt in seq:
Jeremy Hylton628d2892000-10-25 18:02:59 +000010 t = type(elt)
Jeremy Hylton566d9342004-09-07 15:28:01 +000011 if t is tuple or t is list:
Jeremy Hylton628d2892000-10-25 18:02:59 +000012 for elt2 in flatten(elt):
13 l.append(elt2)
14 else:
15 l.append(elt)
16 return l
17
Neil Schemenauerf3694702005-06-02 05:55:20 +000018def flatten_nodes(seq):
19 return [n for n in flatten(seq) if isinstance(n, Node)]
Jeremy Hylton5477f522001-08-29 18:08:02 +000020
Jeremy Hyltonf968e852000-02-04 00:25:23 +000021nodes = {}
22
Jeremy Hylton566d9342004-09-07 15:28:01 +000023class Node:
24 """Abstract base class for ast nodes."""
Jeremy Hylton628d2892000-10-25 18:02:59 +000025 def getChildren(self):
Jeremy Hylton5477f522001-08-29 18:08:02 +000026 pass # implemented by subclasses
Jeremy Hylton566d9342004-09-07 15:28:01 +000027 def __iter__(self):
28 for n in self.getChildren():
29 yield n
30 def asList(self): # for backwards compatibility
31 return self.getChildren()
Jeremy Hylton7713ac22001-08-14 18:59:01 +000032 def getChildNodes(self):
Jeremy Hylton5477f522001-08-29 18:08:02 +000033 pass # implemented by subclasses
Jeremy Hyltonf968e852000-02-04 00:25:23 +000034
35class EmptyNode(Node):
Jeremy Hylton5477f522001-08-29 18:08:02 +000036 pass
Jeremy Hyltonf968e852000-02-04 00:25:23 +000037
Anthony Baxterc2a5a632004-08-02 06:10:11 +000038class Expression(Node):
39 # Expression is an artificial node class to support "eval"
40 nodes["expression"] = "Expression"
41 def __init__(self, node):
42 self.node = node
43
44 def getChildren(self):
45 return self.node,
46
47 def getChildNodes(self):
48 return self.node,
49
50 def __repr__(self):
51 return "Expression(%s)" % (repr(self.node))
52
53class Add(Node):
Antoine Pitroub9d49632010-01-04 23:22:44 +000054 def __init__(self, leftright, lineno=None):
55 self.left = leftright[0]
56 self.right = leftright[1]
Jeremy Hylton566d9342004-09-07 15:28:01 +000057 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +000058
59 def getChildren(self):
60 return self.left, self.right
61
62 def getChildNodes(self):
63 return self.left, self.right
64
65 def __repr__(self):
66 return "Add((%s, %s))" % (repr(self.left), repr(self.right))
67
68class And(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +000069 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +000070 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +000071 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +000072
73 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +000074 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +000075
76 def getChildNodes(self):
77 nodelist = []
78 nodelist.extend(flatten_nodes(self.nodes))
79 return tuple(nodelist)
80
81 def __repr__(self):
82 return "And(%s)" % (repr(self.nodes),)
83
84class AssAttr(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +000085 def __init__(self, expr, attrname, flags, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +000086 self.expr = expr
Anthony Baxterc2a5a632004-08-02 06:10:11 +000087 self.attrname = attrname
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +000088 self.flags = flags
Jeremy Hylton566d9342004-09-07 15:28:01 +000089 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +000090
91 def getChildren(self):
92 return self.expr, self.attrname, self.flags
93
94 def getChildNodes(self):
95 return self.expr,
96
97 def __repr__(self):
98 return "AssAttr(%s, %s, %s)" % (repr(self.expr), repr(self.attrname), repr(self.flags))
99
100class AssList(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000101 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000102 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +0000103 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000104
105 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000106 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000107
108 def getChildNodes(self):
109 nodelist = []
110 nodelist.extend(flatten_nodes(self.nodes))
111 return tuple(nodelist)
112
113 def __repr__(self):
114 return "AssList(%s)" % (repr(self.nodes),)
115
116class AssName(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000117 def __init__(self, name, flags, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000118 self.name = name
119 self.flags = flags
Jeremy Hylton566d9342004-09-07 15:28:01 +0000120 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000121
122 def getChildren(self):
123 return self.name, self.flags
124
125 def getChildNodes(self):
126 return ()
127
128 def __repr__(self):
129 return "AssName(%s, %s)" % (repr(self.name), repr(self.flags))
130
131class AssTuple(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000132 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000133 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +0000134 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000135
136 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000137 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000138
139 def getChildNodes(self):
140 nodelist = []
141 nodelist.extend(flatten_nodes(self.nodes))
142 return tuple(nodelist)
143
144 def __repr__(self):
145 return "AssTuple(%s)" % (repr(self.nodes),)
146
147class Assert(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000148 def __init__(self, test, fail, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000149 self.test = test
150 self.fail = fail
Jeremy Hylton566d9342004-09-07 15:28:01 +0000151 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000152
153 def getChildren(self):
154 children = []
155 children.append(self.test)
156 children.append(self.fail)
157 return tuple(children)
158
159 def getChildNodes(self):
160 nodelist = []
161 nodelist.append(self.test)
Jeremy Hylton566d9342004-09-07 15:28:01 +0000162 if self.fail is not None:
163 nodelist.append(self.fail)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000164 return tuple(nodelist)
165
166 def __repr__(self):
167 return "Assert(%s, %s)" % (repr(self.test), repr(self.fail))
168
169class Assign(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000170 def __init__(self, nodes, expr, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000171 self.nodes = nodes
172 self.expr = expr
Jeremy Hylton566d9342004-09-07 15:28:01 +0000173 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000174
175 def getChildren(self):
176 children = []
177 children.extend(flatten(self.nodes))
178 children.append(self.expr)
179 return tuple(children)
180
181 def getChildNodes(self):
182 nodelist = []
183 nodelist.extend(flatten_nodes(self.nodes))
184 nodelist.append(self.expr)
185 return tuple(nodelist)
186
187 def __repr__(self):
188 return "Assign(%s, %s)" % (repr(self.nodes), repr(self.expr))
189
190class AugAssign(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000191 def __init__(self, node, op, expr, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000192 self.node = node
193 self.op = op
194 self.expr = expr
Jeremy Hylton566d9342004-09-07 15:28:01 +0000195 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000196
197 def getChildren(self):
198 return self.node, self.op, self.expr
199
200 def getChildNodes(self):
201 return self.node, self.expr
202
203 def __repr__(self):
204 return "AugAssign(%s, %s, %s)" % (repr(self.node), repr(self.op), repr(self.expr))
205
206class Backquote(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000207 def __init__(self, expr, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000208 self.expr = expr
Jeremy Hylton566d9342004-09-07 15:28:01 +0000209 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000210
211 def getChildren(self):
212 return self.expr,
213
214 def getChildNodes(self):
215 return self.expr,
216
217 def __repr__(self):
218 return "Backquote(%s)" % (repr(self.expr),)
219
220class Bitand(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000221 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000222 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +0000223 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000224
225 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000226 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000227
228 def getChildNodes(self):
229 nodelist = []
230 nodelist.extend(flatten_nodes(self.nodes))
231 return tuple(nodelist)
232
233 def __repr__(self):
234 return "Bitand(%s)" % (repr(self.nodes),)
235
236class Bitor(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000237 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000238 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +0000239 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000240
241 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000242 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000243
244 def getChildNodes(self):
245 nodelist = []
246 nodelist.extend(flatten_nodes(self.nodes))
247 return tuple(nodelist)
248
249 def __repr__(self):
250 return "Bitor(%s)" % (repr(self.nodes),)
251
252class Bitxor(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000253 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000254 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +0000255 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000256
257 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000258 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000259
260 def getChildNodes(self):
261 nodelist = []
262 nodelist.extend(flatten_nodes(self.nodes))
263 return tuple(nodelist)
264
265 def __repr__(self):
266 return "Bitxor(%s)" % (repr(self.nodes),)
267
268class Break(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000269 def __init__(self, lineno=None):
270 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000271
272 def getChildren(self):
273 return ()
274
275 def getChildNodes(self):
276 return ()
277
278 def __repr__(self):
279 return "Break()"
280
281class CallFunc(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000282 def __init__(self, node, args, star_args = None, dstar_args = None, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000283 self.node = node
284 self.args = args
285 self.star_args = star_args
286 self.dstar_args = dstar_args
Jeremy Hylton566d9342004-09-07 15:28:01 +0000287 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000288
289 def getChildren(self):
290 children = []
291 children.append(self.node)
292 children.extend(flatten(self.args))
293 children.append(self.star_args)
294 children.append(self.dstar_args)
295 return tuple(children)
296
297 def getChildNodes(self):
298 nodelist = []
299 nodelist.append(self.node)
300 nodelist.extend(flatten_nodes(self.args))
Jeremy Hylton566d9342004-09-07 15:28:01 +0000301 if self.star_args is not None:
302 nodelist.append(self.star_args)
303 if self.dstar_args is not None:
304 nodelist.append(self.dstar_args)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000305 return tuple(nodelist)
306
307 def __repr__(self):
308 return "CallFunc(%s, %s, %s, %s)" % (repr(self.node), repr(self.args), repr(self.star_args), repr(self.dstar_args))
309
310class Class(Node):
Facundo Batista4219da42008-02-25 18:06:00 +0000311 def __init__(self, name, bases, doc, code, decorators = None, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000312 self.name = name
313 self.bases = bases
314 self.doc = doc
315 self.code = code
Facundo Batista4219da42008-02-25 18:06:00 +0000316 self.decorators = decorators
Jeremy Hylton566d9342004-09-07 15:28:01 +0000317 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000318
319 def getChildren(self):
320 children = []
321 children.append(self.name)
322 children.extend(flatten(self.bases))
323 children.append(self.doc)
324 children.append(self.code)
Facundo Batista4219da42008-02-25 18:06:00 +0000325 children.append(self.decorators)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000326 return tuple(children)
327
328 def getChildNodes(self):
329 nodelist = []
330 nodelist.extend(flatten_nodes(self.bases))
331 nodelist.append(self.code)
Facundo Batista4219da42008-02-25 18:06:00 +0000332 if self.decorators is not None:
333 nodelist.append(self.decorators)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000334 return tuple(nodelist)
335
336 def __repr__(self):
Facundo Batista4219da42008-02-25 18:06:00 +0000337 return "Class(%s, %s, %s, %s, %s)" % (repr(self.name), repr(self.bases), repr(self.doc), repr(self.code), repr(self.decorators))
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000338
339class Compare(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000340 def __init__(self, expr, ops, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000341 self.expr = expr
342 self.ops = ops
Jeremy Hylton566d9342004-09-07 15:28:01 +0000343 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000344
345 def getChildren(self):
346 children = []
347 children.append(self.expr)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000348 children.extend(flatten(self.ops))
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000349 return tuple(children)
350
351 def getChildNodes(self):
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +0000352 nodelist = []
353 nodelist.append(self.expr)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000354 nodelist.extend(flatten_nodes(self.ops))
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +0000355 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000356
357 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000358 return "Compare(%s, %s)" % (repr(self.expr), repr(self.ops))
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000359
360class Const(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000361 def __init__(self, value, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000362 self.value = value
Jeremy Hylton566d9342004-09-07 15:28:01 +0000363 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000364
365 def getChildren(self):
366 return self.value,
367
368 def getChildNodes(self):
369 return ()
370
371 def __repr__(self):
372 return "Const(%s)" % (repr(self.value),)
373
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000374class Continue(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000375 def __init__(self, lineno=None):
376 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000377
378 def getChildren(self):
379 return ()
380
381 def getChildNodes(self):
382 return ()
383
384 def __repr__(self):
385 return "Continue()"
386
387class Decorators(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000388 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000389 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +0000390 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000391
392 def getChildren(self):
393 return tuple(flatten(self.nodes))
394
395 def getChildNodes(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000396 nodelist = []
397 nodelist.extend(flatten_nodes(self.nodes))
398 return tuple(nodelist)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000399
400 def __repr__(self):
401 return "Decorators(%s)" % (repr(self.nodes),)
402
403class Dict(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000404 def __init__(self, items, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000405 self.items = items
Jeremy Hylton566d9342004-09-07 15:28:01 +0000406 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000407
408 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000409 return tuple(flatten(self.items))
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000410
411 def getChildNodes(self):
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +0000412 nodelist = []
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000413 nodelist.extend(flatten_nodes(self.items))
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +0000414 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000415
416 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000417 return "Dict(%s)" % (repr(self.items),)
418
419class Discard(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000420 def __init__(self, expr, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000421 self.expr = expr
Jeremy Hylton566d9342004-09-07 15:28:01 +0000422 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000423
424 def getChildren(self):
425 return self.expr,
426
427 def getChildNodes(self):
428 return self.expr,
429
430 def __repr__(self):
431 return "Discard(%s)" % (repr(self.expr),)
432
433class Div(Node):
Antoine Pitroub9d49632010-01-04 23:22:44 +0000434 def __init__(self, leftright, lineno=None):
435 self.left = leftright[0]
436 self.right = leftright[1]
Jeremy Hylton566d9342004-09-07 15:28:01 +0000437 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000438
439 def getChildren(self):
440 return self.left, self.right
441
442 def getChildNodes(self):
443 return self.left, self.right
444
445 def __repr__(self):
446 return "Div((%s, %s))" % (repr(self.left), repr(self.right))
447
448class Ellipsis(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000449 def __init__(self, lineno=None):
450 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000451
452 def getChildren(self):
453 return ()
454
455 def getChildNodes(self):
456 return ()
457
458 def __repr__(self):
459 return "Ellipsis()"
460
461class Exec(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000462 def __init__(self, expr, locals, globals, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000463 self.expr = expr
464 self.locals = locals
465 self.globals = globals
Jeremy Hylton566d9342004-09-07 15:28:01 +0000466 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000467
468 def getChildren(self):
469 children = []
470 children.append(self.expr)
471 children.append(self.locals)
472 children.append(self.globals)
473 return tuple(children)
474
475 def getChildNodes(self):
476 nodelist = []
477 nodelist.append(self.expr)
Jeremy Hylton566d9342004-09-07 15:28:01 +0000478 if self.locals is not None:
479 nodelist.append(self.locals)
480 if self.globals is not None:
481 nodelist.append(self.globals)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000482 return tuple(nodelist)
483
484 def __repr__(self):
485 return "Exec(%s, %s, %s)" % (repr(self.expr), repr(self.locals), repr(self.globals))
486
487class FloorDiv(Node):
Antoine Pitroub9d49632010-01-04 23:22:44 +0000488 def __init__(self, leftright, lineno=None):
489 self.left = leftright[0]
490 self.right = leftright[1]
Jeremy Hylton566d9342004-09-07 15:28:01 +0000491 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000492
493 def getChildren(self):
494 return self.left, self.right
495
496 def getChildNodes(self):
497 return self.left, self.right
498
499 def __repr__(self):
500 return "FloorDiv((%s, %s))" % (repr(self.left), repr(self.right))
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000501
502class For(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000503 def __init__(self, assign, list, body, else_, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000504 self.assign = assign
505 self.list = list
506 self.body = body
507 self.else_ = else_
Jeremy Hylton566d9342004-09-07 15:28:01 +0000508 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000509
510 def getChildren(self):
511 children = []
512 children.append(self.assign)
513 children.append(self.list)
514 children.append(self.body)
515 children.append(self.else_)
516 return tuple(children)
517
518 def getChildNodes(self):
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +0000519 nodelist = []
520 nodelist.append(self.assign)
521 nodelist.append(self.list)
522 nodelist.append(self.body)
Jeremy Hylton566d9342004-09-07 15:28:01 +0000523 if self.else_ is not None:
524 nodelist.append(self.else_)
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +0000525 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000526
527 def __repr__(self):
528 return "For(%s, %s, %s, %s)" % (repr(self.assign), repr(self.list), repr(self.body), repr(self.else_))
529
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000530class From(Node):
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000531 def __init__(self, modname, names, level, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000532 self.modname = modname
533 self.names = names
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000534 self.level = level
Jeremy Hylton566d9342004-09-07 15:28:01 +0000535 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000536
537 def getChildren(self):
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000538 return self.modname, self.names, self.level
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000539
540 def getChildNodes(self):
541 return ()
542
543 def __repr__(self):
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000544 return "From(%s, %s, %s)" % (repr(self.modname), repr(self.names), repr(self.level))
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000545
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000546class Function(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000547 def __init__(self, decorators, name, argnames, defaults, flags, doc, code, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000548 self.decorators = decorators
549 self.name = name
550 self.argnames = argnames
551 self.defaults = defaults
552 self.flags = flags
553 self.doc = doc
554 self.code = code
Jeremy Hylton566d9342004-09-07 15:28:01 +0000555 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000556 self.varargs = self.kwargs = None
557 if flags & CO_VARARGS:
558 self.varargs = 1
559 if flags & CO_VARKEYWORDS:
560 self.kwargs = 1
Tim Peters39e11fb2006-03-04 02:43:44 +0000561
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000562
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000563 def getChildren(self):
564 children = []
Jeremy Hylton566d9342004-09-07 15:28:01 +0000565 children.append(self.decorators)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000566 children.append(self.name)
567 children.append(self.argnames)
568 children.extend(flatten(self.defaults))
569 children.append(self.flags)
570 children.append(self.doc)
571 children.append(self.code)
572 return tuple(children)
573
574 def getChildNodes(self):
575 nodelist = []
Jeremy Hylton566d9342004-09-07 15:28:01 +0000576 if self.decorators is not None:
577 nodelist.append(self.decorators)
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000578 nodelist.extend(flatten_nodes(self.defaults))
579 nodelist.append(self.code)
580 return tuple(nodelist)
581
582 def __repr__(self):
583 return "Function(%s, %s, %s, %s, %s, %s, %s)" % (repr(self.decorators), repr(self.name), repr(self.argnames), repr(self.defaults), repr(self.flags), repr(self.doc), repr(self.code))
584
585class GenExpr(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000586 def __init__(self, code, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000587 self.code = code
Jeremy Hylton566d9342004-09-07 15:28:01 +0000588 self.lineno = lineno
Neil Schemenauer4c6b0d52006-08-16 23:38:05 +0000589 self.argnames = ['.0']
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000590 self.varargs = self.kwargs = None
Tim Peters39e11fb2006-03-04 02:43:44 +0000591
Antoine Pitroub9d49632010-01-04 23:22:44 +0000592
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000593 def getChildren(self):
594 return self.code,
595
596 def getChildNodes(self):
597 return self.code,
598
599 def __repr__(self):
600 return "GenExpr(%s)" % (repr(self.code),)
601
602class GenExprFor(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000603 def __init__(self, assign, iter, ifs, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000604 self.assign = assign
605 self.iter = iter
606 self.ifs = ifs
Jeremy Hylton566d9342004-09-07 15:28:01 +0000607 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000608 self.is_outmost = False
609
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000610 def getChildren(self):
611 children = []
612 children.append(self.assign)
613 children.append(self.iter)
614 children.extend(flatten(self.ifs))
615 return tuple(children)
616
617 def getChildNodes(self):
618 nodelist = []
619 nodelist.append(self.assign)
620 nodelist.append(self.iter)
621 nodelist.extend(flatten_nodes(self.ifs))
622 return tuple(nodelist)
623
624 def __repr__(self):
625 return "GenExprFor(%s, %s, %s)" % (repr(self.assign), repr(self.iter), repr(self.ifs))
626
627class GenExprIf(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000628 def __init__(self, test, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000629 self.test = test
Jeremy Hylton566d9342004-09-07 15:28:01 +0000630 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000631
632 def getChildren(self):
633 return self.test,
634
635 def getChildNodes(self):
636 return self.test,
637
638 def __repr__(self):
639 return "GenExprIf(%s)" % (repr(self.test),)
640
641class GenExprInner(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000642 def __init__(self, expr, quals, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000643 self.expr = expr
644 self.quals = quals
Jeremy Hylton566d9342004-09-07 15:28:01 +0000645 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000646
647 def getChildren(self):
648 children = []
649 children.append(self.expr)
650 children.extend(flatten(self.quals))
651 return tuple(children)
652
653 def getChildNodes(self):
654 nodelist = []
655 nodelist.append(self.expr)
656 nodelist.extend(flatten_nodes(self.quals))
657 return tuple(nodelist)
658
659 def __repr__(self):
660 return "GenExprInner(%s, %s)" % (repr(self.expr), repr(self.quals))
661
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000662class Getattr(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000663 def __init__(self, expr, attrname, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000664 self.expr = expr
665 self.attrname = attrname
Jeremy Hylton566d9342004-09-07 15:28:01 +0000666 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000667
668 def getChildren(self):
669 return self.expr, self.attrname
670
671 def getChildNodes(self):
672 return self.expr,
673
674 def __repr__(self):
675 return "Getattr(%s, %s)" % (repr(self.expr), repr(self.attrname))
676
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000677class Global(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000678 def __init__(self, names, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000679 self.names = names
Jeremy Hylton566d9342004-09-07 15:28:01 +0000680 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000681
682 def getChildren(self):
683 return self.names,
684
685 def getChildNodes(self):
686 return ()
687
688 def __repr__(self):
689 return "Global(%s)" % (repr(self.names),)
690
691class If(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000692 def __init__(self, tests, else_, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000693 self.tests = tests
694 self.else_ = else_
Jeremy Hylton566d9342004-09-07 15:28:01 +0000695 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000696
697 def getChildren(self):
698 children = []
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000699 children.extend(flatten(self.tests))
700 children.append(self.else_)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000701 return tuple(children)
702
703 def getChildNodes(self):
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +0000704 nodelist = []
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000705 nodelist.extend(flatten_nodes(self.tests))
Jeremy Hylton566d9342004-09-07 15:28:01 +0000706 if self.else_ is not None:
707 nodelist.append(self.else_)
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +0000708 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000709
710 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000711 return "If(%s, %s)" % (repr(self.tests), repr(self.else_))
712
Thomas Woutersfa0cf4f2006-03-03 18:16:20 +0000713class IfExp(Node):
714 def __init__(self, test, then, else_, lineno=None):
715 self.test = test
716 self.then = then
717 self.else_ = else_
718 self.lineno = lineno
719
720 def getChildren(self):
721 return self.test, self.then, self.else_
722
723 def getChildNodes(self):
724 return self.test, self.then, self.else_
725
726 def __repr__(self):
727 return "IfExp(%s, %s, %s)" % (repr(self.test), repr(self.then), repr(self.else_))
728
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000729class Import(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000730 def __init__(self, names, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000731 self.names = names
Jeremy Hylton566d9342004-09-07 15:28:01 +0000732 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000733
734 def getChildren(self):
735 return self.names,
736
737 def getChildNodes(self):
738 return ()
739
740 def __repr__(self):
741 return "Import(%s)" % (repr(self.names),)
742
743class Invert(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000744 def __init__(self, expr, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000745 self.expr = expr
Jeremy Hylton566d9342004-09-07 15:28:01 +0000746 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000747
748 def getChildren(self):
749 return self.expr,
750
751 def getChildNodes(self):
752 return self.expr,
753
754 def __repr__(self):
755 return "Invert(%s)" % (repr(self.expr),)
756
757class Keyword(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000758 def __init__(self, name, expr, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000759 self.name = name
760 self.expr = expr
Jeremy Hylton566d9342004-09-07 15:28:01 +0000761 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000762
763 def getChildren(self):
764 return self.name, self.expr
765
766 def getChildNodes(self):
767 return self.expr,
768
769 def __repr__(self):
770 return "Keyword(%s, %s)" % (repr(self.name), repr(self.expr))
771
772class Lambda(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000773 def __init__(self, argnames, defaults, flags, code, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000774 self.argnames = argnames
775 self.defaults = defaults
776 self.flags = flags
777 self.code = code
Jeremy Hylton566d9342004-09-07 15:28:01 +0000778 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000779 self.varargs = self.kwargs = None
780 if flags & CO_VARARGS:
781 self.varargs = 1
782 if flags & CO_VARKEYWORDS:
783 self.kwargs = 1
Tim Peters39e11fb2006-03-04 02:43:44 +0000784
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000785
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000786 def getChildren(self):
787 children = []
788 children.append(self.argnames)
789 children.extend(flatten(self.defaults))
790 children.append(self.flags)
791 children.append(self.code)
792 return tuple(children)
793
794 def getChildNodes(self):
795 nodelist = []
796 nodelist.extend(flatten_nodes(self.defaults))
797 nodelist.append(self.code)
798 return tuple(nodelist)
799
800 def __repr__(self):
801 return "Lambda(%s, %s, %s, %s)" % (repr(self.argnames), repr(self.defaults), repr(self.flags), repr(self.code))
802
803class LeftShift(Node):
Antoine Pitroub9d49632010-01-04 23:22:44 +0000804 def __init__(self, leftright, lineno=None):
805 self.left = leftright[0]
806 self.right = leftright[1]
Jeremy Hylton566d9342004-09-07 15:28:01 +0000807 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000808
809 def getChildren(self):
810 return self.left, self.right
811
812 def getChildNodes(self):
813 return self.left, self.right
814
815 def __repr__(self):
816 return "LeftShift((%s, %s))" % (repr(self.left), repr(self.right))
817
818class List(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000819 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000820 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +0000821 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000822
823 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000824 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000825
826 def getChildNodes(self):
827 nodelist = []
828 nodelist.extend(flatten_nodes(self.nodes))
829 return tuple(nodelist)
830
831 def __repr__(self):
832 return "List(%s)" % (repr(self.nodes),)
833
834class ListComp(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000835 def __init__(self, expr, quals, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000836 self.expr = expr
837 self.quals = quals
Jeremy Hylton566d9342004-09-07 15:28:01 +0000838 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000839
840 def getChildren(self):
841 children = []
842 children.append(self.expr)
843 children.extend(flatten(self.quals))
844 return tuple(children)
845
846 def getChildNodes(self):
847 nodelist = []
848 nodelist.append(self.expr)
849 nodelist.extend(flatten_nodes(self.quals))
850 return tuple(nodelist)
851
852 def __repr__(self):
853 return "ListComp(%s, %s)" % (repr(self.expr), repr(self.quals))
854
855class ListCompFor(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000856 def __init__(self, assign, list, ifs, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000857 self.assign = assign
858 self.list = list
859 self.ifs = ifs
Jeremy Hylton566d9342004-09-07 15:28:01 +0000860 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000861
862 def getChildren(self):
863 children = []
864 children.append(self.assign)
865 children.append(self.list)
866 children.extend(flatten(self.ifs))
867 return tuple(children)
868
869 def getChildNodes(self):
870 nodelist = []
871 nodelist.append(self.assign)
872 nodelist.append(self.list)
873 nodelist.extend(flatten_nodes(self.ifs))
874 return tuple(nodelist)
875
876 def __repr__(self):
877 return "ListCompFor(%s, %s, %s)" % (repr(self.assign), repr(self.list), repr(self.ifs))
878
879class ListCompIf(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000880 def __init__(self, test, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000881 self.test = test
Jeremy Hylton566d9342004-09-07 15:28:01 +0000882 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000883
884 def getChildren(self):
885 return self.test,
886
887 def getChildNodes(self):
888 return self.test,
889
890 def __repr__(self):
891 return "ListCompIf(%s)" % (repr(self.test),)
892
Alexandre Vassalottib6465472010-01-11 22:36:12 +0000893class SetComp(Node):
894 def __init__(self, expr, quals, lineno=None):
895 self.expr = expr
896 self.quals = quals
897 self.lineno = lineno
898
899 def getChildren(self):
900 children = []
901 children.append(self.expr)
902 children.extend(flatten(self.quals))
903 return tuple(children)
904
905 def getChildNodes(self):
906 nodelist = []
907 nodelist.append(self.expr)
908 nodelist.extend(flatten_nodes(self.quals))
909 return tuple(nodelist)
910
911 def __repr__(self):
912 return "SetComp(%s, %s)" % (repr(self.expr), repr(self.quals))
913
914class DictComp(Node):
915 def __init__(self, key, value, quals, lineno=None):
916 self.key = key
917 self.value = value
918 self.quals = quals
919 self.lineno = lineno
920
921 def getChildren(self):
922 children = []
923 children.append(self.key)
924 children.append(self.value)
925 children.extend(flatten(self.quals))
926 return tuple(children)
927
928 def getChildNodes(self):
929 nodelist = []
930 nodelist.append(self.key)
931 nodelist.append(self.value)
932 nodelist.extend(flatten_nodes(self.quals))
933 return tuple(nodelist)
934
935 def __repr__(self):
936 return "DictComp(%s, %s, %s)" % (repr(self.key), repr(self.value), repr(self.quals))
937
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000938class Mod(Node):
Antoine Pitroub9d49632010-01-04 23:22:44 +0000939 def __init__(self, leftright, lineno=None):
940 self.left = leftright[0]
941 self.right = leftright[1]
Jeremy Hylton566d9342004-09-07 15:28:01 +0000942 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000943
944 def getChildren(self):
945 return self.left, self.right
946
947 def getChildNodes(self):
948 return self.left, self.right
949
950 def __repr__(self):
951 return "Mod((%s, %s))" % (repr(self.left), repr(self.right))
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000952
953class Module(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000954 def __init__(self, doc, node, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000955 self.doc = doc
956 self.node = node
Jeremy Hylton566d9342004-09-07 15:28:01 +0000957 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000958
959 def getChildren(self):
960 return self.doc, self.node
961
962 def getChildNodes(self):
963 return self.node,
964
965 def __repr__(self):
966 return "Module(%s, %s)" % (repr(self.doc), repr(self.node))
967
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000968class Mul(Node):
Antoine Pitroub9d49632010-01-04 23:22:44 +0000969 def __init__(self, leftright, lineno=None):
970 self.left = leftright[0]
971 self.right = leftright[1]
Jeremy Hylton566d9342004-09-07 15:28:01 +0000972 self.lineno = lineno
Barry Warsaw52acb492001-12-21 20:04:22 +0000973
974 def getChildren(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000975 return self.left, self.right
Barry Warsaw52acb492001-12-21 20:04:22 +0000976
977 def getChildNodes(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000978 return self.left, self.right
Barry Warsaw52acb492001-12-21 20:04:22 +0000979
980 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000981 return "Mul((%s, %s))" % (repr(self.left), repr(self.right))
Barry Warsaw52acb492001-12-21 20:04:22 +0000982
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000983class Name(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000984 def __init__(self, name, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000985 self.name = name
Jeremy Hylton566d9342004-09-07 15:28:01 +0000986 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +0000987
988 def getChildren(self):
989 return self.name,
990
991 def getChildNodes(self):
992 return ()
993
994 def __repr__(self):
995 return "Name(%s)" % (repr(self.name),)
996
997class Not(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +0000998 def __init__(self, expr, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +0000999 self.expr = expr
Jeremy Hylton566d9342004-09-07 15:28:01 +00001000 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001001
1002 def getChildren(self):
1003 return self.expr,
1004
1005 def getChildNodes(self):
1006 return self.expr,
1007
1008 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001009 return "Not(%s)" % (repr(self.expr),)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001010
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001011class Or(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001012 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001013 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +00001014 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001015
1016 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001017 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001018
1019 def getChildNodes(self):
1020 nodelist = []
1021 nodelist.extend(flatten_nodes(self.nodes))
1022 return tuple(nodelist)
1023
1024 def __repr__(self):
1025 return "Or(%s)" % (repr(self.nodes),)
1026
1027class Pass(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001028 def __init__(self, lineno=None):
1029 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001030
1031 def getChildren(self):
1032 return ()
1033
1034 def getChildNodes(self):
1035 return ()
1036
1037 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001038 return "Pass()"
1039
1040class Power(Node):
Antoine Pitroub9d49632010-01-04 23:22:44 +00001041 def __init__(self, leftright, lineno=None):
1042 self.left = leftright[0]
1043 self.right = leftright[1]
Jeremy Hylton566d9342004-09-07 15:28:01 +00001044 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001045
1046 def getChildren(self):
1047 return self.left, self.right
1048
1049 def getChildNodes(self):
1050 return self.left, self.right
1051
1052 def __repr__(self):
1053 return "Power((%s, %s))" % (repr(self.left), repr(self.right))
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001054
1055class Print(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001056 def __init__(self, nodes, dest, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001057 self.nodes = nodes
1058 self.dest = dest
Jeremy Hylton566d9342004-09-07 15:28:01 +00001059 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001060
1061 def getChildren(self):
1062 children = []
1063 children.extend(flatten(self.nodes))
1064 children.append(self.dest)
1065 return tuple(children)
1066
1067 def getChildNodes(self):
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +00001068 nodelist = []
1069 nodelist.extend(flatten_nodes(self.nodes))
Jeremy Hylton566d9342004-09-07 15:28:01 +00001070 if self.dest is not None:
1071 nodelist.append(self.dest)
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +00001072 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001073
1074 def __repr__(self):
1075 return "Print(%s, %s)" % (repr(self.nodes), repr(self.dest))
1076
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001077class Printnl(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001078 def __init__(self, nodes, dest, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001079 self.nodes = nodes
1080 self.dest = dest
Jeremy Hylton566d9342004-09-07 15:28:01 +00001081 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001082
1083 def getChildren(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001084 children = []
1085 children.extend(flatten(self.nodes))
1086 children.append(self.dest)
1087 return tuple(children)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001088
1089 def getChildNodes(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001090 nodelist = []
1091 nodelist.extend(flatten_nodes(self.nodes))
Jeremy Hylton566d9342004-09-07 15:28:01 +00001092 if self.dest is not None:
1093 nodelist.append(self.dest)
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001094 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001095
1096 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001097 return "Printnl(%s, %s)" % (repr(self.nodes), repr(self.dest))
1098
1099class Raise(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001100 def __init__(self, expr1, expr2, expr3, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001101 self.expr1 = expr1
1102 self.expr2 = expr2
1103 self.expr3 = expr3
Jeremy Hylton566d9342004-09-07 15:28:01 +00001104 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001105
1106 def getChildren(self):
1107 children = []
1108 children.append(self.expr1)
1109 children.append(self.expr2)
1110 children.append(self.expr3)
1111 return tuple(children)
1112
1113 def getChildNodes(self):
1114 nodelist = []
Jeremy Hylton566d9342004-09-07 15:28:01 +00001115 if self.expr1 is not None:
1116 nodelist.append(self.expr1)
1117 if self.expr2 is not None:
1118 nodelist.append(self.expr2)
1119 if self.expr3 is not None:
1120 nodelist.append(self.expr3)
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001121 return tuple(nodelist)
1122
1123 def __repr__(self):
1124 return "Raise(%s, %s, %s)" % (repr(self.expr1), repr(self.expr2), repr(self.expr3))
1125
1126class Return(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001127 def __init__(self, value, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001128 self.value = value
Jeremy Hylton566d9342004-09-07 15:28:01 +00001129 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001130
1131 def getChildren(self):
1132 return self.value,
1133
1134 def getChildNodes(self):
1135 return self.value,
1136
1137 def __repr__(self):
1138 return "Return(%s)" % (repr(self.value),)
1139
1140class RightShift(Node):
Antoine Pitroub9d49632010-01-04 23:22:44 +00001141 def __init__(self, leftright, lineno=None):
1142 self.left = leftright[0]
1143 self.right = leftright[1]
Jeremy Hylton566d9342004-09-07 15:28:01 +00001144 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001145
1146 def getChildren(self):
1147 return self.left, self.right
1148
1149 def getChildNodes(self):
1150 return self.left, self.right
1151
1152 def __repr__(self):
1153 return "RightShift((%s, %s))" % (repr(self.left), repr(self.right))
1154
Alexandre Vassalottiee936a22010-01-09 23:35:54 +00001155class Set(Node):
1156 def __init__(self, nodes, lineno=None):
1157 self.nodes = nodes
1158 self.lineno = lineno
1159
1160 def getChildren(self):
1161 return tuple(flatten(self.nodes))
1162
1163 def getChildNodes(self):
1164 nodelist = []
1165 nodelist.extend(flatten_nodes(self.nodes))
1166 return tuple(nodelist)
1167
1168 def __repr__(self):
1169 return "Set(%s)" % (repr(self.nodes),)
1170
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001171class Slice(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001172 def __init__(self, expr, flags, lower, upper, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001173 self.expr = expr
1174 self.flags = flags
1175 self.lower = lower
1176 self.upper = upper
Jeremy Hylton566d9342004-09-07 15:28:01 +00001177 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001178
1179 def getChildren(self):
1180 children = []
1181 children.append(self.expr)
1182 children.append(self.flags)
1183 children.append(self.lower)
1184 children.append(self.upper)
1185 return tuple(children)
1186
1187 def getChildNodes(self):
1188 nodelist = []
1189 nodelist.append(self.expr)
Jeremy Hylton566d9342004-09-07 15:28:01 +00001190 if self.lower is not None:
1191 nodelist.append(self.lower)
1192 if self.upper is not None:
1193 nodelist.append(self.upper)
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001194 return tuple(nodelist)
1195
1196 def __repr__(self):
1197 return "Slice(%s, %s, %s, %s)" % (repr(self.expr), repr(self.flags), repr(self.lower), repr(self.upper))
1198
1199class Sliceobj(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001200 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001201 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +00001202 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001203
1204 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001205 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001206
1207 def getChildNodes(self):
1208 nodelist = []
1209 nodelist.extend(flatten_nodes(self.nodes))
1210 return tuple(nodelist)
1211
1212 def __repr__(self):
1213 return "Sliceobj(%s)" % (repr(self.nodes),)
1214
1215class Stmt(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001216 def __init__(self, nodes, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001217 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +00001218 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001219
1220 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001221 return tuple(flatten(self.nodes))
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001222
1223 def getChildNodes(self):
1224 nodelist = []
1225 nodelist.extend(flatten_nodes(self.nodes))
1226 return tuple(nodelist)
1227
1228 def __repr__(self):
1229 return "Stmt(%s)" % (repr(self.nodes),)
1230
1231class Sub(Node):
Antoine Pitroub9d49632010-01-04 23:22:44 +00001232 def __init__(self, leftright, lineno=None):
1233 self.left = leftright[0]
1234 self.right = leftright[1]
Jeremy Hylton566d9342004-09-07 15:28:01 +00001235 self.lineno = lineno
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001236
1237 def getChildren(self):
1238 return self.left, self.right
1239
1240 def getChildNodes(self):
1241 return self.left, self.right
1242
1243 def __repr__(self):
1244 return "Sub((%s, %s))" % (repr(self.left), repr(self.right))
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001245
1246class Subscript(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001247 def __init__(self, expr, flags, subs, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001248 self.expr = expr
1249 self.flags = flags
1250 self.subs = subs
Jeremy Hylton566d9342004-09-07 15:28:01 +00001251 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001252
1253 def getChildren(self):
1254 children = []
1255 children.append(self.expr)
1256 children.append(self.flags)
1257 children.extend(flatten(self.subs))
1258 return tuple(children)
1259
1260 def getChildNodes(self):
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +00001261 nodelist = []
1262 nodelist.append(self.expr)
1263 nodelist.extend(flatten_nodes(self.subs))
1264 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001265
1266 def __repr__(self):
1267 return "Subscript(%s, %s, %s)" % (repr(self.expr), repr(self.flags), repr(self.subs))
1268
1269class TryExcept(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001270 def __init__(self, body, handlers, else_, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001271 self.body = body
1272 self.handlers = handlers
1273 self.else_ = else_
Jeremy Hylton566d9342004-09-07 15:28:01 +00001274 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001275
1276 def getChildren(self):
1277 children = []
1278 children.append(self.body)
1279 children.extend(flatten(self.handlers))
1280 children.append(self.else_)
1281 return tuple(children)
1282
1283 def getChildNodes(self):
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +00001284 nodelist = []
1285 nodelist.append(self.body)
1286 nodelist.extend(flatten_nodes(self.handlers))
Jeremy Hylton566d9342004-09-07 15:28:01 +00001287 if self.else_ is not None:
1288 nodelist.append(self.else_)
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +00001289 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001290
1291 def __repr__(self):
1292 return "TryExcept(%s, %s, %s)" % (repr(self.body), repr(self.handlers), repr(self.else_))
1293
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001294class TryFinally(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001295 def __init__(self, body, final, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001296 self.body = body
1297 self.final = final
Jeremy Hylton566d9342004-09-07 15:28:01 +00001298 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001299
1300 def getChildren(self):
1301 return self.body, self.final
1302
1303 def getChildNodes(self):
1304 return self.body, self.final
1305
1306 def __repr__(self):
1307 return "TryFinally(%s, %s)" % (repr(self.body), repr(self.final))
1308
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001309class Tuple(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001310 def __init__(self, nodes, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001311 self.nodes = nodes
Jeremy Hylton566d9342004-09-07 15:28:01 +00001312 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001313
1314 def getChildren(self):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001315 return tuple(flatten(self.nodes))
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001316
1317 def getChildNodes(self):
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +00001318 nodelist = []
1319 nodelist.extend(flatten_nodes(self.nodes))
1320 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001321
1322 def __repr__(self):
1323 return "Tuple(%s)" % (repr(self.nodes),)
1324
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001325class UnaryAdd(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001326 def __init__(self, expr, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001327 self.expr = expr
Jeremy Hylton566d9342004-09-07 15:28:01 +00001328 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001329
1330 def getChildren(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001331 return self.expr,
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001332
1333 def getChildNodes(self):
1334 return self.expr,
1335
1336 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001337 return "UnaryAdd(%s)" % (repr(self.expr),)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001338
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001339class UnarySub(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001340 def __init__(self, expr, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001341 self.expr = expr
Jeremy Hylton566d9342004-09-07 15:28:01 +00001342 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001343
1344 def getChildren(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001345 return self.expr,
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001346
1347 def getChildNodes(self):
1348 return self.expr,
1349
1350 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001351 return "UnarySub(%s)" % (repr(self.expr),)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001352
1353class While(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001354 def __init__(self, test, body, else_, lineno=None):
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001355 self.test = test
1356 self.body = body
1357 self.else_ = else_
Jeremy Hylton566d9342004-09-07 15:28:01 +00001358 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001359
1360 def getChildren(self):
1361 children = []
1362 children.append(self.test)
1363 children.append(self.body)
1364 children.append(self.else_)
1365 return tuple(children)
1366
1367 def getChildNodes(self):
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +00001368 nodelist = []
1369 nodelist.append(self.test)
1370 nodelist.append(self.body)
Jeremy Hylton566d9342004-09-07 15:28:01 +00001371 if self.else_ is not None:
1372 nodelist.append(self.else_)
Martin v. Löwis0ac16ec2002-02-23 22:35:33 +00001373 return tuple(nodelist)
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001374
1375 def __repr__(self):
1376 return "While(%s, %s, %s)" % (repr(self.test), repr(self.body), repr(self.else_))
1377
Guido van Rossumc2e20742006-02-27 22:32:47 +00001378class With(Node):
1379 def __init__(self, expr, vars, body, lineno=None):
1380 self.expr = expr
1381 self.vars = vars
1382 self.body = body
1383 self.lineno = lineno
1384
1385 def getChildren(self):
1386 children = []
1387 children.append(self.expr)
1388 children.append(self.vars)
1389 children.append(self.body)
1390 return tuple(children)
1391
1392 def getChildNodes(self):
1393 nodelist = []
1394 nodelist.append(self.expr)
1395 if self.vars is not None:
1396 nodelist.append(self.vars)
1397 nodelist.append(self.body)
1398 return tuple(nodelist)
1399
1400 def __repr__(self):
1401 return "With(%s, %s, %s)" % (repr(self.expr), repr(self.vars), repr(self.body))
1402
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001403class Yield(Node):
Jeremy Hylton566d9342004-09-07 15:28:01 +00001404 def __init__(self, value, lineno=None):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001405 self.value = value
Jeremy Hylton566d9342004-09-07 15:28:01 +00001406 self.lineno = lineno
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001407
1408 def getChildren(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001409 return self.value,
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001410
1411 def getChildNodes(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001412 return self.value,
Jeremy Hyltonc8ed18a2001-09-17 20:17:02 +00001413
1414 def __repr__(self):
Anthony Baxterc2a5a632004-08-02 06:10:11 +00001415 return "Yield(%s)" % (repr(self.value),)
Raymond Hettinger354433a2004-05-19 08:20:33 +00001416
Jeremy Hylton566d9342004-09-07 15:28:01 +00001417for name, obj in globals().items():
1418 if isinstance(obj, type) and issubclass(obj, Node):
Tim Peters0e9980f2004-09-12 03:49:31 +00001419 nodes[name.lower()] = obj