ast · Magic_sword2

[TOC] # 简介说明 原文链接： - [ast](https://docs.python.org/zh-cn/3.9/library/ast.html#module-ast) ``` 版本：ast 作用：抽象语法树 ``` # ast # *parse() ***** 解析字符串成为model对象，与compile()函数一样 ### 返回类型 ``` ast.Module ``` ### 示例内容 1. 解析成AST对象 ``` expr = """ def add(arg1, arg2): return arg1 + arg2 """ expr_ast = ast.parse(expr) ``` # *unparse() ***** 反解析ast树对象生成python代码 # *dump() ***** 返回ast树字符串格式 # *literal_eval() ***** 对表达式节点以及包含Python字面量或容器的字符串进行安全的求值 # *get_docstring() ***** 返回给定节点的文档字符 # *get_source_segment() ***** 获取生成节点的源代码的源代码段 # *increment_lineno() ***** 从节点开始，将树中每个节点的行号和结束行号递增n # *copy_location() ***** 如果可能，将源位置（lineno，col_offset，end_lineno和end_col_offset）从old_node复制到new_node，然后返回new_node # *iter_fields() ***** 迭代字段 # *iter_child_nodes() ***** 迭代子节点 # *walk() ***** 遍历节点 # *fix_missing_locations() ***** 在当前作用域下修复的ast语法树 ### 示例内容 1. 举例说明 ``` import ast expr = """ def add(arg1, arg2): return arg1 + arg2 """ expr_ast = ast.parse(expr) class CrazyTransformer(ast.NodeTransformer): def visit_BinOp(self, node): node.op = ast.Mult() return node transformer = CrazyTransformer() ast.fix_missing_locations(transformer.visit(expr_ast)) a = compile(expr_ast, "<string>", mode="exec") exec(a) print(add(8, 2)) ``` # *[new]-Module() ***** 节点，模块节点 # *[new]-Constant() ***** 节点，常量节点 # *[new]-FormattedValue() ***** 节点，FormattedValue # *[new]-JoinedStr() ***** 节点，JoinedStr # *[new]-List() ***** 节点，List # *[new]-Tuple() ***** 节点，Tuple # *[new]-Set() ***** 节点，Set # *[new]-Dict() ***** 节点，Dict # ast.AST # [new]-AST() ***** 抽象语法树基类 # _fields ***** 所有子节点的名字 # lineno ***** 行号 # end_lineno ***** 结束行号 # col_offset ***** 列偏移 # end_col_offset ***** 列偏移结束位置 # ast.NodeVisitor # [new]-NodeVisitor() ***** 抽象语法树遍历类 # visit() ***** 访问一个节点 # generic_visit() ***** 此访问者在节点的所有子节点上调用 visit() # ast.NodeTransformer # [ext]-NodeVisitor() ***** 抽象语法树遍历类 # [new]-NodeTransformer() ***** 子类 NodeVisitor 用于遍历抽象语法树，并允许修改节点 ### 示例内容 1. 举例说明 ``` import ast expr = """ def add(arg1, arg2): return arg1 + arg2 """ expr_ast = ast.parse(expr) class CrazyTransformer(ast.NodeTransformer): def visit_BinOp(self, node): node.op = ast.Mult() return node transformer = CrazyTransformer() modified = transformer.visit(expr_ast) pro = compile(modified, '<string>', 'exec') exec(pro) print(add(8,2)) ``` 2. 举例说明 ``` import ast class CodeTransformer(ast.NodeTransformer): def visit_BinOp(self, node): if isinstance(node.op, ast.Add): node.op = ast.Sub() self.generic_visit(node) return node def visit_FunctionDef(self, node): self.generic_visit(node) # 这里表示先去访问里面的children node if node.name == 'add': node.name = 'sub' args_num = len(node.args.args) args_num = len(node.args.args) args = tuple([arg.arg for arg in node.args.args]) print(str(args)) func_log_stmt = ''.join(["print('calling func: %s', " % node.name, "'args:'", ", %s" * args_num % args, ')']) node.body.insert(0, ast.parse(func_log_stmt)) # func_log_stmt = ''.join(["print 'calling func: %s', " % node.name, "'args:'", ", %s" * args_num % args]) # node.body.insert(0, ast.parse(func_log_stmt)) return node def visit_Name(self, node): replace = {'add': 'sub', 'x': 'a', 'y': 'b'} re_id = replace.get(node.id, None) node.id = re_id or node.id self.generic_visit(node) return node def visit_arg(self, node): self.generic_visit(node) replace = {'x': 'a', 'y': 'b'} node.arg = replace[node.arg] return node func_def = \ """ def add(x, y): return x+y """ r_node = ast.parse(func_def) transformer = CodeTransformer() r_node = transformer.visit(r_node) source = ast.unparse(r_node) print(source) ```