为文件夹中的每个脚本调用一个函数

Question

有没有办法（仅使用 python。即：没有 bash 脚本或其他语言代码）在文件夹内的每个脚本中调用特定函数，而无需显式导入所有函数。

例如，假设这是我的结构：

main.py
modules/
    module1.py
    module2.py
    module3.py
    module4.py

每个moduleX.py 都有这个代码：

import os

def generic_function(caller):
    print('{} was called by {}'.format(os.path.basename(__file__), caller))

def internal_function():
    print('ERROR: Someone called an internal function')

而main.py 有这个代码：

import modules
import os

for module in modules.some_magic_function():
    module.generic_function(os.path.basename(__file__))

所以如果我运行main.py，我应该得到这个输出：

module1.py was called by main.py
module2.py was called by main.py
module3.py was called by main.py
module4.py was called by main.py

*请注意internal_function()不应被调用（不像question）。另外，我不想显式声明每个模块文件，即使在 __init__.py

顺便说一句，我不介意为此使用类。事实上，它可能会更好。

Answer 1

虽然 sophros 的方法可以快速且足以隐式导入模块，但您可能会遇到与控制每个模块或复杂调用相关的问题（例如为每个调用设置条件）。所以我采用了另一种方法：

首先，我创建了一个声明了函数（现在是方法）的类。有了这个，我可以避免检查该方法是否存在，因为如果我没有声明它，我可以使用默认方法：

# main.py
class BaseModule:
    def __init__(self):
        # Any code
    
    def generic_function(self, caller):
        # This could be a Print (or default return value) or an Exception
        raise Exception('generic_function wasn\'t overridden or it was used with super')
    

然后我创建了另一个扩展 BaseModule 的类。遗憾的是，在不知道子类名称的情况下，我无法找到检查继承的好方法，因此我为每个模块使用了相同的名称：

# modules/moduleX.py
from main import BaseModule

class GenericModule(BaseModule):
    def __init__(self):
        BaseModule.__init__(self)
        # Any code
    
    def generic_function(self, caller):
        print('{} was called by {}'.format(os.path.basename(__file__), caller))

最后，在我的main.py 中，我使用importlib 动态导入模块并为每个模块保存一个实例，以便以后使用它们（为简单起见，我没有将它们保存在下面代码，但它很容易使用列表并在其上附加每个实例）：

# main.py
import importlib
import os

if __name__ == '__main__':
    relPath = 'modules' # This has to be relative to the working directory

    for pyFile in os.listdir('./' + relPath):
        # just load python (.py) files except for __init__.py or similars
        if pyFile.endswith('.py') and not pyFile.startswith('__'):
            # each module has to be loaded with dots instead of slashes in the path and without the extension. Also, modules folder must have a __init___.py file
            module = importlib.import_module('{}.{}'.format(relPath, pyFile[:-3]))
            # we have to test if there is actually a class defined in the module. This was extracted from [1]
            try:
                moduleInstance = module.GenericModule(self)
                moduleInstance.generic_function(os.path.basename(__file__)) # You can actually do whatever you want here. You can save the moduleInstance in a list and call the function (method) later, or save its return value.
            except (AttributeError) as e:
                # NOTE: This will be fired if there is ANY AttributeError exception, including those that are related to a typo, so you should print or raise something here for diagnosting
                print('WARN:', pyFile, 'doesn\'t has GenericModule class or there was a typo in its content')

参考资料：

[1]Check for class existence

[2]Import module dynamically

[3]Method Overriding in Python

Answer 2

您可以使用exec 或eval 来执行此操作。所以它会大致这样（exec）：

def magic_execute():
    import os
    import glob
    for pyfl in glob.glob(os.path(MYPATH, '*.py'):
        with open(pyfl, 'rt') as fh:
            pycode = fh.read()
            pycode += '\ngeneric_function({})'.format(__file__)
            exec(pycode)

这里的假设是您根本不打算导入模块。

请注意，有numerous security issues 与以这种不受限制的方式使用exec 有关。你可以increase security a bit。