class DeclassifyVisitor(CopyVisitor):
def __init__(self, name, finder):
super(DeclassifyVisitor,self).__init__()
self.name = name
self.assignee_visitor = AssigneeVisitor()
self.finder = finder
def preorder(self, tree, outside_scope, *args):
self.assignees = self.assignee_visitor.preorder(tree)
self.outside_scope = outside_scope
return super(DeclassifyVisitor, self).preorder(tree, *args)
def visitVarAssign(self, n):
return AttrAssign(Name(self.name), n.target,
self.dispatch(n.value))
def visitName(self, n):
if n.name in self.assignees:
if n.name in self.outside_scope:
return IfExp(CallHASATTR([Name(self.name), String(n.name)]),
GetAttr(Name(self.name), n.name),
Name(n.name))
else:
return GetAttr(Name(self.name), n.name)
else:
return Name(n.name)
def visitFunction(self, n):
temp = generate_name(n.name + '_temp')
name = n.name
n.name = temp
n = self.finder.preorder(n, self.outside_scope)
return [n, AttrAssign(Name(self.name), name, Name(temp))]
def visitClass(self, n):
temp = generate_name(n.name + '_temp')
name = n.name
n.name = temp
n = self.finder.preorder(n, self.outside_scope)
return n + [AttrAssign(Name(self.name), name, Name(temp))]
def visitLambda(self, n):
return self.finder.preorder_expr(n)
def visitStmtList(self, n):
nodes = []
for s in n.nodes:
if isinstance(s, Class) or isinstance(s, Function):
nodes += self.dispatch(s)
else:
nodes += [self.dispatch(s)]
return StmtList(nodes)
def visitCallFunc(self, n):
return specializeCallFunc(self, n)
class ClassFindVisitor(CopyVisitor):
def __init__(self):
super(ClassFindVisitor,self).__init__()
self.assignee_visitor = AssigneeVisitor()
def preorder(self, tree, outside_scope, *args):
scope = self.assignee_visitor.preorder(tree) | outside_scope
return super(ClassFindVisitor, self).preorder(tree, scope)
def preorder_expr(self, tree, *args):
return super(ClassFindVisitor, self).preorder(tree, *args)
def visitModule(self, n, scope):
return Module(self.dispatch(n.node, scope))
def visitStmtList(self, n, scope):
nodes = []
for s in n.nodes:
if isinstance(s, Class):
nodes += self.dispatch(s, scope)
else:
nodes += [self.dispatch(s, scope)]
return StmtList(nodes)
def visitClass(self, n, scope):
classtemp = generate_name(n.name + '_temp')
bases = map(self.dispatch, n.bases)
stmts = [VarAssign(classtemp,
CallCREATECLASS([List(bases)]))]
stmts += DeclassifyVisitor(classtemp, self).preorder(n.code, scope).nodes
stmts += [VarAssign(n.name, Name(classtemp))]
return stmts
def visitFunction(self, n, scope):
scope = scope | self.assignee_visitor.preorder(n.code)
return Function(n.name, n.args, self.dispatch(n.code, scope))
def visitCallFunc(self, n, *args):
return specializeCallFunc(self,n)
def __init__(self):
super(ClassFindVisitor,self).__init__()
self.assignee_visitor = AssigneeVisitor()
def __init__(self, name, finder):
super(DeclassifyVisitor,self).__init__()
self.name = name
self.assignee_visitor = AssigneeVisitor()
self.finder = finder