def main(projectRoot): print('Creating Conditional Expression Trees for %s' % (projectRoot)) codeTreeWalker = FunctionPythonASTIterator(projectRoot) for picklFilename in codeTreeWalker: location = picklFilename[:-(len('func_ast.pickl'))] print location processor = ConditionalExprTreeGenerator() processor.createTreeForLocation(location)
def main(projectRoot): print('Creating Conditional Expression Trees for %s' %(projectRoot)) codeTreeWalker = FunctionPythonASTIterator(projectRoot) for picklFilename in codeTreeWalker: location = picklFilename[:-(len('func_ast.pickl'))] print location processor = ConditionalExprTreeGenerator() processor.createTreeForLocation(location)
class ExpressionTreeProvider: def __init__(self): self.exprTreeGenerator = ConditionalExprTreeGenerator() self.exprTrees = {} def loadExpressionTree(self, location): filename = location + '/expr_tree.pickl' if filename in self.exprTrees: return self.exprTrees[filename] if not os.path.exists(filename): exprTree = self.exprTreeGenerator.generateTreeForLocation(location) self.exprTreeGenerator.saveTreeForLocation(exprTree, location) else: exprTree = pickle.load(file(filename)) self.exprTrees[filename] = exprTree return self.exprTrees[filename]
def __init__(self): self.exprTreeGenerator = ConditionalExprTreeGenerator() self.exprTrees = {}