如何加载目录中的所有模块并从模块中获取类

Question

我使用的是 Python 2.7。
我需要加载一个目录下的所有模块。有些模块存放在主目录里，有些则在子目录中。
目录结构如下：

dir
   |
   Tool1
   |    |
   |   __init.pyc__
   |   tool1.pyc
   Tool2
   |    |
   |   __init.pyc__
   |   tool2.pyc
   tool3.pyc
   tool4.pyc

每个工具都包含一个带有 Run 方法的类。我可以通过导入工具来加载它，然后查看 dict.keys()，结果看起来像这样：

['__all__', '__builtins__', 'glob', '__file__', 'traceback', 'GlobalTool', 'run_me', '__package__', 'sys', '__path__', 'Tool1', 'time', '__name__', 'csv', 'os', '__doc__']

Tool1 - 是类的名称。
每个类都有一个属性 __VERSION。
我该如何从目录中加载所有模块，并提取出它们的类，然后打印出所有模块的 __VERSION 属性呢？

附加代码信息：
Tool1.py：

import sys, os, csv
from time import time
import traceback
from core.global_tool import GlobalTool

class ParseFiXEDVALUES(GlobalTool):
    _tool_name = 'FIED_VALUES'
    _VERSION = '0.1'

    def run(self):
        """Main function"""
        start_time = time()
        cur = self.db.cursor()
        pass

我的加载工具：

import os
import sys
mods_dir = '/home/test/multitool'
mods = {}
sys.path.append(mods_dir)

for module in os.listdir(mods_dir):
 if '.py' in module and '.pyc' not in module:
     current = module.replace('.py', '')
     modules[current] = __import__(current)

目前它显示所有的 pyc 或 py 文件，但需要处理目录和类的相关操作。

Answer 1

你需要做的就是检查一下这个 module 是否是一个文件夹，并且里面有一个叫 __init__.py 的文件：

import os
import sys
import importlib
import inspect
import logging

def load_module(module_path, filename):
    """ returns the module if filename is a module else None """
    if filename.endswith('.py'):
        module = filename[:-3]
    elif os.path.exists(os.path.join(module_path, filename, '__init__.py')):
        module = filename
    else:
        return None
    try:
        return importlib.import_module(module)
    except:
        logging.exception('Loading %s failed.' % module)
        return None

class PluginManager(object):
    def __init__(self):
        self.modules = {}
        self.classes = {}

    def add_path(self, module_path):
        sys.path.append(module_path)
        for filename in os.listdir(module_path):
            module = load_module(module_path, filename)
            if module:
                self.modules[module.__name__] = module
                self._extract_classes(module)
        sys.path.remove(module_path)

    def _extract_classes(self, module):
        for name in dir(module):
            obj = getattr(module, name)
            if inspect.isclass(obj):
                if hasattr(obj, '_VERSION'):
                    version = getattr(obj, '_VERSION')
                    logging.info("Found %s.%s %s" % (module.__name__, name, version))
                    self.classes[name] = obj

logging.getLogger().level = logging.INFO
plugins = PluginManager()
plugins.add_path('/home/test/multitool')