Làm thế nào để tải tất cả các mô-đun trong một thư mục?

Ai đó có thể cung cấp cho tôi một cách tốt để nhập toàn bộ thư mục mô-đun không?
Tôi có một cấu trúc như thế này:

/Foo
    bar.py
    spam.py
    eggs.py

Tôi đã cố gắng chuyển đổi nó thành một gói bằng cách thêm __init__.pyvà thực hiện from Foo import *nhưng nó không hoạt động như tôi mong đợi.

Liệt kê tất cả .pycác tệp python ( ) trong thư mục hiện tại và đặt chúng dưới dạng __all__biến trong__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')]

Thêm __all__biến để __init__.pychứa:

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

Xem thêm http://docs.python.org/tutorial/modules.html

Cập nhật năm 2017: có lẽ bạn muốn sử dụng importlibthay thế.

Làm cho thư mục Foo một gói bằng cách thêm một __init__.py. Trong đó __init__.pythêm:

import bar
import eggs
import spam

Vì bạn muốn nó động (có thể là một ý tưởng hay), hãy liệt kê tất cả các tệp py với danh sách dir và nhập chúng với nội dung như sau:

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

Sau đó, từ mã của bạn làm điều này:

import Foo

Bây giờ bạn có thể truy cập các mô-đun với

Foo.bar
Foo.eggs
Foo.spam

vv from Foo import *không phải là một ý tưởng tốt vì một số lý do, bao gồm xung đột tên và làm cho nó khó để phân tích mã.

Mở rộng câu trả lời của Mihail, tôi tin rằng cách không hack (như trong, không xử lý trực tiếp đường dẫn tệp) là như sau:

1. tạo một __init__.pytập tin trống dướiFoo/
2. Hành hình

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

Bạn sẽ nhận được:

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

Python, bao gồm tất cả các tệp trong một thư mục:

Đối với những người mới, những người không thể làm cho nó hoạt động, những người cần nắm tay họ.

1. Tạo một thư mục / home / el / foo và tạo một tập tin main.pydưới / home / el / foo Đặt mã này vào đó:

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

2. Tạo một thư mục /home/el/foo/hellokitty

3. Tạo một tập tin __init__.pybên dưới /home/el/foo/hellokittyvà đặt mã này vào đó:

   __all__ = ["spam", "ham"]

4. Tạo hai tệp python: spam.pyvà ham.pydưới/home/el/foo/hellokitty

5. Xác định một chức năng bên trong spam.py:

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

6. Xác định một chức năng bên trong ham.py:

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

7. Chạy nó:

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

Tôi cảm thấy mệt mỏi với vấn đề này, vì vậy tôi đã viết một gói có tên là automodinit để khắc phục nó. Bạn có thể lấy nó từ http://pypi.python.org/pypi/automodinit/ .

Cách sử dụng là như thế này:

1. Bao gồm các automodinitgói vào setup.pyphụ thuộc của bạn .
2. Thay thế tất cả các tệp __init__.py như thế này:

__all__ = ["Tôi sẽ được viết lại"]
# Đừng sửa đổi dòng trên hoặc dòng này!
nhập khẩu tự động
automodinit.automodinit (__ name__, __file__, globalals ())
del automodinit
# Bất cứ điều gì khác bạn muốn có thể đi sau đây, nó sẽ không được sửa đổi.

Đó là nó! Từ giờ, việc nhập mô-đun sẽ đặt __all__ thành danh sách các tệp .py [co] trong mô-đun và cũng sẽ nhập từng tệp đó như thể bạn đã nhập:

for x in __all__: import x

Do đó, hiệu ứng của "từ M nhập *" khớp chính xác với "nhập M".

automodinit rất vui khi chạy từ bên trong kho lưu trữ ZIP và do đó ZIP an toàn.

Không

Tôi biết tôi đang cập nhật một bài viết khá cũ và tôi đã thử sử dụng automodinit , nhưng phát hiện ra quá trình thiết lập của nó bị hỏng đối với python3. Vì vậy, dựa trên câu trả lời của Luca, tôi đã đưa ra một câu trả lời đơn giản hơn - có thể không hoạt động với .zip - cho vấn đề này, vì vậy tôi nghĩ rằng tôi nên chia sẻ nó ở đây:

