Listez tous les modules qui font partie d'un package python?

Existe-t-il un moyen simple de trouver tous les modules qui font partie d'un package python? J'ai trouvé cette vieille discussion , qui n'est pas vraiment concluante, mais j'aimerais avoir une réponse définitive avant de déployer ma propre solution basée sur os.listdir ().

Oui, vous voulez quelque chose basé sur pkgutilou similaire - de cette façon, vous pouvez traiter tous les paquets de la même manière, qu'ils soient dans des œufs ou des zips ou quelque chose de similaire (où os.listdir n'aidera pas).

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)

Comment les importer aussi? Vous pouvez simplement utiliser __import__comme d'habitude:

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

Le bon outil pour ce travail est pkgutil.walk_packages.

Pour lister tous les modules de votre système:

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

Sachez que walk_packages importe tous les sous-packages, mais pas les sous-modules.

Si vous souhaitez lister tous les sous-modules d'un certain package, vous pouvez utiliser quelque chose comme ceci:

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 ne répertorie que les modules qui ont une profondeur d'un niveau. walk_packages récupère tous les sous-modules. Dans le cas de scipy, par exemple, walk_packages renvoie

scipy.stats.stats

tandis que iter_modules ne renvoie que

scipy.stats

La documentation sur pkgutil ( http://docs.python.org/library/pkgutil.html ) ne répertorie pas toutes les fonctions intéressantes définies dans /usr/lib/python2.6/pkgutil.py.

Cela signifie peut-être que les fonctions ne font pas partie de l'interface "publique" et sont susceptibles de changer.

Cependant, au moins à partir de Python 2.6 (et peut-être des versions antérieures?) Pkgutil est livré avec une méthode walk_packages qui parcourt de manière récursive tous les modules disponibles.

Cela fonctionne pour moi:

import types

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

Je cherchais un moyen de recharger tous les sous-modules que j'édite en direct dans mon package. C'est une combinaison des réponses / commentaires ci-dessus, j'ai donc décidé de l'afficher ici comme une réponse plutôt qu'un commentaire.

package=yourPackageName
import importlib
import pkgutil
for importer, modname, ispkg in pkgutil.walk_packages(path=package.__path__, prefix=package.__name__+'.', onerror=lambda x: None):
    try:
        modulesource = importlib.import_module(modname)
        reload(modulesource)
        print("reloaded: {}".format(modname))
    except Exception as e:
        print('Could not load {} {}'.format(modname, e))

Voici une façon, du haut de ma tête:

>>> 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)>]

Il pourrait certainement être nettoyé et amélioré.

EDIT: Voici une version légèrement plus agréable:

>>> [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']

REMARQUE: Cela trouvera également des modules qui ne se trouvent pas nécessairement dans un sous-répertoire du package, s'ils sont extraits dans son __init__.pyfichier, donc cela dépend de ce que vous entendez par «partie» d'un package.