def visit_If(self, node): if not isinstance(node.test, bool): raise ParseError("Unexpected if statement" + debug.dump(node, ast3)) if node.test: return Splice(node.body) else: return Splice(node.orelse)
def main(): parser = argparse.ArgumentParser(usage='%(prog)s [options] input') args, options = arg_parser.add_pytype_and_parse(parser, sys.argv[1:]) filename = args.inputs[0] with open(filename, 'r') as f: src = f.read() module = annotate_ast.annotate_source(src, ast3, options) print(debug.dump(module, ast3))
def enter_If(self, node): # Evaluate the test and preemptively remove the invalid branch so we don't # waste time traversing it. node.test = conditions.evaluate(node.test, self.version, self.platform) if not isinstance(node.test, bool): raise ParseError("Unexpected if statement" + debug.dump(node, ast3)) if node.test: node.orelse = [] else: node.body = []
def parse_pyi_debug( src: str, filename: str, module_name: str, options: Optional[PyiOptions] = None, ) -> Tuple[pytd.TypeDeclUnit, _GeneratePytdVisitor]: """Debug version of parse_pyi.""" options = options or PyiOptions() feature_version = _feature_version(options.python_version) root = _parse(src, feature_version, filename) print(debug.dump(root, ast3, include_attributes=False)) gen_pytd = _GeneratePytdVisitor(src, filename, module_name, options) root = gen_pytd.visit(root) print("---transformed parse tree--------------------") print(root) root = post_process_ast(root, src, module_name) print("---post-processed---------------------") print(root) print("------------------------") print(gen_pytd.defs.type_map) print(gen_pytd.defs.module_path_map) return root, gen_pytd
def parse_pyi_debug( src: str, filename: str, module_name: str, python_version: VersionType, platform: Optional[str] = None ) -> Tuple[pytd.TypeDeclUnit, GeneratePytdVisitor]: """Debug version of parse_pyi.""" feature_version = _feature_version(python_version) python_version = utils.normalize_version(python_version) root = _parse(src, feature_version, filename) print(debug.dump(root, ast3, include_attributes=False)) gen_pytd = GeneratePytdVisitor(src, filename, module_name, python_version, platform) root = gen_pytd.visit(root) print("---transformed parse tree--------------------") print(root) root = post_process_ast(root, src, module_name) print("---post-processed---------------------") print(root) print("------------------------") print(gen_pytd.defs.type_map) print(gen_pytd.defs.module_path_map) return root, gen_pytd
def show(self, node): print(debug.dump(node, ast3, include_attributes=True))