Cantumkan semua modul yang merupakan bagian dari paket python?

Apakah ada cara mudah untuk menemukan semua modul yang merupakan bagian dari paket python? Saya telah menemukan diskusi lama ini , yang tidak benar-benar konklusif, tetapi saya ingin memiliki jawaban yang pasti sebelum saya meluncurkan solusi saya sendiri berdasarkan os.listdir ().

Ya, Anda menginginkan sesuatu yang berdasarkan pkgutilatau serupa - dengan cara ini Anda dapat memperlakukan semua paket dengan sama terlepas dari apakah mereka berada dalam telur atau ritsleting atau lebih (di mana os.listdir tidak akan membantu).

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)

Bagaimana cara mengimpornya juga? Anda bisa menggunakan __import__seperti biasa:

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

Alat yang tepat untuk pekerjaan ini adalah pkgutil.walk_packages.

Untuk mencantumkan semua modul di sistem Anda:

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

Ketahuilah bahwa walk_packages mengimpor semua subpaket, tetapi bukan submodul.

Jika Anda ingin membuat daftar semua submodul dari paket tertentu maka Anda dapat menggunakan sesuatu seperti ini:

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 hanya mencantumkan modul dengan kedalaman satu tingkat. walk_packages mendapatkan semua submodul. Dalam kasus scipy, misalnya, walk_packages akan dikembalikan

scipy.stats.stats

sementara iter_modules hanya kembali

scipy.stats

Dokumentasi di pkgutil ( http://docs.python.org/library/pkgutil.html ) tidak mencantumkan semua fungsi menarik yang didefinisikan di /usr/lib/python2.6/pkgutil.py.

Mungkin ini berarti fungsi tersebut bukan bagian dari antarmuka "publik" dan dapat berubah.

Namun, setidaknya pada Python 2.6 (dan mungkin versi sebelumnya?) Pkgutil hadir dengan metode walk_packages yang secara rekursif berjalan melalui semua modul yang tersedia.

Ini bekerja untuk saya:

import types

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

Saya sedang mencari cara untuk memuat ulang semua submodul yang saya edit langsung di paket saya. Ini adalah kombinasi dari jawaban / komentar di atas, jadi saya memutuskan untuk mempostingnya di sini sebagai jawaban daripada komentar.

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

Inilah salah satu cara, di luar kepala saya:

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

Itu pasti bisa dibersihkan dan diperbaiki.

EDIT: Ini versi yang sedikit lebih bagus:

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

CATATAN: Ini juga akan menemukan modul yang mungkin tidak selalu terletak di subdirektori paket, jika mereka ditarik dalam __init__.pyfilenya, jadi itu tergantung pada apa yang Anda maksud dengan "bagian dari" sebuah paket.