Python?Ast抽象語法樹怎么應(yīng)用

本篇內(nèi)容主要講解“Python Ast抽象語法樹怎么應(yīng)用”，感興趣的朋友不妨來看看。本文介紹的方法操作簡(jiǎn)單快捷，實(shí)用性強(qiáng)。下面就讓小編來帶大家學(xué)習(xí)“Python Ast抽象語法樹怎么應(yīng)用”吧!

引言

Abstract Syntax Trees即抽象語法樹。Ast是python源碼到字節(jié)碼的一種中間產(chǎn)物，借助ast模塊可以從語法樹的角度分析源碼結(jié)構(gòu)。

此外，我們不僅可以修改和執(zhí)行語法樹，還可以將Source生成的語法樹unparse成python源碼。因此ast給python源碼檢查、語法分析、修改代碼以及代碼調(diào)試等留下了足夠的發(fā)揮空間。

1. AST簡(jiǎn)介 

Python官方提供的CPython解釋器對(duì)python源碼的處理過程如下：

Parse source code into a parse tree (Parser/pgen.c)

Transform parse tree into an Abstract Syntax Tree (Python/ast.c)

Transform AST into a Control Flow Graph (Python/compile.c)

Emit bytecode based on the Control Flow Graph (Python/compile.c)

即實(shí)際python代碼的處理過程如下：

源代碼解析 --> 語法樹 --> 抽象語法樹(AST) --> 控制流程圖 --> 字節(jié)碼

上述過程在python2.5之后被應(yīng)用。python源碼首先被解析成語法樹，隨后又轉(zhuǎn)換成抽象語法樹。在抽象語法樹中我們可以看到源碼文件中的python的語法結(jié)構(gòu)。

大部分時(shí)間編程可能都不需要用到抽象語法樹，但是在特定的條件和需求的情況下，AST又有其特殊的方便性。

下面是一個(gè)抽象語法的簡(jiǎn)單實(shí)例。

Module(body=[
    Print(
          dest=None,
          values=[BinOp( left=Num(n=1),op=Add(),right=Num(n=2))],
          nl=True,
 )])

2. 創(chuàng)建AST

2.1 Compile函數(shù)

先簡(jiǎn)單了解一下compile函數(shù)。

compile(source, filename, mode[, flags[, dont_inherit]]) 

• source -- 字符串或者AST（Abstract Syntax Trees）對(duì)象。一般可將整個(gè)py文件內(nèi)容file.read()傳入。

• filename -- 代碼文件名稱，如果不是從文件讀取代碼則傳遞一些可辨認(rèn)的值。

• mode -- 指定編譯代碼的種類?？梢灾付?exec, eval, single。

• flags -- 變量作用域，局部命名空間，如果被提供，可以是任何映射對(duì)象。

• flags和dont_inherit是用來控制編譯源碼時(shí)的標(biāo)志。

func_def = \
"""
def add(x, y):
    return x + y
print add(3, 5)
"""

使用Compile編譯并執(zhí)行：

>>> cm = compile(func_def, '<string>', 'exec')
>>> exec cm
>>> 8

上面func_def經(jīng)過compile編譯得到字節(jié)碼，cm即code對(duì)象，

True == isinstance(cm, types.CodeType)。

compile(source, filename, mode, ast.PyCF_ONLY_AST)  <==> ast.parse(source, filename='<unknown>', mode='exec')

2.2 生成ast

使用上面的func_def生成ast.

r_node = ast.parse(func_def)
print astunparse.dump(r_node)    # print ast.dump(r_node)

 下面是func_def對(duì)應(yīng)的ast結(jié)構(gòu)：

Module(body=[
    FunctionDef(
        name='add',
        args=arguments(
            args=[Name(id='x',ctx=Param()),Name(id='y',ctx=Param())],
            vararg=None,
            kwarg=None,
            defaults=[]),
        body=[Return(value=BinOp(
            left=Name(id='x',ctx=Load()),
            op=Add(),
            right=Name(id='y',ctx=Load())))],
        decorator_list=[]),
    Print(
        dest=None,
        values=[Call(
                func=Name(id='add',ctx=Load()),
                args=[Num(n=3),Num(n=5)],
                keywords=[],
                starargs=None,
                kwargs=None)],
        nl=True)
  ])

 除了ast.dump，有很多dump ast的第三方庫，如astunparse, codegen, unparse等。這些第三方庫不僅能夠以更好的方式展示出ast結(jié)構(gòu)，還能夠?qū)st反向?qū)С鰌ython source代碼。

