Список всех модулей, входящих в пакет python?

Есть ли простой способ найти все модули, входящие в пакет python? Я нашел эту старую дискуссию, которая на самом деле не является окончательной, но я бы хотел, чтобы у меня был определенный ответ, прежде чем я разверну свое собственное решение основанный на os.listdir().

Ответ 1

Да, вы хотите что-то, основанное на pkgutil или аналогичном - таким образом вы можете обрабатывать все пакеты, независимо от того, находятся ли они в яйцах или на молниях или так (где os.listdir не поможет).

import pkgutil

# this is the package we are inspecting -- for example 'email' from stdlib
import email

package = email
for importer, modname, ispkg in pkgutil.iter_modules(package.__path__):
    print "Found submodule %s (is a package: %s)" % (modname, ispkg)

Как импортировать их тоже? Вы можете просто использовать __import__ как обычно:

import pkgutil

# this is the package we are inspecting -- for example 'email' from stdlib
import email

package = email
prefix = package.__name__ + "."
for importer, modname, ispkg in pkgutil.iter_modules(package.__path__, prefix):
    print "Found submodule %s (is a package: %s)" % (modname, ispkg)
    module = __import__(modname, fromlist="dummy")
    print "Imported", module

Ответ 2

Правильным инструментом для этого задания является pkgutil.walk_packages.

Чтобы перечислить все модули в вашей системе:

import pkgutil
for importer, modname, ispkg in pkgutil.walk_packages(path=None, onerror=lambda x: None):
    print(modname)

Помните, что walk_packages импортирует все подпакеты, но не подмодули.

Если вы хотите перечислить все подмодули определенного пакета, вы можете использовать что-то вроде этого:

import pkgutil
import scipy
package=scipy
for importer, modname, ispkg in pkgutil.walk_packages(path=package.__path__,
                                                      prefix=package.__name__+'.',
                                                      onerror=lambda x: None):
    print(modname)

iter_modules отображает только модули, которые являются одноуровневыми. walk_packages получает все подмодули. Например, в случае scipy walk_packages возвращает

scipy.stats.stats

в то время как iter_modules возвращает только

scipy.stats

Документация по pkgutil (http://docs.python.org/library/pkgutil.html) не перечисляет все интересные функции, определенные в /usr/lib/python 2.6/pkgutil.py.

Возможно, это означает, что функции не являются частью "общедоступного" интерфейса и могут быть изменены.

Однако, по крайней мере, от Python 2.6 (и, возможно, более ранних версий?) pkgutil поставляется с методом walk_packages, который рекурсивно просматривает все доступных модулей.

Ответ 3

Это работает для меня:

import types

for key, obj in nltk.__dict__.iteritems():
    if type(obj) is types.ModuleType: 
        print key

Ответ 4

Вот один путь, с моей головы:

>>> import os
>>> filter(lambda i: type(i) == type(os), [getattr(os, j) for j in dir(os)])
[<module 'UserDict' from '/usr/lib/python2.5/UserDict.pyc'>, <module 'copy_reg' from '/usr/lib/python2.5/copy_reg.pyc'>, <module 'errno' (built-in)>, <module 'posixpath' from '/usr/lib/python2.5/posixpath.pyc'>, <module 'sys' (built-in)>]

Его можно было бы очистить и улучшить.

EDIT: Здесь немного более приятная версия:

>>> [m[1] for m in filter(lambda a: type(a[1]) == type(os), os.__dict__.items())]
[<module 'copy_reg' from '/usr/lib/python2.5/copy_reg.pyc'>, <module 'UserDict' from '/usr/lib/python2.5/UserDict.pyc'>, <module 'posixpath' from '/usr/lib/python2.5/posixpath.pyc'>, <module 'errno' (built-in)>, <module 'sys' (built-in)>]
>>> [m[0] for m in filter(lambda a: type(a[1]) == type(os), os.__dict__.items())]
['_copy_reg', 'UserDict', 'path', 'errno', 'sys']

ПРИМЕЧАНИЕ.. Это также найдет модули, которые необязательно могут быть расположены в подкаталоге пакета, если они втянуты в его файл __init__.py, поэтому это зависит от того, что вы имеете в виду "частью" пакета.