From fd2830f48377ea27c3c216eece5528472d7cf85b Mon Sep 17 00:00:00 2001 From: Brett Smith Date: Tue, 31 Mar 2020 14:52:38 -0400 Subject: [PATCH] plugin: Load all existing hooks to registry. --- conservancy_beancount/plugin/__init__.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/conservancy_beancount/plugin/__init__.py b/conservancy_beancount/plugin/__init__.py index 41c7daa..9490b8f 100644 --- a/conservancy_beancount/plugin/__init__.py +++ b/conservancy_beancount/plugin/__init__.py @@ -63,6 +63,9 @@ class HookRegistry: *hook_names: str, package: Optional[str]=__module__, # type:ignore[name-defined] ) -> None: + if not hook_names: + _, _, hook_name = mod_name.rpartition('.') + hook_names = (hook_name.title().replace('_', ''),) module = importlib.import_module(mod_name, package) for hook_name in hook_names: self.add_hook(getattr(module, hook_name)) @@ -94,8 +97,17 @@ class HookRegistry: HOOK_REGISTRY = HookRegistry() -HOOK_REGISTRY.import_hooks('.meta_expense_allocation', 'MetaExpenseAllocation') -HOOK_REGISTRY.import_hooks('.meta_tax_implication', 'MetaTaxImplication') +HOOK_REGISTRY.import_hooks('.meta_approval') +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( entries: List[Directive],