module Python version "$Revision$"
{
  mod = Module(stmt* body)| Expression(expr body)
  stmt = FunctionDef(identifier name, arguments args, stmt* body, expr* decorator_list)
        | ClassDef(identifier name, expr* bases, stmt* body, expr* decorator_list)
        | Return(expr? value)
        | Print(expr? dest, expr* values, bool nl)| For(expr target, expr iter, stmt* body, stmt* orelse)
  expr = BoolOp(boolop op, expr* values)
       | BinOp(expr left, operator op, expr right)| Lambda(arguments args, expr body)| Dict(expr* keys, expr* values)| Num(object n) -- a number as a PyObject.
       | Str(string s) -- need to specify raw, unicode, etc?| Name(identifier id, expr_context ctx)
       | List(expr* elts, expr_context ctx) 
        -- col_offset is the byte offset in the utf8 string the parser uses
        attributes (int lineno, int col_offset)
  expr_context = Load | Store | Del | AugLoad | AugStore | Param
  boolop = And | Or 
  operator = Add | Sub | Mult | Div | Mod | Pow | LShift | RShift | BitOr | BitXor | BitAnd | FloorDiv
  arguments = (expr* args, identifier? vararg, identifier? kwarg, expr* defaults)
}

上面是部分摘自官網(wǎng)的 Abstract Grammar，實(shí)際遍歷ast Node過程中根據(jù)Node的類型訪問其屬性。

 3. 遍歷AST

python提供了兩種方式來遍歷整個(gè)抽象語法樹。

3.1 ast.NodeTransfer

將func_def中的add函數(shù)中的加法運(yùn)算改為減法，同時(shí)為函數(shù)實(shí)現(xiàn)添加調(diào)用日志。

  class CodeVisitor(ast.NodeVisitor):
      def visit_BinOp(self, node):
          if isinstance(node.op, ast.Add):
              node.op = ast.Sub()
          self.generic_visit(node)
      def visit_FunctionDef(self, node):
          print 'Function Name:%s'% node.name
          self.generic_visit(node)
          func_log_stmt = ast.Print(
              dest = None,
              values = [ast.Str(s = 'calling func: %s' % node.name, lineno = 0, col_offset = 0)],
              nl = True,
              lineno = 0,
              col_offset = 0,
          )
          node.body.insert(0, func_log_stmt)
  r_node = ast.parse(func_def)
  visitor = CodeVisitor()
  visitor.visit(r_node)
  # print astunparse.dump(r_node)
  print astunparse.unparse(r_node)
  exec compile(r_node, '<string>', 'exec')

 運(yùn)行結(jié)果：

Function Name:add
def add(x, y):
    print 'calling func: add'
    return (x - y)
print add(3, 5)
calling func: add
-2

3.2 ast.NodeTransformer

使用NodeVisitor主要是通過修改語法樹上節(jié)點(diǎn)的方式改變AST結(jié)構(gòu)，NodeTransformer主要是替換ast中的節(jié)點(diǎn)。

既然func_def中定義的add已經(jīng)被改成一個(gè)減函數(shù)了，那么我們就徹底一點(diǎn)，把函數(shù)名和參數(shù)以及被調(diào)用的函數(shù)都在ast中改掉，并且將添加的函數(shù)調(diào)用log寫的更加復(fù)雜一些，爭(zhēng)取改的面目全非：-）

  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)
          if node.name == 'add':
              node.name = 'sub'
          args_num = len(node.args.args)
          args = tuple([arg.id for arg in node.args.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))
          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
  r_node = ast.parse(func_def)
  transformer = CodeTransformer()
  r_node = transformer.visit(r_node)
  # print astunparse.dump(r_node)
  source = astunparse.unparse(r_node)
  print source
  # exec compile(r_node, '<string>', 'exec')        # 新加入的node func_log_stmt 缺少lineno和col_offset屬性
  exec compile(source, '<string>', 'exec')
  exec compile(ast.parse(source), '<string>', 'exec')

結(jié)果：

def sub(a, b):
    print 'calling func: sub', 'args:', a, b
    return (a - b)
print sub(3, 5)
calling func: sub args: 3 5
-2
calling func: sub args: 3 5
-2

 代碼中能夠清楚的看到兩者的區(qū)別。這里不再贅述。

 4.AST應(yīng)用

AST模塊實(shí)際編程中很少用到，但是作為一種源代碼輔助檢查手段是非常有意義的；語法檢查，調(diào)試錯(cuò)誤，特殊字段檢測(cè)等。

上面通過為函數(shù)添加調(diào)用日志的信息是一種調(diào)試python源代碼的一種方式，不過實(shí)際中我們是通過parse整個(gè)python文件的方式遍歷修改源碼。

4.1 漢字檢測(cè)

下面是中日韓字符的unicode編碼范圍

CJK Unified Ideographs 

Range: 4E00&mdash; 9FFF

Number of characters: 20992

Languages: chinese, japanese, korean, vietnamese

使用 unicode 范圍 \u4e00 - \u9fff 來判別漢字，注意這個(gè)范圍并不包含中文字符(e.g. u'；' == u'\uff1b') .

