Як завантажити всі модулі в папку?

Чи міг би хтось надати мені хороший спосіб імпорту цілого каталогу модулів?
У мене така структура:

/Foo
    bar.py
    spam.py
    eggs.py

Я спробував просто перетворити його в пакет, додавши __init__.pyта зробивши, from Foo import *але він не працював так, як я сподівався.

Перерахуйте всі .pyфайли python ( ) у поточній папці та помістіть їх у __all__змінні__init__.py

from os.path import dirname, basename, isfile, join
import glob
modules = glob.glob(join(dirname(__file__), "*.py"))
__all__ = [ basename(f)[:-3] for f in modules if isfile(f) and not f.endswith('__init__.py')]

Додайте __all__змінну до __init__.pyвмісту:

__all__ = ["bar", "spam", "eggs"]

Дивіться також http://docs.python.org/tutorial/modules.html

Оновлення в 2017 році: ви, ймовірно, хочете використовувати importlibзамість цього.

Зробіть каталог Foo пакетом, додавши __init__.py. У цьому __init__.pyоних:

import bar
import eggs
import spam

Оскільки ви хочете, щоб це було динамічним (що може бути, а може і не бути гарною ідеєю), перерахуйте всі py-файли зі списком dir та імпортуйте їх приблизно так:

import os
for module in os.listdir(os.path.dirname(__file__)):
    if module == '__init__.py' or module[-3:] != '.py':
        continue
    __import__(module[:-3], locals(), globals())
del module

Потім з вашого коду зробіть це:

import Foo

Тепер ви можете отримати доступ до модулів за допомогою

Foo.bar
Foo.eggs
Foo.spam

і т.д. from Foo import *не є хорошою ідеєю з кількох причин, включаючи зіткнення імен і ускладнюючи аналіз коду.

Розкриваючи відповідь Михайла, я вважаю, що не хакітський спосіб (як, наприклад, не обробка напрямками файлів безпосередньо) такий:

1. створити порожній __init__.pyфайл підFoo/
2. Виконати

import pkgutil
import sys


def load_all_modules_from_dir(dirname):
    for importer, package_name, _ in pkgutil.iter_modules([dirname]):
        full_package_name = '%s.%s' % (dirname, package_name)
        if full_package_name not in sys.modules:
            module = importer.find_module(package_name
                        ).load_module(full_package_name)
            print module


load_all_modules_from_dir('Foo')

Ви отримаєте:

<module 'Foo.bar' from '/home/.../Foo/bar.pyc'>
<module 'Foo.spam' from '/home/.../Foo/spam.pyc'>

Python, включайте всі файли в каталозі:

Для новачків, які просто не можуть змусити його працювати, кому потрібно тримати руки.

1. Створіть папку / home / el / foo та зробіть файл main.pyпід / home / el / foo Помістіть цей код туди:

   from hellokitty import *
   spam.spamfunc()
   ham.hamfunc()

2. Складіть каталог /home/el/foo/hellokitty

3. Створіть файл __init__.pyпід /home/el/foo/hellokittyі вставте цей код туди:

   __all__ = ["spam", "ham"]

4. Зробіть два файли python: spam.pyта ham.pyпід/home/el/foo/hellokitty

5. Визначте функцію всередині spam.py:

   def spamfunc():
     print("Spammity spam")

6. Визначте функцію всередині ham.py:

   def hamfunc():
     print("Upgrade from baloney")

7. Виконати:

   el@apollo:/home/el/foo$ python main.py 
   spammity spam
   Upgrade from baloney

Я сам втомився від цієї проблеми, тому написав пакет під назвою automodinit, щоб виправити його. Ви можете отримати його з http://pypi.python.org/pypi/automodinit/ .

Використання таке:

1. Включіть automodinitпакет у свої setup.pyзалежності.
2. Замініть всі файли __init__.py таким чином:

