Liệt kê tất cả các mô-đun là một phần của gói python?

Có cách nào đơn giản để tìm tất cả các mô-đun là một phần của gói python không? Tôi đã tìm thấy cuộc thảo luận cũ này , nó không thực sự kết luận, nhưng tôi muốn có câu trả lời chắc chắn trước khi tôi đưa ra giải pháp của riêng mình dựa trên os.listdir ().

Có, bạn muốn một cái gì đó dựa trên pkgutilhoặc tương tự - bằng cách này, bạn có thể coi tất cả các gói như nhau bất kể chúng nằm trong trứng hay khóa kéo hoặc tương tự (trong đó os.listdir sẽ không giúp được gì).

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)

Làm thế nào để nhập chúng quá? Bạn chỉ có thể sử dụng __import__như bình thường:

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

Công cụ phù hợp cho công việc này là pkgutil.walk_packages.

Để liệt kê tất cả các mô-đun trên hệ thống của bạn:

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

Lưu ý rằng walk_packages nhập tất cả các gói con, nhưng không nhập các mô-đun con.

Nếu bạn muốn liệt kê tất cả các mô-đun con của một gói nhất định thì bạn có thể sử dụng một cái gì đó như sau:

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 chỉ liệt kê các mô-đun sâu một cấp. walk_packages nhận tất cả các mô-đun con. Trong trường hợp scipy, ví dụ: walk_packages trả về

scipy.stats.stats

trong khi iter_modules chỉ trả về

scipy.stats

Tài liệu trên pkgutil ( http://docs.python.org/library/pkgutil.html ) không liệt kê tất cả các hàm thú vị được định nghĩa trong /usr/lib/python2.6/pkgutil.py.

Có lẽ điều này có nghĩa là các chức năng không phải là một phần của giao diện "công khai" và có thể thay đổi.

Tuy nhiên, ít nhất kể từ Python 2.6 (và có lẽ là các phiên bản trước đó?) Pkgutil đi kèm với phương thức walk_packages mà đệ quy duyệt qua tất cả các mô-đun có sẵn.

Điều này phù hợp với tôi:

import types

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

Tôi đang tìm cách tải lại tất cả các mô-đun con mà tôi đang chỉnh sửa trực tiếp trong gói của mình. Nó là sự kết hợp của các câu trả lời / nhận xét ở trên, vì vậy tôi quyết định đăng nó ở đây như một câu trả lời hơn là một bình luận.

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

Đây là một cách, ngoài đầu tôi:

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

Nó chắc chắn có thể được làm sạch và cải thiện.

CHỈNH SỬA: Đây là một phiên bản đẹp hơn một chút:

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

LƯU Ý: Điều này cũng sẽ tìm thấy các mô-đun có thể không nhất thiết phải nằm trong thư mục con của gói, nếu chúng được kéo trong __init__.pytệp của nó , vì vậy nó phụ thuộc vào ý bạn muốn nói đến "một phần" của gói.