Author: Carl Friedrich Bolz <[email protected]>
Branch: py3k-update
Changeset: r83996:72878888045c
Date: 2016-04-28 00:38 +0300
http://bitbucket.org/pypy/pypy/changeset/72878888045c/
Log: merge e35996d1c1b6
diff --git a/pypy/interpreter/astcompiler/astbuilder.py
b/pypy/interpreter/astcompiler/astbuilder.py
--- a/pypy/interpreter/astcompiler/astbuilder.py
+++ b/pypy/interpreter/astcompiler/astbuilder.py
@@ -53,24 +53,24 @@
n = self.root_node
if n.type == syms.file_input:
stmts = []
- for i in range(len(n.children) - 1):
- stmt = n.children[i]
+ for i in range(n.num_children() - 1):
+ stmt = n.get_child(i)
if stmt.type == tokens.NEWLINE:
continue
sub_stmts_count = self.number_of_statements(stmt)
if sub_stmts_count == 1:
stmts.append(self.handle_stmt(stmt))
else:
- stmt = stmt.children[0]
+ stmt = stmt.get_child(0)
for j in range(sub_stmts_count):
- small_stmt = stmt.children[j * 2]
+ small_stmt = stmt.get_child(j * 2)
stmts.append(self.handle_stmt(small_stmt))
return ast.Module(stmts)
elif n.type == syms.eval_input:
- body = self.handle_testlist(n.children[0])
+ body = self.handle_testlist(n.get_child(0))
return ast.Expression(body)
elif n.type == syms.single_input:
- first_child = n.children[0]
+ first_child = n.get_child(0)
if first_child.type == tokens.NEWLINE:
# An empty line.
return ast.Interactive([])
@@ -80,8 +80,8 @@
stmts = [self.handle_stmt(first_child)]
else:
stmts = []
- for i in range(0, len(first_child.children), 2):
- stmt = first_child.children[i]
+ for i in range(0, first_child.num_children(), 2):
+ stmt = first_child.get_child(i)
if stmt.type == tokens.NEWLINE:
break
stmts.append(self.handle_stmt(stmt))
@@ -95,16 +95,16 @@
if stmt_type == syms.compound_stmt:
return 1
elif stmt_type == syms.stmt:
- return self.number_of_statements(n.children[0])
+ return self.number_of_statements(n.get_child(0))
elif stmt_type == syms.simple_stmt:
# Divide to remove semi-colons.
- return len(n.children) // 2
+ return n.num_children() // 2
else:
raise AssertionError("non-statement node")
def error(self, msg, n):
"""Raise a SyntaxError with the lineno and column set to n's."""
- raise SyntaxError(msg, n.lineno, n.column,
+ raise SyntaxError(msg, n.get_lineno(), n.get_column(),
filename=self.compile_info.filename)
def error_ast(self, msg, ast_node):
@@ -130,34 +130,34 @@
self.error_ast("cannot assign to %s" % (e.name,), e.node)
def handle_del_stmt(self, del_node):
- targets = self.handle_exprlist(del_node.children[1], ast.Del)
- return ast.Delete(targets, del_node.lineno, del_node.column)
+ targets = self.handle_exprlist(del_node.get_child(1), ast.Del)
+ return ast.Delete(targets, del_node.get_lineno(),
del_node.get_column())
def handle_flow_stmt(self, flow_node):
- first_child = flow_node.children[0]
+ first_child = flow_node.get_child(0)
first_child_type = first_child.type
if first_child_type == syms.break_stmt:
- return ast.Break(flow_node.lineno, flow_node.column)
+ return ast.Break(flow_node.get_lineno(), flow_node.get_column())
elif first_child_type == syms.continue_stmt:
- return ast.Continue(flow_node.lineno, flow_node.column)
+ return ast.Continue(flow_node.get_lineno(), flow_node.get_column())
elif first_child_type == syms.yield_stmt:
- yield_expr = self.handle_expr(first_child.children[0])
- return ast.Expr(yield_expr, flow_node.lineno, flow_node.column)
+ yield_expr = self.handle_expr(first_child.get_child(0))
+ return ast.Expr(yield_expr, flow_node.get_lineno(),
flow_node.get_column())
elif first_child_type == syms.return_stmt:
- if len(first_child.children) == 1:
+ if first_child.num_children() == 1:
values = None
else:
- values = self.handle_testlist(first_child.children[1])
- return ast.Return(values, flow_node.lineno, flow_node.column)
+ values = self.handle_testlist(first_child.get_child(1))
+ return ast.Return(values, flow_node.get_lineno(),
flow_node.get_column())
elif first_child_type == syms.raise_stmt:
exc = None
cause = None
- child_count = len(first_child.children)
+ child_count = first_child.num_children()
if child_count >= 2:
- exc = self.handle_expr(first_child.children[1])
+ exc = self.handle_expr(first_child.get_child(1))
if child_count >= 4:
- cause = self.handle_expr(first_child.children[3])
- return ast.Raise(exc, cause, flow_node.lineno, flow_node.column)
+ cause = self.handle_expr(first_child.get_child(3))
+ return ast.Raise(exc, cause, flow_node.get_lineno(),
flow_node.get_column())
else:
raise AssertionError("unknown flow statement")
@@ -165,33 +165,33 @@
while True:
import_name_type = import_name.type
if import_name_type == syms.import_as_name:
- name = self.new_identifier(import_name.children[0].value)
- if len(import_name.children) == 3:
+ name =
self.new_identifier(import_name.get_child(0).get_value())
+ if import_name.num_children() == 3:
as_name = self.new_identifier(
- import_name.children[2].value)
- self.check_forbidden_name(as_name, import_name.children[2])
+ import_name.get_child(2).get_value())
+ self.check_forbidden_name(as_name,
import_name.get_child(2))
else:
as_name = None
- self.check_forbidden_name(name, import_name.children[0])
+ self.check_forbidden_name(name, import_name.get_child(0))
return ast.alias(name, as_name)
elif import_name_type == syms.dotted_as_name:
- if len(import_name.children) == 1:
- import_name = import_name.children[0]
+ if import_name.num_children() == 1:
+ import_name = import_name.get_child(0)
continue
- alias = self.alias_for_import_name(import_name.children[0],
+ alias = self.alias_for_import_name(import_name.get_child(0),
store=False)
- asname_node = import_name.children[2]
- alias.asname = self.new_identifier(asname_node.value)
+ asname_node = import_name.get_child(2)
+ alias.asname = self.new_identifier(asname_node.get_value())
self.check_forbidden_name(alias.asname, asname_node)
return alias
elif import_name_type == syms.dotted_name:
- if len(import_name.children) == 1:
- name = self.new_identifier(import_name.children[0].value)
+ if import_name.num_children() == 1:
+ name =
self.new_identifier(import_name.get_child(0).get_value())
if store:
- self.check_forbidden_name(name,
import_name.children[0])
+ self.check_forbidden_name(name,
import_name.get_child(0))
return ast.alias(name, None)
- name_parts = [import_name.children[i].value
- for i in range(0, len(import_name.children), 2)]
+ name_parts = [import_name.get_child(i).get_value()
+ for i in range(0, import_name.num_children(), 2)]
name = ".".join(name_parts)
return ast.alias(name, None)
elif import_name_type == tokens.STAR:
@@ -200,20 +200,20 @@
raise AssertionError("unknown import name")
def handle_import_stmt(self, import_node):
- import_node = import_node.children[0]
+ import_node = import_node.get_child(0)
if import_node.type == syms.import_name:
- dotted_as_names = import_node.children[1]
- aliases = [self.alias_for_import_name(dotted_as_names.children[i])
- for i in range(0, len(dotted_as_names.children), 2)]
- return ast.Import(aliases, import_node.lineno, import_node.column)
+ dotted_as_names = import_node.get_child(1)
+ aliases = [self.alias_for_import_name(dotted_as_names.get_child(i))
+ for i in range(0, dotted_as_names.num_children(), 2)]
+ return ast.Import(aliases, import_node.get_lineno(),
import_node.get_column())
elif import_node.type == syms.import_from:
- child_count = len(import_node.children)
+ child_count = import_node.num_children()
module = None
modname = None
i = 1
dot_count = 0
while i < child_count:
- child = import_node.children[i]
+ child = import_node.get_child(i)
child_type = child.type
if child_type == syms.dotted_name:
module = self.alias_for_import_name(child, False)
@@ -227,16 +227,16 @@
i += 1
dot_count += 1
i += 1
- after_import_type = import_node.children[i].type
+ after_import_type = import_node.get_child(i).type
star_import = False
if after_import_type == tokens.STAR:
- names_node = import_node.children[i]
+ names_node = import_node.get_child(i)
star_import = True
elif after_import_type == tokens.LPAR:
- names_node = import_node.children[i + 1]
+ names_node = import_node.get_child(i + 1)
elif after_import_type == syms.import_as_names:
- names_node = import_node.children[i]
- if len(names_node.children) % 2 == 0:
+ names_node = import_node.get_child(i)
+ if names_node.num_children() % 2 == 0:
self.error("trailing comma is only allowed with "
"surronding parenthesis", names_node)
else:
@@ -244,307 +244,308 @@
if star_import:
aliases = [self.alias_for_import_name(names_node)]
else:
- aliases = [self.alias_for_import_name(names_node.children[i])
- for i in range(0, len(names_node.children), 2)]
+ aliases = [self.alias_for_import_name(names_node.get_child(i))
+ for i in range(0, names_node.num_children(), 2)]
if module is not None:
modname = module.name
return ast.ImportFrom(modname, aliases, dot_count,
- import_node.lineno, import_node.column)
+ import_node.get_lineno(),
import_node.get_column())
else:
raise AssertionError("unknown import node")
def handle_global_stmt(self, global_node):
- names = [self.new_identifier(global_node.children[i].value)
- for i in range(1, len(global_node.children), 2)]
- return ast.Global(names, global_node.lineno, global_node.column)
+ names = [self.new_identifier(global_node.get_child(i).get_value())
+ for i in range(1, global_node.num_children(), 2)]
+ return ast.Global(names, global_node.get_lineno(),
global_node.get_column())
def handle_nonlocal_stmt(self, nonlocal_node):
- names = [self.new_identifier(nonlocal_node.children[i].value)
- for i in range(1, len(nonlocal_node.children), 2)]
- return ast.Nonlocal(names, nonlocal_node.lineno, nonlocal_node.column)
+ names = [self.new_identifier(nonlocal_node.get_child(i).get_value())
+ for i in range(1, nonlocal_node.num_children(), 2)]
+ return ast.Nonlocal(names, nonlocal_node.get_lineno(),
nonlocal_node.get_column())
def handle_assert_stmt(self, assert_node):
- expr = self.handle_expr(assert_node.children[1])
+ expr = self.handle_expr(assert_node.get_child(1))
msg = None
- if len(assert_node.children) == 4:
- msg = self.handle_expr(assert_node.children[3])
- return ast.Assert(expr, msg, assert_node.lineno, assert_node.column)
+ if assert_node.num_children() == 4:
+ msg = self.handle_expr(assert_node.get_child(3))
+ return ast.Assert(expr, msg, assert_node.get_lineno(),
assert_node.get_column())
def handle_suite(self, suite_node):
- first_child = suite_node.children[0]
+ first_child = suite_node.get_child(0)
if first_child.type == syms.simple_stmt:
- end = len(first_child.children) - 1
- if first_child.children[end - 1].type == tokens.SEMI:
+ end = first_child.num_children() - 1
+ if first_child.get_child(end - 1).type == tokens.SEMI:
end -= 1
- stmts = [self.handle_stmt(first_child.children[i])
+ stmts = [self.handle_stmt(first_child.get_child(i))
for i in range(0, end, 2)]
else:
stmts = []
- for i in range(2, len(suite_node.children) - 1):
- stmt = suite_node.children[i]
+ for i in range(2, suite_node.num_children() - 1):
+ stmt = suite_node.get_child(i)
stmt_count = self.number_of_statements(stmt)
if stmt_count == 1:
stmts.append(self.handle_stmt(stmt))
else:
- simple_stmt = stmt.children[0]
- for j in range(0, len(simple_stmt.children), 2):
- stmt = simple_stmt.children[j]
- if not stmt.children:
+ simple_stmt = stmt.get_child(0)
+ for j in range(0, simple_stmt.num_children(), 2):
+ stmt = simple_stmt.get_child(j)
+ if not stmt.num_children():
break
stmts.append(self.handle_stmt(stmt))
return stmts
def handle_if_stmt(self, if_node):
- child_count = len(if_node.children)
+ child_count = if_node.num_children()
if child_count == 4:
- test = self.handle_expr(if_node.children[1])
- suite = self.handle_suite(if_node.children[3])
- return ast.If(test, suite, None, if_node.lineno, if_node.column)
- otherwise_string = if_node.children[4].value
+ test = self.handle_expr(if_node.get_child(1))
+ suite = self.handle_suite(if_node.get_child(3))
+ return ast.If(test, suite, None, if_node.get_lineno(),
if_node.get_column())
+ otherwise_string = if_node.get_child(4).get_value()
if otherwise_string == "else":
- test = self.handle_expr(if_node.children[1])
- suite = self.handle_suite(if_node.children[3])
- else_suite = self.handle_suite(if_node.children[6])
- return ast.If(test, suite, else_suite, if_node.lineno,
- if_node.column)
+ test = self.handle_expr(if_node.get_child(1))
+ suite = self.handle_suite(if_node.get_child(3))
+ else_suite = self.handle_suite(if_node.get_child(6))
+ return ast.If(test, suite, else_suite, if_node.get_lineno(),
+ if_node.get_column())
elif otherwise_string == "elif":
elif_count = child_count - 4
- after_elif = if_node.children[elif_count + 1]
+ after_elif = if_node.get_child(elif_count + 1)
if after_elif.type == tokens.NAME and \
- after_elif.value == "else":
+ after_elif.get_value() == "else":
has_else = True
elif_count -= 3
else:
has_else = False
elif_count /= 4
if has_else:
- last_elif = if_node.children[-6]
+ last_elif = if_node.get_child(-6)
last_elif_test = self.handle_expr(last_elif)
- elif_body = self.handle_suite(if_node.children[-4])
- else_body = self.handle_suite(if_node.children[-1])
+ elif_body = self.handle_suite(if_node.get_child(-4))
+ else_body = self.handle_suite(if_node.get_child(-1))
otherwise = [ast.If(last_elif_test, elif_body, else_body,
- last_elif.lineno, last_elif.column)]
+ last_elif.get_lineno(),
last_elif.get_column())]
elif_count -= 1
else:
otherwise = None
for i in range(elif_count):
offset = 5 + (elif_count - i - 1) * 4
- elif_test_node = if_node.children[offset]
+ elif_test_node = if_node.get_child(offset)
elif_test = self.handle_expr(elif_test_node)
- elif_body = self.handle_suite(if_node.children[offset + 2])
+ elif_body = self.handle_suite(if_node.get_child(offset + 2))
new_if = ast.If(elif_test, elif_body, otherwise,
- elif_test_node.lineno, elif_test_node.column)
+ elif_test_node.get_lineno(),
elif_test_node.get_column())
otherwise = [new_if]
- expr = self.handle_expr(if_node.children[1])
- body = self.handle_suite(if_node.children[3])
- return ast.If(expr, body, otherwise, if_node.lineno,
if_node.column)
+ expr = self.handle_expr(if_node.get_child(1))
+ body = self.handle_suite(if_node.get_child(3))
+ return ast.If(expr, body, otherwise, if_node.get_lineno(),
if_node.get_column())
else:
raise AssertionError("unknown if statement configuration")
def handle_while_stmt(self, while_node):
- loop_test = self.handle_expr(while_node.children[1])
- body = self.handle_suite(while_node.children[3])
- if len(while_node.children) == 7:
- otherwise = self.handle_suite(while_node.children[6])
+ loop_test = self.handle_expr(while_node.get_child(1))
+ body = self.handle_suite(while_node.get_child(3))
+ if while_node.num_children() == 7:
+ otherwise = self.handle_suite(while_node.get_child(6))
else:
otherwise = None
- return ast.While(loop_test, body, otherwise, while_node.lineno,
- while_node.column)
+ return ast.While(loop_test, body, otherwise, while_node.get_lineno(),
+ while_node.get_column())
def handle_for_stmt(self, for_node):
- target_node = for_node.children[1]
+ target_node = for_node.get_child(1)
target_as_exprlist = self.handle_exprlist(target_node, ast.Store)
- if len(target_node.children) == 1:
+ if target_node.num_children() == 1:
target = target_as_exprlist[0]
else:
target = ast.Tuple(target_as_exprlist, ast.Store,
- target_node.lineno, target_node.column)
- expr = self.handle_testlist(for_node.children[3])
- body = self.handle_suite(for_node.children[5])
- if len(for_node.children) == 9:
- otherwise = self.handle_suite(for_node.children[8])
+ target_node.get_lineno(),
target_node.get_column())
+ expr = self.handle_testlist(for_node.get_child(3))
+ body = self.handle_suite(for_node.get_child(5))
+ if for_node.num_children() == 9:
+ otherwise = self.handle_suite(for_node.get_child(8))
else:
otherwise = None
- return ast.For(target, expr, body, otherwise, for_node.lineno,
- for_node.column)
+ return ast.For(target, expr, body, otherwise, for_node.get_lineno(),
+ for_node.get_column())
def handle_except_clause(self, exc, body):
test = None
name = None
suite = self.handle_suite(body)
- child_count = len(exc.children)
+ child_count = exc.num_children()
if child_count >= 2:
- test = self.handle_expr(exc.children[1])
+ test = self.handle_expr(exc.get_child(1))
if child_count == 4:
- name_node = exc.children[3]
- name = self.new_identifier(name_node.value)
+ name_node = exc.get_child(3)
+ name = self.new_identifier(name_node.get_value())
self.check_forbidden_name(name, name_node)
- return ast.ExceptHandler(test, name, suite, exc.lineno, exc.column)
+ return ast.ExceptHandler(test, name, suite, exc.get_lineno(),
exc.get_column())
def handle_try_stmt(self, try_node):
- body = self.handle_suite(try_node.children[2])
- child_count = len(try_node.children)
+ body = self.handle_suite(try_node.get_child(2))
+ child_count = try_node.num_children()
except_count = (child_count - 3 ) // 3
otherwise = None
finally_suite = None
- possible_extra_clause = try_node.children[-3]
+ possible_extra_clause = try_node.get_child(-3)
if possible_extra_clause.type == tokens.NAME:
- if possible_extra_clause.value == "finally":
+ if possible_extra_clause.get_value() == "finally":
if child_count >= 9 and \
- try_node.children[-6].type == tokens.NAME:
- otherwise = self.handle_suite(try_node.children[-4])
+ try_node.get_child(-6).type == tokens.NAME:
+ otherwise = self.handle_suite(try_node.get_child(-4))
except_count -= 1
- finally_suite = self.handle_suite(try_node.children[-1])
+ finally_suite = self.handle_suite(try_node.get_child(-1))
except_count -= 1
else:
- otherwise = self.handle_suite(try_node.children[-1])
+ otherwise = self.handle_suite(try_node.get_child(-1))
except_count -= 1
handlers = []
if except_count:
for i in range(except_count):
base_offset = i * 3
- exc = try_node.children[3 + base_offset]
- except_body = try_node.children[5 + base_offset]
+ exc = try_node.get_child(3 + base_offset)
+ except_body = try_node.get_child(5 + base_offset)
handlers.append(self.handle_except_clause(exc, except_body))
return ast.Try(body, handlers, otherwise, finally_suite,
- try_node.lineno, try_node.column)
+ try_node.get_lineno(), try_node.get_column())
def handle_with_stmt(self, with_node):
- body = self.handle_suite(with_node.children[-1])
- i = len(with_node.children) - 1
+ body = self.handle_suite(with_node.get_child(-1))
+ i = with_node.num_children() - 1
while True:
i -= 2
- item = with_node.children[i]
- test = self.handle_expr(item.children[0])
- if len(item.children) == 3:
- target = self.handle_expr(item.children[2])
+ item = with_node.get_child(i)
+ test = self.handle_expr(item.get_child(0))
+ if item.num_children() == 3:
+ target = self.handle_expr(item.get_child(2))
self.set_context(target, ast.Store)
else:
target = None
- wi = ast.With(test, target, body, with_node.lineno,
- with_node.column)
+ wi = ast.With(test, target, body, with_node.get_lineno(),
+ with_node.get_column())
if i == 1:
break
body = [wi]
return wi
def handle_with_item(self, item_node):
- test = self.handle_expr(item_node.children[0])
- if len(item_node.children) == 3:
- target = self.handle_expr(item_node.children[2])
+ test = self.handle_expr(item_node.get_child(0))
+ if item_node.num_children() == 3:
+ target = self.handle_expr(item_node.get_child(2))
self.set_context(target, ast.Store)
else:
target = None
return ast.withitem(test, target)
def handle_with_stmt(self, with_node):
- body = self.handle_suite(with_node.children[-1])
- items = [self.handle_with_item(with_node.children[i])
- for i in range(1, len(with_node.children)-2, 2)]
- return ast.With(items, body, with_node.lineno, with_node.column)
+ body = self.handle_suite(with_node.get_child(-1))
+ items = [self.handle_with_item(with_node.get_child(i))
+ for i in range(1, with_node.num_children()-2, 2)]
+ return ast.With(items, body, with_node.get_lineno(),
with_node.get_column())
def handle_classdef(self, classdef_node, decorators=None):
- name_node = classdef_node.children[1]
- name = self.new_identifier(name_node.value)
+ name_node = classdef_node.get_child(1)
+ name = self.new_identifier(name_node.get_value())
self.check_forbidden_name(name, name_node)
- if len(classdef_node.children) == 4:
+ if classdef_node.num_children() == 4:
# class NAME ':' suite
- body = self.handle_suite(classdef_node.children[3])
+ body = self.handle_suite(classdef_node.get_child(3))
return ast.ClassDef(name, None, None, None, None, body, decorators,
- classdef_node.lineno, classdef_node.column)
- if classdef_node.children[3].type == tokens.RPAR:
+ classdef_node.get_lineno(),
classdef_node.get_column())
+ if classdef_node.get_child(3).type == tokens.RPAR:
# class NAME '(' ')' ':' suite
- body = self.handle_suite(classdef_node.children[5])
+ body = self.handle_suite(classdef_node.get_child(5))
return ast.ClassDef(name, None, None, None, None, body, decorators,
- classdef_node.lineno, classdef_node.column)
+ classdef_node.get_lineno(),
classdef_node.get_column())
# class NAME '(' arglist ')' ':' suite
# build up a fake Call node so we can extract its pieces
- call_name = ast.Name(name, ast.Load, classdef_node.lineno,
- classdef_node.column)
- call = self.handle_call(classdef_node.children[3], call_name)
- body = self.handle_suite(classdef_node.children[6])
+ call_name = ast.Name(name, ast.Load, classdef_node.get_lineno(),
+ classdef_node.get_column())
+ call = self.handle_call(classdef_node.get_child(3), call_name)
+ body = self.handle_suite(classdef_node.get_child(6))
return ast.ClassDef(
name, call.args, call.keywords, call.starargs, call.kwargs,
- body, decorators, classdef_node.lineno, classdef_node.column)
+ body, decorators, classdef_node.get_lineno(),
classdef_node.get_column())
def handle_class_bases(self, bases_node):
- if len(bases_node.children) == 1:
- return [self.handle_expr(bases_node.children[0])]
+ if bases_node.num_children() == 1:
+ return [self.handle_expr(bases_node.get_child(0))]
return self.get_expression_list(bases_node)
def handle_funcdef(self, funcdef_node, decorators=None):
- name_node = funcdef_node.children[1]
- name = self.new_identifier(name_node.value)
+ name_node = funcdef_node.get_child(1)
+ name = self.new_identifier(name_node.get_value())
self.check_forbidden_name(name, name_node)
- args = self.handle_arguments(funcdef_node.children[2])
+ args = self.handle_arguments(funcdef_node.get_child(2))
suite = 4
returns = None
- if funcdef_node.children[3].type == tokens.RARROW:
- returns = self.handle_expr(funcdef_node.children[4])
+ if funcdef_node.get_child(3).type == tokens.RARROW:
+ returns = self.handle_expr(funcdef_node.get_child(4))
suite += 2
- body = self.handle_suite(funcdef_node.children[suite])
+ body = self.handle_suite(funcdef_node.get_child(suite))
return ast.FunctionDef(name, args, body, decorators, returns,
- funcdef_node.lineno, funcdef_node.column)
+ funcdef_node.get_lineno(),
funcdef_node.get_column())
def handle_decorated(self, decorated_node):
- decorators = self.handle_decorators(decorated_node.children[0])
- definition = decorated_node.children[1]
+ decorators = self.handle_decorators(decorated_node.get_child(0))
+ definition = decorated_node.get_child(1)
if definition.type == syms.funcdef:
node = self.handle_funcdef(definition, decorators)
elif definition.type == syms.classdef:
node = self.handle_classdef(definition, decorators)
else:
raise AssertionError("unkown decorated")
- node.lineno = decorated_node.lineno
- node.col_offset = decorated_node.column
+ node.lineno = decorated_node.get_lineno()
+ node.col_offset = decorated_node.get_column()
return node
def handle_decorators(self, decorators_node):
- return [self.handle_decorator(dec) for dec in decorators_node.children]
+ return [self.handle_decorator(decorators_node.get_child(i))
+ for i in range(decorators_node.num_children())]
def handle_decorator(self, decorator_node):
- dec_name = self.handle_dotted_name(decorator_node.children[1])
- if len(decorator_node.children) == 3:
+ dec_name = self.handle_dotted_name(decorator_node.get_child(1))
+ if decorator_node.num_children() == 3:
dec = dec_name
- elif len(decorator_node.children) == 5:
+ elif decorator_node.num_children() == 5:
dec = ast.Call(dec_name, None, None, None, None,
- decorator_node.lineno, decorator_node.column)
+ decorator_node.get_lineno(),
decorator_node.get_column())
else:
- dec = self.handle_call(decorator_node.children[3], dec_name)
+ dec = self.handle_call(decorator_node.get_child(3), dec_name)
return dec
def handle_dotted_name(self, dotted_name_node):
- base_value = self.new_identifier(dotted_name_node.children[0].value)
- name = ast.Name(base_value, ast.Load, dotted_name_node.lineno,
- dotted_name_node.column)
- for i in range(2, len(dotted_name_node.children), 2):
- attr = dotted_name_node.children[i].value
+ base_value =
self.new_identifier(dotted_name_node.get_child(0).get_value())
+ name = ast.Name(base_value, ast.Load, dotted_name_node.get_lineno(),
+ dotted_name_node.get_column())
+ for i in range(2, dotted_name_node.num_children(), 2):
+ attr = dotted_name_node.get_child(i).get_value()
attr = self.new_identifier(attr)
- name = ast.Attribute(name, attr, ast.Load, dotted_name_node.lineno,
- dotted_name_node.column)
+ name = ast.Attribute(name, attr, ast.Load,
dotted_name_node.get_lineno(),
+ dotted_name_node.get_column())
return name
def handle_arguments(self, arguments_node):
# This function handles both typedargslist (function definition)
# and varargslist (lambda definition).
if arguments_node.type == syms.parameters:
- if len(arguments_node.children) == 2:
+ if arguments_node.num_children() == 2:
return ast.arguments(None, None, None, None, None, None, None,
None)
- arguments_node = arguments_node.children[1]
+ arguments_node = arguments_node.get_child(1)
i = 0
- child_count = len(arguments_node.children)
+ child_count = arguments_node.num_children()
n_pos = 0
n_pos_def = 0
n_kwdonly = 0
# scan args
while i < child_count:
- arg_type = arguments_node.children[i].type
+ arg_type = arguments_node.get_child(i).type
if arg_type == tokens.STAR:
i += 1
if i < child_count:
- next_arg_type = arguments_node.children[i].type
+ next_arg_type = arguments_node.get_child(i).type
if (next_arg_type == syms.tfpdef or
next_arg_type == syms.vfpdef):
i += 1
@@ -557,7 +558,7 @@
n_pos_def += 1
i += 1
while i < child_count:
- arg_type = arguments_node.children[i].type
+ arg_type = arguments_node.get_child(i).type
if arg_type == tokens.DOUBLESTAR:
break
if arg_type == syms.vfpdef or arg_type == syms.tfpdef:
@@ -577,12 +578,12 @@
i = 0
have_default = False
while i < child_count:
- arg = arguments_node.children[i]
+ arg = arguments_node.get_child(i)
arg_type = arg.type
if arg_type == syms.tfpdef or arg_type == syms.vfpdef:
if i + 1 < child_count and \
- arguments_node.children[i + 1].type == tokens.EQUAL:
- default_node = arguments_node.children[i + 2]
+ arguments_node.get_child(i + 1).type == tokens.EQUAL:
+ default_node = arguments_node.get_child(i + 2)
posdefaults.append(self.handle_expr(default_node))
i += 2
have_default = True
@@ -595,32 +596,32 @@
if i + 1 >= child_count:
self.error("named arguments must follow bare *",
arguments_node)
- name_node = arguments_node.children[i + 1]
+ name_node = arguments_node.get_child(i + 1)
keywordonly_args = []
if name_node.type == tokens.COMMA:
i += 2
i = self.handle_keywordonly_args(arguments_node, i, kwonly,
kwdefaults)
else:
- vararg = name_node.children[0].value
+ vararg = name_node.get_child(0).get_value()
vararg = self.new_identifier(vararg)
self.check_forbidden_name(vararg, name_node)
- if len(name_node.children) > 1:
- varargann = self.handle_expr(name_node.children[2])
+ if name_node.num_children() > 1:
+ varargann = self.handle_expr(name_node.get_child(2))
i += 3
if i < child_count:
- next_arg_type = arguments_node.children[i].type
+ next_arg_type = arguments_node.get_child(i).type
if (next_arg_type == syms.tfpdef or
next_arg_type == syms.vfpdef):
i = self.handle_keywordonly_args(arguments_node, i,
kwonly,
kwdefaults)
elif arg_type == tokens.DOUBLESTAR:
- name_node = arguments_node.children[i + 1]
- kwarg = name_node.children[0].value
+ name_node = arguments_node.get_child(i + 1)
+ kwarg = name_node.get_child(0).get_value()
kwarg = self.new_identifier(kwarg)
self.check_forbidden_name(kwarg, name_node)
- if len(name_node.children) > 1:
- kwargann = self.handle_expr(name_node.children[2])
+ if name_node.num_children() > 1:
+ kwargann = self.handle_expr(name_node.get_child(2))
i += 3
else:
raise AssertionError("unknown node in argument list")
@@ -630,24 +631,24 @@
def handle_keywordonly_args(self, arguments_node, i, kwonly, kwdefaults):
if kwonly is None:
self.error("named arguments must follows bare *",
- arguments_node.children[i])
- child_count = len(arguments_node.children)
+ arguments_node.get_child(i))
+ child_count = arguments_node.num_children()
while i < child_count:
- arg = arguments_node.children[i]
+ arg = arguments_node.get_child(i)
arg_type = arg.type
if arg_type == syms.vfpdef or arg_type == syms.tfpdef:
if (i + 1 < child_count and
- arguments_node.children[i + 1].type == tokens.EQUAL):
- expr = self.handle_expr(arguments_node.children[i + 2])
+ arguments_node.get_child(i + 1).type == tokens.EQUAL):
+ expr = self.handle_expr(arguments_node.get_child(i + 2))
kwdefaults.append(expr)
i += 2
else:
kwdefaults.append(None)
ann = None
- if len(arg.children) == 3:
- ann = self.handle_expr(arg.children[2])
- name_node = arg.children[0]
- argname = name_node.value
+ if arg.num_children() == 3:
+ ann = self.handle_expr(arg.get_child(2))
+ name_node = arg.get_child(0)
+ argname = name_node.get_value()
argname = self.new_identifier(argname)
self.check_forbidden_name(argname, name_node)
kwonly.append(ast.arg(argname, ann))
@@ -657,31 +658,31 @@
return i
def handle_arg(self, arg_node):
- name_node = arg_node.children[0]
- name = self.new_identifier(name_node.value)
+ name_node = arg_node.get_child(0)
+ name = self.new_identifier(name_node.get_value())
self.check_forbidden_name(name, arg_node)
ann = None
- if len(arg_node.children) == 3:
- ann = self.handle_expr(arg_node.children[2])
+ if arg_node.num_children() == 3:
+ ann = self.handle_expr(arg_node.get_child(2))
return ast.arg(name, ann)
def handle_stmt(self, stmt):
stmt_type = stmt.type
if stmt_type == syms.stmt:
- stmt = stmt.children[0]
+ stmt = stmt.get_child(0)
stmt_type = stmt.type
if stmt_type == syms.simple_stmt:
- stmt = stmt.children[0]
+ stmt = stmt.get_child(0)
stmt_type = stmt.type
if stmt_type == syms.small_stmt:
- stmt = stmt.children[0]
+ stmt = stmt.get_child(0)
stmt_type = stmt.type
if stmt_type == syms.expr_stmt:
return self.handle_expr_stmt(stmt)
elif stmt_type == syms.del_stmt:
return self.handle_del_stmt(stmt)
elif stmt_type == syms.pass_stmt:
- return ast.Pass(stmt.lineno, stmt.column)
+ return ast.Pass(stmt.get_lineno(), stmt.get_column())
elif stmt_type == syms.flow_stmt:
return self.handle_flow_stmt(stmt)
elif stmt_type == syms.import_stmt:
@@ -695,7 +696,7 @@
else:
raise AssertionError("unhandled small statement")
elif stmt_type == syms.compound_stmt:
- stmt = stmt.children[0]
+ stmt = stmt.get_child(0)
stmt_type = stmt.type
if stmt_type == syms.if_stmt:
return self.handle_if_stmt(stmt)
@@ -719,95 +720,95 @@
raise AssertionError("unknown statment type")
def handle_expr_stmt(self, stmt):
- if len(stmt.children) == 1:
- expression = self.handle_testlist(stmt.children[0])
- return ast.Expr(expression, stmt.lineno, stmt.column)
- elif stmt.children[1].type == syms.augassign:
+ if stmt.num_children() == 1:
+ expression = self.handle_testlist(stmt.get_child(0))
+ return ast.Expr(expression, stmt.get_lineno(), stmt.get_column())
+ elif stmt.get_child(1).type == syms.augassign:
# Augmented assignment.
- target_child = stmt.children[0]
+ target_child = stmt.get_child(0)
target_expr = self.handle_testlist(target_child)
self.set_context(target_expr, ast.Store)
- value_child = stmt.children[2]
+ value_child = stmt.get_child(2)
if value_child.type == syms.testlist:
value_expr = self.handle_testlist(value_child)
else:
value_expr = self.handle_expr(value_child)
- op_str = stmt.children[1].children[0].value
+ op_str = stmt.get_child(1).get_child(0).get_value()
operator = augassign_operator_map[op_str]
return ast.AugAssign(target_expr, operator, value_expr,
- stmt.lineno, stmt.column)
+ stmt.get_lineno(), stmt.get_column())
else:
# Normal assignment.
targets = []
- for i in range(0, len(stmt.children) - 2, 2):
- target_node = stmt.children[i]
+ for i in range(0, stmt.num_children() - 2, 2):
+ target_node = stmt.get_child(i)
if target_node.type == syms.yield_expr:
self.error("assignment to yield expression not possible",
target_node)
target_expr = self.handle_testlist(target_node)
self.set_context(target_expr, ast.Store)
targets.append(target_expr)
- value_child = stmt.children[-1]
+ value_child = stmt.get_child(-1)
if value_child.type == syms.testlist_star_expr:
value_expr = self.handle_testlist(value_child)
else:
value_expr = self.handle_expr(value_child)
- return ast.Assign(targets, value_expr, stmt.lineno, stmt.column)
+ return ast.Assign(targets, value_expr, stmt.get_lineno(),
stmt.get_column())
def get_expression_list(self, tests):
- return [self.handle_expr(tests.children[i])
- for i in range(0, len(tests.children), 2)]
+ return [self.handle_expr(tests.get_child(i))
+ for i in range(0, tests.num_children(), 2)]
def handle_testlist(self, tests):
- if len(tests.children) == 1:
- return self.handle_expr(tests.children[0])
+ if tests.num_children() == 1:
+ return self.handle_expr(tests.get_child(0))
else:
elts = self.get_expression_list(tests)
- return ast.Tuple(elts, ast.Load, tests.lineno, tests.column)
+ return ast.Tuple(elts, ast.Load, tests.get_lineno(),
tests.get_column())
def handle_expr(self, expr_node):
# Loop until we return something.
while True:
expr_node_type = expr_node.type
if expr_node_type == syms.test or expr_node_type ==
syms.test_nocond:
- first_child = expr_node.children[0]
+ first_child = expr_node.get_child(0)
if first_child.type in (syms.lambdef, syms.lambdef_nocond):
return self.handle_lambdef(first_child)
- elif len(expr_node.children) > 1:
+ elif expr_node.num_children() > 1:
return self.handle_ifexp(expr_node)
else:
expr_node = first_child
elif expr_node_type == syms.or_test or \
expr_node_type == syms.and_test:
- if len(expr_node.children) == 1:
- expr_node = expr_node.children[0]
+ if expr_node.num_children() == 1:
+ expr_node = expr_node.get_child(0)
continue
- seq = [self.handle_expr(expr_node.children[i])
- for i in range(0, len(expr_node.children), 2)]
+ seq = [self.handle_expr(expr_node.get_child(i))
+ for i in range(0, expr_node.num_children(), 2)]
if expr_node_type == syms.or_test:
op = ast.Or
else:
op = ast.And
- return ast.BoolOp(op, seq, expr_node.lineno, expr_node.column)
+ return ast.BoolOp(op, seq, expr_node.get_lineno(),
expr_node.get_column())
elif expr_node_type == syms.not_test:
- if len(expr_node.children) == 1:
- expr_node = expr_node.children[0]
+ if expr_node.num_children() == 1:
+ expr_node = expr_node.get_child(0)
continue
- expr = self.handle_expr(expr_node.children[1])
- return ast.UnaryOp(ast.Not, expr, expr_node.lineno,
- expr_node.column)
+ expr = self.handle_expr(expr_node.get_child(1))
+ return ast.UnaryOp(ast.Not, expr, expr_node.get_lineno(),
+ expr_node.get_column())
elif expr_node_type == syms.comparison:
- if len(expr_node.children) == 1:
- expr_node = expr_node.children[0]
+ if expr_node.num_children() == 1:
+ expr_node = expr_node.get_child(0)
continue
operators = []
operands = []
- expr = self.handle_expr(expr_node.children[0])
- for i in range(1, len(expr_node.children), 2):
-
operators.append(self.handle_comp_op(expr_node.children[i]))
- operands.append(self.handle_expr(expr_node.children[i +
1]))
- return ast.Compare(expr, operators, operands, expr_node.lineno,
- expr_node.column)
+ expr = self.handle_expr(expr_node.get_child(0))
+ for i in range(1, expr_node.num_children(), 2):
+
operators.append(self.handle_comp_op(expr_node.get_child(i)))
+ operands.append(self.handle_expr(expr_node.get_child(i +
1)))
+ return ast.Compare(expr, operators, operands,
expr_node.get_lineno(),
+ expr_node.get_column())
elif expr_node_type == syms.star_expr:
return self.handle_star_expr(expr_node)
elif expr_node_type == syms.expr or \
@@ -816,27 +817,27 @@
expr_node_type == syms.shift_expr or \
expr_node_type == syms.arith_expr or \
expr_node_type == syms.term:
- if len(expr_node.children) == 1:
- expr_node = expr_node.children[0]
+ if expr_node.num_children() == 1:
+ expr_node = expr_node.get_child(0)
continue
return self.handle_binop(expr_node)
elif expr_node_type == syms.yield_expr:
is_from = False
- if len(expr_node.children) > 1:
- arg_node = expr_node.children[1] # yield arg
- if len(arg_node.children) == 2:
+ if expr_node.num_children() > 1:
+ arg_node = expr_node.get_child(1) # yield arg
+ if arg_node.num_children() == 2:
is_from = True
- expr = self.handle_expr(arg_node.children[1])
+ expr = self.handle_expr(arg_node.get_child(1))
else:
- expr = self.handle_testlist(arg_node.children[0])
+ expr = self.handle_testlist(arg_node.get_child(0))
else:
expr = None
if is_from:
- return ast.YieldFrom(expr, expr_node.lineno,
expr_node.column)
- return ast.Yield(expr, expr_node.lineno, expr_node.column)
+ return ast.YieldFrom(expr, expr_node.get_lineno(),
expr_node.get_column())
+ return ast.Yield(expr, expr_node.get_lineno(),
expr_node.get_column())
elif expr_node_type == syms.factor:
- if len(expr_node.children) == 1:
- expr_node = expr_node.children[0]
+ if expr_node.num_children() == 1:
+ expr_node = expr_node.get_child(0)
continue
return self.handle_factor(expr_node)
elif expr_node_type == syms.power:
@@ -845,29 +846,29 @@
raise AssertionError("unknown expr")
def handle_star_expr(self, star_expr_node):
- expr = self.handle_expr(star_expr_node.children[1])
- return ast.Starred(expr, ast.Load, star_expr_node.lineno,
- star_expr_node.column)
+ expr = self.handle_expr(star_expr_node.get_child(1))
+ return ast.Starred(expr, ast.Load, star_expr_node.get_lineno(),
+ star_expr_node.get_column())
def handle_lambdef(self, lambdef_node):
- expr = self.handle_expr(lambdef_node.children[-1])
- if len(lambdef_node.children) == 3:
+ expr = self.handle_expr(lambdef_node.get_child(-1))
+ if lambdef_node.num_children() == 3:
args = ast.arguments(None, None, None, None, None, None, None,
None)
else:
- args = self.handle_arguments(lambdef_node.children[1])
- return ast.Lambda(args, expr, lambdef_node.lineno, lambdef_node.column)
+ args = self.handle_arguments(lambdef_node.get_child(1))
+ return ast.Lambda(args, expr, lambdef_node.get_lineno(),
lambdef_node.get_column())
def handle_ifexp(self, if_expr_node):
- body = self.handle_expr(if_expr_node.children[0])
- expression = self.handle_expr(if_expr_node.children[2])
- otherwise = self.handle_expr(if_expr_node.children[4])
- return ast.IfExp(expression, body, otherwise, if_expr_node.lineno,
- if_expr_node.column)
+ body = self.handle_expr(if_expr_node.get_child(0))
+ expression = self.handle_expr(if_expr_node.get_child(2))
+ otherwise = self.handle_expr(if_expr_node.get_child(4))
+ return ast.IfExp(expression, body, otherwise,
if_expr_node.get_lineno(),
+ if_expr_node.get_column())
def handle_comp_op(self, comp_op_node):
- comp_node = comp_op_node.children[0]
+ comp_node = comp_op_node.get_child(0)
comp_type = comp_node.type
- if len(comp_op_node.children) == 1:
+ if comp_op_node.num_children() == 1:
if comp_type == tokens.LESS:
return ast.Lt
elif comp_type == tokens.GREATER:
@@ -880,46 +881,47 @@
return ast.GtE
elif comp_type == tokens.NOTEQUAL:
flufl = self.compile_info.flags &
consts.CO_FUTURE_BARRY_AS_BDFL
- if flufl and comp_node.value == '!=':
+ if flufl and comp_node.get_value() == '!=':
self.error('invalid comparison', comp_node)
- elif not flufl and comp_node.value == '<>':
+ elif not flufl and comp_node.get_value() == '<>':
self.error('invalid comparison', comp_node)
return ast.NotEq
elif comp_type == tokens.NAME:
- if comp_node.value == "is":
+ if comp_node.get_value() == "is":
return ast.Is
- elif comp_node.value == "in":
+ elif comp_node.get_value() == "in":
return ast.In
else:
raise AssertionError("invalid comparison")
else:
raise AssertionError("invalid comparison")
else:
- if comp_op_node.children[1].value == "in":
+ if comp_op_node.get_child(1).get_value() == "in":
return ast.NotIn
- elif comp_node.value == "is":
+ elif comp_node.get_value() == "is":
return ast.IsNot
else:
raise AssertionError("invalid comparison")
def handle_binop(self, binop_node):
- left = self.handle_expr(binop_node.children[0])
- right = self.handle_expr(binop_node.children[2])
- op = operator_map(binop_node.children[1].type)
- result = ast.BinOp(left, op, right, binop_node.lineno,
- binop_node.column)
- number_of_ops = (len(binop_node.children) - 1) / 2
+ left = self.handle_expr(binop_node.get_child(0))
+ right = self.handle_expr(binop_node.get_child(2))
+ op = operator_map(binop_node.get_child(1).type)
+ result = ast.BinOp(left, op, right, binop_node.get_lineno(),
+ binop_node.get_column())
+ number_of_ops = (binop_node.num_children() - 1) / 2
for i in range(1, number_of_ops):
- op_node = binop_node.children[i * 2 + 1]
+ op_node = binop_node.get_child(i * 2 + 1)
op = operator_map(op_node.type)
- sub_right = self.handle_expr(binop_node.children[i * 2 + 2])
- result = ast.BinOp(result, op, sub_right, op_node.lineno,
- op_node.column)
+ sub_right = self.handle_expr(binop_node.get_child(i * 2 + 2))
+ result = ast.BinOp(result, op, sub_right, op_node.get_lineno(),
+ op_node.get_column())
return result
def handle_factor(self, factor_node):
- expr = self.handle_expr(factor_node.children[1])
- op_type = factor_node.children[0].type
+ from pypy.interpreter.pyparser.parser import Terminal
+ expr = self.handle_expr(factor_node.get_child(1))
+ op_type = factor_node.get_child(0).type
if op_type == tokens.PLUS:
op = ast.UAdd
elif op_type == tokens.MINUS:
@@ -928,29 +930,29 @@
op = ast.Invert
else:
raise AssertionError("invalid factor node")
- return ast.UnaryOp(op, expr, factor_node.lineno, factor_node.column)
+ return ast.UnaryOp(op, expr, factor_node.get_lineno(),
factor_node.get_column())
def handle_power(self, power_node):
- atom_expr = self.handle_atom(power_node.children[0])
- if len(power_node.children) == 1:
+ atom_expr = self.handle_atom(power_node.get_child(0))
+ if power_node.num_children() == 1:
return atom_expr
- for i in range(1, len(power_node.children)):
- trailer = power_node.children[i]
+ for i in range(1, power_node.num_children()):
+ trailer = power_node.get_child(i)
if trailer.type != syms.trailer:
break
tmp_atom_expr = self.handle_trailer(trailer, atom_expr)
tmp_atom_expr.lineno = atom_expr.lineno
tmp_atom_expr.col_offset = atom_expr.col_offset
atom_expr = tmp_atom_expr
- if power_node.children[-1].type == syms.factor:
- right = self.handle_expr(power_node.children[-1])
- atom_expr = ast.BinOp(atom_expr, ast.Pow, right, power_node.lineno,
- power_node.column)
+ if power_node.get_child(-1).type == syms.factor:
+ right = self.handle_expr(power_node.get_child(-1))
+ atom_expr = ast.BinOp(atom_expr, ast.Pow, right,
power_node.get_lineno(),
+ power_node.get_column())
return atom_expr
def handle_slice(self, slice_node):
- first_child = slice_node.children[0]
- if len(slice_node.children) == 1 and first_child.type == syms.test:
+ first_child = slice_node.get_child(0)
+ if slice_node.num_children() == 1 and first_child.type == syms.test:
index = self.handle_expr(first_child)
return ast.Index(index)
lower = None
@@ -959,68 +961,69 @@
if first_child.type == syms.test:
lower = self.handle_expr(first_child)
if first_child.type == tokens.COLON:
- if len(slice_node.children) > 1:
- second_child = slice_node.children[1]
+ if slice_node.num_children() > 1:
+ second_child = slice_node.get_child(1)
if second_child.type == syms.test:
upper = self.handle_expr(second_child)
- elif len(slice_node.children) > 2:
- third_child = slice_node.children[2]
+ elif slice_node.num_children() > 2:
+ third_child = slice_node.get_child(2)
if third_child.type == syms.test:
upper = self.handle_expr(third_child)
- last_child = slice_node.children[-1]
+ last_child = slice_node.get_child(-1)
if last_child.type == syms.sliceop:
- if len(last_child.children) != 1:
- step_child = last_child.children[1]
+ if last_child.num_children() != 1:
+ step_child = last_child.get_child(1)
if step_child.type == syms.test:
step = self.handle_expr(step_child)
return ast.Slice(lower, upper, step)
def handle_trailer(self, trailer_node, left_expr):
- first_child = trailer_node.children[0]
+ first_child = trailer_node.get_child(0)
if first_child.type == tokens.LPAR:
- if len(trailer_node.children) == 2:
+ if trailer_node.num_children() == 2:
return ast.Call(left_expr, None, None, None, None,
- trailer_node.lineno, trailer_node.column)
+ trailer_node.get_lineno(),
trailer_node.get_column())
else:
- return self.handle_call(trailer_node.children[1], left_expr)
+ return self.handle_call(trailer_node.get_child(1), left_expr)
elif first_child.type == tokens.DOT:
- attr = self.new_identifier(trailer_node.children[1].value)
+ attr = self.new_identifier(trailer_node.get_child(1).get_value())
return ast.Attribute(left_expr, attr, ast.Load,
- trailer_node.lineno, trailer_node.column)
+ trailer_node.get_lineno(),
trailer_node.get_column())
else:
- middle = trailer_node.children[1]
- if len(middle.children) == 1:
- slice = self.handle_slice(middle.children[0])
+ middle = trailer_node.get_child(1)
+ if middle.num_children() == 1:
+ slice = self.handle_slice(middle.get_child(0))
return ast.Subscript(left_expr, slice, ast.Load,
- middle.lineno, middle.column)
+ middle.get_lineno(), middle.get_column())
slices = []
simple = True
- for i in range(0, len(middle.children), 2):
- slc = self.handle_slice(middle.children[i])
+ for i in range(0, middle.num_children(), 2):
+ slc = self.handle_slice(middle.get_child(i))
if not isinstance(slc, ast.Index):
simple = False
slices.append(slc)
if not simple:
ext_slice = ast.ExtSlice(slices)
return ast.Subscript(left_expr, ext_slice, ast.Load,
- middle.lineno, middle.column)
+ middle.get_lineno(), middle.get_column())
elts = []
for idx in slices:
assert isinstance(idx, ast.Index)
elts.append(idx.value)
- tup = ast.Tuple(elts, ast.Load, middle.lineno, middle.column)
+ tup = ast.Tuple(elts, ast.Load, middle.get_lineno(),
middle.get_column())
return ast.Subscript(left_expr, ast.Index(tup), ast.Load,
- middle.lineno, middle.column)
+ middle.get_lineno(), middle.get_column())
def handle_call(self, args_node, callable_expr):
arg_count = 0
keyword_count = 0
generator_count = 0
- for argument in args_node.children:
+ for i in range(args_node.num_children()):
+ argument = args_node.get_child(i)
if argument.type == syms.argument:
- if len(argument.children) == 1:
+ if argument.num_children() == 1:
arg_count += 1
- elif argument.children[1].type == syms.comp_for:
+ elif argument.get_child(1).type == syms.comp_for:
generator_count += 1
else:
keyword_count += 1
@@ -1035,13 +1038,13 @@
used_keywords = {}
variable_arg = None
keywords_arg = None
- child_count = len(args_node.children)
+ child_count = args_node.num_children()
i = 0
while i < child_count:
- argument = args_node.children[i]
+ argument = args_node.get_child(i)
if argument.type == syms.argument:
- if len(argument.children) == 1:
- expr_node = argument.children[0]
+ if argument.num_children() == 1:
+ expr_node = argument.get_child(0)
if keywords:
self.error("non-keyword arg after keyword arg",
expr_node)
@@ -1049,10 +1052,10 @@
self.error("only named arguments may follow "
"*expression", expr_node)
args.append(self.handle_expr(expr_node))
- elif argument.children[1].type == syms.comp_for:
+ elif argument.get_child(1).type == syms.comp_for:
args.append(self.handle_genexp(argument))
else:
- keyword_node = argument.children[0]
+ keyword_node = argument.get_child(0)
keyword_expr = self.handle_expr(keyword_node)
if isinstance(keyword_expr, ast.Lambda):
self.error("lambda cannot contain assignment",
@@ -1065,13 +1068,13 @@
self.error("keyword argument repeated", keyword_node)
used_keywords[keyword] = None
self.check_forbidden_name(keyword, keyword_node)
- keyword_value = self.handle_expr(argument.children[2])
+ keyword_value = self.handle_expr(argument.get_child(2))
keywords.append(ast.keyword(keyword, keyword_value))
elif argument.type == tokens.STAR:
- variable_arg = self.handle_expr(args_node.children[i + 1])
+ variable_arg = self.handle_expr(args_node.get_child(i + 1))
i += 1
elif argument.type == tokens.DOUBLESTAR:
- keywords_arg = self.handle_expr(args_node.children[i + 1])
+ keywords_arg = self.handle_expr(args_node.get_child(i + 1))
i += 1
i += 1
if not args:
@@ -1127,18 +1130,20 @@
return self.space.call_function(self.space.w_float, w_num_str)
def handle_atom(self, atom_node):
- first_child = atom_node.children[0]
+ first_child = atom_node.get_child(0)
first_child_type = first_child.type
if first_child_type == tokens.NAME:
- name = self.new_identifier(first_child.value)
- return ast.Name(name, ast.Load, first_child.lineno,
- first_child.column)
+ name = self.new_identifier(first_child.get_value())
+ return ast.Name(name, ast.Load, first_child.get_lineno(),
+ first_child.get_column())
elif first_child_type == tokens.STRING:
space = self.space
encoding = self.compile_info.encoding
try:
- sub_strings_w = [parsestring.parsestr(space, encoding, s.value)
- for s in atom_node.children]
+ sub_strings_w = [
+ parsestring.parsestr(
+ space, encoding,
atom_node.get_child(i).get_value())
+ for i in range(atom_node.num_children())]
except error.OperationError, e:
if not (e.match(space, space.w_UnicodeError) or
e.match(space, space.w_ValueError)):
@@ -1159,58 +1164,58 @@
# UnicodeError in literal: turn into SyntaxError
strdata = space.isinstance_w(w_string, space.w_unicode)
node = ast.Str if strdata else ast.Bytes
- return node(w_string, atom_node.lineno, atom_node.column)
+ return node(w_string, atom_node.get_lineno(),
atom_node.get_column())
elif first_child_type == tokens.NUMBER:
- num_value = self.parse_number(first_child.value)
- return ast.Num(num_value, atom_node.lineno, atom_node.column)
+ num_value = self.parse_number(first_child.get_value())
+ return ast.Num(num_value, atom_node.get_lineno(),
atom_node.get_column())
elif first_child_type == tokens.ELLIPSIS:
- return ast.Ellipsis(atom_node.lineno, atom_node.column)
+ return ast.Ellipsis(atom_node.get_lineno(), atom_node.get_column())
elif first_child_type == tokens.LPAR:
- second_child = atom_node.children[1]
+ second_child = atom_node.get_child(1)
if second_child.type == tokens.RPAR:
- return ast.Tuple(None, ast.Load, atom_node.lineno,
- atom_node.column)
+ return ast.Tuple(None, ast.Load, atom_node.get_lineno(),
+ atom_node.get_column())
elif second_child.type == syms.yield_expr:
return self.handle_expr(second_child)
return self.handle_testlist_gexp(second_child)
elif first_child_type == tokens.LSQB:
- second_child = atom_node.children[1]
+ second_child = atom_node.get_child(1)
if second_child.type == tokens.RSQB:
- return ast.List(None, ast.Load, atom_node.lineno,
- atom_node.column)
- if len(second_child.children) == 1 or \
- second_child.children[1].type == tokens.COMMA:
+ return ast.List(None, ast.Load, atom_node.get_lineno(),
+ atom_node.get_column())
+ if second_child.num_children() == 1 or \
+ second_child.get_child(1).type == tokens.COMMA:
elts = self.get_expression_list(second_child)
- return ast.List(elts, ast.Load, atom_node.lineno,
- atom_node.column)
+ return ast.List(elts, ast.Load, atom_node.get_lineno(),
+ atom_node.get_column())
return self.handle_listcomp(second_child)
elif first_child_type == tokens.LBRACE:
- maker = atom_node.children[1]
+ maker = atom_node.get_child(1)
if maker.type == tokens.RBRACE:
- return ast.Dict(None, None, atom_node.lineno, atom_node.column)
- n_maker_children = len(maker.children)
- if n_maker_children == 1 or maker.children[1].type == tokens.COMMA:
+ return ast.Dict(None, None, atom_node.get_lineno(),
atom_node.get_column())
+ n_maker_children = maker.num_children()
+ if n_maker_children == 1 or maker.get_child(1).type ==
tokens.COMMA:
elts = []
for i in range(0, n_maker_children, 2):
- elts.append(self.handle_expr(maker.children[i]))
- return ast.Set(elts, atom_node.lineno, atom_node.column)
- if maker.children[1].type == syms.comp_for:
+ elts.append(self.handle_expr(maker.get_child(i)))
+ return ast.Set(elts, atom_node.get_lineno(),
atom_node.get_column())
+ if maker.get_child(1).type == syms.comp_for:
return self.handle_setcomp(maker)
if (n_maker_children > 3 and
- maker.children[3].type == syms.comp_for):
+ maker.get_child(3).type == syms.comp_for):
return self.handle_dictcomp(maker)
keys = []
values = []
for i in range(0, n_maker_children, 4):
- keys.append(self.handle_expr(maker.children[i]))
- values.append(self.handle_expr(maker.children[i + 2]))
- return ast.Dict(keys, values, atom_node.lineno, atom_node.column)
+ keys.append(self.handle_expr(maker.get_child(i)))
+ values.append(self.handle_expr(maker.get_child(i + 2)))
+ return ast.Dict(keys, values, atom_node.get_lineno(),
atom_node.get_column())
else:
raise AssertionError("unknown atom")
def handle_testlist_gexp(self, gexp_node):
- if len(gexp_node.children) > 1 and \
- gexp_node.children[1].type == syms.comp_for:
+ if gexp_node.num_children() > 1 and \
+ gexp_node.get_child(1).type == syms.comp_for:
return self.handle_genexp(gexp_node)
return self.handle_testlist(gexp_node)
@@ -1219,18 +1224,18 @@
current_for = comp_node
while True:
count += 1
- if len(current_for.children) == 5:
- current_iter = current_for.children[4]
+ if current_for.num_children() == 5:
+ current_iter = current_for.get_child(4)
else:
return count
while True:
- first_child = current_iter.children[0]
+ first_child = current_iter.get_child(0)
if first_child.type == syms.comp_for:
- current_for = current_iter.children[0]
+ current_for = current_iter.get_child(0)
break
elif first_child.type == syms.comp_if:
- if len(first_child.children) == 3:
- current_iter = first_child.children[2]
+ if first_child.num_children() == 3:
+ current_iter = first_child.get_child(2)
else:
return count
else:
@@ -1239,23 +1244,23 @@
def count_comp_ifs(self, iter_node):
count = 0
while True:
- first_child = iter_node.children[0]
+ first_child = iter_node.get_child(0)
if first_child.type == syms.comp_for:
return count
count += 1
- if len(first_child.children) == 2:
+ if first_child.num_children() == 2:
return count
- iter_node = first_child.children[2]
+ iter_node = first_child.get_child(2)
def comprehension_helper(self, comp_node):
fors_count = self.count_comp_fors(comp_node)
comps = []
for i in range(fors_count):
- for_node = comp_node.children[1]
+ for_node = comp_node.get_child(1)
for_targets = self.handle_exprlist(for_node, ast.Store)
- expr = self.handle_expr(comp_node.children[3])
+ expr = self.handle_expr(comp_node.get_child(3))
assert isinstance(expr, ast.expr)
- if len(for_node.children) == 1:
+ if for_node.num_children() == 1:
comp = ast.comprehension(for_targets[0], expr, None)
else:
# Modified in python2.7, see http://bugs.python.org/issue6704
@@ -1266,53 +1271,53 @@
line = expr_node.lineno
target = ast.Tuple(for_targets, ast.Store, line, col)
comp = ast.comprehension(target, expr, None)
- if len(comp_node.children) == 5:
- comp_node = comp_iter = comp_node.children[4]
+ if comp_node.num_children() == 5:
+ comp_node = comp_iter = comp_node.get_child(4)
assert comp_iter.type == syms.comp_iter
ifs_count = self.count_comp_ifs(comp_iter)
if ifs_count:
ifs = []
for j in range(ifs_count):
- comp_node = comp_if = comp_iter.children[0]
- ifs.append(self.handle_expr(comp_if.children[1]))
- if len(comp_if.children) == 3:
- comp_node = comp_iter = comp_if.children[2]
+ comp_node = comp_if = comp_iter.get_child(0)
+ ifs.append(self.handle_expr(comp_if.get_child(1)))
+ if comp_if.num_children() == 3:
+ comp_node = comp_iter = comp_if.get_child(2)
comp.ifs = ifs
if comp_node.type == syms.comp_iter:
- comp_node = comp_node.children[0]
+ comp_node = comp_node.get_child(0)
assert isinstance(comp, ast.comprehension)
comps.append(comp)
return comps
def handle_genexp(self, genexp_node):
- elt = self.handle_expr(genexp_node.children[0])
- comps = self.comprehension_helper(genexp_node.children[1])
- return ast.GeneratorExp(elt, comps, genexp_node.lineno,
- genexp_node.column)
+ elt = self.handle_expr(genexp_node.get_child(0))
+ comps = self.comprehension_helper(genexp_node.get_child(1))
+ return ast.GeneratorExp(elt, comps, genexp_node.get_lineno(),
+ genexp_node.get_column())
def handle_listcomp(self, listcomp_node):
- elt = self.handle_expr(listcomp_node.children[0])
- comps = self.comprehension_helper(listcomp_node.children[1])
- return ast.ListComp(elt, comps, listcomp_node.lineno,
- listcomp_node.column)
+ elt = self.handle_expr(listcomp_node.get_child(0))
+ comps = self.comprehension_helper(listcomp_node.get_child(1))
+ return ast.ListComp(elt, comps, listcomp_node.get_lineno(),
+ listcomp_node.get_column())
def handle_setcomp(self, set_maker):
- elt = self.handle_expr(set_maker.children[0])
- comps = self.comprehension_helper(set_maker.children[1])
- return ast.SetComp(elt, comps, set_maker.lineno,
- set_maker.column)
+ elt = self.handle_expr(set_maker.get_child(0))
+ comps = self.comprehension_helper(set_maker.get_child(1))
+ return ast.SetComp(elt, comps, set_maker.get_lineno(),
+ set_maker.get_column())
def handle_dictcomp(self, dict_maker):
- key = self.handle_expr(dict_maker.children[0])
- value = self.handle_expr(dict_maker.children[2])
- comps = self.comprehension_helper(dict_maker.children[3])
- return ast.DictComp(key, value, comps, dict_maker.lineno,
- dict_maker.column)
+ key = self.handle_expr(dict_maker.get_child(0))
+ value = self.handle_expr(dict_maker.get_child(2))
+ comps = self.comprehension_helper(dict_maker.get_child(3))
+ return ast.DictComp(key, value, comps, dict_maker.get_lineno(),
+ dict_maker.get_column())
def handle_exprlist(self, exprlist, context):
exprs = []
- for i in range(0, len(exprlist.children), 2):
- child = exprlist.children[i]
+ for i in range(0, exprlist.num_children(), 2):
+ child = exprlist.get_child(i)
expr = self.handle_expr(child)
self.set_context(expr, context)
exprs.append(expr)
diff --git a/pypy/interpreter/pyparser/parser.py
b/pypy/interpreter/pyparser/parser.py
--- a/pypy/interpreter/pyparser/parser.py
+++ b/pypy/interpreter/pyparser/parser.py
@@ -44,26 +44,125 @@
class Node(object):
- __slots__ = "type value children lineno column".split()
+ __slots__ = ("type", )
- def __init__(self, type, value, children, lineno, column):
+ def __init__(self, type):
self.type = type
+
+ def __eq__(self, other):
+ raise NotImplementedError("abstract base class")
+
+ def __ne__(self, other):
+ return not self == other
+
+ def get_value(self):
+ return None
+
+ def get_child(self, i):
+ raise NotImplementedError("abstract base class")
+
+ def num_children(self):
+ return 0
+
+ def append_child(self, child):
+ raise NotImplementedError("abstract base class")
+
+ def get_lineno(self):
+ raise NotImplementedError("abstract base class")
+
+ def get_column(self):
+ raise NotImplementedError("abstract base class")
+
+
+class Terminal(Node):
+ __slots__ = ("value", "lineno", "column")
+ def __init__(self, type, value, lineno, column):
+ Node.__init__(self, type)
self.value = value
- self.children = children
self.lineno = lineno
self.column = column
+ def __repr__(self):
+ return "Terminal(type=%s, value=%r)" % (self.type, self.value)
+
def __eq__(self, other):
# For tests.
- return (self.type == other.type and
- self.value == other.value and
- self.children == other.children)
+ return (type(self) == type(other) and
+ self.type == other.type and
+ self.value == other.value)
+
+ def get_value(self):
+ return self.value
+
+ def get_lineno(self):
+ return self.lineno
+
+ def get_column(self):
+ return self.column
+
+
+class AbstractNonterminal(Node):
+ __slots__ = ()
+
+ def get_lineno(self):
+ return self.get_child(0).get_lineno()
+
+ def get_column(self):
+ return self.get_child(0).get_column()
+
+ def __eq__(self, other):
+ # For tests.
+ # grumble, annoying
+ if not isinstance(other, AbstractNonterminal):
+ return False
+ if self.type != other.type:
+ return False
+ if self.num_children() != other.num_children():
+ return False
+ for i in range(self.num_children()):
+ if self.get_child(i) != other.get_child(i):
+ return False
+ return True
+
+
+class Nonterminal(AbstractNonterminal):
+ __slots__ = ("_children", )
+ def __init__(self, type, children):
+ Node.__init__(self, type)
+ self._children = children
def __repr__(self):
- if self.value is None:
- return "Node(type=%s, children=%r)" % (self.type, self.children)
- else:
- return "Node(type=%s, value=%r)" % (self.type, self.value)
+ return "Nonterminal(type=%s, children=%r)" % (self.type,
self._children)
+
+ def get_child(self, i):
+ return self._children[i]
+
+ def num_children(self):
+ return len(self._children)
+
+ def append_child(self, child):
+ self._children.append(child)
+
+
+class Nonterminal1(AbstractNonterminal):
+ __slots__ = ("_child", )
+ def __init__(self, type, child):
+ Node.__init__(self, type)
+ self._child = child
+
+ def __repr__(self):
+ return "Nonterminal(type=%s, children=[%r])" % (self.type, self._child)
+
+ def get_child(self, i):
+ assert i == 0 or i == -1
+ return self._child
+
+ def num_children(self):
+ return 1
+
+ def append_child(self, child):
+ assert 0, "should be unreachable"
+
class ParseError(Exception):
@@ -97,7 +196,7 @@
if start == -1:
start = self.grammar.start
self.root = None
- current_node = Node(start, None, [], 0, 0)
+ current_node = Nonterminal(start, [])
self.stack = []
self.stack.append((self.grammar.dfas[start - 256], 0, current_node))
@@ -164,14 +263,14 @@
def shift(self, next_state, token_type, value, lineno, column):
"""Shift a non-terminal and prepare for the next state."""
dfa, state, node = self.stack[-1]
- new_node = Node(token_type, value, None, lineno, column)
- node.children.append(new_node)
+ new_node = Terminal(token_type, value, lineno, column)
+ node.append_child(new_node)
self.stack[-1] = (dfa, next_state, node)
def push(self, next_dfa, next_state, node_type, lineno, column):
"""Push a terminal and adjust the current state."""
dfa, state, node = self.stack[-1]
- new_node = Node(node_type, None, [], lineno, column)
+ new_node = Nonterminal(node_type, [])
self.stack[-1] = (dfa, next_state, node)
self.stack.append((next_dfa, 0, new_node))
@@ -179,6 +278,10 @@
"""Pop an entry off the stack and make its node a child of the last."""
dfa, state, node = self.stack.pop()
if self.stack:
- self.stack[-1][2].children.append(node)
+ # we are now done with node, so we can store it more efficiently if
+ # it has just one child
+ if node.num_children() == 1:
+ node = Nonterminal1(node.type, node.get_child(0))
+ self.stack[-1][2].append_child(node)
else:
self.root = node
diff --git a/pypy/interpreter/pyparser/pygram.py
b/pypy/interpreter/pyparser/pygram.py
--- a/pypy/interpreter/pyparser/pygram.py
+++ b/pypy/interpreter/pyparser/pygram.py
@@ -28,8 +28,11 @@
class _Symbols(object):
pass
+rev_lookup = {}
for sym_name, idx in python_grammar.symbol_ids.iteritems():
setattr(_Symbols, sym_name, idx)
+ rev_lookup[idx] = sym_name
syms = _Symbols()
+syms._rev_lookup = rev_lookup # for debugging
del _get_python_grammar, _Tokens, tok_name, sym_name, idx
diff --git a/pypy/interpreter/pyparser/test/test_parser.py
b/pypy/interpreter/pyparser/test/test_parser.py
--- a/pypy/interpreter/pyparser/test/test_parser.py
+++ b/pypy/interpreter/pyparser/test/test_parser.py
@@ -52,24 +52,23 @@
value = "\n"
else:
value = ""
- children = None
+ n = parser.Terminal(tp, value, 0, 0)
else:
tp = gram.symbol_ids[data[0]]
- value = None
children = []
- n = parser.Node(tp, value, children, 0, 0)
+ n = parser.Nonterminal(tp, children)
new_indent = count_indent(line)
if new_indent >= last_indent:
if new_indent == last_indent and node_stack:
node_stack.pop()
if node_stack:
- node_stack[-1].children.append(n)
+ node_stack[-1].append_child(n)
node_stack.append(n)
else:
diff = last_indent - new_indent
pop_nodes = diff // 4 + 1
del node_stack[-pop_nodes:]
- node_stack[-1].children.append(n)
+ node_stack[-1].append_child(n)
node_stack.append(n)
last_indent = new_indent
return node_stack[0]
diff --git a/pypy/module/parser/pyparser.py b/pypy/module/parser/pyparser.py
--- a/pypy/module/parser/pyparser.py
+++ b/pypy/module/parser/pyparser.py
@@ -15,21 +15,21 @@
@specialize.arg(3)
def _build_app_tree(self, space, node, seq_maker, with_lineno,
with_column):
- if node.children is not None:
- seq_w = [None]*(len(node.children) + 1)
+ if node.num_children():
+ seq_w = [None]*(node.num_children() + 1)
seq_w[0] = space.wrap(node.type)
- for i in range(1, len(node.children) + 1):
- seq_w[i] = self._build_app_tree(space, node.children[i - 1],
+ for i in range(1, node.num_children() + 1):
+ seq_w[i] = self._build_app_tree(space, node.get_child(i - 1),
seq_maker, with_lineno,
with_column)
else:
seq_w = [None]*(2 + with_lineno + with_column)
seq_w[0] = space.wrap(node.type)
- seq_w[1] = space.wrap(node.value)
+ seq_w[1] = space.wrap(node.get_value())
if with_lineno:
- seq_w[2] = space.wrap(node.lineno)
+ seq_w[2] = space.wrap(node.get_lineno())
if with_column:
- seq_w[3] = space.wrap(node.column)
+ seq_w[3] = space.wrap(node.get_column())
return seq_maker(seq_w)
def descr_issuite(self, space):
_______________________________________________
pypy-commit mailing list
[email protected]
https://mail.python.org/mailman/listinfo/pypy-commit