__all__ = ["Я перепишу"]
# Не змінюйте рядок вище, або цей рядок!
імпорт автомобілей
automodinit.automodinit (__ ім'я__, __file__, глобальні ())
del automodinit
# Все, що ви хочете, може продовжити тут, воно не буде змінено.

Це воно! Відтепер імпорт модуля встановить __all__ у список .py [co] файлів у модулі, а також імпортує кожен із цих файлів так, як ніби ви ввели:

for x in __all__: import x

Тому ефект "від М імпорту *" точно відповідає "імпорту М".

automodinit із задоволенням працює з архівів ZIP, тому він безпечний.

Найл

Я знаю, що я оновлював досить стару публікацію, і я спробував використовувати automodinit, але дізнався, що процес налаштування порушено для python3. Отже, виходячи з відповіді Лука, я придумав простішу відповідь - яка може не працювати з .zip - на це питання, тому я подумав, що мені слід поділитися нею тут:

в __init__.pyмодулі від yourpackage:

#!/usr/bin/env python
import os, pkgutil
__all__ = list(module for _, module, _ in pkgutil.iter_modules([os.path.dirname(__file__)]))

і в іншому пакеті нижче yourpackage:

from yourpackage import *

Тоді ви будете завантажувати всі модулі, розміщені в пакеті, і якщо ви напишете новий модуль, він також буде імпортований автоматично. Звичайно, використовувати такі речі обережно, з великими повноваженнями постає велика відповідальність.

import pkgutil
__path__ = pkgutil.extend_path(__path__, __name__)
for imp, module, ispackage in pkgutil.walk_packages(path=__path__, prefix=__name__+'.'):
  __import__(module)

Я також стикався з цією проблемою, і це було моє рішення:

import os

def loadImports(path):
    files = os.listdir(path)
    imps = []

    for i in range(len(files)):
        name = files[i].split('.')
        if len(name) > 1:
            if name[1] == 'py' and name[0] != '__init__':
               name = name[0]
               imps.append(name)

    file = open(path+'__init__.py','w')

    toWrite = '__all__ = '+str(imps)

    file.write(toWrite)
    file.close()

Ця функція створює файл (у наданій папці) з іменем __init__.py, який містить __all__змінну, яка містить кожен модуль у папці.

Наприклад, у мене є папка з назвою, Test яка містить:

Foo.py
Bar.py

Тож у сценарії я хочу імпортувати модулі, я напишу:

loadImports('Test/')
from Test import *

Це імпортуватиме все, з Testчого __init__.pyфайл Testтепер буде містити:

__all__ = ['Foo','Bar']

Приклад Анурага з кількома виправленнями:

import os, glob

modules = glob.glob(os.path.join(os.path.dirname(__file__), "*.py"))
__all__ = [os.path.basename(f)[:-3] for f in modules if not f.endswith("__init__.py")]

Відповідь Anurag Uniyal із запропонованими поліпшеннями!

#!/usr/bin/python
# -*- encoding: utf-8 -*-

import os
import glob

all_list = list()
for f in glob.glob(os.path.dirname(__file__)+"/*.py"):
    if os.path.isfile(f) and not os.path.basename(f).startswith('_'):
        all_list.append(os.path.basename(f)[:-3])

__all__ = all_list  

Дивіться, що ваш __init__.pyвизначає __all__. Модулі - пакети док говорить

Ці __init__.pyфайли необхідні , щоб Python лікувати каталоги як містять пакети; це робиться для того, щоб каталоги із загальною назвою, наприклад, рядком, ненавмисно приховували дійсні модулі, що з’являються пізніше на шляху пошуку модулів. У найпростішому випадку це __init__.pyможе бути просто порожній файл, але він також може виконати код ініціалізації для пакета або встановити __all__змінну, описану пізніше.

...

Єдине рішення полягає в тому, щоб автор пакету надавав явний індекс пакету. У заяві про імпорт використовується наступна конвенція: якщо __init__.pyкод пакета визначає список з назвою __all__, він вважається списком імен модулів, які слід імпортувати, коли зіткнувся з імпортом пакета *. Автор цього пакета повинен постійно оновлювати цей список, коли виходить нова версія пакета. Автори пакунків можуть також вирішити не підтримувати його, якщо вони не бачать використання для імпорту * зі свого пакета. Наприклад, файл sounds/effects/__init__.pyможе містити такий код:

__all__ = ["echo", "surround", "reverse"]

Це означає, що from sound.effects import *було б імпортувати три названі підмодулі звукового пакету.

Це найкращий спосіб, який я знайшов поки що:

from os.path import dirname, join, isdir, abspath, basename
from glob import glob
pwd = dirname(__file__)
for x in glob(join(pwd, '*.py')):
    if not x.startswith('__'):
        __import__(basename(x)[:-3], globals(), locals())

Використання importlibєдиного, що ви повинні додати, - це

from importlib import import_module
from pathlib import Path

__all__ = [
    import_module(f".{f.stem}", __package__)
    for f in Path(__file__).parent.glob("*.py")
    if "__" not in f.stem
]
del import_module, Path

Подивіться на модуль pkgutil зі стандартної бібліотеки. Це дозволить робити саме те, що ви хочете, доки у вас є __init__.pyфайл у каталозі. __init__.pyФайл може бути порожнім.

Я створив для цього модуль, який не покладається на __init__.py(або будь-який інший допоміжний файл) і змушує мене вводити лише наступні два рядки:

import importdir
importdir.do("Foo", globals())

Не соромтесь повторно використовувати або сприяти: http://gitlab.com/aurelien-lourot/importdir

Просто імпортуйте їх importlib та додайте їх до __all__( addдія необов'язково) повторно в __init__.pyпакеті.

/Foo
    bar.py
    spam.py
    eggs.py
    __init__.py

# __init__.py
import os
import importlib
pyfile_extes = ['py', ]
__all__ = [importlib.import_module('.%s' % filename, __package__) for filename in [os.path.splitext(i)[0] for i in os.listdir(os.path.dirname(__file__)) if os.path.splitext(i)[1] in pyfile_extes] if not filename.startswith('__')]
del os, importlib, pyfile_extes

Коли from . import *недостатньо добре, це є поліпшенням у відповідь Теда . Зокрема, використання __all__цього підходу не є необхідним.

"""Import all modules that exist in the current directory."""
# Ref https://stackoverflow.com/a/60861023/
from importlib import import_module
from pathlib import Path

for f in Path(__file__).parent.glob("*.py"):
    module_name = f.stem
    if (not module_name.startswith("_")) and (module_name not in globals()):
        import_module(f".{module_name}", __package__)
    del f, module_name
del import_module, Path

Зауважте, що module_name not in globals()призначено уникати повторного імпорту модуля, якщо він вже імпортований, оскільки це може загрожувати циклічним імпортом.