NPO-Accounting/conservancy_beancount
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

plugin: Load all existing hooks to registry.

Browse source
This commit is contained in:
Brett Smith 2020-03-31 14:52:38 -04:00
parent 5566672cd6
commit fd2830f483
1 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
conservancy_beancount/plugin/__init__.py
View file

@ -63,6 +63,9 @@ class HookRegistry:
*hook_names: str, *hook_names: str,
package: Optional[str]=__module__, # type:ignore[name-defined] package: Optional[str]=__module__, # type:ignore[name-defined]
) -> None: ) -> None:
if not hook_names:
_, _, hook_name = mod_name.rpartition('.')
hook_names = (hook_name.title().replace('_', ''),)
module = importlib.import_module(mod_name, package) module = importlib.import_module(mod_name, package)
for hook_name in hook_names: for hook_name in hook_names:
self.add_hook(getattr(module, hook_name)) self.add_hook(getattr(module, hook_name))
@ -94,8 +97,17 @@ class HookRegistry:
HOOK_REGISTRY = HookRegistry() HOOK_REGISTRY = HookRegistry()
HOOK_REGISTRY.import_hooks('.meta_expense_allocation', 'MetaExpenseAllocation') HOOK_REGISTRY.import_hooks('.meta_approval')
HOOK_REGISTRY.import_hooks('.meta_tax_implication', 'MetaTaxImplication') HOOK_REGISTRY.import_hooks('.meta_entity')
HOOK_REGISTRY.import_hooks('.meta_expense_allocation')
HOOK_REGISTRY.import_hooks('.meta_income_type')
HOOK_REGISTRY.import_hooks('.meta_invoice')
HOOK_REGISTRY.import_hooks('.meta_project')
HOOK_REGISTRY.import_hooks('.meta_receipt')
HOOK_REGISTRY.import_hooks('.meta_receivable_documentation')
HOOK_REGISTRY.import_hooks('.meta_repo_links')
HOOK_REGISTRY.import_hooks('.meta_rt_links', 'MetaRTLinks')
HOOK_REGISTRY.import_hooks('.meta_tax_implication')
def run( def run(
entries: List[Directive], entries: List[Directive],