我可以一次从一个文件夹中导入所有 Python 文件吗？

Question

我想知道是否可以从文件夹中导入所有 Python 文件，还是需要单独导入每个文件？ 我建立了一个项目，其中有一个文件夹“website_functions”，其中包含 20 个 Python 文件。 其中每个包含一个function（我使用Selenium访问不同的网站，每个文件对应一个网站）。 不必单独导入它们会使其更快。

提前感谢您的回答

Answer 1

我会说这是一个非常糟糕的主意，它违反了 pep8 样式指南

但如果你真的必须这样做，你可以像这样破解它：

import importlib
from glob import glob

# Get list of *.py files
modules = glob('*.py')
# Get rid of *.py extension
modules = [module[:-3] for module in modules]

for module in modules:
    importlib.import_module(module)

Answer 2

是的，这是可能的。 在包含您要导入的文件的文件夹中，在__init__.py文件中名为__all__的列表中指定所有文件名。 从您希望将文件导入到的程序中，您现在可以使用from directory import *导入它们。

考虑以下示例。

项目目录结构：

.
|-- main.py
`-- modules
    |-- __init__.py
    |-- f.py
    `-- g.py

1 directory, 4 files

主文件

from modules import *

f.f1()
f.f2()
g.g1()
g.g2()

模块/f.py

def f1():

    print("This is f1() from modules/f.py.")

def f2():

    print("This is f2() from modules/f.py.")

模块/g.py

def g1():

    print("This is g1() from modules/g.py.")

def g2():

    print("This is g2() from modules/g.py.")

模块/__init__.py

__all__ = ["f", "g"]

main.py的output如下。

This is f1() from modules/f.py.
This is f2() from modules/f.py.
This is g1() from modules/g.py.
This is g2() from modules/g.py.

如果您希望每次运行程序时自动生成__init__.py中的__all__列表，您可以使用os库来实现。 例如：

（我已经从这个例子中删除了 f.py 以免干扰modules的f属性。）

模块/__init__.py

import os

__all__ = []
dirname = os.path.dirname(os.path.abspath(__file__))

for f in os.listdir(dirname):
    if f != "__init__.py" and os.path.isfile("%s/%s" % (dirname, f)) and f[-3:] == ".py":
        __all__.append(f[:-3])

主文件

from modules import *

g.g1()
g.g2()

main.py的output如下。

This is g1() from modules/g.py.
This is g2() from modules/g.py.