importer.py

import inspect
import logging
from os import listdir
from os.path import isfile, join, abspath, dirname, sep
import sys
from traceback import format_exception


def moduleQualifies(dir, file, excludes, includes):
    return isfile(join(dir, file))                  \
        and file.endswith(".py")                    \
        and file not in excludes                    \
        and file[0] != "_"                          \
        and (file in includes or len(includes) == 0)


def getPluginList(plugindir, excludes, includes):
    basedir = dirname(abspath(inspect.getsourcefile(lambda _: None)))
    dir = basedir + sep + plugindir
    return [f[:-3]
            for f in listdir(dir)
            if moduleQualifies(dir, f, excludes, includes)]


def ImportPlugins(dir, excludes, includes, progress, matrix, config):
    output = {}
    excludes.append("__init__.py")

    plugins = getPluginList(dir, excludes, includes)
    total = len(plugins)
    for index, plugin in enumerate(plugins):
        try:
            module = __import__(
                dir+'.'+plugin,
                globals(),
                locals(),
                ["Art"],
                0)
            obj = module.Art(matrix, config)

            desc = getattr(obj, "description", None)
            if desc is not None:
                logging.info("%s: %s" % (plugin, desc))
            else:
                logging.info("%s: [None]" % (plugin))

            if progress is not None:
                progress(index, total)

            output[plugin] = obj
        except Exception:  # yes, we want to catch everything
            logging.error("%s: import failed, details follow: " % (plugin))

            etype, evalue, etraceback = sys.exc_info()
            for line in format_exception(etype, evalue, etraceback):
                logging.error('    Exception: '+line.rstrip('\n'))

    return output