Small generator function that recursively scans a directory for all python modules.

scan_for_modules.py

def scan_for_modules(path):
    """
    Generator function which scans a directory for all python modules and returns them successively.
    """
    modules = []
    for finder, pkgname, is_module in pkgutil.iter_modules([path]):
        if pkgname in ['setup']:
            continue

        mod = importlib.import_module(pkgname)
        if not hasattr(mod, '__path__'):
            continue

        yield mod
        modules.append(mod)

    while modules:
        mod = modules.pop()
        for finder, pkgname, is_module in pkgutil.iter_modules(mod.__path__):
            child_mod = importlib.import_module('%s.%s' % (mod.__name__, pkgname))

            yield child_mod
            if is_module and 'test' != pkgname:
                modules.append(child_mod)