trong __init__.pymô-đun từ yourpackage:

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

và trong một gói khác dưới đây yourpackage:

from yourpackage import *

Sau đó, bạn sẽ có tất cả các mô-đun được đặt trong gói được tải và nếu bạn viết một mô-đun mới, nó cũng sẽ được nhập tự động. Tất nhiên, sử dụng những thứ đó một cách cẩn thận, với sức mạnh lớn đi kèm với trách nhiệm lớn.

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

Tôi cũng đã gặp phải vấn đề này và đây là giải pháp của tôi:

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

Hàm này tạo một tệp (trong thư mục được cung cấp) có tên __init__.py, chứa một __all__biến chứa mọi mô-đun trong thư mục.

Ví dụ: tôi có một thư mục có tên Test :

Foo.py
Bar.py

Vì vậy, trong kịch bản tôi muốn các mô-đun được nhập vào tôi sẽ viết:

loadImports('Test/')
from Test import *

Điều này sẽ nhập mọi thứ từ Testvà __init__.pytệp trong đó Testsẽ chứa:

__all__ = ['Foo','Bar']

Ví dụ của Anurag với một vài sửa chữa:

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 trả lời với những cải tiến được đề xuất!

#!/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  

Xem __init__.pyđịnh nghĩa của bạn __all__. Các mô-đun - gói doc nói

Các __init__.pytệp được yêu cầu để làm cho Python coi các thư mục là chứa các gói; điều này được thực hiện để ngăn các thư mục có tên chung, chẳng hạn như chuỗi, vô tình ẩn các mô-đun hợp lệ xảy ra sau này trên đường dẫn tìm kiếm mô-đun. Trong trường hợp đơn giản nhất, __init__.pycó thể chỉ là một tệp trống, nhưng nó cũng có thể thực thi mã khởi tạo cho gói hoặc đặt __all__biến, được mô tả sau.

...

Giải pháp duy nhất là để tác giả gói cung cấp một chỉ mục rõ ràng của gói. Câu lệnh nhập sử dụng quy ước sau: nếu __init__.pymã của gói xác định danh sách có tên __all__, thì nó được coi là danh sách tên mô-đun nên được nhập khi gặp phải nhập gói *. Tùy thuộc vào tác giả gói để giữ cho danh sách này được cập nhật khi một phiên bản mới của gói được phát hành. Các tác giả gói cũng có thể quyết định không hỗ trợ nó, nếu họ không thấy việc sử dụng để nhập * từ gói của họ. Ví dụ: tệp sounds/effects/__init__.pycó thể chứa mã sau:

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

Điều này có nghĩa là from sound.effects import *sẽ nhập ba mô hình con có tên của gói âm thanh.

Đây là cách tốt nhất mà tôi đã tìm thấy cho đến nay:

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

Sử dụng importlibthứ duy nhất bạn phải thêm là

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

Nhìn vào mô-đun pkgutil từ thư viện tiêu chuẩn. Nó sẽ cho phép bạn làm chính xác những gì bạn muốn miễn là bạn có một __init__.pytập tin trong thư mục. Các __init__.pytập tin có thể để trống.

Tôi đã tạo một mô-đun cho điều đó, không dựa vào __init__.py(hoặc bất kỳ tệp phụ trợ nào khác) và khiến tôi chỉ gõ hai dòng sau:

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

Vui lòng sử dụng lại hoặc đóng góp: http://gitlab.com/aurelien-lourot/importdir

Chỉ cần nhập chúng bằng cách importlib và thêm chúng vào __all__( addhành động này là không bắt buộc) trong recurse trong __init__.pycác gói.

/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

Khi from . import *không đủ tốt, đây là một cải tiến so với câu trả lời của ted . Cụ thể, việc sử dụng __all__là không cần thiết với phương pháp này.

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

Lưu ý rằng module_name not in globals()nhằm tránh nhập lại mô-đun nếu nó đã được nhập, vì điều này có thể gây rủi ro cho việc nhập theo chu kỳ.