class TreeTransformController(object):
def __init__(self):
self.ast_tree_manager = AstTreeManager()
self.ast_transformer_manager = AstTransformerManager()
def clear(self):
self.ast_tree_manager.clear()
self.ast_transformer_manager.clear()
def apply_transform(self, tree=None, transform=None, name=None):
"""
creates a new ast_tree item by applying
transform to tree, updates the controllers
internals to track that
"""
if isinstance(tree, int):
tree = self.ast_tree_manager[tree]
if isinstance(transform, int):
transform = self.ast_transformer_manager[transform]
new_ast_tree = self.ast_tree_manager.create_transformed_child(
tree, transform, name=name
)
return new_ast_tree
def load_transforms(self, key):
self.ast_transformer_manager.load_transforms(key)
class TreeTransformController(object):
def __init__(self):
self.ast_tree_manager = AstTreeManager()
self.ast_transformer_manager = AstTransformerManager()
def clear(self):
self.ast_tree_manager.clear()
self.ast_transformer_manager.clear()
def apply_transform(self, tree=None, transform=None, name=None):
"""
creates a new ast_tree item by applying
transform to tree, updates the controllers
internals to track that
"""
if isinstance(tree, int):
tree = self.ast_tree_manager[tree]
if isinstance(transform, int):
transform = self.ast_transformer_manager[transform]
new_ast_tree = self.ast_tree_manager.create_transformed_child(
tree, transform, name=name)
return new_ast_tree
def load_transforms(self, key):
self.ast_transformer_manager.load_transforms(key)
def test_basic_load(self):
trees = AstTreeManager()
assert_equal(trees.count(), 0)
test_ast = trees.new_item_from_source(inspect.getsource(sample_code))
assert_equal(trees.count(), 1)
assert_is_not_none(test_ast)
ti = AstTransformerItem(NoisyIdentityTransform)
trees.create_transformed_child(test_ast, ast_transform_item=ti)
assert_equal(trees.count(), 2)
def __init__(self):
self.ast_tree_manager = AstTreeManager()
self.ast_transformer_manager = AstTransformerManager()
def __init__(self):
self.ast_tree_manager = AstTreeManager()
self.ast_transformer_manager = AstTransformerManager()
