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) ```