有没有一种标准的方法可以在软件包中列出Python模块的名称？

Question

有没有一种简单的方法可以列出软件包中所有模块的名称，而无需使用__all__？

例如，给定此程序包：

/testpkg
/testpkg/__init__.py
/testpkg/modulea.py
/testpkg/moduleb.py

我想知道是否有标准或内置的方式来做这样的事情：

>>> package_contents("testpkg")
['modulea', 'moduleb']

手动方法是遍历模块搜索路径，以找到包的目录。然后可以列出该目录中的所有文件，过滤出唯一命名为py / pyc /
pyo的文件，剥离扩展名，然后返回该列表。但这对于模块导入机制已经在内部完成的工作来说似乎是相当多的工作。该功能在任何地方都可以使用吗？

Answer 1

也许这会满足您的需求？

import imp
import os
MODULE_EXTENSIONS = ('.py', '.pyc', '.pyo')

def package_contents(package_name):
    file, pathname, description = imp.find_module(package_name)
    if file:
        raise ImportError('Not a package: %r', package_name)
    # Use a set because some may be both source and compiled.
    return set([os.path.splitext(module)[0]
        for module in os.listdir(pathname)
        if module.endswith(MODULE_EXTENSIONS)])