def _ParseAndVisit(source): mod = pythonparser.parse(source) future_features = stmt.visit_future(mod) b = block.ModuleBlock('__main__', '<test>', source, future_features) visitor = stmt.StatementVisitor(b) visitor.visit(mod) return visitor
def _ParseAndVisit(source): mod = ast.parse(source) future_features = stmt.visit_future(mod) b = block.ModuleBlock('__main__', 'grumpy', 'grumpy/lib', '<test>', source.split('\n'), future_features) visitor = stmt.StatementVisitor(b) visitor.visit(mod) return visitor
def testWriteBlock(self): writer = util.Writer() mod_block = block.ModuleBlock(None, '__main__', '<test>', '', imputil.FutureFeatures()) writer.write_block(mod_block, 'BODY') output = writer.getvalue() dispatch = 'switch πF.State() {\n\tcase 0:\n\tdefault: panic' self.assertIn(dispatch, output)
def testWriteBlock(self): writer = util.Writer() writer.write_block(block.ModuleBlock( '__main__', 'grumpy', 'grumpy/lib', '<test>', []), 'BODY') output = writer.out.getvalue() dispatch = 'switch πF.State() {\n\tcase 0:\n\tdefault: panic' self.assertIn(dispatch, output) self.assertIn('return nil, nil\n}', output)
def _ParseAndVisit(source): mod = pythonparser.parse(source) _, future_features = imputil.parse_future_features(mod) importer = imputil.Importer(None, 'foo', 'foo.py', False) b = block.ModuleBlock(importer, '__main__', '<test>', source, future_features) visitor = stmt.StatementVisitor(b) visitor.visit(mod) return visitor
def testWriteBlock(self): writer = util.Writer() mod_block = block.ModuleBlock(imputil_test.MockPath(), '__main__', '<test>', '', stmt.FutureFeatures()) writer.write_block(mod_block, 'BODY') output = writer.getvalue() dispatch = 'switch πF.State() {\n\tcase 0:\n\tdefault: panic' self.assertIn(dispatch, output) self.assertIn('return nil, nil\n}', output)
def _MakeModuleBlock(): return block.ModuleBlock(None, '__main__', '<test>', '', imputil.FutureFeatures())
def _MakeModuleBlock(): return block.ModuleBlock('__main__', 'grumpy', 'grumpy/lib', '<test>', [], stmt.FutureFeatures())
def _MakeModuleBlock(): importer = imputil.Importer(None, '__main__', '/tmp/foo.py', False) return block.ModuleBlock(importer, '__main__', '<test>', '', imputil.FutureFeatures())
def _ParseAndVisit(source): b = block.ModuleBlock('__main__', 'grumpy', 'grumpy/lib', '<test>', source.split('\n')) visitor = stmt.StatementVisitor(b) visitor.visit(ast.parse(source)) return visitor
def _MakeModuleBlock(): return block.ModuleBlock('__main__', 'grumpy', 'grumpy/lib', '<test>', [])
def _MakeModuleBlock(): return block.ModuleBlock(imputil_test.MockPath(), '__main__', '<test>', '', stmt.FutureFeatures())
def _MakeModuleBlock(): return block.ModuleBlock('__main__', '<test>', '', stmt.FutureFeatures())
def main(script=None, modname=None): assert script and modname, 'Script "%s" or Modname "%s" is empty' % ( script, modname) gopath = os.getenv('GOPATH', None) if not gopath: print >> sys.stderr, 'GOPATH not set' return 1 with open(script) as py_file: py_contents = py_file.read() try: mod = pythonparser.parse(py_contents) except SyntaxError as e: print >> sys.stderr, '{}: line {}: invalid syntax: {}'.format( e.filename, e.lineno, e.text) return 2 # Do a pass for compiler directives from `from __future__ import *` statements try: future_node, future_features = imputil.parse_future_features(mod) except util.CompileError as e: print >> sys.stderr, str(e) return 2 importer = imputil.Importer(gopath, modname, script, future_features.absolute_import) full_package_name = modname.replace('.', '/') mod_block = block.ModuleBlock(importer, full_package_name, script, py_contents, future_features) visitor = stmt.StatementVisitor(mod_block, future_node) # Indent so that the module body is aligned with the goto labels. with visitor.writer.indent_block(): try: visitor.visit(mod) except util.ParseError as e: print >> sys.stderr, str(e) return 2 writer = util.Writer(sys.stdout) tmpl = textwrap.dedent("""\ package $package import πg "grumpy" var Code *πg.Code func init() { \tCode = πg.NewCode("<module>", $script, nil, 0, func(πF *πg.Frame, _ []*πg.Object) (*πg.Object, *πg.BaseException) { \t\tvar πR *πg.Object; _ = πR \t\tvar πE *πg.BaseException; _ = πE""") writer.write_tmpl(tmpl, package=modname.split('.')[-1], script=util.go_str(script)) with writer.indent_block(2): for s in sorted(mod_block.strings): writer.write('ß{} := πg.InternStr({})'.format(s, util.go_str(s))) writer.write_temp_decls(mod_block) writer.write_block(mod_block, visitor.writer.getvalue()) writer.write_tmpl(textwrap.dedent("""\ \t\treturn nil, πE \t}) \tπg.RegisterModule($modname, Code) }"""), modname=util.go_str(modname)) return 0