Как импортировать элементы всех модулей в пакете?

Я хочу, чтобы члены всех модулей в пакете находились в пространстве имен пакета, и я не хочу, чтобы сами модули находились в пространстве имен.

Мне удалось это сделать, адаптировав кое-что, что я использовал в Python 2, для автоматического импорта подключаемых модулей, чтобы они также работали в Python 3.

Вкратце, вот как это работает:

1. Файл __init__.py пакета импортирует все остальные файлы Python в том же каталоге пакета, за исключением тех, чьи имена начинаются с символа '_' (подчеркивание).

2. Затем он добавляет любые имена в пространстве имен импортированного модуля к имени модуля __init__ (которое также является пространством имен пакета). Обратите внимание, что мне пришлось сделать модуль example_module явно import foo из .foo_module.

Одним из важных аспектов такого подхода является осознание того, что он динамичен и не требует, чтобы имена модулей пакета были жестко закодированы в файле __init__.py. Конечно, это требует большего количества кода для выполнения, но также делает его очень универсальным и способным работать практически с любым (одноуровневым) пакетом - поскольку он автоматически импортирует новые модули при их добавлении и больше не пытается импортировать любые удаленные из каталога.

test.py:

from package import *

print(example('derp'))

__init__.py:

def _import_all_modules():
    """ Dynamically imports all modules in this package. """
    import traceback
    import os
    global __all__
    __all__ = []
    globals_, locals_ = globals(), locals()

    # Dynamically import all the package modules in this file's directory.
    for filename in os.listdir(__name__):
        # Process all python files in directory that don't start
        # with underscore (which also prevents this module from
        # importing itself).
        if filename[0] != '_' and filename.split('.')[-1] in ('py', 'pyw'):
            modulename = filename.split('.')[0]  # Filename sans extension.
            package_module = '.'.join([__name__, modulename])
            try:
                module = __import__(package_module, globals_, locals_, [modulename])
            except:
                traceback.print_exc()
                raise
            for name in module.__dict__:
                if not name.startswith('_'):
                    globals_[name] = module.__dict__[name]
                    __all__.append(name)

_import_all_modules()

foo_module.py:

def foo(bar):
    return bar

example_module.py:

from .foo_module import foo  # added

def example(arg):
    return foo(arg)

Я думаю, вы можете получить нужные значения, не загромождая пространство имен, используя импорт стиля from module import name. Я думаю, что этот импорт будет работать для того, о чем вы просите:

Импорт для example_module.py:

from package.foo_module import foo

Импорт для __init__.py:

from package.foo_module import foo
from package.example_module import example

__all__ = [foo, example] # not strictly necessary, but makes clear what is public

Импорт для test.py:

from package import example

Обратите внимание, что это работает, только если вы используете test.py (или что-то еще на том же уровне иерархии пакетов). В противном случае вам нужно будет убедиться, что папка, содержащая package, находится в пути поиска модуля python (либо установив пакет в том месте, где Python будет его искать, либо добавив соответствующую папку в sys.path).