blob: 93437d6ceef07800e8d6ebaa9b6e6716438aff05 [file] [log] [blame]
Tor Norbye3a2425a2013-11-04 10:16:08 -08001"""Python abstract syntax node definitions
2
3This file is automatically generated by Tools/compiler/astgen.py
4"""
5from compiler.consts import CO_VARARGS, CO_VARKEYWORDS
6
7def flatten(seq):
8 l = []
9 for elt in seq:
10 t = type(elt)
11 if t is tuple or t is list:
12 for elt2 in flatten(elt):
13 l.append(elt2)
14 else:
15 l.append(elt)
16 return l
17
18def flatten_nodes(seq):
19 return [n for n in flatten(seq) if isinstance(n, Node)]
20
21nodes = {}
22
23class Node:
24 """Abstract base class for ast nodes."""
25 def getChildren(self):
26 pass # implemented by subclasses
27 def __iter__(self):
28 for n in self.getChildren():
29 yield n
30 def asList(self): # for backwards compatibility
31 return self.getChildren()
32 def getChildNodes(self):
33 pass # implemented by subclasses
34
35class EmptyNode(Node):
36 pass
37
38class 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):
54 def __init__(self, (left, right), lineno=None):
55 self.left = left
56 self.right = right
57 self.lineno = lineno
58
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):
69 def __init__(self, nodes, lineno=None):
70 self.nodes = nodes
71 self.lineno = lineno
72
73 def getChildren(self):
74 return tuple(flatten(self.nodes))
75
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):
85 def __init__(self, expr, attrname, flags, lineno=None):
86 self.expr = expr
87 self.attrname = attrname
88 self.flags = flags
89 self.lineno = lineno
90
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):
101 def __init__(self, nodes, lineno=None):
102 self.nodes = nodes
103 self.lineno = lineno
104
105 def getChildren(self):
106 return tuple(flatten(self.nodes))
107
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):
117 def __init__(self, name, flags, lineno=None):
118 self.name = name
119 self.flags = flags
120 self.lineno = lineno
121
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):
132 def __init__(self, nodes, lineno=None):
133 self.nodes = nodes
134 self.lineno = lineno
135
136 def getChildren(self):
137 return tuple(flatten(self.nodes))
138
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):
148 def __init__(self, test, fail, lineno=None):
149 self.test = test
150 self.fail = fail
151 self.lineno = lineno
152
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)
162 if self.fail is not None:
163 nodelist.append(self.fail)
164 return tuple(nodelist)
165
166 def __repr__(self):
167 return "Assert(%s, %s)" % (repr(self.test), repr(self.fail))
168
169class Assign(Node):
170 def __init__(self, nodes, expr, lineno=None):
171 self.nodes = nodes
172 self.expr = expr
173 self.lineno = lineno
174
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):
191 def __init__(self, node, op, expr, lineno=None):
192 self.node = node
193 self.op = op
194 self.expr = expr
195 self.lineno = lineno
196
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):
207 def __init__(self, expr, lineno=None):
208 self.expr = expr
209 self.lineno = lineno
210
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):
221 def __init__(self, nodes, lineno=None):
222 self.nodes = nodes
223 self.lineno = lineno
224
225 def getChildren(self):
226 return tuple(flatten(self.nodes))
227
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):
237 def __init__(self, nodes, lineno=None):
238 self.nodes = nodes
239 self.lineno = lineno
240
241 def getChildren(self):
242 return tuple(flatten(self.nodes))
243
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):
253 def __init__(self, nodes, lineno=None):
254 self.nodes = nodes
255 self.lineno = lineno
256
257 def getChildren(self):
258 return tuple(flatten(self.nodes))
259
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):
269 def __init__(self, lineno=None):
270 self.lineno = lineno
271
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):
282 def __init__(self, node, args, star_args = None, dstar_args = None, lineno=None):
283 self.node = node
284 self.args = args
285 self.star_args = star_args
286 self.dstar_args = dstar_args
287 self.lineno = lineno
288
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))
301 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)
305 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):
311 def __init__(self, name, bases, doc, code, lineno=None):
312 self.name = name
313 self.bases = bases
314 self.doc = doc
315 self.code = code
316 self.lineno = lineno
317
318 def getChildren(self):
319 children = []
320 children.append(self.name)
321 children.extend(flatten(self.bases))
322 children.append(self.doc)
323 children.append(self.code)
324 return tuple(children)
325
326 def getChildNodes(self):
327 nodelist = []
328 nodelist.extend(flatten_nodes(self.bases))
329 nodelist.append(self.code)
330 return tuple(nodelist)
331
332 def __repr__(self):
333 return "Class(%s, %s, %s, %s)" % (repr(self.name), repr(self.bases), repr(self.doc), repr(self.code))
334
335class Compare(Node):
336 def __init__(self, expr, ops, lineno=None):
337 self.expr = expr
338 self.ops = ops
339 self.lineno = lineno
340
341 def getChildren(self):
342 children = []
343 children.append(self.expr)
344 children.extend(flatten(self.ops))
345 return tuple(children)
346
347 def getChildNodes(self):
348 nodelist = []
349 nodelist.append(self.expr)
350 nodelist.extend(flatten_nodes(self.ops))
351 return tuple(nodelist)
352
353 def __repr__(self):
354 return "Compare(%s, %s)" % (repr(self.expr), repr(self.ops))
355
356class Const(Node):
357 def __init__(self, value, lineno=None):
358 self.value = value
359 self.lineno = lineno
360
361 def getChildren(self):
362 return self.value,
363
364 def getChildNodes(self):
365 return ()
366
367 def __repr__(self):
368 return "Const(%s)" % (repr(self.value),)
369
370class Continue(Node):
371 def __init__(self, lineno=None):
372 self.lineno = lineno
373
374 def getChildren(self):
375 return ()
376
377 def getChildNodes(self):
378 return ()
379
380 def __repr__(self):
381 return "Continue()"
382
383class Decorators(Node):
384 def __init__(self, nodes, lineno=None):
385 self.nodes = nodes
386 self.lineno = lineno
387
388 def getChildren(self):
389 return tuple(flatten(self.nodes))
390
391 def getChildNodes(self):
392 nodelist = []
393 nodelist.extend(flatten_nodes(self.nodes))
394 return tuple(nodelist)
395
396 def __repr__(self):
397 return "Decorators(%s)" % (repr(self.nodes),)
398
399class Dict(Node):
400 def __init__(self, items, lineno=None):
401 self.items = items
402 self.lineno = lineno
403
404 def getChildren(self):
405 return tuple(flatten(self.items))
406
407 def getChildNodes(self):
408 nodelist = []
409 nodelist.extend(flatten_nodes(self.items))
410 return tuple(nodelist)
411
412 def __repr__(self):
413 return "Dict(%s)" % (repr(self.items),)
414
415class Discard(Node):
416 def __init__(self, expr, lineno=None):
417 self.expr = expr
418 self.lineno = lineno
419
420 def getChildren(self):
421 return self.expr,
422
423 def getChildNodes(self):
424 return self.expr,
425
426 def __repr__(self):
427 return "Discard(%s)" % (repr(self.expr),)
428
429class Div(Node):
430 def __init__(self, (left, right), lineno=None):
431 self.left = left
432 self.right = right
433 self.lineno = lineno
434
435 def getChildren(self):
436 return self.left, self.right
437
438 def getChildNodes(self):
439 return self.left, self.right
440
441 def __repr__(self):
442 return "Div((%s, %s))" % (repr(self.left), repr(self.right))
443
444class Ellipsis(Node):
445 def __init__(self, lineno=None):
446 self.lineno = lineno
447
448 def getChildren(self):
449 return ()
450
451 def getChildNodes(self):
452 return ()
453
454 def __repr__(self):
455 return "Ellipsis()"
456
457class Exec(Node):
458 def __init__(self, expr, locals, globals, lineno=None):
459 self.expr = expr
460 self.locals = locals
461 self.globals = globals
462 self.lineno = lineno
463
464 def getChildren(self):
465 children = []
466 children.append(self.expr)
467 children.append(self.locals)
468 children.append(self.globals)
469 return tuple(children)
470
471 def getChildNodes(self):
472 nodelist = []
473 nodelist.append(self.expr)
474 if self.locals is not None:
475 nodelist.append(self.locals)
476 if self.globals is not None:
477 nodelist.append(self.globals)
478 return tuple(nodelist)
479
480 def __repr__(self):
481 return "Exec(%s, %s, %s)" % (repr(self.expr), repr(self.locals), repr(self.globals))
482
483class FloorDiv(Node):
484 def __init__(self, (left, right), lineno=None):
485 self.left = left
486 self.right = right
487 self.lineno = lineno
488
489 def getChildren(self):
490 return self.left, self.right
491
492 def getChildNodes(self):
493 return self.left, self.right
494
495 def __repr__(self):
496 return "FloorDiv((%s, %s))" % (repr(self.left), repr(self.right))
497
498class For(Node):
499 def __init__(self, assign, list, body, else_, lineno=None):
500 self.assign = assign
501 self.list = list
502 self.body = body
503 self.else_ = else_
504 self.lineno = lineno
505
506 def getChildren(self):
507 children = []
508 children.append(self.assign)
509 children.append(self.list)
510 children.append(self.body)
511 children.append(self.else_)
512 return tuple(children)
513
514 def getChildNodes(self):
515 nodelist = []
516 nodelist.append(self.assign)
517 nodelist.append(self.list)
518 nodelist.append(self.body)
519 if self.else_ is not None:
520 nodelist.append(self.else_)
521 return tuple(nodelist)
522
523 def __repr__(self):
524 return "For(%s, %s, %s, %s)" % (repr(self.assign), repr(self.list), repr(self.body), repr(self.else_))
525
526class From(Node):
527 def __init__(self, modname, names, level, lineno=None):
528 self.modname = modname
529 self.names = names
530 self.level = level
531 self.lineno = lineno
532
533 def getChildren(self):
534 return self.modname, self.names, self.level
535
536 def getChildNodes(self):
537 return ()
538
539 def __repr__(self):
540 return "From(%s, %s, %s)" % (repr(self.modname), repr(self.names), repr(self.level))
541
542class Function(Node):
543 def __init__(self, decorators, name, argnames, defaults, flags, doc, code, lineno=None):
544 self.decorators = decorators
545 self.name = name
546 self.argnames = argnames
547 self.defaults = defaults
548 self.flags = flags
549 self.doc = doc
550 self.code = code
551 self.lineno = lineno
552 self.varargs = self.kwargs = None
553 if flags & CO_VARARGS:
554 self.varargs = 1
555 if flags & CO_VARKEYWORDS:
556 self.kwargs = 1
557
558
559
560 def getChildren(self):
561 children = []
562 children.append(self.decorators)
563 children.append(self.name)
564 children.append(self.argnames)
565 children.extend(flatten(self.defaults))
566 children.append(self.flags)
567 children.append(self.doc)
568 children.append(self.code)
569 return tuple(children)
570
571 def getChildNodes(self):
572 nodelist = []
573 if self.decorators is not None:
574 nodelist.append(self.decorators)
575 nodelist.extend(flatten_nodes(self.defaults))
576 nodelist.append(self.code)
577 return tuple(nodelist)
578
579 def __repr__(self):
580 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))
581
582class GenExpr(Node):
583 def __init__(self, code, lineno=None):
584 self.code = code
585 self.lineno = lineno
586 self.argnames = ['.0']
587 self.varargs = self.kwargs = None
588
589 def getChildren(self):
590 return self.code,
591
592 def getChildNodes(self):
593 return self.code,
594
595 def __repr__(self):
596 return "GenExpr(%s)" % (repr(self.code),)
597
598class GenExprFor(Node):
599 def __init__(self, assign, iter, ifs, lineno=None):
600 self.assign = assign
601 self.iter = iter
602 self.ifs = ifs
603 self.lineno = lineno
604 self.is_outmost = False
605
606
607 def getChildren(self):
608 children = []
609 children.append(self.assign)
610 children.append(self.iter)
611 children.extend(flatten(self.ifs))
612 return tuple(children)
613
614 def getChildNodes(self):
615 nodelist = []
616 nodelist.append(self.assign)
617 nodelist.append(self.iter)
618 nodelist.extend(flatten_nodes(self.ifs))
619 return tuple(nodelist)
620
621 def __repr__(self):
622 return "GenExprFor(%s, %s, %s)" % (repr(self.assign), repr(self.iter), repr(self.ifs))
623
624class GenExprIf(Node):
625 def __init__(self, test, lineno=None):
626 self.test = test
627 self.lineno = lineno
628
629 def getChildren(self):
630 return self.test,
631
632 def getChildNodes(self):
633 return self.test,
634
635 def __repr__(self):
636 return "GenExprIf(%s)" % (repr(self.test),)
637
638class GenExprInner(Node):
639 def __init__(self, expr, quals, lineno=None):
640 self.expr = expr
641 self.quals = quals
642 self.lineno = lineno
643
644 def getChildren(self):
645 children = []
646 children.append(self.expr)
647 children.extend(flatten(self.quals))
648 return tuple(children)
649
650 def getChildNodes(self):
651 nodelist = []
652 nodelist.append(self.expr)
653 nodelist.extend(flatten_nodes(self.quals))
654 return tuple(nodelist)
655
656 def __repr__(self):
657 return "GenExprInner(%s, %s)" % (repr(self.expr), repr(self.quals))
658
659class Getattr(Node):
660 def __init__(self, expr, attrname, lineno=None):
661 self.expr = expr
662 self.attrname = attrname
663 self.lineno = lineno
664
665 def getChildren(self):
666 return self.expr, self.attrname
667
668 def getChildNodes(self):
669 return self.expr,
670
671 def __repr__(self):
672 return "Getattr(%s, %s)" % (repr(self.expr), repr(self.attrname))
673
674class Global(Node):
675 def __init__(self, names, lineno=None):
676 self.names = names
677 self.lineno = lineno
678
679 def getChildren(self):
680 return self.names,
681
682 def getChildNodes(self):
683 return ()
684
685 def __repr__(self):
686 return "Global(%s)" % (repr(self.names),)
687
688class If(Node):
689 def __init__(self, tests, else_, lineno=None):
690 self.tests = tests
691 self.else_ = else_
692 self.lineno = lineno
693
694 def getChildren(self):
695 children = []
696 children.extend(flatten(self.tests))
697 children.append(self.else_)
698 return tuple(children)
699
700 def getChildNodes(self):
701 nodelist = []
702 nodelist.extend(flatten_nodes(self.tests))
703 if self.else_ is not None:
704 nodelist.append(self.else_)
705 return tuple(nodelist)
706
707 def __repr__(self):
708 return "If(%s, %s)" % (repr(self.tests), repr(self.else_))
709
710class IfExp(Node):
711 def __init__(self, test, then, else_, lineno=None):
712 self.test = test
713 self.then = then
714 self.else_ = else_
715 self.lineno = lineno
716
717 def getChildren(self):
718 return self.test, self.then, self.else_
719
720 def getChildNodes(self):
721 return self.test, self.then, self.else_
722
723 def __repr__(self):
724 return "IfExp(%s, %s, %s)" % (repr(self.test), repr(self.then), repr(self.else_))
725
726class Import(Node):
727 def __init__(self, names, lineno=None):
728 self.names = names
729 self.lineno = lineno
730
731 def getChildren(self):
732 return self.names,
733
734 def getChildNodes(self):
735 return ()
736
737 def __repr__(self):
738 return "Import(%s)" % (repr(self.names),)
739
740class Invert(Node):
741 def __init__(self, expr, lineno=None):
742 self.expr = expr
743 self.lineno = lineno
744
745 def getChildren(self):
746 return self.expr,
747
748 def getChildNodes(self):
749 return self.expr,
750
751 def __repr__(self):
752 return "Invert(%s)" % (repr(self.expr),)
753
754class Keyword(Node):
755 def __init__(self, name, expr, lineno=None):
756 self.name = name
757 self.expr = expr
758 self.lineno = lineno
759
760 def getChildren(self):
761 return self.name, self.expr
762
763 def getChildNodes(self):
764 return self.expr,
765
766 def __repr__(self):
767 return "Keyword(%s, %s)" % (repr(self.name), repr(self.expr))
768
769class Lambda(Node):
770 def __init__(self, argnames, defaults, flags, code, lineno=None):
771 self.argnames = argnames
772 self.defaults = defaults
773 self.flags = flags
774 self.code = code
775 self.lineno = lineno
776 self.varargs = self.kwargs = None
777 if flags & CO_VARARGS:
778 self.varargs = 1
779 if flags & CO_VARKEYWORDS:
780 self.kwargs = 1
781
782
783
784 def getChildren(self):
785 children = []
786 children.append(self.argnames)
787 children.extend(flatten(self.defaults))
788 children.append(self.flags)
789 children.append(self.code)
790 return tuple(children)
791
792 def getChildNodes(self):
793 nodelist = []
794 nodelist.extend(flatten_nodes(self.defaults))
795 nodelist.append(self.code)
796 return tuple(nodelist)
797
798 def __repr__(self):
799 return "Lambda(%s, %s, %s, %s)" % (repr(self.argnames), repr(self.defaults), repr(self.flags), repr(self.code))
800
801class LeftShift(Node):
802 def __init__(self, (left, right), lineno=None):
803 self.left = left
804 self.right = right
805 self.lineno = lineno
806
807 def getChildren(self):
808 return self.left, self.right
809
810 def getChildNodes(self):
811 return self.left, self.right
812
813 def __repr__(self):
814 return "LeftShift((%s, %s))" % (repr(self.left), repr(self.right))
815
816class List(Node):
817 def __init__(self, nodes, lineno=None):
818 self.nodes = nodes
819 self.lineno = lineno
820
821 def getChildren(self):
822 return tuple(flatten(self.nodes))
823
824 def getChildNodes(self):
825 nodelist = []
826 nodelist.extend(flatten_nodes(self.nodes))
827 return tuple(nodelist)
828
829 def __repr__(self):
830 return "List(%s)" % (repr(self.nodes),)
831
832class ListComp(Node):
833 def __init__(self, expr, quals, lineno=None):
834 self.expr = expr
835 self.quals = quals
836 self.lineno = lineno
837
838 def getChildren(self):
839 children = []
840 children.append(self.expr)
841 children.extend(flatten(self.quals))
842 return tuple(children)
843
844 def getChildNodes(self):
845 nodelist = []
846 nodelist.append(self.expr)
847 nodelist.extend(flatten_nodes(self.quals))
848 return tuple(nodelist)
849
850 def __repr__(self):
851 return "ListComp(%s, %s)" % (repr(self.expr), repr(self.quals))
852
853class ListCompFor(Node):
854 def __init__(self, assign, list, ifs, lineno=None):
855 self.assign = assign
856 self.list = list
857 self.ifs = ifs
858 self.lineno = lineno
859
860 def getChildren(self):
861 children = []
862 children.append(self.assign)
863 children.append(self.list)
864 children.extend(flatten(self.ifs))
865 return tuple(children)
866
867 def getChildNodes(self):
868 nodelist = []
869 nodelist.append(self.assign)
870 nodelist.append(self.list)
871 nodelist.extend(flatten_nodes(self.ifs))
872 return tuple(nodelist)
873
874 def __repr__(self):
875 return "ListCompFor(%s, %s, %s)" % (repr(self.assign), repr(self.list), repr(self.ifs))
876
877class ListCompIf(Node):
878 def __init__(self, test, lineno=None):
879 self.test = test
880 self.lineno = lineno
881
882 def getChildren(self):
883 return self.test,
884
885 def getChildNodes(self):
886 return self.test,
887
888 def __repr__(self):
889 return "ListCompIf(%s)" % (repr(self.test),)
890
891class Mod(Node):
892 def __init__(self, (left, right), lineno=None):
893 self.left = left
894 self.right = right
895 self.lineno = lineno
896
897 def getChildren(self):
898 return self.left, self.right
899
900 def getChildNodes(self):
901 return self.left, self.right
902
903 def __repr__(self):
904 return "Mod((%s, %s))" % (repr(self.left), repr(self.right))
905
906class Module(Node):
907 def __init__(self, doc, node, lineno=None):
908 self.doc = doc
909 self.node = node
910 self.lineno = lineno
911
912 def getChildren(self):
913 return self.doc, self.node
914
915 def getChildNodes(self):
916 return self.node,
917
918 def __repr__(self):
919 return "Module(%s, %s)" % (repr(self.doc), repr(self.node))
920
921class Mul(Node):
922 def __init__(self, (left, right), lineno=None):
923 self.left = left
924 self.right = right
925 self.lineno = lineno
926
927 def getChildren(self):
928 return self.left, self.right
929
930 def getChildNodes(self):
931 return self.left, self.right
932
933 def __repr__(self):
934 return "Mul((%s, %s))" % (repr(self.left), repr(self.right))
935
936class Name(Node):
937 def __init__(self, name, lineno=None):
938 self.name = name
939 self.lineno = lineno
940
941 def getChildren(self):
942 return self.name,
943
944 def getChildNodes(self):
945 return ()
946
947 def __repr__(self):
948 return "Name(%s)" % (repr(self.name),)
949
950class Not(Node):
951 def __init__(self, expr, lineno=None):
952 self.expr = expr
953 self.lineno = lineno
954
955 def getChildren(self):
956 return self.expr,
957
958 def getChildNodes(self):
959 return self.expr,
960
961 def __repr__(self):
962 return "Not(%s)" % (repr(self.expr),)
963
964class Or(Node):
965 def __init__(self, nodes, lineno=None):
966 self.nodes = nodes
967 self.lineno = lineno
968
969 def getChildren(self):
970 return tuple(flatten(self.nodes))
971
972 def getChildNodes(self):
973 nodelist = []
974 nodelist.extend(flatten_nodes(self.nodes))
975 return tuple(nodelist)
976
977 def __repr__(self):
978 return "Or(%s)" % (repr(self.nodes),)
979
980class Pass(Node):
981 def __init__(self, lineno=None):
982 self.lineno = lineno
983
984 def getChildren(self):
985 return ()
986
987 def getChildNodes(self):
988 return ()
989
990 def __repr__(self):
991 return "Pass()"
992
993class Power(Node):
994 def __init__(self, (left, right), lineno=None):
995 self.left = left
996 self.right = right
997 self.lineno = lineno
998
999 def getChildren(self):
1000 return self.left, self.right
1001
1002 def getChildNodes(self):
1003 return self.left, self.right
1004
1005 def __repr__(self):
1006 return "Power((%s, %s))" % (repr(self.left), repr(self.right))
1007
1008class Print(Node):
1009 def __init__(self, nodes, dest, lineno=None):
1010 self.nodes = nodes
1011 self.dest = dest
1012 self.lineno = lineno
1013
1014 def getChildren(self):
1015 children = []
1016 children.extend(flatten(self.nodes))
1017 children.append(self.dest)
1018 return tuple(children)
1019
1020 def getChildNodes(self):
1021 nodelist = []
1022 nodelist.extend(flatten_nodes(self.nodes))
1023 if self.dest is not None:
1024 nodelist.append(self.dest)
1025 return tuple(nodelist)
1026
1027 def __repr__(self):
1028 return "Print(%s, %s)" % (repr(self.nodes), repr(self.dest))
1029
1030class Printnl(Node):
1031 def __init__(self, nodes, dest, lineno=None):
1032 self.nodes = nodes
1033 self.dest = dest
1034 self.lineno = lineno
1035
1036 def getChildren(self):
1037 children = []
1038 children.extend(flatten(self.nodes))
1039 children.append(self.dest)
1040 return tuple(children)
1041
1042 def getChildNodes(self):
1043 nodelist = []
1044 nodelist.extend(flatten_nodes(self.nodes))
1045 if self.dest is not None:
1046 nodelist.append(self.dest)
1047 return tuple(nodelist)
1048
1049 def __repr__(self):
1050 return "Printnl(%s, %s)" % (repr(self.nodes), repr(self.dest))
1051
1052class Raise(Node):
1053 def __init__(self, expr1, expr2, expr3, lineno=None):
1054 self.expr1 = expr1
1055 self.expr2 = expr2
1056 self.expr3 = expr3
1057 self.lineno = lineno
1058
1059 def getChildren(self):
1060 children = []
1061 children.append(self.expr1)
1062 children.append(self.expr2)
1063 children.append(self.expr3)
1064 return tuple(children)
1065
1066 def getChildNodes(self):
1067 nodelist = []
1068 if self.expr1 is not None:
1069 nodelist.append(self.expr1)
1070 if self.expr2 is not None:
1071 nodelist.append(self.expr2)
1072 if self.expr3 is not None:
1073 nodelist.append(self.expr3)
1074 return tuple(nodelist)
1075
1076 def __repr__(self):
1077 return "Raise(%s, %s, %s)" % (repr(self.expr1), repr(self.expr2), repr(self.expr3))
1078
1079class Return(Node):
1080 def __init__(self, value, lineno=None):
1081 self.value = value
1082 self.lineno = lineno
1083
1084 def getChildren(self):
1085 return self.value,
1086
1087 def getChildNodes(self):
1088 return self.value,
1089
1090 def __repr__(self):
1091 return "Return(%s)" % (repr(self.value),)
1092
1093class RightShift(Node):
1094 def __init__(self, (left, right), lineno=None):
1095 self.left = left
1096 self.right = right
1097 self.lineno = lineno
1098
1099 def getChildren(self):
1100 return self.left, self.right
1101
1102 def getChildNodes(self):
1103 return self.left, self.right
1104
1105 def __repr__(self):
1106 return "RightShift((%s, %s))" % (repr(self.left), repr(self.right))
1107
1108class Slice(Node):
1109 def __init__(self, expr, flags, lower, upper, lineno=None):
1110 self.expr = expr
1111 self.flags = flags
1112 self.lower = lower
1113 self.upper = upper
1114 self.lineno = lineno
1115
1116 def getChildren(self):
1117 children = []
1118 children.append(self.expr)
1119 children.append(self.flags)
1120 children.append(self.lower)
1121 children.append(self.upper)
1122 return tuple(children)
1123
1124 def getChildNodes(self):
1125 nodelist = []
1126 nodelist.append(self.expr)
1127 if self.lower is not None:
1128 nodelist.append(self.lower)
1129 if self.upper is not None:
1130 nodelist.append(self.upper)
1131 return tuple(nodelist)
1132
1133 def __repr__(self):
1134 return "Slice(%s, %s, %s, %s)" % (repr(self.expr), repr(self.flags), repr(self.lower), repr(self.upper))
1135
1136class Sliceobj(Node):
1137 def __init__(self, nodes, lineno=None):
1138 self.nodes = nodes
1139 self.lineno = lineno
1140
1141 def getChildren(self):
1142 return tuple(flatten(self.nodes))
1143
1144 def getChildNodes(self):
1145 nodelist = []
1146 nodelist.extend(flatten_nodes(self.nodes))
1147 return tuple(nodelist)
1148
1149 def __repr__(self):
1150 return "Sliceobj(%s)" % (repr(self.nodes),)
1151
1152class Stmt(Node):
1153 def __init__(self, nodes, lineno=None):
1154 self.nodes = nodes
1155 self.lineno = lineno
1156
1157 def getChildren(self):
1158 return tuple(flatten(self.nodes))
1159
1160 def getChildNodes(self):
1161 nodelist = []
1162 nodelist.extend(flatten_nodes(self.nodes))
1163 return tuple(nodelist)
1164
1165 def __repr__(self):
1166 return "Stmt(%s)" % (repr(self.nodes),)
1167
1168class Sub(Node):
1169 def __init__(self, (left, right), lineno=None):
1170 self.left = left
1171 self.right = right
1172 self.lineno = lineno
1173
1174 def getChildren(self):
1175 return self.left, self.right
1176
1177 def getChildNodes(self):
1178 return self.left, self.right
1179
1180 def __repr__(self):
1181 return "Sub((%s, %s))" % (repr(self.left), repr(self.right))
1182
1183class Subscript(Node):
1184 def __init__(self, expr, flags, subs, lineno=None):
1185 self.expr = expr
1186 self.flags = flags
1187 self.subs = subs
1188 self.lineno = lineno
1189
1190 def getChildren(self):
1191 children = []
1192 children.append(self.expr)
1193 children.append(self.flags)
1194 children.extend(flatten(self.subs))
1195 return tuple(children)
1196
1197 def getChildNodes(self):
1198 nodelist = []
1199 nodelist.append(self.expr)
1200 nodelist.extend(flatten_nodes(self.subs))
1201 return tuple(nodelist)
1202
1203 def __repr__(self):
1204 return "Subscript(%s, %s, %s)" % (repr(self.expr), repr(self.flags), repr(self.subs))
1205
1206class TryExcept(Node):
1207 def __init__(self, body, handlers, else_, lineno=None):
1208 self.body = body
1209 self.handlers = handlers
1210 self.else_ = else_
1211 self.lineno = lineno
1212
1213 def getChildren(self):
1214 children = []
1215 children.append(self.body)
1216 children.extend(flatten(self.handlers))
1217 children.append(self.else_)
1218 return tuple(children)
1219
1220 def getChildNodes(self):
1221 nodelist = []
1222 nodelist.append(self.body)
1223 nodelist.extend(flatten_nodes(self.handlers))
1224 if self.else_ is not None:
1225 nodelist.append(self.else_)
1226 return tuple(nodelist)
1227
1228 def __repr__(self):
1229 return "TryExcept(%s, %s, %s)" % (repr(self.body), repr(self.handlers), repr(self.else_))
1230
1231class TryFinally(Node):
1232 def __init__(self, body, final, lineno=None):
1233 self.body = body
1234 self.final = final
1235 self.lineno = lineno
1236
1237 def getChildren(self):
1238 return self.body, self.final
1239
1240 def getChildNodes(self):
1241 return self.body, self.final
1242
1243 def __repr__(self):
1244 return "TryFinally(%s, %s)" % (repr(self.body), repr(self.final))
1245
1246class Tuple(Node):
1247 def __init__(self, nodes, lineno=None):
1248 self.nodes = nodes
1249 self.lineno = lineno
1250
1251 def getChildren(self):
1252 return tuple(flatten(self.nodes))
1253
1254 def getChildNodes(self):
1255 nodelist = []
1256 nodelist.extend(flatten_nodes(self.nodes))
1257 return tuple(nodelist)
1258
1259 def __repr__(self):
1260 return "Tuple(%s)" % (repr(self.nodes),)
1261
1262class UnaryAdd(Node):
1263 def __init__(self, expr, lineno=None):
1264 self.expr = expr
1265 self.lineno = lineno
1266
1267 def getChildren(self):
1268 return self.expr,
1269
1270 def getChildNodes(self):
1271 return self.expr,
1272
1273 def __repr__(self):
1274 return "UnaryAdd(%s)" % (repr(self.expr),)
1275
1276class UnarySub(Node):
1277 def __init__(self, expr, lineno=None):
1278 self.expr = expr
1279 self.lineno = lineno
1280
1281 def getChildren(self):
1282 return self.expr,
1283
1284 def getChildNodes(self):
1285 return self.expr,
1286
1287 def __repr__(self):
1288 return "UnarySub(%s)" % (repr(self.expr),)
1289
1290class While(Node):
1291 def __init__(self, test, body, else_, lineno=None):
1292 self.test = test
1293 self.body = body
1294 self.else_ = else_
1295 self.lineno = lineno
1296
1297 def getChildren(self):
1298 children = []
1299 children.append(self.test)
1300 children.append(self.body)
1301 children.append(self.else_)
1302 return tuple(children)
1303
1304 def getChildNodes(self):
1305 nodelist = []
1306 nodelist.append(self.test)
1307 nodelist.append(self.body)
1308 if self.else_ is not None:
1309 nodelist.append(self.else_)
1310 return tuple(nodelist)
1311
1312 def __repr__(self):
1313 return "While(%s, %s, %s)" % (repr(self.test), repr(self.body), repr(self.else_))
1314
1315class With(Node):
1316 def __init__(self, expr, vars, body, lineno=None):
1317 self.expr = expr
1318 self.vars = vars
1319 self.body = body
1320 self.lineno = lineno
1321
1322 def getChildren(self):
1323 children = []
1324 children.append(self.expr)
1325 children.append(self.vars)
1326 children.append(self.body)
1327 return tuple(children)
1328
1329 def getChildNodes(self):
1330 nodelist = []
1331 nodelist.append(self.expr)
1332 if self.vars is not None:
1333 nodelist.append(self.vars)
1334 nodelist.append(self.body)
1335 return tuple(nodelist)
1336
1337 def __repr__(self):
1338 return "With(%s, %s, %s)" % (repr(self.expr), repr(self.vars), repr(self.body))
1339
1340class Yield(Node):
1341 def __init__(self, value, lineno=None):
1342 self.value = value
1343 self.lineno = lineno
1344
1345 def getChildren(self):
1346 return self.value,
1347
1348 def getChildNodes(self):
1349 return self.value,
1350
1351 def __repr__(self):
1352 return "Yield(%s)" % (repr(self.value),)
1353
1354for name, obj in globals().items():
1355 if isinstance(obj, type) and issubclass(obj, Node):
1356 nodes[name.lower()] = obj