Question

我使用python 2.7。我需要加载目录中的所有模块。一些模块存储在目录中，一些存储在子目录中。结构

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

每个工具都包含带有Run方法的类。我可以使用import加载工具，然后查看 dict .keys（）它看起来像：

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

Tool1 - 是类名。每个类都有__VERSION属性。如何从目录加载所有模块并从中提取类并打印所有模块__VERSTION属性？

其他代码信息： 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

MyLoad工具：

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

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

1 个答案: