將簡單命令轉換爲python中的AST

Question

我想要一個Python函數，它將字符串命令轉換爲AST（抽象語法樹）。

該命令的語法如下：

commandName(3, "hello", 5.0, x::int) 

命令可以接受任何數量的逗號分隔值的，可以是任一

1. 整數
2. 字符串
3. 浮標
4. 類型

假設函數被調用convert_to_ast，然後

convert_to_ast('commandName(3, "hello", 5.0, x::int)') 

應該產生以下AST：

{ 
    'type': 'command', 
    'name': 'commandName', 
    'args': [{ 
    'type': 'int', 
    'value': 3 
    }, { 
    'type': 'str', 
    'value': 'Hello' 
    }, { 
    'type': 'float', 
    'value': 5.0 
    }, { 
    'type': 'var', 
    'kind': 'int', 
    'name': 'x 
    }] 

Answer 1

好像你可以只是評估字符串，然後從那裏摘掉類型：

>>> items = ast.literal_eval('(404.5, "Hello", 5)') 
>>> [{'type': type(item).__name__, 'value': item} for item in items] 
[{'type': 'float', 'value': 404.5}, {'type': 'str', 'value': 'Hello'}, {'type': 'int', 'value': 5}] 

當然，如果你想做更多有趣的事情，你可以訪問AST di rectly：

>>> ast.dump(ast.parse('(404.5, "Hello", 5)')) 
"Module(body=[Expr(value=Tuple(elts=[Num(n=404.5), Str(s='Hello'), Num(n=5)], ctx=Load()))])" 
>>> ast.parse('(404.5, "Hello", 5)').body[0].value.elts 
[<_ast.Num object at 0x107fa1250>, <_ast.Str object at 0x107fa1290>, <_ast.Num object at 0x107fa12d0>] 

對於除解析一個元組（如你加入到這個問題），我們仍然可以使用Python的AST解析這個更一般的事情（只要你的語法是有效的Python） 。在這種情況下，我們將創建一個ast.NodeVisitor，它將提取我們在訪問我們關心的Python AST的每個節點時的信息。在這種情況下，我們關心Call，Num，Str和Name節點：

import ast 

class Parser(ast.NodeVisitor): 

    def __init__(self): 
     self.calls = [] 
     self.current_command = None 

    def visit_Call(self, node): 
     name = node.func.id 
     self.current_command = { 
      'type': 'command', 
      'name': name, 
      'args': [] 
     } 
     self.calls.append(self.current_command) 
     for arg in node.args: 
      self.visit(arg) 
     self.current_command = None 

    def visit_Num(self, node): 
     if not self.current_command: 
      return 
     args = self.current_command['args'] 
     arg = { 
      'type': type(node.n).__name__, 
      'value': node.n 
     } 
     args.append(arg) 

    def visit_Str(self, node): 
     if not self.current_command: 
      return 
     args = self.current_command['args'] 
     arg = { 
      'type': 'str', 
      'value': node.s 
     } 
     args.append(arg) 

    def visit_Name(self, node): 
     if not self.current_command: 
      return 
     args = self.current_command['args'] 
     arg = { 
      'type': 'type', 
      'kind': node.id 
     } 
     args.append(arg) 


S = 'commandName(3, "hello", 5.0, int)' 

tree = ast.parse(S) 
p = Parser() 
p.visit(tree) 
print p.calls