从未知的python文件中提取所有类和函数

Question

我正在使用 AST 库来提取这样一个文件的所有细节，

import ast

file = open("TestFile.py", "r")

f = file.read()

p = ast.parse(f)

classFunc = [node.name for node in ast.walk(p) if isinstance(node, ast.ClassDef) or isinstance(node, ast.FunctionDef)]

print classFunc

这给了我一个输出，

['adf', 'A', 'message', 'dghe', '__init__', 'mess', 'B', 'teqwtdg']

这里'adf'和'A'是主类，'message'和'dghe'是'adf'下的函数，' init '和'mess'是A下的函数，'B'是'下的一个类A'和'teqwtdg'是'B'下的一个函数。

所以，现在我的任务是编写一个 python 文件，在其中创建类实例并调用这些函数（这些来自未知文件）。我想安排这个列表，以便我可以很容易地知道哪些是主类，哪些是子类以及哪个函数属于哪个类。我怎样才能做到这一点？

Answer 1

以下代码将遍历文件并在层次结构中创建一个对象。import astimport pprintdef create_py_object(node_to_traverse, current_object):    for node in node_to_traverse.body:        if isinstance(node, ast.ClassDef):            current_object.append({node.name: []})            create_py_object(node, current_object[-1:][0][node.name])        if isinstance(node, ast.FunctionDef):            current_object.append({node.name: 'func'})file = open("TestFile.py", "r")f = file.read()node_to_traverse = ast.parse(f)py_file_structure = []create_py_object(node_to_traverse, py_file_structure)pprint.pprint(py_file_structure)输出 ：[{'adf': [{'message': 'func'}, {'dghe': 'func'}]}, {'A': [{'__init__': 'func'}, {'mess': 'func'}, {'B': [{'teqwtdg': 'func'}]}]}]