tools/make_stmt_transformer: pass args/kwargs

This commit is contained in:
Sebastien Bourdeauducq 2014-06-21 12:13:28 +02:00
parent c3092ae13f
commit c71eb702bb

View File

@ -25,8 +25,8 @@ def value_to_ast(value):
return None
def make_stmt_transformer(transformer_class):
def stmt_transformer(stmts):
transformer = transformer_class()
def stmt_transformer(stmts, *args, **kwargs):
transformer = transformer_class(*args, **kwargs)
new_stmts = [transformer.visit(stmt) for stmt in stmts]
stmts[:] = new_stmts
return stmt_transformer