解析所有方法和类的python文件

Question

我正在尝试构建一个程序，允许用户浏览到包含 python 模块的文件夹。选择文件夹后，它将列出该文件夹中的所有 python 文件以及每个模块的所有类和方法。我的问题是，有什么方法可以在不打开每个文件并解析“def”或“class”的情况下做到这一点？我注意到有一个名为 mro 的函数，它返回一个类的属性，但这需要我通过导入访问该类。那么有什么办法可以得到相同的结果吗？提前谢谢！

Answer 1

这是我使用 AST 模块提出的，它正是我想要的。

def fillClassList(file):
    classList = []
    className = None
    mehotdName = None
    fileName = "C:\Transcriber\Framework\ctetest\RegressionTest\GeneralTest\\" + file
    fileObject = open(fileName,"r")
    text = fileObject.read()
    p = ast.parse(text)
    node = ast.NodeVisitor()
    for node in ast.walk(p):
        if isinstance(node, ast.FunctionDef) or isinstance(node, ast.ClassDef):
            if isinstance(node, ast.ClassDef):
                className = node.name
            else:
                methodName = node.name
            if className != None and methodName != None:
                subList = (methodName , className)
                classList.append(subList)
    return classList

Answer 2

如果你想知道文件的内容，没有办法查看文件:)

您的选择取决于您是要自己解析出感兴趣的内容，还是要让 Python 加载文件然后询问它找到了什么。

对于一个非常简单的 Python 文件，比如下面的 testme.py，你可以这样做（警告：不适合那些胃不好的人）：

testme.py：

class Foo (object):
    pass

def bar():
    pass 

分析.py：

import os.path

files = ['testme.py']
for f in files: 
    print f
    modname = os.path.splitext(f)[0]
    exec('import ' + modname)
    mod = eval(modname)
    for symbol in dir(mod):
        if symbol.startswith('__'):
            continue
        print '   ', symbol, type(eval(modname + '.' + symbol))

输出：

testme.py
   Foo <type 'type'>
   bar <type 'function'>

但是，当您扩展它以处理嵌套的包和模块以及损坏的代码和等等等等时，这将开始变得非常糟糕。 grep class 和/或 def 可能更容易，然后从那里开始。

玩得开心！我 :heart: 元编程

Answer 3

大部分 Python 的实现（包括解析器）都可以在 stdlib 中找到，所以仔细阅读 modules index 你应该会找到你需要的。首先想到的模块/包是importlib、inspect 和ast，但肯定还有其他感兴趣的模块。

Answer 4

我不得不在我的一个模块中替换很多代码，这是我获取类和方法的方式：

def listClass(file):

    with open(file,"r") as f:
        p = ast.parse(f.read())

    # get all classes from the given python file.
    classes = [c for c in ast.walk(p) if isinstance(c,ast.ClassDef)]

    out = dict()
    for x in classes:
        out[x.name] = [fun.name for fun in ast.walk(x) if isinstance(fun,ast.FunctionDef)]

    return out

样本pprint 输出：

{'Alert': ['__init__',
           'fg',
           'fg',
           'bg',
           'bg',
           'paintEvent',
           'drawBG',
           'drawAlert'],
 'AlertMouse': ['__init__', 'paintEvent', 'mouseMoveEvent'],
 'AlertPopup': ['__init__', 'mousePressEvent', 'keyPressEvent', 'systemInfo']}

Answer 5

谢谢，这是第一次作为用户的有用示例。上面的代码带有导入、打印输出，并且没有 1 个拼写错误；-)

import ast

classList = []
className = None
methodName = None
fileName = "C:\\fullPathToAPythonFile.py"
fileObject = open(fileName ,"r")
text = fileObject.read()
p = ast.parse(text)
node = ast.NodeVisitor()
for node in ast.walk(p):
    if isinstance(node, ast.FunctionDef) or isinstance(node, ast.ClassDef):
        if isinstance(node, ast.ClassDef):
            className = node.name
        else:
            methodName = node.name
        if className != None and methodName != None:
            subList = (methodName , className)
            classList.append(subList)
            print("class: " + className + ", method: " + methodName)