如何解析 ast.Assign 个对象？

Question

我正在尝试使用 ast 提取有关我的 python 代码的顶级分配的信息，但我不太熟悉如何有效地解析它。

例如

import some_module

a = 1
b = some_module.load_number("some_path")

def plus(a,b):
   return a + b

c = plus(a,b)

我希望能够从中提取具有如下数据的赋值表达式列表：

{
'targets' : 'a',
'type' : 'static_value',
'value' : 1
},

{
'targets' : 'b',
'type' : 'function',
'value' : 'some_module.load_number'
'inputs' : ['some_path']
},

{
'targets' : 'c',
'type' : 'function',
'value' : 'plus'
'inputs' : ['var.a', 'var.b']
},

我只关心作业。我目前的方法是这样的。

import ast
with open("test_code.py", "rb") as f:
   content = f.read()
code = ast.parse(content)
results = []

def parse_assignment(node):
   targets = [x.id for x in node.targets] #extract variables names being assigned to

   ### Extract assigned values for primitives; only str shown for brevity
   if isinstance(node.value, ast.Str):
      values = [node.value.s]
      type = 'str'
   
   ### Function call version
   elif isinstance(node.value, ast.Call):
      values = [node.value.func.value.id],
      type = ['function']

   return {'targets' : targets, 'values' : values, 'types' : types}

for node in ast.walk(code):
   if isinstance(node, ast.Assign):
      print(node)
      results = results + parse_assignment(node)

这里显示的方法有两个问题：

我认为 ast.walk 在这里不是一个好主意，因为它似乎是递归的，并且可能会在函数定义或其他内容中获取分配。我只想要顶级作业。
我解析函数名称的方式似乎不正确。在此示例中，我想解析 some_module.load_number，但我得到的是 some_module。如何从 ast.Call 对象中获取完整的函数名称？

Answer 1

而不是 ast.walk，只需遍历 code.body 并挑选出 Assign 个对象。

关于函数名，Call值有一个func属性，其值为Attribute节点，value是模块名，value =18=] 是实际引用函数的属性。

>>> >>> call = code.body[2].value.func
>>> f'{call.value.id}.{call.attr}'
'some_module.load_number'

请注意，一般来说，Call 节点的 func 属性可以是任意表达式，其 run-time 值是可调用的。

如何解析 ast.Assign 个对象？

How can I parse ast.Assign objects?

python

abstract-syntax-tree

python-3.x