forked from gaugendre/ofx-processor
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
import importlib
|
||
import pkgutil
|
||
|
||
import click
|
||
|
||
import ofx_processor
|
||
|
||
|
||
def discover_processors(cli: click.Group):
|
||
"""
|
||
Discover processors. To be discovered, CLIs need to match the following structure:
|
||
|
||
ofx_processor
|
||
├── __init__.py
|
||
└── <package_name>_processor
|
||
├── __init__.py
|
||
├── <module_name>_cli.py
|
||
└── <module_name>_processor.py
|
||
|
||
Preferably, package_name and module_name are the same, but there is no restriction.
|
||
Finally, a "main" function will be looked for inside the <module_name>_cli module.
|
||
This function must be a Click command with a unique name and will be added to the
|
||
main command line interface.
|
||
|
||
:param cli: The main CLI to add discovered processors to.
|
||
"""
|
||
prefix = ofx_processor.__name__ + "."
|
||
for package in pkgutil.iter_modules(ofx_processor.__path__, prefix):
|
||
if package.name.endswith("_processor"):
|
||
module_prefix = package.name + "."
|
||
package = importlib.import_module(package.name)
|
||
for module in pkgutil.iter_modules(package.__path__, module_prefix):
|
||
if module.name.endswith("_cli"):
|
||
module = importlib.import_module(module.name)
|
||
if "main" in dir(module):
|
||
cli.add_command(module.main)
|