給定路徑計算python文件的函數數量的好方法

Question

我不想導入我的模塊。 我必須計算給定 .py 文件路徑的函數數量。 這樣做的最佳方法是什么？

我想到的一件事是計算代碼中“def”的數量，但這似乎不是解決此問題的最佳方法。 有沒有更好的方法來計算函數的數量？

Answer 1

您可以使用ast.NodeVisitor ：

import inspect
import importlib
import ast

class CountFunc(ast.NodeVisitor):
    func_count = 0
    def visit_FunctionDef(self, node):
        self.func_count += 1


mod = "/path/to/some.py"

p = ast.parse(open(mod).read())

f = CountFunc()
f.visit(p)

print(f.func_count)

如果您想包含 lambda，您需要添加一個visit_Lambda ：

 def visit_Lambda(self, node):
    self.func_count += 1

這將找到所有定義，包括任何類的方法，我們可以添加更多限制來禁止：

class CountFunc(ast.NodeVisitor):
    func_count = 0

    def visit_ClassDef(self, node):
        return 

    def visit_FunctionDef(self, node):
        self.func_count += 1


    def visit_Lambda(self, node):
        self.func_count += 1

您可以根據自己的喜好定制代碼， greentreesnakes 文檔中描述了所有節點及其屬性

Answer 2

要計算頂級定義，請使用ast模塊，如下所示：

import ast

with open(filename) as f:
    tree = ast.parse(f.read())
    sum(isinstance(exp, ast.FunctionDef) for exp in tree.body)

Answer 3

Padraic 擊敗了我，但這是我的代碼，它適用於 Python 2 和 Python 3。

from __future__ import print_function
import ast

class FunctionCounter(ast.NodeVisitor):
    def __init__(self, filename):
        self.function_count = 0
        with open(filename) as f:
            module = ast.parse(f.read())
            self.visit(module)

    def visit_FunctionDef(self, node):
        print('function: {}'.format(node.name))
        self.function_count += 1

    # Uncomment this to disable counting methods, properties within a
    # class
    # def visit_ClassDef(self, node):
    #     pass

if __name__ == '__main__':
    counter = FunctionCounter('simple.py')
    print('Number of functions: {}'.format(counter.function_count))

討論

• 此代碼不僅計算模塊級別的函數，還計算嵌套在類定義中的函數（方法和屬性）。
• 要在類定義中禁用計數功能，請取消注釋visit_ClassDef的 2 行

Answer 4

您可以使用pyclbr模塊來獲取模塊的結果，唯一的問題是您需要使用模塊的名稱，因為您將導入它而不是文件路徑，這也得益於from X import Y識別 python 源基於模塊（不是像math這樣的內置模塊）

from pyclbr import readmodule_ex, Function
#readmodule_ex is function 1

def test(): #2
    pass
def other_func(): #3
    pass

class Thing:
    def method(self):
        pass


result = readmodule_ex("test") #this would be it's own file if it is test.py

funcs = sum(isinstance(v,Function) for v in result.values())

print(funcs)

Answer 5

你可以使用

len(dir(module))

希望它有幫助

Answer 6

我的回答是對@ Naveen Kumar 的回答的編輯，因為我目前無法編輯他的回答。 您可以使用

import module

len(dir(module))

IE：

import math
print(len(dir(math)))

輸出：

 63