Question

My python project has the following file structure:

/main.py
/functions
/functions/func1.py
/functions/func2.py
/functions/func3.py
/funcitons/__init__.py

Every func.py file has a variable ‘CAN_USE’. In some files it’s true in other false.
How can I check inside of my main.py which func.py files has ‘CAN_USE’ variable equals true?

Answer 1

Using pkgutil you can find all modules in a package:

import pkgutil

def usable_modules(package_name):
    modules = pkgutil.iter_modules([package_name])
    usable = []
    for importer, name, ispkg in modules:
        module = pkgutil.find_loader('{0}.{1}'.format(package_name, name)).\
                                                             load_module(name)
        if hasattr(module, 'CAN_USE') and module.CAN_USE:
            usable.append(module)
    return usable

print(usable_modules('functions'))

Note that this also checks other modules in your package (e.g. __init__.py). You can filter them out in your loop if you want to (e.g. if not name.startswith('func'): continue).