下面是一個(gè)判斷字符串中是否包含中文字符的一個(gè)類CNCheckHelper：

  class CNCheckHelper(object):
      # 待檢測(cè)文本可能的編碼方式列表
      VALID_ENCODING = ('utf-8', 'gbk')
      def _get_unicode_imp(self, value, idx = 0):
          if idx < len(self.VALID_ENCODING):
              try:
                  return value.decode(self.VALID_ENCODING[idx])
              except:
                  return self._get_unicode_imp(value, idx + 1)
      def _get_unicode(self, from_str):
          if isinstance(from_str, unicode):
              return None
          return self._get_unicode_imp(from_str)
      def is_any_chinese(self, check_str, is_strict = True):
          unicode_str = self._get_unicode(check_str)
          if unicode_str:
              c_func = any if is_strict else all
              return c_func(u'\u4e00' <= char <= u'\u9fff' for char in unicode_str)
          return False

 接口is_any_chinese有兩種判斷模式，嚴(yán)格檢測(cè)只要包含中文字符串就可以檢查出，非嚴(yán)格必須全部包含中文。

下面我們利用ast來遍歷源文件的抽象語法樹，并檢測(cè)其中字符串是否包含中文字符。

  class CodeCheck(ast.NodeVisitor):
      def __init__(self):
          self.cn_checker = CNCheckHelper()
      def visit_Str(self, node):
          self.generic_visit(node)
          # if node.s and any(u'\u4e00' <= char <= u'\u9fff' for char in node.s.decode('utf-8')):
          if self.cn_checker.is_any_chinese(node.s, True):
              print 'line no: %d, column offset: %d, CN_Str: %s' % (node.lineno, node.col_offset, node.s)
  project_dir = './your_project/script'
  for root, dirs, files in os.walk(project_dir):
      print root, dirs, files
      py_files = filter(lambda file: file.endswith('.py'), files)
      checker = CodeCheck()
      for file in py_files:
          file_path = os.path.join(root, file)
          print 'Checking: %s' % file_path
          with open(file_path, 'r') as f:
              root_node = ast.parse(f.read())
              checker.visit(root_node)

 上面這個(gè)例子比較的簡(jiǎn)單，但大概就是這個(gè)意思。

關(guān)于CPython解釋器執(zhí)行源碼的過程可以參考官網(wǎng)描述：PEP 339

4.2 Closure 檢查

一個(gè)函數(shù)中定義的函數(shù)或者lambda中引用了父函數(shù)中的local variable，并且當(dāng)做返回值返回。特定場(chǎng)景下閉包是非常有用的，但是也很容易被誤用。

關(guān)于python閉包的概念可以參考我的另一篇文章：理解Python閉包概念

這里簡(jiǎn)單介紹一下如何借助ast來檢測(cè)lambda中閉包的引用。代碼如下：

  class LambdaCheck(ast.NodeVisitor):
      def __init__(self):
          self.illegal_args_list = []
          self._cur_file = None
          self._cur_lambda_args = []
      def set_cur_file(self, cur_file):
          assert os.path.isfile(cur_file)， cur_file
          self._cur_file = os.path.realpath(cur_file)
      def visit_Lambda(self, node):
          """
          lambda 閉包檢查原則：
          只需檢測(cè)lambda expr body中args是否引用了lambda args list之外的參數(shù)
          """
          self._cur_lambda_args =[a.id for a in node.args.args]
          print astunparse.unparse(node)
          # print astunparse.dump(node)
          self.get_lambda_body_args(node.body)
          self.generic_visit(node)
      def record_args(self, name_node):
          if isinstance(name_node, ast.Name) and name_node.id not in self._cur_lambda_args:
              self.illegal_args_list.append((self._cur_file, 'line no:%s' % name_node.lineno, 'var:%s' % name_node.id))
      def _is_args(self, node):
          if isinstance(node, ast.Name):
              self.record_args(node)
              return True
          if isinstance(node, ast.Call):
              map(self.record_args, node.args)
              return True
          return False
      def get_lambda_body_args(self, node):
          if self._is_args(node): return
          # for cnode in ast.walk(node):
          for cnode in ast.iter_child_nodes(node):
              if not self._is_args(cnode):
                  self.get_lambda_body_args(cnode)

 遍歷工程文件：

  project_dir = './your project/script'
  for root, dirs, files in os.walk(project_dir):
      py_files = filter(lambda file: file.endswith('.py'), files)
      checker = LambdaCheck()
      for file in py_files:
          file_path = os.path.join(root, file)
          checker.set_cur_file(file_path)
          with open(file_path, 'r') as f:
              root_node = ast.parse(f.read())
              checker.visit(root_node)
      res = '\n'.join([' ## '.join(info) for info in checker.illegal_args_list])
      print res

由于Lambda(arguments args, expr body)中的body expression可能非常復(fù)雜，上面的例子中僅僅處理了比較簡(jiǎn)單的body expr?？筛鶕?jù)自己工程特點(diǎn)修改和擴(kuò)展檢查規(guī)則。為了更加一般化可以單獨(dú)寫一個(gè)visitor類來遍歷lambda節(jié)點(diǎn)。

到此，相信大家對(duì)“Python Ast抽象語法樹怎么應(yīng)用”有了更深的了解，不妨來實(shí)際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！